JMergeTuneHV.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <vector>
#include <set>

#include "JSystem/JTime.hh"

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

#include "JLang/JPredicate.hh"

#include "JSupport/JFilenameSupportkit.hh"

#include "JDetector/JDetectorCalibration.hh"

#include "JDB/JDB.hh"
#include "JDB/JDBToolkit.hh"
#include "JDB/JDBincludes.hh"
#include "JDB/JSelector.hh"
#include "JDB/JSelectorSupportkit.hh"

#include "JSon/JSon.hh"


/**
 * \file
 *
 * Auxiliary program to merge multiple HV tuning calibrations.
 *
 * Note that only the calibration in the first given file will be maintained\n
 * in case a PMT appears in multiple files.
 * \author bjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  string         usr;
  string         pwd;
  string         cookie;
  
  vector<string> inputFiles;
  string         outputFile;
  
  int            debug;

  try {

    JParser<> zap("Auxiliary program to merge high-voltage calibration results.");

    zap['u'] = make_field(usr)                                = "";
    zap['!'] = make_field(pwd)                                = "";
    zap['C'] = make_field(cookie)                             = "";
    zap['f'] = make_field(inputFiles,   "input  files"); 
    zap['o'] = make_field(outputFile,   "output file");
    zap['d'] = make_field(debug,        "debug")              = 2;

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

  JDateAndTime timer;

  // Retrieve personalia
  JPersons person;

  try {
    
    JDB::reset(usr, pwd, cookie);
    
    JSelector  selector = getSelector<JPersons>(JDB::get()->User());
    ResultSet& rs       = getResultSet(getTable<JPersons>(), selector);
    rs >> person;
    rs.Close();
    
  } catch (const exception& error) {
    
    FATAL(error.what() << endl);
  }

  // Merge high-voltage calibrations
  JHVCalibration merged;
  set<string>    testTypes;
  
  for (vector<string>::const_iterator fileIt = inputFiles.cbegin(); fileIt != inputFiles.cend(); ++fileIt) {

    if (isJSONFile(fileIt->c_str())) {

      json     js;
      ifstream ifs(fileIt->c_str());
      
      ifs >> js;
      ifs.close();

      json::const_iterator DBtype = js.find(Test_t + Type_t);

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

        testTypes.insert(DBtype->get<string>());
        
      } else {

        WARNING("No database test type specified for " << *fileIt << endl);
      }
      
      JHVCalibration HVcals;

      json::const_iterator data = js.find(Tests_t);
      
      if (data != js.end()) {
    
        HVcals = data->get<JHVCalibration>();
    
      } else {

        WARNING("HV calibration data for " << *fileIt << "could not be found; skip." << endl);
        continue;
      }

      for (JHVCalibration::const_iterator cal = HVcals.cbegin(); cal != HVcals.cend(); ++cal) {
        
        if (find_if(merged.cbegin(), merged.cend(), make_predicate(&JHVCalibration_t::getNumber, cal->getNumber())) == merged.cend()) {
          merged.push_back(*cal);
        }
      }

    } else {

      WARNING(*fileIt << " is not a JSON file; skip" << endl);
      continue;
    }
  }

  if (testTypes.size() > 1) {
   
    WARNING("Database test type is ambiguous; Assuming " << *testTypes.cbegin() << endl);
    
  } else if (testTypes.size() == 0) {
    
    ERROR("No database test type specified.");
  }

  json js;

  js[User_t]          = person.LOGIN;
  js[Location_t]      = person.LOCATIONID;
  js[StartTime_t]     = timer.toString();
  js[EndTime_t]       = timer().toString();
  js[Test_t + Type_t] = *testTypes.cbegin();
  js[Tests_t]         = json(merged);
  
  ofstream ofs(outputFile.c_str());

  ofs << setw(2) << setprecision(8);
  ofs << js;

  ofs.close();

  return 0;
}