JTuna2Graph.cc File Reference

Auxiliary program to convert ROOT TTree with slow control data to ROOT TGraph's. More...

#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"

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Detailed Description

Auxiliary program to convert ROOT TTree with slow control data to ROOT TGraph's.

Author

mdejong

Definition in file JTuna2Graph.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 115 of file JTuna2Graph.cc.

{
  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();
}