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);
    }

    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);
  }
}