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 "JDAQ/JDAQSummarysliceIO.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 "JReconstruction/JEvt.hh"
#include "JReconstruction/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, JDAQSummaryslice, 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<JTYPELIST<JDAQSummaryslice, JMeta>::typelist> io(inputFile);

  io >> outputFile;

  outputFile.close();
}