JAcousticsTriggerProcessor.cc File Reference

Main program to trigger acoustic data. More...

#include <iostream>
#include <iomanip>
#include <vector>
#include <map>
#include <algorithm>
#include "TROOT.h"
#include "TFile.h"
#include "JLang/JComparator.hh"
#include "JDB/JToAshort.hh"
#include "JDB/JSupport.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JDetectorSupportkit.hh"
#include "JDetector/JHydrophone.hh"
#include "JLang/JPredicate.hh"
#include "JTools/JRange.hh"
#include "JTools/JQuantile.hh"
#include "JTools/JHashMap.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMeta.hh"
#include "JAcoustics/JTransmission.hh"
#include "JAcoustics/JReceiver.hh"
#include "JAcoustics/JSoundVelocity.hh"
#include "JAcoustics/JAcousticsToolkit.hh"
#include "JAcoustics/JAcousticsSupportkit.hh"
#include "JAcoustics/JTriggerParameters.hh"
#include "JAcoustics/JEvent.hh"
#include "JAcoustics/JTriggerOutput.hh"
#include "JAcoustics/JSupport.hh"
#include "Jeep/JContainer.hh"
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "JTrigger/JMatch.hh"
#include "JTrigger/JAlgorithm.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...
struct	JACOUSTICS::JEventOverlap
	Match of two events considering overlap in time. More...

Namespaces
	JACOUSTICS

Functions
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 JAcousticsTriggerProcessor.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 183 of file JAcousticsTriggerProcessor.cc.

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

  typedef JContainer< vector<JHydrophone> >  hydrophones_container;

  JMultipleFileScanner<JToAshort> inputFile;
  JLimit_t&                numberOfEvents = inputFile.getLimit();
  JTriggerParameters       parameters;
  JFileRecorder<typelist>  outputFile;
  JSoundVelocity           V = getSoundVelocity;   // default sound velocity
  string                   detectorFile;
  hydrophones_container    hydrophones;            // hydrophones
  double                   precision;
  int                      ID;
  set<int>                 waveforms;
  int                      debug;

  try {

    JParser<> zap("Main program to trigger acoustic data.");
    
    zap['f'] = make_field(inputFile,    "output of JConvertDB -q toashort");
    zap['n'] = make_field(numberOfEvents)                                    = JLimit::max();    
    zap['@'] = make_field(parameters,   "trigger parameters");
    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['H'] = make_field(hydrophones,  "hydrophone data")                   = JPARSER::initialised();
    zap['p'] = make_field(precision,    "precision time-of-arrival")         = 1.0e-6;
    zap['D'] = make_field(ID)                                                = 10000;
    zap['w'] = make_field(waveforms)                                         = JPARSER::initialised();
    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 int           FACTORY_LIMIT  =  10000;
  const double        TMaxExtra_s    =  1.0e-3;

  const JMatch3D      match(V, TMaxExtra_s);
  const JEventOverlap overlap;                   // (parameters.TMax_s);

    
  JHashMap<int, JReceiver>   receivers;

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

    JPosition3D pos(0.0, 0.0, 0.0);

    if (i->getFloor() == 0) {                    // get relative position of hydrophone

      try {
        pos = getPosition(hydrophones.begin(),
                          hydrophones.end(),
                          make_predicate(&JHydrophone::getString, i->getString()));
      }
      catch(const exception&) {
        continue;                                // if no position avialable, discard hydrophone
      }
    }

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


  outputFile.open();

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

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


  // input data

  string oid;                                    // detector identifier

  typedef vector<hit_type> buffer_type;          // data type

  map<int, buffer_type>  f1;                     // receiver -> data

  while (inputFile.hasNext()) {

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

    JToAshort* parameters = inputFile.next();

    if (oid == "") {
      oid = parameters->DETID;
    }

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

    if (waveforms.empty() || waveforms.count(parameters->EMITTERID)) {
      
      if (receivers.has(parameters->DOMID)) {

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

        const JTransmission transmission(parameters->RUNNUMBER,
                                         parameters->DOMID,
                                         parameters->QUALITYFACTOR,
                                         parameters->UNIXTIMEBASE  +  receiver.getT(parameters->TOA_S),
                                         parameters->UNIXTIMEBASE  +  receiver.getT(parameters->TOA_S));

        f1[receiver.getID()].push_back(hit_type(receiver.getPosition(), transmission));
      }
    }
  }
  STATUS(endl);


  // filter similar hits

  for (map< int, buffer_type>::iterator receiver = f1.begin(); receiver != f1.end(); ++receiver) {

    buffer_type& buffer = receiver->second;

    sort(buffer.begin(), buffer.end(), JTransmission::compare(precision));

    buffer.erase(unique(buffer.begin(), buffer.end(), JTransmission::equals(precision)), buffer.end());
  }


  buffer_type data;

  for (map<int, buffer_type>::iterator receiver = f1.begin(); receiver != f1.end(); ++receiver) {

    // selection based on quality

    buffer_type&    f2   = receiver->second;
    double          Qmin = parameters.Q;

    if (parameters.Q > 0.0 && parameters.Q < 1.0) {

      const JQuantile Q1("quality", make_array(f2.begin(), f2.end(), &JTransmission::getQ), true);

      Qmin = Q1.getQuantile(parameters.Q);
    }

    buffer_type::iterator __end = partition(f2.begin(), f2.end(), make_predicate(&JTransmission::getQ, Qmin, JComparison::gt()));

    copy(f2.begin(), __end, back_inserter(data));
  }


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


  buffer_type buffer(FACTORY_LIMIT);               // local buffer of hits
  JEvent      out[2];                              // FIFO of events
  int         number_of_events = 0;
 
  for (buffer_type::const_iterator p = data.begin(); p != data.end(); ++p) {

    if (distance(data.cbegin(),p)%1000 == 0) {
      STATUS("processed: " << FIXED(5,1) << (double) (100 * distance(data.cbegin(),p)) / (double) data.size() << "%" << '\r' << flush); DEBUG(endl);
    }
    
    buffer_type::const_iterator q = p;

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

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

      if (distance(p,q) < FACTORY_LIMIT) {

        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 (match(*p,*i)) {
            *__q = *i;
            ++__q;
          }
        }

        if (distance(root,__q) >= parameters.numberOfHits) { 
        
          __q = clusterize(__p, __q, match, parameters.numberOfHits - 1);

          if (distance(root,__q) >= parameters.numberOfHits) {
            out[1] = JEvent(oid, ++number_of_events, ID, root, __q);
          }
        }

      } else {

        out[1] = JEvent(oid, ++number_of_events, ID, p, q);
      }

      if (!overlap(out[0],out[1])) {

        if (!out[0].empty()) {
          outputFile.put(out[0]);                  // write
        }

        out[0] = out[1];                           // shift
              
      } else {
              
        out[0].merge(out[1]);                      // merge
      }

      out[1].clear();
    }
  }

  if (!out[0].empty()) {
    outputFile.put(out[0]);                        // write
  }
  STATUS(endl);

  STATUS("triggers: " << setw(7) << number_of_events << endl);
  
  JMultipleFileScanner<JMeta> io(inputFile);

  io >> outputFile;

  outputFile.close();
}