#ifndef DataTreeCbmInterface_H
#define DataTreeCbmInterface_H 1

#include "CbmKFPartEfficiencies.h"
#include "CbmKFParticleFinder.h"

#include "FairTask.h"
#include "CbmVertex.h"
#include <vector>
#include "TLorentzVector.h"
#include <map>
#include <cstring>

#include <TGeoManager.h>
#include "TGraphErrors.h"
#include <iostream>
#include "TClonesArray.h"
#include "TProfile.h"
#include "TH2F.h"
#include "TCanvas.h"
#include "TFile.h"
#include "FairMCEventHeader.h"

#include "CbmKFVertex.h"
#include "CbmTrackMatch.h"

class DataTreeEvent;
class TClonesArray;
class CbmVertex;
class TDirectory;
class TH1F;
class TProfile;
class TH2F;


class DataTreeCbmInterface : public FairTask
{

public:

  DataTreeCbmInterface();
  ~DataTreeCbmInterface();

  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  virtual void Finish();

  void SetOutputFile(const TString filename) { fOutputFileName = filename; }

  void LoadGeo(const TString &geoFile);

  void SetKFParticleFinderTOF(const CbmKFParticleFinder* finder) { fFinderTOF = finder; }
  void SetKFParticleFinderMC(const CbmKFParticleFinder* finder) { fFinderMC = finder; }

private:

  void InitInput();
  void InitOutput();
  void InitOutputTree();
  void InitDataTreeEvent();
  void ClearEvent();
  void ReadEvent();
  void ReadPSD();
  void ReadTracks();
  void LinkSTS();
  void ReadTOF();
  void ReadMC();
  int GetMCTrackMatch(const int idx);
  void ReadV0(const int UseMCpid = 0);
  void ReadPsdPrimaryParticles();

  TString fOutputFileName {""};
  TFile* fTreeFile {nullptr};
  TTree* fDataTree {nullptr};

  CbmVertex* fPrimVtx {nullptr};
  FairMCEventHeader* fHeader {nullptr};
  TClonesArray* flistPSDhit {nullptr};
  TClonesArray* flistPSDdigit {nullptr};
  TClonesArray* flistMCtrack {nullptr};
  TClonesArray* flistSTSRECOtrack {nullptr};
  TClonesArray* flistSTStrackMATCH {nullptr};
  TClonesArray *fGlobalTrackArray {nullptr};
  TClonesArray *fTofHitArray {nullptr};
  TClonesArray *fTofHitMatchArray {nullptr};
  TClonesArray *fPsdPointArray {nullptr};

  DataTreeEvent* fDTEvent {nullptr};

  int fPsdModules{0};
  TVector3 fPsdPosition;
  std::map <int, TVector3 > fPsdModulePositions;


  std::vector<int> fMCTrackIDs;
  std::vector<int> fTrackIDs;

  const CbmKFParticleFinder* fFinderTOF{nullptr};
  const CbmKFParticleFinder* fFinderMC{nullptr};

  ClassDef(DataTreeCbmInterface, 1)
};

#endif