TrackMlData.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2019 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 <cstdint>

#include <dfe/dfe_namedtuple.hpp>

namespace ActsExamples {

struct ParticleData {
  uint64_t particle_id;
  int32_t particle_type;
  uint32_t process = 0u;
  float vx, vy, vz;
  // Production time in ns. Not available in the TrackML datasets.
  float vt = 0.0f;
  float px, py, pz;
  float m = 0.0f;
  float q;

  DFE_NAMEDTUPLE(ParticleData, particle_id, particle_type, process, vx, vy, vz,
                 vt, px, py, pz, m, q);
};

struct TruthHitData {
  uint64_t hit_id;
  uint64_t geometry_id = 0u;
  uint64_t particle_id;
  float tx, ty, tz;
  // True global hit time in ns. Not available in the TrackML datasets.
  float tt = 0.0f;
  float tpx, tpy, tpz;
  float te = 0.0f;
  float deltapx = 0.0f;
  float deltapy = 0.0f;
  float deltapz = 0.0f;
  float deltae = 0.0f;
  // Hit index along the trajectory. Not available in the TrackML datasets.
  int32_t index = -1;

  DFE_NAMEDTUPLE(TruthHitData, hit_id, particle_id, geometry_id, tx, ty, tz, tt,
                 tpx, tpy, tpz, te, deltapx, deltapy, deltapz, deltae, index);
};

struct HitData {
  uint64_t hit_id;
  uint64_t geometry_id = 0u;
  uint32_t volume_id, layer_id, module_id;
  float x, y, z;
  float t = 0.0f;

  DFE_NAMEDTUPLE(HitData, hit_id, geometry_id, volume_id, layer_id, module_id,
                 x, y, z, t);
};

struct CellData {
  uint64_t hit_id;
  int32_t ch0, ch1;
  int32_t timestamp = 0;
  int32_t value;

  DFE_NAMEDTUPLE(CellData, hit_id, ch0, ch1, timestamp, value);
};

struct SurfaceData {
  uint64_t geometry_id;
  uint32_t volume_id, layer_id, module_id;
  float cx, cy, cz;
  float rot_xu, rot_xv, rot_xw;
  float rot_yu, rot_yv, rot_yw;
  float rot_zu, rot_zv, rot_zw;
  float module_t = -1;
  float module_minhu = -1;
  float module_maxhu = -1;
  float module_hv = -1;
  float pitch_u = -1;
  float pitch_v = -1;

  DFE_NAMEDTUPLE(SurfaceData, geometry_id, volume_id, layer_id, module_id, cx,
                 cy, cz, rot_xu, rot_xv, rot_xw, rot_yu, rot_yv, rot_yw, rot_zu,
                 rot_zv, rot_zw, module_t, module_minhu, module_maxhu,
                 module_hv, pitch_u, pitch_v);
};

}  // namespace ActsExamples