PixelCPEClusterRepair.h

Go to the documentation of this file.

#ifndef RecoLocalTracker_SiPixelRecHits_PixelCPEClusterRepair_H
#define RecoLocalTracker_SiPixelRecHits_PixelCPEClusterRepair_H

#include "RecoLocalTracker/SiPixelRecHits/interface/PixelCPEBase.h"

// Already in the base class
//#include "Geometry/CommonDetUnit/interface/GeomDetType.h"
//#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
//#include "Geometry/TrackerGeometryBuilder/interface/RectangularPixelTopology.h"
//#include "Geometry/CommonDetAlgo/interface/MeasurementPoint.h"
//#include "Geometry/CommonDetAlgo/interface/MeasurementError.h"
//#include "Geometry/Surface/interface/GloballyPositioned.h"
//#include "FWCore/ParameterSet/interface/ParameterSet.h"

// The template header files
//
#include "RecoLocalTracker/SiPixelRecHits/interface/SiPixelTemplateReco.h"
#include "RecoLocalTracker/SiPixelRecHits/interface/SiPixelTemplateReco2D.h"
#include "CondFormats/SiPixelTransient/interface/SiPixelTemplate.h"
#include "CondFormats/SiPixelTransient/interface/SiPixelTemplate2D.h"
#include "CondFormats/SiPixelObjects/interface/SiPixel2DTemplateDBObject.h"

#include <utility>
#include <vector>


#if 0

#endif

class MagneticField;
class PixelCPEClusterRepair : public PixelCPEBase
{
public:
   struct ClusterParamTemplate : ClusterParam
   {
      ClusterParamTemplate(const SiPixelCluster & cl) : ClusterParam(cl){}
      // The result of PixelTemplateReco2D
      float templXrec_ ;
      float templYrec_ ;
      float templSigmaX_ ;
      float templSigmaY_ ;
      // Add new information produced by SiPixelTemplateReco::PixelTempReco2D &&&
      // These can only be accessed if we change silicon pixel data formats and add them to the rechit
      int   ierr;
      
      // 2D fit stuff.
      float templProbXY_ ;
      bool  recommended2D_ ;
      int   ierr2;
   };
   
   // PixelCPEClusterRepair( const DetUnit& det );
   PixelCPEClusterRepair(edm::ParameterSet const& conf, const MagneticField *, const TrackerGeometry&, const TrackerTopology&,
             const SiPixelLorentzAngle *, const SiPixelTemplateDBObject *, const SiPixel2DTemplateDBObject * );
   
   ~PixelCPEClusterRepair() override;
   
private:
   ClusterParam * createClusterParam(const SiPixelCluster & cl) const override;
   
   // Calculate local position.  (Calls TemplateReco)
   LocalPoint localPosition (DetParam const & theDetParam, ClusterParam & theClusterParam) const override;
   // Calculate local error. Note: it MUST be called AFTER localPosition() !!!
   LocalError localError   (DetParam const & theDetParam, ClusterParam & theClusterParam) const override;
   
   // Helper functions: 

   // Call vanilla template reco, then clean-up
   void callTempReco1D( DetParam const & theDetParam, 
            ClusterParamTemplate & theClusterParam, 
            SiPixelTemplateReco::ClusMatrix & clusterPayload,
            int ID, LocalPoint & lp ) const;

   // Call 2D template reco, then clean-up
   void callTempReco2D( DetParam const & theDetParam, 
            ClusterParamTemplate & theClusterParam, 
            SiPixelTemplateReco2D::ClusMatrix & clusterPayload,
            int ID, LocalPoint & lp ) const;

   //Helper function to see if 2D reco should be run
   void checkRecommend2D( DetParam const & theDetParam, 
            ClusterParamTemplate & theClusterParam, 
            SiPixelTemplateReco::ClusMatrix & clusterPayload,
            int ID) const;

   //Fill IDs for 2D template
   void fill2DTemplIDs();


   

   // Template storage
   std::vector< SiPixelTemplateStore >   thePixelTemp_;
   std::vector< SiPixelTemplateStore2D > thePixelTemp2D_;

   int speed_ ;
   
   bool UseClusterSplitter_;

   // Template file management (when not getting the templates from the DB)
   int barrelTemplateID_ ;
   int forwardTemplateID_ ;
   std::string templateDir_ ;

   const SiPixel2DTemplateDBObject * templateDBobject2D_;

   // Configure 2D reco.
   float minChargeRatio_;
   float maxSizeMismatchInY_ ;
   
   //bool DoCosmics_;
   //bool LoadTemplatesFromDB_;

   // read sub-detectors to recommend 2D
   class Rule {
   public:
     // parse a rule from a string
     Rule(const std::string &str) ;
     // check this DetId to recommend 2D or not (default false)
     bool recommend(DetId detid, const TrackerTopology &tTopo) const {
       // check detector
       if (detid.subdetId() == subdet_) {
     // check layer
     if ( (layer_ == 0) || (layer_ == int(tTopo.layer(detid))) ) {
       return true;
     }
     else return false;
       }
       else return false;
     }
   private:
     int  subdet_;
     int  layer_;
   };
   std::vector<Rule> recommend2D_;

   // run on damaged hits or not
   bool runDamagedClusters_;
};

#endif