Differential D21182 Diff 50444 src/difference/ArcanistDiffVectorTree.php

Changeset View

Standalone View

src/difference/ArcanistDiffVectorTree.php

This file was added.

				<?php

				final class ArcanistDiffVectorTree
				extends Phobject {

				private $vectors;

				public function addVector(array $vector) {
				$this->vectors[] = $vector;
				return $this;
				}

				public function newDisplayList() {
				$root = new ArcanistDiffVectorNode();

				foreach ($this->vectors as $vector) {
				$root->addChild($vector, count($vector), 0);
				}

				foreach ($root->getChildren() as $child) {
				$this->compressTree($child);
				}

				$root->setDisplayDepth(-1);
				foreach ($root->getChildren() as $child) {
				$this->updateDisplayDepth($child);
				}

				return $this->getDisplayList($root);
				}

				private function compressTree(ArcanistDiffVectorNode $node) {
				$display_node = $node;

				$children = $node->getChildren();
				if ($children) {
				$parent = $node->getParentNode();
				if ($parent) {
				$siblings = $parent->getChildren();
				if (count($siblings) === 1) {
				if (!$parent->getValueNode()) {
				$parent_display = $parent->getDisplayNode();
				if ($parent_display) {
				$display_node = $parent_display;
				if ($node->getValueNode()) {
				$parent->setValueNode($node->getValueNode());
				}
				}
				}
				}
				}
				}

				$node->setDisplayNode($display_node);

				$display_element = last($node->getVector());
				$display_node->appendDisplayElement($display_element);

				foreach ($children as $child) {
				$this->compressTree($child);
				}
				}

				private function updateDisplayDepth(ArcanistDiffVectorNode $node) {
				$parent_depth = $node->getParentNode()->getDisplayDepth();

				if ($node->getDisplayVector() === null) {
				$display_depth = $parent_depth;
				} else {
				$display_depth = $parent_depth + 1;
				}

				$node->setDisplayDepth($display_depth);

				foreach ($node->getChildren() as $child) {
				$this->updateDisplayDepth($child);
				}
				}

				private function getDisplayList(ArcanistDiffVectorNode $node) {
				$result = array();

				foreach ($node->getChildren() as $child) {
				if ($child->getDisplayVector() !== null) {
				$result[] = $child;
				}
				foreach ($this->getDisplayList($child) as $item) {
				$result[] = $item;
				}
				}

				return $result;
				}

				}