HepMC3Vertex.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file HepMC3Vertex.cpp

// This file is part of the Acts project.
//
// Copyright (C) 2018 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include "ActsExamples/Io/HepMC3/HepMC3Vertex.hpp"

#include "ActsExamples/Io/HepMC3/HepMC3Particle.hpp"

namespace {

std::vector<ActsExamples::SimParticle> genParticlesToActs(
    const std::vector<HepMC3::GenParticlePtr>& genParticles) {
  std::vector<ActsExamples::SimParticle> actsParticles;
  // Translate all particles
  for (auto& genParticle : genParticles) {
    actsParticles.push_back(ActsExamples::HepMC3Particle::particle(
        std::make_shared<HepMC3::GenParticle>(*genParticle)));
  }
  return actsParticles;
}

HepMC3::GenParticlePtr actsParticleToGen(
    const std::shared_ptr<ActsExamples::SimParticle>& actsParticle) {
  // Extract momentum and energy from Acts particle for HepMC3::FourVector
  const auto mom = actsParticle->fourMomentum();
  const HepMC3::FourVector vec(mom[0], mom[1], mom[2], mom[3]);
  // Create HepMC3::GenParticle
  auto genParticle =
      std::make_shared<HepMC3::GenParticle>(vec, actsParticle->pdg());
  genParticle->set_generated_mass(actsParticle->mass());

  return genParticle;
}

HepMC3::GenParticlePtr matchParticles(
    const std::vector<HepMC3::GenParticlePtr>& genParticles,
    const std::shared_ptr<ActsExamples::SimParticle>& actsParticle) {
  const auto id = actsParticle->particleId();
  // Search HepMC3::GenParticle with the same id as the Acts particle
  for (auto& genParticle : genParticles) {
    if (genParticle->id() == id) {
      // Return particle if found
      return genParticle;
    }
  }
  return nullptr;
}

}  // namespace

std::unique_ptr<ActsExamples::SimVertex>
ActsExamples::HepMC3Vertex::processVertex(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  SimVertex vtx({vertex->position().x(), vertex->position().y(),
                 vertex->position().z(), vertex->position().t()});
  vtx.incoming = genParticlesToActs(vertex->particles_in());
  vtx.outgoing = genParticlesToActs(vertex->particles_out());
  // Create Acts vertex
  return std::make_unique<SimVertex>(std::move(vtx));
}

bool ActsExamples::HepMC3Vertex::inEvent(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  return vertex->in_event();
}

int ActsExamples::HepMC3Vertex::id(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  return vertex->id();
}

std::vector<ActsExamples::SimParticle> ActsExamples::HepMC3Vertex::particlesIn(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  return genParticlesToActs(vertex->particles_in());
}

std::vector<ActsExamples::SimParticle> ActsExamples::HepMC3Vertex::particlesOut(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  return genParticlesToActs(vertex->particles_out());
}

Acts::Vector3 ActsExamples::HepMC3Vertex::position(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  Acts::Vector3 vec;
  vec(0) = vertex->position().x();
  vec(1) = vertex->position().y();
  vec(2) = vertex->position().z();
  return vec;
}

double ActsExamples::HepMC3Vertex::time(
    const std::shared_ptr<HepMC3::GenVertex>& vertex) {
  return vertex->position().t();
}

void ActsExamples::HepMC3Vertex::addParticleIn(
    const std::shared_ptr<HepMC3::GenVertex>& vertex,
    const std::shared_ptr<SimParticle>& particle) {
  vertex->add_particle_in(actsParticleToGen(particle));
}

void ActsExamples::HepMC3Vertex::addParticleOut(
    const std::shared_ptr<HepMC3::GenVertex>& vertex,
    const std::shared_ptr<SimParticle>& particle) {
  vertex->add_particle_out(actsParticleToGen(particle));
}

void ActsExamples::HepMC3Vertex::removeParticleIn(
    const std::shared_ptr<HepMC3::GenVertex>& vertex,
    const std::shared_ptr<SimParticle>& particle) {
  // Remove particle if it exists
  if (HepMC3::GenParticlePtr genParticle =
          matchParticles(vertex->particles_in(), particle)) {
    vertex->remove_particle_in(genParticle);
  }
}

void ActsExamples::HepMC3Vertex::removeParticleOut(
    const std::shared_ptr<HepMC3::GenVertex>& vertex,
    const std::shared_ptr<SimParticle>& particle) {
  // Remove particle if it exists
  if (HepMC3::GenParticlePtr genParticle =
          matchParticles(vertex->particles_out(), particle)) {
    vertex->remove_particle_out(genParticle);
  }
}

void ActsExamples::HepMC3Vertex::position(
    const std::shared_ptr<HepMC3::GenVertex>& vertex,
    const Acts::Vector3& pos) {
  HepMC3::FourVector fVec(pos(0), pos(1), pos(2), vertex->position().t());
  vertex->set_position(fVec);
}

void ActsExamples::HepMC3Vertex::time(
    const std::shared_ptr<HepMC3::GenVertex>& vertex, double time) {
  HepMC3::FourVector fVec(vertex->position().x(), vertex->position().y(),
                          vertex->position().z(), time);
  vertex->set_position(fVec);
}