File Coverage

lib/GraphViz/ISA/Multi.pm

Criterion	Covered	Total	%
statement	4	6	66.6
branch			n/a
condition			n/a
subroutine	2	2	100.0
pod			n/a
total	6	8	75.0

package GraphViz::ISA::Multi;
use strict;
use GraphViz;

use Carp;

BEGIN {
	use Exporter ();
	use vars qw ($VERSION @ISA);
	$VERSION     = 0.02;
}

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

	my $self = bless ({}, ref ($class) || $class);

	$self->{data} = {};
	$self->{ignore} = $parameters{ignore};
	$self->{changed} = 0;
	return ($self);
}


sub graph
{
    my $self = shift;
    return $self->{g} unless $self->{changed};

    $self->{g} = GraphViz->new();

    ### draw all nodes:
    foreach my $module (sort keys %{$self->{data}}) {
	$self->{g}->add_node($module);
	foreach my $nod (@{$self->{data}->{$module}}) {
	    $self->{g}->add_node($nod) ;
	}
    }

    ### draw the edges
    foreach my $module (sort keys %{$self->{data}}) {
	$self->{g}->add_edge($_, $module) for
	    @{$self->{data}->{$module}};
    }
    $self->{changed} = 0;

    return $self->{g};
}


sub add
{
    my $self = shift;
    my @to_add = @_;

    foreach my $module (@to_add) {
	next if grep /$module/i, @{$self->{ignore}};
	next if $self->{data}->{$module};
	my $filename = $module;
	$filename =~ s!::!/!g;
	$filename .= ".pm";

	my @pkg = ($module);
	eval {
	    require $filename;
	    # if we got more packages in a file, find them
	    open my $fg, "<$INC{$filename}" or croak "$!\n";
	    foreach my $line (<$fg>) {
		if ($line =~ /package (.+);/) {
		    push @pkg, $1 unless $1 eq $module;
		}
	    }
	    close $fg;
        };
	if ($@) {
	    return undef;
	}

	no strict 'refs';
	foreach my $mod (@pkg) {
	    if (@{$mod . "::ISA"} > 0) {
		$self->{data}->{$mod} = \@{$mod."::ISA"};
		foreach my $ign (@{$self->{ignore}}) {
		    @{$self->{data}->{$mod}} =
			grep $_ !~ /$ign/, @{$self->{data}->{$mod}};
		}
		$self->add($_) foreach @{$mod."::ISA"};
	    }
	}
    }
    $self->{changed} = 1;
    return $self->{data};
}

sub AUTOLOAD
{
    no strict 'vars';
    my $self = shift;
    my $n = $AUTOLOAD;
    $n =~ s/.*:://g;

    return if $n =~ /DESTROY/;

    if ($n =~ /as_/) {
	$self->graph();

    }
    ### give it down to GraphViz:
    $self->{g}->$n(@_);
}


1; #this line is important and will help the module return a true value
__END__