//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Pattern recognition = track finding in the 2 Silicon station
//                            of the TestBench and adding GEM clusters
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Maxence Vandenbroucke TUM            (original author)
//
//
//-----------------------------------------------------------

#ifndef TBTRACK_HH
#define TBTRACK_HH

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

// Collaborating Class Headers -------
#include <ostream> // remove if you do not need streaming op

// Collaborating Class Declarations --
#include "TpcRiemannTrack.h"
#include "TBSICluster.h"
#include "TBGEMCluster.h"
#include "FairGeanePro.h"
class TClonesArray;
class TpcFrontend;
class GFRecoHitFactory;
class TH1I;
class TH1D;
class TBGEMCluster;
class TBSICluster;

class TBTrackInitTask : public FairTask {
public:

  // Constructors/Destructors ---------
  TBTrackInitTask();
  ~TBTrackInitTask();

  // Operators
  
  // Accessors -----------------------


  // Modifiers -----------------------
  void SetSIClusterBranchName(const TString& name) {_clusterSIBranchName=name;}
  void SetGEMClusterBranchName(const TString& name) {_clusterGEMBranchName=name;}
  void SetRiemannBranchName(const TString& name) {_riemannBranchName=name;}
  void SetOutBranchNames(const TString& tracks)
  {_trackBranchName=tracks;}
  
  void SetPersistence(Bool_t opt=kTRUE) {_persistence=opt;} // store GFTracks with GFTrackCands  
  
  // Operations ----------------------
  virtual InitStatus Init();
  
  virtual void SetParContainers();
  
  virtual void Exec(Option_t* opt);
  
  void SetPDG(int pdg) {_pdg=pdg;} // use hypothesis
  void SetDefMomemtum(double mom) {fdefmom=mom;} // use default momentum
  void SetSmoothing(bool s=true) {_smoothing=s;} // use smoothing for the Kalman
  void SetDebug(Bool_t s=true) {fDEBUG=s;} // use smoothing for the Kalman
  
 private:
  
  // Private Data Members ------------
  TString _clusterGEMBranchName;
  TString _clusterSIBranchName;
  TString _riemannBranchName;
  TString _trackBranchName;
  

  TClonesArray* _clusterGEMArray;
  TClonesArray* _clusterSIArray;
  TClonesArray* _trackArray;
  TClonesArray* _riemannTrackArray;
  TClonesArray* _riemannHitArray;
  
  Bool_t _persistence;
  Bool_t fDEBUG;
  int _pdg;
  double fdefmom;
  double Bz;
  
  GFRecoHitFactory* _theRecoHitFactory;
  
  bool _smoothing;
  
  int counter;
  // Private Methods -----------------

public:
  ClassDef(TBTrackInitTask,1)

};

#endif

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