examples/JTimeslice/JRandomTimesliceWriter.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
 
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TRandom3.h"
 
#include "km3net-dataformat/online/JDAQ.hh"
#include "km3net-dataformat/online/JDAQClock.hh"
#include "km3net-dataformat/online/JDAQTimeslice.hh"
#include "JTimeslice/JRandomTimeslice.hh"
#include "JROOT/JManager.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 *
 * Example program to test generation of KM3NETDAQ::JDAQTimeslice with random data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;
 
  string                 outputFile;
  Long64_t               numberOfSlices;
  double                 rate_Hz;
  pair<size_t, double>   recycling;
  UInt_t                 seed;
  int                    debug;
 
  try { 
 
    JParser<> zap("Example program to test generation of time slices with random data.");
    
    zap['o'] = make_field(outputFile,          "output file")                                          = "timeslice.root";
    zap['n'] = make_field(numberOfSlices);
    zap['B'] = make_field(rate_Hz,             "background rate [Hz]");
    zap['N'] = make_field(recycling,           "number of recycles / time interval for sampling data") = make_pair(0, 0.0);
    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);
 
  const JDAQHit::JPMT_t  PMT_L0[] =  { 0, 1 };
 
  JManager<size_t, TH1D> H0(new TH1D("h0[%]", NULL, 100, 0.0, getFrameTime()));
 
  for (Long64_t counter = 0; counter != numberOfSlices; ++counter) {
 
    STATUS("slice: " << setw(10) << counter << '\r'); DEBUG(endl);
 
    JRandomTimeslice timeslice;
 
    timeslice.resize(1);
 
 
    vector<JDAQHit> buffer;
 
    if (rate_Hz > 0.0) {
 
      double period_ns = 1.0e9 / (NUMBER_OF_PMTS * rate_Hz);
 
      for (double t1 = 0.0 * getFrameTime() + gRandom->Exp(period_ns); t1 < 0.5 * getFrameTime(); t1 += gRandom->Exp(period_ns)) {
        buffer.push_back(JDAQHit(PMT_L0[0], t1, 0));
      }
 
      period_ns *= 2.0;
 
      for (double t1 = 0.5 * getFrameTime() + gRandom->Exp(period_ns); t1 < 1.0 * getFrameTime(); t1 += gRandom->Exp(period_ns)) {
        buffer.push_back(JDAQHit(PMT_L0[1], t1, 0));
      }
    }
 
    timeslice[0].add(buffer.size(), buffer.data());
 
    TH1D* h0 = H0[0];
 
    for (JDAQSuperFrame::const_iterator hit = timeslice[0].begin(); hit != timeslice[0].end(); ++hit) {
      h0->Fill((Double_t) hit->getT());
    }
 
    for (size_t i = 1; i <= recycling.first; ++i) {
      
      timeslice.recycle(recycling.second);
 
      TH1D* h0 = H0[i];
 
      for (JDAQSuperFrame::const_iterator hit = timeslice[0].begin(); hit != timeslice[0].end(); ++hit) {
        h0->Fill((Double_t) hit->getT());
      }
    }
  }
  STATUS(endl);
  
 
  TFile out(outputFile.c_str(), "recreate");
 
  out << H0;
 
  out.Write();
  out.Close();
}