JBillabong.cc File Reference

Main program to trigger acoustic data. More...

#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
#include <chrono>
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JModuleSupportkit.hh"
#include "JLang/JPredicate.hh"
#include "JLang/JComparator.hh"
#include "JTools/JHashMap.hh"
#include "JTools/JAbstractHistogram.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMeta.hh"
#include "JGeometry3D/JCylinder3D.hh"
#include "JGeometry3D/JOmega3D.hh"
#include "JGeometry3D/JRotator3D.hh"
#include "JAcoustics/JToA.hh"
#include "JAcoustics/JTransmission.hh"
#include "JAcoustics/JReceiver.hh"
#include "JAcoustics/JSoundVelocity.hh"
#include "JAcoustics/JAcousticsToolkit.hh"
#include "JAcoustics/JEvent.hh"
#include "JAcoustics/JEventOverlap.hh"
#include "JAcoustics/JSupport.hh"
#include "Jeep/JProperties.hh"
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "JTrigger/JMatch.hh"
#include "JTrigger/JAlgorithm.hh"
#include "JTrigger/JBind2nd.hh"

Go to the source code of this file.

Classes
struct	JACOUSTICS::hit_type
	Acoustic hit. More...
class	JACOUSTICS::JMatch3D
	3D match criterion for acoustic signals. More...
class	JACOUSTICS::JMatch1D
	1D match criterion for acoustic signals. More...
struct	JACOUSTICS::JCheck
	Auxiliary data structure for final check on event. More...

Namespaces
	JACOUSTICS
	Auxiliary classes and methods for acoustic position calibration.

Typedefs
typedef JAbstractHistogram < double >	JACOUSTICS::JHistogram_t
	Type definition for scan along axis. More...

Functions
void	JACOUSTICS::print (std::ostream &out, const JEvent &event)
	Print event. More...
int	main (int argc, char **argv)

Detailed Description

Main program to trigger acoustic data.

An acoustic event is based on coincidences between times of arrival.
If the number of coincident times of arrival exceeds a preset minimum, the event is triggered and subsequently stored in the output file.

Author

mdejong

Definition in file JBillabong.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 315 of file JBillabong.cc.

