HIProtoTrackSelector Class Reference

#include <HIProtoTrackSelector.h>

Public Types
typedef reco::TrackCollection	collection
typedef container::const_iterator	const_iterator
typedef std::vector< const reco::Track * >	container

Public Member Functions
const_iterator	begin () const
const_iterator	end () const
	HIProtoTrackSelector (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
void	select (edm::Handle< reco::TrackCollection > &TCH, const edm::Event &iEvent, const edm::EventSetup &iSetup)
size_t	size () const

Private Attributes
edm::InputTag	beamSpot_
edm::EDGetTokenT< reco::BeamSpot >	beamSpotToken_
double	maxD0Significance_
double	minZCut_
double	nSigmaZ_
double	ptMin_
container	selected_
edm::InputTag	vertexCollection_
edm::EDGetTokenT< reco::VertexCollection >	vertexCollectionToken_

Detailed Description

Selector to select prototracks that pass certain kinematic cuts based on fast vertex

Definition at line 26 of file HIProtoTrackSelector.h.

Member Typedef Documentation

collection

typedef reco::TrackCollection HIProtoTrackSelector::collection

Definition at line 29 of file HIProtoTrackSelector.h.

const_iterator

typedef container::const_iterator HIProtoTrackSelector::const_iterator

Definition at line 35 of file HIProtoTrackSelector.h.

container

typedef std::vector<const reco::Track*> HIProtoTrackSelector::container

Definition at line 32 of file HIProtoTrackSelector.h.

Constructor & Destructor Documentation

HIProtoTrackSelector()

HIProtoTrackSelector::HIProtoTrackSelector ( const edm::ParameterSet &  iConfig, edm::ConsumesCollector &&  iC  )

inline

Definition at line 38 of file HIProtoTrackSelector.h.

      : vertexCollection_(iConfig.getParameter<edm::InputTag>("VertexCollection")),
        vertexCollectionToken_(iC.consumes<reco::VertexCollection>(vertexCollection_)),
        beamSpot_(iConfig.getParameter<edm::InputTag>("beamSpotLabel")),
        beamSpotToken_(iC.consumes<reco::BeamSpot>(beamSpot_)),
        ptMin_(iConfig.getParameter<double>("ptMin")),
        nSigmaZ_(iConfig.getParameter<double>("nSigmaZ")),
        minZCut_(iConfig.getParameter<double>("minZCut")),
        maxD0Significance_(iConfig.getParameter<double>("maxD0Significance")){};

Member Function Documentation

begin()

const_iterator HIProtoTrackSelector::begin ( void  ) const

inline

Definition at line 119 of file HIProtoTrackSelector.h.

References selected_.

{ return selected_.begin(); }

end()

const_iterator HIProtoTrackSelector::end ( void  ) const

inline

Definition at line 122 of file HIProtoTrackSelector.h.

References selected_.

{ return selected_.end(); }

select()

void HIProtoTrackSelector::select ( edm::Handle< reco::TrackCollection > &  TCH, const edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

inline

Definition at line 49 of file HIProtoTrackSelector.h.

References pwdgSkimBPark_cfi::beamSpot, beamSpot_, beamSpotToken_, HltBtagPostValidation_cff::c, d0, iEvent, edm::HandleBase::isValid(), SiStripPI::max, maxD0Significance_, minZCut_, nSigmaZ_, edm::Handle< T >::product(), ptMin_, selected_, mathSSE::sqrt(), vertexCollectionToken_, and AlignmentTracksFromVertexSelector_cfi::vertices.

                                                                                                        {
    selected_.clear();

    const collection& c = *(TCH.product());

    // Get fast reco vertex
    edm::Handle<reco::VertexCollection> vc;
    iEvent.getByToken(vertexCollectionToken_, vc);
    const reco::VertexCollection* vertices = vc.product();

    math::XYZPoint vtxPoint(0.0, 0.0, 0.0);
    double vzErr = 0.0;

    if (!vertices->empty()) {
      vtxPoint = vertices->begin()->position();
      vzErr = vertices->begin()->zError();
      edm::LogInfo("HeavyIonVertexing") << "Select prototracks compatible with median vertex"
                                        << "\n   vz = " << vtxPoint.Z() << "\n   " << nSigmaZ_
                                        << " vz sigmas = " << vzErr * nSigmaZ_
                                        << "\n   cut at = " << std::max(vzErr * nSigmaZ_, minZCut_);
    }
    // Supress this warning, since events w/ no vertex are expected
    //else {
    //edm::LogError("HeavyIonVertexing") << "No vertex found in collection '" << vertexCollection_ << "'";
    //}

    // Get beamspot
    reco::BeamSpot beamSpot;
    edm::Handle<reco::BeamSpot> beamSpotHandle;
    iEvent.getByToken(beamSpotToken_, beamSpotHandle);

    math::XYZPoint bsPoint(0.0, 0.0, 0.0);
    double bsWidth = 0.0;

    if (beamSpotHandle.isValid()) {
      beamSpot = *beamSpotHandle;
      bsPoint = beamSpot.position();
      bsWidth = sqrt(beamSpot.BeamWidthX() * beamSpot.BeamWidthY());
      edm::LogInfo("HeavyIonVertexing") << "Select prototracks compatible with beamspot"
                                        << "\n   (x,y,z) = (" << bsPoint.X() << "," << bsPoint.Y() << "," << bsPoint.Z()
                                        << ")"
                                        << "\n   width = " << bsWidth
                                        << "\n   cut at d0/d0sigma = " << maxD0Significance_;
    } else {
      edm::LogError("HeavyIonVertexing") << "No beam spot available from '" << beamSpot_ << "\n";
    }

    // Do selection
    int nSelected = 0;
    int nRejected = 0;
    double d0 = 0.0;
    double d0sigma = 0.0;
    for (reco::TrackCollection::const_iterator trk = c.begin(); trk != c.end(); ++trk) {
      d0 = -1. * trk->dxy(bsPoint);
      d0sigma = sqrt(trk->d0Error() * trk->d0Error() + bsWidth * bsWidth);
      if (trk->pt() > ptMin_                          // keep only tracks above ptMin
          && fabs(d0 / d0sigma) < maxD0Significance_  // keep only tracks with D0 significance less than cut
          && fabs(trk->dz(vtxPoint)) < std::max(vzErr * nSigmaZ_, minZCut_)  // within minZCut, nSigmaZ of fast vertex
      ) {
        nSelected++;
        selected_.push_back(&*trk);
      } else
        nRejected++;
    }

    edm::LogInfo("HeavyIonVertexing") << "selected " << nSelected << " prototracks out of " << nRejected + nSelected
                                      << "\n";
  }

size()

size_t HIProtoTrackSelector::size ( void  ) const

inline

Definition at line 125 of file HIProtoTrackSelector.h.

References selected_.

Referenced by ntupleDataFormat._Collection::__iter__(), and ntupleDataFormat._Collection::__len__().

{ return selected_.size(); }

Member Data Documentation

beamSpot_

edm::InputTag HIProtoTrackSelector::beamSpot_

private

Definition at line 131 of file HIProtoTrackSelector.h.

Referenced by select().

beamSpotToken_

edm::EDGetTokenT<reco::BeamSpot> HIProtoTrackSelector::beamSpotToken_

private

Definition at line 132 of file HIProtoTrackSelector.h.

Referenced by select().

maxD0Significance_

double HIProtoTrackSelector::maxD0Significance_

private

Definition at line 136 of file HIProtoTrackSelector.h.

Referenced by select().

minZCut_

double HIProtoTrackSelector::minZCut_

private

Definition at line 135 of file HIProtoTrackSelector.h.

Referenced by select().

nSigmaZ_

double HIProtoTrackSelector::nSigmaZ_

private

Definition at line 134 of file HIProtoTrackSelector.h.

Referenced by select().

ptMin_

double HIProtoTrackSelector::ptMin_

private

Definition at line 133 of file HIProtoTrackSelector.h.

Referenced by select().

selected_

container HIProtoTrackSelector::selected_

private

Definition at line 128 of file HIProtoTrackSelector.h.

Referenced by begin(), end(), select(), and size().

vertexCollection_

edm::InputTag HIProtoTrackSelector::vertexCollection_

private

Definition at line 129 of file HIProtoTrackSelector.h.

vertexCollectionToken_

edm::EDGetTokenT<reco::VertexCollection> HIProtoTrackSelector::vertexCollectionToken_

private

Definition at line 130 of file HIProtoTrackSelector.h.

Referenced by select().