DuplicateTrackMerger Class Reference

Inheritance diagram for DuplicateTrackMerger:

Public Types
typedef std::vector< std::pair < TrackCandidate, std::pair < reco::TrackRef, reco::TrackRef > > >	CandidateToDuplicate
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T...>	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T...>	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT < LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache
Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Public Member Functions
	DuplicateTrackMerger (const edm::ParameterSet &iPara)
	constructor More...
virtual	~DuplicateTrackMerger ()
	destructor More...
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducerBase ()
Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	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< ConsumesInfo >	consumesInfo () 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 ()

Protected Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override
	produce one event More...
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)

Private Attributes
std::string	dbFileName_
	MVA weights file. More...
const GBRForest *	forest_
	MVA discriminator. More...
std::string	forestLabel_
float *	gbrVals_
edm::ESHandle< MagneticField >	magfield_
double	maxDCA_
	max distance between two tracks at closest approach More...
double	maxDdsz_
	max difference in longitudinal impact parameter between two tracks More...
double	maxDdxy_
	max difference in transverse impact parameter between two tracks More...
double	maxDLambda_
	max difference in Lambda between two tracks More...
double	maxDPhi_
	max difference in phi between two tracks More...
double	maxDQoP_
	max difference in q/p between two tracks More...
TrackMerger	merger_
	Merger. More...
double	minBDTG_
	minBDTG cut value More...
double	minDeltaR3d_
	minDeltaR3d cut value More...
double	minP_
	min p cut value More...
double	minpT_
	min pT cut value More...
float	tmva_d3dr_
float	tmva_d3dz_
float	tmva_ddsz_
	MVA input variables. More...
float	tmva_ddxy_
float	tmva_dlambda_
float	tmva_dphi_
float	tmva_dqoverp_
float	tmva_inner_nMissingOuter_
float	tmva_outer_nMissingInner_
edm::EDGetTokenT < reco::TrackCollection >	trackSource_
	track input collection More...
bool	useForestFromDB_

Additional Inherited Members
Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

Detailed Description

selects pairs of tracks that should be single tracks

Author

Matthew Walker

Definition at line 33 of file DuplicateTrackMerger.cc.

Member Typedef Documentation

typedef std::vector<std::pair<TrackCandidate,std::pair<reco::TrackRef,reco::TrackRef> > > DuplicateTrackMerger::CandidateToDuplicate

Definition at line 40 of file DuplicateTrackMerger.cc.

Constructor & Destructor Documentation

DuplicateTrackMerger::DuplicateTrackMerger ( const edm::ParameterSet &  iPara )

explicit

constructor

Definition at line 107 of file DuplicateTrackMerger.cc.

References dbFileName_, edm::ParameterSet::exists(), forestLabel_, gbrVals_, edm::ParameterSet::getParameter(), maxDCA_, maxDdsz_, maxDdxy_, maxDLambda_, maxDPhi_, maxDQoP_, minBDTG_, minDeltaR3d_, minP_, minpT_, AlCaHLTBitMon_QueryRunRegistry::string, trackSource_, and useForestFromDB_.

                                                                       : forest_(nullptr), gbrVals_(nullptr), merger_(iPara)
{
  forestLabel_ = "MVADuplicate";
  useForestFromDB_ = true;

  minDeltaR3d_ = -4.0;
  minBDTG_ = -0.96;
  minpT_ = 0.2;
  minP_ = 0.4;
  maxDCA_ = 30.0;
  maxDPhi_ = 0.30;
  maxDLambda_ = 0.30;
  maxDdsz_ = 10.0;
  maxDdxy_ = 10.0;
  maxDQoP_ = 0.25;
  if(iPara.exists("minpT"))minpT_ = iPara.getParameter<double>("minpT");
  if(iPara.exists("minP"))minP_ = iPara.getParameter<double>("minP");
  if(iPara.exists("maxDCA"))maxDCA_ = iPara.getParameter<double>("maxDCA");
  if(iPara.exists("maxDPhi"))maxDPhi_ = iPara.getParameter<double>("maxDPhi");
  if(iPara.exists("maxDLambda"))maxDLambda_ = iPara.getParameter<double>("maxDLambda");
  if(iPara.exists("maxDdsz"))maxDdsz_ = iPara.getParameter<double>("maxDdsz");
  if(iPara.exists("maxDdxy"))maxDdxy_ = iPara.getParameter<double>("maxDdxy");
  if(iPara.exists("maxDQoP"))maxDQoP_ = iPara.getParameter<double>("maxDQoP");
  if(iPara.exists("source"))trackSource_ = consumes<reco::TrackCollection>(iPara.getParameter<edm::InputTag>("source"));
  if(iPara.exists("minDeltaR3d"))minDeltaR3d_ = iPara.getParameter<double>("minDeltaR3d");
  if(iPara.exists("minBDTG"))minBDTG_ = iPara.getParameter<double>("minBDTG");

  produces<std::vector<TrackCandidate> >("candidates");
  produces<CandidateToDuplicate>("candidateMap");

  gbrVals_ = new float[9];
  dbFileName_ = "";
  if(iPara.exists("GBRForestLabel"))forestLabel_ = iPara.getParameter<std::string>("GBRForestLabel");
  if(iPara.exists("GBRForestFileName")){
    dbFileName_ = iPara.getParameter<std::string>("GBRForestFileName");
    useForestFromDB_ = false;
  }

  /*
  tmvaReader_ = new TMVA::Reader("!Color:Silent");
  tmvaReader_->AddVariable("ddsz",&tmva_ddsz_);
  tmvaReader_->AddVariable("ddxy",&tmva_ddxy_);
  tmvaReader_->AddVariable("dphi",&tmva_dphi_);
  tmvaReader_->AddVariable("dlambda",&tmva_dlambda_);
  tmvaReader_->AddVariable("dqoverp",&tmva_dqoverp_);
  tmvaReader_->AddVariable("delta3d_r",&tmva_d3dr_);
  tmvaReader_->AddVariable("delta3d_z",&tmva_d3dz_);
  tmvaReader_->AddVariable("outer_nMissingInner",&tmva_outer_nMissingInner_);
  tmvaReader_->AddVariable("inner_nMissingOuter",&tmva_inner_nMissingOuter_);
  tmvaReader_->BookMVA("BDTG",mvaFilePath);
  */

}

