JDetectorDB.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <map>
#include <algorithm>

#include "JDB/JDB.hh"
#include "JDB/JSelector.hh"
#include "JDB/JSelectorSupportkit.hh"
#include "JDB/JCLBMap.hh"
#include "JDB/JDetectorIntegration_t.hh"
#include "JDB/JPBSSequence.hh"
#include "JDB/JProductRouter.hh"
#include "JDB/JDBToolkit.hh"

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

#include "JLang/JComparator.hh"
#include "JLang/JPredicate.hh"

#include "JSupport/JMeta.hh"

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

namespace {

  using JDATABASE::JUPI_t;

  /**
   * Get delay time for acoustic sensor with given %UPI.
   *
   * \param  upi            %UPI
   * \return                delay time [ns]
   */
  inline double getDelayTime(const JUPI_t& upi)
  {
    using namespace std;
    using namespace JPP;

    if      (upi.getPBS() == PBS::HYDROPHONE)
      return HYDROPHONE_DELAYTIME_US * 1.0e3;
    else if (upi.getPBS() == PBS::ACOUSTIC_SENSOR && upi.getVariant().find("V1") != string::npos)
      return PIEZO_V1_DELAYTIME_US   * 1.0e3;
    else if (upi.getPBS() == PBS::ACOUSTIC_SENSOR && upi.getVariant().find("V2") != string::npos)
      return PIEZO_V2_DELAYTIME_US   * 1.0e3;
    else
      THROW(JValueOutOfRange, "Invalid UPI " << upi);
  }
}


