LayerMeasurements.cc

Go to the documentation of this file.

#include "TrackingTools/MeasurementDet/interface/LayerMeasurements.h"
#include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.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/DetLayers/interface/GeometricSearchDet.h"
#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "TrackingTools/DetLayers/interface/DetGroup.h"

#include "TrackingTools/TransientTrackingRecHit/interface/InvalidTransientRecHit.h"


#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <algorithm>

using namespace std;


namespace {
  typedef GeometricSearchDet::DetWithState DetWithState;
  inline
  void addInvalidMeas( std::vector<TrajectoryMeasurement>& result, 
               const TrajectoryStateOnSurface& ts, const GeomDet* det, const DetLayer& layer) {
    result.emplace_back(ts, std::make_shared<InvalidTrackingRecHit>(*det, TrackingRecHit::missing), 0.F,&layer);
  }
  
  
  
  inline
  std::vector<TrajectoryMeasurement> 
  get(const MeasurementDetSystem* theDetSystem, 
      const MeasurementTrackerEvent* theData,
      const DetLayer& layer,
      std::vector<DetWithState> const& compatDets,
      const TrajectoryStateOnSurface& ts, 
      const Propagator& prop, 
      const MeasurementEstimator& est) {
    std::vector<TrajectoryMeasurement> result;
    typedef TrajectoryMeasurement      TM;

    tracking::TempMeasurements tmps;

    for ( auto const & ds : compatDets) {
      MeasurementDetWithData mdet = theDetSystem->idToDet(ds.first->geographicalId(), *theData);
      if unlikely(mdet.isNull()) {
    throw MeasurementDetException( "MeasurementDet not found");
      }
      
      if (mdet.measurements(ds.second, est,tmps))
    for (std::size_t i=0; i!=tmps.size(); ++i)
      result.emplace_back(ds.second,std::move(tmps.hits[i]),tmps.distances[i],&layer);
      tmps.clear();
    }
    // 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(),layer);
    }
    else {
      // invalid state on first compatible Det
      addInvalidMeas( result, compatDets.front().second, compatDets.front().first,layer);
    }
  
    return result;
  }

}
 

// return just valid hits, no sorting (for seeding mostly)
bool LayerMeasurements::recHits(SimpleHitContainer & result,
                 const DetLayer& layer, 
                 const TrajectoryStateOnSurface& startingState,
                 const Propagator& prop, 
                 const MeasurementEstimator& est) const {

  auto  const & compatDets = layer.compatibleDets( startingState, prop, est);  
  if (compatDets.empty()) return false;
  bool ret=false;
  for ( auto const & ds : compatDets) {
    auto mdet = theDetSystem->idToDet(ds.first->geographicalId(), *theData);
    ret |=mdet.recHits(result,ds.second,est);
  }
  return ret;
}


vector<TrajectoryMeasurement>
LayerMeasurements::measurements( const DetLayer& layer, 
                 const TrajectoryStateOnSurface& startingState,
                 const Propagator& prop, 
                 const MeasurementEstimator& est) const {

  typedef DetLayer::DetWithState   DetWithState;

  vector<DetWithState>  const & compatDets = layer.compatibleDets( startingState, prop, est);
  
  if (!compatDets.empty())  return get(theDetSystem, theData, layer, compatDets, startingState, prop, est);
    
  vector<TrajectoryMeasurement> result;
  pair<bool, TrajectoryStateOnSurface> compat = layer.compatible( startingState, prop, est);
  
  if ( compat.first) {
    result.push_back( TrajectoryMeasurement( compat.second, 
                         std::make_shared<InvalidTrackingRecHitNoDet>(layer.surface(), TrackingRecHit::inactive), 0.F,
                         &layer));
    LogDebug("LayerMeasurements")<<"adding a missing hit.";
  }else LogDebug("LayerMeasurements")<<"adding not measurement.";


  return result;  
  
}


vector<TrajectoryMeasurementGroup>
LayerMeasurements::groupedMeasurements( const DetLayer& layer, 
                    const TrajectoryStateOnSurface& startingState,
                    const Propagator& prop, 
                    const MeasurementEstimator& est) const {
  vector<TrajectoryMeasurementGroup> result;
  
  vector<DetGroup> && groups = layer.groupedCompatibleDets( startingState, prop, est);
  result.reserve(groups.size());

  tracking::TempMeasurements tmps;
  for (auto&  grp : groups) {
    if ( grp.empty() )  continue;
    
    vector<TrajectoryMeasurement> tmpVec;
    for (auto const & det : grp) {
      MeasurementDetWithData mdet = theDetSystem->idToDet(det.det()->geographicalId(), *theData);
      if (mdet.isNull()) {
    throw MeasurementDetException( "MeasurementDet not found");
      }      
      if (mdet.measurements( det.trajectoryState(), est,tmps))
    for (std::size_t i=0; i!=tmps.size(); ++i)
      tmpVec.emplace_back(det.trajectoryState(),std::move(tmps.hits[i]),tmps.distances[i],&layer);
      tmps.clear();
    }
    
    // sort the final result
    LogDebug("LayerMeasurements")<<"Sorting " << tmpVec.size() << " measurements in this grp.";
    sort( tmpVec.begin(), tmpVec.end(), TrajMeasLessEstim());
    addInvalidMeas( tmpVec, grp,layer); 
    result.emplace_back(std::move(tmpVec), std::move(grp));
  }


  // if the result is empty check if the layer is compatible (for invalid measurement)
  if (result.empty()) {
    pair<bool, TrajectoryStateOnSurface> compat = layer.compatible( startingState, prop, est);
    if ( compat.first) {
      vector<TrajectoryMeasurement> tmVec;
      tmVec.emplace_back(compat.second, std::make_shared<InvalidTrackingRecHitNoDet>(layer.surface(), TrackingRecHit::inactive), 0.F,&layer);
      result.emplace_back(std::move(tmVec), DetGroup());
    }
  }
  return result;
}

void LayerMeasurements::addInvalidMeas( vector<TrajectoryMeasurement>& measVec,
                    const DetGroup& group,
                    const DetLayer& layer) const
{
  if (!measVec.empty()) {
    // invalidMeas on Det of most compatible hit
    measVec.emplace_back(measVec.front().predictedState(), 
             std::make_shared<InvalidTrackingRecHit>(*measVec.front().recHit()->det(), TrackingRecHit::missing),
             0.,&layer);
  }
  else if (!group.empty()) {
    // invalid state on first compatible Det
    measVec.emplace_back(group.front().trajectoryState(), 
             std::make_shared<InvalidTrackingRecHit>(*group.front().det(), TrackingRecHit::missing), 0.,&layer);
  }
}