StiKalmanTrackFinder.h

#ifndef StiKalmanTrackFinder_H_INCLUDED
#define StiKalmanTrackFinder_H_INCLUDED
#include "Sti/StiTrackFinder.h"
#include "Sti/Base/Named.h"
#include "Sti/Base/Described.h"
class TStopwatch;
class StiDetector;
class StiDetectorBuilder;
class StiDetectorContainer;
class StiToolkit;
class StiKalmanTrackFactory;
class StiHitContainer;
class StiTrackContainer;
class StiTrack;
class StiKalmanTrack;
class StiKalmanTrackNode;
template<class Factorized>class Factory;


class StiKalmanTrackFinder : public StiTrackFinder
{
public:
  StiKalmanTrackFinder() {}
  StiKalmanTrackFinder(StiToolkit *toolkit);
  void addSeedFinder(StiTrackFinder* sf) {_seedFinders.push_back(sf);}
  virtual ~StiKalmanTrackFinder() {}
  virtual void initialize();
          void setTiming();
  virtual void findTracks(); 
          bool find(StiTrack *track, int direction, double rmin=0);
  virtual StiTrack * findTrack(double rMin=0); 
  void extendSeeds (double rMin);
  void extendTracks(double rMin);
  int extendTrack(StiKalmanTrack *track,double rMin);
  void extendTracksToVertices(const std::vector<StiHit*> &vertices);
  int getNTracks() const ;
  int getNPrims()  const { return _nPrimTracks;}
  virtual void reset();
  virtual void unset(){}
  
  virtual void clear();
  virtual void finish() const;
  virtual Int_t Fit(StiKalmanTrack *track, Double_t rMin=0);
  virtual Filter<StiTrack> * getTrackFilter();
  static void setDebug(int m = 0) {_debug = m;}
  static int  debug() {return _debug;}
  typedef enum{ // type of return value for the Fit() procedure
    kNoErrors = 0,
    kApproxFail,
    kFitFail,
    kExtendFail,
    kCheckFail
  } TFitStatus;

  typedef enum{ // type of return value for the extendTrack() procedure
    kExtended,
    kNotExtended,
    kNotRefitedIn,
    kNotRefitedOut
  } TExtendStatus;

  
private:
class QAFind;
  void find(StiKalmanTrack *track, int direction,StiKalmanTrackNode *node,QAFind &qa);
  void nodeQA(StiKalmanTrackNode *node, int position,int active,QAFind &qa);
  int  compQA(QAFind &qaBest,QAFind &qaTry,double maxChi2);
 
 protected:

    void printState();
    StiToolkit                  * _toolkit;
    Filter<StiTrack>            * _trackFilter;
    Factory<StiKalmanTrackNode> * _trackNodeFactory;
    StiDetectorContainer        * _detectorContainer;
    StiHitContainer             * _hitContainer;
    StiTrackContainer           * _trackContainer;
    std::vector<StiTrackFinder*>  _seedFinders;
    int                           _nPrimTracks;
    int         mEventPerm;     //Count number of permutations

private:
        
    double    chi2;
    TStopwatch *mTimg[3];       //seeds,traks,prims
    int         mTrackPerm;     //Count number of permutations
    int         mUseComb;       //useComb() saved 
    static int   _debug;
};

inline Filter<StiTrack> * StiKalmanTrackFinder::getTrackFilter() 
{
  return _trackFilter;
}


class CloserAngle
{
  public:
  CloserAngle(double refAngle);
  bool operator()(const StiDetector*lhs, const StiDetector* rhs);
 protected:
  double _refAngle;
};

#endif