// -------------------------------------------------------------------------
// -----                    CbmTrdSetTracksPidWkn header file                -----
// -----                  Created 13/02/2007  by F. Uhlig              -----
// -------------------------------------------------------------------------


/** CbmTrdSetTracksPidWkn
 *@author F.Uhlig <F.Uhlig@gsi.de>
 **
 ** Task class for PID calculation for tracks using the Wkn method.
 ** Input: TClonesArray of CbmTrdTrack
 ** Parameters of these objects are updated
 **
 **/


#ifndef CBMTRDSETTRACKSPIDWKN
#define CBMTRDSETTRACKSPIDWKN 1


#include "CbmTask.h"

class CbmTrdSetTracksPidWkn : public CbmTask
{

 public:

  /** Default constructor **/
  CbmTrdSetTracksPidWkn();


  /** Standard constructor 
   **
   *@param name   Name of class
   *@param title  Task title
   **/
  CbmTrdSetTracksPidWkn(const char* name, const char* title = "CbmTask");


  /** Destructor **/
  virtual ~CbmTrdSetTracksPidWkn();


  /** Initialisation at beginning of each event **/
  virtual InitStatus Init();


  /** Task execution **/
  virtual void Exec(Option_t* opt);


  /** Finish at the end of each event **/
  virtual void Finish();


  /** SetParContainers **/
  virtual void SetParContainers();


  /** Accessors **/
  Int_t GetNofTracks()           { return fNofTracks; };
  Double_t GetEloss_const1()        { return Eloss_const1; };
  Double_t GetEloss_const2()        { return Eloss_const2; };
  Double_t GetEloss_const3()        { return Eloss_const3; };
  Double_t GetEloss_const4()        { return Eloss_const4; };

  /** Setters **/
  void SetEloss_const1(Double_t eloss) { Eloss_const1=eloss; };
  void SetEloss_const2(Double_t eloss) { Eloss_const2=eloss; };
  void SetEloss_const3(Double_t eloss) { Eloss_const3=eloss; };
  void SetEloss_const4(Double_t eloss) { Eloss_const4=eloss; };



 private:

  TClonesArray* fTrackArray;     // Input array of TRD tracks
  TClonesArray* fTrdHitArray;     // Input array of TRD Hits
  Int_t fNofTracks;              // Number of tracks successfully fitted
  Double_t Eloss_const1;         // Parameter1 for Wkn method
  Double_t Eloss_const2;         // Parameter2 for Wkn method
  Double_t Eloss_const3;         // Parameter3 for Wkn method
  Double_t Eloss_const4;         // Parameter4 for Wkn method

  Double_t CbmWknStat(Double_t* Y, Int_t k, Int_t n);

  ClassDef(CbmTrdSetTracksPidWkn,1);

};

#endif