GroupedDAFHitCollector.cc

Go to the documentation of this file.

#include "RecoTracker/SiTrackerMRHTools/interface/GroupedDAFHitCollector.h"
#include "RecoTracker/SiTrackerMRHTools/interface/SiTrackerMultiRecHitUpdator.h"
#include "RecoTracker/SiTrackerMRHTools/interface/MeasurementByLayerGrouper.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TSiTrackerMultiRecHit.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/PatternTools/interface/MeasurementEstimator.h"
#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/MeasurementDet/interface/TrajectoryMeasurementGroup.h"
#include "TrackingTools/TransientTrackingRecHit/interface/InvalidTransientRecHit.h"

#include <vector>
#include <map>

#define _debug_GroupedDAFHitCollector_ 

 using namespace std;


vector<TrajectoryMeasurement> GroupedDAFHitCollector::recHits(const Trajectory& traj) const{

    //it assumes that the measurements are sorted in the smoothing direction
    const vector<TrajectoryMeasurement>& meas = traj.measurements();
    const Propagator* forwardPropagator = getPropagator();
    const Propagator* backwardPropagator = getReversePropagator();
    if (traj.direction() == alongMomentum){
        forwardPropagator = getReversePropagator();
        backwardPropagator = getPropagator(); 
    }  
    if (meas.empty()) //return TransientTrackingRecHit::ConstRecHitContainer(); 
        return vector<TrajectoryMeasurement>();

    vector<pair<const DetLayer*, vector<TrajectoryMeasurement> > > mol = MeasurementByLayerGrouper(getMeasurementTracker()->geometricSearchTracker())(meas);


    //TransientTrackingRecHit::ConstRecHitContainer result;
    vector<TrajectoryMeasurement> result;

    //add a protection if all the measurement are on the same layer
    if(mol.size()<2)return vector<TrajectoryMeasurement>();

    //first layer
    //  cout<<"DAFHitCollectionFromRecTrack: first layer"<<endl;

    //it assumes that the measurements are sorted in the smoothing direction
    //TrajectoryStateOnSurface current = (*(mol.begin()+1)).second.front().updatedState();
    TrajectoryStateOnSurface current = (*(mol.rbegin()+1)).second.back().updatedState();
    //if (current.isValid()) current.rescaleError(10);

    
    vector<TrajectoryMeasurementGroup> groupedMeas;
    //protection for layers with invalid meas with no id associated
    //to be fixed
    //for the moment no hit are lookerd for in these layers
    //remind that:
    //groupedMeasurements will return at least a measurement with an invalid hit with no detid
    LogDebug("MultiRecHitCollector") << "Layer "  << mol.back().first  << " has " << mol.back().second.size() << " measurements";
    //debug
        LogTrace("MultiRecHitCollector") << "Original measurements are:";
        vector<TrajectoryMeasurement>::const_iterator ibeg = mol.back().second.begin();
        vector<TrajectoryMeasurement>::const_iterator iend = mol.back().second.end();
        for (vector<TrajectoryMeasurement>::const_iterator imeas = ibeg; imeas != iend; ++imeas){
            if (imeas->recHit()->isValid()){
                  LogTrace("MultiRecHitCollector") << "Valid Hit with DetId " << imeas->recHit()->geographicalId().rawId()
                                                   << " local position " << imeas->recHit()->hit()->localPosition();
            //                         << " on " << giulioGetLayer(imeas->recHit()->geographicalId());
                } else {
                  LogTrace("MultiRecHitCollector") << "Invalid Hit with DetId " << imeas->recHit()->geographicalId().rawId(); 
            //" on " << giulioGetLayer(imeas->recHit()->geographicalId());
                }
        }
        //debug      
    if (mol.back().first) groupedMeas = theLM.groupedMeasurements(*(mol.back().first), current, *backwardPropagator, *(getEstimator()));
    //in this case we have to sort the detGroups in the opposite way (according the forward propagator, not the backward one)
    vector<TrajectoryMeasurementGroup> sortedgroupedMeas; 
    for (vector<TrajectoryMeasurementGroup>::reverse_iterator iter = groupedMeas.rbegin();  iter != groupedMeas.rend(); iter++){
        sortedgroupedMeas.push_back(*iter);
    }
    buildMultiRecHits(sortedgroupedMeas, result);
        

    //other layers
    current = mol.back().second.front().updatedState();
    //if (current.isValid()) current.rescaleError(10);
    for(vector<pair<const DetLayer*, vector<TrajectoryMeasurement> > >::reverse_iterator imol = mol.rbegin() + 1; imol != mol.rend(); imol++) {
        const DetLayer* lay = (*imol).first;
        LogDebug("MultiRecHitCollector") << "Layer "  << lay << " has " << (*imol).second.size() << " measurements";
        //debug
                LogTrace("MultiRecHitCollector") << "Original measurements are:";
                vector<TrajectoryMeasurement>::const_iterator ibeg = (*imol).second.begin();
                vector<TrajectoryMeasurement>::const_iterator iend = (*imol).second.end();
                for (vector<TrajectoryMeasurement>::const_iterator imeas = ibeg; imeas != iend; ++imeas){
                        if (imeas->recHit()->isValid()){
                          LogTrace("MultiRecHitCollector") << "Valid Hit with DetId " << imeas->recHit()->geographicalId().rawId()
                               << " local position " << imeas->recHit()->hit()->localPosition();
                //<< " on " << giulioGetLayer(imeas->recHit()->geographicalId());
                        } else {
                          LogTrace("MultiRecHitCollector") << "Invalid Hit with DetId " << imeas->recHit()->geographicalId().rawId();
              //<< " on " << giulioGetLayer(imeas->recHit()->geographicalId());
                        }
                }
                //debug 
        vector<TrajectoryMeasurementGroup> currentLayerMeas;
        if (lay) currentLayerMeas = theLM.groupedMeasurements(*lay, current, *forwardPropagator, *(getEstimator()));
            buildMultiRecHits(currentLayerMeas, result);
        current = (*imol).second.front().updatedState();
        //if (current.isValid()) current.rescaleError(10);
    }
    LogTrace("MultiRecHitCollector") << "Original Measurement size "  << meas.size() << " GroupedDAFHitCollector returned " << result.size() << " measurements";
    //results are sorted in the fitting direction

 // adding a protection against too few hits and invalid hits (due to failed propagation on the same surface of the original hits)
        if (result.size()>2)
          {
            int hitcounter=0;
            //check if the vector result has more than 3 valid hits
            for (vector<TrajectoryMeasurement>::const_iterator iimeas = result.begin(); iimeas != result.end(); ++iimeas)
              {
                if(iimeas->recHit()->isValid()) hitcounter++;
              }
            
            if(hitcounter>2) 
              {return result;} 
            
            else return vector<TrajectoryMeasurement>();
          }
        
        else{return vector<TrajectoryMeasurement>();}
    
}

