BasicGPUToAthenaImporter.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++ -*-
//
 
//Dear emacs, this is -*-c++-*-
 
#ifndef CALORECGPU_BASICGPUTOATHENAIMPORTER_H
#define CALORECGPU_BASICGPUTOATHENAIMPORTER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "CaloRecGPU/CaloClusterGPUTransformers.h"
#include "StoreGate/ReadHandleKey.h"
#include "CaloRecGPU/CaloGPUTimed.h"
 
#include "AthCUDAInterfaces/IStreamTool.h"
 
class CaloCell_ID;

 
class BasicGPUToAthenaImporter :
  public extends<AthAlgTool, ICaloClusterGPUOutputTransformer>, public CaloGPUTimed
{
 public:
 
  BasicGPUToAthenaImporter(const std::string & type, const std::string & name, const IInterface * parent);
 
  virtual StatusCode initialize() override;
 
  virtual StatusCode convert (const EventContext & ctx, const CaloRecGPU::ConstantDataHolder & constant_data,
                              CaloRecGPU::EventDataHolder & event_data, xAOD::CaloClusterContainer * cluster_collection) const override;
 
  virtual StatusCode finalize() override;
 
  virtual ~BasicGPUToAthenaImporter() = default;
 
 private:

  Gaudi::Property<bool> m_useCPUPropertiesCalculation {this, "UseCPUClusterPropertiesCalculation", false, "Use CaloClusterKineHelper::calculateKine instead of GPU-calculated cluster properties"};

  SG::ReadHandleKey<CaloCellContainer> m_cellsKey {this, "CellsName", "", "Name(s) of Cell Containers"};

  Gaudi::Property<std::string> m_clusterSizeString {this, "ClusterSize", "Topo_420", "The size/type of the clusters"};
 
  xAOD::CaloCluster::ClusterSize m_clusterSize;

  Gaudi::Property<std::vector<int>> m_missingCellsToFill {this, "MissingCellsToFill", {}, "Force fill these cells as disabled on empty containers."};

  const CaloCell_ID * m_calo_id {nullptr};

  Gaudi::Property<bool> m_saveUncalibrated {this, "SaveUncalibratedSignalState", true, "Use CaloClusterKineHelper::calculateKine instead of GPU-calculated cluster properties"};

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