JMergeDetector.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JModuleRouter.hh"
#include "JDetector/JPMTRouter.hh"
#include "JDetector/JDetectorCalibration.hh"
 
#include "JLang/JEquation_t.hh"
 
#include "JSupport/JMeta.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
namespace {
 
  using namespace JPP;
 
  /**
   * List of valid keys.
   */
  static const std::set<std::string> keys = { TCAL, PCAL, RCAL, ACAL, CCAL, SCAL };
}
 
 
/**
 * \file
 *
 * Auxiliary program to merge detector files.
 *
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  vector<string>      detectorFile;
  vector<JEquation_t> calset;
  string              outputFile;
  int                 debug;
 
  try {
 
    JParser<> zap("Auxiliary program to merge detector files.");
 
    zap['a'] = make_field(detectorFile);
    zap['o'] = make_field(outputFile);
    zap['@'] = make_field(calset,        "import calibration sets \"<key> = <file name>[; <key> = <file name>]\"" << endl
                          << "possible keys: " << JEEPZ() << keys)                                      = JPARSER::initialised();
    zap['d'] = make_field(debug)          = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  map<string, string> calibration;
 
  for (vector<JEquation_t>::const_iterator i = calset.begin(); i != calset.end(); ++i) {
 
    if (keys.count(i->getKey()) == 0)
      FATAL("Invalid calibration set \"" << i->getKey() << "\"" << endl);
    else
      calibration[i->getKey()] = trim(i->getValue());
  }
 
 
  JDetector detector;
 
  for (vector<string>::const_iterator i = detectorFile.begin(); i != detectorFile.end(); ++i) {
 
    JDetector buffer;
 
    try {
      load(*i, buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    if (detector.empty())
      detector = buffer;
    else
      copy(buffer.begin(), buffer.end(), back_inserter(detector));
  }
 
 
  if (calibration.count(TCAL)) {
 
    JDetector buffer;
 
    try {
      load(calibration[TCAL], buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    JPMTRouter router(detector);
 
    for (const auto& module : buffer) {
      for (const auto& pmt : module) {
        if (router.hasPMT(pmt.getID()))
          detector.getPMT(router.getAddress(pmt.getID())).setCalibration(pmt.getCalibration());
        else
          FATAL("Missing PMT " << pmt.getID() << endl);
      }
    }
  }
 
 
  if (calibration.count(PCAL)) {
 
    JDetector buffer;
 
    try {
      load(calibration[PCAL], buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    JModuleRouter router(detector);
 
    for (const auto& module : buffer) {
      if (router.hasModule(module.getID())) {
 
        if (module.getFloor() != 0)
          detector.getModule(router.getAddress(module.getID())).set(module.getCenter());
        else
          detector.getModule(router.getAddress(module.getID())).set(module.getPosition());
 
      } else
        FATAL("Missing module " << module.getID() << endl);
    }
  }
 
 
  if (calibration.count(RCAL)) {
 
    JDetector buffer;
 
    try {
      load(calibration[RCAL], buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    JModuleRouter router(detector);
 
    for (const auto& module : buffer) {
      if (router.hasModule(module.getID())) {
 
        if (module.getFloor() != 0) {
 
          JModule& object = detector.getModule(router.getAddress(module.getID()));
          
          const JRotation3D R      = getRotation(object, module);
          const JPosition3D center = object.getCenter();
 
          object.sub(center);
          object.rotate(R);
          object.add(center);
        }
 
      } else
        FATAL("Missing module " << module.getID() << endl);
    }
  }
 
 
  if (calibration.count(ACAL)) {
 
    JDetector buffer;
 
    try {
      load(calibration[ACAL], buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    JModuleRouter router(detector);
 
    for (const auto& module : buffer) {
      if (router.hasModule(module.getID())) {
 
        detector.getModule(router.getAddress(module.getID())).setT0(module.getT0());
 
      } else
        FATAL("Missing module " << module.getID() << endl);
    }
  }
 
 
  if (calibration.count(CCAL)) {
 
    JDetector buffer;
 
    try {
      load(calibration[CCAL], buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    JModuleRouter router(detector);
 
    for (const auto& module : buffer) {
      if (router.hasModule(module.getID())) {
 
        detector.getModule(router.getAddress(module.getID())).setQuaternion(module.getQuaternion());
 
      } else
        FATAL("Missing module " << module.getID() << endl);
    }
  }
 
 
  if (calibration.count(SCAL)) {
 
    JDetector buffer;
 
    try {
      load(calibration[SCAL], buffer);
    }
    catch(const JException& error) {
      FATAL(error);
    }
    {
      JModuleRouter router(detector);
 
      for (const auto& module : buffer) {
        if (router.hasModule(module.getID()))
          detector.getModule(router.getAddress(module.getID())).setStatus(module.getStatus());
        else
          FATAL("Missing module " << module.getID() << endl);
      }
    }
    {
      JPMTRouter router(detector);
 
      for (const auto& module : buffer) {
        for (const auto& pmt : module) {
          if (router.hasPMT(pmt.getID()))
            detector.getPMT(router.getAddress(pmt.getID())).setStatus(pmt.getStatus());
          else
            FATAL("Missing PMT " << pmt.getID() << endl);
        }
      }
    }
  }
 
 
  detector.comment.add(JMeta(argc,argv));
 
  try {
    store(outputFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
}