Skip to content

Commit

Permalink
Adapt to new (v00-15) podio generated class names
Browse files Browse the repository at this point in the history
- Replace some dedicated types with  and rename others (Cluster -> MutableCluster, ConstCluster -> Cluster) to adapt to the new naming scheme introduced in AIDASoft/podio#205.
  • Loading branch information
vvolkl committed Feb 2, 2022
1 parent 3af7e26 commit d481b1e
Show file tree
Hide file tree
Showing 12 changed files with 15 additions and 19 deletions.
2 changes: 1 addition & 1 deletion Detector/DetComponents/src/MergeCells.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ StatusCode MergeCells::execute() {
uint debugIter = 0;

for (const auto& hit : *inHits) {
edm4hep::CalorimeterHit newHit = outHits->create();
auto newHit = outHits->create();
newHit.setEnergy(hit.getEnergy());
newHit.setEnergyError(hit.getEnergyError());
newHit.setPosition(hit.getPosition());
Expand Down
2 changes: 1 addition & 1 deletion Detector/DetComponents/src/MergeLayers.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ StatusCode MergeLayers::execute() {
unsigned int debugIter = 0;

for (const auto& hit : *inHits) {
edm4hep::CalorimeterHit newHit = outHits->create();
auto newHit = outHits->create();
newHit.setEnergy(hit.getEnergy());
newHit.setEnergyError(hit.getEnergyError());
newHit.setPosition(hit.getPosition());
Expand Down
2 changes: 1 addition & 1 deletion Detector/DetComponents/src/RedoSegmentation.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ StatusCode RedoSegmentation::execute() {
dd4hep::DDSegmentation::CellID oldid = 0;
uint debugIter = 0;
for (const auto& hit : *inHits) {
edm4hep::SimCalorimeterHit newHit = outHits->create();
auto newHit = outHits->create();
newHit.setEnergy(hit.getEnergy());
// SimCalorimeterHit type (needed for createCaloCells which runs after RedoSegmentation) has no time member
//newHit.setTime(hit.getTime());
Expand Down
2 changes: 1 addition & 1 deletion Detector/DetComponents/src/RewriteBitfield.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ StatusCode RewriteBitfield::execute() {
uint64_t oldid = 0;
uint debugIter = 0;
for (const auto& hit : *inHits) {
edm4hep::CalorimeterHit newHit = outHits->create();
auto newHit = outHits->create();
newHit.setEnergy(hit.getEnergy());
newHit.setTime(hit.getTime());
dd4hep::DDSegmentation::CellID cID = hit.getCellID();
Expand Down
5 changes: 2 additions & 3 deletions SimG4Common/include/SimG4Common/ParticleInformation.h
Original file line number Diff line number Diff line change
Expand Up @@ -29,15 +29,14 @@ class ParticleInformation : public G4VUserPrimaryParticleInformation {
* @param[in] aMCpart EDM MCParticle.
*/
explicit ParticleInformation(const edm4hep::MCParticle& aMCpart);
explicit ParticleInformation(edm4hep::ConstMCParticle& aMCpart);
/// A destructor
virtual ~ParticleInformation();
/// A printing method
virtual void Print() const final;
/** Getter of the MCParticle.
* @returns EDM MCParticle.
*/
edm4hep::ConstMCParticle& mcParticle();
edm4hep::MCParticle& mcParticle();
/** Setter of the end-of-tracking momentum (used for fast simulation).
* @param[in] aMom Particle momentum.
*/
Expand Down Expand Up @@ -73,7 +72,7 @@ class ParticleInformation : public G4VUserPrimaryParticleInformation {

private:
/// EDM MC particle
edm4hep::ConstMCParticle m_mcParticle;
edm4hep::MCParticle m_mcParticle;
/// Particle momentum at the end of tracking (filled for fast-sim)
CLHEP::Hep3Vector m_endMomentum;
/// Particle vertex position saved at the end of tracking (filled for fast-sim)
Expand Down
4 changes: 1 addition & 3 deletions SimG4Common/src/ParticleInformation.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,11 @@
namespace sim {
ParticleInformation::ParticleInformation(const edm4hep::MCParticle& aMCpart) : m_mcParticle(aMCpart), m_smeared(false) {}

ParticleInformation::ParticleInformation(edm4hep::ConstMCParticle& aMCpart) : m_smeared(false) {
m_mcParticle = aMCpart.clone();}
ParticleInformation::~ParticleInformation() {}

void ParticleInformation::Print() const {}

edm4hep::ConstMCParticle& ParticleInformation::mcParticle() { return m_mcParticle; }
edm4hep::MCParticle& ParticleInformation::mcParticle() { return m_mcParticle; }
void ParticleInformation::setEndMomentum(const CLHEP::Hep3Vector& aMom) { m_endMomentum = aMom; }
const CLHEP::Hep3Vector& ParticleInformation::endMomentum() const { return m_endMomentum; }
void ParticleInformation::setVertexPosition(const CLHEP::Hep3Vector& aPos) { m_vertexPosition = aPos; }
Expand Down
2 changes: 1 addition & 1 deletion SimG4Components/src/SimG4SaveCalHits.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ StatusCode SimG4SaveCalHits::saveOutput(const G4Event& aEvent) {
<< endmsg;
for (size_t iter_hit = 0; iter_hit < n_hit; iter_hit++) {
hit = dynamic_cast<k4::Geant4CaloHit*>(collect->GetHit(iter_hit));
edm4hep::SimCalorimeterHit edmHit = edmHits->create();
auto edmHit = edmHits->create();
edmHit.setCellID(hit->cellID);
//todo
//edmHitCore.bits = hit->trackId;
Expand Down
6 changes: 3 additions & 3 deletions SimG4Components/src/SimG4SaveSmearedParticles.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -40,9 +40,9 @@ StatusCode SimG4SaveSmearedParticles::saveOutput(const G4Event& aEvent) {
const G4PrimaryParticle* g4particle = aEvent.GetPrimaryVertex(i)->GetPrimary(j);
sim::ParticleInformation* info = dynamic_cast<sim::ParticleInformation*>(g4particle->GetUserInformation());
if (info->smeared()) {
edm4hep::ConstMCParticle& MCparticle = info->mcParticle();
edm4hep::ReconstructedParticle particle = particles->create();
edm4hep::MCRecoParticleAssociation association = associations->create();
auto & MCparticle = info->mcParticle();
auto particle = particles->create();
auto association = associations->create();
association.setRec(particle);
association.setSim(MCparticle);
particle.setCharge(g4particle->GetCharge());
Expand Down
2 changes: 1 addition & 1 deletion SimG4Components/src/SimG4SaveTrackerHits.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ StatusCode SimG4SaveTrackerHits::saveOutput(const G4Event& aEvent) {
<< collect->GetName() << endmsg;
for (size_t iter_hit = 0; iter_hit < n_hit; iter_hit++) {
hit = dynamic_cast<k4::Geant4PreDigiTrackHit*>(collect->GetHit(iter_hit));
edm4hep::SimTrackerHit edmHit = edmHits->create();
auto edmHit = edmHits->create();
edmHit.setCellID(hit->cellID);
edmHit.setEDep(hit->energyDeposit * sim::g42edm::energy);
/// workaround, store trackid in an unrelated field
Expand Down
2 changes: 1 addition & 1 deletion SimG4Components/src/SimG4SaveTrajectory.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ StatusCode SimG4SaveTrajectory::saveOutput(const G4Event& aEvent) {
G4VTrajectory* theTrajectory = (*trajectoryContainer)[trajectoryIndex];
for (int pointIndex = 0; pointIndex < theTrajectory->GetPointEntries(); ++pointIndex) {
auto trajectoryPoint = theTrajectory->GetPoint(pointIndex)->GetPosition();
edm4hep::TrackerHit edmHit = edmPositions->create();
auto edmHit = edmPositions->create();
edmHit.setCellID(0);
edmHit.setEDep(0);
edmHit.setTime(0);
Expand Down
2 changes: 1 addition & 1 deletion SimG4Components/src/SimG4SingleParticleGeneratorTool.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ G4Event* SimG4SingleParticleGeneratorTool::g4Event() {
StatusCode SimG4SingleParticleGeneratorTool::saveToEdm(const G4PrimaryVertex* aVertex,
const G4PrimaryParticle* aParticle) {
edm4hep::MCParticleCollection* particles = new edm4hep::MCParticleCollection();
edm4hep::MCParticle particle = particles->create();
auto particle = particles->create();
particle.setVertex({
aVertex->GetX0() * sim::g42edm::length,
aVertex->GetY0() * sim::g42edm::length,
Expand Down
3 changes: 1 addition & 2 deletions SimG4Components/src/SimG4SmearGenParticles.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ StatusCode SimG4SmearGenParticles::execute() {

// todo: replace with copy / ctor method when available in podio
// relations currently not set!
edm4hep::MCParticle particle;
auto particle = particles->create();
particle.setCharge(j.getCharge());
particle.setPDG(j.getPDG());
particle.setMass(j.getMass());
Expand All @@ -71,7 +71,6 @@ StatusCode SimG4SmearGenParticles::execute() {
});

n_part++;
particles->push_back(particle);
}
}

Expand Down

0 comments on commit d481b1e

Please sign in to comment.