GPUClusterInfoAndMomentsCalculator.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
//
// Dear emacs, this is -*- c++ -*-
//
 
#ifndef CALORECGPU_GPUCLUSTERINFOANDMOMENTSCALCULATOR_H
#define CALORECGPU_GPUCLUSTERINFOANDMOMENTSCALCULATOR_H
 
#include "CxxUtils/checker_macros.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CaloRecGPU/CaloClusterGPUProcessor.h"
#include "CaloRecGPU/CaloGPUTimed.h"
#include "GPUClusterInfoAndMomentsCalculatorImpl.h"
 
#include "AthCUDAInterfaces/IStreamTool.h"
 
#include "CaloRecGPU/IGPUKernelSizeOptimizerSvc.h"
 
#include "GaudiKernel/ServiceHandle.h"
 
#include "CaloRecGPU/CaloGPUCUDAInitialization.h"
 
#include "CLHEP/Units/SystemOfUnits.h"

 
 
class GPUClusterInfoAndMomentsCalculator:
  public extends<AthAlgTool, CaloClusterGPUProcessor>, public CaloGPUTimed, public CaloGPUCUDAInitialization
{
 public:
 
  GPUClusterInfoAndMomentsCalculator(const std::string & type, const std::string & name, const IInterface * parent);
 
  virtual StatusCode initialize() override
  {
    return CaloGPUCUDAInitialization::initialize();
  }
 
  virtual StatusCode initialize_non_CUDA() override;
 
  virtual StatusCode initialize_CUDA() override;
 
  virtual StatusCode execute (const EventContext & ctx,
                              const CaloRecGPU::ConstantDataHolder & constant_data,
                              CaloRecGPU::EventDataHolder & event_data,
                              void * temporary_buffer) const override;
 
  virtual StatusCode finalize() override;
 
  virtual ~GPUClusterInfoAndMomentsCalculator() = default;
 
 private:
 

  Gaudi::Property<double> m_maxAxisAngle {this, "MaxAxisAngle", 20 * CLHEP::deg, "The maximal allowed deviation from the IP-to-ClusterCenter-axis"};

  Gaudi::Property<double> m_minRLateral {this, "MinRLateral", 4 * CLHEP::cm, "The minimal r in the definition of the Lateral moment"};

  Gaudi::Property<double> m_minLLongitudinal {this, "MinLLongitudinal", 10 * CLHEP::cm, "The minimal lambda in the definition of the Longitudinal moment"};

  Gaudi::Property<double> m_minBadLArQuality {this, "MinBadLArQuality", 4000, "The minimal cell quality in the LAr for declaring a cell bad"};

   Gaudi::Property<bool> m_absOpt {this, "WeightingOfNegClusters", false, "If set to true use abs E value of cells to calculate cluster moments"};

   Gaudi::Property<double> m_etaInnerWheel {this, "EMECAbsEtaWheelTransition", 2.52, "Transition from outer to inner wheel in EME2"};

   Gaudi::Property<bool> m_twoGaussianNoise {this, "TwoGaussianNoise", false, "If set to true use 2-gaussian noise description for TileCal"};

  Gaudi::Property<bool> m_skipInvalidClusters {this, "SkipInvalidClusters", true, "Check for and skip invalid clusters during moments calculation."};

  ClusterMomentsCalculator::CMCOptionsHolder m_options;

  ServiceHandle<IGPUKernelSizeOptimizerSvc> m_kernelSizeOptimizer { this, "KernelSizeOptimizer", "GPUKernelSizeOptimizerSvc", "CUDA kernel size optimization service." };

  ToolHandle<AthCUDA::IStreamTool> m_streamTool {this, "StreamTool", "", "CUDA stream tool."};
};
 
#endif //CALORECGPU_TOPOAUTOMATONCLUSTERING_H