BasicGPUClusterInfoCalculator.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
//
// Dear emacs, this is -*- c++ -*-
//
 
 
#ifndef CALORECGPU_BASICGPUCLUSTERINFOCALCULATOR_H
#define CALORECGPU_BASICGPUCLUSTERINFOCALCULATOR_H
 
#include "CxxUtils/checker_macros.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CaloRecGPU/CaloClusterGPUProcessor.h"
#include "CaloRecGPU/CaloGPUTimed.h"
#include "BasicGPUClusterInfoCalculatorImpl.h"
#include "CaloRecGPU/CaloGPUCUDAInitialization.h"
 
#include "GaudiKernel/ServiceHandle.h"
 
#include "CaloRecGPU/IGPUKernelSizeOptimizerSvc.h"
 
#include "CLHEP/Units/SystemOfUnits.h"
 
class BasicGPUClusterInfoCalculator:
  public AthAlgTool, virtual public CaloClusterGPUProcessor, public CaloGPUTimed, public CaloGPUCUDAInitialization
{
 public:
 
  BasicGPUClusterInfoCalculator(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 ~BasicGPUClusterInfoCalculator() = default;
 
  virtual size_t size_of_temporaries() const override
  {
    if (m_preserveClusterMoments)
      {
        return sizeof(BasicClusterInfoCalculator::ClusterInfoCalculatorTemporaries);
      }
    else
      {
        return 0;
      }
  };
  
 private:
 
  Gaudi::Property<bool> m_preserveClusterMoments {this, "PreserveClusterMoments", false, "Do not use cluster moments array as temporary storage"};
 
  Gaudi::Property<bool> m_cutClustersInAbsE {this, "ClusterCutsInAbsEt", true, "Do cluster cuts in Abs Et instead of Et"};
 
  Gaudi::Property<float> m_clusterETThreshold {this, "ClusterEtorAbsEtCut", 0.*CLHEP::MeV, "Cluster E_t or Abs E_t cut"};
 
  ServiceHandle<IGPUKernelSizeOptimizerSvc> m_kernelSizeOptimizer { this, "KernelSizeOptimizer", "GPUKernelSizeOptimizerSvc", "CUDA kernel size optimization service." };
 
};
 
#endif //CALORECGPU_TOPOAUTOMATONCLUSTERING_H