File Coverage

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

Criterion	Covered	Total	%
statement	30	30	100.0
branch	3	6	50.0
condition	2	6	33.3
subroutine	8	8	100.0
pod	1	1	100.0
total	44	51	86.2

package Tree::Visualize::GraphViz::Layouts::Simple::Tree;

use strict;
use warnings;

use Tree::Visualize::Exceptions;

our $VERSION = '0.01';

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

sub draw {
    my ($self, $tree) = @_;
    (defined($tree) && ref($tree) && UNIVERSAL::isa($tree, "Tree::Simple"))
            || throw Tree::Visualize::IncorrectObjectType "argument must be Tree::Simple instance";
    return $self->_draw($tree);
}

sub _draw {
    my ($self, $tree) = @_;
    my $output = "digraph test {\n";
    $output .= "node_" . $tree->getUID() . " [ label = \"" . $tree->getNodeValue() . "\" ];\n";
    $output .= $self->_draw_tree($tree);
    $output .= "}\n";
    return $output;
}

sub _draw_tree {
    my ($self, $tree) = @_;
    my $output = "";
    $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;
}



1;

__END__