TrackMaker Class Reference

#include <TrackMaker.h>

Public Member Functions
void	SetVars (HWW &, const edm::Event &, const edm::EventSetup &)
	TrackMaker (const edm::ParameterSet &, edm::ConsumesCollector)

Public Attributes
edm::EDGetTokenT< edm::View < reco::Track > >	TrackCollection_

Detailed Description

Definition at line 8 of file TrackMaker.h.

Constructor & Destructor Documentation

TrackMaker::TrackMaker	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector	iCollector
	)

Definition at line 17 of file TrackMaker.cc.

References edm::ConsumesCollector::consumes(), edm::ParameterSet::getParameter(), and TrackCollection_.

                                                                                     {

  TrackCollection_ = iCollector.consumes<edm::View<reco::Track> >(iConfig.getParameter<edm::InputTag>("trackInputTag"));

}

Member Function Documentation

void TrackMaker::SetVars	(	HWW &	hww,
		const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

Definition at line 23 of file TrackMaker.cc.

References alongMomentum, AnalyticalPropagator_cfi::AnalyticalPropagator, Reference_intrackfit_cff::barrel, Cylinder::build(), newFWLiteAna::build, SiStripRecHit1D::cluster(), SiStripRecHit2D::cluster(), edm::EventSetup::get(), edm::Event::getByToken(), reco::HitPattern::getHitPattern(), i, gen::k, HWW::Load_trks_charge(), HWW::Load_trks_chi2(), HWW::Load_trks_d0(), HWW::Load_trks_d0Err(), HWW::Load_trks_d0phiCov(), HWW::Load_trks_etaErr(), HWW::Load_trks_ndof(), HWW::Load_trks_nlayers(), HWW::Load_trks_phiErr(), HWW::Load_trks_qualityMask(), HWW::Load_trks_trk_p4(), HWW::Load_trks_valid_pixelhits(), HWW::Load_trks_vertex_p4(), HWW::Load_trks_z0(), HWW::Load_trks_z0Err(), M_PI, NULL, reco::HitPattern::numberOfValidPixelHits(), listBenchmarks::pattern, reco::HitPattern::pixelHitFilter(), CosmicsPD_Skims::radius, mathSSE::sqrt(), reco::HitPattern::stripHitFilter(), TrackCollection_, HWW::trks_charge(), HWW::trks_chi2(), HWW::trks_d0(), HWW::trks_d0Err(), HWW::trks_d0phiCov(), HWW::trks_etaErr(), HWW::trks_ndof(), HWW::trks_nlayers(), HWW::trks_phiErr(), HWW::trks_qualityMask(), HWW::trks_trk_p4(), HWW::trks_valid_pixelhits(), HWW::trks_vertex_p4(), HWW::trks_z0(), HWW::trks_z0Err(), and reco::HitPattern::validHitFilter().

Referenced by HWWAnalyzer::analyze().

                                                                                     {

  using namespace edm;

  hww.Load_trks_trk_p4();
  hww.Load_trks_vertex_p4();
  hww.Load_trks_d0();
  hww.Load_trks_chi2();
  hww.Load_trks_ndof();
  hww.Load_trks_z0();
  hww.Load_trks_d0Err();
  hww.Load_trks_z0Err();
  hww.Load_trks_etaErr();
  hww.Load_trks_phiErr();
  hww.Load_trks_d0phiCov();
  hww.Load_trks_charge();
  hww.Load_trks_qualityMask();
  hww.Load_trks_valid_pixelhits();
  hww.Load_trks_nlayers();


  // Get Tracks //

  Handle<edm::View<reco::Track> > track_h;
  iEvent.getByToken(TrackCollection_, track_h); 
  if( !track_h.isValid() ) {
    LogInfo("OutputInfo") << " failed to retrieve track collection";
    LogInfo("OutputInfo") << " TrackMaker cannot continue...!";
    return;
  }

  // Get B Field //

  ESHandle<MagneticField> theMagField;
  iSetup.get<IdealMagneticFieldRecord>().get(theMagField);
  if( !theMagField.isValid() ) {
    LogInfo("OutputInfo") << " failed to retrieve the magnetic field";
    LogInfo("OutputInfo") << " TrackMaker cannot continue...!";
    return;
  }
  const MagneticField* bf = theMagField.product();

  // Get Tracker Geometry //

  ESHandle<TrackerGeometry> theG;
  iSetup.get<TrackerDigiGeometryRecord>().get(theG);
  
  // Loop over Tracks //

  View<reco::Track>::const_iterator tracks_end = track_h->end();
  unsigned int iTIndex=-1;
  for (View<reco::Track>::const_iterator i = track_h->begin(); i != tracks_end; ++i) {

    iTIndex++;

    hww.trks_trk_p4()       .push_back( LorentzVector( i->px(), i->py(), i->pz(), i->p() )       );
    hww.trks_vertex_p4()    .push_back( LorentzVector(i->vx(),i->vy(), i->vz(), 0.)              );
    hww.trks_d0()           .push_back( i->d0()                                                  );
    hww.trks_chi2()         .push_back( i->chi2()                                                );
    hww.trks_ndof()         .push_back( i->ndof()                                                );
    hww.trks_z0()           .push_back( i->dz()                                                  );
    hww.trks_d0Err()        .push_back( i->d0Error()                                             );
    hww.trks_z0Err()        .push_back( i->dzError()                                             );
    hww.trks_etaErr()       .push_back( i->etaError()                                            );
    hww.trks_phiErr()       .push_back( i->phiError()                                            );
    hww.trks_d0phiCov()     .push_back( -i->covariance(TrackBase::i_phi, TrackBase::i_dxy)       ); 
    hww.trks_charge()       .push_back( i->charge()                                              );
    hww.trks_qualityMask()  .push_back( i->qualityMask()                                         );

    GlobalPoint  tpVertex   ( i->vx(), i->vy(), i->vz() );
    GlobalVector tpMomentum ( i->px(), i->py(), i->pz() );
    int tpCharge ( i->charge() );
    
    FreeTrajectoryState fts ( tpVertex, tpMomentum, tpCharge, bf);

    const float zdist  = 314.;
    const float radius = 130.;
    const float corner = 1.479;

    Plane::PlanePointer lendcap      = Plane::build( Plane::PositionType (0, 0, -zdist), Plane::RotationType () );    
    Plane::PlanePointer rendcap      = Plane::build( Plane::PositionType (0, 0,  zdist), Plane::RotationType () );
    Cylinder::CylinderPointer barrel = Cylinder::build( Cylinder::PositionType (0, 0, 0), Cylinder::RotationType (), radius);
    AnalyticalPropagator myAP (bf, alongMomentum, 2*M_PI);
    TrajectoryStateOnSurface tsos;    
        
    if( i->eta() < -corner ) {
      tsos = myAP.propagate( fts, *lendcap);
    }
    else if( fabs(i->eta()) < corner ) {
      tsos = myAP.propagate( fts, *barrel);
    }
    else if( i->eta() > corner ) {
      tsos = myAP.propagate( fts, *rendcap);
    }
    
    const reco::HitPattern& pattern = i->hitPattern();    
    hww.trks_valid_pixelhits() .push_back(pattern.numberOfValidPixelHits());
      
    if(i->extra().isAvailable()) {
      bool valid_hit      = false;
      uint32_t hit_pattern; 
      int i_layer       = 1;
      bool pixel_hit   = false;
      bool strip_hit   = false;


      typedef Ref<edmNew::DetSetVector<SiStripCluster>,SiStripCluster > ClusterRef;
      typedef Ref<edmNew::DetSetVector<SiPixelCluster>, SiPixelCluster > pixel_ClusterRef;


      for(trackingRecHit_iterator ihit = i->recHitsBegin(); ihit != i->recHitsEnd(); ++ihit){
        if(i_layer > 1) break;
        int k = ihit-i->recHitsBegin();
        hit_pattern = pattern.getHitPattern(k);
        valid_hit = pattern.validHitFilter(hit_pattern);
        pixel_hit = pattern.pixelHitFilter(hit_pattern);
        strip_hit = pattern.stripHitFilter(hit_pattern);
        if(!valid_hit) continue;
        if(pixel_hit){
          const SiPixelRecHit *pixel_hit_cast = dynamic_cast<const SiPixelRecHit*>(&(**ihit));
          if (pixel_hit_cast == NULL){
            LogInfo("OutputInfo") << " pixel_hit_cast is NULL, TrackMaker quitting";
            return;
          } 
          if(i_layer == 1){
            i_layer++;

          }
        }
        else if (strip_hit){
          const SiStripRecHit1D *strip_hit_cast = dynamic_cast<const SiStripRecHit1D*>(&(**ihit));
          const SiStripRecHit2D *strip2d_hit_cast = dynamic_cast<const SiStripRecHit2D*>(&(**ihit));
          ClusterRef cluster;
          if(strip_hit_cast == NULL) {
            if(strip2d_hit_cast == NULL) {
              LogInfo("OutputInfo") << " strip2d_hit_cast is NULL, TrackMaker quitting";
              return;
            }
            cluster = strip2d_hit_cast->cluster();
          }
          else { 
            cluster = strip_hit_cast->cluster();
          }
          int cluster_size   = (int)cluster->amplitudes().size();
          int cluster_charge = 0;
          double   cluster_weight_size = 0.0;
          int max_strip_i = std::max_element(cluster->amplitudes().begin(),cluster->amplitudes().end())-cluster->amplitudes().begin();
          for(int istrip = 0; istrip < cluster_size; istrip++){
            cluster_charge += (int)cluster->amplitudes().at(istrip);
            cluster_weight_size += (istrip-max_strip_i)*(istrip-max_strip_i)*(cluster->amplitudes().at(istrip));
          }
          cluster_weight_size = sqrt(cluster_weight_size/cluster_charge);
          if(i_layer == 1){
            i_layer++;
          }
        }
      }
    }
    
    hww.trks_nlayers()    .push_back( i->hitPattern().trackerLayersWithMeasurement() );

  } // End loop on tracks

}

Member Data Documentation

edm::EDGetTokenT<edm::View<reco::Track> > TrackMaker::TrackCollection_

Definition at line 15 of file TrackMaker.h.

Referenced by SetVars(), and TrackMaker().