TRGDummy2DB.cc

Go to the documentation of this file.

#ifndef RecoLuminosity_LumiProducer_TRGDummy2DB_h
#define RecoLuminosity_LumiProducer_TRGDummy2DB_h
#include "RelationalAccess/ConnectionService.h"
#include "RelationalAccess/ISessionProxy.h"
#include "RelationalAccess/ITransaction.h"
#include "RelationalAccess/ISchema.h"
#include "RelationalAccess/ITable.h"
#include "RelationalAccess/ITableDataEditor.h"
#include "RelationalAccess/IBulkOperation.h"
#include "RelationalAccess/ITypeConverter.h"
#include "CoralBase/AttributeList.h"
#include "CoralBase/AttributeSpecification.h"
#include "CoralBase/Attribute.h"
#include "CoralBase/Exception.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 <cstdio>
namespace lumi {
  class TRGDummy2DB : public DataPipe {
  public:
    TRGDummy2DB(const std::string& dest);
    unsigned long long retrieveData(unsigned int) override;
    const std::string dataType() const override;
    const std::string sourceType() const override;
    ~TRGDummy2DB() override;
  };  //cl TRGDummy2DB
  //
  //implementation
  //
  TRGDummy2DB::TRGDummy2DB(const std::string& dest) : DataPipe(dest) {}
  unsigned long long TRGDummy2DB::retrieveData(unsigned int runnum) {
    //
    //generate dummy data of trg for the given run and write data to LumiDB
    //
    coral::ConnectionService* svc = new coral::ConnectionService;
    lumi::DBConfig dbconf(*svc);

    if (!m_authpath.empty()) {
      dbconf.setAuthentication(m_authpath);
    }
    coral::ISessionProxy* session = svc->connect(m_dest, coral::Update);
    coral::ITypeConverter& tpc = session->typeConverter();
    tpc.setCppTypeForSqlType("unsigned int", "NUMBER(7)");
    tpc.setCppTypeForSqlType("unsigned int", "NUMBER(10)");
    tpc.setCppTypeForSqlType("unsigned long long", "NUMBER(20)");
    try {
      unsigned int totalcmsls = 32;
      session->transaction().start(false);
      coral::ISchema& schema = session->nominalSchema();
      lumi::idDealer idg(schema);
      coral::ITable& trgtable = schema.tableHandle(LumiNames::trgTableName());
      coral::AttributeList trgData;
      trgData.extend("TRG_ID", typeid(unsigned long long));
      trgData.extend("RUNNUM", typeid(unsigned int));
      trgData.extend("CMSLSNUM", typeid(unsigned int));
      trgData.extend("BITNUM", typeid(unsigned int));
      trgData.extend("BITNAME", typeid(std::string));
      trgData.extend("TRGCOUNT", typeid(unsigned int));
      trgData.extend("DEADTIME", typeid(unsigned long long));
      trgData.extend("PRESCALE", typeid(unsigned int));
      coral::IBulkOperation* trgInserter =
          trgtable.dataEditor().bulkInsert(trgData, totalcmsls * (lumi::N_TRGALGOBIT + lumi::N_TRGTECHBIT));
      //loop over lumi LS
      unsigned long long& trg_id = trgData["TRG_ID"].data<unsigned long long>();
      unsigned int& trgrunnum = trgData["RUNNUM"].data<unsigned int>();
      unsigned int& cmslsnum = trgData["CMSLSNUM"].data<unsigned int>();
      unsigned int& bitnum = trgData["BITNUM"].data<unsigned int>();
      std::string& bitname = trgData["BITNAME"].data<std::string>();
      unsigned int& count = trgData["TRGCOUNT"].data<unsigned int>();
      unsigned long long& deadtime = trgData["DEADTIME"].data<unsigned long long>();
      unsigned int& prescale = trgData["PRESCALE"].data<unsigned int>();

      for (unsigned int i = 1; i <= totalcmsls; ++i) {
        for (unsigned int j = 0; j < (lumi::N_TRGALGOBIT + lumi::N_TRGTECHBIT); ++j) {  //total n of trg bits
          trg_id = idg.generateNextIDForTable(LumiNames::trgTableName());
          trgrunnum = runnum;
          cmslsnum = i;
          bitnum = j;
          char c[10];
          if (j > (lumi::N_TRGALGOBIT - 1)) {
            ::sprintf(c, "%d", j - (lumi::N_TRGALGOBIT - 1));
            bitname = std::string(c);
          } else {
            ::sprintf(c, "%d", j);
            bitname = std::string("FakeTRGBIT_") + std::string(c);
          }
          count = 12345;
          deadtime = 6785;
          prescale = 1;
          trgInserter->processNextIteration();
        }
      }
      trgInserter->flush();
      delete trgInserter;
    } catch (const coral::Exception& er) {
      std::cout << "database problem " << er.what() << std::endl;
      session->transaction().rollback();
      delete session;
      delete svc;
      throw er;
    }
    //delete detailInserter;
    session->transaction().commit();
    delete session;
    delete svc;
    return 0;
  }
  const std::string TRGDummy2DB::dataType() const { return "TRG"; }
  const std::string TRGDummy2DB::sourceType() const { return "DUMMY"; }
  TRGDummy2DB::~TRGDummy2DB() {}
}  // namespace lumi
#include "RecoLuminosity/LumiProducer/interface/DataPipeFactory.h"
DEFINE_EDM_PLUGIN(lumi::DataPipeFactory, lumi::TRGDummy2DB, "TRGDummy2DB");
#endif