CaloGPUOutput.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_CALOGPUOUTPUT_H
#define CALORECGPU_CALOGPUOUTPUT_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CaloRecGPU/CaloClusterGPUProcessor.h"
#include <string>
#include <mutex>
#include <atomic>

 
class CaloGPUOutput :
  public extends<AthAlgTool, CaloClusterGPUProcessor>
{
 public:
 
  CaloGPUOutput(const std::string & type, const std::string & name, const IInterface * parent);
 
  virtual StatusCode execute (const EventContext & ctx,
                              const CaloRecGPU::ConstantDataHolder & constant_data,
                              CaloRecGPU::EventDataHolder & event_data,
                              void * temporary_buffer) const override;
 
  virtual ~CaloGPUOutput() = default;
 
 private:
 

  Gaudi::Property<std::string> m_savePath{this, "SavePath", "./saved_clusters", "Path to where the files should be saved"};

  Gaudi::Property<std::string> m_filePrefix{this, "FilePrefix", "", "Prefix of the saved files"};

  Gaudi::Property<std::string> m_fileSuffix{this, "FileSuffix", "", "Suffix of the saved files"};

  Gaudi::Property<unsigned int> m_numWidth{this, "NumberWidth", 9, "The number of digits to reserve for the events"};

  Gaudi::Property<bool> m_sortedAndCutClusters {this, "UseSortedAndCutClusters", true, "Sort the clusters by transverse energy, apply a cut and ensure contiguous tags"};

  Gaudi::Property<bool> m_outputTags{this, "OutputCellsAsTags", true, "Whether to output cell assignment as tags instead of a list of indices per cluster."};
  
  Gaudi::Property<bool> m_onlyCellInfo {this, "OnlyOutputCellInfo", false, "Only output cell info"};
  
  mutable std::atomic<bool> m_constantDataSaved;

  mutable std::mutex m_mutex;
 
};
 
#endif //CALORECGPU_CALOGPUOUTPUT_H