Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 13:30:40

0001 #ifndef L1Trigger_TrackFindingTMTT_ChiSquaredFitBase_h
0002 #define L1Trigger_TrackFindingTMTT_ChiSquaredFitBase_h
0003 
0004 ///=== This is the base class for the linearised chi-squared track fit algorithms.
0005 
0006 ///=== Written by: Sioni Summers and Alexander D. Morton.
0007 
0008 #include "L1Trigger/TrackFindingTMTT/interface/Stub.h"
0009 #include "L1Trigger/TrackFindingTMTT/interface/TrackFitGeneric.h"
0010 #include "L1Trigger/TrackFindingTMTT/interface/Settings.h"
0011 #include "L1Trigger/TrackFindingTMTT/interface/L1track3D.h"
0012 #include <TMatrixD.h>
0013 #include <TVectorD.h>
0014 
0015 #include <vector>
0016 #include <map>
0017 #include <utility>
0018 
0019 namespace tmtt {
0020 
0021   class ChiSquaredFitBase : public TrackFitGeneric {
0022   public:
0023     enum PAR_IDS { INVR, PHI0, T, Z0, D0 };
0024 
0025   public:
0026     ChiSquaredFitBase(const Settings* settings, const uint nPar);
0027 
0028     L1fittedTrack fit(const L1track3D& l1track3D) override;
0029 
0030   protected:
0031     /* Methods */
0032     virtual TVectorD seed(const L1track3D& l1track3D) = 0;
0033     virtual TVectorD residuals(const TVectorD& x) = 0;  // Stub residuals/uncertainty
0034     virtual TMatrixD D(const TVectorD& x) = 0;          // derivatives
0035     virtual TMatrixD Vinv() = 0;                        // Covariances
0036 
0037     /* Variables */
0038     double qOverPt_seed_;
0039     std::vector<Stub*> stubs_;
0040     TVectorD trackParams_;
0041     uint nPar_;
0042     float largestresid_;
0043     int ilargestresid_;
0044     double chiSq_;
0045 
0046   private:
0047     void calculateChiSq(const TVectorD& resids);
0048     void calculateDeltaChiSq(const TVectorD& deltaX, const TVectorD& covX);
0049 
0050     int numFittingIterations_;
0051     int killTrackFitWorstHit_;
0052     double generalResidualCut_;
0053     double killingResidualCut_;
0054 
0055     unsigned int minStubLayers_;
0056     unsigned int minStubLayersRed_;
0057   };
0058 
0059 }  // namespace tmtt
0060 
0061 #endif