DuplicateTrackMerger::~DuplicateTrackMerger ( )

virtual

destructor

Definition at line 161 of file DuplicateTrackMerger.cc.

References forest_, gbrVals_, and useForestFromDB_.

{

  if(gbrVals_)delete [] gbrVals_;
  if(!useForestFromDB_ && forest_) delete forest_;
  /* no op */

}

Member Function Documentation

void DuplicateTrackMerger::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprotectedvirtual

produce one event

Implements edm::stream::EDProducerBase.

Definition at line 170 of file DuplicateTrackMerger.cc.

References reco::TrackBase::charge(), CommonMethods::cp(), dbFileName_, forest_, forestLabel_, gbrVals_, edm::EventSetup::get(), edm::Event::getByToken(), GBRForest::GetClassifier(), patZpeak::handle, reco::TrackBase::hitPattern(), i, TrackMerger::init(), trajectoryStateTransform::innerFreeState(), reco::Track::innerMomentum(), reco::Track::innerPosition(), TrajectoryStateClosestToPoint::isValid(), j, M_PI, PV3DBase< T, PVType, FrameType >::mag(), mag(), magfield_, maxDCA_, maxDdsz_, maxDdxy_, maxDLambda_, maxDPhi_, maxDQoP_, TrackMerger::merge(), merger_, minBDTG_, minDeltaR3d_, minP_, minpT_, reco::HitPattern::MISSING_INNER_HITS, reco::HitPattern::MISSING_OUTER_HITS, FreeTrajectoryState::momentum(), TrajectoryStateClosestToPoint::momentum(), reco::HitPattern::numberOfLostHits(), trajectoryStateTransform::outerFreeState(), reco::Track::outerPosition(), PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::phi(), FreeTrajectoryState::position(), funct::pow(), edm::ESHandle< class >::product(), TrajectoryStateClosestToPoint::pt(), edm::Event::put(), FreeTrajectoryState::signedInverseMomentum(), mathSSE::sqrt(), TrajectoryStateClosestToPoint::theState(), PV3DBase< T, PVType, FrameType >::theta(), tmva_d3dr_, tmva_d3dz_, tmva_ddsz_, tmva_ddxy_, tmva_dlambda_, tmva_dphi_, tmva_dqoverp_, tmva_inner_nMissingOuter_, tmva_outer_nMissingInner_, trackSource_, useForestFromDB_, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

