File Coverage

blib/lib/Chart/Gnuplot/Pie.pm

Criterion	Covered	Total	%
statement	63	137	45.9
branch	9	32	28.1
condition	9	27	33.3
subroutine	10	13	76.9
pod	3	3	100.0
total	94	212	44.3

package Chart::Gnuplot::Pie;
use strict;
use vars qw($VERSION);
use base 'Chart::Gnuplot';
use Carp;
$VERSION = '0.04';


sub new
{
    my ($self, %opt) = @_;
    my $obj = $self->SUPER::new(%opt);
    $obj->set(
        parametric => '',
        xyplane    => 'at 0',
        urange     => '[0:1]',
        vrange     => '[0:1]',
        zrange     => '[-1:1]',
        cbrange    => '[-1:1]',
    );
    $obj->command(join("\n", (
        'unset border',
        'unset tics',
        'unset key',
        'unset colorbox',
    )));
    return($obj);
}


# Plot 2D pie chart
sub plot2d
{
    my ($self, $dataSet) = @_;
    $self->set(
        xrange => '[-1.5:1.5]',
        yrange => '[-1.5:1.5]',
        size   => 'square',
        view   => 'map',
    );
    $self->SUPER::_setChart([$dataSet]);

    open(CHT, ">>$self->{_script}") || confess("Can't write $self->{_script}");
    print CHT "set multiplot\n";
    print CHT $dataSet->_thaw2d($self);
    print CHT "unset multiplot\n";
    close(CHT);

    $self->SUPER::execute();
    return($self);
}


# Plot 3D pie chart
sub plot3d
{
    my ($self, $dataSet) = @_;
    $self->set(
        xrange => '[-1:1]',
        yrange => '[-1:1]',
    );
    $self->SUPER::_setChart([$dataSet]);

    open(CHT, ">>$self->{_script}") || confess("Can't write $self->{_script}");
    print CHT "set multiplot\n";
    print CHT $dataSet->_thaw3d($self);
    print CHT "unset multiplot\n";
    close(CHT);

    $self->SUPER::execute();
    return($self);
}

1;

##############################################################

package Chart::Gnuplot::Pie::DataSet;
use strict;
use base 'Chart::Gnuplot::DataSet';


# Plot 2D pie chart
sub _thaw2d
{
    my ($self, $chart) = @_;
    my $string = '';
    my $rotate = (defined $self->{rotate})? $self->{rotate} : 0;

    my $pt = $self->{data};
    my $sum = 0;
    for (my $i = 0; $i < @$pt; $i++)
    {
        $sum += $$pt[$i][1];
    }

    # Print label
    my $s = my $start = $rotate/360;
    my (@r, @g, @b) = ();
    for (my $i = 0; $i < @$pt; $i++)
    {
        my $e = $$pt[$i][1]/$sum + $s;

        # Print label
        my $pos = "cos(($s+$e)*pi)*1.1, sin(($s+$e)*pi)*1.1";
        $pos .= ", -0.1" if ($s+$e > 1 && $s+$e < 2);
        $pos .= ", 0.2" if ($s+$e < 1 || $s+$e > 2);
        $pos .= " right" if ($s+$e > 0.5 && $s+$e < 1.5);
        $pos .= " front";
        $chart->label(
            text     => $$pt[$i][0],
            position => $pos,
        );
        $string .= "set label${$chart->{_labels}}[-1]\n";
        $s = $e;
    }

    # Draw top surface
    $s = $start;
    for (my $i = 0; $i < @$pt; $i++)
    {
        my $e = $$pt[$i][1]/$sum + $s;

        # Set colors of the slices
        # - Initialize random color if not specified
        my ($r, $g, $b);
        if (!defined $self->{colors} || ${$self->{colors}}[$i] eq '')
        {
            $r = rand();
            $g = rand();
            $b = rand();
        }
        else
        {
            ($r, $g, $b) = &_rgb2real(${$self->{colors}}[$i]);
        }

        # Draw top surface
        $string .= "set palette model RGB functions $r, $g, $b\n";
        $string .= "splot cos(2*pi*(($e-$s)*u+$s))*v, ".
            "sin(2*pi*(($e-$s)*u+$s))*v, 0.1 with pm3d\n";
        $s = $e;
    }

    # Draw border around slice
    if (defined $self->{border} && $self->{border} ne 'off')
    {
        # Set line properties
        my $border = $self->{border};
        my $linecolor = "black";
        my $linewidth = 1;
        if (ref($border) eq 'HASH')
        {
            $linecolor = $$border{color} if (defined $$border{color});
            $linewidth = $$border{width} if (defined $$border{width});
        }

        $s = $start;
        for (my $i = 0; $i < @$pt; $i++)
        {
            my $e = $$pt[$i][1]/$sum + $s;
            $string .= "splot cos(2*pi*(($e-$s)*u+$s)), ".
                "sin(2*pi*(($e-$s)*u+$s)), 0.1 with lines lt 1 ".
                "lw $linewidth lc rgb \"$linecolor\"";
            $string .= ", u*cos(2*pi*$s), u*sin(2*pi*$s), 0.1 ".
                "with lines lt 1 lw $linewidth lc rgb \"$linecolor\"\n";
            $s = $e;
        }
    }

    return($string);
}


