//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Kalman Filter Task for the TC test Bench prototypes
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Maxence Vandenbroucke    TUM            (original author)
//
//
//-----------------------------------------------------------

#ifndef TBKALMANTASK_HH
#define TBKALMANTASK_HH

// Base Class Headers ----------------
#include "FairTask.h"

// Collaborating Class Headers -------
#include <map>
#include "TString.h"

// Collaborating Class Declarations --
class TClonesArray;
class GFRecoHitFactory;
class TGeoManager; 

class TBKalmanTask : public FairTask {
public:

  // Constructors/Destructors ---------
  TBKalmanTask();
  ~TBKalmanTask();
  
  // Operators
  
  
  // Accessors -----------------------
  // Modifiers -----------------------
  void SetTrackBranchName(const TString& name) {_trackBranchName=name;}
  void SetOutBranchName(const TString& name) {_outBranchName=name;}
  void SetSIBranchName(const TString& name) {fSIBranchName=name;}
  void SetGEMBranchName(const TString& name) {fGEMBranchName=name;}
  void SetPersistence(Bool_t opt=kTRUE) {_persistence=opt;}
  void SetLazy(Bool_t opt){_lazy=opt;}
  void SetNumIterations(Int_t it){_numIt=it;}

  // Operations ----------------------
  
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  
private:

  // Private Data Members ------------
  TClonesArray* _trackArray;
  TClonesArray* _trackOutArray;
  TString _trackBranchName;
  TString _outBranchName;
  TString fSIBranchName;
  TString fGEMBranchName;
  
  Bool_t _persistence;
  Bool_t _lazy;
  Int_t _numIt; // number of fitting iterations (back and forth)
  
  GFRecoHitFactory* _theRecoHitFactory;
  
  int _trackcount;
  
  // Private Methods -----------------

 public:
  ClassDef(TBKalmanTask,1)
    
};

#endif

//--------------------------------------------------------------
// $Log$
//--------------------------------------------------------------