DSTTrackInfoReader.h

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

#ifndef G4EVAL_DSTTRACKINFOREADER_H
#define G4EVAL_DSTTRACKINFOREADER_H

// #include "DSTContainerv3.h"

#include <fun4all/SubsysReco.h>
#include <trackbase/TrkrClusterContainerv4.h>
#include <trackbase/TrkrClusterv5.h>
#include <trackbase/TrkrDefs.h>
#include <trackbase_historic/TrackInfoContainer_v1.h>
#include <trackbase_historic/TrackSeedContainer_v1.h>
#include <trackbase_historic/TrackSeed_v1.h>

#include <map>
#include <set>
#include <string>
#include <vector>

class PHG4Hit;
class PHG4HitContainer;
class PHG4Particle;
class PHG4TruthInfoContainer;
class SvtxTrack;
class SvtxTrackMap;
// class DSTContainer;
class TrkrCluster;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class TrkrHitSetContainer;
class TrkrHitTruthAssoc;

class DSTTrackInfoReader : public SubsysReco
{
 public:
  DSTTrackInfoReader(const std::string& = "DSTTrackInfoReader");

  int InitRun(PHCompositeNode*) override;

  int process_event(PHCompositeNode*) override;

  enum Flags
  {
    WriteEvent = 1 << 0,
    WriteClusters = 1 << 1,
    WriteTracks = 1 << 2
  };

  void set_flags(int flags)
  {
    m_flags = flags;
  }

 private:
  int load_nodes(PHCompositeNode*);

  void evaluate_track_info();

  // SvtxTrack recover_track(DSTContainerv3::TrackStruct);

  // TrkrCluster recover_cluster(DSTContainerv3::ClusterStruct);

  TrackInfoContainer_v1* m_track_info_container = nullptr;
  // DSTContainer* m_container = nullptr;

  int m_flags = WriteEvent | WriteClusters | WriteTracks;


  // debugging helpers
  // bool dryrun = false;
  // bool generateKey = false;
};

#endif  // G4EVAL_DSTTRACKINFOREADER_H