JConstructDetector.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>
#include <fstream>

#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorCalibration.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JDetectorAddressMapToolkit.hh"
#include "JDetector/JModuleRouter.hh"
#include "JDetector/JPMTRouter.hh"
#include "JSupport/JMeta.hh"

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


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

  string   detectorFile;
  string   inputFile;
  string   outputFile;
  int      debug;

  try {

    JParser<> zap("Auxiliary program to compose detector from separate calibrations.");

    zap['a'] = make_field(detectorFile,  "detector file (module identifiers, module locations and PMT identifiers are used)");
    zap['f'] = make_field(inputFile,     "detector calibrations file (json format)");
    zap['o'] = make_field(outputFile,    "detector file")   = "";
    zap['d'] = make_field(debug)                            = 1;

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


  JDetector detector;

  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }

  if (!hasDetectorAddressMap(detector.getID())) {
    FATAL("No detector address map for detector identier " << detector.getID() << endl);
  }

  const JDetectorAddressMap& demo = getDetectorAddressMap(detector.getID());

  const JModuleRouter moduleRouter(detector);
  const JPMTRouter    pmtRouter   (detector);


  json js;

  istream* in = open<istream>(inputFile.c_str());

  *in >> js;

  close(in);

  DEBUG(setw(4) << js << endl);

  JPMTCalibration  calibration;
  JModulePosition  position;
  JModuleRotation  rotation;  

  if (is_valid(js)) {

    json::const_iterator data = js.find(Data_t);

    if (data != js.end()) {

      for (size_t i = 0; i != data->size(); ++i) {

        if (data->at(i).contains(PMTT0s_t)) {
          calibration = data->at(i)[PMTT0s_t].get<JPMTCalibration>();
        }

        if (data->at(i).contains(DOMPositions_t)) {
          position = data->at(i)[DOMPositions_t].get<JModulePosition>();
        }

        if (data->at(i).contains(DOMRotations_t)) {
          rotation = data->at(i)[DOMRotations_t].get<JModuleRotation>();
        }
      }
    }
  }


  for (JModuleRotation::const_iterator i = rotation.begin(); i != rotation.end(); ++i) {

    if (moduleRouter.hasModule(i->getID())) {

      JModule&      module = detector.getModule(moduleRouter.getAddress(i->getID()));
      const JModule buffer = getModule(demo.get(module.getID()), module.getID());

      if (module.size() != buffer.size()) {
        FATAL("Module size " << module.size() << " != " << buffer.size() << endl);
      }

      for (size_t i = 0; i != module.size(); ++i) {
        module.getPMT(i).setAxis(buffer.getPMT(i).getAxis());
      }

      module.rotate(*i);
    }
  }

  for (JModulePosition::const_iterator i = position.begin(); i != position.end(); ++i) {

    if (moduleRouter.hasModule(i->getID())) {

      JModule& module = detector.getModule(moduleRouter.getAddress(i->getID()));

      module.add(*i);
    }
  }

  for (JPMTCalibration::const_iterator i = calibration.begin(); i != calibration.end(); ++i) {

    if (pmtRouter.hasPMT(i->getID())) {

      JPMT& pmt = detector.getPMT(pmtRouter.getAddress(i->getID()));

      pmt.setCalibration(*i);
    }
  }


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

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