/**
* \file CbmAnaConversion.h
*
* \brief Optimization of the RICH geometry.
*
* \author Tariq Mahmoud<t.mahmoud@gsi.de>
* \date 2014
**/

#ifndef CBM_ANA_CONVERSION
#define CBM_ANA_CONVERSION

#include "FairTask.h"
#include "CbmMCTrack.h"
#include "../dielectron/CbmAnaDielectronTask.h" 
class TH1;
class TH2;
class TH3;
class TH1D;
class TH2D;
class TH2I;
class TH3D;
class TClonesArray;
class CbmRichRing;
class TCanvas;

#include <vector>
#include <map>

using namespace std;

/**
* \class CbmAnaConversion
*
* \brief Optimization of the RICH geometry.
*
* \author Tariq Mahmoud<t.mahmoud@gsi.de>
* \date 2014
**/

/*class KinematicParams{
public:
   Double_t momentumMag; // Absolute value of momentum
   Double_t pt; // Transverse momentum
   Double_t rapidity; // Rapidity
   Double_t minv; // Invariant mass
   Double_t angle; // Opening angle
};*/

class CbmAnaConversion : public FairTask
{

public:
   /**
    * \brief Standard constructor.
    */
	CbmAnaConversion();

   /**
    * \brief Standard destructor.
    */
   virtual ~CbmAnaConversion();

   /**
    * \brief Inherited from FairTask.
    */
   virtual InitStatus Init();

   /**
    * \brief Inherited from FairTask.
    */
   virtual void Exec(
		   Option_t* option);
		   
	KinematicParams CalculateKinematicParams(const CbmMCTrack* mctrackP, const CbmMCTrack* mctrackM);
        
	Double_t	Invmass_2gammas(const CbmMCTrack* gamma1, const CbmMCTrack* gamma2);
	Double_t	Invmass_gepem(const CbmMCTrack* mctrack1, const CbmMCTrack* mctrack2, const CbmMCTrack* mctrack3);
	Double_t	Invmass_2particles(const CbmMCTrack* mctrack1, const CbmMCTrack* mctrack2);
	Double_t	Invmass_4particles(const CbmMCTrack* mctrack1, const CbmMCTrack* mctrack2, const CbmMCTrack* mctrack3, const CbmMCTrack* mctrack4);
	Double_t	Invmass_4particlesRECO(const TVector3 part1, const TVector3 part2, const TVector3 part3, const TVector3 part4);


	void	CalculateInvMass_MC_2particles();

   /**
    * \brief Inherited from FairTask.
    */
   virtual void Finish();
   
   void 	TomographyMC(CbmMCTrack* mctrack);
   void 	TomographyReco(CbmMCTrack* mctrack);
   void 	Probability();
   void 	FillMCTracklist(CbmMCTrack* mctrack);
   void 	FillMCTracklist_all(CbmMCTrack* mctrack);
   void 	FillMCTracklist_omega(CbmMCTrack* mctrack);
   void 	FillRecoTracklist(CbmMCTrack* mtrack);
   void 	FillRecoTracklistEPEM(CbmMCTrack* mctrack, TVector3 stsMomentum);
   void 	InvariantMassTest();
   void 	InvariantMassTest_4epem();
   void 	InvariantMassTest_omega();
   void 	InvariantMassTestReco();
   int 		GetTest();
   int 		GetNofEvents();
   void		SetMode(int mode);



private:
   TH1D * fhGammaZ;
   TH1D * fhGammaZ_selected;
   TH2D * fTest;
   
   TH1D * fhNofElPrim;
   TH1D * fhNofElSec;
   TH1D * fhNofElAll;
   
   TH3D * fhTomography;
   TH2D * fhTomography_XZ;
   TH2D * fhTomography_YZ;
   TH2D * fhTomography_uptoRICH;
   TH2D * fhTomography_RICH_complete;
   TH2D * fhTomography_RICH_beampipe;
   TH2D * fhTomography_STS_end;
   TH2D * fhTomography_STS_lastStation;
   TH2D * fhTomography_RICH_frontplate;
   TH2D * fhTomography_RICH_backplate;
      
   TH3D * fhTomography_reco;
   TH2D * fhTomography_reco_XZ;
   TH2D * fhTomography_reco_YZ;
   
   TH1D * fhConversion;
   TH1D * fhConversion_prob;
   TH1D * fhConversion_energy;
   TH1D * fhConversion_p;
   
   TH1D * fhInvariantMass_test;
   TH1D * fhInvariantMass_test2;
   TH1D * fhInvariantMass_test3;
   TH1D * fhInvariantMassReco_test;
   TH1D * fhInvariantMassReco_test2;
   TH1D * fhInvariantMassReco_test3;
      
   TH1D * fhInvariantMass_MC_all;
   TH1D * fhInvariantMass_MC_omega;
   TH1D * fhInvariantMass_MC_pi0;
   
   TH1D * fhInvariantMassReco_pi0;
   
   TH2D * fhMomentum_MCvsReco;
   TH1D * fhMomentum_MCvsReco_diff;
   
   TH1D * fhInvariantMass_recoMomentum1;
   TH1D * fhInvariantMass_recoMomentum2;

   TClonesArray* fRichPoints;
   TClonesArray* fMcTracks;
   TClonesArray* fStsTracks;
   TClonesArray* fStsTrackMatches;
   TClonesArray* fGlobalTracks;

   Int_t fEventNum;
   
   Int_t test;
   
   int testint;
   
   Int_t fAnalyseMode;



   /**
    * \brief Initialize histograms.
    */
   void InitHistograms();
   
   vector<TH1*> fHistoList;				// list of all histograms
   vector<TH1*> fHistoList_MC;			// list of all histograms generated with MC data
   vector<TH1*> fHistoList_tomography;	// list of all histograms of tomography data (photon conversion)
   
   vector<CbmMCTrack*> fMCTracklist;
   vector<CbmMCTrack*> fMCTracklist_all;
   vector<CbmMCTrack*> fMCTracklist_omega;
   vector<CbmMCTrack*> fRecoTracklist;
   vector<CbmMCTrack*> fRecoTracklistEPEM;
   
   vector<TVector3> fRecoMomentum;


   /**
    * \brief Copy constructor.
    */
   CbmAnaConversion(const CbmAnaConversion&);

   /**
    * \brief Assignment operator.
    */
   CbmAnaConversion& operator=(const CbmAnaConversion&);


	ClassDef(CbmAnaConversion,1)
};

#endif