IMuonCombinedInDetExtensionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef IRECMUONCOMBINEDINDETEXTENSIONTOOL_H
#define IRECMUONCOMBINEDINDETEXTENSIONTOOL_H
 
#include "GaudiKernel/IAlgTool.h"
#include "MuonCombinedEvent/InDetCandidateCollection.h"
#include "MuonCombinedEvent/InDetCandidateToTagMap.h"
#include "MuonPrepRawData/MMPrepDataContainer.h"
#include "MuonPrepRawData/MuonPrepDataContainer.h"
#include "MuonPrepRawData/sTgcPrepDataContainer.h"
#include "TrkSegment/SegmentCollection.h"
#include "TrkTrack/TrackCollection.h"
 
namespace MuonCombined {

 
    class IMuonCombinedInDetExtensionTool : virtual public IAlgTool {
    public:
        struct MuonPrdData {
            MuonPrdData() : mdtPrds(nullptr), rpcPrds(nullptr), tgcPrds(nullptr), cscPrds(nullptr), stgcPrds(nullptr), mmPrds(nullptr) {}
            const Muon::MdtPrepDataContainer* mdtPrds;
            const Muon::RpcPrepDataContainer* rpcPrds;
            const Muon::TgcPrepDataContainer* tgcPrds;
            const Muon::CscPrepDataContainer* cscPrds;
            const Muon::sTgcPrepDataContainer* stgcPrds;
            const Muon::MMPrepDataContainer* mmPrds;
        };
 
        static const InterfaceID& interfaceID() {
            static const InterfaceID IID_IMuonCombinedInDetExtensionTool("MuonCombined::IMuonCombinedInDetExtensionTool", 1, 0);
            return IID_IMuonCombinedInDetExtensionTool;
        }

        virtual void extend(const InDetCandidateCollection& inDetCandidates, InDetCandidateToTagMap* tagMap, TrackCollection* combTracks,
                            TrackCollection* meTracks, Trk::SegmentCollection* segments, const EventContext& ctx) const = 0;
 
        /*New interface including PRDs for MuGirl segment-finding*/
        virtual void extendWithPRDs(const InDetCandidateCollection& inDetCandidates, InDetCandidateToTagMap* tagMap,
                                    IMuonCombinedInDetExtensionTool::MuonPrdData prdData, TrackCollection* combTracks,
                                    TrackCollection* meTracks, Trk::SegmentCollection* segments, const EventContext& ctx) const = 0;
 
        virtual ~IMuonCombinedInDetExtensionTool() = default;
    };
 
}  // namespace MuonCombined
 
#endif