CTPPS: backport of diamond track fitting to 8_0_X

DataFormats/CTPPSReco/interface/CTPPSDiamondLocalTrack.h

@@ -0,0 +1,101 @@
+/****************************************************************************
+ *
+ * This is a part of CTPPS offline software.
+ * Authors:
+ *   Laurent Forthomme (laurent.forthomme@cern.ch)
+ *   Nicola Minafra nicola.minafra@cern.ch)
+ *
+ ****************************************************************************/
+
+#ifndef DataFormats_CTPPSReco_CTPPSDiamondLocalTrack
+#define DataFormats_CTPPSReco_CTPPSDiamondLocalTrack
+
+#include "DataFormats/Math/interface/Point3D.h"
+#include "DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h"
+
+//----------------------------------------------------------------------------------------------------
+
+class CTPPSDiamondLocalTrack
+{
+  public:
+    CTPPSDiamondLocalTrack() :
+      chi_squared_( 0. ), valid_( true ), t_( 0. ), t_sigma_( 0. ), ts_index_( 0 ), mh_( 0 ) {}
+    CTPPSDiamondLocalTrack( const math::XYZPoint& pos0, const math::XYZPoint& pos0_sigma, float chisq, float t, float t_sigma, int oot_idx, int mult_hits ) :
+      pos0_( pos0 ), pos0_sigma_( pos0_sigma ),
+      chi_squared_( chisq ), valid_( false ),
+      t_( t ), t_sigma_( t_sigma ), ts_index_( oot_idx ), mh_( mult_hits ) {}
+    virtual ~CTPPSDiamondLocalTrack() {}
+
+    //--- spatial get'ters
+
+    inline float getX0() const { return pos0_.x(); }
+    inline float getX0Sigma() const { return pos0_sigma_.x(); }
+
+    inline float getY0() const { return pos0_.y(); }
+    inline float getY0Sigma() const { return pos0_sigma_.y(); }
+
+    inline float getZ0() const { return pos0_.z(); }
+
+    inline float getChiSquared() const { return chi_squared_; }
+
+    //--- spatial set'ters
+
+    inline void setPosition( const math::XYZPoint& pos0 ) { pos0_ = pos0; }
+    inline void setPositionSigma( const math::XYZPoint& pos0_sigma ) { pos0_sigma_ = pos0_sigma; }
+
+    inline void setChiSquared( const float chisq ) { chi_squared_ = chisq; }
+
+    inline bool isValid() const { return valid_; }
+    inline void setValid( bool valid ) { valid_ = valid; }
+
+    //--- temporal get'ters
+
+    inline float getT() const { return t_; }
+    inline float getTSigma() const { return t_sigma_; }
+
+    //--- temporal set'ters
+
+    inline void setT( const float t ) { t_ = t; }
+    inline void setTSigma( const float t_sigma ) { t_sigma_ = t_sigma; }
+
+    inline void setOOTIndex( const int i ) { ts_index_ = i; }
+    inline int getOOTIndex() const { return ts_index_; }
+
+    inline void setMultipleHits( const int i ) { mh_ = i; }
+    inline int getMultipleHits() const { return mh_; }
+
+  private:
+    //--- spatial information
+
+    /// initial track position
+    math::XYZPoint pos0_;
+    /// error on the initial track position
+    math::XYZPoint pos0_sigma_;
+
+    /// fit chi^2
+    float chi_squared_;
+
+    /// fit valid?
+    bool valid_;
+
+    //--- timing information
+
+    float t_;
+    float t_sigma_;
+    /// Time slice index
+    int ts_index_;
+    /// Multiple hits counter
+    int mh_;
+
+};
+
+inline bool operator<( const CTPPSDiamondLocalTrack& lhs, const CTPPSDiamondLocalTrack& rhs )
+{
+  // start to sort by temporal coordinate
+  if ( lhs.getT() < rhs.getT() ) return true;
+  if ( lhs.getT() > rhs.getT() ) return false;
+  // then sort by x-position
+  return ( lhs.getX0() < rhs.getX0() );
+}
+
+#endif

DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h

@@ -19,12 +19,12 @@ class CTPPSDiamondRecHit
     CTPPSDiamondRecHit() :
       x_( 0. ), x_width_( 0. ), y_( 0. ), y_width_( 0. ),
       t_( 0. ), tot_( 0. ),
-      ts_index_( 0 ), hptdc_err_( 0 )
+      ts_index_( 0 ), hptdc_err_( 0 ), mh_( false )
     {}
-    CTPPSDiamondRecHit( float x, float x_width, float y, float y_width, float t, float tot, int oot_idx, const HPTDCErrorFlags& hptdc_err ) :
+    CTPPSDiamondRecHit( float x, float x_width, float y, float y_width, float t, float tot, int oot_idx, const HPTDCErrorFlags& hptdc_err, const bool mh ) :
       x_( x ), x_width_( x_width ), y_( y ), y_width_( y_width ),
       t_( t ), tot_( tot ),
