File Manager

Editing File: /home/ubuntu/code_review/arcanist/src/difference/ArcanistDiffVectorTree.php

<?php

final class ArcanistDiffVectorTree
  extends Phobject {

private $vectors = array();

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;
  }

}

Back to Directory File Manager