File Coverage

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

Criterion	Covered	Total	%
statement	37	39	94.8
branch	10	16	62.5
condition	2	3	66.6
subroutine	7	7	100.0
pod	2	2	100.0
total	58	67	86.5

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

use strict;
use warnings;

our $VERSION = '0.01';

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

use Tree::Visualize::Exceptions;
use Tree::Visualize::Connector::Factory;

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 => 'TopDown',
                                connector_type => 'LeftRightConnector'
                                )->drawLeftConnector($current, $left)
                                    if defined $left;
    # get the right connection if we need it
    $right_connection = $conn_factory->get(
                                output => 'ASCII',
                                layout => 'TopDown',
                                connector_type => 'LeftRightConnector'
                                )->drawRightConnector($current, $right)
                                    if defined $right;
    # return our connections
    return ($left_connection, $right_connection);
}

sub assembleDrawing {
    my ($self, $current, $children, $connections) = @_;
    # get the children ...
    my ($left, $right) = @{$children};
    # if we have no left of no right, then just return
    # otherwise we need to process them ...
    unless (!$left && !$right) {
        # stash the width of the current for later
        my $orig_current_width = $current->width();
        # get the connections ...
        my ($left_connection, $right_connection) = @{$connections};
        # paste the connections
        $current->pasteLeft($left_connection)   if defined $left_connection;
        $current->pasteRight($right_connection) if defined $right_connection;

        my $bottom = "";
        if ($left) {
            $left->padRight(" " x $orig_current_width);
            if ($right) {
                $bottom = $left->pasteRight($right);
            }
            else {
                $bottom = $left;
            }
        }
        elsif ($right) {
            $bottom = $right->padLeft(" " x $orig_current_width);
        }
        # however, if we have them, then add them to the output
        $current = $current->pasteBottom($bottom);
    }
#     # just debugging
#     if (DEBUG) {
#         debug->log("=" x 80);
#         debug->log(join "\n" => map { "[$_]" } $current->getLinesAsArray());
#         debug->log("=" x 80);
#     }
    return $current;
}

1;

__END__