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;


    float x = 0;
    float y = 0;
    float z = 0;
    float r = 0;
    float phi = 0;
    float phi_error = 0;
    float 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 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;

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