Inheritance diagram for BtoCharmDecayVertexMerger:

Classes
struct	vertexProxy

Public Member Functions
	BtoCharmDecayVertexMerger (const edm::ParameterSet &params)

virtual void	produce (edm::Event &event, const edm::EventSetup &es)

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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
	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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
GlobalVector	flightDirection (reco::Vertex &pv, reco::Vertex &sv)

void	resolveBtoDchain (std::vector< vertexProxy > &coll, unsigned int i, unsigned int k)

Private Attributes
double	maxPtreltomerge

double	minCosPAtomerge

double	minDRForUnique

reco::Vertex	pv

edm::EDGetTokenT < reco::VertexCollection >	token_primaryVertex

edm::EDGetTokenT < reco::VertexCollection >	token_secondaryVertex

double	vecSumIMCUTForUnique

Friends
bool	operator< (vertexProxy v1, vertexProxy v2)

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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

Definition at line 23 of file BtoCharmDecayVertexMerger.cc.

Constructor & Destructor Documentation

BtoCharmDecayVertexMerger::BtoCharmDecayVertexMerger ( const edm::ParameterSet & params )

Definition at line 58 of file BtoCharmDecayVertexMerger.cc.

                                                                                   :
        token_primaryVertex(consumes<reco::VertexCollection>(params.getParameter<edm::InputTag>("primaryVertices"))),
        token_secondaryVertex(consumes<reco::VertexCollection>(params.getParameter<edm::InputTag>("secondaryVertices"))),
        minDRForUnique(params.getUntrackedParameter<double>("minDRUnique",0.4)),
        vecSumIMCUTForUnique(params.getUntrackedParameter<double>("minvecSumIMifsmallDRUnique",5.5)),
        minCosPAtomerge(params.getUntrackedParameter<double>("minCosPAtomerge",0.99)),
        maxPtreltomerge(params.getUntrackedParameter<double>("maxPtreltomerge",7777.0))
        // maxFraction(params.getParameter<double>("maxFraction")),
        // minSignificance(params.getParameter<double>("minSignificance"))
 {
        produces<reco::VertexCollection>();
 }

Member Function Documentation

GlobalVector BtoCharmDecayVertexMerger::flightDirection	(	reco::Vertex &	pv,
		reco::Vertex &	sv
	)

private

Definition at line 233 of file BtoCharmDecayVertexMerger.cc.

References reco::Vertex::position().

Referenced by resolveBtoDchain().

                                                                         {
   GlobalVector res(sv.position().X() - pv.position().X(),
                     sv.position().Y() - pv.position().Y(),
                     sv.position().Z() - pv.position().Z());
   return res;
 }

void BtoCharmDecayVertexMerger::produce	(	edm::Event &	event,
		const edm::EventSetup &	es
	)

virtual

Implements edm::EDProducer.

Definition at line 71 of file BtoCharmDecayVertexMerger.cc.

