EnergyInCaloLayers.h

Go to the documentation of this file.

#ifndef DETSTUDIES_ENERGYINCALOLAYERS_H
#define DETSTUDIES_ENERGYINCALOLAYERS_H
 
// GAUDI
#include "Gaudi/Algorithm.h"
 
// Key4HEP
#include "k4FWCore/DataHandle.h"
#include "k4Interface/IGeoSvc.h"
 
// EDM4hep & Podio
#include "edm4hep/CalorimeterHitCollection.h"
#include "edm4hep/MCParticleCollection.h"
#include "podio/UserDataCollection.h"
 
class EnergyInCaloLayers : public Gaudi::Algorithm {
public:
  explicit EnergyInCaloLayers(const std::string&, ISvcLocator*);
  virtual ~EnergyInCaloLayers();
  virtual StatusCode initialize() final;
  virtual StatusCode execute(const EventContext&) const final;
  virtual StatusCode finalize() final;
 
private:
  mutable k4FWCore::DataHandle<edm4hep::CalorimeterHitCollection> m_deposits{"det/caloDeposits",
                                                                             Gaudi::DataHandle::Reader, this};
  mutable k4FWCore::DataHandle<edm4hep::MCParticleCollection> m_particle{"det/particles", Gaudi::DataHandle::Reader,
                                                                         this};
  mutable k4FWCore::DataHandle<podio::UserDataCollection<double>> m_energyInLayer{"energyInLayer",
                                                                                  Gaudi::DataHandle::Writer, this};
  mutable k4FWCore::DataHandle<podio::UserDataCollection<double>> m_energyInCryo{"energyInCryo",
                                                                                 Gaudi::DataHandle::Writer, this};
  mutable k4FWCore::DataHandle<podio::UserDataCollection<double>> m_particleVec{"particleVec",
                                                                                Gaudi::DataHandle::Writer, this};
 
  ServiceHandle<IGeoSvc> m_geoSvc;
 
  Gaudi::Property<size_t> m_numLayers{this, "numLayers", 11, "Number of layers"};
  Gaudi::Property<std::vector<double>> m_samplingFractions{
      this, "samplingFractions", {}, "Values of sampling fraction per layer"};
  Gaudi::Property<std::string> m_readoutName{this, "readoutName", "", "Name of the readout"};
};
 
#endif /* DETSTUDIES_ENERGYINCALOLAYERS_H */