HepMC3Event.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file HepMC3Event.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 "Acts/Utilities/Units.hpp"
#include "ActsExamples/EventData/SimParticle.hpp"
#include "ActsExamples/EventData/SimVertex.hpp"

#include <HepMC3/FourVector.h>
#include <HepMC3/GenEvent.h>
#include <HepMC3/GenParticle.h>
#include <HepMC3/GenVertex.h>
#include <HepPID/ParticleIDMethods.hh>

namespace ActsExamples {

struct HepMC3Event {
 public:

  void momentumUnit(std::shared_ptr<HepMC3::GenEvent> event,
                    const double momentumUnit);

  void lengthUnit(std::shared_ptr<HepMC3::GenEvent> event,
                  const double lengthUnit);

  void shiftPositionBy(std::shared_ptr<HepMC3::GenEvent> event,
                       const Acts::Vector3D& deltaPos, const double deltaTime);

  void shiftPositionTo(std::shared_ptr<HepMC3::GenEvent> event,
                       const Acts::Vector3D& pos, const double time);

  void shiftPositionTo(std::shared_ptr<HepMC3::GenEvent> event,
                       const Acts::Vector3D& pos);

  void shiftPositionTo(std::shared_ptr<HepMC3::GenEvent> event,
                       const double time);


  void addParticle(std::shared_ptr<HepMC3::GenEvent> event,
                   std::shared_ptr<SimParticle> particle);

  void addVertex(std::shared_ptr<HepMC3::GenEvent> event,
                 const std::shared_ptr<SimVertex> vertex);

  void removeParticle(std::shared_ptr<HepMC3::GenEvent> event,
                      const std::shared_ptr<SimParticle>& particle);

  void removeVertex(std::shared_ptr<HepMC3::GenEvent> event,
                    const std::shared_ptr<SimVertex>& vertex);


  double momentumUnit(const std::shared_ptr<HepMC3::GenEvent> event);

  double lengthUnit(const std::shared_ptr<HepMC3::GenEvent> event);

  Acts::Vector3D eventPos(const std::shared_ptr<HepMC3::GenEvent> event);

  double eventTime(const std::shared_ptr<HepMC3::GenEvent> event);

  std::vector<std::unique_ptr<SimParticle>> particles(
      const std::shared_ptr<HepMC3::GenEvent> event);

  std::vector<std::unique_ptr<SimVertex>> vertices(
      const std::shared_ptr<HepMC3::GenEvent> event);

  std::vector<std::unique_ptr<SimParticle>> beams(
      const std::shared_ptr<HepMC3::GenEvent> event);

  std::vector<std::unique_ptr<SimParticle>> finalState(
      const std::shared_ptr<HepMC3::GenEvent> event);

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

  HepMC3::GenVertexPtr createGenVertex(
      const std::shared_ptr<SimVertex>& actsVertex);

  bool compareVertices(const std::shared_ptr<SimVertex>& actsVertex,
                       const HepMC3::GenVertexPtr& genVertex);
};
}  // namespace ActsExamples