CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Private Member Functions | Private Attributes
ProbeTreeProducer Class Reference

#include <ProbeTreeProducer.cc>

Inheritance diagram for ProbeTreeProducer:
edm::EDFilter edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 ProbeTreeProducer (const edm::ParameterSet &)
 
 ~ProbeTreeProducer ()
 
- Public Member Functions inherited from edm::EDFilter
 EDFilter ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDFilter ()
 
- Public Member Functions inherited from edm::ProducerBase
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription
const &)> 
registrationCallback () const
 used by the fwk to register list of products More...
 
virtual ~ProducerBase ()
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () 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 (const std::string &iProcessName, std::vector< const char * > &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 Member Functions

virtual void endJob () override
 
virtual bool filter (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

StringCutObjectSelector
< reco::Candidate, true > 
cut_
 The selector object. More...
 
bool filter_
 Specifies whether this module should filter. More...
 
int32_t maxProbes_
 The number of first probes used to fill the tree. More...
 
std::auto_ptr
< tnp::BaseTreeFiller
probeFiller_
 The object that actually computes variables and fills the tree for the probe. More...
 
edm::EDGetTokenT
< reco::CandidateView
probesToken_
 InputTag to the collection of all probes. More...
 
std::string sortDescendingBy_
 Name of the reco::Candidate function used for sorting. More...
 
StringObjectFunction
< reco::Candidate, true > 
sortFunction_
 The StringObjectFunction itself. More...
 

Additional Inherited Members

- Public Types inherited from edm::EDFilter
typedef EDFilter ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef
ProductRegistryHelper::TypeLabelList 
TypeLabelList
 
- Static Public Member Functions inherited from edm::EDFilter
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &)
 
- Protected Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
- 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

Description: TTree producer based on input probe parameters

Implementation: <Notes on="" implementation>="">

Definition at line 29 of file ProbeTreeProducer.cc.

Constructor & Destructor Documentation

ProbeTreeProducer::ProbeTreeProducer ( const edm::ParameterSet iConfig)
explicit

Definition at line 60 of file ProbeTreeProducer.cc.

60  :
61  probesToken_(consumes<reco::CandidateView>(iConfig.getParameter<edm::InputTag>("src"))),
62  cut_(iConfig.existsAs<std::string>("cut") ? iConfig.getParameter<std::string>("cut") : ""),
63  filter_(iConfig.existsAs<bool>("filter") ? iConfig.getParameter<bool>("filter") : false),
64  sortDescendingBy_(iConfig.existsAs<std::string>("sortDescendingBy") ? iConfig.getParameter<std::string>("sortDescendingBy") : ""),
65  sortFunction_(sortDescendingBy_.size()>0 ? sortDescendingBy_ : "pt"), //need to pass a valid default
66  maxProbes_(iConfig.existsAs<int32_t>("maxProbes") ? iConfig.getParameter<int32_t>("maxProbes") : -1),
67  probeFiller_(new tnp::BaseTreeFiller("probe_tree", iConfig, consumesCollector()))
68 {
69 }
T getParameter(std::string const &) const
bool existsAs(std::string const &parameterName, bool trackiness=true) const
checks if a parameter exists as a given type
Definition: ParameterSet.h:185
StringObjectFunction< reco::Candidate, true > sortFunction_
The StringObjectFunction itself.
int32_t maxProbes_
The number of first probes used to fill the tree.
ConsumesCollector consumesCollector()
Use a ConsumesCollector to gather consumes information from helper functions.
StringCutObjectSelector< reco::Candidate, true > cut_
The selector object.
std::auto_ptr< tnp::BaseTreeFiller > probeFiller_
The object that actually computes variables and fills the tree for the probe.
std::string sortDescendingBy_
Name of the reco::Candidate function used for sorting.
bool filter_
Specifies whether this module should filter.
edm::EDGetTokenT< reco::CandidateView > probesToken_
InputTag to the collection of all probes.
ProbeTreeProducer::~ProbeTreeProducer ( )

Definition at line 71 of file ProbeTreeProducer.cc.

71  {
72 }

Member Function Documentation

void ProbeTreeProducer::endJob ( void  )
overrideprivatevirtual

Reimplemented from edm::EDFilter.

Definition at line 99 of file ProbeTreeProducer.cc.

References edm::getProcessParameterSet(), and probeFiller_.

99  {
100  // ask to write the current PSet info into the TTree header
101  probeFiller_->writeProvenance(edm::getProcessParameterSet());
102 }
ParameterSet const & getProcessParameterSet()
Definition: Registry.cc:85
std::auto_ptr< tnp::BaseTreeFiller > probeFiller_
The object that actually computes variables and fills the tree for the probe.
bool ProbeTreeProducer::filter ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivatevirtual

Implements edm::EDFilter.

Definition at line 74 of file ProbeTreeProducer.cc.

References cut_, filter_, plotBeamSpotDB::first, edm::Event::getByToken(), i, edm::HandleBase::isValid(), maxProbes_, min(), probeFiller_, probesToken_, query::result, edm::second(), python.multivaluedict::sort(), sortDescendingBy_, and sortFunction_.

74  {
75  bool result = !filter_;
77  iEvent.getByToken(probesToken_, probes);
78  if(!probes.isValid()) return result;
79  probeFiller_->init(iEvent);
80  // select probes and calculate the sorting value
81  typedef std::pair<reco::CandidateBaseRef, double> Pair;
82  std::vector<Pair> selectedProbes;
83  for (size_t i = 0; i < probes->size(); ++i){
84  const reco::CandidateBaseRef &probe = probes->refAt(i);
85  if(cut_(*probe)){
86  selectedProbes.push_back(Pair(probe, sortFunction_(*probe)));
87  }
88  }
89  // sort only if a function was provided
90  if(sortDescendingBy_.size()>0) sort(selectedProbes.begin(), selectedProbes.end(), boost::bind(&Pair::second, _1) > boost::bind(&Pair::second, _2));
91  // fill the first maxProbes_ into the tree
92  for (size_t i = 0; i < (maxProbes_<0 ? selectedProbes.size() : std::min((size_t)maxProbes_, selectedProbes.size())); ++i){
93  probeFiller_->fill(selectedProbes[i].first);
94  result = true;
95  }
96  return result;
97 }
int i
Definition: DBlmapReader.cc:9
StringObjectFunction< reco::Candidate, true > sortFunction_
The StringObjectFunction itself.
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:449
int32_t maxProbes_
The number of first probes used to fill the tree.
U second(std::pair< T, U > const &p)
tuple result
Definition: query.py:137
T min(T a, T b)
Definition: MathUtil.h:58
bool isValid() const
Definition: HandleBase.h:75
StringCutObjectSelector< reco::Candidate, true > cut_
The selector object.
std::auto_ptr< tnp::BaseTreeFiller > probeFiller_
The object that actually computes variables and fills the tree for the probe.
std::string sortDescendingBy_
Name of the reco::Candidate function used for sorting.
bool filter_
Specifies whether this module should filter.
edm::EDGetTokenT< reco::CandidateView > probesToken_
InputTag to the collection of all probes.

Member Data Documentation

StringCutObjectSelector<reco::Candidate, true> ProbeTreeProducer::cut_
private

The selector object.

Definition at line 42 of file ProbeTreeProducer.cc.

Referenced by filter().

bool ProbeTreeProducer::filter_
private

Specifies whether this module should filter.

Definition at line 45 of file ProbeTreeProducer.cc.

Referenced by filter().

int32_t ProbeTreeProducer::maxProbes_
private

The number of first probes used to fill the tree.

Definition at line 54 of file ProbeTreeProducer.cc.

Referenced by filter().

std::auto_ptr<tnp::BaseTreeFiller> ProbeTreeProducer::probeFiller_
private

The object that actually computes variables and fills the tree for the probe.

Definition at line 57 of file ProbeTreeProducer.cc.

Referenced by endJob(), and filter().

edm::EDGetTokenT<reco::CandidateView> ProbeTreeProducer::probesToken_
private

InputTag to the collection of all probes.

Definition at line 39 of file ProbeTreeProducer.cc.

Referenced by filter().

std::string ProbeTreeProducer::sortDescendingBy_
private

Name of the reco::Candidate function used for sorting.

Definition at line 48 of file ProbeTreeProducer.cc.

Referenced by filter().

StringObjectFunction<reco::Candidate, true> ProbeTreeProducer::sortFunction_
private

The StringObjectFunction itself.

Definition at line 51 of file ProbeTreeProducer.cc.

Referenced by filter().