CMS 3D CMS Logo

List of all members | Public Member Functions | Protected Member Functions | Private Attributes
PhysicsTools::MVATrainerContainerSave Class Referenceabstract

#include <MVATrainerContainerSave.h>

Inheritance diagram for PhysicsTools::MVATrainerContainerSave:
edm::EDAnalyzer edm::EDConsumerBase PhysicsTools::MVATrainerContainerSaveImpl< Record_t >

Public Member Functions

void analyze (const edm::Event &iEvent, const edm::EventSetup &iSetup) override
 
void endJob () override
 
 MVATrainerContainerSave (const edm::ParameterSet &params)
 
- 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
 
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
 
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)
 
void updateLookup (eventsetup::ESRecordsToProxyIndices const &)
 
virtual ~EDConsumerBase () noexcept(false)
 

Protected Member Functions

virtual std::string getRecordName () const =0
 
virtual const Calibration::MVAComputerContainergetToCopy (const edm::EventSetup &es) const =0
 
virtual const Calibration::MVAComputerContainergetToPut (const edm::EventSetup &es) const =0
 
- 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)
 

Private Attributes

std::unique_ptr< Calibration::MVAComputerContainercalib
 
bool saved
 
std::vector< std::string > toCopy
 
std::vector< std::string > toPut
 

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 ()
 

Detailed Description

Definition at line 17 of file MVATrainerContainerSave.h.

Constructor & Destructor Documentation

PhysicsTools::MVATrainerContainerSave::MVATrainerContainerSave ( const edm::ParameterSet params)
explicit

Definition at line 26 of file MVATrainerContainerSave.cc.

27  :
28  toPut(params.getParameter<std::vector<std::string> >("toPut")),
29  toCopy(params.getParameter<std::vector<std::string> >("toCopy")),
30  saved(false)
31 {
32 }
T getParameter(std::string const &) const

Member Function Documentation

void PhysicsTools::MVATrainerContainerSave::analyze ( const edm::Event iEvent,
const edm::EventSetup iSetup 
)
override

Definition at line 34 of file MVATrainerContainerSave.cc.

References calib, PhysicsTools::Calibration::MVAComputerContainer::find(), getToCopy(), getToPut(), PhysicsTools::MVATrainerLooper::isUntrained(), eostools::move(), saved, toCopy, and toPut.

36 {
37  if (calib.get() || saved)
38  return;
39 
40  const Calibration::MVAComputerContainer *toPutCalib = nullptr;
41  if (!toPut.empty()) {
42  toPutCalib = getToPut(es);
43  if (MVATrainerLooper::isUntrained(toPutCalib))
44  return;
45  }
46 
47  const Calibration::MVAComputerContainer *toCopyCalib = nullptr;
48  if (!toCopy.empty())
49  toCopyCalib = getToCopy(es);
50 
51  edm::LogInfo("MVATrainerSave") << "Got the trained calibration data";
52 
53  std::unique_ptr<Calibration::MVAComputerContainer> calib(
54  new Calibration::MVAComputerContainer);
55 
56  for(std::vector<std::string>::const_iterator iter = toCopy.begin();
57  iter != toCopy.end(); iter++)
58  calib->add(*iter) = toCopyCalib->find(*iter);
59 
60  for(std::vector<std::string>::const_iterator iter = toPut.begin();
61  iter != toPut.end(); iter++)
62  calib->add(*iter) = toPutCalib->find(*iter);
63 
64  this->calib = std::move(calib);
65 }
std::unique_ptr< Calibration::MVAComputerContainer > calib
virtual const Calibration::MVAComputerContainer * getToCopy(const edm::EventSetup &es) const =0
static bool isUntrained(const T *ptr)
virtual const Calibration::MVAComputerContainer * getToPut(const edm::EventSetup &es) const =0
def move(src, dest)
Definition: eostools.py:511
void PhysicsTools::MVATrainerContainerSave::endJob ( void  )
overridevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 67 of file MVATrainerContainerSave.cc.

References cond::service::PoolDBOutputService::beginOfTime(), cond::service::PoolDBOutputService::createNewIOV(), cond::service::PoolDBOutputService::endOfTime(), getRecordName(), edm::Service< T >::isAvailable(), and saved.

68 {
69  if (!calib.get() || saved)
70  return;
71 
72  edm::LogInfo("MVATrainerSave") << "Saving calibration data in CondDB.";
73 
75  if (!dbService.isAvailable())
76  throw cms::Exception("MVATrainerContainerSave")
77  << "DBService unavailable" << std::endl;
78 
79  dbService->createNewIOV<Calibration::MVAComputerContainer>(
80  calib.release(), dbService->beginOfTime(),
81  dbService->endOfTime(), getRecordName());
82 
83  saved = true;
84 }
virtual std::string getRecordName() const =0
bool isAvailable() const
Definition: Service.h:40
void createNewIOV(T *firstPayloadObj, cond::Time_t firstSinceTime, cond::Time_t firstTillTime, const std::string &recordName, bool withlogging=false)
virtual std::string PhysicsTools::MVATrainerContainerSave::getRecordName ( ) const
protectedpure virtual
virtual const Calibration::MVAComputerContainer* PhysicsTools::MVATrainerContainerSave::getToCopy ( const edm::EventSetup es) const
protectedpure virtual
virtual const Calibration::MVAComputerContainer* PhysicsTools::MVATrainerContainerSave::getToPut ( const edm::EventSetup es) const
protectedpure virtual

Member Data Documentation

std::unique_ptr<Calibration::MVAComputerContainer> PhysicsTools::MVATrainerContainerSave::calib
private

Definition at line 38 of file MVATrainerContainerSave.h.

Referenced by analyze().

bool PhysicsTools::MVATrainerContainerSave::saved
private

Definition at line 39 of file MVATrainerContainerSave.h.

Referenced by analyze(), and endJob().

std::vector<std::string> PhysicsTools::MVATrainerContainerSave::toCopy
private

Definition at line 37 of file MVATrainerContainerSave.h.

Referenced by analyze().

std::vector<std::string> PhysicsTools::MVATrainerContainerSave::toPut
private

Definition at line 36 of file MVATrainerContainerSave.h.

Referenced by analyze().