JRandomToAWriter.cc

Go to the documentation of this file.

#include <string>
#include <iostream>

#include "TRandom3.h"

#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"

#include "JAcoustics/JToA.hh"
#include "JAcoustics/JSupport.hh"
#include "JAcoustics/JReceiver.hh"

#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMeta.hh"

#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

/**
 * \file
 * Auxiliary program to write random acoustic data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef JFileRecorder<JTYPELIST<JToA, JMeta>::typelist>  JFileRecorder_t;

  JFileRecorder_t  outputFile;
  string           detectorFile;
  double           rate_Hz;
  double           T_s;
  int              run;
  int              id;
  UInt_t           seed;
  int              debug;

  try {

    JParser<> zap(" Auxiliary program to write random acoustic data.");

    zap['o'] = make_field(outputFile,          "output file");
    zap['a'] = make_field(detectorFile,        "detector.");
    zap['R'] = make_field(run,                 "run number")  = -1;
    zap['B'] = make_field(rate_Hz,             "background rate  [Hz]");
    zap['T'] = make_field(T_s,                 "run duration [s]");
    zap['W'] = make_field(id,                  "waveform identifier");
    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);
  }

  if (rate_Hz <= 0.0) {
    FATAL("Invalid rate [Hz] " << rate_Hz << endl);
  }

  const double t_s = 1.0 / rate_Hz;

  gRandom->SetSeed(seed);

  JDetector detector;

  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }

  outputFile.open();

  outputFile.put(JMeta(argc, argv));

  JToA toa;

  toa.DETID                 = detector.getID();
  toa.RUN                   = run;
  toa.WAVEFORMID            = id;
  toa.QUALITYFACTOR         = 2.0e3;
  toa.QUALITYNORMALISATION  = 0.0e3;
  
  for (JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {

    if (module->getFloor() != 0) {

      const JReceiver receiver(module->getID(), module->getPosition(), module->getT0() * 1.0e-9);
    
      for (double t1 = 0.0 + gRandom->Exp(t_s); t1 < T_s; t1 += gRandom->Exp(t_s)) {

        toa.DOMID        = receiver.getID();
        toa.TOA_NS       = llrint(1E9*receiver.putT(t1));

        outputFile.put(toa);
      }
    }
  }
  
  outputFile.close();
  
  return 0;
}