#include <RecoVertex/KVFTest/src/KVFTest.cc>

Inheritance diagram for KVFTest:

Public Member Functions
void	analyze (const edm::Event &, const edm::EventSetup &) override

void	beginJob () override

void	endJob () override

	KVFTest (const edm::ParameterSet &)

	~KVFTest () override

Public Member Functions inherited from edm::one::EDAnalyzer<>
	EDAnalyzer ()=default

	EDAnalyzer (const EDAnalyzer &)=delete

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

const EDAnalyzer &	operator= (const EDAnalyzer &)=delete

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsInputProcessBlocks () const final

bool	wantsProcessBlocks () const final

Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDAnalyzerBase () override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () 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

std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (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::array< std::vector< ModuleDescription const * > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, 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	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
TrackingVertex	getSimVertex (const edm::Event &iEvent) const

Private Attributes
const edm::ESGetToken < MagneticField, IdealMagneticFieldRecord >	estoken_MF

const edm::ESGetToken < TransientTrackBuilder, TransientTrackRecord >	estoken_TTB

edm::ParameterSet	kvfPSet

std::string	outputFile_

TFile *	rootFile_

edm::ParameterSet	theConfig

edm::EDGetTokenT < reco::TrackToTrackingParticleAssociator >	token_associatorForParamAtPca

edm::EDGetTokenT < reco::TrackCollection >	token_tracks

edm::EDGetTokenT < TrackingParticleCollection >	token_TrackTruth

edm::EDGetTokenT < TrackingVertexCollection >	token_VertexTruth

std::unique_ptr< SimpleVertexTree >	tree

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

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<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

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)

void	resetItemsToGetFrom (BranchType iType)

Detailed Description

This is a very simple test analyzer mean to test the KalmanVertexFitter

Description: steers tracker primary vertex reconstruction and storage

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

Definition at line 38 of file KVFTest.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 19 of file KVFTest.cc.

References edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), HLT_FULL_cff::InputTag, kvfPSet, outputFile_, rootFile_, AlCaHLTBitMon_QueryRunRegistry::string, token_associatorForParamAtPca, token_tracks, token_TrackTruth, and token_VertexTruth.

     : estoken_MF(esConsumes()),
       estoken_TTB(esConsumes(edm::ESInputTag("", "TransientTrackBuilder"))),
       theConfig(iConfig) {
   token_tracks = consumes<TrackCollection>(iConfig.getParameter<string>("TrackLabel"));
   outputFile_ = iConfig.getUntrackedParameter<std::string>("outputFile");
   kvfPSet = iConfig.getParameter<edm::ParameterSet>("KVFParameters");
   rootFile_ = TFile::Open(outputFile_.c_str(), "RECREATE");
   edm::LogInfo("RecoVertex/KVFTest") << "Initializing KVF TEST analyser  - Output file: " << outputFile_ << "\n";
 
   token_TrackTruth = consumes<TrackingParticleCollection>(edm::InputTag("trackingtruth", "TrackTruth"));
   token_VertexTruth = consumes<TrackingVertexCollection>(edm::InputTag("trackingtruth", "VertexTruth"));
   token_associatorForParamAtPca =
       consumes<reco::TrackToTrackingParticleAssociator>(edm::InputTag("trackAssociatorByChi2"));
 }

KVFTest::~KVFTest ( )

override

Definition at line 35 of file KVFTest.cc.

References rootFile_.

35 { delete rootFile_; }

KVFTest::rootFile_

TFile * rootFile_

Definition: KVFTest.h:57

Member Function Documentation

void KVFTest::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overridevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 45 of file KVFTest.cc.

