JRandomTimesliceWriter.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <fstream>
 
#include "TRandom3.h"
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JPMTParametersMap.hh"
#include "JDetector/JK40DefaultSimulator.hh"
#include "JDetector/JPMTDefaultSimulator.hh"
#include "JDetector/JCLBDefaultSimulator.hh"
#include "JDetector/JK40RunByRunSimulator.hh"
#include "JDetector/JPMTRunByRunSimulator.hh"
#include "JDetector/JDetectorSimulator.hh"
#include "km3net-dataformat/online/JDAQClock.hh"
#include "JDAQ/JDAQSummarysliceIO.hh"
#include "JDAQ/JDAQTimesliceIO.hh"
#include "JTimeslice/JRandomTimeslice.hh"
#include "JTrigger/JSummaryRouter.hh"
#include "JTrigger/JRunByRun.hh"
#include "JSupport/JSingleFileScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JSupport.hh"
#include "JSupport/JMeta.hh"
 
#include "Jeep/JTimer.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 *
 * Auxiliary program to write KM3NETDAQ::JDAQTimeslice with random data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;
 
  JFileRecorder<JTYPELIST<JDAQTimesliceL0, JMeta, JRootTypes_t>::typelist> outputFile;
  string                 detectorFile;
  Long64_t               numberOfEvents;
  JDAQHit::JTDC_t        TCLB_ns;
  JPMTParametersMap      pmtParameters;
  JK40Rates              rates_Hz;
  int                    run;
  JRunByRun              runbyrun;
  UInt_t                 seed;
  int                    debug;
 
  try { 
 
    JParser<> zap("Auxiliary program to write time slices with random data.");
    
    zap['o'] = make_field(outputFile,          "output file")                                                      = "timeslice.root";
    zap['n'] = make_field(numberOfEvents)                     = JLimit::max();
    zap['a'] = make_field(detectorFile,        "detecto.");
    zap['R'] = make_field(run,                 "run number")  = -1;
    zap['r'] = make_field(runbyrun,            "option for run-by-run mode")                                       = JPARSER::initialised();
    zap['P'] = make_field(pmtParameters,       "PMT simulation data (or corresponding file name)")                 = JPARSER::initialised();
    zap['B'] = make_field(rates_Hz,            "background rates [Hz]")                                            = JPARSER::initialised();
    zap['T'] = make_field(TCLB_ns)             = 256;     // [ns]
    zap['S'] = make_field(seed,                "seed")        = 0;
    zap['d'] = make_field(debug,               "debug")       = 0;
    
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  gRandom->SetSeed(seed);
 
 
  cout.tie(&cerr);
 
  setDAQLongprint(debug >= JEEP::debug_t);
 
  if (pmtParameters.getQE() != 1.0) {
 
    WARNING("Correct background rates with global efficiency " << pmtParameters.getQE() << endl);
    
    rates_Hz.correct(pmtParameters.getQE());
  }
 
  DEBUG("PMT parameters: " << endl << pmtParameters << endl);
  DEBUG("K40 rates:      " << endl << rates_Hz      << endl);
 
  JDetector detector;
 
  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
 
  JPMTParametersMap::Throw(false);
  
  JDetectorSimulator  simbad(detector);
  JSummaryRouter      summaryRouter;
 
  if (runbyrun.is_valid()) {
 
    NOTICE("Using run-by-run:" << endl << runbyrun << endl);
 
    if (!runbyrun.hasNext()) {
      FATAL("Run-by-run simulation yields no input." << endl);
    }
 
    if (rates_Hz.getSinglesRate() != 0.0) {
      WARNING("Run-by-run simulation discards singles rate [Hz] " << rates_Hz.getSinglesRate() << endl);
    }
 
    try {
      simbad.reset(new JK40RunByRunSimulator(summaryRouter, rates_Hz));
      simbad.reset(new JPMTRunByRunSimulator(summaryRouter, pmtParameters, detector, runbyrun.range_Hz));
      simbad.reset(new JCLBDefaultSimulator());
    }
    catch(const JException& error) {
      FATAL(error.what() << endl);
    }
 
  } else {
 
    NOTICE("Using fixed rates [Hz]: " << rates_Hz << endl);
 
    try {
      simbad.reset(new JK40DefaultSimulator(rates_Hz));
      simbad.reset(new JPMTDefaultSimulator(pmtParameters, detector));
      simbad.reset(new JCLBDefaultSimulator());
    }
    catch(const JException& error) {
      FATAL(error.what() << endl);
    }
  }
 
 
  JTimer timerco("constructor");
  JTimer timerIO("I/O");
 
 
  outputFile.open();
 
  if (!outputFile.is_open()) {
    FATAL("Error opening file " << outputFile << endl);
  }
 
  outputFile.put(*gRandom);
  outputFile.put(JMeta(argc, argv));
 
  int count = 1;
 
  for ( ; count <= numberOfEvents; ++count) {
 
    STATUS("event: " << setw(10) << count << '\r'); DEBUG(endl);    
 
    int frame_index = count;
 
    if (runbyrun.hasNext()) {
 
      summaryRouter.update(runbyrun.next());
    
      summaryRouter.correct(dynamic_cast<const JPMTDefaultSimulatorInterface&>(simbad.getPMTSimulator()));
 
      frame_index = summaryRouter.getFrameIndex();
      run         = summaryRouter.getRunNumber();
    }
 
    const JDAQChronometer chronometer(detector.getID(), run, frame_index, JDAQUTCExtended(getTimeOfFrame(frame_index)));
 
    timerco.start();
 
    JRandomTimeslice timeslice(chronometer, simbad); 
 
    timerco.stop();
 
    timerIO.start();
      
    outputFile.put(timeslice);
 
    timerIO.stop();
  }
  STATUS(endl);
 
  if (debug >= debug_t && count > 1) {
 
    const double factor = 1.0 / (double) (count - 1);
 
    timerco.print(cout, factor);
    timerIO.print(cout, factor);
  }
 
  outputFile.put(*gRandom);
  outputFile.close();
}