// -------------------------------------------------------------------------
// -----                CbmD0CandidateSelection  header file           -----
// -----                  Created 09/02/07  by C. Dritsa               -----
// -----                    Update 06/15  by P.Sitzmann                -----
// -------------------------------------------------------------------------


/**  CbmD0CandidateSelection.h
 *@author C.Dritsa <c.dritsa@gsi.de>
 * update 2015 P.Sitzmann <p.sitzmann@gsi.de>
 *
 *  Selection of candidate pairs
 *  for D0 reconstruction
 *
 **/

#ifndef CBMD0CANDIDATESELECTION_H
#define CBMD0CANDIDATESELECTION_H

#include "FairTask.h"
#include "CbmMCTrack.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TString.h"
#include "TNtuple.h"
#include "FairField.h"

class TClonesArray;
class CbmVertex;
class FairTrackParam;
class CbmD0Tools;
class KFParticle;

class CbmD0CandidateSelection : public FairTask
{
 public:
  CbmD0CandidateSelection();
  CbmD0CandidateSelection(char* name, Int_t iVerbose, Double_t cutIPD, Double_t cutSVZ);

  ~CbmD0CandidateSelection();

  InitStatus Init();

  void Exec(Option_t* option);                                   
  void SetTestMode ( Bool_t testMode ) { bTestMode = testMode; };
  void Finish();

private:
    Int_t fEventNumber;
    TClonesArray* fKaonParticleArray;
    TClonesArray* fPionParticleArray;
    TClonesArray* fKaonTrackArray;
    TClonesArray* fPionTrackArray;
    TClonesArray* fD0Candidates;
    TClonesArray* fD0KFCandidates;
    TClonesArray* fListMCTracksPos;
    TClonesArray* fListMCTracksNeg;
    TClonesArray* fListMCTracks;
    TClonesArray* fListD0TrackCandidate;
    TClonesArray* fD0TrackArray;
    TClonesArray* fStsTrackMatches;
    TClonesArray* fPosStsTracks;
    TClonesArray* fNegStsTracks;
    CbmVertex* fPrimVtx;
    float fvtx[3];
    CbmD0Tools* tools;
    FairField* fMF;
    Bool_t bTestMode;
    Bool_t f_particleIsMCD0;
    Double_t fcutIPD0;
    Double_t fcutSVZ;

    /** Register the output arrays to the IOManager **/
    void Register();
    void CheckIfParticleIsD0(Int_t iNeg, Int_t iPos);
    Double_t GetPairImpactParameterR(KFParticle* particle );
    //Double_t GetPairPt(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2);
    //Double_t GetPairPz(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2);
    //Double_t GetPairTx(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2);
    //Double_t GetPairTy(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2);
    //Double_t GetMomentum(CbmD0TrackCandidate* tr1);
    //Double_t GetPairMomentum(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2);
    //Double_t GetEnergy(CbmD0TrackCandidate* tr1);
    //Double_t GetPairEnergy( CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2 );
    //Double_t GetIM(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2);
    //Double_t GetCos12(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2 );
    //Double_t GetIPAngle(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2 );
    //Double_t GetChi2Topo(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2 );
    //Double_t GetAPalpha(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2 );
    //Double_t GetAPptt(CbmD0TrackCandidate* tr1, CbmD0TrackCandidate* tr2 );
    void SetCuts(Double_t ipD0, Double_t SVZ);


    CbmD0CandidateSelection(const CbmD0CandidateSelection&);
    CbmD0CandidateSelection& operator=(const CbmD0CandidateSelection&);

  ClassDef(CbmD0CandidateSelection,1);
};

#endif