software/JDB/JPMTThreshold.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <vector>
#include <map>

#include "TF1.h"

#include "JDB/JDB.hh"
#include "JDB/JSelector.hh"
#include "JDB/JSelectorSupportkit.hh"
#include "JDB/JDBToolkit.hh"
#include "JDB/JDetectors.hh"
#include "JDB/JRuns.hh"
#include "JDB/JAllParams.hh"
#include "JDB/JRunsetupParams.hh"
#include "JDB/JDetectorIntegration.hh"

#include "JDetector/JPMTParametersMap.hh"

#include "JSupport/JMeta.hh"

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


/**
 * \file
 * Auxiliary program to set PMT thresholds according DAQ configuration.
 * \author mdejong
 */
int main(const int argc, const char * const argv[])
{
  using namespace std;
  using namespace JPP;

  string   usr;
  string   pwd;
  string   cookie;
  string   detid;
  int      run;
  string   pmtFile;
  string   formula;
  int      debug;

  try {

    JParser<> zap("Auxiliary program to set PMT thresholds according DAQ configuration.");

    zap['u'] = make_field(usr)        = "";
    zap['!'] = make_field(pwd)        = "";
    zap['C'] = make_field(cookie)     = "";
    zap['D'] = make_field(detid);
    zap['r'] = make_field(run);
    zap['P'] = make_field(pmtFile,    "PMT parameters file")  =  "";
    zap['F'] = make_field(formula,    "transfer function")    =  "((x * 1.0)/255.0 * 1.6 + 0.8 - 0.989) / 0.44"; // "((x * 1.0)/255.0 * 1.6 + 0.8 - 1) / 0.44";
    zap['d'] = make_field(debug)      = 1;

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


  JPMTParametersMap parameters;

  if (pmtFile != "") {
    parameters.load(pmtFile.c_str());
  }


  TF1 f1("user", formula.c_str());

  if (!f1.IsValid()) {
    FATAL("Formula <" << formula << "> invalid." << endl);
  }


  try {
    JDB::reset(usr, pwd, cookie);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }


  string rs_oid;

  try {

    JRuns parameters;

    ResultSet& rs = getResultSet(getTable<JRuns>(), getSelector<JRuns>(is_integer(detid) ? to_value<int>(detid) : getDetector(detid), run));

    if (rs >> parameters) {
      rs_oid = parameters.RUNSETUPID;
    }

    rs.Close();

    DEBUG("Run setup " << rs_oid << endl);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }


  JAllParams upars;

  try {

    JAllParams parameters;

    ResultSet& rs = getResultSet(getTable<JAllParams>(), getSelector<JAllParams>("OPTICS"));

    while (rs >> parameters) {
      if (parameters.NAME.find("PMT_THRESHOLD") != string::npos) {
        upars = parameters;
      }
    }

    rs.Close();
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }


  map<string, int> umap;

  try {

    JRunsetupParams parameters;

    ResultSet& rs = getResultSet(getTable<JRunsetupParams>(), getSelector<JRunsetupParams>(is_integer(detid) ? getDetector(to_value<int>(detid)) : detid, rs_oid));

    while (rs >> parameters) {

      if (parameters.PAR_OID == upars.OID && parameters.ISINPUT == 'Y') {
        if (parameters.VALUE != "") {
          umap[parameters.UPIFILTER] = to_value<int>(parameters.VALUE);
        }
      }
    }

    rs.Close();
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }

  for (map<string, int>::const_iterator i = umap.begin(); i != umap.end(); ++i) {
    DEBUG(left << setw(20) << i->first << ' ' << right << setw(3) << i->second << endl);
  }


  vector<JDetectorIntegration> detector;

  try {
    
    JDetectorIntegration parameters;

    ResultSet& rs = getResultSet(getTable<JDetectorIntegration>(), getSelector<JDetectorIntegration>(is_integer(detid) ? to_value<int>(detid) : getDetector(detid)));

    while (rs >> parameters) {
      detector.push_back(parameters);
    }

    rs.Close();
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }



  for (vector<JDetectorIntegration>::const_iterator i = detector.begin(); i != detector.end(); ++i) {

    if (i->PMTID != -1) {

      const JUPI_t upi = i->PMTUPI;

      map<string, int>::const_iterator p = umap.find(upi.toString());

      bool is_default = (p == umap.end());

      if (p == umap.end()) {

        ostringstream os;

        os << upi.getPBS()     << JUPI_t::SEPARATOR
           << upi.getVariant() << JUPI_t::SEPARATOR
           << JUPI_t::DOT;

        p = umap.find(os.str());
      }
      
      if (p != umap.end()) {

        const double threshold = f1.Eval((double) p->second);

        if (debug >= debug_t) {

          cout << setw(4) << i->DUID      << ' ' 
               << setw(2) << i->FLOORID   << ' ' 
               << setw(2) << i->CABLEPOS  << ' ' 
               << setw(3) << p->second    << ' '
               << FIXED(5,2) << threshold << ' ';

          if (is_default) {
            cout << "* (" << FIXED(5,2) << JPMTParameters().threshold << ")";
          }

          cout << endl;
        }

        if (!is_default) {

          const JPMTIdentifier id(i->DOMID, i->CABLEPOS);

          parameters[id].threshold = threshold;
        }

      } else {

        ERROR("Missing threshold data PMT " << upi << endl);
      }
    }
  }

  if (pmtFile != "") {

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

    parameters.store(pmtFile.c_str());

  } else {

    cout << parameters << endl;
  }
}