ThreeHitSeedGrower.h

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

#ifndef __THREEHITSEEDGROWER__
#define __THREEHITSEEDGROWER__

#include "HelixHough.h"
#include <vector>
#include <set>

class ThreeHitSeedGrower : public HelixHough
{
  public:
    ThreeHitSeedGrower(std::vector<float>& detrad, unsigned int n_phi, unsigned int n_d, unsigned int n_k, unsigned int n_dzdl, unsigned int n_z0, HelixResolution& min_resolution, HelixResolution& max_resolution, HelixRange& range);
    virtual ~ThreeHitSeedGrower(){}
    

    bool GrowTrack(SimpleTrack3D seed_track, std::vector<unsigned int>& nhit_layer, std::vector<SimpleHit3D>& hits, std::vector<SimpleTrack3D>& tracks, unsigned int c_hit);
    bool addOneHit(SimpleTrack3D & seed_track, std::vector<unsigned int> & nhit_layer, unsigned int c_hit, std::vector<SimpleHit3D>& hits);
    void finalize(std::vector<SimpleTrack3D>& input, std::vector<SimpleTrack3D>& output);
    void findTracks(std::vector<SimpleHit3D>& hits, std::vector<SimpleTrack3D>& tracks);
    void initEvent(std::vector<SimpleHit3D>& hits)
    {
      combos.clear();
    }
    
    void setUsingVertex(bool usevtx){using_vertex = usevtx;}

    void setMaxHits(unsigned int max_hits){_max_hits = max_hits;}
    
    void setLayerResolution(std::vector<float>& lxy, std::vector<float>& lz)
    {
      layer_xy_resolution = lxy;
      layer_z_resolution = lz;
    }
    
    void setVertexResolution(float vxy, float vz)
    {
      vertex_sigma_xy = vxy;
      vertex_sigma_z = vz;
    }
    
    double fitTrack(SimpleTrack3D & track, std::vector<double>& chi2_hit);
    
    void setChi2Cut(double c){chi2_cut=c;}
    
  private:
    bool using_vertex;
    std::vector<float> detector_radii;
    std::vector<float> layer_xy_resolution;
    std::vector<float> layer_z_resolution;
    double vertex_sigma_xy, vertex_sigma_z;
    double chi2_cut;
    unsigned int _max_hits;
    std::set<std::vector<unsigned int> > combos; 
};


#endif