KVFTest Class Reference

#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::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	EDAnalyzer ()
SerialTaskQueue *	globalLuminosityBlocksQueue ()
SerialTaskQueue *	globalRunsQueue ()
ModuleDescription const &	moduleDescription () const
std::string	workerType () const
	~EDAnalyzer () 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
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
EDConsumerBase &	operator= (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)

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

Private Attributes
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::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 ()
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)

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 37 of file KVFTest.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 25 of file KVFTest.cc.

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

  : 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 42 of file KVFTest.cc.

References rootFile_.

                  {
  delete rootFile_;
}

Member Function Documentation

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

override

Definition at line 58 of file KVFTest.cc.

References reco::TrackToTrackingParticleAssociator::associateRecoToSim(), gather_cfg::cout, edm::EventSetup::get(), edm::Event::getByToken(), getSimVertex(), edm::EventBase::id(), edm::HandleBase::isValid(), TransientVertex::position(), edm::Handle< T >::product(), edm::ESHandle< T >::product(), pfDeepBoostedJetPreprocessParams_cfi::sv, token_associatorForParamAtPca, token_tracks, token_TrackTruth, and KalmanVertexFitter::vertex().

{
  edm::Handle<reco::TrackToTrackingParticleAssociator> associatorForParamAtPca;
  iEvent.getByToken(token_associatorForParamAtPca,associatorForParamAtPca);

  if(not tree) {
    edm::ESHandle<MagneticField> magField;
    iSetup.get<IdealMagneticFieldRecord>().get(magField);
    tree.reset( new SimpleVertexTree("VertexFitter", magField.product()) );
  }



  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";
    std::cout << "got " << (*tks).size() << " tracks " << std::endl;
    
    // Transform Track to TransientTrack

    //get the builder:
    edm::ESHandle<TransientTrackBuilder> theB;
    iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder",theB);
    //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);

      std::cout << "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::EDAnalyzer.

Definition at line 46 of file KVFTest.cc.

                      {
}

void KVFTest::endJob ( void  )

overridevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 50 of file KVFTest.cc.

                     {
}

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

private

Definition at line 126 of file KVFTest.cc.

References DEFINE_FWK_MODULE, 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());
}

Member Data Documentation

edm::ParameterSet KVFTest::kvfPSet

private

Definition at line 52 of file KVFTest.h.

Referenced by KVFTest().

std::string KVFTest::outputFile_

private

Definition at line 56 of file KVFTest.h.

Referenced by KVFTest().

TFile* KVFTest::rootFile_

private

Definition at line 54 of file KVFTest.h.

Referenced by KVFTest(), and ~KVFTest().

edm::ParameterSet KVFTest::theConfig

private

Definition at line 51 of file KVFTest.h.

edm::EDGetTokenT<reco::TrackToTrackingParticleAssociator> KVFTest::token_associatorForParamAtPca

private

Definition at line 60 of file KVFTest.h.

Referenced by analyze(), and KVFTest().

edm::EDGetTokenT<reco::TrackCollection> KVFTest::token_tracks

private

Definition at line 57 of file KVFTest.h.

Referenced by analyze(), and KVFTest().

edm::EDGetTokenT<TrackingParticleCollection> KVFTest::token_TrackTruth

private

Definition at line 58 of file KVFTest.h.

Referenced by analyze(), and KVFTest().

edm::EDGetTokenT<TrackingVertexCollection> KVFTest::token_VertexTruth

private

Definition at line 59 of file KVFTest.h.

Referenced by getSimVertex(), and KVFTest().

std::unique_ptr<SimpleVertexTree> KVFTest::tree

private

Definition at line 53 of file KVFTest.h.

Referenced by svgfig.SVG::__str__(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::beginLoop(), MetTreeProducer.MetTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::fillTree(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::process(), MetTreeProducer.MetTreeProducer::process(), and python.cmstools.EventTree::SetAlias().