#ifndef CBMSTTFITVERTICES
#define CBMSTTFITVERTICES

#include "CbmTask.h"

class CbmSttVertexFitter;

class CbmSttFitVertices : public CbmTask
{
 public:
  /** Default constructor **/
  CbmSttFitVertices();

  /** Standard constructor 
   **
   *@param name   Name of class
   *@param title  Task title
   *@param fitter Pointer to STT vertex fitter concrete class
   **/
  CbmSttFitVertices(const char* name, const char* title = "CbmTask",
		    CbmSttVertexFitter* fitter = NULL);

  /** Destructor **/
  virtual ~CbmSttFitVertices();

  /** Initialisation at beginning of each event **/
  virtual InitStatus Init();

  /** Task execution **/
  virtual void Exec(Option_t* opt);

  /** Finish at the end of each event **/
  virtual void Finish();

  /** Accessors **/
  CbmSttVertexFitter* GetFitter() { return fFitter; };
  Int_t GetNofVertices()          { return fNofVertices; };

  /** Set concrete vertex finder **/
  void UseFitter(CbmSttVertexFitter* fitter) { fFitter = fitter; };

 private:
  CbmSttVertexFitter* fFitter;    // Pointer to VertexFinder concrete class
  TClonesArray* fVertexArray;     // Input array of STT tracks
  Int_t fNofVertices;             // Number of vertices successfully fitted

  ClassDef(CbmSttFitVertices,1);
};

#endif