File Coverage

blib/lib/Tree/Visualize.pm

Criterion	Covered	Total	%
statement	35	36	97.2
branch	3	6	50.0
condition	5	14	35.7
subroutine	9	9	100.0
pod	2	2	100.0
total	54	67	80.6

package Tree::Visualize;

use strict;
use warnings;

our $VERSION = '0.01';

use Tree::Visualize::Exceptions;
use Tree::Visualize::Config;
use Tree::Visualize::Layout::Factory;

sub new {
    my ($_class, $tree, $output_format, $layout) = @_;
    my $class = ref($_class) || $_class;
    my $visulization = {};
    bless($visulization, $class);
    $visulization->_init($tree, $output_format, $layout);
    return $visulization;
}

sub _init {
    my ($self, $tree, $output_format, $layout) = @_;
    (defined($tree) && ref($tree) && defined($output_format))
        || throw Tree::Visualize::InsufficientArguments "bad arguments for Tree::Visualize->new";

    $self->{tree} = $tree;
    $self->{tree_type} = $Tree::Visualize::Config::TREE_TYPES{ref($tree)};
    $self->{output_format} = $output_format;
    $self->{layout} = $layout;
}

sub draw {
    my ($self) = @_;
    my $output;
    eval {
        $output = $self->_getVisualizationInstance()->draw($self->{tree});
    };
    throw Tree::Visualize::Exception ("Visualization for tree (" . $self->{tree} . ") failed", $@) if $@;
    return $output->getAsString() if ref($output) && UNIVERSAL::isa($output, 'Tree::Visualize::ASCII::BoundingBox');
    return $output;
}


sub _getVisualizationInstance {
    my ($self) = @_;
    return Tree::Visualize::Layout::Factory->new()->get(
                            output => $self->{output_format},
                            tree_type => $self->{tree_type},
                            layout => $self->{layout} || "TopDown"
                            );
}

1;

__END__