File Coverage

blib/lib/Tree/Simple/Visitor/ToNestedArray.pm

Criterion	Covered	Total	%
statement	32	32	100.0
branch	10	10	100.0
condition	4	6	66.6
subroutine	7	7	100.0
pod	2	2	100.0
total	55	57	96.4

package Tree::Simple::Visitor::ToNestedArray;

use strict;
use warnings;

our $VERSION = '0.16';

use Scalar::Util qw(blessed);

use base qw(Tree::Simple::Visitor);

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

sub visit {
	my ($self, $tree) = @_;
	(blessed($tree) && $tree->isa("Tree::Simple"))
		|| die "Insufficient Arguments : You must supply a valid Tree::Simple object";
    # grab our filter (if we have one)
    my $filter = $self->getNodeFilter();
    my @results;
    # get the array
    $self->_buildArray($tree, \@results, $filter);
    # add the trunk if we need to
    @results = (
            ((defined($filter)) ?
                    $filter->($tree)
                    :
                    $tree->getNodeValue()),
            [ @results ]
        ) if $self->includeTrunk();
    # set results
    $self->setResults(\@results);
}

sub _buildArray {
    my ($self, $tree, $accumulator, $filter) = @_;
    foreach my $child ($tree->getAllChildren()) {
        push @{$accumulator} => (defined($filter) ? $filter->($child) : $child->getNodeValue());
        push @{$accumulator} => $self->_buildArray($child, [], $filter) unless $child->isLeaf();
    }
    return $accumulator;
}

1;

__END__