File Coverage

blib/lib/Tree/Visualize/ASCII/Layouts/Binary/Diagonal.pm

Criterion	Covered	Total	%
statement	42	42	100.0
branch	10	10	100.0
condition			n/a
subroutine	10	10	100.0
pod	3	3	100.0
total	65	65	100.0

package Tree::Visualize::ASCII::Layouts::Binary::Diagonal;

use strict;
use warnings;

use Tree::Visualize::Exceptions;
use Tree::Visualize::ASCII::BoundingBox;

use Tree::Visualize::Node::Factory;
use Tree::Visualize::Connector::Factory;

our $VERSION = '0.01';

use base qw(Tree::Visualize::ASCII::Layouts::Binary);

sub drawNode {
    my ($self, $tree) = @_;
    my $node = Tree::Visualize::Node::Factory->new()->get(output => 'ASCII', node_type => 'Parens', args => [ $tree ]);
    return Tree::Visualize::ASCII::BoundingBox->new($node->draw());
}

sub drawConnections {
    my ($self, $current, $left, $right) = @_;
    # prepare the connections
    my ($left_connection, $right_connection) = (undef, undef);
    my $conn_factory = Tree::Visualize::Connector::Factory->new();
    # get the left connection if we need it
    $left_connection = $conn_factory->get(
                                output => 'ASCII',
                                layout => 'Diagonal',
                                connector_type => 'LeftRightConnector'
                                )->drawLeftConnector($current, $left, $right)
                                    if defined $left;

    # get the right connection if we need it
    $right_connection = $conn_factory->get(
                                output => 'ASCII',
                                layout => 'Diagonal',
                                connector_type => 'LeftRightConnector'
                                )->drawRightConnector($current, $right, $left)
                                    if defined $right;
    # return our connections
    return ($left_connection, $right_connection);
}

sub assembleDrawing {
    my ($self, $current, $children, $connections) = @_;

    my ($left, $right) = @{$children};
    my ($left_connection, $right_connection) = @{$connections};

    if (defined($right)) {
        $current->pasteRight($right_connection);
    }
    if (defined($left)) {
        $current->pasteBottom($left_connection)
                ->pasteBottom($left);
    }
    if (defined($right)) {
        $current->pasteRight($right);
    }

    return $current;
}

1;

__END__