PixelMatchNextLayers.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    EgammaElectronAlgos
// Class:      PixelMatchNextLayers
// 
//
// Original Author:  Ursula Berthon, Claude Charlot
//         Created:  Mon Mar 27 13:22:06 CEST 2006
// $Id: PixelMatchNextLayers.cc,v 1.17 2011/04/08 08:54:09 innocent Exp $
//
//

#include "TrackingTools/DetLayers/interface/DetLayer.h"
#include "TrackingTools/MeasurementDet/interface/LayerMeasurements.h" 
#include "TrackingTools/MaterialEffects/interface/PropagatorWithMaterial.h"
#include "RecoEgamma/EgammaElectronAlgos/interface/PixelMatchNextLayers.h"

#include "DataFormats/SiStripDetId/interface/TIDDetId.h"
#include "DataFormats/SiStripDetId/interface/TECDetId.h"

#include <iostream> 
#include <algorithm>

 
PixelMatchNextLayers::PixelMatchNextLayers(const LayerMeasurements * theLayerMeasurements,const DetLayer* ilayer, 
                       FreeTrajectoryState & aFTS,
                       const PropagatorWithMaterial *aProp,  
                       const BarrelMeasurementEstimator *aBarrelMeas,
                       const ForwardMeasurementEstimator *aForwardMeas,
                       bool searchInTIDTEC)
 {

  typedef std::vector<TrajectoryMeasurement>::const_iterator aMeas;
  std::vector<const DetLayer*> allayers;
  std::vector<const DetLayer*> nl = ilayer->nextLayers( aFTS, alongMomentum);
  for (std::vector<const DetLayer*>::const_iterator il = nl.begin(); il != nl.end(); il++) {
    allayers.push_back(*il);
    std::vector<const DetLayer*> n2l = (*il)->nextLayers( aFTS, alongMomentum);
    for (std::vector<const DetLayer*>::const_iterator i2l = n2l.begin(); i2l != n2l.end(); i2l++) {
      allayers.push_back(*i2l);
    }
  }

  const TrajectoryStateOnSurface tsos(aFTS,ilayer->surface());
  
  if (tsos.isValid()) 
    {    
      for (std::vector<const DetLayer*>::const_iterator il = allayers.begin(); il != allayers.end(); il++) 
    {
      if ( (*il)->subDetector()==GeomDetEnumerators::PixelBarrel || (*il)->subDetector()==GeomDetEnumerators::PixelEndcap ) {
        
        std::vector<TrajectoryMeasurement> pixelMeasurements;
        if ((*il)->subDetector()==GeomDetEnumerators::PixelBarrel) {
          pixelMeasurements = theLayerMeasurements->measurements( **il, tsos , *aProp, *aBarrelMeas); 
        } else {
          pixelMeasurements = theLayerMeasurements->measurements( **il, tsos, *aProp, *aForwardMeas);
        }
        for (aMeas m=pixelMeasurements.begin(); m!=pixelMeasurements.end(); m++){
          if (m == pixelMeasurements.begin()){
        CLHEP::Hep3Vector prediction(m->forwardPredictedState().globalPosition().x(),
                      m->forwardPredictedState().globalPosition().y(),
                      m->forwardPredictedState().globalPosition().z());
        predictionHere.push_back( prediction);
          }
          if (m->recHit()->isValid()) {
        measurementsHere.push_back( *m);
        hitsHere.push_back( m->recHit());

        //std::cout<<"\n SH B-D "<<std::endl;

          } else {
        badMeasurementsHere.push_back( *m);
          }
        }
      }
      if (searchInTIDTEC) {   
      //additional search in the TID layers
      if ( ((*il)->subDetector())==GeomDetEnumerators::TID && (ilayer->location()) == GeomDetEnumerators::endcap)
        {
          std::vector<TrajectoryMeasurement> pixelMeasurements;
          pixelMeasurements = theLayerMeasurements->measurements( (**il), tsos , *aProp, *aForwardMeas); 
          
          for (aMeas m=pixelMeasurements.begin(); m!=pixelMeasurements.end(); m++)
        {
          // limit search in first ring
          if (TIDDetId(m->recHit()->geographicalId()).ring() > 1) continue;
          if (m == pixelMeasurements.begin())
            {
              CLHEP::Hep3Vector prediction(m->forwardPredictedState().globalPosition().x(),
                        m->forwardPredictedState().globalPosition().y(),
                        m->forwardPredictedState().globalPosition().z());
              predictionHere.push_back( prediction);
            }
          if (m->recHit()->isValid()) 
            {
              measurementsHere.push_back( *m);
              hitsHere.push_back(m->recHit());
            }
          // else{ std::cout<<" 2H not valid "<<std::endl;}
        }
        } //end of TID search
      
      //additional search in the TEC layers
      if ( ((*il)->subDetector())==GeomDetEnumerators::TEC && (ilayer->location()) == GeomDetEnumerators::endcap)
        {
          std::vector<TrajectoryMeasurement> pixelMeasurements;
          pixelMeasurements = theLayerMeasurements->measurements( (**il), tsos , *aProp, *aForwardMeas); 
                  
          for (aMeas m=pixelMeasurements.begin(); m!=pixelMeasurements.end(); m++)
        {
          // limit search in first ring and first third wheels
          if (TECDetId(m->recHit()->geographicalId()).ring() > 1) continue;
          if (TECDetId(m->recHit()->geographicalId()).wheel() > 3) continue;
          if (m == pixelMeasurements.begin())
            {
              CLHEP::Hep3Vector prediction(m->forwardPredictedState().globalPosition().x(),
                        m->forwardPredictedState().globalPosition().y(),
                        m->forwardPredictedState().globalPosition().z());
              predictionHere.push_back( prediction);
            }
          if (m->recHit()->isValid()) 
            {
              measurementsHere.push_back( *m);
              hitsHere.push_back(m->recHit());

              //std::cout<<"\n SH TEC "<<std::endl;

            }
          // else{ std::cout<<" 2H not valid "<<std::endl;}
        }
        } //end of TEC search
      }
    }
    } 
}

  
std::vector<TrajectoryMeasurement> PixelMatchNextLayers::measurementsInNextLayers() const {

  return measurementsHere;
}

std::vector<TrajectoryMeasurement> PixelMatchNextLayers::badMeasurementsInNextLayers() const {

  return badMeasurementsHere;
}

TransientTrackingRecHit::RecHitContainer PixelMatchNextLayers::hitsInNextLayers() const {

  return hitsHere;
}

std::vector<CLHEP::Hep3Vector> PixelMatchNextLayers::predictionInNextLayers() const {

  return predictionHere;
}