source: trunk/xgraph/jpgraph/jpgraph_stock.php @ 42

<?php
/*=======================================================================
 // File:        JPGRAPH_STOCK.PHP
 // Description: Stock plot extension for JpGraph
 // Created:     2003-01-27
 // Ver:         $Id: jpgraph_stock.php 1364 2009-06-24 07:07:44Z ljp $
 //
 // Copyright (c) Asial Corporation. All rights reserved.
 //========================================================================
 */

//===================================================
// CLASS StockPlot
//===================================================
class StockPlot extends Plot {
    protected $iTupleSize = 4;
    private $iWidth=9;
    private $iEndLines=1;
    private $iStockColor1='white',$iStockColor2='darkred',$iStockColor3='darkred';
    //---------------
    // CONSTRUCTOR
    function __construct($datay,$datax=false) {
        if( count($datay) % $this->iTupleSize ) {
            JpGraphError::RaiseL(21001,$this->iTupleSize);
            //('Data values for Stock charts must contain an even multiple of '.$this->iTupleSize.' data points.');
        }
        parent::__construct($datay,$datax);
        $this->numpoints /= $this->iTupleSize;
    }
    //---------------
    // PUBLIC METHODS

    function SetColor($aColor,$aColor1='white',$aColor2='darkred',$aColor3='darkred') {
        $this->color = $aColor;
        $this->iStockColor1 = $aColor1;
        $this->iStockColor2 = $aColor2;
        $this->iStockColor3 = $aColor3;
    }

    function SetWidth($aWidth) {
        // Make sure it's odd
        $this->iWidth = 2*floor($aWidth/2)+1;
    }

    function HideEndLines($aHide=true) {
        $this->iEndLines = !$aHide;
    }

    // Gets called before any axis are stroked
    function PreStrokeAdjust($graph) {
        if( $this->center ) {
            $a=0.5; $b=0.5;
            $this->numpoints++;
        } else {
            $a=0; $b=0;
        }
        $graph->xaxis->scale->ticks->SetXLabelOffset($a);
        $graph->SetTextScaleOff($b);
    }

    // Method description
    function Stroke($img,$xscale,$yscale) {
        $n=$this->numpoints;
        if( $this->center ) $n--;
        if( isset($this->coords[1]) ) {
            if( count($this->coords[1])!=$n ) {
                JpGraphError::RaiseL(2003,count($this->coords[1]),$n);
                // ("Number of X and Y points are not equal. Number of X-points:".count($this->coords[1])." Number of Y-points:$numpoints");
            }
            else {
                $exist_x = true;
            }
        }
        else {
            $exist_x = false;
        }

        if( $exist_x ) {
            $xs=$this->coords[1][0];
        }
        else {
            $xs=0;
        }

        $ts = $this->iTupleSize;
        $this->csimareas = '';
        for( $i=0; $i<$n; ++$i) {

            //If value is NULL, then don't draw a bar at all
            if ($this->coords[0][$i*$ts] === null) continue;

            if( $exist_x ) {
                $x=$this->coords[1][$i];
                                if ($x === null) continue;
            }
            else {
                $x=$i;
            }
            $xt = $xscale->Translate($x);

            $neg = $this->coords[0][$i*$ts] > $this->coords[0][$i*$ts+1] ;
            $yopen  = $yscale->Translate($this->coords[0][$i*$ts]);
            $yclose = $yscale->Translate($this->coords[0][$i*$ts+1]);
            $ymin   = $yscale->Translate($this->coords[0][$i*$ts+2]);
            $ymax   = $yscale->Translate($this->coords[0][$i*$ts+3]);

            $dx = floor($this->iWidth/2);
            $xl = $xt - $dx;
            $xr = $xt + $dx;

            if( $neg ) {
                $img->SetColor($this->iStockColor3);
            }
            else {
                $img->SetColor($this->iStockColor1);
            }
            $img->FilledRectangle($xl,$yopen,$xr,$yclose);
            $img->SetLineWeight($this->weight);
            if( $neg ) {
                $img->SetColor($this->iStockColor2);
            }
            else {
                $img->SetColor($this->color);
            }

            $img->Rectangle($xl,$yopen,$xr,$yclose);

            if( $yopen < $yclose ) {
                $ytop = $yopen ;
                $ybottom = $yclose ;
            }
            else {
                $ytop = $yclose ;
                $ybottom = $yopen ;
            }
            $img->SetColor($this->color);
            $img->Line($xt,$ytop,$xt,$ymax);
            $img->Line($xt,$ybottom,$xt,$ymin);

            if( $this->iEndLines ) {
                $img->Line($xl,$ymax,$xr,$ymax);
                $img->Line($xl,$ymin,$xr,$ymin);
            }

            // A chance for subclasses to add things to the bar
            // for data point i
            $this->ModBox($img,$xscale,$yscale,$i,$xl,$xr,$neg);

            // Setup image maps
            if( !empty($this->csimtargets[$i]) ) {
                $this->csimareas.= '<area shape="rect" coords="'.
                round($xl).','.round($ytop).','.
                round($xr).','.round($ybottom).'" ';
                $this->csimareas .= ' href="'.$this->csimtargets[$i].'"';
                if( !empty($this->csimalts[$i]) ) {
                    $sval=$this->csimalts[$i];
                    $this->csimareas .= " title=\"$sval\" alt=\"$sval\" ";
                }
                $this->csimareas.= "  />\n";
            }
        }
        return true;
    }

    // A hook for subclasses to modify the plot
    function ModBox($img,$xscale,$yscale,$i,$xl,$xr,$neg) {}

} // Class

//===================================================
// CLASS BoxPlot
//===================================================
class BoxPlot extends StockPlot {
    private $iPColor='black',$iNColor='white';

    function __construct($datay,$datax=false) {
        $this->iTupleSize=5;
        parent::__construct($datay,$datax);
    }

    function SetMedianColor($aPos,$aNeg) {
        $this->iPColor = $aPos;
        $this->iNColor = $aNeg;
    }

    function ModBox($img,$xscale,$yscale,$i,$xl,$xr,$neg) {
        if( $neg )
        $img->SetColor($this->iNColor);
        else
        $img->SetColor($this->iPColor);

        $y = $yscale->Translate($this->coords[0][$i*5+4]);
        $img->Line($xl,$y,$xr,$y);
    }
}

/* EOF */
?>