MuonPhiHitSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MuonSegmentCleaner_MuonPhiHitSelector_H
#define MuonSegmentCleaner_MuonPhiHitSelector_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonRecToolInterfaces/IMuonClusterOnTrackCreator.h"
#include "MuonRecToolInterfaces/IMuonCompetingClustersOnTrackCreator.h"
#include "MuonRecToolInterfaces/IMuonHitSelector.h"
 
namespace Trk {
class PrepRawData;
class RIO_OnTrack;
}  // namespace Trk
 
class MuonPhiHitSelector : public AthAlgTool, virtual public Muon::IMuonHitSelector {
  public:
    MuonPhiHitSelector(const std::string&, const std::string&, const IInterface*);
    virtual ~MuonPhiHitSelector() = default;
 
    StatusCode initialize() override;

    std::vector<std::unique_ptr<const Trk::MeasurementBase>> select_rio(
        const double pmom, const std::vector<const Trk::RIO_OnTrack*>& associatedHits,
        const std::vector<const Trk::PrepRawData*>& unassociatedHits) const override;
 
  private:
    ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{
        this,
        "MuonIdHelperSvc",
        "Muon::MuonIdHelperSvc/MuonIdHelperSvc",
    };

    ToolHandle<Muon::IMuonCompetingClustersOnTrackCreator> m_competingRIOsOnTrackTool{
        this,
        "MuonCompetingClustersOnTrackCreator",
        "Muon::MuonCompetingClustersOnTrackCreator/MuonCompetingClustersOnTrackCreator",
    };

    ToolHandle<Muon::IMuonClusterOnTrackCreator> m_clusterCreator{this, "MuonClusterOnTrackCreator",""};
    
    ToolHandle<Muon::IMuonClusterOnTrackCreator> m_cscRotCreator{this, "CscRotCreator",""};

    Gaudi::Property<bool> m_summary{this, "DoSummary", false};
    Gaudi::Property<bool> m_cosmics{this, "DoCosmics", false};
    Gaudi::Property<bool> m_makeClusters{this, "MakeClusters", false};
    Gaudi::Property<bool> m_competingRios{this, "CompetingRios", false};

    void fitRecPhi(const double pmom, const std::vector<Identifier>& phiId, const std::vector<double>& phiHitx,
                   const std::vector<double>& phiHity, const std::vector<double>& phiHitz,
                   const std::vector<double>& phiError, std::vector<int>& quality, const int nphi,
                   std::vector<double>& phiPull, std::vector<int>& phiMult, std::vector<int>& phiSelect, double& chi2,
                   double& r0, double& phi, std::vector<double>& errorM, int& nfit) const;

    void fitPhiSL(const double pmom, const std::vector<Identifier>& id, const std::vector<double>& hitx,
                  const std::vector<double>& hity, const std::vector<double>& hitz, const std::vector<double>& error,
                  std::vector<int>& select, const int n, std::vector<double>& pull, int& imax, double& chi2, double& r0,
                  double& phi, std::vector<double>& errorM, bool fast) const;

 
    void clusterPhi(const std::vector<Identifier>& id, const std::vector<double>& hitx, const std::vector<double>& hity,
                    const std::vector<double>& hitz, const std::vector<double>& error, const std::vector<double>& pull,
                    std::vector<int>& select, const int n, std::vector<double>& clusterX, std::vector<double>& clusterY,
                    std::vector<double>& clusterZ, std::vector<double>& clusterError,
                    std::vector<Identifier>& clusterId, std::vector<int>& clusterHits, std::vector<int>& clusterSelect,
                    std::vector<int>& clusterInt, int& ncl) const;
};
 
#endif  // MuonSegmentCleaner_MuonPhiHitSelector_H