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 = nullptr);

  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