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 24 of file HIProtoTrackSelector.h.

Member Typedef Documentation

typedef reco::TrackCollection HIProtoTrackSelector::collection

Definition at line 29 of file HIProtoTrackSelector.h.

typedef container::const_iterator HIProtoTrackSelector::const_iterator

Definition at line 35 of file HIProtoTrackSelector.h.

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

Definition at line 32 of file HIProtoTrackSelector.h.

Constructor & Destructor Documentation

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

const_iterator HIProtoTrackSelector::begin ( void  ) const

inline

Definition at line 125 of file HIProtoTrackSelector.h.

References selected_.

{ return selected_.begin(); }

const_iterator HIProtoTrackSelector::end ( void  ) const

inline

Definition at line 128 of file HIProtoTrackSelector.h.

References selected_.

{ return selected_.end(); }

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

inline

Definition at line 50 of file HIProtoTrackSelector.h.

References ecalDrivenElectronSeedsParameters_cff::beamSpot, beamSpot_, beamSpotToken_, reco::BeamSpot::BeamWidthX(), reco::BeamSpot::BeamWidthY(), EnergyCorrector::c, allConversions_cfi::d0, edm::Event::getByToken(), edm::HandleBase::isValid(), hpstanc_transforms::max, maxD0Significance_, minZCut_, nSigmaZ_, reco::BeamSpot::position(), edm::Handle< T >::product(), ptMin_, selected_, mathSSE::sqrt(), vertexCollectionToken_, and primaryVertexAssociation_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->size()>0) {
    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_t HIProtoTrackSelector::size ( void  ) const

inline

Definition at line 131 of file HIProtoTrackSelector.h.

References selected_.

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

{ return selected_.size(); }

Member Data Documentation

edm::InputTag HIProtoTrackSelector::beamSpot_

private

Definition at line 138 of file HIProtoTrackSelector.h.

Referenced by select().

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

private

Definition at line 139 of file HIProtoTrackSelector.h.

Referenced by select().

double HIProtoTrackSelector::maxD0Significance_

private

Definition at line 143 of file HIProtoTrackSelector.h.

Referenced by select().

double HIProtoTrackSelector::minZCut_

private

Definition at line 142 of file HIProtoTrackSelector.h.

Referenced by select().

double HIProtoTrackSelector::nSigmaZ_

private

Definition at line 141 of file HIProtoTrackSelector.h.

Referenced by select().

double HIProtoTrackSelector::ptMin_

private

Definition at line 140 of file HIProtoTrackSelector.h.

Referenced by select().

container HIProtoTrackSelector::selected_

private

Definition at line 135 of file HIProtoTrackSelector.h.

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

edm::InputTag HIProtoTrackSelector::vertexCollection_

private

Definition at line 136 of file HIProtoTrackSelector.h.

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

private

Definition at line 137 of file HIProtoTrackSelector.h.

Referenced by select().