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: " << std::hex << lastHit.geographicalId().rawId()
                                        << std::dec << ' ' << surface.position();
    theValid = false;
    return;
  }
  theValid = true;

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

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