// -------------------------------------------------------------------------
// -----          PndFwdPrepareKalmanTracks  header file               -----
// -----            Created 30.09.2008  by A. Wronska                  -----
// -----        based on the recotasks/demo code  by S.Neubert         -----
// -------------------------------------------------------------------------

/** \class PndFwdPrepareKalmanTracks2
 *  \author A.Wronska <aleksandra.wronska@uj.edu.pl>
 *  \date 15.05.2008
 *  \modified by Radoslaw Karabowicz <r.karabowicz@gsi.de> for GEM on 03/04/2009
 *  \brief PndFwdPrepareKalmanTracks2 creates and initializes Tracks and is a pre-step for genfit
 **/
#ifndef PNDFWDPREPAREKALMANTRACKS_H
#define PNDFWDPREPAREKALMANTRACKS_H

// Pnd Headres
#include "FairTask.h"

// c++ headers
#include <map>

class TClonesArray;
class RecoHitFactory;
class AbsBFieldIfc;
class FairField;
class FairGeanePro;

class PndFwdPrepareKalmanTracks : public FairTask {
public:

  /**  Constructor  **/
  PndFwdPrepareKalmanTracks();

  /**  Destructor **/
  ~PndFwdPrepareKalmanTracks();

  /** public method SetPersistence(Bool_t opt) **/
  void SetPersistence(Bool_t opt=kTRUE) {fPersistence=opt;}

  /** public method SetField(FairField* f) to keep pointer to the B field **/
  void SetField(FairField* f){fField=f;}

  /** public method UseGeane(Bool_t f) to switch on the use of Geane **/
  void UseGeane(Bool_t f=kTRUE){fUseGeane=f;}

  /** public method UseMC(Bool_t f) to switch on init from mc or to 0 if not needed**/
  void UseMC(Bool_t f=kTRUE){fUseMC=f;}

  /** public method UseGemDchMC(Bool_t f, Bool_t h) to in/exclude one of the detectors from reco**/
  void UseGemDch(Bool_t f, Bool_t h){fUseGem=f; fUseDch=h;}

  /** public method SetMinNuOfHits(Int_t nu) sets lower limit 
      for number of hits in a processable track **/
  void SetMinNofGemHits(Int_t nu=6){fMinNofGemHits=nu;}

  void SetMinNofDchHits(Int_t nu=6){fMinNofDchHits=nu;}

  /** public method SetPDG(Int_t f) if useMC == kTRUE it is ignored **/
  void SetPDG(Int_t f=11){fPDG=f;}

  /** public method Init()
      fetches and creates all necessary arrays, creates Geane **/
  virtual InitStatus Init();

  /** public method Exec(Option_t* opt)
      creates TrackCands, find starting values for Kalman and creates Track objects **/
  virtual void Exec(Option_t* opt);


private:

  TClonesArray* fTrackArray;          ///< output array of genfit Tracks

  TClonesArray* fGemPointArray;       ///< input array of PndGemPoints (needed for initialization of parameters)
  TClonesArray* fGemTrackArray;       ///< input array of PndGemTracks to be processed by this task
  TClonesArray* fDchPointArray;       ///< input array of PndDchPoints (needed for initialization of parameters)
  TClonesArray* fDchTrackArray;       ///< input array of PndDchTracks to be processed by this task
  TClonesArray* fDchTrackMatchArray;  ///<input array of PndDchTrackMatches (needed for initialization) 
  TClonesArray* fMcArray;             ///<input array of MCTracks (needed for initialization) 
  TClonesArray* fGemHitArray;            ///<input array of PndGemHits
  TClonesArray* fDchHitArray;            ///<input array of PndDchHits

  Int_t fMinNofGemHits;    ///< minimum number of hits to further process a track
  Int_t fMinNofDchHits;    ///< minimum number of hits to further process a track
  Bool_t fPersistence;    ///< persistance flag
  Bool_t fUseGeane;       ///< flag switching on/off use of Geane
  Bool_t fUseMC;          ///< flag switching on/off initialisation of track params from MC
  Bool_t fUseGem;          ///< flag switching on/off GEMs from reco
  Bool_t fUseDch;          ///< flag switching on/off DCHs from reco
  Int_t fPDG;  ///< hipothesis of PDG value 
    //It  (shouldn't be here)!!!!

  FairField*    fField;    ///< pointer to field to be used by Geane 
  FairGeanePro* fGeanePro; ///< pointer to Geane object


public:
  ClassDef(PndFwdPrepareKalmanTracks,1)

};

#endif