-      ts_index_( oot_idx ), hptdc_err_( hptdc_err )
+      ts_index_( oot_idx ), hptdc_err_( hptdc_err ), mh_( mh )
     {}
 
     inline void setX( const float& x ) { x_ = x; }
@@ -47,6 +47,9 @@ class CTPPSDiamondRecHit
 
     inline void setOOTIndex( const int& i ) { ts_index_ = i; }
     inline int getOOTIndex() const { return ts_index_; }
+
+    inline void setMultipleHits( const bool mh ) { mh_ = mh; }
+    inline bool getMultipleHits() const { return mh_; }
 
     inline void setHPTDCErrorFlags( const HPTDCErrorFlags& err ) { hptdc_err_ = err; }
     inline HPTDCErrorFlags getHPTDCErrorFlags() const { return hptdc_err_; }
@@ -58,6 +61,7 @@ class CTPPSDiamondRecHit
     /// Time slice index
     int ts_index_;
     HPTDCErrorFlags hptdc_err_;
+    bool mh_;
 };
 
 //----------------------------------------------------------------------------------------------------

DataFormats/CTPPSReco/src/classes.h

@@ -10,6 +10,7 @@
 #include "DataFormats/CTPPSReco/interface/TotemRPLocalTrack.h"
 
 #include "DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h"
+#include "DataFormats/CTPPSReco/interface/CTPPSDiamondLocalTrack.h"
 
 #include "DataFormats/CTPPSReco/interface/CTPPSLocalTrackLite.h"
 
