TauShotFinder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUSHOTFINDER_H
#define TAURECTOOLS_TAUSHOTFINDER_H
 
#include "tauRecTools/TauRecToolBase.h"
 
#include "xAODPFlow/PFOAuxContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "CaloInterface/IHadronicCalibrationTool.h"
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadCondHandleKey.h"
 
class CaloCell_ID;
 
class TauShotFinder : public TauRecToolBase {
 
public:
  
  ASG_TOOL_CLASS2(TauShotFinder, TauRecToolBase, ITauToolBase);
 
  TauShotFinder(const std::string& name);
  virtual ~TauShotFinder() = default;
 
  virtual StatusCode initialize() override;
  virtual StatusCode executeShotFinder(xAOD::TauJet& pTau, xAOD::CaloClusterContainer& tauShotCaloClusContainer, xAOD::PFOContainer& tauShotPFOContainer) const override;
 
private:
 
  struct ptSort
  { 
    ptSort( const TauShotFinder& info );
    const TauShotFinder& m_info;
    bool operator()( const CaloCell* c1, const CaloCell* c2 );
  };
 
  StatusCode selectCells(const xAOD::TauJet& tau, const CaloCellContainer& cellContainer,
                                           const CaloDetDescrManager* detMgr, 
                                           std::vector<const CaloCell*>& cells) const;
 
  StatusCode selectSeedCells(const xAOD::TauJet& tau, const CaloCellContainer& cellContainer,
                                               const CaloDetDescrManager* detMgr,
                                                std::vector<const CaloCell*>& seedCells) const;
 
  bool isPhiNeighbour(IdentifierHash cell1Hash, IdentifierHash cell2Hash) const;
 
  const CaloCell* getPhiNeighbour(const CaloCell& seedCell, const std::vector<const CaloCell*>& seedCells) const;
  
  std::vector<const CaloCell*> getEtaNeighbours(const CaloCell& cell, const CaloCellContainer& cellContainer, int maxDepth) const;
 
  void addEtaNeighbours(const CaloCell& cell,
                        const CaloCellContainer& cellContainer,
                        std::vector<const CaloCell*>& cells,
                        int depth,
                        int maxDepth,
                        bool next) const;
 
  xAOD::CaloCluster* createShotCluster(const CaloCell* cell,
                       const CaloCell* phiNeighCell,
                       const CaloCellContainer& cellContainer,
                       xAOD::CaloClusterContainer* clusterContainer) const;
 
  int getEtaBin(float eta) const;
 
  int getNPhotons(float eta, float energy) const;
 
  Gaudi::Property<int> m_nCellsInEta {this, "NCellsInEta"};
  Gaudi::Property<std::vector<float>> m_minPtCut {this, "MinPtCut"};
  Gaudi::Property<std::vector<float>> m_doubleShotCut {this, "AutoDoubleShotCut"};
  Gaudi::Property<bool> m_removeElectronCells {this, "RemoveElectronCells", false};
 
  SG::ReadHandleKey<CaloCellContainer> m_caloCellInputContainer{this,"Key_caloCellInputContainer", "AllCalo", "input vertex container key"};
  SG::ReadCondHandleKey<CaloDetDescrManager> m_caloMgrKey{this,"CaloDetDescrManager", "CaloDetDescrManager"};
  ToolHandle<IHadronicCalibrationTool> m_caloWeightTool {this, "CaloWeightTool", "H1WeightToolCSC12Generic"};
  SG::ReadHandleKey<xAOD::CaloClusterContainer> m_removedClusterInputContainer {this,"Key_RemovedClusterInputContainer", "", "input removed cluster key"};
  
  const CaloCell_ID* m_calo_id = nullptr;
 
};
 
#endif  // TAURECTOOLS_TAUSHOTFINDER_H