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" );