#ifndef PndIsochroneTrackFinderTask_H_
#define PndIsochroneTrackFinderTask_H_

#include "FairTask.h"
#include "PndIsochroneTrackFinder.h"

#include "PndGeoSttPar.h"

#include <vector>

class PndIsochroneTrackFinderTask: public FairTask {
public:

	PndIsochroneTrackFinderTask() :
			FairTask("Stt Isochrone Track Finder"), fTrackCandArray(0), fSttTubeArray(0), fSttParameters(0), fVerbose(0), fPersistence(kTRUE){};

	virtual ~PndIsochroneTrackFinderTask() {};

	/** Virtual method Init **/
	virtual void SetParContainers();
	virtual InitStatus Init();
	//virtual InitStatus ReInit();

	/** Virtual method Exec **/
	virtual void Exec(Option_t* opt);
	virtual void FinishEvent();

	void AddHitBranch(TString branchName){fHitBranch.push_back(branchName);};//Has to be called before Init() is used otherwise the default names are taken!

	void SetVerbose(Int_t verbose) {
		fVerbose = verbose;
	};
	void SetSaveHistograms(bool value = true) {
		fSaveHistograms = value;
	};
	void SetPersistence(Bool_t val) {
		fPersistence = val;
	};

protected:
    void FillHitArray();
    void InitHitArray(TString branchName);

private:
	std::vector<TString> fHitBranch;
	std::vector<TClonesArray*> fHitArray;

	TClonesArray* fSttTubeArray;	//< Tube parameters


	PndGeoSttPar* fSttParameters;

//	TClonesArray* fSttHitArray;

	TClonesArray* fTrackArray;
	TClonesArray* fTrackCandArray;
	TClonesArray* fSttSkewedHitsArray;

	Int_t fVerbose;
	Bool_t fPersistence; // safe data?
	bool fSaveHistograms;

	PndIsochroneTrackFinder fTrackFinder;

	TClonesArray* fEventHeader;

ClassDef(PndIsochroneTrackFinderTask,1);

};

#endif /*PndIsochroneTrackFinderTask_H_*/