{

  merger_.init(iSetup);

  if(!forest_){
    if(useForestFromDB_){
      edm::ESHandle<GBRForest> forestHandle;
      iSetup.get<GBRWrapperRcd>().get(forestLabel_,forestHandle);
      forest_ = forestHandle.product();
    }else{
      TFile gbrfile(dbFileName_.c_str());
      forest_ = dynamic_cast<const GBRForest*>(gbrfile.Get(forestLabel_.c_str()));
    }
  }

  //edm::Handle<edm::View<reco::Track> >handle;
  edm::Handle<reco::TrackCollection >handle;
  iEvent.getByToken(trackSource_,handle);
  reco::TrackRefProd refTrks(handle);

  iSetup.get<IdealMagneticFieldRecord>().get(magfield_);
  TwoTrackMinimumDistance ttmd;
  TSCPBuilderNoMaterial tscpBuilder;
  std::auto_ptr<std::vector<TrackCandidate> > out_duplicateCandidates(new std::vector<TrackCandidate>());

  std::auto_ptr<CandidateToDuplicate> out_candidateMap(new CandidateToDuplicate());

  for(int i = 0; i < (int)handle->size(); i++){
    const reco::Track *rt1 = &(handle->at(i));
    if(rt1->innerMomentum().Rho() < minpT_)continue;
    if(rt1->innerMomentum().R() < minP_)continue;
    for(int j = i+1; j < (int)handle->size();j++){
      const reco::Track *rt2 = &(handle->at(j));
      if(rt2->innerMomentum().Rho() < minpT_)continue;
      if(rt2->innerMomentum().R() < minP_)continue;
      if(rt1->charge() != rt2->charge())continue;
      const reco::Track* t1,*t2;
      if(rt1->outerPosition().Rho() < rt2->outerPosition().Rho()){
    t1 = rt1;
    t2 = rt2;
      }else{
    t1 = rt2;
    t2 = rt1;
      }
      double deltaR3d = sqrt(pow(t1->outerPosition().x()-t2->innerPosition().x(),2) + pow(t1->outerPosition().y()-t2->innerPosition().y(),2) + pow(t1->outerPosition().z()-t2->innerPosition().z(),2));

      if(t1->outerPosition().Rho() > t2->innerPosition().Rho())deltaR3d *= -1.0;
      if(deltaR3d < minDeltaR3d_)continue;
      
      FreeTrajectoryState fts1 = trajectoryStateTransform::outerFreeState(*t1, &*magfield_,false);
      FreeTrajectoryState fts2 = trajectoryStateTransform::innerFreeState(*t2, &*magfield_,false);
      GlobalPoint avgPoint((t1->outerPosition().x()+t2->innerPosition().x())*0.5,(t1->outerPosition().y()+t2->innerPosition().y())*0.5,(t1->outerPosition().z()+t2->innerPosition().z())*0.5);
      TrajectoryStateClosestToPoint TSCP1 = tscpBuilder(fts1, avgPoint);
      TrajectoryStateClosestToPoint TSCP2 = tscpBuilder(fts2, avgPoint);
      if(!TSCP1.isValid())continue;
      if(!TSCP2.isValid())continue;

      const FreeTrajectoryState ftsn1 = TSCP1.theState();
      const FreeTrajectoryState ftsn2 = TSCP2.theState();
 
      if ( (ftsn2.position()-ftsn1.position()).mag() > maxDCA_ ) continue;

      double qoverp1 = ftsn1.signedInverseMomentum();
      double qoverp2 = ftsn2.signedInverseMomentum();
      tmva_dqoverp_ = qoverp1-qoverp2;
      if ( fabs(tmva_dqoverp_) > maxDQoP_ ) continue;

      double lambda1 =  M_PI/2 - ftsn1.momentum().theta();
      double lambda2 =  M_PI/2 - ftsn2.momentum().theta();
      tmva_dlambda_ = lambda1-lambda2;
      if ( fabs(tmva_dlambda_) > maxDLambda_ ) continue;

      double phi1 = ftsn1.momentum().phi();
      double phi2 = ftsn2.momentum().phi();
      tmva_dphi_ = phi1-phi2;
      if(fabs(tmva_dphi_) > M_PI) tmva_dphi_ = 2*M_PI - fabs(tmva_dphi_);
      if ( fabs(tmva_dphi_) > maxDPhi_ ) continue;

      double dxy1 = (-ftsn1.position().x() * ftsn1.momentum().y() + ftsn1.position().y() * ftsn1.momentum().x())/TSCP1.pt();
      double dxy2 = (-ftsn2.position().x() * ftsn2.momentum().y() + ftsn2.position().y() * ftsn2.momentum().x())/TSCP2.pt();
      tmva_ddxy_ = dxy1-dxy2;
      if ( fabs(tmva_ddxy_) > maxDdxy_ ) continue;

      double dsz1 = ftsn1.position().z() * TSCP1.pt() / TSCP1.momentum().mag() - (ftsn1.position().x() * ftsn1.momentum().y() + ftsn1.position().y() * ftsn1.momentum().x())/TSCP1.pt() * ftsn1.momentum().z()/ftsn1.momentum().mag();
      double dsz2 = ftsn2.position().z() * TSCP2.pt() / TSCP2.momentum().mag() - (ftsn2.position().x() * ftsn2.momentum().y() + ftsn2.position().y() * ftsn2.momentum().x())/TSCP2.pt() * ftsn2.momentum().z()/ftsn2.momentum().mag();
      tmva_ddsz_ = dsz1-dsz2;
      if ( fabs(tmva_ddsz_) > maxDdsz_ ) continue;

      tmva_d3dr_ = avgPoint.perp();
      tmva_d3dz_ = avgPoint.z();
      tmva_outer_nMissingInner_ = t2->hitPattern().numberOfLostHits(reco::HitPattern::MISSING_INNER_HITS);
      tmva_inner_nMissingOuter_ = t1->hitPattern().numberOfLostHits(reco::HitPattern::MISSING_OUTER_HITS);
      

      gbrVals_[0] = tmva_ddsz_;
      gbrVals_[1] = tmva_ddxy_;
      gbrVals_[2] = tmva_dphi_;
      gbrVals_[3] = tmva_dlambda_;
      gbrVals_[4] = tmva_dqoverp_;
      gbrVals_[5] = tmva_d3dr_;
      gbrVals_[6] = tmva_d3dz_;
      gbrVals_[7] = tmva_outer_nMissingInner_;
      gbrVals_[8] = tmva_inner_nMissingOuter_;


      double mvaBDTG = forest_->GetClassifier(gbrVals_);
      if(mvaBDTG < minBDTG_)continue;
      
      
      TrackCandidate mergedTrack = merger_.merge(*t1,*t2);
      out_duplicateCandidates->push_back(mergedTrack);
      std::pair<TrackRef,TrackRef> trackPair(TrackRef(refTrks,i),TrackRef(refTrks,j));
      std::pair<TrackCandidate, std::pair<TrackRef,TrackRef> > cp(mergedTrack,trackPair);
      out_candidateMap->push_back(cp);
    }
  }
  iEvent.put(out_duplicateCandidates,"candidates");
  iEvent.put(out_candidateMap,"candidateMap");

}

