d0/d2b/RootParticleWriter_8hpp_source.html

// This file is part of the Acts project.

//

// Copyright (C) 2017-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/Framework/WriterT.hpp"


#include <cstdint>

#include <mutex>

#include <string>


class TFile;

class TTree;


namespace ActsExamples {


class RootParticleWriter final : public WriterT<SimParticleContainer> {

 public:

  struct Config {

    std::string inputParticles;

    std::string filePath;

    std::string fileMode = "RECREATE";

    std::string treeName = "particles";

  };


  RootParticleWriter(const Config& cfg, Acts::Logging::Level lvl);


  ~RootParticleWriter() final override;


  ProcessCode endRun() final override;


 protected:

  ProcessCode writeT(const AlgorithmContext& ctx,

                     const SimParticleContainer& particles) final override;


 private:

  Config m_cfg;

  std::mutex m_writeMutex;

  TFile* m_outputFile = nullptr;

  TTree* m_outputTree = nullptr;

  uint32_t m_eventId;

  uint64_t m_particleId;

  int32_t m_particleType;

  uint32_t m_process;

  float m_vx, m_vy, m_vz;

  // Production time in ns.

  float m_vt;

  float m_px, m_py, m_pz;

  float m_m;

  float m_q;

  // Derived kinematic quantities

  float m_eta;

  float m_phi;

  float m_pt;

  // Decoded particle identifier; see Barcode definition for details.

  uint32_t m_vertexPrimary;

  uint32_t m_vertexSecondary;

  uint32_t m_particle;

  uint32_t m_generation;

  uint32_t m_subParticle;

};


}  // namespace ActsExamples