PndSttFitTracks.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                    PndSttFitTracks header file                -----
// -----                  Created 18/02/05  by V. Friese               -----
// -------------------------------------------------------------------------


#ifndef PNDSTTFITTRACKS
#define PNDSTTFITTRACKS 1


#include "FairTask.h"
#include "PndGeoSttPar.h"

#include <string>
#include <vector>

class PndSttTrackFitter;
class TClonesArray;

class PndSttFitTracks : public FairTask
{

 public:

  PndSttFitTracks();


  PndSttFitTracks(const char* name, const char* title = "FairTask",
                  PndSttTrackFitter* fitter = NULL);


  virtual ~PndSttFitTracks();


  virtual InitStatus Init();


  virtual void Exec(Option_t* opt);


  virtual void Finish();


  PndSttTrackFitter* GetFitter() { return fFitter; };
  Int_t GetNofTracks()           { return fNofTracks; };


  void UseFitter(PndSttTrackFitter* fitter) { fFitter = fitter; };

  void AddHitCollectionName(char *hitCollectionName);


 private:
  void AddAllCollections(); 
  void AddHitCollection(char const *collectionName);

  void SetParContainers();

  PndSttTrackFitter* fFitter;    // Pointer to TrackFinder concrete class
  TClonesArray* fTrackCandArray; // Input array of STT track candidates 
  TClonesArray* fTrackArray;     // Output array of STT tracks  
  TClonesArray *fHitArray;
  Int_t fNofTracks;              // Number of tracks successfully fitted
  std::vector<std::string> fHitCollectionNames;
  Bool_t fCollectionsComplete;
  //  CHECK added
  TClonesArray* fTubeArray;

  PndGeoSttPar *fSttParameters;  //  CHECK added

  ClassDef(PndSttFitTracks,1);

};

#endif