MuonCalibTrack_E.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCALIBEXTRATREEEVENT_MUONCALIBTRACK_E_H
#define MUONCALIBEXTRATREEEVENT_MUONCALIBTRACK_E_H
 
#include <EventPrimitives/EventPrimitives.h>
#include <MuonCalibExtraTreeEvent/MuonCalibHit_E.h>
#include <MuonCalibExtraTreeEvent/MuonCalibHole_E.h>
 
#include <cmath>
#include <memory>
#include <vector>
namespace MuonCalib {
    class MuonCalibSegment;
    class MuonCalibTrack_E {
    public:
        using CalibHitPtr = std::shared_ptr<const MuonCalibHit_E>;
        using CalibHolePtr = std::shared_ptr<MuonCalibHole_E>;
        using CalibSegPtr = std::shared_ptr<MuonCalibSegment>;
        using HitVector = std::vector<CalibHitPtr>;    
        using HoleVector = std::vector<CalibHolePtr>;  
        using SegmentVector = std::vector<CalibSegPtr>;
        MuonCalibTrack_E() = default;  
 
        static constexpr float dummy_chi2{9999.};
        struct defineParams {
            float x0{0.};
            float y0{0.};
            float z0{0.};
            float phi{5. * M_PI};
            float theta{0.};
            float qOverP{0.};
 
            int author{0};
            float chi2{dummy_chi2};
            int ndof{1};
            AmgSymMatrix(5) cov{};
        };
 
        MuonCalibTrack_E(defineParams pars);
 
        MuonCalibTrack_E(const MuonCalibTrack_E& pat) = default;
        virtual ~MuonCalibTrack_E() = default;
 
        const HitVector& hits() const;
        HitVector& hits();
 
        const HoleVector& holes() const;
        HoleVector& holes();
 
        const SegmentVector& segmetnsOnTrack() const;
        SegmentVector& segmetnsOnTrack();
 
        MuonCalibTrack_E& operator=(const MuonCalibTrack_E& rhs) = default;  
        unsigned int nrHits() const { return m_hits.size(); }                
        unsigned int nrHoles() const { return m_holes.size(); }              
 
        virtual float d0() const;    
        virtual float z0ip() const;  
        float x0() const;            
        float y0() const;            
        float z0() const;            
        float phi() const;           
        float theta() const;         
        float qOverP() const;        
        float p() const;             
        float pt() const;            
        float errInvP() const;       
        int author() const;          
        float chi2() const;          
        int ndof() const;            
 
        float cov00() const { return m_params.cov(0, 0); }  
        float cov01() const { return m_params.cov(0, 1); }  
        float cov02() const { return m_params.cov(0, 2); }  
        float cov03() const { return m_params.cov(0, 3); }  
        float cov04() const { return m_params.cov(0, 4); }  
        float cov11() const { return m_params.cov(1, 1); }  
        float cov12() const { return m_params.cov(1, 2); }  
        float cov13() const { return m_params.cov(1, 3); }  
        float cov14() const { return m_params.cov(1, 4); }  
        float cov22() const { return m_params.cov(2, 2); }  
        float cov23() const { return m_params.cov(2, 3); }  
        float cov24() const { return m_params.cov(2, 4); }  
        float cov33() const { return m_params.cov(3, 3); }  
        float cov34() const { return m_params.cov(3, 4); }  
        float cov44() const { return m_params.cov(4, 4); }  
 
        void setX0(float x0);          
        void setY0(float y0);          
        void setZ0(float z0);          
        void setPhi(float phi);        
        void setTheta(float theta);    
        void setQOverP(float qOverP);  
        void setAuthor(int author);    
 
        void setChi2(float chi2);  
        void setNdof(int ndof);    
 
        const Amg::Vector3D& position() const;
        const Amg::Vector3D& direction() const;
 
        void setCov00(float cov00) { m_params.cov(0, 0) = cov00; }  
        void setCov01(float cov01) { m_params.cov(0, 1) = cov01; }  
        void setCov02(float cov02) { m_params.cov(0, 2) = cov02; }  
        void setCov03(float cov03) { m_params.cov(0, 3) = cov03; }  
        void setCov04(float cov04) { m_params.cov(0, 4) = cov04; }  
        void setCov11(float cov11) { m_params.cov(1, 1) = cov11; }  
        void setCov12(float cov12) { m_params.cov(1, 2) = cov12; }  
        void setCov13(float cov13) { m_params.cov(1, 3) = cov13; }  
        void setCov14(float cov14) { m_params.cov(1, 4) = cov14; }  
        void setCov22(float cov22) { m_params.cov(2, 2) = cov22; }  
        void setCov23(float cov23) { m_params.cov(2, 3) = cov23; }  
        void setCov24(float cov24) { m_params.cov(2, 4) = cov24; }  
        void setCov33(float cov33) { m_params.cov(3, 3) = cov33; }  
        void setCov34(float cov34) { m_params.cov(3, 4) = cov34; }  
        void setCov44(float cov44) { m_params.cov(4, 4) = cov44; }  
 
        void addHit(const CalibHitPtr& hit);           
        void addHole(const CalibHolePtr& hole);        
        void addSegmentOnTrack(const CalibSegPtr& s);  
 
    private:
        void clear();
 
        HitVector m_hits;    
        HoleVector m_holes;  
        SegmentVector m_segments_on_track;
 
        defineParams m_params{};
        Amg::Vector3D m_pos{0., 0., 0.};
        Amg::Vector3D m_dir{0., 0., 1.};
    };
 
}  // namespace MuonCalib
 
#endif  // MUONCALIBTRACK_E_H