JKexing.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>
 
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
#include "TF1.h"
 
#include "JSupernova.hh"
 
#include "JDAQ/JDAQClock.hh"
#include "JDAQ/JDAQEvaluator.hh"
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
 
#include "JGizmo/JManager.hh"
 
#include "JROOT/JROOTClassSelector.hh"
#include "JTools/JQuantile.hh"
 
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JTreeScanner.hh"
#include "JSupport/JAutoTreeScanner.hh"
#include "JSupport/JSupport.hh"
 
#include "JLang/JObjectMultiplexer.hh"
#include "JMath/JMathToolkit.hh"
 
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
using namespace std;
using namespace JPP;
using namespace KM3NETDAQ;
using namespace JSUPERNOVA;
 
/**
 * \author mlincett
 * \file
 * Example application to test supernova trigger on run files.
 */
int main(int argc, char **argv)
{
 
  typedef JRange<int>    JRange_t;
 
  JMultipleFileScanner<> inputFile;
  JLimit_t&              numberOfEvents = inputFile.getLimit();
  string                 outputFile;
  string                 detectorFile;
  double                 TMax_ns;
  double                 TVeto_ns;
  JROOTClassSelector     selector;
  int                    debug;
  int                    preTriggerThreshold;
  JRange_t               M;
  
 
  // Check input parameters
 
  try { 
 
    JParser<> zap("Example program test supernova triggers.");
    
    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile)          = "jsn.root";
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['a'] = make_field(detectorFile);
    zap['T'] = make_field(TMax_ns)             = 10.0;
    zap['V'] = make_field(TVeto_ns)            = 1000.0;  
    zap['C'] = make_field(selector)            = getROOTClassSelection<JDAQTimesliceTypes_t>();
    zap['d'] = make_field(debug)               = 1;
    zap['M'] = make_field(M)                  = JRange<int>(6,10);
    zap['t'] = make_field(preTriggerThreshold) = 4;
                            
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  cout.tie(&cerr);
 
  JDetector detector;
 
  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
 
  // Configure input streams
 
  JTreeScanner<>::debug = debug;
 
  JAutoTreeScanner<JDAQTimeslice, JDAQEvaluator> zmap = JType<JDAQTimesliceTypes_t>();
 
  JTreeScannerInterface<JDAQTimeslice>* pts = zmap[selector]; 
 
  pts->configure(inputFile);
  
  // Configure routers
 
  const JModuleRouter moduleRouter(detector);
  
  const JDAQHitRouter hitRouter(detector);
  
  // -----------------------------------
  // STEP 1: building vetoes from events
  // -----------------------------------
 
  TH1D* h_vtr = new TH1D("VetoTimeRange","VetoTimeRange", 10000, 0, 10000);
 
  JTreeScanner<JDAQEvent, JDAQEvaluator> evIn(inputFile);
 
  map<int, JVetoSet > triggeredEvents; 
 
  for (; evIn.hasNext(); ) {
 
    STATUS("event: " << setw(10) << evIn.getCounter() << '\r'); DEBUG(endl);
    
    JDAQEvent* event = evIn.next();
 
    JVeto vp(*event, hitRouter);
 
    triggeredEvents[event->getFrameIndex()].push_back(vp);
 
    h_vtr->Fill(vp.getLength());
 
  } 
 
  STATUS(triggeredEvents.size() << " JDAQEvents loaded in veto buffer." << endl);
 
  //--------------------------------
  // STEP 2: timeslice processing
  // -------------------------------
 
  // 0 -> no filter
  // 1 -> count once per track
  // 2 -> suppress track
  // 3 -> suppress based on trigger veto
  
  // Output data structures
  
  typedef JManager<int, TH1D>  JManager_t;
 
  JManager_t SNTH(new TH1D("SNT_F%", NULL, 100, 0.0, 100));
 
  JManager_t MUL(new TH1D("MUL_F%", NULL, 1 + 31, -0.5, 31 + 1 - 0.5));
 
  const int nStages = 5;
  vector<vector<int> > trgHistory(nStages);
 
  // Loop
 
  counter_type counter = 0;
 
  for ( ; pts->hasNext() && counter != inputFile.getLimit(); ++counter) {
 
    STATUS("timeslice: " << setw(10) << counter << '\r'); DEBUG(endl);
 
    const JDAQTimeslice* timeslice = pts->next();
 
    int fIndex = timeslice->getFrameIndex();
 
    JDataSN preTrigger(TMax_ns, preTriggerThreshold); 
 
    preTrigger(timeslice, moduleRouter);
 
    JTriggerSN trigger(M, TVeto_ns);
 
    trigger(preTrigger);
 
    // generate and configure event-based veto
 
    JVetoSet veto;
    
    if (triggeredEvents.count(fIndex)) { 
      veto = triggeredEvents.at(fIndex);
    }
 
    trigger.setVeto(veto);
    
    // count trigger
 
    JSNFilterM trgOp(M);
 
    int rawCount = count_if(preTrigger.begin(), preTrigger.end(), trgOp);
 
    vector<int> trgCount = trigger.benchmark();
 
    trgHistory[0].push_back(rawCount);
    
    for (int i = 0; i < 3; i++) {
      trgHistory[i + 1].push_back(trgCount[i]);
    }
 
    trgHistory[4].push_back(trigger.getModules().size());
 
  } 
 
  //-------------------------------------
  // STEP 3: generating trigger summaries
  // ------------------------------------
  
  for (int i = 0; i < nStages; i++) {
    for (unsigned j = 0; j < trgHistory[i].size(); j++) {
      SNTH[i]->Fill(trgHistory[i][j]);
    }
  }
 
  if (outputFile != "") {
 
    TFile out(outputFile.c_str(), "RECREATE");
 
    h_vtr->Write();
    SNTH.Write(out);
    MUL.Write(out);
 
    out.Close();
 
  }
 
}