PHTpcTrackSeedCircleFit.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef PHTPCTRACKSEEDCIRCLEFIT_H
#define PHTPCTRACKSEEDCIRCLEFIT_H

#include <fun4all/SubsysReco.h>
#include <trackbase/ActsSurfaceMaps.h>
#include <trackbase/ActsTrackingGeometry.h>
#include <tpc/TpcDistortionCorrection.h>

#include <string>
#include <vector>

class PHCompositeNode;
class SvtxTrackMap;
class SvtxTrack;
class TrkrCluster;
class TF1;
class TrkrClusterContainer;

class PHTpcTrackSeedCircleFit : public SubsysReco
{
 public:

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

  ~PHTpcTrackSeedCircleFit() override = default;

  int InitRun(PHCompositeNode* topNode) override;

  int process_event(PHCompositeNode*) override;

  int End(PHCompositeNode*) override;

  void use_truth_clusters(bool truth)
  { _use_truth_clusters = truth; }

  void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }
  void SetIteration(int iter){_n_iteration = iter;} 

 private:

  int GetNodes(PHCompositeNode* topNode);
  std::vector<TrkrCluster*> getTrackClusters(SvtxTrack *);
  Acts::Vector3D getGlobalPosition( TrkrCluster* cluster ) const;
                
  ActsSurfaceMaps *_surfmaps{nullptr};
  ActsTrackingGeometry *_tGeometry{nullptr};
  SvtxTrackMap *_track_map{nullptr};
  
  bool _use_truth_clusters = false;
  bool _are_clusters_corrected = true;
  TrkrClusterContainer *_cluster_map = nullptr;
  TpcDistortionCorrectionContainer* _dcc = nullptr;
  TpcDistortionCorrection _distortionCorrection;

  int _n_iteration = 0;
  std::string _track_map_name = "SvtxTrackMap";

};

#endif // PHTRACKSEEDVERTEXASSOCIATION_H