Back to home page

sPhenix code displayed by LXR

 
 

    


Warning, /coresoftware/offline/packages/trackreco/PHTpcTrackSeedCircleFit.h.outdated is written in an unsupported language. File is not indexed.

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef PHTPCTRACKSEEDCIRCLEFIT_H
0004 #define PHTPCTRACKSEEDCIRCLEFIT_H
0005 
0006 #include <fun4all/SubsysReco.h>
0007 #include <trackbase/ActsSurfaceMaps.h>
0008 #include <trackbase/ActsTrackingGeometry.h>
0009 #include <tpc/TpcDistortionCorrection.h>
0010 
0011 #include <string>
0012 #include <vector>
0013 
0014 class PHCompositeNode;
0015 class TrackSeed;
0016 class TrackSeedContainer;
0017 class TrkrCluster;
0018 class TF1;
0019 class TrkrClusterContainer;
0020 
0021 class PHTpcTrackSeedCircleFit : public SubsysReco
0022 {
0023  public:
0024 
0025   PHTpcTrackSeedCircleFit(const std::string &name = "PHTpcTrackSeedCircleFit");
0026 
0027   ~PHTpcTrackSeedCircleFit() override = default;
0028 
0029   int InitRun(PHCompositeNode* topNode) override;
0030 
0031   int process_event(PHCompositeNode*) override;
0032 
0033   int End(PHCompositeNode*) override;
0034 
0035   void use_truth_clusters(bool truth)
0036   { _use_truth_clusters = truth; }
0037 
0038   void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }
0039   void SetIteration(int iter){_n_iteration = iter;} 
0040 
0041  private:
0042 
0043   int GetNodes(PHCompositeNode* topNode);
0044 
0045   Acts::Vector3 getGlobalPosition( TrkrDefs::cluskey, TrkrCluster* cluster ) const;
0046                                                     
0047   ActsSurfaceMaps *_surfmaps{nullptr};
0048   ActsTrackingGeometry *_tGeometry{nullptr};
0049   TrackSeedContainer *_track_map{nullptr};
0050   
0051   bool _use_truth_clusters = false;
0052   TrkrClusterContainer *_cluster_map = nullptr;
0053   /// distortion correction container
0054   TpcDistortionCorrectionContainer* _dcc = nullptr;
0055  /// tpc distortion correction utility class
0056   TpcDistortionCorrection _distortionCorrection;
0057 
0058   int _n_iteration = 0;
0059   std::string _track_map_name = "SvtxTrackMap";
0060 
0061 };
0062 
0063 #endif // PHTRACKSEEDVERTEXASSOCIATION_H