CMS 3D CMS Logo

List of all members | Classes | Public Member Functions | Private Types | Private Member Functions | Private Attributes
GBRForestWriter Class Reference

#include <GBRForestWriter.h>

Inheritance diagram for GBRForestWriter:
edm::EDAnalyzer edm::EDConsumerBase

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 ()
 
SerialTaskQueueglobalLuminosityBlocksQueue ()
 
SerialTaskQueueglobalRunsQueue ()
 
ModuleDescription const & moduleDescription () const
 
std::string workerType () const
 
 ~EDAnalyzer () override
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () 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
 
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
 
EDConsumerBaseoperator= (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)
 
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 ESProduct , Transition Tr = Transition::Event>
auto esConsumes (eventsetup::EventSetupRecordKey const &, 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 21 of file GBRForestWriter.h.

Member Typedef Documentation

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

Definition at line 33 of file GBRForestWriter.h.

Constructor & Destructor Documentation

GBRForestWriter::GBRForestWriter ( const edm::ParameterSet cfg)

Definition at line 12 of file GBRForestWriter.cc.

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

13  : moduleLabel_(cfg.getParameter<std::string>("@module_label"))
14 {
15  edm::VParameterSet cfgJobs = cfg.getParameter<edm::VParameterSet>("jobs");
16  for (edm::VParameterSet::const_iterator cfgJob = cfgJobs.begin(); cfgJob != cfgJobs.end(); ++cfgJob) {
17  jobEntryType* job = new jobEntryType(*cfgJob);
18  jobs_.push_back(job);
19  }
20 }
T getParameter(std::string const &) const
std::vector< ParameterSet > VParameterSet
Definition: ParameterSet.h:33
std::string moduleLabel_
std::vector< jobEntryType * > jobs_
GBRForestWriter::~GBRForestWriter ( )
override

Definition at line 22 of file GBRForestWriter.cc.

References jobs_.

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

Member Function Documentation

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

Definition at line 29 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().

30 {
31 
32  for (std::vector<jobEntryType*>::iterator job = jobs_.begin(); job != jobs_.end(); ++job) {
33  std::map<std::string, const GBRForest*> gbrForests; // key = name
34  for (std::vector<categoryEntryType*>::iterator category = (*job)->categories_.begin();
35  category != (*job)->categories_.end(); ++category) {
36  const GBRForest* gbrForest = nullptr;
37  if ((*category)->inputFileType_ == categoryEntryType::kXML) {
38  gbrForest = createGBRForest((*category)->inputFileName_).release();
39  } else if ((*category)->inputFileType_ == categoryEntryType::kGBRForest) {
40  TFile* inputFile = new TFile((*category)->inputFileName_.data());
41  // gbrForest = dynamic_cast<GBRForest*>(inputFile->Get((*category)->gbrForestName_.data())); // CV:
42  // dynamic_cast<GBRForest*> fails for some reason ?!
43  gbrForest = (GBRForest*)inputFile->Get((*category)->gbrForestName_.data());
44  delete inputFile;
45  }
46  if (!gbrForest)
47  throw cms::Exception("GBRForestWriter")
48  << " Failed to load GBRForest = " << (*category)->gbrForestName_.data()
49  << " from file = " << (*category)->inputFileName_ << " !!\n";
50  gbrForests[(*category)->gbrForestName_] = gbrForest;
51  }
52  if ((*job)->outputFileType_ == jobEntryType::kGBRForest) {
53  TFile* outputFile = new TFile((*job)->outputFileName_.data(), "RECREATE");
54 
55  for (std::map<std::string, const GBRForest*>::iterator gbrForest = gbrForests.begin();
56  gbrForest != gbrForests.end(); ++gbrForest) {
57  outputFile->WriteObject(gbrForest->second, gbrForest->first.data());
58  }
59  delete outputFile;
60  } else if ((*job)->outputFileType_ == jobEntryType::kSQLLite) {
62  if (!dbService.isAvailable())
63  throw cms::Exception("GBRForestWriter") << " Failed to access PoolDBOutputService !!\n";
64 
65  for (std::map<std::string, const GBRForest*>::iterator gbrForest = gbrForests.begin();
66  gbrForest != gbrForests.end(); ++gbrForest) {
67  std::string outputRecord = (*job)->outputRecord_;
68  if (gbrForests.size() > 1)
69  outputRecord.append("_").append(gbrForest->first);
70  dbService->writeOne(gbrForest->second, dbService->beginOfTime(), outputRecord);
71  }
72  }
73 
74  // gbrforest deletion
75  for (std::map<std::string, const GBRForest*>::iterator gbrForest = gbrForests.begin();
76  gbrForest != gbrForests.end(); ++gbrForest) {
77  delete gbrForest->second;
78  }
79  }
80 }
bool isAvailable() const
Definition: Service.h:46
void writeOne(T *payload, Time_t time, const std::string &recordName, bool withlogging=false)
std::vector< jobEntryType * > jobs_
std::unique_ptr< const GBRForest > createGBRForest(const std::string &weightsFile)

Member Data Documentation

bool GBRForestWriter::hasRun_
private

Definition at line 31 of file GBRForestWriter.h.

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

Definition at line 114 of file GBRForestWriter.h.

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

std::string GBRForestWriter::moduleLabel_
private