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 "JSupport/JMeta.hh"

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


/**
 * \file
 *
 * Auxiliary program to merge detector files.
 *
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  vector<string> detectorFile;
  string         tcalset;
  string         pcalset;
  string         rcalset;
  string         outputFile;
  int            debug;

  try {

    JParser<> zap("Auxiliary program to merge detector files.");

    zap['a'] = make_field(detectorFile);
    zap['o'] = make_field(outputFile);
    zap['t'] = make_field(tcalset,       "tcal: t0 (time offsets) calibration set from file")          = "";
    zap['p'] = make_field(pcalset,       "pcal: position (x,y,z) calibration set from file")           = "";
    zap['q'] = make_field(rcalset,       "rcal: orientation (quaterions) calibration set from file")   = "";
    zap['d'] = make_field(debug)          = 1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }


  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 (tcalset != "") {

    JDetector buffer;

    try {
      load(tcalset, 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 (pcalset != "") {

    JDetector buffer;

    try {
      load(pcalset, 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 (rcalset != "") {

    JDetector buffer;

    try {
      load(rcalset, 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())).rotate(getRotation(detector.getModule(router.getAddress(module.getID())),module));
        }

      } else
        FATAL("Missing module " << module.getID() << endl);
    }
  }

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

  try {
    store(outputFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
}