Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/Exception.h"

#include "RecoTracker/TransientTrackingRecHit/interface/ProjectedRecHit2D.h"

#include "Alignment/ReferenceTrajectories/interface/TrajectoryFactoryBase.h"

TrajectoryFactoryBase::TrajectoryFactoryBase(const edm::ParameterSet& config, const edm::ConsumesCollector& iC)
    : TrajectoryFactoryBase(config, 1, iC) {}

TrajectoryFactoryBase::TrajectoryFactoryBase(const edm::ParameterSet& config,
                                             unsigned int tracksPerTrajectory,
                                             const edm::ConsumesCollector& iC)
    : cfg_(config),
      tracksPerTrajectory_(tracksPerTrajectory),
      materialEffects_(materialEffects(config.getParameter<std::string>("MaterialEffects"))),
      propDir_(propagationDirection(config.getParameter<std::string>("PropagationDirection"))),
      useWithoutDet_(config.getParameter<bool>("UseHitWithoutDet")),
      useInvalidHits_(config.getParameter<bool>("UseInvalidHits")),
      useProjectedHits_(config.getParameter<bool>("UseProjectedHits")),
      useBeamSpot_(config.getParameter<bool>("UseBeamSpot")),
      includeAPEs_(config.getParameter<bool>("IncludeAPEs")),
      allowZeroMaterial_(config.getParameter<bool>("AllowZeroMaterial")) {
  edm::LogInfo("Alignment") << "@SUB=TrajectoryFactoryBase"
                            << "TrajectoryFactory '" << cfg_.getParameter<std::string>("TrajectoryFactoryName")
                            << "' with following settings:"
                            << "\nmaterial effects: " << cfg_.getParameter<std::string>("MaterialEffects")
                            << "\npropagation: " << cfg_.getParameter<std::string>("PropagationDirection")
                            << "\nuse hits without det: " << (useWithoutDet_ ? "yes" : "no")
                            << "\nuse invalid hits: " << (useInvalidHits_ ? "yes" : "no")
                            << "\nuse projected hits: " << (useProjectedHits_ ? "yes" : "no")
                            << "\nuse beamspot: " << (useBeamSpot_ ? "yes" : "no")
                            << "\ninclude APEs: " << (includeAPEs_ ? "yes" : "no")
                            << "\nallow zero material: " << (allowZeroMaterial_ ? "yes" : "no");
}

TrajectoryFactoryBase::~TrajectoryFactoryBase(void) {}

const TrajectoryFactoryBase::TrajectoryInput TrajectoryFactoryBase::innermostStateAndRecHits(
    const ConstTrajTrackPair& track) const {
  TrajectoryInput result;

  // get the trajectory measurements in the correct order, i.e. reverse if needed
  Trajectory::DataContainer trajectoryMeasurements = this->orderedTrajectoryMeasurements(*track.first);
  Trajectory::DataContainer::iterator itM = trajectoryMeasurements.begin();

  // get the innermost valid trajectory state - the corresponding hit must be o.k. as well
  while (itM != trajectoryMeasurements.end()) {
    if ((*itM).updatedState().isValid() && useRecHit((*itM).recHit()))
      break;
    ++itM;
  }
  if (itM != trajectoryMeasurements.end())
    result.first = (*itM).updatedState();

  // get the valid RecHits
  while (itM != trajectoryMeasurements.end()) {
    TransientTrackingRecHit::ConstRecHitPointer aRecHit = (*itM).recHit();
    if (useRecHit(aRecHit))
      result.second.push_back(aRecHit);
    ++itM;
  }

  return result;
}

const Trajectory::DataContainer TrajectoryFactoryBase::orderedTrajectoryMeasurements(
    const Trajectory& trajectory) const {
  const PropagationDirection dir = trajectory.direction();
  const bool hitsAreReverse = ((dir == propDir_ || propDir_ == anyDirection) ? false : true);

  const Trajectory::DataContainer& original = trajectory.measurements();

  if (hitsAreReverse) {
    // Simply use this line instead of the copying by hand?
    // const Trajectory::DataContainer reordered(original.rbegin(), original.rend());
    Trajectory::DataContainer reordered;
    reordered.reserve(original.size());

    Trajectory::DataContainer::const_reverse_iterator itM;
    for (itM = original.rbegin(); itM != original.rend(); ++itM) {
      reordered.push_back(*itM);
    }
    return reordered;
  }

  return original;
}

bool TrajectoryFactoryBase::sameSurface(const Surface& s1, const Surface& s2) const {
  // - Should use perp2() instead of perp()
  // - Should not rely on floating point equality, but make a minimal range, e.g. 1.e-6 ?
  return (s1.eta() == s2.eta()) && (s1.phi() == s2.phi()) && (s1.position().perp() == s2.position().perp());
}

bool TrajectoryFactoryBase::useRecHit(const TransientTrackingRecHit::ConstRecHitPointer& hitPtr) const {
  const GeomDet* det = hitPtr->det();
  if (!det && !useWithoutDet_)
    return false;

  if (!(useInvalidHits_ || hitPtr->isValid()))
    return false;

  if (!useProjectedHits_) {
    if (trackerHitRTTI::isProjected(*hitPtr))
      return false;
  }

  return true;
}

TrajectoryFactoryBase::MaterialEffects TrajectoryFactoryBase::materialEffects(const std::string& strME) const {
  if (strME == "MultipleScattering")
    return ReferenceTrajectoryBase::multipleScattering;
  if (strME == "EnergyLoss")
    return ReferenceTrajectoryBase::energyLoss;
  if (strME == "Combined")
    return ReferenceTrajectoryBase::combined;
  if (strME == "None")
    return ReferenceTrajectoryBase::none;
  if (strME == "BreakPoints")
    return ReferenceTrajectoryBase::breakPoints;
  if (strME == "BrokenLines")
    return ReferenceTrajectoryBase::brokenLinesCoarse;
  if (strME == "BrokenLinesCoarse")
    return ReferenceTrajectoryBase::brokenLinesCoarse;
  if (strME == "BrokenLinesFine")
    return ReferenceTrajectoryBase::brokenLinesFine;
  if (strME == "LocalGBL")
    return ReferenceTrajectoryBase::localGBL;
  if (strME == "CurvlinGBL")
    return ReferenceTrajectoryBase::curvlinGBL;

  throw cms::Exception("BadConfig") << "[TrajectoryFactoryBase::materialEffects] Unknown parameter: " << strME;
}

PropagationDirection TrajectoryFactoryBase::propagationDirection(const std::string& strPD) const {
  if (strPD == "oppositeToMomentum")
    return oppositeToMomentum;
  if (strPD == "alongMomentum")
    return alongMomentum;
  if (strPD == "anyDirection")
    return anyDirection;

  throw cms::Exception("BadConfig") << "[TrajectoryFactoryBase::propagationDirection] Unknown parameter: " << strPD;
}