GSFTrackMaker.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/ESHandle.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "DQM/PhysicsHWW/interface/GSFTrackMaker.h"

typedef math::XYZTLorentzVectorF LorentzVector;
typedef math::XYZPoint Point;

GSFTrackMaker::GSFTrackMaker(const edm::ParameterSet& iConfig, edm::ConsumesCollector iCollector) {

  GSFTrack_ = iCollector.consumes<edm::View<reco::GsfTrack> > (iConfig.getParameter<edm::InputTag>("gsftracksInputTag"));

}

void GSFTrackMaker::SetVars(HWW& hww, const edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  using namespace edm;
  using std::vector;
  using reco::Track;
  using reco::TrackBase;

  hww.Load_gsftrks_p4();
  hww.Load_gsftrks_vertex_p4();
  hww.Load_gsftrks_d0();
  hww.Load_gsftrks_z0();
  hww.Load_gsftrks_d0Err();
  hww.Load_gsftrks_z0Err();
  hww.Load_gsftrks_etaErr();
  hww.Load_gsftrks_phiErr();
  hww.Load_gsftrks_d0phiCov();

  bool validToken;

  Handle<edm::View<reco::GsfTrack> > track_h;
  validToken = iEvent.getByToken(GSFTrack_, track_h);
  if(!validToken) return;

  edm::ESHandle<MagneticField> theMagField;
  iSetup.get<IdealMagneticFieldRecord>().get(theMagField);
  
  edm::ESHandle<TrackerGeometry> theG;
  iSetup.get<TrackerDigiGeometryRecord>().get(theG);
  
  edm::View<reco::GsfTrack>::const_iterator tracks_end = track_h->end();
  
  for (edm::View<reco::GsfTrack>::const_iterator i = track_h->begin(); i != tracks_end; ++i) {

    hww.gsftrks_p4()           .push_back( LorentzVector( i->px(), i->py(), i->pz(), i->p() )       );
    hww.gsftrks_vertex_p4()    .push_back( LorentzVector(i->vx(),i->vy(), i->vz(), 0.)              );
    hww.gsftrks_d0()           .push_back( i->d0()                                                  );
    hww.gsftrks_z0()           .push_back( i->dz()                                                  );
    hww.gsftrks_d0Err()        .push_back( i->d0Error()                                             );
    hww.gsftrks_z0Err()        .push_back( i->dzError()                                             );
    hww.gsftrks_etaErr()       .push_back( i->etaError()                                            );
    hww.gsftrks_phiErr()       .push_back( i->phiError()                                            );
    hww.gsftrks_d0phiCov()     .push_back( -i->covariance(TrackBase::i_phi, TrackBase::i_dxy)       );

  }
}