GeometricSearchDetMeasurements.h

Go to the documentation of this file.

#ifndef GeometricSearchDetMeasurements_H
#define GeometricSearchDetMeasurements_H

#include "TrackingTools/DetLayers/interface/GeometricSearchDet.h"
#include "TrackingTools/PatternTools/interface/TrajMeasLessEstim.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/MeasurementDet/interface/MeasurementDetException.h"
#include "TrackingTools/MeasurementDet/interface/MeasurementDetSystem.h"
#include "TrackingTools/MeasurementDet/interface/MeasurementDet.h"
#include "TrackingTools/MeasurementDet/interface/TrajectoryMeasurementGroup.h"
#include "TrackingTools/TransientTrackingRecHit/interface/InvalidTransientRecHit.h"


#include <algorithm>

class GeometricSearchDetMeasurements {
public:

  typedef GeometricSearchDet::DetWithState DetWithState;
  typedef TrajectoryStateOnSurface   TSOS;
  typedef TrajectoryMeasurement      TM;
  typedef TrajectoryMeasurementGroup TMG;

  GeometricSearchDetMeasurements( const MeasurementDetSystem* detSysytem) :
    theDetSystem(detSysytem) {}



  /*
  template <class TrajectoryState> 
  std::vector<DetWithState> getDets( const GeometricSearchDet& det,
                const TrajectoryState& ts, 
                const Propagator& prop, 
                const MeasurementEstimator& est) const {
    pair<bool, TSOS> compat = det.compatible( ts, prop, est);
    if ( compat.first) return det.fastCompatibleDets( compat.second, ts, prop, est);
    else return std::vector<DetWithState>();
  }

  template <class TrajectoryState> 
  std::vector<DetWithState> getDets( const GeometricSearchDet& det,
                const TrajectoryStateOnSurface& stateOnDet,
                const TrajectoryState& ts, 
                const Propagator& prop, 
                const MeasurementEstimator& est) const {
    return det.fastCompatibleDets( stateOnDet, ts, prop, est);
  }

  template <class TrajectoryState>
  std::vector<TrajectoryMeasurement> get( const GeometricSearchDet& det,
                     const TrajectoryState& ts, 
                     const Propagator& prop, 
                     const MeasurementEstimator& est) const {
    pair<bool, TSOS> compat = det.compatible( ts, prop, est);
    if ( compat.first) {
      return det.fastMeasurements( compat.second, ts, prop, est);
    }
    else return std::vector<TrajectoryMeasurement>();
  }
    
  template <class TrajectoryState>
  std::vector<TrajectoryMeasurement> get( const GeometricSearchDet& det,
                     const TrajectoryStateOnSurface& stateOnDet,
                     const TrajectoryState& ts, 
                     const Propagator& prop, 
                     const MeasurementEstimator& est) const {
    std::vector<DetWithState> compatDets = det.fastCompatibleDets( stateOnDet, ts, prop, est);
    if (!compatDets.empty()) {
      return get( det, compatDets, ts, prop, est);
    }
    else {
      std::vector<TrajectoryMeasurement> result;
      addInvalidMeas( result, stateOnDet, &det);
      return result;
    }
  }
  */

  template <class TrajectoryState>
  std::vector<TrajectoryMeasurement> get( const GeometricSearchDet& layer,
                      const std::vector<DetWithState>& compatDets,
                      const TrajectoryState& ts, 
                      const Propagator& prop, 
                      const MeasurementEstimator& est) const;
  /*
  template <class TrajectoryState>
  std::vector<TMG>
  getGrouped( const GeometricSearchDet& det, const std::vector<DetWithState>& dg,
          const TrajectoryState& ts, const Propagator& prop,
          const MeasurementEstimator& est) const {
    if (!dg.empty()) {
      std::vector<TMG> result(1);
      result[0] = TMG( get( det, dg, ts, prop, est), dg);
      return result;
    }
    else return std::vector<TMG>();
  }
  */

  
  void addInvalidMeas( std::vector<TrajectoryMeasurement>& result, 
               const TrajectoryStateOnSurface& ts, const GeomDet* det) const {
    result.push_back( TM( ts, InvalidTransientRecHit::build(det, TrackingRecHit::missing), 0.F,0));
  }
  

private:

  const MeasurementDetSystem* theDetSystem;

};


template <class TrajectoryState>
std::vector<TrajectoryMeasurement> 
GeometricSearchDetMeasurements::get( const GeometricSearchDet& layer,
                     const std::vector<DetWithState>& compatDets,
                     const TrajectoryState& ts, 
                     const Propagator& prop, 
                     const MeasurementEstimator& est) const
{
  std::vector<TrajectoryMeasurement> result;
  if (!compatDets.empty()) {
    for ( std::vector<DetWithState>::const_iterator i=compatDets.begin();
      i != compatDets.end(); i++) {
      const MeasurementDet* mdet = theDetSystem->idToDet(i->first->geographicalId());
      if (mdet == 0) {
    throw MeasurementDetException( "MeasurementDet not found");
      }

      std::vector<TM> tmp = mdet->fastMeasurements( i->second, ts, prop, est);
      if ( !tmp.empty()) {
    // only collect valid RecHits
    std::vector<TM>::iterator end = (tmp.back().recHit()->getType() != TrackingRecHit::missing ? tmp.end() : tmp.end()-1);
    result.insert( result.end(), tmp.begin(), end);
      }
    }
    // WARNING: we might end up with more than one invalid hit of type 'inactive' in result
    // to be fixed in order to avoid usless double traj candidates.

    // sort the final result
    if ( result.size() > 1) {
      sort( result.begin(), result.end(), TrajMeasLessEstim());
    }

    
    if ( !result.empty()) {
      // invalidMeas on Det of most compatible hit
      addInvalidMeas( result, result.front().predictedState(), result.front().recHit()->det());
    }
    else {
      // invalid state on first compatible Det
      addInvalidMeas( result, compatDets.front().second, compatDets.front().first);
    }
    

  }
  else {
    // this case should have been handled by the caller!
    throw MeasurementDetException("GeometricSearchDetMeasurements::get called with empty std::vector<DetWithState>");
  }
  return result;
}


#endif