Member Data Documentation

std::string DuplicateTrackMerger::dbFileName_

private

MVA weights file.

Definition at line 66 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

const GBRForest* DuplicateTrackMerger::forest_

private

MVA discriminator.

Definition at line 48 of file DuplicateTrackMerger.cc.

Referenced by produce(), and ~DuplicateTrackMerger().

std::string DuplicateTrackMerger::forestLabel_

private

Definition at line 68 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

float* DuplicateTrackMerger::gbrVals_

private

Definition at line 61 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), produce(), and ~DuplicateTrackMerger().

edm::ESHandle<MagneticField> DuplicateTrackMerger::magfield_

private

Definition at line 90 of file DuplicateTrackMerger.cc.

Referenced by produce().

double DuplicateTrackMerger::maxDCA_

private

max distance between two tracks at closest approach

Definition at line 78 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::maxDdsz_

private

max difference in longitudinal impact parameter between two tracks

Definition at line 86 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::maxDdxy_

private

max difference in transverse impact parameter between two tracks

Definition at line 84 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::maxDLambda_

private

max difference in Lambda between two tracks

Definition at line 82 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::maxDPhi_

private

max difference in phi between two tracks

Definition at line 80 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::maxDQoP_

private

max difference in q/p between two tracks

Definition at line 88 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

TrackMerger DuplicateTrackMerger::merger_

private

Merger.

Definition at line 93 of file DuplicateTrackMerger.cc.

Referenced by produce().

double DuplicateTrackMerger::minBDTG_

private

minBDTG cut value

Definition at line 72 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::minDeltaR3d_

private

minDeltaR3d cut value

Definition at line 70 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::minP_

private

min p cut value

Definition at line 76 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

double DuplicateTrackMerger::minpT_

private

min pT cut value

Definition at line 74 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

float DuplicateTrackMerger::tmva_d3dr_

private

Definition at line 56 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_d3dz_

private

Definition at line 57 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_ddsz_

private

MVA input variables.

Definition at line 51 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_ddxy_

private

Definition at line 52 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_dlambda_

private

Definition at line 54 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_dphi_

private

Definition at line 53 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_dqoverp_

private

Definition at line 55 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_inner_nMissingOuter_

private

Definition at line 59 of file DuplicateTrackMerger.cc.

Referenced by produce().

float DuplicateTrackMerger::tmva_outer_nMissingInner_

private

Definition at line 58 of file DuplicateTrackMerger.cc.

Referenced by produce().

edm::EDGetTokenT<reco::TrackCollection> DuplicateTrackMerger::trackSource_

private

track input collection

Definition at line 64 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), and produce().

bool DuplicateTrackMerger::useForestFromDB_

private

Definition at line 67 of file DuplicateTrackMerger.cc.

Referenced by DuplicateTrackMerger(), produce(), and ~DuplicateTrackMerger().