MuonSegmentCombinationCleanerTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONSEGMENTCOMBINATIONCLEANERTOOL_H
#define MUONSEGMENTCOMBINATIONCLEANERTOOL_H
 
#include <map>
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonRecHelperTools/IMuonEDMHelperSvc.h"
#include "MuonRecHelperTools/MuonEDMPrinterTool.h"
#include "MuonSegment/MuonSegmentCombinationCollection.h"
#include "MuonSegmentCombinerToolInterfaces/IMuonSegmentCombinationCleanerTool.h"
#include "MuonSegmentMakerToolInterfaces/IMuonSegmentOverlapRemovalTool.h"
#include "MuonSegmentMakerUtils/MuonSegmentCombiOverlapSummary.h"
#include "MuonSegmentMakerUtils/MuonSegmentCombiSummary.h"
 
namespace Muon {
 
    class MuonSegment;
 
    class MuonSegmentCombinationCleanerTool : virtual public IMuonSegmentCombinationCleanerTool, public AthAlgTool {
    public:
        MuonSegmentCombinationCleanerTool(const std::string&, const std::string&, const IInterface*);
 
        virtual ~MuonSegmentCombinationCleanerTool() = default;
 
        virtual StatusCode initialize() override;
 
        std::unique_ptr<MuonSegmentCombinationCollection> clean(const MuonSegmentCombinationCollection& combiCol,
                                                                MuonSegmentCombPatternCombAssociationMap* segPattMap) const override;
 
        std::unique_ptr<MuonSegmentCombination> removeDuplicateSegments(MuonSegmentCombination& combi) const;
 
    private:
        void cleanAndMergeCombis(const MuonSegmentCombinationCollection& combiCol,
                                 std::unique_ptr<MuonSegmentCombinationCollection>& combiCleanCol,
                                 MuonSegmentCombPatternCombAssociationMap* segPattMap) const;
 
        bool subSetOfPrevious(MuonSegmentCombiSummary& summary, std::vector<MuonSegmentCombiSummary>& summaries) const;
 
        MuonSegmentCombiOverlapSummary calculateOverlap(MuonSegmentCombiSummary& summary1, MuonSegmentCombiSummary& summary2) const;
 
        /* resolve overlap between two vectors of segments */
        void resolveLayerOverlap(const std::vector<MuonSegment*>& chamberVec1, const std::vector<MuonSegment*>& chamberVec2,
                                 MuonSegmentCombiOverlapSummary& summary) const;
 
        void fillSummary(MuonSegmentCombiSummary& combi) const;
 
        std::string print(MuonSegmentCombiSummary& summary) const;
        static std::string print(MuonSegmentCombiOverlapSummary& summary) ;
 
        PublicToolHandle<Muon::MuonEDMPrinterTool> m_printer{
            this,
            "Printer",
            "Muon::MuonEDMPrinterTool/MuonEDMPrinterTool",
        };
        ToolHandle<IMuonSegmentOverlapRemovalTool> m_overlapRemovalTool{
            this,
            "SegmentOverlapRemovalTool",
            "Muon::MuonSegmentOverlapRemovalTool/MuonSegmentOverlapRemovalTool",
            "tool to removal overlaps in segment combinations",
        };
 
        ServiceHandle<Muon::IMuonEDMHelperSvc> m_edmHelperSvc{
            this,
            "edmHelper",
            "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc",
            "Handle to the service providing the IMuonEDMHelperSvc interface",
        };
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{
            this,
            "MuonIdHelperSvc",
            "Muon::MuonIdHelperSvc/MuonIdHelperSvc",
        };
 
        Gaudi::Property<bool> m_mergeAllCombis{this, "MergeAllCombis", false, "merge all combinations into one large combination"};
    };
 
}  // namespace Muon
 
#endif  // MUONSEGMENTCOMBINATIONCLEANER_H