# Plot 3D pie chart
sub _thaw3d
{
    my ($self, $chart) = @_;
    my $string = '';
    my $rotate = (defined $self->{rotate})? $self->{rotate} : 0;

    my $pt = $self->{data};
    my $sum = 0;

    for (my $i = 0; $i < @$pt; $i++)
    {
        $sum += $$pt[$i][1];
    }

    # Print label and draw side sureface
    my $s = my $start = $rotate/360;
    my (@r, @g, @b) = ();
    for (my $i = 0; $i < @$pt; $i++)
    {
        my $e = $$pt[$i][1]/$sum + $s;

        # Print label
        my $pos = "cos(($s+$e)*pi)*1.1, sin(($s+$e)*pi)*1.1";
        $pos .= ", -0.1" if ($s+$e > 1 && $s+$e < 2);
        $pos .= ", 0.2" if ($s+$e < 1 || $s+$e > 2);
        $pos .= " right" if ($s+$e > 0.5 && $s+$e < 1.5);
        $pos .= " front";
        $chart->label(
            text     => $$pt[$i][0],
            position => $pos,
        );
        $string .= "set label${$chart->{_labels}}[-1]\n";

        # Set colors of the slices
        # - Initialize random color if not specified
        if (!defined $self->{colors} || ${$self->{colors}}[$i] eq '')
        {
            push(@r, rand());
            push(@g, rand());
            push(@b, rand());
        }
        else
        {
            my ($r, $g, $b) = &_rgb2real(${$self->{colors}}[$i]);
            push(@r, $r);
            push(@g, $g);
            push(@b, $b);
        }
        $string .= "set palette model RGB functions ".
            "$r[$i]*0.8, $g[$i]*0.8, $b[$i]*0.8\n";

        # Draw side surface
        $string .= "splot cos(2*pi*(($e-$s)*u+$s)), ".
            "sin(2*pi*(($e-$s)*u+$s)), v*0.2 with pm3d\n";
        $s = $e;
    }

    # Draw top surface
    $s = $start;
    for (my $i = 0; $i < @$pt; $i++)
    {
        my $e = $$pt[$i][1]/$sum + $s;

        # Draw top surface
        $string .= "set palette model RGB functions ".
            "$r[$i], $g[$i], $b[$i]\n";
        $string .= "splot cos(2*pi*(($e-$s)*u+$s))*v, ".
            "sin(2*pi*(($e-$s)*u+$s))*v, 0.2 with pm3d\n";
        $s = $e;
    }

    return($string);
}


# Transform #RRGGBB to (0-1, 0-1, 0-1)
# - called by _thaw2d() and _thaw3d()
sub _rgb2real
{
    my ($rgb) = @_;
    my ($r, $g, $b) = ($rgb =~ /^#(.{2})(.{2})(.{2})/);
    return(&_16to1($r)/255, &_16to1($g)/255, &_16to1($b)/255);
}


# Transform 0-H to 0-255
# - called by _rgb2real()
sub _16to1
{
    my ($x) = @_;
    my %tran = (
        0 => 0, 1 => 1, 2 => 2, 3 => 3, 4 => 4,
        5 => 5, 6 => 6, 7 => 7, 8 => 8, 9 => 9,
        A => 10, B => 11, C => 12, D => 13, E => 14, F => 15,
        a => 10, b => 11, c => 12, d => 13, e => 14, f => 15,
    );
    my ($a, $b) = ($x =~ /^(.)(.)$/);
    return($tran{$a}*16 + $tran{$b});
}


1;

__END__