MuonMillepedeTrackRefitter.cc

Go to the documentation of this file.

#include "Alignment/MuonAlignmentAlgorithms/plugins/MuonMillepedeTrackRefitter.h"

// Collaborating Class Header
#include "Alignment/MuonAlignmentAlgorithms/interface/SegmentToTrackAssociator.h"

#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "Geometry/Records/interface/GlobalTrackingGeometryRecord.h"
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include "Geometry/CommonDetUnit/interface/GeomDet.h"

#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "DataFormats/TrackReco/interface/Track.h"

#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"

#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "DataFormats/MuonDetId/interface/DTChamberId.h"
#include "DataFormats/GeometrySurface/interface/Cylinder.h"
#include "DataFormats/GeometrySurface/interface/Plane.h"
#include "DataFormats/GeometrySurface/interface/Cone.h"

#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"

#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHitBuilder.h"
#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHit.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"

#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"

#include <vector>
#include "TH2D.h"

MuonMillepedeTrackRefitter::MuonMillepedeTrackRefitter(const edm::ParameterSet& pset) {
  SACollectionTag = pset.getParameter<edm::InputTag>("SATrackCollectionTag");

  //Products
  produces<std::vector<Trajectory>>();
  produces<TrajTrackAssociationCollection>();
}

// Destructor
MuonMillepedeTrackRefitter::~MuonMillepedeTrackRefitter() {}

void MuonMillepedeTrackRefitter::produce(edm::Event& event, const edm::EventSetup& eventSetup) {
  //Get collections from the event

  edm::Handle<reco::TrackCollection> tracksSA;
  event.getByLabel(SACollectionTag, tracksSA);

  edm::ESHandle<MagneticField> theMGField;
  eventSetup.get<IdealMagneticFieldRecord>().get(theMGField);

  edm::ESHandle<GlobalTrackingGeometry> theTrackingGeometry;
  eventSetup.get<GlobalTrackingGeometryRecord>().get(theTrackingGeometry);

  edm::RefProd<std::vector<Trajectory>> trajectoryCollectionRefProd = event.getRefBeforePut<std::vector<Trajectory>>();

  //Allocate collection of tracks
  auto trajectoryCollection = std::make_unique<std::vector<Trajectory>>();
  // Association map between Trajectory and Track
  auto trajTrackMap = std::make_unique<TrajTrackAssociationCollection>();

  //Create the propagator

  std::map<edm::Ref<std::vector<Trajectory>>::key_type, edm::Ref<reco::TrackCollection>::key_type> trajToTrack_map;

  edm::Ref<std::vector<Trajectory>>::key_type trajectoryIndex = 0;

  reco::TrackRef::key_type trackIndex = 0;

  for (reco::TrackCollection::const_iterator trackSA = tracksSA->begin(); trackSA != tracksSA->end(); ++trackSA) {
    reco::TransientTrack tTrackSA(*trackSA, &*theMGField, theTrackingGeometry);

    //Create an empty trajectory
    Trajectory myTraj;

    TrajectoryStateOnSurface innerTSOS = tTrackSA.innermostMeasurementState();

    for (trackingRecHit_iterator theHit = tTrackSA.recHitsBegin(); theHit != tTrackSA.recHitsEnd(); ++theHit) {
      TrackingRecHit* myClone = (*theHit)->clone();
      const GeomDet* myDet = theTrackingGeometry->idToDet((*theHit)->geographicalId());
      TrajectoryMeasurement myMeas(innerTSOS,
                                   MuonTransientTrackingRecHit::specificBuild(myDet, (TrackingRecHit*)&*myClone));
      myTraj.push(myMeas);
    }

    trajectoryCollection->push_back(myTraj);
    trajToTrack_map[trajectoryIndex] = trackIndex;
    ++trajectoryIndex;
    ++trackIndex;
  }

  edm::OrphanHandle<std::vector<Trajectory>> trajsRef = event.put(std::move(trajectoryCollection));

  for (trajectoryIndex = 0; trajectoryIndex < tracksSA->size(); ++trajectoryIndex) {
    edm::Ref<reco::TrackCollection>::key_type trackCounter = trajToTrack_map[trajectoryIndex];
    trajTrackMap->insert(edm::Ref<std::vector<Trajectory>>(trajsRef, trajectoryIndex),
                         edm::Ref<reco::TrackCollection>(tracksSA, trackCounter));
  }

  event.put(std::move(trajTrackMap));
}

DEFINE_FWK_MODULE(MuonMillepedeTrackRefitter);