CMSRunSummary2DB.cc

Go to the documentation of this file.

#ifndef RecoLuminosity_LumiProducer_CMSRunSummary2DB_h
#define RecoLuminosity_LumiProducer_CMSRunSummary2DB_h
#include "CoralBase/AttributeList.h"
#include "CoralBase/Attribute.h"
#include "CoralBase/AttributeSpecification.h"
#include "CoralBase/Exception.h"
#include "CoralBase/TimeStamp.h"
#include "RelationalAccess/ConnectionService.h"
#include "RelationalAccess/ISessionProxy.h"
#include "RelationalAccess/ITransaction.h"
#include "RelationalAccess/ITypeConverter.h"
#include "RelationalAccess/IQuery.h"
#include "RelationalAccess/ICursor.h"
#include "RelationalAccess/ISchema.h"
#include "RelationalAccess/IView.h"
#include "RelationalAccess/ITable.h"
#include "RelationalAccess/ITableDataEditor.h"
#include "RelationalAccess/IBulkOperation.h"
#include "RecoLuminosity/LumiProducer/interface/DataPipe.h"
#include "RecoLuminosity/LumiProducer/interface/LumiNames.h"
#include "RecoLuminosity/LumiProducer/interface/idDealer.h"
#include "RecoLuminosity/LumiProducer/interface/Exception.h"
#include "RecoLuminosity/LumiProducer/interface/DBConfig.h"
#include "RecoLuminosity/LumiProducer/interface/ConstantDef.h"
#include "RecoLuminosity/LumiProducer/interface/DataPipe.h"
#include "RecoLuminosity/LumiProducer/interface/LumiNames.h"
#include "RecoLuminosity/LumiProducer/interface/idDealer.h"
#include "RecoLuminosity/LumiProducer/interface/Exception.h"
#include "RecoLuminosity/LumiProducer/interface/DBConfig.h"
#include "RecoLuminosity/LumiProducer/interface/ConstantDef.h"
#include <iostream>
#include <sstream>
#include <fstream>
#include <vector>
#include <string>
#include <boost/regex.hpp>
#include <boost/tokenizer.hpp>

namespace lumi {
  class CMSRunSummary2DB : public DataPipe {
  public:
    CMSRunSummary2DB(const std::string& dest);
    unsigned long long retrieveData(unsigned int runnumber) override;
    const std::string dataType() const override;
    const std::string sourceType() const override;
    unsigned int str2int(const std::string& s) const;
    ~CMSRunSummary2DB() override;

