source: trunk/lib/Gnuplot.cc @ 34

// $Id: Gnuplot.cc 31 2006-01-11 23:45:03Z jari $

#include "Gnuplot.h"

#include <cstdio>
#include <fstream>
#include <vector>
#include <unistd.h>


namespace theplu {
namespace svnstat {


  Gnuplot::Gnuplot(void)
    : linestyle_("lines"), pipe_(NULL)
  {
    acquire_program_path("gnuplot");
    if (gnuplot_binary_.empty())
      throw GnuplotException("Can't find gnuplot in your PATH");

    pipe_=popen(gnuplot_binary_.c_str(),"w");
    if (!pipe_)
      throw GnuplotException("Could'nt open connection to gnuplot");
  }


  Gnuplot::~Gnuplot(void)
  {
    if (pclose(pipe_) == -1)
      throw GnuplotException("Problem closing communication to gnuplot");
    for (std::list<std::string>::iterator i=tempfiles_.begin();
         i!=tempfiles_.end(); i++)
      remove(i->c_str());
  }


  void Gnuplot::acquire_program_path(const std::string& progname)
  {
    char* env_path=getenv("PATH");  // is there a need to free this memory?
    if (!env_path)
      throw GnuplotException("Environment variable PATH is not set");

    std::list<std::string> paths;
    tokenizer(env_path,paths);
    for (std::list<std::string>::const_iterator i=paths.begin();
         i!=paths.end(); ++i) {
      std::string tmp((*i) + '/' + progname);
      if (!access(tmp.c_str(),X_OK)) {
        gnuplot_binary_=tmp;
        break;
      }
    }

    if (gnuplot_binary_.empty())
      throw GnuplotException("Cannot find '" + progname + "' in PATH");
  }


  void Gnuplot::tokenizer(const std::string& in,
                          std::list<std::string>& tokens,
                          const std::string& delimiters)
  {
    std::string::size_type previous_pos=in.find_first_not_of(delimiters,0);
    std::string::size_type position=in.find_first_of(delimiters,previous_pos);
    while ((std::string::npos!=position) || (std::string::npos!=previous_pos)) {
      tokens.push_back(in.substr(previous_pos, position-previous_pos));
      previous_pos=in.find_first_not_of(delimiters,position);
      position=in.find_first_of(delimiters,previous_pos);
    }
  }


  void Gnuplot::command(std::string cmdstr)
  {
    if (*(cmdstr.rbegin())!='\n')
      cmdstr+='\n';
    fputs(cmdstr.c_str(),pipe_);
    fflush(pipe_);
  }


  void Gnuplot::plot_xy(const std::vector<double>& x, 
                        const std::vector<double>& y, bool replot)
  {
    char name[]="/tmp/svnstatXXXXXX";
    if (mkstemp(name) == -1)
      throw GnuplotException(std::string("Failed to get unique filename: ") +
                             name);
    std::ofstream tmp(name);
    if (tmp.bad())
      throw GnuplotException(std::string("Failed to create file: ") + name);
    for (std::vector<double>::size_type i=0; i<y.size(); ++i)
      tmp << x[i] << '\t' << y[i] << '\n';
    tmp.close();
    std::string cmdstring((replot ? "replot" : "plot"));
    cmdstring+=(std::string(" '") + name + "' title '" + linetitle_ + "' with " +
                linestyle_ + "\n");
    command(cmdstring);

    // need to keep track of created files since the gnuplot command
    // is not executed until pclose on MacOSX, shouldn't the fflush
    // fix this in the command() member function?
    tempfiles_.push_back(name);
  }


  void Gnuplot::plot_y(const std::vector<double>& y, bool replot)
  {
    char name[]="/tmp/svnstatXXXXXX";
    if (mkstemp(name) == -1)
      throw GnuplotException(std::string("Failed to get unique filename: ") +
                             name);
    std::ofstream tmp(name);
    if (tmp.bad())
      throw GnuplotException(std::string("Failed to create file: ") + name);
    for (std::vector<double>::size_type i=0; i<y.size(); ++i)
      tmp << y[i] << '\n';
    tmp.close();
    std::string cmdstring((replot ? "replot" : "plot"));
    cmdstring+=(std::string(" '") + name + "' title '" + linetitle_ + "' with " +
                linestyle_ + "\n");
    command(cmdstring);

    // need to keep track of created files since the gnuplot command
    // is not executed until pclose on MacOSX, shouldn't the fflush
    // fix this in the command() member function?
    tempfiles_.push_back(name);
  }


}} // end of namespace svnstat and namespace theplu