TrackEvaluationContainerv1.h

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

// Tell emacs that this is a C++ source
// -*- C++ -*-.

#ifndef G4EVAL_TRACKEVALUATIONCONTAINERV1_H
#define G4EVAL_TRACKEVALUATIONCONTAINERV1_H

#include "TrackEvaluationContainer.h"

#include <vector>


class TrackEvaluationContainerv1 : public TrackEvaluationContainer
{
 public:
  explicit TrackEvaluationContainerv1()
  {
    // only one event structure per event (!)
    m_events.reserve(1);
  }

  void Reset() override;


  class EventStruct
  {
   public:
    using List = std::vector<EventStruct>;
    static constexpr size_t max_layer = 57;

    // constructor
    EventStruct()
    {
      for (size_t i = 0; i < max_layer; ++i)
      {
        nclusters[i] = 0;
      }
    }

    int nclusters[max_layer];

    int nclusters_mvtx = 0;

    int nclusters_intt = 0;

    int nclusters_tpc = 0;

    int nclusters_micromegas = 0;
  };


  class ClusterStruct
  {
   public:
    using List = std::vector<ClusterStruct>;

    unsigned int layer = 0;

    unsigned int size = 0;

    unsigned int truth_size = 0;

    int phi_size = 0;
    int z_size = 0;

    unsigned int adc = 0;
    unsigned int max_adc = 0;
    int ovlp = 0;
    int edge = 0;


    float x = 0;
    float y = 0;
    float z = 0;
    float r = 0;
    float phi = 0;
    float phi_error = 0;
    float z_error = 0;
    float para_phi_error = 0;
    float para_z_error = 0;


    float trk_x = 0;
    float trk_y = 0;
    float trk_z = 0;
    float trk_r = 0;
    float trk_phi = 0;

    float trk_phi_error = 0;
    float trk_z_error = 0;

    float trk_alpha = 0;

    float trk_beta = 0;

    float trk_radius = 0;



    float truth_x = 0;
    float truth_y = 0;
    float truth_z = 0;
    float truth_r = 0;
    float truth_phi = 0;

    float truth_alpha = 0;

    float truth_beta = 0;



    //* maximum charge on strip
    float energy_max = 0;

    //* sum of strip charges
    float energy_sum = 0;



    float trk_px = 0;
    float trk_py = 0;
    float trk_pz = 0;


    float truth_px = 0;
    float truth_py = 0;
    float truth_pz = 0;
  };


  class TrackStruct
  {
   public:
    // constructor
    explicit TrackStruct()
    {
      // allocate enough size for the clusters
      static constexpr int max_layers = 60;
      clusters.reserve(max_layers);
    }

    using List = std::vector<TrackStruct>;

    int charge = 0;
    int nclusters = 0;
    int64_t mask = 0;

    int nclusters_mvtx = 0;
    int nclusters_intt = 0;
    int nclusters_tpc = 0;
    int nclusters_micromegas = 0;

    float chisquare = 0;
    int ndf = 0;


    float x = 0;
    float y = 0;
    float z = 0;
    float r = 0;
    float phi = 0;


    float px = 0;
    float py = 0;
    float pz = 0;
    float pt = 0;
    float p = 0;
    float eta = 0;


    int pid = 0;
    int embed = 0;
    bool is_primary = false;
    int gtrackID = 0;
    float truth_t = 0;

    // number of g4hits from this MC track that match
    int contributors = 0;

    float truth_px = 0;
    float truth_py = 0;
    float truth_pz = 0;
    float truth_pt = 0;
    float truth_p = 0;
    float truth_eta = 0;

    // associate clusters
    ClusterStruct::List clusters;
  };



  const EventStruct::List& events() const
  {
    return m_events;
  }

  const ClusterStruct::List& clusters() const
  {
    return m_clusters;
  }

  const TrackStruct::List& tracks() const
  {
    return m_tracks;
  }




  void addEvent(const EventStruct& event)
  {
    m_events.push_back(event);
  }

  void addCluster(const ClusterStruct& cluster)
  {
    m_clusters.push_back(cluster);
  }

  void addTrack(const TrackStruct& track)
  {
    m_tracks.push_back(track);
  }

  void clearEvents()
  {
    m_events.clear();
  }

  void clearClusters()
  {
    m_clusters.clear();
  }

  void clearTracks()
  {
    m_tracks.clear();
  }


 private:
  /* there is only one element per event in this array */
  EventStruct::List m_events;

  ClusterStruct::List m_clusters;

  TrackStruct::List m_tracks;

  ClassDefOverride(TrackEvaluationContainerv1, 1)
};

#endif