File Coverage

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

Criterion	Covered	Total	%
statement	47	47	100.0
branch	14	14	100.0
condition	4	6	66.6
subroutine	10	10	100.0
pod	4	4	100.0
total	79	81	97.5

package Tree::Simple::Visitor::VariableDepthClone;

use strict;
use warnings;

use Scalar::Util 'blessed';

our $VERSION = '0.16';

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 _init {
    my ($self) = @_;
    $self->{clone_depth} = undef;
    $self->SUPER::_init();
}

sub setCloneDepth {
    my ($self, $clone_depth) = @_;
    (defined($clone_depth))
    	|| die "Insufficient Arguments : you must supply a clone depth";
    $self->{clone_depth} = $clone_depth;
}

sub getClone {
    my ($self) = @_;
    return $self->getResults()->[0];
}

sub visit {
    my ($self, $tree) = @_;
    (blessed($tree) && $tree->isa("Tree::Simple"))
    	|| die "Insufficient Arguments : You must supply a valid Tree::Simple object";

    my $filter = $self->getNodeFilter();

    # get a new instance of the root tree type
    my $new_root = blessed($tree)->new($tree->ROOT);
    my $new_tree = $new_root;

    if ($self->includeTrunk()) {
        my $cloned_trunk = blessed($tree)->new();
        $cloned_trunk->setNodeValue(
            Tree::Simple::_cloneNode($tree->getNodeValue())
        );
        $filter->($tree, $cloned_trunk) if defined $filter;
        $new_tree->addChild($cloned_trunk);
        $new_tree = $cloned_trunk;
    }

    $self->_cloneTree($tree, $new_tree, $self->{clone_depth}, $filter);

    $self->setResults($new_root);
}

sub _cloneTree {
    my ($self, $tree, $clone, $depth, $filter) = @_;
    return if $depth <= 0;
    foreach my $child ($tree->getAllChildren()) {
        my $cloned_child = blessed($child)->new();
        $cloned_child->setNodeValue(
            Tree::Simple::_cloneNode($child->getNodeValue())
        );
        $filter->($child, $cloned_child) if defined $filter;
        $clone->addChild($cloned_child);
        $self->_cloneTree($child, $cloned_child, $depth - 1, $filter) unless $child->isLeaf();
    }
}

1;

__END__