JConvertEvt.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <iomanip>
#include <algorithm>
 
#include "TROOT.h"
#include "TFile.h"
 
#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"
#include "km3net-dataformat/offline/io_ascii.hh"
#include "km3net-dataformat/offline/io_online.hh"
 
#include "JAAnet/JHead.hh"
#include "JAAnet/JHeadToolkit.hh"
 
#include "JDAQ/JDAQEventIO.hh"
 
#include "JSupport/JParallelFileScanner.hh"
#include "JSupport/JTreeScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
#include "JSupport/JSupportToolkit.hh"
#include "JSupport/JSupport.hh"
#include "JSupport/JMeta.hh"
 
#include "JFit/JEvt.hh"
#include "JFit/JEvtToolkit.hh"
 
#include "Jeep/JeepToolkit.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \author mdejong
 */
 
/**
 * Auxiliary program to convert fit results to AAnet format.
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  JMultipleFileScanner<>                               inputFile;
  JFileRecorder<JTYPELIST<Head, Evt, JMeta>::typelist> outputFile;
  JLimit_t&              numberOfEvents = inputFile.getLimit();
  JQualitySorter         qualitySorter;
  size_t                 numberOfFits;
  int                    debug;
 
  try {
 
    JParser<> zap("Auxiliary program to convert fit results to AAnet format.\
                  \nThe option -L corresponds to the names of a shared library \
                  \nand function so to rearrange the order of fit results.");
 
    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile)          = "aanet.root";
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['L'] = make_field(qualitySorter)       = JPARSER::initialised();
    zap['N'] = make_field(numberOfFits)        = 0;
    zap['d'] = make_field(debug)               = 2;
 
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }
 
  using namespace KM3NETDAQ;
 
  typedef JParallelFileScanner< JTypeList<JDAQEvent, JEvt> >  JParallelFileScanner_t;
  typedef JParallelFileScanner_t::multi_pointer_type          multi_pointer_type;
 
  outputFile.open();
 
  Head header;
 
  try {
    header = getHeader(inputFile);
  } catch(const exception& error) {}
  {
    JHead buffer(header);
 
    buffer.DAQ.livetime_s = getLivetime(inputFile.begin(), inputFile.end());
    buffer.push(&JHead::DAQ);
    
    copy(buffer, header);
  }
  
  outputFile.put(header);
 
 
  outputFile.put(JMeta(argc, argv));
 
 
  counter_type number_of_events = 0;
 
  for (JMultipleFileScanner<>::const_iterator i = inputFile.begin(); i != inputFile.end(); ++i) {
    
    STATUS("Processing: " << *i << endl);
    
    JParallelFileScanner_t in(*i);
    JTreeScanner<Evt>      mc(*i);
 
    Vec center(0,0,0);
 
    int mc_run_id = 0;
    
    try {
 
      const JHead head(getHeader(*i));
    
      center    = get<Vec>(head);
      mc_run_id = head.start_run.run_id;
      
    } catch(const exception& error) {}
 
    
    while (in.hasNext()) {
 
      STATUS("event: " << setw(10) << in.getCounter() << '\r'); DEBUG(endl);
 
      multi_pointer_type ps = in.next();
 
      JDAQEvent* tev   = ps;
      JEvt*      evt   = ps;
 
      JEvt::iterator __end = evt->end();
      
      if (numberOfFits > 0) {
        advance(__end = evt->begin(), min(numberOfFits, evt->size()));
      }
 
      if (qualitySorter.is_valid()) {
        partial_sort(evt->begin(), __end, evt->end(), qualitySorter);
      }
      
      Evt out;
 
      if (mc.getEntries() != 0) {
 
        Evt* event = mc.getEntry(tev->getCounter());
 
        if (event != NULL) {
 
          out = *event;
 
          if (out.mc_run_id == 0) {
            out.mc_run_id = mc_run_id;
          }
 
          for (vector<Trk>::iterator i = out.mc_trks.begin(); i != out.mc_trks.end(); ++i) {
            i->pos += center;
          }
        }
      }
      
      read(out, *tev);
      
      for (JEvt::const_iterator fit = evt->begin(); fit != __end; ++fit) {
        
        Trk tr;
        
        tr.id        = out.trks.size() + 1;
        tr.pos       = Vec(fit->getX(),  fit->getY(),  fit->getZ());
        tr.dir       = Vec(fit->getDX(), fit->getDY(), fit->getDZ());
        tr.t         = fit->getT();
        tr.E         = fit->getE();
        tr.lik       = fit->getQ();
 
        tr.rec_type  = JPP_RECONSTRUCTION_TYPE;
        
        for (JHistory::const_iterator i = fit->getHistory().begin(); i != fit->getHistory().end(); ++i) {
          tr.rec_stages.push_back(i->type);
        }
        
        for (int i = 0; i != fit->getN(); ++i) {
          tr.fitinf.push_back(fit->getW(i));
        }
        
        out.trks.push_back(tr);
      }
 
      out.id = ++number_of_events;
 
      outputFile.put(out);
    }
  }
 
  STATUS(endl);
 
  JMultipleFileScanner<JMeta> io(inputFile);
 
  io >> outputFile;
 
  outputFile.close();
}