CondTools/Ecal/interface/PopConESTransitionSourceHandler.h - master - githubmirror/cmssw

CondTools/Ecal/interface/PopConESTransitionSourceHandler.h
164 строки9 KB
Giacomo Govi
Port CMSSW to renewed CondDB interface
09 мар 2020, 17:40
09 мар 2020, 17:40e702860
#ifndef PopConBTransitionSourceHandler_H
#define PopConBTransitionSourceHandler_H

#include "CondCore/CondDB/interface/ConnectionPool.h"
#include "CondCore/CondDB/interface/IOVProxy.h"
#include "CondCore/PopCon/interface/PopConSourceHandler.h"
#include "CondFormats/RunInfo/interface/RunInfo.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include <string>

namespace popcon {
  template <class T>
  class PopConESTransitionSourceHandler : public PopConSourceHandler<T> {
  public:
    PopConESTransitionSourceHandler(edm::ParameterSet const& pset)
        : m_run(pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<unsigned long long>("runNumber")),
          //      m_currentThreshold( pset.getParameter<edm::ParameterSet>( "ESTransition" ).getUntrackedParameter<double>( "currentThreshold", 18000. ) ),
          m_tagForRunInfo(
              pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<std::string>("tagForRunInfo")),
          m_ESGain(pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<std::string>("ESGain")),
          m_tagForLowGain(
              pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<std::string>("ESLowGainTag")),
          m_tagForHighGain(
              pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<std::string>("ESHighGainTag")),
          m_connectionString(pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<std::string>("connect")),
          m_connectionPset(
              pset.getParameter<edm::ParameterSet>("ESTransition").getParameter<edm::ParameterSet>("DBParameters")) {
      edm::LogInfo("PopConESTransitionSourceHandler") << "["
                                                      << "PopConESTransitionSourceHandler:" << __func__ << "]: "
                                                      << "Initialising Connection Pool" << std::endl;
      m_connection.setParameters(m_connectionPset);
      m_connection.configure();
    }

    ~PopConESTransitionSourceHandler() override {}

    std::string id() const final { return std::string("PopConESTransitionSourceHandler"); }

    bool checkLowGain() {
      //the output boolean is set to true as default
      bool isLowGain = true;
      cond::persistency::Session& session = PopConSourceHandler<T>::dbSession();
      //reading RunInfo from Conditions
      cond::persistency::TransactionScope trans(session.transaction());
      edm::LogInfo("PopConESTransitionSourceHandler")
          << "["
          << "PopConESTransitionSourceHandler::" << __func__ << "]: "
          << "Loading tag for RunInfo " << m_tagForRunInfo << " and IOV valid for run number: " << m_run << std::endl;
      cond::persistency::IOVProxy iov = session.readIov(m_tagForRunInfo);
      cond::Iov_t currentIov = iov.getInterval(m_run);
      LogDebug("PopConESTransitionSourceHandler")
          << "Loaded IOV sequence from tag " << m_tagForRunInfo << " with size: " << iov.loadedSize()
          << ", IOV valid for run number " << m_run << " starting from: " << currentIov.since
          << ", with corresponding payload hash: " << currentIov.payloadId << std::endl;
      /*    accessing the average magnet current for the run
            double current_default = -1;
           double avg_current = current_default;
           avg_current = session.fetchPayload<RunInfo>( currentIov.payloadId )->m_avg_current;
            LogDebug( "PopConESTransitionSourceHandler" ) << "Comparing value of magnet current: " << avg_current << " A for run: " << m_run
                                                   << " with the corresponding threshold: "<< m_currentThreshold << " A." << std::endl;
      comparing the magnet current with the user defined threshold
      if( avg_current != current_default && avg_current <= m_currentThreshold ) isLowGain = false;
      edm::LogInfo( "PopConESTransitionSourceHandler" ) << "[" << "PopConESTransitionSourceHandler::" << __func__ << "]: "
                                                       << "The magnet was " << ( isBOn ? "ON" : "OFF" )
						       << " during run " << m_run << std::endl;   */
      if (m_ESGain == "HIGH") {
        isLowGain = false;
      } else {
        isLowGain = true;
      }
      trans.close();
      return isLowGain;
    }

