SeedFromProtoTrack.cc

Go to the documentation of this file.

#include "RecoTracker/TkSeedGenerator/interface/SeedFromProtoTrack.h"

#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"

#include "TrackingTools/MaterialEffects/interface/PropagatorWithMaterial.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
#include "DataFormats/TrackReco/interface/Track.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

SeedFromProtoTrack::SeedFromProtoTrack(const reco::Track & proto,  
  const SeedingHitSet & hits, const edm::EventSetup& es)
  : theValid(true)
{
  for (unsigned int i= 0, n = hits.size(); i< n; ++i) {
    const TrackingRecHit * trh = hits[i]->hit();
   theHits.push_back( trh->clone() ); 
  }
  init(proto, es);
}

SeedFromProtoTrack::SeedFromProtoTrack(const reco::Track & proto,  const edm::EventSetup& es)
  : theValid(true)
{
  const TrackingRecHit* hit = nullptr;
  for (unsigned int iHit = 0, nHits = proto.recHitsSize(); iHit < nHits; ++iHit) {
    TrackingRecHitRef refHit = proto.recHit(iHit);
    hit = &(*refHit);
    theHits.push_back( hit->clone() );
  }
  init(proto,es);
}


void SeedFromProtoTrack::init(const reco::Track & proto, const edm::EventSetup& es)
{
  edm::ESHandle<TrackerGeometry> tracker;
  es.get<TrackerDigiGeometryRecord>().get(tracker);

  edm::ESHandle<Propagator>  propagatorHandle;
  es.get<TrackingComponentsRecord>().get("PropagatorWithMaterial",propagatorHandle);
  const Propagator*  propagator = &(*propagatorHandle);

  edm::ESHandle<MagneticField> field;
  es.get<IdealMagneticFieldRecord>().get(field);//fixme

  const reco::TrackBase::Point&  vtx = proto.referencePoint();
  const reco::TrackBase::Vector& mom = proto.momentum();
  GlobalTrajectoryParameters gtp( 
      GlobalPoint(vtx.x(),vtx.y(),vtx.z()),
      GlobalVector(mom.x(),mom.y(),mom.z()),
      proto.charge(),  &(*field) ); 

  CurvilinearTrajectoryError err = proto.covariance();

  FreeTrajectoryState fts( gtp, err);

  const TrackingRecHit & lastHit = theHits.back();

  TrajectoryStateOnSurface outerState =
      propagator->propagate(fts, tracker->idToDet(lastHit.geographicalId())->surface());

  if (!outerState.isValid()){    
    const Surface & surface = tracker->idToDet(lastHit.geographicalId())->surface();
    edm::LogError("SeedFromProtoTrack")<<" was trying to create a seed from:\n"<<fts<<"\n propagating to: "<<lastHit.geographicalId()<<surface.position();
    theValid = false;
    return ;
  }
  theValid = true;

  thePTraj = trajectoryStateTransform::persistentState(outerState, lastHit.geographicalId().rawId());


}

TrajectorySeed SeedFromProtoTrack::trajectorySeed() const 
{
  return TrajectorySeed( trajectoryState(), hits(), direction());
}