CaloCPUOutput.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_CALOCPUOUTPUT_H
#define CALORECGPU_CALOCPUOUTPUT_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CaloUtils/CaloClusterCollectionProcessor.h"
#include <string>
 
#include "StoreGate/ReadCondHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
 
class CaloCell_ID;

 
class CaloCPUOutput :
  public extends<AthAlgTool, CaloClusterCollectionProcessor>
{
 public:
 
  CaloCPUOutput(const std::string & type, const std::string & name, const IInterface * parent);
  
  using CaloClusterCollectionProcessor::execute;
  
  virtual StatusCode initialize() override;
  
  virtual StatusCode execute (const EventContext& ctx, xAOD::CaloClusterContainer* cluster_collection) const override;
 
  virtual ~CaloCPUOutput() = 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_saveCellInfo{this, "AlsoOutputCellInfo", false,
                                       "Whether to output cell energies, gains, times, etc., as this information is repeated in the GPU side too."};
  
  Gaudi::Property<bool> m_outputTags{this, "OutputCellsAsTags", true, "Whether to output cell assignment as tags instead of a list of indices per cluster."};
  
  SG::ReadHandleKey<CaloCellContainer> m_cellsKey {this, "CellsName", "", "Name(s) of Cell Containers"};
  
  const CaloCell_ID* m_calo_id {nullptr};
 
};
 
#endif //CALORECGPU_CALOCPUOUTPUT_H