    virtual void getObjectsForESTransition(bool isLowGain) {
      //reading payloads for 0T and 3.8T from Conditions
      cond::persistency::Session& session = PopConSourceHandler<T>::dbSession();
      cond::persistency::TransactionScope trans(session.transaction());
      edm::LogInfo("PopConESTransitionSourceHandler") << "["
                                                      << "PopConESTransitionSourceHandler::" << __func__ << "]: "
                                                      << "Loading tag for ES " << (isLowGain ? "LowGain" : "HighGain")
                                                      << ": " << (isLowGain ? m_tagForLowGain : m_tagForHighGain)
                                                      << " and IOV valid for run number: " << m_run << std::endl;
      cond::persistency::IOVProxy iov = session.readIov(isLowGain ? m_tagForLowGain : m_tagForHighGain);
      cond::Iov_t currentIov = iov.getInterval(m_run);
      LogDebug("PopConESTransitionSourceHandler")
          << "Loaded IOV sequence from tag " << (isLowGain ? m_tagForLowGain : m_tagForHighGain)
          << " with size: " << iov.loadedSize() << ", IOV valid for run number " << m_run
          << " starting from: " << currentIov.since << ", with corresponding payload hash: " << currentIov.payloadId
          << std::endl;
      std::string destTag = this->tagInfo().name;
      if (currentIov.payloadId != this->tagInfo().lastInterval.payloadId) {
        std::ostringstream ss;
        ss << "Adding iov with since " << m_run << " pointing to hash " << currentIov.payloadId
           << " corresponding to the ES Gain " << (isLowGain ? "LOW" : "HIGH");
        edm::LogInfo("PopConESTransitionSourceHandler")
            << "["
            << "PopConESTransitionSourceHandler::" << __func__ << "]: " << ss.str() << std::endl;
        cond::persistency::IOVEditor editor;
        if (session.existsIov(destTag)) {
          editor = session.editIov(destTag);
        } else {
          editor = session.createIov<T>(destTag, iov.tagInfo().timeType);
          editor.setDescription("Tag created by PopConESTransitionSourceHandler");
        }
        editor.insert(m_run, currentIov.payloadId);
        editor.flush();
        this->m_userTextLog = ss.str();
      } else {
        edm::LogInfo("PopConESTransitionSourceHandler")
            << "["
            << "PopConESTransitionSourceHandler::" << __func__ << "]: "
            << "The payload with hash " << currentIov.payloadId << " corresponding to ES Gain"
            << (isLowGain ? "LOW" : "HIGH") << " is still valid for run " << m_run << " in the destination tag "
            << destTag << ".\nNo transfer needed." << std::endl;
      }
      trans.close();
    }

    void getNewObjects() final {
      //check whats already inside of database
      edm::LogInfo("PopConESTransitionSourceHandler")
          << "["
          << "PopConESTransitionSourceHandler::" << __func__ << "]: "
          << "Destination Tag Info: name " << this->tagInfo().name << ", size " << this->tagInfo().size
          << ", last object valid since " << this->tagInfo().lastInterval.since << ", hash "
          << this->tagInfo().lastInterval.payloadId << std::endl;
      //check if a transfer is needed:
      //if the new run number is smaller than or equal to the latest IOV, exit.
      //This is needed as now the IOV Editor does not always protect for insertions:
      //ANY and VALIDATION sychronizations are allowed to write in the past.
      if (this->tagInfo().size > 0 && this->tagInfo().lastInterval.since >= m_run) {
        edm::LogInfo("PopConESTransitionSourceHandler")
            << "["
            << "PopConESTransitionSourceHandler::" << __func__ << "]: "
            << "last IOV " << this->tagInfo().lastInterval.since
            << (this->tagInfo().lastInterval.since == m_run ? " is equal to" : " is larger than")
            << " the run proposed for insertion " << m_run << ". No transfer needed." << std::endl;
        return;
      }
      bool isLowGain = checkLowGain();
      getObjectsForESTransition(isLowGain);
      edm::LogInfo("PopConESTransitionSourceHandler") << "["
                                                      << "PopConESTransitionSourceHandler::" << __func__ << "]: "
                                                      << "END." << std::endl;
    }

  private:
    unsigned long long m_run;
    double m_currentThreshold;
    // for reading from CondDB the current from RunInfo
    std::string m_tagForRunInfo;
    // for reading from CondDB the Low Gain and High Gain payloads
    std::string m_ESGain;
    std::string m_tagForLowGain;
    std::string m_tagForHighGain;
    std::string m_connectionString;
    edm::ParameterSet m_connectionPset;
    cond::persistency::ConnectionPool m_connection;
  };
}  //namespace popcon

#endif  //PopConESTransitionSourceHandler_H