JRunCalibration.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <algorithm>

#include "dbclient/KM3NeTDBClient.h"

#include "JDB/JDB.hh"
#include "JDB/JDBToolkit.hh"
#include "JDB/JSelector.hh"
#include "JDB/JSelectorSupportkit.hh"
#include "JDB/JCalibration.hh"
#include "JDB/JRunCalibration.hh"
#include "JDB/JSonSupportkit.hh"

#include "JSon/JSon.hh"
#include "JSon/JSupport.hh"

#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"

#include "JLang/JException.hh"
#include "JLang/JPredicate.hh"
#include "JLang/JManip.hh"

#include "JSupport/JMeta.hh"

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

namespace {

  using namespace JPP;

  static const char* const RUN_t = "RUN";

  /**
   * Auxiliary data structure for run calibrations.
   */
  struct calibration_type :
    public std::vector<APIV2::JRunCalibration_t> 
  {
    /**
     * Get calibration identifier.
     *
     * \param  type        calibration type
     * \return             calibration idientifier
     */
    std::string operator[](const std::string& type) const
    {
      using namespace std;

      const_iterator p = find_if(this->begin(), this->end(), make_predicate(&APIV2::JRunCalibration_t::CalibrationType, type));

      if (p != this->end())
        return p->CalibrationId;
      else
        THROW(JException, "No calibration type " << type << endl);
    }
  };

}


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

  JServer         server;
  string          usr;
  string          pwd;
  string          cookie;
  string          detid;
  int             run;
  int             ranking;
  string          variant;
  string          outputFile;
  vector<string>  format;
  int             debug;

  try {

    JParser<> zap("Auxiliary program to download official detector.");
    
    zap['s'] = make_field(server)     = getServernames();
    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['R'] = make_field(ranking,       "ranking (1 is best)")      = 1;
    zap['V'] = make_field(variant,       "detector version")         = getDetectorVersions<string>();
    zap['o'] = make_field(outputFile,    "detector file or JSon files (must then contain wildcard \'" << FILENAME_WILDCARD << "\')")  =  "";
    zap['F'] = make_field(format,     "column names: " 
                          << RUN_t                      << ' '
                          << PMT_T0_CALIBRATION_t       << ' '
                          << DOM_POSITION_CALIBRATION_t << ' '
                          << DOM_ROTATION_CALIBRATION_t << ' '
                          << ACOUSTIC_T0_CALIBRATION_t  << ' '
                          << COMPASS_CALIBRATION_t      << ' '
                          << STATUS_CALIBRATION_t)  = JPARSER::initialised();
    zap['d'] = make_field(debug)      = 1;

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

  try {

    JDB::reset(usr, pwd, cookie);

    detid = getDetector<string>(detid);

    json js;

    to_json(js, APIV2::JRunCalibration::getName(), getSelector<APIV2::JRunCalibration>(detid, run, ranking));

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

    if (!is_valid(js)) {
      FATAL("JSon error code not okay." << endl);
    }

    if (!js.contains(Data_t) || js[Data_t].empty()) {
      FATAL("JSon output does not contain data." << endl);
    }

    const calibration_type calibration = js[Data_t].get<calibration_type>();

    if (debug >= debug_t) {
      for (const auto& element : calibration) {
        cout << setw(24) << left << element.CalibrationType << ' ' << right << element.CalibrationId << endl;
      }
    }

    if (outputFile != "") {

      if (!hasWildCard(outputFile)) {

        JDetector detector;

        *(JDB::get()->DetX)(detid.c_str(),
                            calibration[PMT_T0_CALIBRATION_t]      .c_str(),
                            calibration[DOM_POSITION_CALIBRATION_t].c_str(),
                            calibration[DOM_ROTATION_CALIBRATION_t].c_str(),
                            calibration[ACOUSTIC_T0_CALIBRATION_t] .c_str(),
                            calibration[COMPASS_CALIBRATION_t]     .c_str(),
                            calibration[STATUS_CALIBRATION_t]      .c_str(),
                            getDetectorVersion(variant))  >>  detector;
      
        detector.comment.add(JMeta(argc,argv));

        for (const auto& i : { PMT_T0_CALIBRATION_t, DOM_POSITION_CALIBRATION_t, DOM_ROTATION_CALIBRATION_t, ACOUSTIC_T0_CALIBRATION_t, COMPASS_CALIBRATION_t, STATUS_CALIBRATION_t }) {
          detector.comment.add(MAKE_STRING(i << "=" << calibration[i]));
        }

        if (detector.setToLatestVersion()) {
          NOTICE("Set detector version to " << detector.getVersion() << endl);
        }

        store(outputFile, detector);

      } else {

        for (const auto& element : calibration) {

          json buffer;

          to_json(buffer, APIV2::JCalibration_t::getName(), getSelector<APIV2::JCalibration_t>(detid, element.CalibrationId));

          if (is_valid(buffer))
            store(setWildCard(outputFile.c_str(), getCalibrationType.getNickname(element.CalibrationType)), buffer);
          else
            ERROR("Invalid JSon data " << setw(2) << buffer);
        }
      }
    }

    if (!format.empty()) {

      for (vector<string>::const_iterator i = format.begin(); i != format.end(); ++i) {
      
        if      (*i == RUN_t)
          cout << ' ' << run;
        else
          cout << ' ' << calibration[*i];
      }

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

  return 0;
}