// *************************************************************************
// Author: Ralf Kliemt r.kliemt(at)gsi(dot)de
//   
// ideal tracker using MC id & track info
// gaussian momentum smearing & fake efficiency included
//
// Created: 20.03.2013
// Modified:
//
// *************************************************************************

#ifndef STTMVDGEMFTSTRACKINGIDEAL
#define STTMVDGEMFTSTRACKINGIDEAL

#include "TVector3.h"
#include "TDatabasePDG.h"
#include "FairTask.h"
#include <cmath>

class PndSttMvdGemFtsTrackingIdeal : public FairTask {
  
public:
  
  PndSttMvdGemFtsTrackingIdeal();
  virtual ~PndSttMvdGemFtsTrackingIdeal();
  
  virtual void Exec(Option_t * option);
  virtual InitStatus Init();              
  virtual void Finish(); 
  
  void Reset();                    
  void Register();                 
  
  void SetTrackOutput(TString name = "STTTrkIdeal") { fTracksArrayName = name; };
  
  void SetMomentumSmearing(Double_t sigmax = -1.,Double_t sigmay = -1.,Double_t sigmaz = -1.) { fMomSigma.SetXYZ(fabs(sigmax),fabs(sigmay),fabs(sigmaz)); fRelative=kFALSE; }; // in GeV
  void SetRelativeMomentumSmearing(Double_t dpop = -1.) { fDPoP=fabs(dpop); fRelative=kTRUE;}; // in GeV
  void SetVertexSmearing(Double_t sigmax = -1.,Double_t sigmay = -1.,Double_t sigmaz = -1.) { fVtxSigma.SetXYZ(fabs(sigmax),fabs(sigmay),fabs(sigmaz)); }; // in cm
  void SetTrackingEfficiency(Double_t eff = 1.) { fEfficiency=eff; };
  
  void SetSttActivity(Bool_t act=kTRUE){fBranchActive[0]=act;}
  void SetGemActivity(Bool_t act=kTRUE){fBranchActive[1]=act;}
  void SetMvdActivity(Bool_t act=kTRUE){fBranchActive[2]=act;fBranchActive[3]=act;}
  void SetFtsActivity(Bool_t act=kTRUE){fBranchActive[4]=act;}
  
protected:
  
  void SmearVector(TVector3 &vec, const TVector3 &sigma); // smearing with doubled sigma in z direction
  
  TClonesArray*  fMCTracks;         //! Array of PndMCTrack
  TClonesArray*  fMCPoints[5];      //! Array of event's points
  TClonesArray*  fHits[5];          //! Array of event's hits
  Int_t          fBranchIDs[5];     //! Array of Branch IDs
  Bool_t         fBranchActive[5];  //! Array of Branch Activeness
  
  TClonesArray  *fTrackCands;   //! Array of found track candidates
  TClonesArray  *fTracks;       //! Array of found tracks
  TClonesArray  *fTrackIds;     //! Array of track IDs (Links)

  // Parameters for fake tracking
  TVector3 fMomSigma;          // Momentum smearing sigma [GeV]
  Double_t fDPoP;              // Relative momentum Smearing
  Bool_t fRelative;            // falg
  TVector3 fVtxSigma;          // Vertex smearing sigma [cm]
  Double_t fEfficiency;        // Tracking efficiency - if (0 <= e < 1), some tracks will be discarded
  
  TString fTracksArrayName;     // Branch name where to store the Track candidates
  TDatabasePDG *pdg;            //! Particle DB
  
  PndSttMvdGemFtsTrackingIdeal(const  PndSttMvdGemFtsTrackingIdeal& L);
  PndSttMvdGemFtsTrackingIdeal& operator= (const  PndSttMvdGemFtsTrackingIdeal&) {return *this;}
  
  int eventcount;

  ClassDef(PndSttMvdGemFtsTrackingIdeal,2);
  
};
#endif