File Coverage

blib/lib/Tree/Visualize/GraphViz/Layouts/Binary/Tree.pm

Criterion	Covered	Total	%
statement	25	25	100.0
branch	6	8	75.0
condition	3	9	33.3
subroutine	7	7	100.0
pod	1	1	100.0
total	42	50	84.0

package Tree::Visualize::GraphViz::Layouts::Binary::Tree;

use strict;
use warnings;

use Tree::Visualize::Exceptions;

our $VERSION = '0.01';

use base qw(Tree::Visualize::Layout::ILayout);

sub draw {
    my ($self, $binary_tree) = @_;
    (defined($binary_tree) && ref($binary_tree) &&
        (UNIVERSAL::isa($binary_tree, "Tree::Binary") || UNIVERSAL::isa($binary_tree, "Tree::Binary::Search")))
            || throw Tree::Visualize::IncorrectObjectType "argument must be Tree::Binary or Tree::Binary::Search instance";
    # if its a binary search tree,
    # we need to extract the binary
    # tree in it
    $binary_tree = $binary_tree->getTree() if $binary_tree->isa("Tree::Binary::Search");
    # call our private method here
    $self->_draw($binary_tree);
}

sub _draw {
    my ($self, $tree) = @_;
    my $output = "digraph test {\n";
    $tree->traverse(sub {
        my ($tree) = @_;
        my $tree_id = $tree->getUID();
        my $parent_id = $tree->getParent()->getUID() unless $tree->isRoot();
        $output .= "node_$tree_id [ label = \"" . $tree->getNodeValue() . "\" ];\n";
        $output .= "node_${parent_id} -> node_${tree_id};\n" if $parent_id;
    });
    return $output . "}\n";
}

1;

__END__