JTriggerReprocessor.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <iomanip>
 
#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/MultiHead.hh"
#include "km3net-dataformat/offline/Evt.hh"
#include "km3net-dataformat/offline/Hit.hh"
#include "km3net-dataformat/online/JDAQ.hh"
#include "km3net-dataformat/online/JDAQClock.hh"
 
#include "JDAQ/JDAQTimesliceIO.hh"
#include "JDAQ/JDAQEventIO.hh"
#include "JDAQ/JDAQSummarysliceIO.hh"
#include "JDAQ/JDAQEvaluator.hh"
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JModuleRouter.hh"
 
#include "JTrigger/JHit.hh"
#include "JTrigger/JHitToolkit.hh"
#include "JTrigger/JSuperFrame1D.hh"
#include "JTrigger/JSuperFrame2D.hh"
#include "JTrigger/JTimeslice.hh"
#include "JTrigger/JHitL0.hh"
#include "JTrigger/JHitL1.hh"
#include "JTrigger/JBuildL1.hh"
#include "JTrigger/JBuildL2.hh"
#include "JTrigger/JTrigger3DShower.hh"
#include "JTrigger/JTriggerMXShower.hh"
#include "JTrigger/JTrigger3DMuon.hh"
#include "JTrigger/JTriggerNB.hh"
#include "JTrigger/JTriggerBits.hh"
#include "JTrigger/JEventOverlap.hh"
#include "JTrigger/JTimesliceRouter.hh"
#include "JTrigger/JTriggeredEvent.hh"
#include "JTrigger/JTimesliceL1.hh"
#include "JTrigger/JTriggerParameters.hh"
#include "JTrigger/JTriggerToolkit.hh"
 
#include "JLang/JPipe.hh"
#include "JROOT/JROOTClassSelector.hh"
 
