JNTupleToDetector.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/JDetectorAddressMapToolkit.hh"
#include "JDetector/JModuleGeometry.hh"
#include "JIO/JFileStreamIO.hh"

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


/**
 * \file
 *
 * Program to convert ntuples to detector.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;

  string         detectorFile;
  string         ntupleFile;
  string         outputFile;
  int            debug;

  try {

    JParser<> zap("Program to convert ntuples to detector.");

    zap['a'] = make_field(detectorFile);
    zap['f'] = make_field(ntupleFile);
    zap['o'] = make_field(outputFile)  = "";
    zap['d'] = make_field(debug)       = 0;

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

  
  using namespace JPP;


  JDetector detector;

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


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

  
  JTimer timer("module");

  JFileStreamReader in(ntupleFile.c_str());

  for (JModuleGeometry geometry; in.load(geometry); ) {

    DEBUG(geometry << endl);
    
    const int id = geometry.getID();

    if (router.hasModule(id)) {

      timer.start();

      const JEulerMatrix3D R(geometry.getEulerAngle());
      
      JModule master = getModule(demo.get(id), id);
      
      master.rotate(R);
      master.add(geometry.getPosition());
   
      JModule& module = detector.getModule(router.getAddress(id)); 
   
      if (module.size() == master.size()) {

        for (int pmt = 0; pmt != (int) master.size(); ++pmt) {
          module.getPMT(pmt).setAxis(master.getPMT(pmt).getAxis());
        }

        module.compile();

      } else {

        ERROR("Module sizes inconsistent " << module.size() << " != " << master.size() << endl);
      }

      timer.stop();

    } else {
      
      ERROR("Missing module " << geometry.getID() << endl);
    }
  }

  in.close();

  timer.print(cout);

  if (outputFile != "") {

    if (outputFile == "%") {
      outputFile = detectorFile;
    }

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