References edm::Event::getByToken(), edm::Event::put(), pv, dt_dqm_sourceclient_common_cff::reco, resolveBtoDchain(), python.multivaluedict::sort(), token_primaryVertex, and token_secondaryVertex.

                                                                                     {
 
   using namespace reco;
   // PV
   edm::Handle<reco::VertexCollection> PVcoll;
   iEvent.getByToken(token_primaryVertex, PVcoll);
 
   if(PVcoll->size()!=0) {
 
   const reco::VertexCollection pvc = *( PVcoll.product());
   pv = pvc[0];
 
   // get the IVF collection
   edm::Handle<reco::VertexCollection> secondaryVertices;
   iEvent.getByToken(token_secondaryVertex, secondaryVertices);
 
 
 
   //loop over vertices,  fill into own collection for sorting
   std::vector<vertexProxy> vertexProxyColl;
   for(std::vector<reco::Vertex>::const_iterator sv = secondaryVertices->begin();
       sv != secondaryVertices->end(); ++sv) {
     vertexProxy vtx = {*sv,(*sv).p4().M(),(*sv).tracksSize()};
     vertexProxyColl.push_back( vtx );
   }
 
   // sort the vertices by mass and track multiplicity
   sort( vertexProxyColl.begin(), vertexProxyColl.end());
 
 
   // loop forward over all vertices
   for(unsigned int iVtx=0; iVtx < vertexProxyColl.size(); iVtx++){
 
     // nested loop backwards (in order to start from light masses)
     // check all vertices against each other for B->D chain
     for(unsigned int kVtx=vertexProxyColl.size()-1; kVtx>iVtx; kVtx--){
       // remove D vertices from the collection and add the tracks to the original one
       resolveBtoDchain(vertexProxyColl, iVtx, kVtx);
     }
   }
 
   // now create new vertex collection and add to event
   VertexCollection *bvertices = new VertexCollection();
   for(std::vector<vertexProxy>::iterator it=vertexProxyColl.begin(); it!=vertexProxyColl.end(); it++) bvertices->push_back((*it).vert);
   std::auto_ptr<VertexCollection> bvertColl(bvertices);
   iEvent.put(bvertColl);
   }
   else{
     std::auto_ptr<VertexCollection> bvertCollEmpty(new VertexCollection);
     iEvent.put(bvertCollEmpty);
   }
 }

void BtoCharmDecayVertexMerger::resolveBtoDchain	(	std::vector< vertexProxy > &	coll,
		unsigned int	i,
		unsigned int	k
	)

private

Definition at line 126 of file BtoCharmDecayVertexMerger.cc.

References Geom::deltaR(), reco::SecondaryVertex::dist3d(), dot(), spr::find(), flightDirection(), i, gen::k, maxPtreltomerge, minCosPAtomerge, minDRForUnique, pv, dt_dqm_sourceclient_common_cff::reco, reco::Vertex::refittedTrack(), mathSSE::sqrt(), lumiQTWidget::t, groupFilesInBlocks::temp, tracks_, reco::Vertex::tracks_begin(), reco::Vertex::tracks_end(), reco::Vertex::trackWeight(), Measurement1D::value(), vecSumIMCUTForUnique, and histoStyle::weight.

