BzeroReferenceTrajectoryFactory.cc

Go to the documentation of this file.

// Local include from plugins directory...:
#include "BzeroReferenceTrajectoryFactory.h"
 
#include "Alignment/ReferenceTrajectories/interface/BzeroReferenceTrajectory.h"
#include "Alignment/ReferenceTrajectories/interface/TrajectoryFactoryPlugin.h"
 
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
 
#include "Alignment/ReferenceTrajectories/interface/TrajectoryFactoryBase.h"
 
 
 
BzeroReferenceTrajectoryFactory::BzeroReferenceTrajectoryFactory(const edm::ParameterSet &config)
    : TrajectoryFactoryBase(config),
      theMass(config.getParameter<double>("ParticleMass")),
      theMomentumEstimate(config.getParameter<double>("MomentumEstimate")) {
  edm::LogInfo("Alignment") << "@SUB=BzeroReferenceTrajectoryFactory"
                            << "mass: " << theMass << "\nmomentum: " << theMomentumEstimate;
}
 
BzeroReferenceTrajectoryFactory::~BzeroReferenceTrajectoryFactory(void) {}
 
const BzeroReferenceTrajectoryFactory::ReferenceTrajectoryCollection BzeroReferenceTrajectoryFactory::trajectories(
    const edm::EventSetup &setup, const ConstTrajTrackPairCollection &tracks, const reco::BeamSpot &beamSpot) const {
  ReferenceTrajectoryCollection trajectories;
 
  edm::ESHandle<MagneticField> magneticField;
  setup.get<IdealMagneticFieldRecord>().get(magneticField);
 
  ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
 
  while (itTracks != tracks.end()) {
    TrajectoryInput input = this->innermostStateAndRecHits(*itTracks);
    // Check input: If all hits were rejected, the TSOS is initialized as invalid.
    if (input.first.isValid()) {
      ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass, theMomentumEstimate);
      config.useBeamSpot = useBeamSpot_;
      config.includeAPEs = includeAPEs_;
      config.allowZeroMaterial = allowZeroMaterial_;
      // set the flag for reversing the RecHits to false, since they are already in the correct order.
      config.hitsAreReverse = false;
      trajectories.push_back(ReferenceTrajectoryPtr(
          new BzeroReferenceTrajectory(input.first, input.second, magneticField.product(), beamSpot, config)));
    }
 
    ++itTracks;
  }
 
  return trajectories;
}
 
const BzeroReferenceTrajectoryFactory::ReferenceTrajectoryCollection BzeroReferenceTrajectoryFactory::trajectories(
    const edm::EventSetup &setup,
    const ConstTrajTrackPairCollection &tracks,
    const ExternalPredictionCollection &external,
    const reco::BeamSpot &beamSpot) const {
  ReferenceTrajectoryCollection trajectories;
 
  if (tracks.size() != external.size()) {
    edm::LogInfo("ReferenceTrajectories")
        << "@SUB=BzeroReferenceTrajectoryFactory::trajectories"
        << "Inconsistent input:\n"
        << "\tnumber of tracks = " << tracks.size() << "\tnumber of external predictions = " << external.size();
    return trajectories;
  }
 
  edm::ESHandle<MagneticField> magneticField;
  setup.get<IdealMagneticFieldRecord>().get(magneticField);
 
  ConstTrajTrackPairCollection::const_iterator itTracks = tracks.begin();
  ExternalPredictionCollection::const_iterator itExternal = external.begin();
 
  while (itTracks != tracks.end()) {
    TrajectoryInput input = innermostStateAndRecHits(*itTracks);
    // Check input: If all hits were rejected, the TSOS is initialized as invalid.
    if (input.first.isValid()) {
      if ((*itExternal).isValid() && sameSurface((*itExternal).surface(), input.first.surface())) {
        ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass, theMomentumEstimate);
        config.useBeamSpot = useBeamSpot_;
        config.includeAPEs = includeAPEs_;
        config.allowZeroMaterial = allowZeroMaterial_;
        // set the flag for reversing the RecHits to false, since they are already in the correct order.
        config.hitsAreReverse = false;
        ReferenceTrajectoryPtr refTraj(
            new BzeroReferenceTrajectory(*itExternal, input.second, magneticField.product(), beamSpot, config));
 
        AlgebraicSymMatrix externalParamErrors(asHepMatrix<5>((*itExternal).localError().matrix()));
        refTraj->setParameterErrors(externalParamErrors.sub(2, 5));
 
        trajectories.push_back(refTraj);
      } else {
        ReferenceTrajectoryBase::Config config(materialEffects(), propagationDirection(), theMass, theMomentumEstimate);
        config.useBeamSpot = useBeamSpot_;
        config.includeAPEs = includeAPEs_;
        config.allowZeroMaterial = allowZeroMaterial_;
        // set the flag for reversing the RecHits to false, since they are already in the correct order.
        config.hitsAreReverse = false;
        trajectories.push_back(ReferenceTrajectoryPtr(
            new BzeroReferenceTrajectory(input.first, input.second, magneticField.product(), beamSpot, config)));
      }
    }
 
    ++itTracks;
    ++itExternal;
  }
 
  return trajectories;
}
 
DEFINE_EDM_PLUGIN(TrajectoryFactoryPlugin, BzeroReferenceTrajectoryFactory, "BzeroReferenceTrajectoryFactory");