File Coverage

blib/lib/Data/RandomPerson/Choice.pm

Criterion	Covered	Total	%
statement	29	29	100.0
branch	2	2	100.0
condition	2	2	100.0
subroutine	8	8	100.0
pod	5	5	100.0
total	46	46	100.0

package Data::RandomPerson::Choice;

use strict;
use warnings;

use List::Util::WeightedChoice qw(choose_weighted);

sub new {
    my ($class) = @_;

    return bless {}, $class;
}

sub add {
    my ( $self, $value, $count ) = @_;

    $count ||= 1;

    push @{ $self->{data} }, $value;
    push @{ $self->{weight} }, $count;
    $self->{size}++;
}

sub add_list {
    my ( $self, @list ) = @_;

    push @{ $self->{data} }, @list;
    push @{ $self->{weight} }, 1 for 1..scalar(@list);
    $self->{size} += scalar(@list);
}

sub size {
    my ($self) = @_;

    return $self->{size};
}

sub pick {
    my ($self) = @_;

    die "No data has been added to the list" unless $self->{size};

    return choose_weighted($self->{data}, $self->{weight});
}

1;

__END__