#include "JSupport/JSingleFileScanner.hh"
#include "JSupport/JTreeScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JSupport.hh"
#include "JSupport/JMeta.hh"
#include "JSupport/JTriggerParametersSupportkit.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 * Auxiliary program to re-trigger KM3NETDAQ::JDAQEvent data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;
 
  typedef JAllTypes_t  typelist;
 
  JSingleFileScanner<JDAQEvent> inputFile;
  JFileRecorder<typelist>       outputFile;
  JLimit_t&           numberOfEvents = inputFile.getLimit();
  string              detectorFile;
  JTriggerParameters  parameters;
  bool                reuse_parameters;
  bool                snapshot;
  JROOTClassSelection selection      = getROOTClassSelection<typelist>();
  int                 debug;
 
  try { 
 
    JParser<> zap("Auxiliary program to re-trigger event data.");
 
    zap['f'] = make_field(inputFile,        "input file.");
    zap['o'] = make_field(outputFile,       "output file.")                                 = "trigger_reprocessor.root";
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['a'] = make_field(detectorFile,     "detector file.");
    zap['@'] = make_field(parameters,       "trigger parameters")                           = JPARSER::initialised();
    zap['U'] = make_field(reuse_parameters, "reuse trigger parameters from input file.");
    zap['S'] = make_field(snapshot,         "use snapshot hits instead of triggered hits.");
    zap['C'] = make_field(selection,        "selection of data types for output.")          = JPARSER::initialised();
    zap['d'] = make_field(debug,            "debug flag.")                                  = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  setDAQLongprint(debug >= JEEP::debug_t);
 
  JDetector detector;
 
  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
 
  if (reuse_parameters) {
 
    try {
 
      parameters = getTriggerParameters(inputFile);
 
      NOTICE("Set trigger parameters from input." << endl);
    }
    catch(const JException& error) {
      FATAL("No trigger parameters from input." << endl);
    }
  }
 
  // detector
 
  if (parameters.disableHighRateVeto) {
 
    NOTICE("Disabling high-rate veto of all PMTs." << endl);
 
    detector.setPMTStatus(HIGH_RATE_VETO_DISABLE);
  }
  
  parameters.set(getMaximalDistance(detector));
 
  parameters.triggerNB.write.prescale = 1;
 
  DEBUG("Frame time [ms] " << getFrameTime() * 1e-6 << endl);
  DEBUG("Reset time [ms] " << getRTS()       * 1e-6 << endl);
  DEBUG("Trigger"          << endl << parameters    << endl);
 
  const JModuleRouter moduleRouter(detector);
 
  if (parameters.writeSummary()) { WARNING("Discard writeSummary option during reprocesing of data." << endl); }
  if (parameters.writeL1())      { WARNING("Discard writeL1 option during reprocesing of data."      << endl); }
  if (parameters.writeL2())      { WARNING("Discard writeL2 option during reprocesing of data."      << endl); }
  if (parameters.writeSN())      { WARNING("Discard writeSN option during reprocesing of data."      << endl); }
 
  //typedef JHit        hit_type;
  //typedef int         hit_type;
  typedef double        hit_type;
 
  typedef JSuperFrame1D<hit_type>   JSuperFrame1D_t;
  typedef JSuperFrame2D<hit_type>   JSuperFrame2D_t;
  typedef JTimeslice   <hit_type>   JTimeslice_t;
  typedef JBuildL1     <hit_type>   JBuildL1_t;
  typedef JBuildL2     <hit_type>   JBuildL2_t;
 
  const JBuildL1_t buildL1(parameters);
  const JBuildL2_t buildL2(parameters.L2);
  const JBuildL2_t buildSN(parameters.SN);
  const JBuildL2_t buildNB(parameters.NB);
 
  JTimesliceRouter timesliceRouter(parameters.numberOfBins);
 
  const JTriggerNB       triggerNB      (parameters);
  const JTrigger3DMuon   trigger3DMuon  (parameters);
  const JTrigger3DShower trigger3DShower(parameters);
  const JTriggerMXShower triggerMXShower(parameters, detector);
 
 
  outputFile.open();
 
  if (!outputFile.is_open()) {
    FATAL("Error opening file " << outputFile << endl);
  }
 
  outputFile.put(JMeta(argc, argv));
  outputFile.put(parameters);
 
 
  JTreeScanner<JDAQSummaryslice, JDAQEvaluator>::debug = debug;
  
  JTreeScanner<JDAQSummaryslice, JDAQEvaluator> scan(inputFile);
 
  if (scan.getEntries() == 0) {
    FATAL("No summary data." << endl);
  }
  
  while (inputFile.hasNext()) {
 
    STATUS("event: " << setw(10) << inputFile.getCounter() << '\r'); DEBUG(endl);
 
    JDAQEvent*        evt     = inputFile.next();
    Long64_t          index   = scan.find(*evt);
    JDAQSummaryslice* summary = scan.getEntry(index);
 
    if (evt->getFrameIndex() != summary->getFrameIndex()) {
      DEBUG("Frame indices " << evt->getFrameIndex() << " != " << summary->getFrameIndex() << endl);
    }
    
    JDAQTimeslice timeslice(*evt, *summary, snapshot);
 
    DEBUG(timeslice << endl);
 
    timesliceRouter.configure(timeslice);
 
 
    JTimeslice_t    timesliceL0(timeslice.getDAQChronometer());
    JTimeslice_t    timesliceL1(timeslice.getDAQChronometer());
    JTimeslice_t    timesliceL2(timeslice.getDAQChronometer());
    JTimeslice_t    timesliceSN(timeslice.getDAQChronometer());
    JTimeslice_t    timesliceNB(timeslice.getDAQChronometer());
    
    for (JDAQTimeslice::const_iterator super_frame = timeslice.begin(); super_frame != timeslice.end(); ++super_frame) {
 
      if (moduleRouter.hasModule(super_frame->getModuleID())) {
 
        // calibration
      
        const JModule&   module = moduleRouter.getModule(super_frame->getModuleID()); 
        JSuperFrame2D_t& buffer = JSuperFrame2D_t::demultiplex(*super_frame, module);
      
        // Apply high-rate veto
 
        buffer.applyHighRateVeto(parameters.highRateVeto_Hz);
 
        // L0
      
        timesliceL0.push_back(JSuperFrame1D_t(buffer));
 
        // Nano-beacon trigger
 
        if (parameters.triggerNB.enabled) {
 
          JSuperFrame2D_t::iterator __end = partition(buffer.begin(), buffer.end(), parameters.triggerNB.pmts);
 
          if (buffer.begin() != __end) {
 
            timesliceNB.push_back(JSuperFrame1D_t(super_frame->getDAQChronometer(),
                                                  super_frame->getModuleIdentifier(),
                                                  module.getPosition()));
 
 
            JSuperFrame1D_t zbuf;
 
            buildL1(buffer.begin(), __end, back_inserter(zbuf));
 
            buildNB(buffer.begin(), __end, zbuf, back_inserter(*timesliceNB.rbegin()));
          }
        }
 
        // L1
      
        timesliceL1.push_back(JSuperFrame1D_t(super_frame->getDAQChronometer(),
                                              super_frame->getModuleIdentifier(),
                                              module.getPosition()));
      
        buildL1(*timesliceL0.rbegin(), back_inserter(*timesliceL1.rbegin()));
 
        // L2
 
        timesliceL2.push_back(JSuperFrame1D_t(super_frame->getDAQChronometer(),
                                              super_frame->getModuleIdentifier(),
                                              module.getPosition()));
 
        buildL2(buffer, *timesliceL1.rbegin(), back_inserter(*timesliceL2.rbegin()));
 
        // SN
        {
          JTimeslice_t::value_type tv(super_frame->getDAQChronometer(),
                                      super_frame->getModuleIdentifier(),
                                      module.getPosition());
 
          buildSN(buffer, *timesliceL1.rbegin(), back_inserter(tv));
 
          if (!tv.empty()) {
            timesliceSN.push_back(tv);
          }
        }
      }
    }
 
 
    // Trigger
 
    if (parameters.triggerNB.enabled) {
 
      const JTriggerInput  trigger_input(timesliceNB);
 
      for (JTriggerInput::const_iterator hit = trigger_input.begin(); hit != trigger_input.end(); ++hit) {
 
        if (parameters.triggerNB.write()) {
 
          JTriggeredEvent tev(timesliceNB.getDAQChronometer(),
                              getTriggerMask(triggerNB.getTriggerBit()),
                              *hit,
                              timesliceRouter,
                              moduleRouter,
                              parameters.TMaxLocal_ns,
                              parameters.triggerNB.DMax_m,
                              getTimeRange(parameters.triggerNB));
 
          outputFile.put(tev);
        }
      }
    }
 
    JTriggerInput  trigger_input(timesliceL2);
    JTriggerOutput trigger_output;
 
    trigger3DMuon  (trigger_input, back_inserter(trigger_output));
    trigger3DShower(trigger_input, back_inserter(trigger_output));
    triggerMXShower(trigger_input, timesliceL0, back_inserter(trigger_output));
 
    trigger_output.merge(JEventOverlap(parameters.TMaxEvent_ns));
 
    DEBUG("Number of triggers: " << trigger_output.size() << endl);
 
    for (JTriggerOutput::const_iterator event = trigger_output.begin(); event != trigger_output.end(); ++event) {
 
      JTriggeredEvent tev(*event, 
                          timesliceRouter, 
                          moduleRouter, 
                          parameters.TMaxLocal_ns, 
                          getTimeRange(parameters));
 
      tev.setCounter(evt->getCounter());
 
      outputFile.put(tev);
    }
  }
  STATUS(endl);
 
  JSingleFileScanner<typelist> io(inputFile);
  
  selection.remove<JDAQEvent>();
  
  io | JValve<typelist>(selection) | outputFile;
 
  outputFile.close();
}