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/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"


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

    JParser<> zap("Auxiliary program to print detector from database.");
    
    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 (quarterions) calibration set from DB")    = "";
    zap['a'] = make_field(detectorFile)  = "";
    zap['o'] = make_field(outputFile)    = "";
    zap['W'] = make_field(wizzard);
    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) {

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

      if (i->FLOORID == 0 && 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(),
                                       TBARZ_M));

        const double t0 = 0.0;

        module.setCalibration(t0  -  HYDROPHONE_DELAYTIME_US * 1.0e+3);

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

        detector.push_back(module);
      }
    }
  }

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