RedoSegmentation.h

Go to the documentation of this file.

#ifndef DETCOMPONENTS_REDOSEGMENTATION_H
#define DETCOMPONENTS_REDOSEGMENTATION_H
 
// GAUDI
#include "Gaudi/Algorithm.h"
#include "GaudiKernel/ToolHandle.h"
 
// k4FWCore
#include "k4FWCore/DataHandle.h"
#include "k4FWCore/MetaDataHandle.h"
#include "k4Interface/IGeoSvc.h"
 
// DD4hep
#include "DD4hep/Readout.h"
#include "DD4hep/Segmentations.h"
 
// EDM4hep
#include "edm4hep/CalorimeterHitCollection.h"
#include "edm4hep/Constants.h"
#include "edm4hep/SimCalorimeterHitCollection.h"
 
class RedoSegmentation : public Gaudi::Algorithm {
public:
  explicit RedoSegmentation(const std::string&, ISvcLocator*);
  virtual ~RedoSegmentation();
  virtual StatusCode initialize() final;
  virtual StatusCode execute(const EventContext&) const final;
  virtual StatusCode finalize() final;
 
private:
  uint64_t volumeID(uint64_t aCellId) const;
  ServiceHandle<IGeoSvc> m_geoSvc;
  mutable DataHandle<edm4hep::CalorimeterHitCollection> m_inHits{"hits/caloInHits", Gaudi::DataHandle::Reader, this};
  mutable DataHandle<edm4hep::SimCalorimeterHitCollection> m_outHits{"hits/caloOutHits", Gaudi::DataHandle::Writer,
                                                                     this};
  MetaDataHandle<std::string> m_outHitsCellIDEncoding{m_outHits, edm4hep::labels::CellIDEncoding,
                                                      Gaudi::DataHandle::Writer};
  dd4hep::DDSegmentation::Segmentation* m_segmentation;
  int m_segmentationType; // use enum instead? defined in some namespace?
  Gaudi::Property<std::string> m_oldReadoutName{this, "oldReadoutName", "",
                                                "Name of the detector readout used in simulation"};
  dd4hep::DDSegmentation::Segmentation* m_oldSegmentation;
  int m_oldSegmentationType; // use enum instead? defined in some namespace?
  Gaudi::Property<std::string> m_newReadoutName{this, "newReadoutName", "", "Name of the new detector readout"};
  dd4hep::DDSegmentation::BitFieldCoder* m_oldDecoder;
  Gaudi::Property<std::vector<std::string>> m_oldIdentifiers{
      this, "oldSegmentationIds", {}, "Segmentation fields that are going to be replaced by the new segmentation"};
  std::vector<std::string> m_detectorIdentifiers;
  Gaudi::Property<uint> m_debugPrint{this, "debugPrint", 10, "Limit of debug printing"};
};
#endif /* DETCOMPONENTS_REDOSEGMENTATION_H */