{
  using namespace std;
  using namespace JPP;
  
  typedef JTYPELIST<JEvent, TH1D, JMeta>::typelist  typelist;

  JMultipleFileScanner<JToA> inputFile;
  JLimit_t&                  numberOfEvents = inputFile.getLimit();

  struct {
    int                      factoryLimit   = 100000;
    int                      numberOfHits   =   0;
    double                   Q              =   0.0;
    double                   TMax_s         =   0.0;
    double                   DMax_m         = 500.0;
    double                   ZMax_m         = 200.0;
    double                   TMaxExtra_s    = 100.0e-6; 
    double                   TMaxVertex_s   =   1.0e-2;
    double                   gridAngle_deg  =  10.0;
  } parameters;

  JHistogram_t               X;
  JHistogram_t               Y;
  JHistogram_t               Z;
  JFileRecorder<typelist>    outputFile;
  JSoundVelocity             V = getSoundVelocity;   // default sound velocity
  string                     detectorFile;
  int                        waveform;
  int                        debug;

  try {

    JProperties properties;

    properties.insert(gmake_property(parameters.factoryLimit));
    properties.insert(gmake_property(parameters.numberOfHits));
    properties.insert(gmake_property(parameters.Q));
    properties.insert(gmake_property(parameters.TMax_s));
    properties.insert(gmake_property(parameters.DMax_m));
    properties.insert(gmake_property(parameters.ZMax_m));
    properties.insert(gmake_property(parameters.TMaxExtra_s));
    properties.insert(gmake_property(parameters.TMaxVertex_s));
    properties.insert(gmake_property(parameters.gridAngle_deg));
    
    JParser<> zap("Main program to trigger acoustic data.");
    
    zap['f'] = make_field(inputFile);
    zap['n'] = make_field(numberOfEvents)                                    = JLimit::max();    
    zap['@'] = make_field(properties,   "trigger parameters")                = JPARSER::initialised();
    zap['X'] = make_field(X,            "x-scan");
    zap['Y'] = make_field(Y,            "y-scan");
    zap['Z'] = make_field(Z,            "z-scan");
    zap['o'] = make_field(outputFile,   "output file")                       = "event.root";
    zap['V'] = make_field(V,            "sound velocity")                    = JPARSER::initialised();
    zap['a'] = make_field(detectorFile, "detector file");
    zap['W'] = make_field(waveform,     "waveform identifier");
    zap['d'] = make_field(debug)                                             = 1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  JDetector detector;

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

  V.set(detector.getUTMZ());
  
  const JCylinder3D   cylinder(detector.begin(), detector.end());
  
  const double v0 = V(cylinder.getZmax());

  if (parameters.TMax_s < parameters.DMax_m / v0) {

    parameters.TMax_s = parameters.DMax_m / v0;

    NOTICE("Set maximal time [s] " << FIXED(7,3) << parameters.TMax_s << endl);
  }
  
  const JMatch3D      match3D(v0, parameters.DMax_m, parameters.TMaxExtra_s);
  const JMatch1D      match1D(v0, parameters.DMax_m, parameters.ZMax_m, parameters.TMaxExtra_s);

  const JEventOverlap overlap(parameters.TMax_s);

  const JCheck        check(X, Y, Z, v0, parameters.TMaxVertex_s, parameters.numberOfHits);

  const JOmega3D      omega(JAngle3D(0.0, 0.0), JOmega3D::range_type(0.0, 0.5*PI), parameters.gridAngle_deg * PI / 180.0);
  const JRotator3D    rotator(omega);
  

  JHashMap<int, JReceiver>   receivers;

  const JPosition3D   center(cylinder.getX(), cylinder.getY(), 0.5 * (cylinder.getZmin() + cylinder.getZmax()));

  for (JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {

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

      receivers[module->getID()] = JReceiver(module->getID(), 
                                             module->getPosition() + getPiezoPosition() - center,
                                             module->getT0() * 1.0e-9);
    }
  }

  TH1D h0("h0", NULL, 21, -0.5, 20.5);

  vector<TH1D*> M1(10, NULL);

  for (size_t i = 2; i <= 5; ++i) {
    M1[i] = new TH1D(MAKE_CSTRING("M[" << i << "]"), NULL, 1001, -0.5, 1000.5);
  }
      
  outputFile.open();

  if (!outputFile.is_open()) {
    FATAL("Error opening file " << outputFile << endl);
  }

  outputFile.put(JMeta(argc, argv));
  
  // input data

  typedef vector<hit_type> buffer_type;          // data type

  buffer_type data;

  while (inputFile.hasNext()) {

    if (inputFile.getCounter()%1000 == 0) {
      STATUS("counter: " << setw(8) << inputFile.getCounter() << '\r' << flush); DEBUG(endl);
    }

    const JToA* p = inputFile.next();

    if (detector.getID() != p->DETID) {          // consistency check
      FATAL("Invalid detector identifier " << p->DETID << " != " << detector.getID() << endl);
    }

    if (p->WAVEFORMID == waveform && receivers.has(p->DOMID)) {

      if (p->QUALITYFACTOR >= parameters.Q * (parameters.Q <= 1.0 ? p->QUALITYNORMALISATION : 1.0)) {

        const JReceiver&    receiver = receivers[p->DOMID];

    const double toa = p->TOA_S();

        const JTransmission transmission(p->RUN,
                                         p->DOMID,
                                         p->QUALITYFACTOR,
                                         p->QUALITYNORMALISATION,
                                         receiver.getT(toa),
                                         receiver.getT(toa));
        
        data.push_back(hit_type(receiver.getPosition(), transmission));
      }
    }
  }
  STATUS(endl);


  sort(data.begin(), data.end(), make_comparator(&hit_type::getToA, JComparison::lt()));          // sort according time-of-arrival


  buffer_type buffer(parameters.factoryLimit);    // local buffer of hits
  JEvent      out[2];                             // FIFO of events
 
  const chrono::steady_clock::time_point t0 = chrono::steady_clock::now();

  size_t trigger_counter = 0;

  for (buffer_type::const_iterator p = data.begin(), q = p; p != data.end(); ++p) {

    if (distance(data.cbegin(),p)%10000 == 0) {
      //STATUS("processed: " << FIXED(5,1) << (double) (100 * distance(data.cbegin(),p)) / (double) data.size() << "%" << ' ' << FIXED(12,2) << p->getToA() << " [s]" << '\r' << flush); DEBUG(endl);

      const chrono::steady_clock::time_point t1 = chrono::steady_clock::now();

      STATUS("processed: "
             << FIXED(5,1) << (double) (100 * distance(data.cbegin(),p)) / (double) data.size() << "%"    << ' '
             << FIXED(8,1) << p->getToA()                                                       << " [s]" << ' '
             << setw(6)    << chrono::duration_cast<chrono::seconds>(t1 - t0).count()           << " [s]" << ' '
             << setw(6)    << trigger_counter                                                             << endl);
    }

    h0.Fill(0.0);

    while (q != data.end() && q->getToA() - p->getToA() <= parameters.TMax_s) { ++q; }

    if (distance(p,q) >= parameters.numberOfHits) {

      h0.Fill(1.0);

      if (distance(p,q) < parameters.factoryLimit) {

        h0.Fill(2.0);

        buffer_type::iterator root = buffer.begin();
        buffer_type::iterator __p  = buffer.begin();
        buffer_type::iterator __q  = buffer.begin();

        *root = *p;
        
        ++__p;
        ++__q;

        for (buffer_type::const_iterator i = p; ++i != q; ) {
          if (match3D(*root,*i)) {
            *(__q++) = *i;
          }
        }

        M1[2]->Fill(distance(root,__q));

        if (distance(root,__q) >= parameters.numberOfHits) { 

          h0.Fill(3.0);

          __q = clusterize(__p, __q, match3D);

          M1[3]->Fill(distance(root,__q));

          if (distance(root,__q) >= parameters.numberOfHits) {
            
            h0.Fill(4.0);

            const double W = 1.0 / (double) rotator.size();

            for (const JRotation3D& R : rotator) {

              for (buffer_type::iterator i = root; i != __q; ++i) {
                i->rotate(R);
              }

              buffer_type::iterator __z = partition(__p, __q, JBind2nd(match1D,*root));
              
              M1[4]->Fill(distance(root,__z), W);

              if (distance(root,__z) >= parameters.numberOfHits) {

                h0.Fill(5.0, W);
                
                __z = clusterize(__p, __z, match1D);

                M1[5]->Fill(distance(root,__z), W);

                if (distance(root,__z) >= parameters.numberOfHits) {

                  h0.Fill(6.0, W);

                  if (!check.is_valid() || check(*root, __p, __z)) {

                    trigger_counter += 1;

                    h0.Fill(7.0, W);

                    out[1] = JEvent(detector.getID(), out[0].getCounter() + 1, waveform, p, q);

                    break;
                  }
                }
              }
            }
          }
        }
          
      } else {

        trigger_counter += 1;

        out[1] = JEvent(detector.getID(), out[0].getCounter() + 1, waveform, p, q);
      }

      if (!out[1].empty()) {

        if (out[0].empty()) {
          
          out[0] = out[1];                         // shift

        } else if (overlap(out[0],out[1])) {

          out[0].merge(out[1]);                    // merge

        } else {

          if (debug >= notice_t) { print(cout, out[0]); }

          outputFile.put(out[0]);                  // write

          out[0] = out[1];                         // shift
        }
      }

      out[1].clear();
    }
  }

  if (!out[0].empty()) {

    if (debug >= notice_t) { print(cout, out[0]); }

    outputFile.put(out[0]);                        // write
  }
  STATUS(endl);

  outputFile.put(h0);

  for (TH1D* h1 : M1) {
    if (h1 != NULL) {
      outputFile.put(*h1);
    }
  }
  
  JMultipleFileScanner<JMeta> io(inputFile);

  io >> outputFile;

  outputFile.close();
}