source: trunk/lib/Stats.cc @ 41

//$Id: Stats.cc 41 2006-01-13 16:41:57Z peter $

#include "Stats.h"
#include "Gnuplot.h"
#include "utility.h"

#include <cstdio>
#include <fstream>

#include <algorithm>
#include <map>
#include <numeric>
#include <sstream>
#include <string>
#include <utility>
#include <vector>

#include <iostream>

namespace theplu{
namespace svnstat{

  Gnuplot Stats::gnuplot_pipe_;
  u_int Stats::latest_revision_=0;

  std::vector<double> Stats::accumulated(void) const
  {
    if (map_.empty())
      return std::vector<double>();

    // sum of all users
    std::vector<u_int> sum(latest_revision_+1,0);
    sum=std::accumulate(map_.begin(), map_.end(), sum,
                        PairValuePlus<std::string,u_int>());

    // calculate accumulated sum
    std::vector<double> accum(sum.size());
    std::partial_sum(sum.begin(),sum.end(),accum.begin());    
    return accum;
  }

  std::vector<double> Stats::accumulated(const std::string& user) const
  {
    if (!map_.count(user))
      return std::vector<double>();
    std::vector<u_int> vec=(map_.find(user))->second;
  
    vec.reserve(latest_revision_+1);
    if (vec.size() < latest_revision_+1)
      vec.insert(vec.end(), latest_revision_+1-vec.size(), 0);

    std::vector<double> accum(vec.size());
    std::partial_sum(vec.begin(),vec.end(),accum.begin());
    return accum;
  }

  void Stats::add(const std::string& user, const u_int& rev)
  {
    std::vector<u_int> vec = map_[user];
    if (vec.size() < rev+1){
      u_int i=vec.size();
      vec.resize(rev+1);
      for (; i<rev; i++)
        vec[i]=0;
      vec[rev]=1;
      latest_revision_ = std::max(latest_revision_,rev);
    }
    else
      vec[rev]++;
    map_[user]=vec;
  }

  std::string Stats::plot(void) const
  {
    char name[]="svnstatXXXXXX";
    if (!strlen(mktemp(name))) {
      std::cerr << "Failed to get unique filename: " << name << std::endl;
      exit(-1);
    }
    std::string cmd=std::string("set term png; set output '")+name+".png'";
    gnuplot_pipe_.command(cmd);
    gnuplot_pipe_.command("set key left Left reverse");
    std::vector<double> x=accumulated();
    std::stringstream sa;
    sa << x.back() << " total";
    gnuplot_pipe_.linetitle(sa.str());
    gnuplot_pipe_.plot(x);
    for (MapConstIter_ i= map_.begin(); i != map_.end(); i++) {
      x=accumulated(i->first);
      std::stringstream s;
      s << x.back() << " " << i->first;
      gnuplot_pipe_.linetitle(s.str());
      gnuplot_pipe_.replot(x);
    }

    // Jari, must rewrite output once since the replots above are not
    // added to the plot. Rather, the plot only contains the result
    // from the first 'plot' call.
    gnuplot_pipe_.command(cmd);
    gnuplot_pipe_.command("replot");
    return std::string(name)+".png";
  }

  void Stats::print(std::ostream& os) const
  {
    os << "<p><img src='" << plot() << "' alt='[svnstat plot]' border=0></p>\n";
  }

  Stats& Stats::operator+=(const Stats& other)
  {
    for (MapConstIter_ o_i= other.map_.begin(); o_i != other.map_.end(); ++o_i)
    {
      std::pair<MapIter_,bool> result = map_.insert(*o_i);
      if (!result.second)
        map_[(*(result.first)).first] = 
          VectorPlus<u_int>()( (*(result.first)).second, (*o_i).second );
  
    }
    return *this;
  }

}} // end of namespace svnstat and namespace theplu