void GroupedDAFHitCollector::buildMultiRecHits(const vector<TrajectoryMeasurementGroup>& measgroup, vector<TrajectoryMeasurement>& result) const {

    unsigned int initial_size = result.size();
    //TransientTrackingRecHit::ConstRecHitContainer rhits;
    if (measgroup.empty()) {
        LogTrace("MultiRecHitCollector") << "No TrajectoryMeasurementGroups found for this layer" ; 
        //should we do something?
        //result.push_back(InvalidTransientRecHit::build(0,TrackingRecHit::missing));
        return;
    } 

    //we build a MultiRecHit out of each group
    //groups are sorted along momentum or opposite to momentum, 
    //measurements in groups are sorted with increating chi2
    LogTrace("MultiRecHitCollector") << "Found " << measgroup.size() << " groups for this layer";
    //trajectory state to store the last valid TrajectoryState (if present) to be used 
    //to add an invalid Measurement in case no valid state or no valid hits are found in any group
    for (vector<TrajectoryMeasurementGroup>::const_iterator igroup = measgroup.begin(); igroup != measgroup.end(); igroup++ ){
        //the TrajectoryState is the first one
        TrajectoryStateOnSurface state = igroup->measurements().front().predictedState();
        if (!state.isValid()){
            LogTrace("MultiRecHitCollector") << "Something wrong! no valid TSOS found in current group ";
                        continue;       
        }
        //debug
                LogTrace("MultiRecHitCollector") << "This group has " << igroup->measurements().size() << " measurements";
                LogTrace("MultiRecHitCollector") << "This group has the following " << igroup->detGroup().size() << " detector ids: " << endl;
                for (DetGroup::const_iterator idet = igroup->detGroup().begin(); idet != igroup->detGroup().end(); ++idet){
          LogTrace("MultiRecHitCollector") << idet->det()->geographicalId().rawId();
              // " on " << giulioGetLayer(idet->det()->geographicalId());
                }
                //debug
        vector<const TrackingRecHit*> hits;
        for (vector<TrajectoryMeasurement>::const_iterator imeas = igroup->measurements().begin(); imeas != igroup->measurements().end(); imeas++){
            //collect the non missing hits to build the MultiRecHits
            //we ese the recHits method; anyway only simple hits, not MultiHits should be present 
            if (imeas->recHit()->getType() != TrackingRecHit::missing) {
                LogTrace("MultiRecHitCollector") << "This hit is valid ";
                hits.push_back(imeas->recHit()->hit());
            }
        }
        if (hits.empty()){
            LogTrace("MultiRecHitCollector") << "No valid hits found in current group ";
            continue;
        }
        LogTrace("MultiRecHitCollector") << "The best TSOS in this group is " << state << " it lays on surface located at " << state.surface().position();
#ifdef _debug_GroupedDAFHitCollector_   
        LogTrace("MultiRecHitCollector") << "For the MRH on this group the following hits will be used"; 
        for (vector<const TrackingRecHit*>::iterator iter = hits.begin(); iter != hits.end(); iter++){  
            string validity = "valid";
            if ((*iter)->getType() == TrackingRecHit::missing ) validity = "missing !should not happen!";
            else if ((*iter)->getType() == TrackingRecHit::inactive) validity = "inactive";
            else if ((*iter)->getType() == TrackingRecHit::bad) validity = "bad";   
            LogTrace("MultiRecHitCollector") << "DetId " << (*iter)->geographicalId().rawId()  << " validity: " << validity 
                << " surface position " << getMeasurementTracker()->geomTracker()->idToDet((*iter)->geographicalId())->position()  
                << " hit local position " << (*iter)->localPosition();
        }
#endif
        
        result.push_back(TrajectoryMeasurement(state,theUpdator->buildMultiRecHit(hits, state)));
    }

    //can this happen? it means that the measgroup was not empty but no valid measurement was found inside
    //in this case we add an invalid measuremnt for this layer 
    if (result.size() == initial_size){
        LogTrace("MultiRecHitCollector") << "no valid measuremnt or no valid TSOS in none of the groups";
        //measgroup has been already checked for size != 0
        if (measgroup.back().measurements().size() != 0){   
            result.push_back(measgroup.back().measurements().back());
        } 
    }
}