TracksToTrajectories Class Reference

Inheritance diagram for TracksToTrajectories:

Public Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override
	Convert a reco::TrackCollection into std::vector<Trajectory> More...
	TracksToTrajectories (const edm::ParameterSet &, const Count *)
	Constructor. More...
	~TracksToTrajectories () override
	Destructor. More...
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< Count > >
	EDProducer ()=default
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndRuns () const final

Static Public Member Functions
static void	globalEndJob (Count const *iCount)
static std::unique_ptr< Count >	initializeGlobalCache (edm::ParameterSet const &)

Private Attributes
edm::EDGetTokenT< reco::TrackCollection >	theTracksToken
std::unique_ptr< TrackTransformerBase >	theTrackTransformer

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< Count > >
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

This class, which is a EDProducer, takes a reco::TrackCollection from the Event and refits the rechits strored in the reco::Tracks. The final result is a std::vector of Trajectories (objs of the type "Trajectory"), which is loaded into the Event in a transient way

Author

R. Bellan - INFN Torino ricca.nosp@m.rdo..nosp@m.bella.nosp@m.n@ce.nosp@m.rn.ch

Definition at line 42 of file TracksToTrajectories.cc.

Constructor & Destructor Documentation

TracksToTrajectories::TracksToTrajectories	(	const edm::ParameterSet &	parameterSet,
		const Count *
	)

Constructor.

Definition at line 67 of file TracksToTrajectories.cc.

References Exception, and edm::ParameterSet::getParameter().

                                                                                         {
  theTracksToken = consumes<reco::TrackCollection>(parameterSet.getParameter<InputTag>("Tracks"));

  ParameterSet trackTransformerParam = parameterSet.getParameter<ParameterSet>("TrackTransformer");

  string type = parameterSet.getParameter<string>("Type");

  if (type == "Default")
    theTrackTransformer = std::make_unique<TrackTransformer>(trackTransformerParam);
  else if (type == "GlobalCosmicMuonsForAlignment")
    theTrackTransformer = std::make_unique<TrackTransformerForGlobalCosmicMuons>(trackTransformerParam);
  else if (type == "CosmicMuonsForAlignment")
    theTrackTransformer = std::make_unique<TrackTransformerForCosmicMuons>(trackTransformerParam);
  else {
    throw cms::Exception("TracksToTrajectories")
        << "The selected algorithm does not exist"
        << "\n"
        << "Possible choices are:"
        << "\n"
        << "Type = [Default, GlobalCosmicMuonsForAlignment, CosmicMuonsForAlignment]";
  }

  produces<vector<Trajectory>>("Refitted");
  produces<TrajTrackAssociationCollection>("Refitted");
}

TracksToTrajectories::~TracksToTrajectories ( )

override

Destructor.

Definition at line 94 of file TracksToTrajectories.cc.

{}

Member Function Documentation

void TracksToTrajectories::globalEndJob ( Count const *  iCount )

static

Definition at line 96 of file TracksToTrajectories.cc.

References constexpr, LogTrace, metname, tracksToTrajectories::Count::theNFailures, and tracksToTrajectories::Count::theNTracks.

                                                           {
  constexpr char metname[] = "Reco|TrackingTools|TracksToTrajectories";

  auto theNFailures = iCount->theNFailures.load();
  auto theNTracks = iCount->theNTracks.load();

  if (theNFailures != 0)
    LogWarning(metname) << "During the refit there were " << theNFailures << " out of " << theNTracks
                        << " tracks, i.e. failure rate is: " << double(theNFailures) / theNTracks;
  else {
    LogTrace(metname) << "Refit of the tracks done without any failure";
  }
}

static std::unique_ptr<Count> TracksToTrajectories::initializeGlobalCache ( edm::ParameterSet const &  )

inlinestatic

Definition at line 50 of file TracksToTrajectories.cc.

{ return std::make_unique<Count>(); }

void TracksToTrajectories::produce ( edm::Event &  event, const edm::EventSetup &  setup  )

override

Convert a reco::TrackCollection into std::vector<Trajectory>

Convert Tracks into Trajectories.

Definition at line 111 of file TracksToTrajectories.cc.

References constexpr, DEFINE_FWK_MODULE, LogTrace, metname, eostools::move(), and PDWG_EXOHSCP_cff::tracks.

                                                                        {
#ifdef EDM_ML_DEBUG
  constexpr char metname[] = "Reco|TrackingTools|TracksToTrajectories";
#endif

  theTrackTransformer->setServices(setup);

  // Collection of Trajectory
  auto trajectoryCollection = std::make_unique<vector<Trajectory>>();

  // Get the reference
  RefProd<vector<Trajectory>> trajectoryCollectionRefProd = event.getRefBeforePut<vector<Trajectory>>("Refitted");

  // Get the RecTrack collection from the event
  Handle<reco::TrackCollection> tracks;
  event.getByToken(theTracksToken, tracks);

  // Association map between Trajectory and Track
  auto trajTrackMap = std::make_unique<TrajTrackAssociationCollection>(trajectoryCollectionRefProd, tracks);

  Ref<vector<Trajectory>>::key_type trajectoryIndex = 0;
  reco::TrackRef::key_type trackIndex = 0;

  // Loop over the Rec tracks
  for (auto const& newTrack : *tracks) {
    ++(globalCache()->theNTracks);

    auto const& trajectoriesSM = theTrackTransformer->transform(newTrack);

    if (!trajectoriesSM.empty()) {
      // Load the trajectory in the Trajectory Container
      trajectoryCollection->push_back(trajectoriesSM.front());

      // Make the association between the Trajectory and the original Track
      trajTrackMap->insert(Ref<vector<Trajectory>>(trajectoryCollectionRefProd, trajectoryIndex++),
                           reco::TrackRef(tracks, trackIndex++));
    } else {
      LogTrace(metname) << "Error in the Track refitting. This should not happen";
      ++(globalCache()->theNFailures);
    }
  }
  LogTrace(metname) << "Load the Trajectory Collection";
  event.put(move(trajectoryCollection), "Refitted");
  event.put(move(trajTrackMap), "Refitted");
}

Member Data Documentation

edm::EDGetTokenT<reco::TrackCollection> TracksToTrajectories::theTracksToken

private

Definition at line 59 of file TracksToTrajectories.cc.

std::unique_ptr<TrackTransformerBase> TracksToTrajectories::theTrackTransformer

private

Definition at line 60 of file TracksToTrajectories.cc.