JDetector/JPlotPMTParameters.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>
 
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
 
#include "km3net-dataformat/online/JDAQ.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JPMTParametersMap.hh"
#include "JDetector/JPMTParametersToolkit.hh"
#include "JDetector/JDetectorAddressMap.hh"
#include "JDetector/JDetectorSupportkit.hh"
#include "JROOT/JManager.hh"
#include "JGizmo/JGizmoToolkit.hh"
 
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
namespace {
 
  using JDETECTOR::JPMTParameters;
 
  /**
   * Get efficiency.
   *
   * \param  parameters        PMT parameters
   * \return                   efficiency
   */
  inline double getEfficiency(const JPMTParameters& parameters)
  {
    return getSurvivalProbability(parameters) * parameters.QE;
  }
 
  static const char* const EFFICIENCY = "EFFICIENCY";
}
 
 
/**
 * \file
 * Auxiliary application to plot PMT parameters.
 * \author mdejong
 */
 
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;
 
  string               detectorFile;
  JPMTParametersMap    parameters;
  string               outputFile;
  bool                 showPMTAddress;
  int                  debug;
 
  try { 
 
    JParser<> zap("Auxiliary application to plot PMT parameters.");
 
    zap['a'] = make_field(detectorFile,      "detector file.");
    zap['o'] = make_field(outputFile,        "output file.")      = "pmt_parameters.root";
    zap['P'] = make_field(parameters,        "PMT calibration data (or corresponding file name)");
    zap['A'] = make_field(showPMTAddress,    "show PMT address on y-axis");
    zap['d'] = make_field(debug,             "debug")             = 0;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  JDetector detector;
 
  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
 
  if (detector.empty()) {
    FATAL("Empty detector." << endl);
  }
 
  JManager<string, TH1D> manager(new TH1D("%", NULL, NUMBER_OF_PMTS, -0.5, NUMBER_OF_PMTS - 0.5));
 
  for (JDetector::iterator module = detector.begin(); module != detector.end(); ++module) {
 
    DEBUG("Module " << setw(10) << module->getID() << endl);
 
    for (int pmt = 0; pmt != NUMBER_OF_PMTS; ++pmt) {
 
      const JProperties properties = parameters.getPMTParameters(JPMTIdentifier(module->getID(), pmt)).getProperties();
 
      for (JProperties::const_iterator p = properties.begin(); p != properties.end(); ++p) {
 
        double value = 0.0;
 
        try { value = (p->second.getValue<const double>()); }           catch(const exception& error) {}
        try { value = (p->second.getValue<const bool>() ? 1.0 : 0.0); } catch(const exception& error) {}
 
        manager[MAKE_CSTRING(module->getID() << "." << p->first)]->SetBinContent(pmt + 1, value);
      }
 
      double value = getEfficiency(parameters.getPMTParameters(JPMTIdentifier(module->getID(), pmt)));
      
      manager[MAKE_CSTRING(module->getID() << "." << EFFICIENCY)]->SetBinContent(pmt + 1, value);
    }
  }
 
  if (showPMTAddress) {
 
    const JDetectorAddressMap& demo = getDetectorAddressMap(detector.getID());
 
    for (JManager<string, TH1D>::iterator i = manager.begin(); i != manager.end(); ++i) {
 
      int id;
 
      istringstream(i->first) >> id;
 
      setAxisLabels(*i->second, "X", demo.get(id));
    }
  }
 
  manager.Write(outputFile.c_str());
}