JConvertAcoustics.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <memory>
 
#include "TROOT.h"
#include "TFile.h"
 
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMeta.hh"
 
#include "JAcoustics/JEvent.hh"
#include "JAcoustics/JEvt.hh"
#include "JAcoustics/JSuperEvt.hh"
#include "JAcoustics/JSupport.hh"
 
#include "JLang/JPipe.hh"
#include "JLang/JObjectSelector.hh"
#include "JROOT/JROOTClassSelector.hh"
#include "JTools/JRange.hh"
 
#include "Jeep/JContainer.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
namespace {
 
  using namespace JLANG;
  using namespace JACOUSTICS;
 
  /**
   * Type definition for UTC time range.
   */
  typedef JTOOLS::JRange<double>               JTimeRange;
 
  /**
   * Auxiliary data structure for UTC time ranges.
   */
  struct veto_type :
    public std::vector<JTimeRange>
  {
    veto_type() {}
 
    veto_type(const veto_type& veto) : 
      std::vector<JTimeRange>(static_cast<const std::vector<JTimeRange>&>(veto))
    {}
 
    // allow use of virtual base class
    veto_type(veto_type&&) = delete;
    veto_type& operator=(veto_type&&) = delete;
  };
 
 
  /**
   * Default implementation for veto of acoustics objects based on UTC time ranges.
   */
  template<class T>
  struct JSelector :
    public virtual JObjectSelector<T>,
    public virtual veto_type
  {
    virtual bool accept(const T& object) const override
    {
      return true;
    }
  };
 
  /**
   * Custom implementation for veto of acoustics objects based on UTC time ranges.
   */
  template<>
  struct JSelector<JEvent> :
    public virtual JObjectSelector<JEvent>,
    public virtual veto_type
  {
    virtual bool accept(const JEvent& event) const override
    {
      if (!event.empty()) {
        for (veto_type::const_iterator i = this->begin(); i != this->end(); ++i) {
          for (JEvent::const_iterator hit = event.begin(); hit != event.end(); ++hit) {
            if ((*i)(hit->getToA())) {
              return false;
            }
          }
        }
      }
 
      return true;
    }
  };
 
  /**
   * Custom implementation for veto of acoustics objects based on UTC time ranges.
   */
  template<>
  struct JSelector<JEvt> :
    public virtual JObjectSelector<JEvt>,
    public virtual veto_type
  {
    virtual bool accept(const JEvt& evt) const override
    {
      for (veto_type::const_iterator i = this->begin(); i != this->end(); ++i) {
        if ((*i)(evt.UNIXTimeStart) || (*i)(evt.UNIXTimeStop)) {
          return false;
        }
      }
 
      return true;
    }
  };
 
  /**
   * Implementation for multiple veto of acoustics objects based on UTC time ranges.
   */
  template<>
  struct JSelector<JSuperEvt> :
    public virtual JObjectSelector<JSuperEvt>,
    public virtual veto_type
  {
    virtual bool accept(const JSuperEvt& evt) const override
    {
      for (veto_type::const_iterator i = this->begin(); i != this->end(); ++i) {
        if ((*i)(evt.UNIXTimeStart) || (*i)(evt.UNIXTimeStop)) {
          return false;
        }
      }
 
      return true;
    }
  };
 
  /**
   * Implementation for multiple selection of objects based on UTC time range.
   */
  template<class JHead_t, class JTail_t>
  struct JSelector< JTypeList<JHead_t, JTail_t> > :
    public virtual JObjectSelector< JTypeList<JHead_t, JTail_t> >,
    public JSelector<JHead_t>,
    public JSelector<JTail_t>
  {};
 
  /**
   * Terminator class of recursive JSelector class.
   */
  template<class JHead_t>
  struct JSelector< JTypeList<JHead_t, JNullType> > :
    public JSelector<JHead_t>
  {};
}
 
/**
 * \file
 *
 * Auxiliary program to convert acoustic events.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  typedef JTYPELIST<JMeta, JEvent, JEvt, JSuperEvt>::typelist        typelist;
  typedef JContainer<JSelector<typelist>, std::vector<JTimeRange> >  JSelector_t;
 
  JMultipleFileScanner<typelist> inputFile;
  JFileRecorder       <typelist> outputFile;
  JLimit_t&           numberOfEvents = inputFile.getLimit();
  JROOTClassSelection selection      = getROOTClassSelection<typelist>();
  JSelector_t         selector;
  int                 debug;
 
  try {
 
    JParser<> zap("Auxiliary program to convert acoustic events.");
 
    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile);
    zap['n'] = make_field(numberOfEvents)   = JLimit::max();
    zap['C'] = make_field(selection,
                          "Precede name of data structure by a '+' or '-' "
                          "to add or remove data types in the output, respectively."
                          "\nROOT wildcards are accepted.") = JPARSER::initialised();
    zap['r'] = make_field(selector,
                          "veto UTC time ranges [s]")       = JPARSER::initialised();
    zap['d'] = make_field(debug)            = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  outputFile.open();
 
  outputFile.put(JMeta(argc,argv));
 
  inputFile | JValve<typelist>(selection) | selector | outputFile;
 
  outputFile.close();
}