ITrackDepositInCaloTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef CALOTRKMUIDTOOLS_ITRACKDEPOSITINCALOTOOL_H
#define CALOTRKMUIDTOOLS_ITRACKDEPOSITINCALOTOOL_H
 
#include "CaloDetDescr/CaloDetDescriptor.h"
#include "CaloEvent/CaloCell.h"
#include "CaloEvent/CaloCellContainer.h"
#include "GaudiKernel/IAlgTool.h"
#include "GaudiKernel/StatusCode.h"
#include "TrkCaloExtension/CaloExtensionCollection.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkTrack/Track.h"
#include "muonEvent/DepositInCalo.h"
#include "xAODTracking/TrackParticle.h"
 
class ITrackDepositInCaloTool : virtual public IAlgTool {
public:
    enum CaloSurfaceType { Entrance = 0, Middle = 1, Exit = 2, Inside = 3, Outside = 4 };
    virtual ~ITrackDepositInCaloTool() = default;
 
    static const InterfaceID& interfaceID() {
        static const InterfaceID IID_ITrackDepositInCaloTool("ITrackDepositInCaloTool", 1, 0);
        return IID_ITrackDepositInCaloTool;
    }

    virtual std::vector<DepositInCalo> deposits(const Trk::TrackParameters* par,
                                                const CaloCellContainer* cellContainer = nullptr) const = 0;
    virtual std::vector<DepositInCalo> getDeposits(const Trk::TrackParameters* par,
                                                   const CaloCellContainer* caloCellCont = nullptr) const = 0;
 
    virtual std::vector<DepositInCalo> getDeposits(const xAOD::TrackParticle* tp, const CaloCellContainer* caloCellCont = nullptr,
                                                   const CaloExtensionCollection* extensionCache = nullptr) const = 0;

    virtual Trk::Surface* createSurface(const CaloDetDescriptor* descr, CaloSurfaceType type) const = 0;
    virtual double calcEnergy(const Trk::TrackParameters* par, const Trk::ParticleHypothesis& particleHypo) const = 0;

    typedef std::map<double, std::vector<const CaloDetDescriptor*> > CaloLayerMap;
    typedef std::map<double, std::vector<const CaloDetDescriptor*> >::const_iterator CaloLayerMapIt;
};
 
#endif