File Coverage

blib/lib/Graph/Traverse.pm

Criterion	Covered	Total	%
statement	73	74	98.6
branch	46	50	92.0
condition	34	39	87.1
subroutine	6	6	100.0
pod	1	1	100.0
total	160	170	94.1

package Graph::Traverse 0.04 {

    use warnings;
    use strict;

    use parent 'Graph';
    use Carp;

    sub traverse {
        # Use as: $graph->search( START_VERTEX, [OPTS])
        #
        # Traverses edges from the start vertex (or verticess) [either
        # a scalar with a single vertex's name, or an array of vertex
        # names, may be passed], finding adjacent vertices using the
        # 'next' function (by default, 'successors'), until either a
        # maximum accumulated edge weight ('max' option, if given) is
        # exceeded (by default using the 'weight' attribute, or
        # specify an 'attribute'), or until a callback function ('cb')
        # returns a nonzero value.  Default is to return the list of
        # vertices encountered in the search; use option 'weights' to
        # return a list of vertex=>weight_value.
        #
        # Use option 'hash=>1' to return a hash where keys are vertex
        # names, and values are a hash containing the 'path' to that
        # vertex from the starting vertex, the 'weight' at that
        # vertex, and 'terminal' the value of the callback function
        # returned for that vertex (if nonzero, further nodes on that
        # branch are not searched).  Note that as we traverse the
        # graph, we may encounter the same vertex several times, but
        # only the shortest path (lowest weight) will be retained in
        # the final hash.

        my ($self, $vertex, $opts) = @_;
        carp "Must pass starting vertex" unless defined $vertex;

        my $cb_check = $opts->{cb} if defined $opts;

        my $return_weights = (defined $opts && $opts->{weights});
        my $return_hash = (defined $opts && $opts->{hash});
        my $save_paths = ($return_hash) ? [] : undef;
        # Save all nodes (default), or only the callback-flagged terminal nodes?
        my $save_all = (defined $opts && ($opts->{all} // 1));

        # If option 'attribute' is defined, we accumulate weights from each edge.
        # Define 'max' to terminate when a maximum weight is achieved.
        # Define 'vertex' to accumulate vertex weights rather than edge weights.
        # Define 'incr' to change the default weight value from 1.
        my $attr = (defined $opts) ? ($opts->{attribute} // 'weight') : 'weight';
        my $max_weight = $opts->{max} if defined $opts;
        my $add_vertex = $opts->{vertex} if defined $opts;
        my $incr = (defined $opts) ? ($opts->{default} // 1) : 1;  # default weight for each edge

        # Use a method that will return a list of adjacent vertices.
        # Other useful values are 'predecessors' and 'neighbors'.
        my $next = (defined $opts ? ($opts->{next}) : undef) // 'successors';

        my (%todo, %path, %weight);
        foreach my $s (@{ref $vertex ? $vertex : [$vertex]}) {
            $todo{$s} = $s;
            $path{$s} = [$s];
            $weight{$s} = 0;
        }
        my %terminal;
        my %seen;
        my %init = %todo;
        while (keys %todo) {
            my @todo = values %todo;
            for my $t (@todo) {
                $seen{$t} = delete $todo{$t};
                for my $s ($self->$next($t)) {
                    next unless $self->has_vertex($s);
                    my $newvalue;
                    if (defined $attr) {
                        if ($add_vertex) { # Add vertex attribute value
                            $newvalue = $weight{$t} + ($self->get_vertex_attribute($s, $attr) // $incr);
                        } else {           # Add edge attribute value (default 'weight', default value 1)
                            # Note, if our search function is 'predecessors' or 'neighbors' we
                            # may find nodes in reverse direction, but we want the edge attributes
                            # in either case
                            $newvalue = $weight{$t} + ($self->get_edge_attribute($t, $s, $attr) //
                                                      $self->get_edge_attribute($s, $t, $attr) //
                                                      $incr);
                        }
                    } else {
                        $newvalue = $weight{$t} + $incr;
                    }
                    # If callback function returns nonzero, do not traverse beyond this node.
                    if (defined $cb_check) {
                        if ($terminal{$s} = &$cb_check($self, $s, $newvalue, $opts)) {
                            $seen{$s} = $s;
                        }
                    }
                    # Do not save vertices beyond a defined maximum weight
                    next if (defined $max_weight) && ($newvalue > $max_weight);
                    # Always save the found vertices.  As we traverse,
                    # we may later encounter shortcuts which we must
                    # discard before the final return (see below).
                    if (defined $save_paths) {
                        my $this_node = { vertex => $s,
                                          path => [@{$path{$t}}, $s],
                                          weight => $newvalue };
                        $this_node->{terminal} = $terminal{$s} if exists $terminal{$s};
                        push @{$save_paths}, $this_node if ($save_all || $terminal{$s});
                    }
                    # Only save new paths, and shorter-than-previously-found paths.
                    if ((!defined $path{$s}) || ($newvalue < $weight{$s} )) {
                        # If new path is shorter than we previously
                        # found, then retrace all paths from this
                        # vertex onward.
                        delete $seen{$s} if (defined $weight{$s} && $newvalue < $weight{$s});
                        $weight{$s} = $newvalue;
                        $path{$s} = [@{$path{$t}}, $s];
                    }
                    # If callback function returns nonzero, do not
                    # traverse beyond this node.  NOTE: In the case of
                    # multiple paths to the following node, the above
                    # does track the shortest path to the node, but
                    # the caller will not receive every combination of
                    # paths *through* the node.
                    next if ($terminal{$s});
                    $todo{$s} = $s unless exists $seen{$s};
                }
            }
        }
        for my $v (keys %init) {
            delete $seen{$v};
            delete $weight{$v};
        }
        # return $save_paths if defined $return_all;
        if ($return_hash) {
            # Scan list of found vertices, overwriting higher-valued
            # (longer) paths with lower (shorter) ones which were
            # found later.
            my $ret = {};
            foreach my $v (@{$save_paths}) {
                $ret->{$v->{vertex}} = $v if (!defined $ret->{$v->{vertex}} || ($ret->{$v->{vertex}}->{weight} > $v->{weight}));
            }
            return $ret;
        }
        return $return_weights ? (%weight) : (values %seen);
    }

    if (Graph->can('traverse')) {
        carp ('Graph already has a traverse method.');
    } else {
        no warnings 'redefine', 'once'; ## no critic
        *Graph::traverse = \&traverse;
    }

};

1;

__END__