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, 
                         std::vector<const CaloCell*>& cells) const;

  StatusCode selectSeedCells(const xAOD::TauJet& tau, 
                     const CaloCellContainer& cellContainer,
                             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};
  Gaudi::Property<float> m_dRThreshold{this, "DRThreshold", 0.4, "dR(cell,tau) to select cells"};
  Gaudi::Property<float> m_energyThreshold{this, "EnergyThreshold", 100., "energy threshold (in MeV) to select cells"};
   
  SG::ReadHandleKey<CaloCellContainer> m_caloCellInputContainer{this,"Key_caloCellInputContainer", "AllCalo", "input calo cell container key"};
  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