JTuna2Graph.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <sstream>
#include <iomanip>
#include <iterator>
#include <set>
#include <map>

#include "TROOT.h"
#include "TFile.h"
#include "TGraph.h"
#include "TTimeStamp.h"
#include "TNamed.h"

#include "JDB/JDatalog.hh"
#include "JDB/JSupport.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JROOT/JRootToolkit.hh"
#include "JTools/JRange.hh"

#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"


namespace {

  /**
   * Auxiliary data structure for book keeping of histograms.
   */
  struct JKey_t {
    /**
     * Constuctor.
     *
     * \param  string          string
     * \param  floor           floor
     * \param  parameter       parameter
     */
    JKey_t(const int          string,
           const int          floor,
           const std::string& parameter) :
      string   (string),
      floor    (floor),
      parameter(parameter)
    {}


    /**
     * Less-than operator.
     *
     * \param  first           first  key
     * \param  second          second key
     * \return                 true if first key before second key; else false
     */
    friend inline bool operator<(const JKey_t& first, const JKey_t& second)
    {
      if (first.string == second.string) {

        if (first.floor == second.floor)
          return first.parameter < second.parameter;
        else
          return first.floor     < second.floor;

      } else {

        return first.string < second.string;
      }
    }


    /**
     * Convert to string.
     *
     * \return                 string
     */
    std::string toString() const
    {
      std::ostringstream os;

      os << *this;

      return os.str();
    }


    /**
     * Write key to output stream.
     *
     * \param  output          output stream
     * \param  object          key
     * \return                 output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JKey_t& object)
    {
      using namespace std;

      return out << setw(3) << setfill('0') << object.string << '.'
                 << setw(2) << setfill('0') << object.floor  << '.'
                 << setfill(' ') << object.parameter;
    }

    int         string;
    int         floor;
    std::string parameter;
  };
}


/**
 * \file
 *
 * Auxiliary program to convert ROOT TTree with slow control data to ROOT TGraph's.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  JMultipleFileScanner<JDatalog> inputFile;
  JLimit_t&   numberOfEvents = inputFile.getLimit();
  string      outputFile;
  set<string> filter;
  int         debug;

  try {

    JParser<> zap("Auxiliary program to convert ROOT TTree with slow control data to ROOT TGraph's.");
    
    zap['f'] = make_field(inputFile,  "ROOT input file (output file of JTuna).");
    zap['n'] = make_field(numberOfEvents)        = JLimit::max();
    zap['o'] = make_field(outputFile, "ROOT output file");
    zap['F'] = make_field(filter,     "filter")  = JPARSER::initialised();
    zap['d'] = make_field(debug)                 = 2;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }


  struct JGraph_t {

    void put(const Double_t x, const Double_t y)
    {
      X.push_back(x);
      Y.push_back(y);
    }

    vector<double> X;
    vector<double> Y;
  };


  map<JKey_t, JGraph_t> data;

  set<int> strings;
  set<int> floors;

  long long int counter = 0;

  for (inputFile.rewind(); inputFile.hasNext(); ++counter) {

    STATUS(setw(10) << counter << '\r'); DEBUG(endl);

    JDatalog* p  = inputFile.next();

    if (filter.count(p->parameter) == 0) {
    
      JGraph_t& g1 = data[JKey_t(p->string, p->floor, p->parameter)];

      g1.put(p->getTime(), p->value);

      strings.insert(p->string);
      floors .insert(p->floor);
    }
  }
  STATUS(endl);


  TFile out(outputFile.c_str(), "recreate");

  for (map<JKey_t, JGraph_t>::iterator i = data.begin(); i != data.end(); ++i) {

    TGraph g1(i->second.X.size(), i->second.X.data(), i->second.Y.data());

    g1.SetName(i->first.toString().c_str());

    const JRange<double> range(i->second.Y.begin(), i->second.Y.end());

    g1.SetMinimum(range.getLowerLimit());
    g1.SetMaximum(range.getUpperLimit());

    out << g1;
  }

  ostringstream os;

  os << "set_variable NUMBER_OF_STRINGS " << setw(4) <<  strings.size()   << ";" << endl;
  os << "set_variable NUMBER_OF_FLOORS  " << setw(4) <<  floors. size()   << ";" << endl;
  if (!strings.empty()) {
    os << "set_variable FIRST_STRING      " << setw(4) << *strings. begin() << ";" << endl;
    os << "set_variable LAST_STRING       " << setw(4) << *strings.rbegin() << ";" << endl;
  }
  if (!floors.empty()) {
    os << "set_variable FIRST_FLOOR       " << setw(4) << *floors.  begin() << ";" << endl;
    os << "set_variable LAST_FLOOR        " << setw(4) << *floors. rbegin() << ";" << endl;
  }
  os << "set_array    STRINGS           ";
  copy(strings.begin(), strings.end(), ostream_iterator<int>(os, " "));
  os << endl;

  TNamed meta("TUNA", os.str().c_str());

  out << meta;

  out.Write();
  out.Close();
}