Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make the renaming of TrackerHit -> TrackerHit3D transparent #1219

Merged
merged 2 commits into from
Feb 8, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions DDDigi/io/DigiEdm4hepOutput.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@
#include <podio/Frame.h>
#include <edm4hep/SimTrackerHit.h>
#include <edm4hep/MCParticleCollection.h>
#include <edm4hep/TrackerHitCollection.h>
#include <edm4hep/EventHeaderCollection.h>
#include <edm4hep/CalorimeterHitCollection.h>
#include <edm4hep/CaloHitContributionCollection.h>
Expand Down Expand Up @@ -64,7 +63,7 @@ namespace dd4hep {
/// MC particle collection
particlecollection_t m_particles { };
/// Collection of all edm4hep tracker object collections
std::map<std::string, std::unique_ptr<edm4hep::TrackerHitCollection> > m_tracker_collections;
std::map<std::string, std::unique_ptr<edm4hep::TrackerHit3DCollection> > m_tracker_collections;
/// Collection of all edm4hep calorimeter object collections
std::map<std::string, std::unique_ptr<edm4hep::CalorimeterHitCollection> > m_calo_collections;
/// Output section name
Expand Down Expand Up @@ -122,7 +121,7 @@ namespace dd4hep {
m_particles = std::make_pair(nam, std::make_unique<edm4hep::MCParticleCollection>());
}
else if ( typ == "TrackerHits" ) {
m_tracker_collections.emplace(nam, std::make_unique<edm4hep::TrackerHitCollection>());
m_tracker_collections.emplace(nam, std::make_unique<edm4hep::TrackerHit3DCollection>());
}
else if ( typ == "CalorimeterHits" ) {
m_calo_collections.emplace(nam, std::make_unique<edm4hep::CalorimeterHitCollection>());
Expand Down Expand Up @@ -286,7 +285,7 @@ namespace dd4hep {
template <typename T> void
DigiEdm4hepOutputProcessor::convert_depos(const T& cont,
const predicate_t& predicate,
edm4hep::TrackerHitCollection* collection) const
edm4hep::TrackerHit3DCollection* collection) const
{
std::array<float,6> covMat = {0., 0., m_pointResoutionRPhi*m_pointResoutionRPhi,
0., 0., m_pointResoutionZ*m_pointResoutionZ
Expand Down Expand Up @@ -320,7 +319,7 @@ namespace dd4hep {
if ( !cont.empty() ) {
switch(cont.data_type) {
case SegmentEntry::TRACKER_HITS:
convert_depos(cont, predicate, static_cast<edm4hep::TrackerHitCollection*>(coll));
convert_depos(cont, predicate, static_cast<edm4hep::TrackerHit3DCollection*>(coll));
break;
case SegmentEntry::CALORIMETER_HITS:
convert_depos(cont, predicate, static_cast<edm4hep::CalorimeterHitCollection*>(coll));
Expand Down
13 changes: 10 additions & 3 deletions DDDigi/io/DigiEdm4hepOutput.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,17 @@
#include <DDDigi/DigiOutputAction.h>

/// C/C++ include files
#if __has_include("edm4hep/TrackerHitCollection.h")
#include <edm4hep/TrackerHitCollection.h>
namespace edm4hep {
using TrackerHit3DCollection = edm4hep::TrackerHitCollection;
}
#else
#include <edm4hep/TrackerHit3DCollection.h>
#endif

/// Forward declarations from edm4hep
namespace edm4hep {
class TrackerHitCollection;
class CalorimeterHitCollection;
}

Expand All @@ -35,7 +42,7 @@ namespace dd4hep {
* Supported output containers types are:
* - edm4hep::MCParticles aka "MCParticles"
* - edm4hep::CalorimeterHitCollection aka "CalorimeterHits"
* - edm4hep::TrackerHitCollection aka "TracketHits"
* - edm4hep::TrackerHit3DCollection aka "TracketHits"
*
* This entity actually is only the work dispatcher:
* It opens files and dumps data into
Expand Down Expand Up @@ -103,7 +110,7 @@ namespace dd4hep {

/// Convert tracker hits to edm4hep
template <typename T> void
convert_depos(const T& cont, const predicate_t& predicate, edm4hep::TrackerHitCollection* collection) const;
convert_depos(const T& cont, const predicate_t& predicate, edm4hep::TrackerHit3DCollection* collection) const;

/// Convert calorimeter hits to edm4hep
template <typename T> void
Expand Down
9 changes: 8 additions & 1 deletion DDDigi/io/DigiIO.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,14 @@
#include <edm4hep/SimTrackerHit.h>
#include <edm4hep/MCParticle.h>
#include <edm4hep/MCParticleCollection.h>
#if __has_include("edm4hep/TrackerHitCollection.h")
#include <edm4hep/TrackerHitCollection.h>
namespace edm4hep {
using TrackerHit3DCollection = edm4hep::TrackerHitCollection;
}
#else
#include <edm4hep/TrackerHit3DCollection.h>
#endif
#include <edm4hep/SimTrackerHitCollection.h>
#include <edm4hep/CalorimeterHitCollection.h>
#include <edm4hep/SimCalorimeterHitCollection.h>
Expand Down Expand Up @@ -224,7 +231,7 @@ namespace dd4hep {
template <> template <>
void data_io<edm4hep_input>::_to_edm4hep(const std::pair<const CellID, EnergyDeposit>& dep,
const std::array<float, 6>& covMat,
edm4hep::TrackerHitCollection& collection,
edm4hep::TrackerHit3DCollection& collection,
int hit_type)

{
Expand Down
Loading