@@ -52,11 +53,22 @@ namespace DataFormats_CTPPSReco {
     CTPPSDiamondRecHit ctd_rh;
     edm::Ptr<CTPPSDiamondRecHit> ptr_ctd_rh;
     edm::Wrapper<CTPPSDiamondRecHit> wrp_ctd_rh;
-    std::vector<CTPPSDiamondRecHit> vec_rh;
-    std::vector< edm::DetSet<CTPPSDiamondRecHit> > vec_ds_rh;
-    edm::DetSet<CTPPSDiamondRecHit> ds_rh;
+    std::vector<CTPPSDiamondRecHit> vec_ctd_rh;
+    std::vector< edm::DetSet<CTPPSDiamondRecHit> > vec_ds_ctd_rh;
+    edm::DetSet<CTPPSDiamondRecHit> ds_ctd_rh;
     edm::DetSetVector<CTPPSDiamondRecHit> dsv_ctd_rh;
     edm::Wrapper< edm::DetSetVector<CTPPSDiamondRecHit> > wrp_dsv_ctd_rh;
+    edm::Wrapper< std::vector<CTPPSDiamondRecHit> > wrp_vec_ctd_rh;
+
+    CTPPSDiamondLocalTrack ctd_lt;
+    edm::Ptr<CTPPSDiamondLocalTrack> ptr_ctd_lt;
+    edm::Wrapper<CTPPSDiamondLocalTrack> wrp_ctd_lt;
+    std::vector<CTPPSDiamondLocalTrack> vec_ctd_lt;
+    edm::DetSet<CTPPSDiamondLocalTrack> ds_ctd_lt;
+    std::vector< edm::DetSet<CTPPSDiamondLocalTrack> > vec_ds_ctd_lt;
+    edm::Wrapper< std::vector<CTPPSDiamondLocalTrack> > wrp_vec_ctd_lt;
+    edm::DetSetVector<CTPPSDiamondLocalTrack> dsv_ctd_lt;
+    edm::Wrapper<edm::DetSetVector<CTPPSDiamondLocalTrack> > wrp_dsv_ctd_lt;
 
     //--- common objects
 

DataFormats/CTPPSReco/src/classes_def.xml

@@ -45,8 +45,9 @@
 
 <!-- diamonds objects -->
 
-  <class name="CTPPSDiamondRecHit" ClassVersion="3">
+  <class name="CTPPSDiamondRecHit" ClassVersion="4">
     <version ClassVersion="3" checksum="2377257106"/>
+    <version ClassVersion="4" checksum="2150979176"/>
   </class>
   <class name="edm::Ptr<CTPPSDiamondRecHit>"/>
   <class name="edm::Wrapper<CTPPSDiamondRecHit>"/>
@@ -55,6 +56,19 @@
   <class name="edm::DetSet<CTPPSDiamondRecHit>"/>
   <class name="edm::DetSetVector<CTPPSDiamondRecHit>"/>
   <class name="edm::Wrapper< edm::DetSetVector<CTPPSDiamondRecHit> >"/>
+  <class name="edm::Wrapper< std::vector<CTPPSDiamondRecHit> >"/>
+
+  <class name="CTPPSDiamondLocalTrack" ClassVersion="3">
+    <version ClassVersion="3" checksum="2739193553"/>
+  </class>
+  <class name="edm::Ptr<CTPPSDiamondLocalTrack>"/>
+  <class name="edm::Wrapper<CTPPSDiamondLocalTrack>"/>
+  <class name="std::vector<CTPPSDiamondLocalTrack>"/>
+  <class name="edm::DetSet<CTPPSDiamondLocalTrack>"/>
+  <class name="std::vector< edm::DetSet<CTPPSDiamondLocalTrack> >"/>
+  <class name="edm::Wrapper< std::vector<CTPPSDiamondLocalTrack> >"/>
+  <class name="edm::DetSetVector<CTPPSDiamondLocalTrack>"/>
+  <class name="edm::Wrapper< edm::DetSetVector<CTPPSDiamondLocalTrack> >"/>
 
 <!-- common objects -->
 

RecoCTPPS/Configuration/python/RecoCTPPS_EventContent_cff.py

@@ -19,6 +19,7 @@
     'keep CTPPSDiamondDigiedmDetSetVector_ctppsDiamondRawToDigi_*_*',
     'keep TotemVFATStatusedmDetSetVector_ctppsDiamondRawToDigi_*_*',
     'keep CTPPSDiamondRecHitedmDetSetVector_ctppsDiamondRecHits_*_*',
+    'keep CTPPSDiamondLocalTrackedmDetSetVector_ctppsDiamondLocalTracks_*_*',
 
     # CTPPS common
     'keep CTPPSLocalTrackLites_ctppsLocalTrackLiteProducer_*_*',
@@ -45,6 +46,7 @@
     'keep CTPPSDiamondDigiedmDetSetVector_ctppsDiamondRawToDigi_*_*',
     'keep TotemVFATStatusedmDetSetVector_ctppsDiamondRawToDigi_*_*',
     'keep CTPPSDiamondRecHitedmDetSetVector_ctppsDiamondRecHits_*_*',
+    'keep CTPPSDiamondLocalTrackedmDetSetVector_ctppsDiamondLocalTracks_*_*',
 
     # CTPPS common
     'keep CTPPSLocalTrackLites_ctppsLocalTrackLiteProducer_*_*',
@@ -71,6 +73,7 @@
     'keep CTPPSDiamondDigiedmDetSetVector_ctppsDiamondRawToDigi_*_*',
     'keep TotemVFATStatusedmDetSetVector_ctppsDiamondRawToDigi_*_*',
     'keep CTPPSDiamondRecHitedmDetSetVector_ctppsDiamondRecHits_*_*',
+    'keep CTPPSDiamondLocalTrackedmDetSetVector_ctppsDiamondLocalTracks_*_*',
 
     # CTPPS common
     'keep CTPPSLocalTrackLites_ctppsLocalTrackLiteProducer_*_*',

RecoCTPPS/TotemRPLocal/interface/CTPPSDiamondTrackRecognition.h

@@ -0,0 +1,73 @@
+/****************************************************************************
+ *
+ * This is a part of CTPPS offline software.
+ * Authors:
+ *   Laurent Forthomme (laurent.forthomme@cern.ch)
+ *   Nicola Minafra (nicola.minafra@cern.ch)
+ *
+ ****************************************************************************/
+
+#ifndef RecoCTPPS_TotemRPLocal_CTPPSDiamondTrackRecognition
+#define RecoCTPPS_TotemRPLocal_CTPPSDiamondTrackRecognition
+
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+
+#include "DataFormats/Common/interface/DetSet.h"
+#include "DataFormats/Common/interface/DetSetVector.h"
+
+#include "DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h"
+#include "DataFormats/CTPPSReco/interface/CTPPSDiamondLocalTrack.h"
+
+#include <vector>
+#include <unordered_map>
+#include "TF1.h"
+
+/**
+ * \brief Class performing smart reconstruction for CTPPS Diamond Detectors.
+ * \date Jan 2017
+**/
+class CTPPSDiamondTrackRecognition
+{
+  public:
+    CTPPSDiamondTrackRecognition( const edm::ParameterSet& );
+    ~CTPPSDiamondTrackRecognition();
+
+    /// Reset the list of hits
+    void clear();
+
+    /// Feed a new hit to the tracks recognition algorithm
+    void addHit( const CTPPSDiamondRecHit& recHit );
+
+    /// Produce a collection of tracks for the current station, given its hits collection
+    int produceTracks( edm::DetSet<CTPPSDiamondLocalTrack>& tracks );
+
+  private:
+    struct HitParameters {
+      HitParameters( const float center, const float width ) :
+        center( center ), width( width ) {}
+      float center;
+      float width;
+    };
+    typedef std::vector<HitParameters> HitParametersVector;
+    typedef std::unordered_map<int,HitParametersVector> HitParametersVectorMap;
+
+    /// Default hit function accounting for the pad spatial efficiency
+    static const std::string pixelEfficiencyDefaultFunction_;
+
+    const float threshold_;
+    const float thresholdFromMaximum_;
+    const float resolution_;
+    const float sigma_;
+    const float startFromX_;
+    const float stopAtX_;
+
+    float yPosition_;
+    float yWidth_;
+
+    /// Function for pad efficiency
+    TF1 hit_f_;
+    HitParametersVectorMap hitParametersVectorMap_;
+    std::unordered_map<int,int> mhMap_;
+};
+
+#endif