MBHFramework / structures

use Mbh\Tree\Interfaces\Node as NodeInterface;

use Mbh\Tree\Visitor;

class InOrder extends Visitor

{

    public function visit(NodeInterface $node)

    {

        if ($node->isLeaf()) {

            return [$node];

        }

        $yield = [];

        foreach ($node->getChildren() as $child) {

            $yield = array_merge($yield, $child->accept($this));

        }

        return $yield;

    }

}

use Mbh\Tree\Interfaces\Node as NodeInterface;

use Mbh\Tree\Visitor;

class PostOrder extends Visitor

{

    public function visit(NodeInterface $node)

    {

        $nodes = [];

        foreach ($node->getChildren() as $child) {

            $nodes = array_merge(

                $nodes,

                $child->accept($this)

            );

        }

        $nodes[] = $node;

        return $nodes;

    }

}

use Mbh\Tree\Interfaces\Node as NodeInterface;

use Mbh\Tree\Visitor;

class PreOrder extends Visitor

{

    public function visit(NodeInterface $node)

    {

        $nodes = [$node];

        foreach ($node->getChildren() as $child) {

            $nodes = array_merge(

                $nodes,

                $child->accept($this)

            );

        }

        return $nodes;

    }

}