  private:
    struct cmsrunsum {
      std::string l1key;
      std::string amodetag;
      int egev;
      std::string hltkey;
      std::string fillnumber;  //convert to number when write into lumi
      std::string sequence;
      std::string fillscheme;
      int ncollidingbunches;
      coral::TimeStamp startT;
      coral::TimeStamp stopT;
    };
    bool isCollisionRun(const lumi::CMSRunSummary2DB::cmsrunsum& rundata);
    void parseFillCSV(const std::string& csvsource, cmsrunsum& result);
  };  //cl CMSRunSummary2DB
  //
  //implementation
  //
  void CMSRunSummary2DB::parseFillCSV(const std::string& csvsource, lumi::CMSRunSummary2DB::cmsrunsum& result) {
    result.fillscheme = std::string("");
    result.ncollidingbunches = 0;
    std::ifstream csvfile;
    csvfile.open(csvsource.c_str());
    if (!csvfile) {
      std::cout << "[warning] unable to open file: " << csvsource << std::endl;
      return;
    }
    typedef boost::tokenizer<boost::escaped_list_separator<char> > Tokenizer;
    std::vector<std::string> record;
    std::string line;
    while (std::getline(csvfile, line)) {
      Tokenizer tok(line);
      record.assign(tok.begin(), tok.end());
      if (record.size() < 3)
        continue;
      std::string fillnum = record[0];
      if (fillnum == result.fillnumber) {
        result.fillscheme = record[1];
        std::string ncollidingbunchesStr = record[2];
        result.ncollidingbunches = str2int(ncollidingbunchesStr);
        break;
      }
    }
  }
  CMSRunSummary2DB::CMSRunSummary2DB(const std::string& dest) : DataPipe(dest) {}
  bool CMSRunSummary2DB::isCollisionRun(const lumi::CMSRunSummary2DB::cmsrunsum& rundata) {
    bool isCollision = false;
    bool isPhysics = false;
    std::string hk = rundata.hltkey;
    std::string lk = rundata.l1key;
    boost::match_results<std::string::const_iterator> what;
    const boost::regex lexpr("^TSC_.+_collisions_.+");
    boost::regex_match(lk, what, lexpr, boost::match_default);
    if (what[0].matched)
      isCollision = true;
    const boost::regex hexpr("^/cdaq/physics/.+");
    boost::regex_match(hk, what, hexpr, boost::match_default);
    if (what[0].matched)
      isPhysics = true;
    return (isCollision && isPhysics);
  }
  unsigned long long CMSRunSummary2DB::retrieveData(unsigned int runnumber) {
    cmsrunsum result;
    std::string runinfoschema("CMS_RUNINFO");
    std::string runsessionParamTable("RUNSESSION_PARAMETER");
    coral::ConnectionService* svc = new coral::ConnectionService;
    lumi::DBConfig dbconf(*svc);
    if (!m_authpath.empty()) {
      dbconf.setAuthentication(m_authpath);
    }

    //std::cout<<"m_source "<<m_source<<std::endl;
    std::string::size_type cutpos = m_source.find(';');
    std::string dbsource = m_source;
    std::string csvsource("");
    if (cutpos != std::string::npos) {
      dbsource = m_source.substr(0, cutpos);
      csvsource = m_source.substr(cutpos + 1);
    }
    //std::cout<<"dbsource: "<<dbsource<<" , csvsource: "<<csvsource<<std::endl;
    coral::ISessionProxy* runinfosession = svc->connect(dbsource, coral::ReadOnly);
    try {
      coral::ITypeConverter& tpc = runinfosession->typeConverter();
      tpc.setCppTypeForSqlType("unsigned int", "NUMBER(38)");
      runinfosession->transaction().start(true);
      coral::ISchema& runinfoschemaHandle = runinfosession->schema(runinfoschema);
      if (!runinfoschemaHandle.existsTable(runsessionParamTable)) {
        throw lumi::Exception(
            std::string("non-existing table " + runsessionParamTable), "CMSRunSummary2DB", "retrieveData");
      }
      coral::IQuery* amodetagQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList amodetagOutput;
      amodetagOutput.extend("amodetag", typeid(std::string));
      coral::AttributeList amodetagCondition;
      amodetagCondition = coral::AttributeList();
      amodetagCondition.extend("name", typeid(std::string));
      amodetagCondition.extend("runnumber", typeid(unsigned int));
      amodetagCondition["name"].data<std::string>() = std::string("CMS.SCAL:AMODEtag");
      amodetagCondition["runnumber"].data<unsigned int>() = runnumber;
      amodetagQuery->addToOutputList("distinct(STRING_VALUE)");
      amodetagQuery->setCondition("NAME=:name AND RUNNUMBER=:runnumber", amodetagCondition);
      //amodetagQuery->limitReturnedRows(1);
      amodetagQuery->defineOutput(amodetagOutput);
      coral::ICursor& amodetagCursor = amodetagQuery->execute();
      std::vector<std::string> amodes;
      while (amodetagCursor.next()) {
        const coral::AttributeList& row = amodetagCursor.currentRow();
        amodes.push_back(row["amodetag"].data<std::string>());
        //result.amodetag=row["amodetag"].data<std::string>();
      }
      //
      //priority pick the one contains PHYS if not found pick the first
      //
      std::string amd;
      for (std::vector<std::string>::iterator it = amodes.begin(); it != amodes.end(); ++it) {
        if (it->find("PHYS") == std::string::npos)
          continue;
        amd = *it;
      }
      if (amd.empty() && !amodes.empty()) {
        amd = *(amodes.begin());
      }
      if (amd.empty()) {
        amd = std::string("PROTPHYS");  //last resort
      }
      //std::cout<<"amd "<<amd<<std::endl;
      result.amodetag = amd;
      delete amodetagQuery;

      coral::IQuery* egevQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList egevOutput;
      egevOutput.extend("egev", typeid(std::string));
      coral::AttributeList egevCondition;
      egevCondition = coral::AttributeList();
      egevCondition.extend("name", typeid(std::string));
      egevCondition.extend("runnumber", typeid(unsigned int));
      egevCondition["name"].data<std::string>() = std::string("CMS.SCAL:EGEV");
      egevCondition["runnumber"].data<unsigned int>() = runnumber;
      egevQuery->addToOutputList("distinct(STRING_VALUE)");
      egevQuery->setCondition("NAME=:name AND RUNNUMBER=:runnumber", egevCondition);
      egevQuery->defineOutput(egevOutput);
      coral::ICursor& egevCursor = egevQuery->execute();
      result.egev = 0;
      while (egevCursor.next()) {
        const coral::AttributeList& row = egevCursor.currentRow();
        std::string egevstr = row["egev"].data<std::string>();
        int tmpgev = str2int(egevstr);
        if (tmpgev > result.egev) {
          result.egev = tmpgev;
        }
      }
      if (result.egev == 0) {
        result.egev = 3500;  //last resort
      }
      delete egevQuery;

      coral::IQuery* seqQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList seqBindVariableList;
      seqBindVariableList.extend("runnumber", typeid(unsigned int));
      seqBindVariableList.extend("name", typeid(std::string));

      seqBindVariableList["runnumber"].data<unsigned int>() = runnumber;
      seqBindVariableList["name"].data<std::string>() = std::string("CMS.LVL0:SEQ_NAME");
      seqQuery->setCondition("RUNNUMBER =:runnumber AND NAME =:name", seqBindVariableList);
      seqQuery->addToOutputList("STRING_VALUE");
      coral::ICursor& seqCursor = seqQuery->execute();

      while (seqCursor.next()) {
        const coral::AttributeList& row = seqCursor.currentRow();
        result.sequence = row["STRING_VALUE"].data<std::string>();
      }
      delete seqQuery;

      coral::IQuery* hltkeyQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList hltkeyBindVariableList;
      hltkeyBindVariableList.extend("runnumber", typeid(unsigned int));
      hltkeyBindVariableList.extend("name", typeid(std::string));

      hltkeyBindVariableList["runnumber"].data<unsigned int>() = runnumber;
      hltkeyBindVariableList["name"].data<std::string>() = std::string("CMS.LVL0:HLT_KEY_DESCRIPTION");
      hltkeyQuery->setCondition("RUNNUMBER =:runnumber AND NAME =:name", hltkeyBindVariableList);
      hltkeyQuery->addToOutputList("STRING_VALUE");
      coral::ICursor& hltkeyCursor = hltkeyQuery->execute();

      while (hltkeyCursor.next()) {
        const coral::AttributeList& row = hltkeyCursor.currentRow();
        result.hltkey = row["STRING_VALUE"].data<std::string>();
      }
      delete hltkeyQuery;

      coral::IQuery* fillQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList fillBindVariableList;
      fillBindVariableList.extend("runnumber", typeid(unsigned int));
      fillBindVariableList.extend("name", typeid(std::string));

      fillBindVariableList["runnumber"].data<unsigned int>() = runnumber;
      fillBindVariableList["name"].data<std::string>() = std::string("CMS.SCAL:FILLN");
      fillQuery->setCondition("RUNNUMBER =:runnumber AND NAME =:name", fillBindVariableList);
      fillQuery->addToOutputList("STRING_VALUE");
      fillQuery->addToOrderList("TIME");
      //fillQuery->limitReturnedRows(1);
      coral::ICursor& fillCursor = fillQuery->execute();
      unsigned int cc = 0;
      while (fillCursor.next()) {
        const coral::AttributeList& row = fillCursor.currentRow();
        if (cc == 0) {
          result.fillnumber = row["STRING_VALUE"].data<std::string>();
        }
        ++cc;
      }
      delete fillQuery;
      if (result.fillnumber.empty()) {
        throw nonCollisionException("retrieveData", "CMSRunSummary2DB");
      }
      coral::IQuery* startTQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList startTVariableList;
      startTVariableList.extend("runnumber", typeid(unsigned int));
      startTVariableList.extend("name", typeid(std::string));

      startTVariableList["runnumber"].data<unsigned int>() = runnumber;
      startTVariableList["name"].data<std::string>() = std::string("CMS.LVL0:START_TIME_T");
      startTQuery->setCondition("RUNNUMBER =:runnumber AND NAME =:name", startTVariableList);
      startTQuery->addToOutputList("TIME");
      coral::ICursor& startTCursor = startTQuery->execute();

      while (startTCursor.next()) {
        const coral::AttributeList& row = startTCursor.currentRow();
        result.startT = row["TIME"].data<coral::TimeStamp>();
      }
      delete startTQuery;
      coral::IQuery* stopTQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList stopTVariableList;
      stopTVariableList.extend("runnumber", typeid(unsigned int));
      stopTVariableList.extend("name", typeid(std::string));

      stopTVariableList["runnumber"].data<unsigned int>() = runnumber;
      stopTVariableList["name"].data<std::string>() = std::string("CMS.LVL0:STOP_TIME_T");
      stopTQuery->setCondition("RUNNUMBER =:runnumber AND NAME =:name", stopTVariableList);
      stopTQuery->addToOutputList("TIME");
      coral::ICursor& stopTCursor = stopTQuery->execute();

      while (stopTCursor.next()) {
        const coral::AttributeList& row = stopTCursor.currentRow();
        result.stopT = row["TIME"].data<coral::TimeStamp>();
      }
      delete stopTQuery;
      coral::IQuery* l1keyQuery = runinfoschemaHandle.tableHandle(runsessionParamTable).newQuery();
      coral::AttributeList l1keyOutput;
      l1keyOutput.extend("l1key", typeid(std::string));
      coral::AttributeList l1keyCondition;
      l1keyCondition = coral::AttributeList();
      l1keyCondition.extend("name", typeid(std::string));
      l1keyCondition.extend("runnumber", typeid(unsigned int));
      l1keyCondition["name"].data<std::string>() = std::string("CMS.TRG:TSC_KEY");
      l1keyCondition["runnumber"].data<unsigned int>() = runnumber;
      l1keyQuery->addToOutputList("STRING_VALUE");
      l1keyQuery->setCondition("NAME=:name AND RUNNUMBER=:runnumber", l1keyCondition);
      //l1keyQuery->limitReturnedRows(1);
      l1keyQuery->defineOutput(l1keyOutput);
      coral::ICursor& l1keyCursor = l1keyQuery->execute();
      while (l1keyCursor.next()) {
        const coral::AttributeList& row = l1keyCursor.currentRow();
        result.l1key = row["l1key"].data<std::string>();
      }
      delete l1keyQuery;
    } catch (const coral::Exception& er) {
      runinfosession->transaction().rollback();
      delete runinfosession;
      delete svc;
      throw er;
    }
    runinfosession->transaction().commit();
    delete runinfosession;

    if (!csvsource.empty()) {
      parseFillCSV(csvsource, result);
    } else {
      result.fillscheme = std::string("");
      result.ncollidingbunches = 0;
    }
    std::cout << "result for run " << runnumber << " : sequence : " << result.sequence
              << " : hltkey : " << result.hltkey << " : fillnumber : " << result.fillnumber
              << " : l1key : " << result.l1key << " : amodetag :" << result.amodetag << " : egev : " << result.egev
              << " : fillscheme " << result.fillscheme << " : ncollidingbunches : " << result.ncollidingbunches
              << std::endl;

    //std::cout<<"connecting to dest "<<m_dest<<std::endl;
    coral::ISessionProxy* destsession = svc->connect(m_dest, coral::Update);

    coral::ITypeConverter& desttpc = destsession->typeConverter();
    desttpc.setCppTypeForSqlType("unsigned int", "NUMBER(10)");
    try {
      destsession->transaction().start(false);
      coral::ISchema& destschema = destsession->nominalSchema();
      coral::ITable& destruntable = destschema.tableHandle(LumiNames::cmsrunsummaryTableName());
      coral::AttributeList runData;
      destruntable.dataEditor().rowBuffer(runData);
      runData["RUNNUM"].data<unsigned int>() = runnumber;
      runData["FILLNUM"].data<unsigned int>() = str2int(result.fillnumber);
      runData["SEQUENCE"].data<std::string>() = result.sequence;
      runData["HLTKEY"].data<std::string>() = result.hltkey;
      runData["STARTTIME"].data<coral::TimeStamp>() = result.startT;
      runData["STOPTIME"].data<coral::TimeStamp>() = result.stopT;
      runData["AMODETAG"].data<std::string>() = result.amodetag;
      runData["EGEV"].data<unsigned int>() = (unsigned int)result.egev;
      runData["L1KEY"].data<std::string>() = result.l1key;
      runData["FILLSCHEME"].data<std::string>() = result.fillscheme;
      runData["NCOLLIDINGBUNCHES"].data<unsigned int>() = result.ncollidingbunches;
      destruntable.dataEditor().insertRow(runData);
    } catch (const coral::Exception& er) {
      std::cout << "database problem " << er.what() << std::endl;
      destsession->transaction().rollback();
      delete destsession;
      delete svc;
      throw er;
    }
    destsession->transaction().commit();
    delete svc;
    return 0;
  }
  const std::string CMSRunSummary2DB::dataType() const { return "CMSRUNSUMMARY"; }
  const std::string CMSRunSummary2DB::sourceType() const { return "DB"; }
  unsigned int CMSRunSummary2DB::str2int(const std::string& s) const {
    std::istringstream myStream(s);
    unsigned int i;
    if (myStream >> i) {
      return i;
    } else {
      throw lumi::Exception(std::string("str2int error"), "str2int", "CMSRunSummary2DB");
    }
  }
  CMSRunSummary2DB::~CMSRunSummary2DB() {}
}  // namespace lumi
#include "RecoLuminosity/LumiProducer/interface/DataPipeFactory.h"
DEFINE_EDM_PLUGIN(lumi::DataPipeFactory, lumi::CMSRunSummary2DB, "CMSRunSummary2DB");
#endif