Referenced by produce().

                                                                                                             {
   using namespace reco;
 
   GlobalVector momentum1 = GlobalVector(coll[i].vert.p4().X(), coll[i].vert.p4().Y(), coll[i].vert.p4().Z());
   GlobalVector momentum2 = GlobalVector(coll[k].vert.p4().X(), coll[k].vert.p4().Y(), coll[k].vert.p4().Z());
 
   reco::SecondaryVertex sv1(pv, coll[i].vert, momentum1 , true);
   reco::SecondaryVertex sv2(pv, coll[k].vert, momentum2 , true);
 
 
   // find out which one is near and far
   reco::SecondaryVertex svNear = sv1;
   reco::SecondaryVertex svFar  = sv2;
   GlobalVector momentumNear  = momentum1;
   GlobalVector momentumFar   = momentum2;
 
   // swap if it is the other way around
   if(sv1.dist3d().value() >= sv2.dist3d().value()){
     svNear = sv2;
     svFar = sv1;
     momentumNear = momentum2;
     momentumFar = momentum1;
   }
   GlobalVector nearToFar = flightDirection( svNear, svFar);
   GlobalVector pvToNear  = flightDirection( pv, svNear);
 
   double cosPA =  nearToFar.dot(momentumFar) / momentumFar.mag()/ nearToFar.mag();
   double cosa  =  pvToNear. dot(momentumFar) / pvToNear.mag()   / momentumFar.mag();
   double ptrel = sqrt(1.0 - cosa*cosa)* momentumFar.mag();
 
   double vertexDeltaR = Geom::deltaR(flightDirection(pv, sv1), flightDirection(pv, sv2) );
 
   // create a set of all tracks from both vertices, avoid double counting by using a std::set<>
   std::set<reco::TrackRef> trackrefs;
   // first vertex
   for(reco::Vertex::trackRef_iterator ti = sv1.tracks_begin(); ti!=sv1.tracks_end(); ti++){
     if(sv1.trackWeight(*ti)>0.5){
       reco::TrackRef t = ti->castTo<reco::TrackRef>();
       trackrefs.insert(t);
     }
   }
   // second vertex
   for(reco::Vertex::trackRef_iterator ti = sv2.tracks_begin(); ti!=sv2.tracks_end(); ti++){
     if(sv2.trackWeight(*ti)>0.5){
       reco::TrackRef t = ti->castTo<reco::TrackRef>();
       trackrefs.insert(t);
     }
   }
 
   // now calculate one LorentzVector from the track momenta
   ROOT::Math::LorentzVector<ROOT::Math::PxPyPzM4D<double> > mother;
   for(std::set<reco::TrackRef>::const_iterator it = trackrefs.begin(); it!= trackrefs.end(); it++){
     ROOT::Math::LorentzVector<ROOT::Math::PxPyPzM4D<double> >  temp ( (*it)->px(),(*it)->py(),(*it)->pz(), 0.13957 );
     mother += temp;
   }
 
 
   //   check if criteria for merging are fulfilled
   if(vertexDeltaR<minDRForUnique && mother.M()<vecSumIMCUTForUnique && cosPA>minCosPAtomerge && fabs(ptrel)<maxPtreltomerge ) {
 
 
     // add tracks of second vertex which are missing to first vertex
     // loop over the second
     bool bFoundDuplicate=false;
     for(reco::Vertex::trackRef_iterator ti = sv2.tracks_begin(); ti!=sv2.tracks_end(); ti++){
       reco::Vertex::trackRef_iterator it = find(sv1.tracks_begin(), sv1.tracks_end(), *ti);
       if (it==sv1.tracks_end()) coll[i].vert.add( *ti, sv2.refittedTrack(*ti), sv2.trackWeight(*ti) );
       else bFoundDuplicate=true;
     }
     // in case a duplicate track is found, need to create the full track list in the vertex from scratch because we need to modify the weight.
     // the weight must be the larger one, otherwise we may have outliers which are not real outliers
 
     if(bFoundDuplicate){
       // create backup track containers from main vertex
       std::vector<TrackBaseRef > tracks_;
       std::vector<Track> refittedTracks_;
       std::vector<float> weights_;
       for(reco::Vertex::trackRef_iterator it = coll[i].vert.tracks_begin(); it!=coll[i].vert.tracks_end(); it++) {
        tracks_.push_back( *it);
        refittedTracks_.push_back( coll[i].vert.refittedTrack(*it));
        weights_.push_back( coll[i].vert.trackWeight(*it) );
       }
       // delete tracks and add all tracks back, and check in which vertex the weight is larger
       coll[i].vert.removeTracks();
       std::vector<Track>::iterator it2 = refittedTracks_.begin();
       std::vector<float>::iterator it3 = weights_.begin();
       for(reco::Vertex::trackRef_iterator it = tracks_.begin(); it!=tracks_.end(); it++, it2++, it3++){
        float weight = *it3;
        float weight2= sv2.trackWeight(*it);
        Track refittedTrackWithLargerWeight = *it2;
        if( weight2 >weight) {
          weight = weight2;
          refittedTrackWithLargerWeight = sv2.refittedTrack(*it);
        }
        coll[i].vert.add(*it , refittedTrackWithLargerWeight  , weight);
       }
     }
 
     // remove the second vertex from the collection
     coll.erase( coll.begin() + k  );
   }
 
 }

Friends And Related Function Documentation

bool operator<	(	vertexProxy	v1,
		vertexProxy	v2
	)

friend

Definition at line 46 of file BtoCharmDecayVertexMerger.cc.

                                                             {
          if(v1.ntracks>2 && v2.ntracks<3) return true;
          if(v1.ntracks<3 && v2.ntracks>2) return false;
          return (v1.invm>v2.invm);
        }