GBRForestWriter Class Reference

#include <GBRForestWriter.h>

Inheritance diagram for GBRForestWriter:

Classes
struct	categoryEntryType
struct	jobEntryType

Public Member Functions
	GBRForestWriter (const edm::ParameterSet &)
	~GBRForestWriter ()
Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	EDAnalyzer ()
ModuleDescription const &	moduleDescription () const
std::string	workerType () const
virtual	~EDAnalyzer ()
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (std::string const &iProcessName, std::string const &iModuleLabel, bool iPrint, std::vector< char const * > &oModuleLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

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

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

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 &)
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 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 34 of file GBRForestWriter.h.

Constructor & Destructor Documentation

GBRForestWriter::GBRForestWriter

(

const edm::ParameterSet &

cfg

)

Definition at line 21 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

(

)

Definition at line 32 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 &    )

privatevirtual

Implements edm::EDAnalyzer.

Definition at line 40 of file GBRForestWriter.cc.

References cond::service::PoolDBOutputService::beginOfTime(), python.rootplot.argparse::category, Exception, analyzePatCleaning_cfg::inputFile, edm::Service< T >::isAvailable(), jobs_, GBRForestWriter::categoryEntryType::kGBRForest, GBRForestWriter::jobEntryType::kGBRForest, GBRForestWriter::jobEntryType::kSQLLite, GBRForestWriter::categoryEntryType::kXML, reco::details::loadTMVAWeights(), 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 ) {
    TMVA::Tools::Instance();
    TMVA::Reader* mvaReader = new TMVA::Reader("!V:!Silent");   
    std::vector<Float_t> dummyVariables;
    for ( vstring::const_iterator inputVariable = (*category)->inputVariables_.begin();
          inputVariable != (*category)->inputVariables_.end(); ++inputVariable ) {
      dummyVariables.push_back(0.);
      mvaReader->AddVariable(inputVariable->data(), &dummyVariables.back());
    }
    for ( vstring::const_iterator spectatorVariable = (*category)->spectatorVariables_.begin();
          spectatorVariable != (*category)->spectatorVariables_.end(); ++spectatorVariable ) {
      dummyVariables.push_back(0.);
      mvaReader->AddSpectator(spectatorVariable->data(), &dummyVariables.back());
    }
    reco::details::loadTMVAWeights(mvaReader, (*category)->methodName_.data(), (*category)->inputFileName_.data());
    TMVA::MethodBDT* bdt = dynamic_cast<TMVA::MethodBDT*>(mvaReader->FindMVA((*category)->methodName_.data()));
    if ( !bdt )
      throw cms::Exception("GBRForestWriter") 
        << "Failed to load MVA = " << (*category)->methodName_.data() << " from file = " << (*category)->inputFileName_ << " !!\n";
    gbrForest = new GBRForest(bdt);  
    delete mvaReader;
    TMVA::Tools::DestroyInstance();
      } 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 32 of file GBRForestWriter.h.

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

private

Definition at line 110 of file GBRForestWriter.h.

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

std::string GBRForestWriter::moduleLabel_

private

Definition at line 30 of file GBRForestWriter.h.