GBRForestWriter Class Reference

Inheritance diagram for GBRForestWriter:

Classes
struct	categoryEntryType
struct	jobEntryType

Public Member Functions
	GBRForestWriter (const edm::ParameterSet &)
	~GBRForestWriter () override
Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	EDAnalyzer ()
SerialTaskQueue *	globalLuminosityBlocksQueue ()
SerialTaskQueue *	globalRunsQueue ()
ModuleDescription const &	moduleDescription () const
std::string	workerType () const
	~EDAnalyzer () override
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const
ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)
virtual	~EDConsumerBase () noexcept(false)

Private Types
typedef std::vector< std::string >	vstring

Private Member Functions
void	analyze (const edm::Event &, const edm::EventSetup &) override

Private Attributes
bool	hasRun_
std::vector< jobEntryType * >	jobs_
std::string	moduleLabel_

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &)
static bool	wantsGlobalLuminosityBlocks ()
static bool	wantsGlobalRuns ()
static bool	wantsStreamLuminosityBlocks ()
static bool	wantsStreamRuns ()
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Read GBRForest objects from ROOT file input and store it in SQL-lite output file

Authors

Christian Veelken, LLR

Definition at line 24 of file GBRForestWriter.cc.

Member Typedef Documentation

typedef std::vector<std::string> GBRForestWriter::vstring

private

Definition at line 36 of file GBRForestWriter.cc.

Constructor & Destructor Documentation

GBRForestWriter::GBRForestWriter

(

const edm::ParameterSet &

cfg

)

Definition at line 116 of file GBRForestWriter.cc.

References edm::ParameterSet::getParameter(), and jobs_.

    : moduleLabel_(cfg.getParameter<std::string>("@module_label")) {
  edm::VParameterSet cfgJobs = cfg.getParameter<edm::VParameterSet>("jobs");
  for (edm::VParameterSet::const_iterator cfgJob = cfgJobs.begin(); cfgJob != cfgJobs.end(); ++cfgJob) {
    jobEntryType* job = new jobEntryType(*cfgJob);
    jobs_.push_back(job);
  }
}

GBRForestWriter::~GBRForestWriter ( )

override

Definition at line 125 of file GBRForestWriter.cc.

References jobs_.

                                  {
  for (std::vector<jobEntryType*>::iterator it = jobs_.begin(); it != jobs_.end(); ++it) {
    delete (*it);
  }
}

Member Function Documentation

void GBRForestWriter::analyze ( const edm::Event &  , const edm::EventSetup &    )

overrideprivate

Definition at line 131 of file GBRForestWriter.cc.

References cond::service::PoolDBOutputService::beginOfTime(), taus_updatedMVAIds_cff::category, createGBRForest(), DEFINE_FWK_MODULE, Exception, analyzePatCleaning_cfg::inputFile, edm::Service< T >::isAvailable(), jobs_, GBRForestWriter::categoryEntryType::kGBRForest, GBRForestWriter::jobEntryType::kGBRForest, GBRForestWriter::jobEntryType::kSQLLite, GBRForestWriter::categoryEntryType::kXML, download_sqlite_cfg::outputFile, writeAntiElectronDiscrMVA_cfi::outputRecord, AlCaHLTBitMon_QueryRunRegistry::string, and cond::service::PoolDBOutputService::writeOne().

                                                                   {
  for (std::vector<jobEntryType*>::iterator job = jobs_.begin(); job != jobs_.end(); ++job) {
    std::map<std::string, const GBRForest*> gbrForests;  // key = name
    for (std::vector<categoryEntryType*>::iterator category = (*job)->categories_.begin();
         category != (*job)->categories_.end();
         ++category) {
      const GBRForest* gbrForest = nullptr;
      if ((*category)->inputFileType_ == categoryEntryType::kXML) {
        gbrForest = createGBRForest((*category)->inputFileName_).release();
      } else if ((*category)->inputFileType_ == categoryEntryType::kGBRForest) {
        TFile* inputFile = new TFile((*category)->inputFileName_.data());
        // gbrForest = dynamic_cast<GBRForest*>(inputFile->Get((*category)->gbrForestName_.data())); // CV:
        // dynamic_cast<GBRForest*> fails for some reason ?!
        gbrForest = (GBRForest*)inputFile->Get((*category)->gbrForestName_.data());
        delete inputFile;
      }
      if (!gbrForest)
        throw cms::Exception("GBRForestWriter") << " Failed to load GBRForest = " << (*category)->gbrForestName_.data()
                                                << " from file = " << (*category)->inputFileName_ << " !!\n";
      gbrForests[(*category)->gbrForestName_] = gbrForest;
    }
    if ((*job)->outputFileType_ == jobEntryType::kGBRForest) {
      TFile* outputFile = new TFile((*job)->outputFileName_.data(), "RECREATE");

      for (std::map<std::string, const GBRForest*>::iterator gbrForest = gbrForests.begin();
           gbrForest != gbrForests.end();
           ++gbrForest) {
        outputFile->WriteObject(gbrForest->second, gbrForest->first.data());
      }
      delete outputFile;
    } else if ((*job)->outputFileType_ == jobEntryType::kSQLLite) {
      edm::Service<cond::service::PoolDBOutputService> dbService;
      if (!dbService.isAvailable())
        throw cms::Exception("GBRForestWriter") << " Failed to access PoolDBOutputService !!\n";

      for (std::map<std::string, const GBRForest*>::iterator gbrForest = gbrForests.begin();
           gbrForest != gbrForests.end();
           ++gbrForest) {
        std::string outputRecord = (*job)->outputRecord_;
        if (gbrForests.size() > 1)
          outputRecord.append("_").append(gbrForest->first);
        dbService->writeOne(gbrForest->second, dbService->beginOfTime(), outputRecord);
      }
    }

    // gbrforest deletion
    for (std::map<std::string, const GBRForest*>::iterator gbrForest = gbrForests.begin();
         gbrForest != gbrForests.end();
         ++gbrForest) {
      delete gbrForest->second;
    }
  }
}

Member Data Documentation

bool GBRForestWriter::hasRun_

private

Definition at line 34 of file GBRForestWriter.cc.

std::vector<jobEntryType*> GBRForestWriter::jobs_

private

Definition at line 113 of file GBRForestWriter.cc.

Referenced by analyze(), GBRForestWriter(), and ~GBRForestWriter().

std::string GBRForestWriter::moduleLabel_

private

Definition at line 32 of file GBRForestWriter.cc.

Referenced by ExternalGeneratorFilter.ExternalGeneratorFilter::insertInto(), Modules.SwitchProducer::insertInto(), Mixins._TypedParameterizable::insertInto(), Modules.ESSource::nameInProcessDesc_(), and Modules.ESProducer::nameInProcessDesc_().