HepMC3Vertex.hpp

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

// 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/.

#pragma once

#include "ActsExamples/EventData/SimParticle.hpp"
#include "ActsExamples/EventData/SimVertex.hpp"

#include <HepMC3/FourVector.h>
#include <HepMC3/GenParticle.h>
#include <HepMC3/GenVertex.h>

namespace ActsExamples {

struct HepMC3Vertex {
 public:
  std::unique_ptr<SimVertex> processVertex(
      const std::shared_ptr<HepMC3::GenVertex> vertex);

  bool inEvent(const std::shared_ptr<HepMC3::GenVertex> vertex);

  int id(const std::shared_ptr<HepMC3::GenVertex> vertex);

  std::vector<SimParticle> particlesIn(
      const std::shared_ptr<HepMC3::GenVertex> vertex);

  std::vector<SimParticle> particlesOut(
      const std::shared_ptr<HepMC3::GenVertex> vertex);

  Acts::Vector3D position(const std::shared_ptr<HepMC3::GenVertex> vertex);

  double time(const std::shared_ptr<HepMC3::GenVertex> vertex);

  void addParticleIn(std::shared_ptr<HepMC3::GenVertex> vertex,
                     std::shared_ptr<SimParticle> particle);

  void addParticleOut(std::shared_ptr<HepMC3::GenVertex> vertex,
                      std::shared_ptr<SimParticle> particle);

  void removeParticleIn(std::shared_ptr<HepMC3::GenVertex> vertex,
                        std::shared_ptr<SimParticle> particle);

  void removeParticleOut(std::shared_ptr<HepMC3::GenVertex> vertex,
                         std::shared_ptr<SimParticle> particle);

  void position(const std::shared_ptr<HepMC3::GenVertex> vertex,
                Acts::Vector3D pos);

  void time(const std::shared_ptr<HepMC3::GenVertex> vertex, double time);

 private:
  std::vector<SimParticle> genParticlesToActs(
      const std::vector<HepMC3::GenParticlePtr>& genParticles);

  HepMC3::GenParticlePtr actsParticleToGen(
      std::shared_ptr<SimParticle> actsParticle);

  HepMC3::GenParticlePtr matchParticles(
      const std::vector<HepMC3::GenParticlePtr>& genParticles,
      std::shared_ptr<SimParticle> actsParticle);
};
}  // namespace ActsExamples