References estoken_MF, estoken_TTB, edm::Event::getByToken(), edm::EventSetup::getData(), getSimVertex(), edm::EventBase::id(), edm::HandleBase::isValid(), TransientVertex::position(), edm::Handle< T >::product(), token_associatorForParamAtPca, token_tracks, token_TrackTruth, tree, and KalmanVertexFitter::vertex().

                                                                          {
   edm::Handle<reco::TrackToTrackingParticleAssociator> associatorForParamAtPca;
   iEvent.getByToken(token_associatorForParamAtPca, associatorForParamAtPca);
 
   if (not tree) {
     tree = std::make_unique<SimpleVertexTree>("VertexFitter", &iSetup.getData(estoken_MF));
   }
 
   edm::LogInfo("RecoVertex/KVFTest") << "Reconstructing event number: " << iEvent.id() << "\n";
 
   // get RECO tracks from the event
   // `tks` can be used as a ptr to a reco::TrackCollection
   edm::Handle<edm::View<reco::Track> > tks;
   iEvent.getByToken(token_tracks, tks);
   if (!tks.isValid()) {
     edm::LogInfo("RecoVertex/KVFTest") << "Exception during event number: " << iEvent.id() << "\n";
   } else {
     edm::LogInfo("RecoVertex/KVFTest") << "Found: " << (*tks).size() << " reconstructed tracks"
                                        << "\n";
     edm::LogPrint("RecoVertex/KVFTest") << "got " << (*tks).size() << " tracks " << std::endl;
 
     // Transform Track to TransientTrack
 
     //get the builder:
     const auto& theB = &iSetup.getData(estoken_TTB);
     //do the conversion:
     std::vector<TransientTrack> t_tks = theB->build(tks);
 
     edm::LogInfo("RecoVertex/KVFTest") << "Found: " << t_tks.size() << " reconstructed tracks"
                                        << "\n";
 
     // Call the KalmanVertexFitter if more than 1 track
     if (t_tks.size() > 1) {
       //      KalmanVertexFitter kvf(kvfPSet);
       KalmanVertexFitter kvf(true);
       TransientVertex tv = kvf.vertex(t_tks);
 
       edm::LogPrint("RecoVertex/KVFTest") << "Position: " << Vertex::Point(tv.position()) << "\n";
 
       // For the analysis: compare to your SimVertex
       TrackingVertex sv = getSimVertex(iEvent);
       edm::Handle<TrackingParticleCollection> TPCollectionH;
       iEvent.getByToken(token_TrackTruth, TPCollectionH);
       if (!TPCollectionH.isValid()) {
         edm::LogInfo("RecoVertex/KVFTest") << "Exception during event number: " << iEvent.id() << "\n";
       } else {
         const TrackingParticleCollection tPC = *(TPCollectionH.product());
         reco::RecoToSimCollection recSimColl = associatorForParamAtPca->associateRecoToSim(tks, TPCollectionH);
         tree->fill(tv, &sv, &recSimColl);
       }
     }
   }
 }

void KVFTest::beginJob ( void )

overridevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 37 of file KVFTest.cc.

37 {}

void KVFTest::endJob ( void )

overridevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 39 of file KVFTest.cc.

39 {}

TrackingVertex KVFTest::getSimVertex ( const edm::Event & iEvent ) const

private

Definition at line 101 of file KVFTest.cc.

References edm::Event::getByToken(), edm::Handle< T >::product(), and token_VertexTruth.

Referenced by analyze().

                                                                  {
   // get the simulated vertices
   edm::Handle<TrackingVertexCollection> TVCollectionH;
   iEvent.getByToken(token_VertexTruth, TVCollectionH);
   const TrackingVertexCollection tPC = *(TVCollectionH.product());
 
   //    Handle<edm::SimVertexContainer> simVtcs;
   //    iEvent.getByLabel("g4SimHits", simVtcs);
   //    std::cout << "SimVertex " << simVtcs->size() << std::endl;
   //    for(edm::SimVertexContainer::const_iterator v=simVtcs->begin();
   //        v!=simVtcs->end(); ++v){
   //      std::cout << "simvtx "
   //           << v->position().x() << " "
   //           << v->position().y() << " "
   //           << v->position().z() << " "
   //           << v->parentIndex() << " "
   //           << v->noParent() << " "
   //               << std::endl;
   //    }
   return *(tPC.begin());
 }