File Coverage

blib/lib/Iterator/RoundRobin.pm

Criterion	Covered	Total	%
statement	57	58	98.2
branch	15	18	83.3
condition			n/a
subroutine	7	7	100.0
pod	4	4	100.0
total	83	87	95.4

package Iterator::RoundRobin;

use strict;
use warnings;

our $VERSION = '0.2';

use overload
    'eq' => \&next,
    'ne' => \&next,
    "cmp" => \&next,
    "<=>" => \&next,
    '""' => \&next;

sub new {
    my ($class, @arr) = @_;
    my %args = (
        'arrays' => \@arr,
        'completed' => [],
        'max' => 0,
        'current' => 0,
        'cols' => 0,
        'track_completed' => 1,
    );
    my $self = bless { %args }, $class;
    for my $arr (@arr) {
      $self->{'cols'}++;
      my $count = scalar @{$arr};
      if ($count > $self->{'max'}) { $self->{'max'} = $count; }
    }
    if (! $self->{'col'}) { $self->{'col'} = 0; }
    return $self;
}

sub isempty {
    my ($self) = @_;
    if (scalar @{$self->{'arrays'}}) { return 0; }
    return 1;
}

sub next {
    my ($self) = @_;
    if ($self->isempty()) { return undef; }
    if (my $data = shift @{ $self->{'arrays'}->[$self->{'col'}] }) {
        my $oldcol = $self->{'col'};
        $self->{'col'}++;
        unless ( scalar @{ $self->{'arrays'}->[ $oldcol ] }) { $self->rebuildarrays(); $self->{'col'} = $oldcol; }
        if ($self->{'track_completed'}) {
            push @{ $self->{'completed'} }, $data;
        }
        if ($self->{'col'} == $self->{'cols'}) {
            $self->{'col'} = 0;
        }
        return $data;
    }
    return undef;
}

sub rebuildarrays {
    my ($self) = @_;
    my @arrays = ();
    $self->{'cols'} = 0;
    my $col = 1;
    for my $list (@{$self->{'arrays'}}) {
        my $count = scalar @{$list};
        if ($count) {
            push @arrays, $list; $self->{'cols'}++;
        }
        $col++;
    }
    $self->{'arrays'} = \@arrays;
    return 1;
}

1;
__END__