File Coverage

blib/lib/Language/Homespring/Visualise/GraphViz.pm

Criterion	Covered	Total	%
statement	18	40	45.0
branch	0	4	0.0
condition	0	8	0.0
subroutine	6	9	66.6
pod	2	3	66.6
total	26	64	40.6

package Language::Homespring::Visualise::GraphViz;

$VERSION = 0.04;

use warnings;
use strict;
use GraphViz;

sub new {
        my $class = shift;
        my $self = bless {}, $class;

        my $options = shift;
        $self->{interp}		= $options->{interp};
	$self->{spring_col}	= $options->{spring_col} || '#C0C0FF';
	$self->{node_col}	= $options->{node_col} || 'white';

	$self->{fontname}	= $options->{fontname} || 'Times';
	$self->{fontsize}	= $options->{fontsize} || '12';

        return $self;
}

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

	$self->{graph}	= GraphViz->new(
			directed => 0,
			layout => 'dot',
			rankdir => 1,
			epsilon => 1,
		);

	$self->add_node($self->{interp}->{root_node}, 1);

	return $self->{graph};
}

sub add_node {
	my ($self, $node, $rank) = @_;

	for(@{$node->{child_nodes}}){
		my $label = $_->{node_name_safe};
		$label =~ s/\\/\\\\/g;

		my $fillcolor = $_->{spring}?$self->{spring_col}:$self->{node_col};

		$self->{graph}->add_node(
			$_->{uid},
			label		=> $label,
			rank		=> $rank,
			fillcolor	=> $fillcolor,
			style		=> 'filled',
			fontname	=> $self->{fontname},
			fontsize	=> $self->{fontsize},
		);

		$self->{graph}->add_edge(
			$node->{uid} => $_->{uid},
			arrowtail => 'normal',
		) if ($node->{uid} != $self->{interp}->{root_node}->{uid});

		$self->add_node($_, $rank+1);
	}
}

__END__