PHTpcDeltaZCorrection.h

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

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

#ifndef TRACKRECO_PHTPCDELTAZCORRECTION_H
#define TRACKRECO_PHTPCDELTAZCORRECTION_H

#include <fun4all/SubsysReco.h>
#include <phparameter/PHParameterInterface.h>
#include <trackbase/ActsGeometry.h>
#include <trackbase/TrkrDefs.h>

class TrackSeedContainer;
class TrkrClusterContainer;
class TrackSeed;

class PHTpcDeltaZCorrection : public SubsysReco, public PHParameterInterface
{
 public:

  PHTpcDeltaZCorrection(const std::string &name = "PHTpcDeltaZCorrection");

  ~PHTpcDeltaZCorrection() override = default;

  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;
  void SetDefaultParameters() override;

  private:

  int load_nodes( PHCompositeNode* );

  void process_tracks();

  void process_track( unsigned int, TrackSeed* );

  ActsGeometry *m_tGeometry = nullptr;

  TrackSeedContainer *m_track_map = nullptr;

  TrkrClusterContainer *m_cluster_map = nullptr;


  std::set<TrkrDefs::cluskey> m_corrected_clusters;

};

#endif // PHTpcDeltaZCorrection_H