TreeSplitter Class Reference

#include <MuonAnalysis/MomentumScaleCalibration/plugins/TreeSplitter.cc>

Inheritance diagram for TreeSplitter:

Public Member Functions
	TreeSplitter (const edm::ParameterSet &)
	~TreeSplitter ()
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 ()
	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 &	itemsToGetFromEvent () 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
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
virtual	~EDConsumerBase () noexcept(false)

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

Private Attributes
int32_t	maxEvents_
TString	outputFileName_
uint32_t	subSampleFirstEvent_
uint32_t	subSampleMaxEvents_
TString	treeFileName_

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

Description: <one line="" class="" summary>="">

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

Definition at line 28 of file TreeSplitter.h.

Constructor & Destructor Documentation

TreeSplitter::TreeSplitter ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 6 of file TreeSplitter.cc.

                                                         :
  treeFileName_( iConfig.getParameter<std::string>("InputFileName") ),
  outputFileName_( iConfig.getParameter<std::string>("OutputFileName") ),
  maxEvents_( iConfig.getParameter<int32_t>("MaxEvents") ),
  subSampleFirstEvent_( iConfig.getParameter<uint32_t>("SubSampleFirstEvent") ),
  subSampleMaxEvents_( iConfig.getParameter<uint32_t>("SubSampleMaxEvents") )
{
}

TreeSplitter::~TreeSplitter

(

)

Definition at line 15 of file TreeSplitter.cc.

{
}

Member Function Documentation

virtual void TreeSplitter::analyze ( const edm::Event &  , const edm::EventSetup &    )

inlineprivatevirtual

Definition at line 35 of file TreeSplitter.h.

References endJob().

{};

void TreeSplitter::endJob ( void  )

privatevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 19 of file TreeSplitter.cc.

References gather_cfg::cout, DEFINE_FWK_MODULE, mps_fire::i, maxEvents_, outputFileName_, RootTreeHandler::readTree(), subSampleFirstEvent_, subSampleMaxEvents_, treeFileName_, and RootTreeHandler::writeTree().

Referenced by analyze().

{
  std::cout << "Reading muon pairs from Root Tree in " << treeFileName_ << std::endl;
  RootTreeHandler rootTreeHandler;

  typedef std::vector<std::pair<lorentzVector,lorentzVector> > MuonPairVector;
  // MuonPairVector savedPair;
  std::vector<MuonPair> savedPair;
  rootTreeHandler.readTree(maxEvents_, treeFileName_, &savedPair, 0);
  // rootTreeHandler.readTree(maxEvents, inputRootTreeFileName_, &savedPair, &(MuScleFitUtils::genPair));

  // Loop on all the pairs
  std::vector<MuonPair> newSavedPair;
  // MuonPairVector newSavedPair;
  unsigned int i = 0;
  // MuonPairVector::iterator it = savedPair.begin();
  std::vector<MuonPair>::iterator it = savedPair.begin();
  std::cout << "Starting loop on " << savedPair.size() << " muons" << std::endl;
  uint32_t lastEvent = subSampleFirstEvent_ + subSampleMaxEvents_;
  for( ; it != savedPair.end(); ++it, ++i ) {
    // Save only events in the selected range
    if( i >= subSampleFirstEvent_ && i < lastEvent ) {
      newSavedPair.push_back(*it);
    }
  }
  rootTreeHandler.writeTree(outputFileName_, &newSavedPair, 0);
}

Member Data Documentation

int32_t TreeSplitter::maxEvents_

private

Definition at line 40 of file TreeSplitter.h.

Referenced by endJob().

TString TreeSplitter::outputFileName_

private

Definition at line 39 of file TreeSplitter.h.

Referenced by endJob().

uint32_t TreeSplitter::subSampleFirstEvent_

private

Definition at line 41 of file TreeSplitter.h.

Referenced by endJob().

uint32_t TreeSplitter::subSampleMaxEvents_

private

Definition at line 42 of file TreeSplitter.h.

Referenced by endJob().

TString TreeSplitter::treeFileName_

private

Definition at line 38 of file TreeSplitter.h.

Referenced by endJob().