JMergeCalibrateToT.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <map>
 
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TKey.h"
#include "TString.h"
#include "TMath.h"
 
#include "km3net-dataformat/online/JDAQHeader.hh"
 
#include "JROOT/JRootFileReader.hh"
#include "JROOT/JRootFileWriter.hh"
#include "JSupport/JMeta.hh"
 
#include "JCalibrate/JCalibrateToT.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \author mdejong
 * Auxiliary program to merge time-over-threshold data.
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;
 
  vector<string>     inputFile;
  string             outputFile;
  int                debug;
 
  try { 
 
    JParser<> zap("Auxiliary program to merge time-over-threshold data.");
    
    zap['f'] = make_field(inputFile,         "input file (output from JCalibrateToT).");
    zap['o'] = make_field(outputFile,        "output file (input to JFitToT).")   = "merge-tot.root";
    zap['d'] = make_field(debug,             "debug.")                            = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  gErrorIgnoreLevel = kError;
 
  typedef map<TString, TH1*>  map_type;
 
  map_type zmap;
 
 
  for (vector<string>::const_iterator i = inputFile.begin(); i != inputFile.end(); ++i) {
 
    DEBUG("Processing " << *i << endl) ;
 
    TFile in(i->c_str(), "read");
    
    TIter iter(in.GetListOfKeys());
 
    for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {
 
      if (TString(key->GetName()).EndsWith(_2SToT)) {
 
        TH2* h2 = dynamic_cast<TH2*>(key->ReadObj());
 
        map_type::iterator p = zmap.find(h2->GetName());
 
        if (p == zmap.end()) {
 
          DEBUG("Clone " << h2->GetName() << endl);
 
          p = zmap.insert(make_pair(h2->GetName(), (TH2*) h2->Clone())).first;
 
        } else {
 
          DEBUG("Add   " << h2->GetName() << endl);
 
          p->second->Add(h2);
        }
      }
    }
    
    for (map_type::iterator i = zmap.begin(); i != zmap.end(); ++i) {
      i->second->SetDirectory(0);
    }
    
    in.Close();
  }
 
  // write file
 
  TFile out(outputFile.c_str(), "recreate");
 
  for (map_type::iterator i = zmap.begin(); i != zmap.end(); ++i) {
 
    if (i->first.EndsWith(_2SToT)) {
 
      TH2D* h2s = (TH2D*) i->second;
 
      h2s->Write();
    } 
  }
 
  putObject(out, JMeta(argc, argv));
 
  for (vector<string>::const_iterator i = inputFile.begin(); i != inputFile.end(); ++i) {
 
    JMeta::copy(i->c_str(), out);
 
    for (JRootFileReader<JDAQHeader> in(i->c_str()); in.hasNext(); ) {
      putObject(out, *in.next());
    }
  }
 
  out.Write();
  out.Close();
}