/**
 * \file
 *
 * Auxiliary program to print detector from database.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  
  JServer  server;
  string   usr;
  string   pwd;
  string   cookie;
  string   id;
  string   tcalset;
  string   pcalset;
  string   rcalset;
  string   detectorFile;
  string   outputFile;
  int      run;
  JCounter wizzard;
  int      debug;
  
  try {

    JParser<> zap("Auxiliary program to print detector from database.");
    
    zap['s'] = make_field(server)        = getServernames();
    zap['u'] = make_field(usr)           = "";
    zap['!'] = make_field(pwd)           = "";
    zap['C'] = make_field(cookie)        = "";
    zap['D'] = make_field(id,            "detector identifier")                                        = "";
    zap['r'] = make_field(run,           "run number")                                                 = -1;
    zap['t'] = make_field(tcalset,       "tcal: t0 (time offsets) calibration set from DB")            = "";
    zap['p'] = make_field(pcalset,       "pcal: position (x,y,z) calibration set from DB")             = "";
    zap['q'] = make_field(rcalset,       "rcal: orientation (quaterions) calibration set from DB")     = "";
    zap['a'] = make_field(detectorFile)  = "";
    zap['o'] = make_field(outputFile)    = "";
    zap['W'] = make_field(wizzard,       "-W add base modules; -WW set time offsets of acoustic sensors");
    zap['d'] = make_field(debug)         = 2;
    
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }


  const int file   = ((detectorFile != "" ? 1 : 0));
  const int calset = ((tcalset.empty() ? 0 : 1)  +
                      (pcalset.empty() ? 0 : 1)  +
                      (rcalset.empty() ? 0 : 1));
  const int daq    = ((run != -1 ? 1 : 0));

  if (file   != 0  &&  (file   != 1))             { FATAL("Inconsistent specificaton of detector file."); }
  if (calset != 0  &&  (calset != 3 || id == "")) { FATAL("Inconsistent specificaton of detector identifier, tcal, pcal and rcal calibration sets."); }
  if (daq    != 0  &&  (daq    != 1 || id == "")) { FATAL("Inconsistent specificaton of detector identifier and run number."); }

  int np = 0;

  if (file   == 1) { ++np; }
  if (calset == 3) { ++np; }
  if (daq    == 1) { ++np; }

  if (np != 1) { FATAL("Specify detector file; OR detector identifier, tcal, pcal and rcal calibration set; OR detector identifier and run number."); }


  try {
    JDB::reset(usr, pwd, cookie);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }

  JDetector detector;

  try {

    if      (file   == 1)
      load(detectorFile, detector);
    else if (calset == 3)
      *(JDB::get()->DetX)(id.c_str(), tcalset.c_str(), pcalset.c_str(), rcalset.c_str())  >>  detector;
    else if (daq    == 1)
      *(JDB::get()->DetX)(id.c_str(), run)                                                >>  detector;
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }

  detector.comment.add(JMeta(argc,argv));

  if (detector.setToLatestVersion()) {
    NOTICE("Set detector version to " << detector.getVersion() << endl);
  }

  if (wizzard) {

    vector<JCLBMap> clbmap;

    try {

      JSelector selector = getSelector<JCLBMap>(getDetector(detector.getID()));

      selector.add(&JCLBMap::FLOORID, 0);

      ResultSet& rs  = getResultSet(getTable<JCLBMap>(), selector);

      rs >> clbmap;

      rs.Close();
    }
    catch(const exception& error) {
      FATAL(error.what() << endl);
    }

    map<int, map<int, JPosition3D> > zmap;

    for (JDetector::const_iterator module = detector.begin(); module != detector.end(); ++module) {
      zmap[module->getString()][module->getFloor()] = module->getPosition();
    }

    for (vector<JCLBMap>::const_iterator i = clbmap.begin(); i != clbmap.end(); ++i) {

      if (i->FLOORID == 0) {

        JDetector::iterator p = find_if(detector.begin(), detector.end(), make_predicate(&JModule::getID, i->DOMID, JComparison::eq()));

        if (p == detector.end()) {

          JModule module(i->DOMID, JLocation(i->DUID, i->FLOORID));

          module.setPosition(JPosition3D(zmap[i->DUID].begin()->second.getX(),
                                         zmap[i->DUID].begin()->second.getY(),
                                         isARCADetector(detector) ? ARCA_TBARZ_M :
                                         isORCADetector(detector) ? ORCA_TBARZ_M :
                                         0.0));

          NOTICE("Adding module " << module << endl);

          detector.push_back(module);
        }
      }
    }
  }

  if (wizzard > 1) {

    try {

      NOTICE("Setting time offsets of acoustic sensors... " << flush);

      JDetectorIntegration_t detector_t;

      ResultSet& rs  = getResultSet(getTable<JDetectorIntegration_t>());

      if (! (rs >> detector_t)) {
        THROW(JDatabaseException, "Error reading " << getTable<JDetectorIntegration_t>());
      }

      detector_t.configure(getDetector<string>(id));

      for (const JPBS_t& pbs : { PBS::ACOUSTIC_SENSOR, PBS::HYDROPHONE }) {

        const JProductRouter router(detector_t, getPBSSequences(pbs));

        JDetectorIntegration_t::range_type range = detector_t.find(pbs);

        for (JDetectorIntegration_t::range_const_iterator i = range.first; i != range.second; ++i) {

          const JLocation_t location = router.getLocation(i->first);

          JDetector::iterator p = find_if(detector.begin(), 
                                          detector.end(), 
                                          make_predicate(&JModule::getLocation, 
                                                         JLocation(location.string, (pbs != PBS::HYDROPHONE ? location.floor : 0)), 
                                                         JComparison::eq()));
        
          if (p != detector.end())
            p->setCalibration(getAverage(make_array(p->begin(), p->end(), &JModule::getT0), 0.0)  -  getDelayTime(i->first));
          else
            FATAL("No module for UPI " << i->first << endl);
        }
      }

      NOTICE("OK" << endl);
    }
    catch(const exception& error) {
      FATAL(error.what() << endl);
    }
  }

  if (outputFile != "")
    store(outputFile, detector);
  else
    cout << detector;
}