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

/** \class PndDchPrepareKalmanTracks2
 *  \author A.Wronska <aleksandra.wronska@uj.edu.pl>
 *  \date 15.05.2008
 *  \brief PndDchPrepareKalmanTracks2 creates and initializes Tracks and is a pre-step for genfit
 **/
#ifndef PNDDCHPREPAREKALMANTRACKS2_HH
#define PNDDCHPREPAREKALMANTRACKS2_HH

// Pnd Headres
#include "FairTask.h"

// c++ headers
#include <map>

class TClonesArray;
class GFRecoHitFactory;
class AbsBFieldIfc;
class FairField;
class FairGeanePro;

class PndDchPrepareKalmanTracks2 : public FairTask {
public:

  /**  Constructor  **/
  PndDchPrepareKalmanTracks2();

  /**  Destructor **/
  ~PndDchPrepareKalmanTracks2();

  /** public method AddHitBranch(detID, hitBranchName) adds 
   more hit branches to be included in the Kalman analysis,
  here used as if private **/
  void AddHitBranch(unsigned int detId, const TString& m){fHitBranchNameMap[detId]=m;};

  /** 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 SetMinNuOfHits(Int_t nu) sets lower limit 
      for number of hits in a processable track **/
  void SetMinNuOfHits(Int_t nu=30){fMinNuOfHits=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 GFTrackCands, find starting values for Kalman and creates Track objects **/
  virtual void Exec(Option_t* opt);


private:

  TClonesArray* fTrackArray;          ///< output array of genfit Tracks
  TClonesArray* fDchPointArray;       ///< input array of PndDchPoints (needed for initialization of parameters)
  TClonesArray* fDchTrackArray;       ///< input array of PndTrackCands to be processed by this task
  TClonesArray* fMcArray;             ///<input array of MCTracks (needed for initialization) 
  TClonesArray* fCHitArray;           ///<input array of PndDchCylinderHits

  std::map<unsigned int,TString> fHitBranchNameMap;   ///< map of relevant <detID,name of hit branch>
  std::map<unsigned int,TClonesArray*> fHitBranchMap; ///< map of relevant <detID, hit array>
    Int_t fMinNuOfHits;    ///< 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
  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(PndDchPrepareKalmanTracks2,1)

};

#endif