SegmentRefitTest.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRACKFINDINGTEST_SEGMENTREFITTEST_H
#define MUONTRACKFINDINGTEST_SEGMENTREFITTEST_H
 
#include "AthenaBaseComps/AthHistogramAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "xAODMuon/MuonSegmentContainer.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonTesterTree/MuonTesterTree.h"
 
namespace MuonValR4{
    class SegmentRefitTest: public AthHistogramAlgorithm {
        public:
            using AthHistogramAlgorithm::AthHistogramAlgorithm;
            virtual StatusCode initialize() override final;
            virtual StatusCode execute() override final;
            virtual StatusCode finalize() override final;
 
        private:
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_preFitKey{this, "PreFitContainer", "MuonSegmentsFromR4"};
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_postFitKey{this, "PostFitContainer", "ActsRefitSegments"};
            SG::ReadDecorHandleKey<xAOD::MuonSegmentContainer> m_linkKey{this, "Link", m_postFitKey, "prefitSegmentLink"};
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "IdHelperSvc",  "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};

            MuonVal::MuonTesterTree m_tree{"SegmentRefitTree", "SegmentRefitTest"};
            MuonVal::ScalarBranch<int>& m_chamberIndex{m_tree.newScalar<int>("chamberIndex")};
            MuonVal::ScalarBranch<short>& m_stationSide{m_tree.newScalar<short>("stationSide")};
            MuonVal::ScalarBranch<int>& m_stationPhi{m_tree.newScalar<int>("stationPhi")};
            MuonVal::ScalarBranch<float>& m_preFitLocX{m_tree.newScalar<float>("preFitLocX")};
            MuonVal::ScalarBranch<float>& m_preFitLocY{m_tree.newScalar<float>("preFitLocY")}; 
            MuonVal::ScalarBranch<float>& m_preFitTheta{m_tree.newScalar<float>("preFitTheta")};
            MuonVal::ScalarBranch<float>& m_preFitPhi{m_tree.newScalar<float>("preFitPhi")};
            MuonVal::ScalarBranch<float>& m_uncertLocX{m_tree.newScalar<float>("uncertLocX")};
            MuonVal::ScalarBranch<float>& m_uncertLocY{m_tree.newScalar<float>("uncertLocY")};
            MuonVal::ScalarBranch<float>& m_uncertTheta{m_tree.newScalar<float>("uncertTheta")};
            MuonVal::ScalarBranch<float>& m_uncertPhi{m_tree.newScalar<float>("uncertPhi")};

            MuonVal::ScalarBranch<float>& m_preFitChi2{m_tree.newScalar<float>("preFitChi2")};
            MuonVal::ScalarBranch<unsigned short>& m_preFitNdoF{m_tree.newScalar<unsigned short>("preFitNdoF")};
            MuonVal::ScalarBranch<unsigned short>& m_preFitNPrecHits{m_tree.newScalar<unsigned short>("preFitNPrecHits")};
            MuonVal::ScalarBranch<unsigned short>& m_preFitNTrigEtaHits{m_tree.newScalar<unsigned short>("preFitNTrigEtaHits")};
            MuonVal::ScalarBranch<unsigned short>& m_preFitNTrigPhiHits{m_tree.newScalar<unsigned short>("preFitNTrigPhiHits")};
            MuonVal::ScalarBranch<unsigned char>& m_goodFit{m_tree.newScalar<unsigned char>("goodFit", false)};
            MuonVal::ScalarBranch<float>& m_seedFitLocX{m_tree.newScalar<float>("seedFitLocX", 0.f)};
            MuonVal::ScalarBranch<float>& m_seedFitLocY{m_tree.newScalar<float>("seedFitLocY", 0.f)}; 
            MuonVal::ScalarBranch<float>& m_seedFitTheta{m_tree.newScalar<float>("seedFitTheta", 0.f)};
            MuonVal::ScalarBranch<float>& m_seedFitPhi{m_tree.newScalar<float>("seedFitPhi", 0.f)};
            MuonVal::ScalarBranch<float>& m_postFitLocX{m_tree.newScalar<float>("postFitLocX", 0.f)};
            MuonVal::ScalarBranch<float>& m_postFitLocY{m_tree.newScalar<float>("postFitLocY", 0.f)}; 
            MuonVal::ScalarBranch<float>& m_postFitTheta{m_tree.newScalar<float>("postFitTheta", 0.f)};
            MuonVal::ScalarBranch<float>& m_postFitPhi{m_tree.newScalar<float>("postFitPhi", 0.f)};
            MuonVal::ScalarBranch<float>& m_postFitChi2{m_tree.newScalar<float>("postFitChi2", 0.f)};
            MuonVal::ScalarBranch<unsigned short>& m_postFitNdoF{m_tree.newScalar<unsigned short>("postFitNdoF", 0)};
            MuonVal::ScalarBranch<unsigned short>& m_postFitNPrecHits{m_tree.newScalar<unsigned short>("postFitNPrecHits", 0)};
            MuonVal::ScalarBranch<unsigned short>& m_postFitNTrigEtaHits{m_tree.newScalar<unsigned short>("postFitNTrigEtaHits", 0)};
            MuonVal::ScalarBranch<unsigned short>& m_postFitNTrigPhiHits{m_tree.newScalar<unsigned short>("postFitNTrigPhiHits", 0)};
    };
}
 
#endif