source: trunk/lib/Graph.cc @ 881

// $Id: Graph.cc 881 2009-11-24 11:19:52Z jari $

/*
  Copyright (C) 2009 Jari Häkkinen

  This file is part of svndigest, http://dev.thep.lu.se/svndigest

  svndigest is free software; you can redistribute it and/or modify it
  under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 3 of the License, or
  (at your option) any later version.

  svndigest is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with svndigest. If not, see <http://www.gnu.org/licenses/>.
*/

#include "Graph.h"

#include "Date.h"

#include <cmath>

namespace theplu {
namespace svndigest {

  std::vector<std::string> Graph::xticks_;

  Graph::Graph(const std::string& filename)
    : plots_(0), pls_(1,1,"svg",filename.c_str()), title_(filename)
  {
    // we use colour map 0 position 0 for background colour
    pls_.scolbga(230,230,230,1);
    pls_.init();
    pls_.adv(0);
    pls_.vsta();
  }


  Graph::~Graph(void)
  {
  }


  bool Graph::date_xticks(void)
  {
    return xticks_.size() != 0;
  }


  void Graph::current_colour(unsigned char r, unsigned char g, unsigned char b)
  {
    // we use colour map 0 position 1 for current colour
    pls_.scol0a(1,r,g,b,1.0);
    pls_.col0(1);
  }


  void Graph::plot(const std::vector<unsigned int>& y, const std::string& legend,
                   const std::string& timeformat)
  {
    unsigned int xmin= date_xticks() ? Date(xticks_.front()).seconds() : 0;
    unsigned int xmax= date_xticks() ? Date(xticks_.back()).seconds() : y.size();
    pls_.wind(xmin, xmax, 0, yrange_);

    if (!plots_) {
      // draw plot frame, x and y ticks only for the first plot
      pls_.scol0a(2,0,0,0,1.0);
      pls_.col0(2);

      if (date_xticks())
        pls_.timefmt(timeformat.c_str());

      unsigned int ytickspacing=tick_spacing(yrange_);
      unsigned int xtickspacing=tick_spacing(xmax-xmin);
      pls_.box("bcnstd", xtickspacing, 1, "bcnstv", ytickspacing, 2);
      pls_.lab("Date", "Number of lines", title_.c_str());
    }
    ++plots_;

    print_legend(xmin,0,xmax,yrange_,legend);
    pls_.col0(1);
    for (unsigned int i=1; i<y.size(); ++i) {
      PLFLT x0=i-1;
      PLFLT x1=i;
      if (date_xticks()) {
        x0=Date(xticks_[i-1]).seconds();
        x1=Date(xticks_[i]).seconds();
      }
      pls_.join(x0, y[i-1], x0, y[i]);
      pls_.join(x0, y[i]  , x1, y[i]);
    }
  }


  void Graph::print_legend(PLFLT xmin, PLFLT ymin, PLFLT xmax, PLFLT ymax,
                           const std::string& legend)
  {
    PLFLT xrange=(xmax-xmin);
    PLFLT length=0.05*xrange;
    PLFLT dx=0.01*xrange;
    PLFLT x=4*dx+xmin+length;
    PLFLT yrange=(ymax-ymin);
    PLFLT dy=0.001*yrange;
    PLFLT y=(1.-0.05*plots_)*yrange;
    pls_.col0(1);
    pls_.join(x-dx, y-3*dy, x-dx-length, y-3*dy);
    pls_.col0(2);
    pls_.ptex(x, y, 0, 0, 0, legend.c_str());
  }


  void Graph::set_dates(const std::vector<std::string>& date)
  {
    xticks_=date;
  }


  unsigned int Graph::tick_spacing(const double range) const
  {
    double frac=range/5;
    unsigned char characteristic=log10(frac);
    unsigned int power=pow(10,characteristic);
    unsigned char msn=frac/power;
    return power*msn;
  }


  const std::vector<std::string>& Graph::xticks(void)
  {
    return xticks_;
  }


  double Graph::yrange(double ymax)
  {
    return yrange_=ymax;
  }

}} // end of namespace svndigest and namespace theplu