#ifndef RICH_CbmRichUtil
#define RICH_CbmRichUtil

#include "TObject.h"
#include "TH2.h"
#include "TCanvas.h"
#include <string>
#include "FairLogger.h"
#include "TClonesArray.h"
#include "FairRootManager.h"
#include "CbmGlobalTrack.h"
#include "FairTrackParam.h"

using namespace std;

class CbmRichUtil {
    
public:
    
	static Double_t GetRingTrackDistance(Int_t globalTrackId)
	{
		vector<Double_t> v = GetRingTrackDistanceImpl(globalTrackId);
		return v[0];
	}

	static Double_t GetRingTrackDistanceX(Int_t globalTrackId)
	{
		vector<Double_t> v = GetRingTrackDistanceImpl(globalTrackId);
		return v[1];
	}

	static Double_t GetRingTrackDistanceY(Int_t globalTrackId)
	{
		vector<Double_t> v = GetRingTrackDistanceImpl(globalTrackId);
		return v[2];
	}

private:

	/**
	 * \brief Return a vector with total distance and x, y components. [0] - total distance, [1] - x component, [2] - y component
	 */
	static vector<Double_t> GetRingTrackDistanceImpl(Int_t globalTrackId)
	{
		vector<Double_t> errorVec = {999., 999., 999.};
		FairRootManager* ioman = FairRootManager::Instance();
		if (ioman == NULL) return errorVec;
		// Do we really need static here, depends on ioman->GetObject() method
		static TClonesArray* globalTracks = NULL;
		static TClonesArray* richRings = NULL;
		static TClonesArray* richProjections = NULL;

		if (globalTracks == NULL || richRings == NULL || richProjections == NULL) {
			//cout << "globalTracks == NULL || richRings == NULL || richProjections == NULL" << endl;
			globalTracks = (TClonesArray*) ioman->GetObject("GlobalTrack");
			richRings = (TClonesArray*) ioman->GetObject("RichRing");
			richProjections = (TClonesArray*) ioman->GetObject("RichProjection");
		} else {
			//cout << "globalTracks, richRings, richProjections NOT NULL" << endl;
		}

		if (globalTracks == NULL || richRings == NULL || richProjections == NULL) {
			LOG(error) << "CbmRichUtil::GetRingTrackDistance globalTracks, richRings, richProjections NOT INITIALIZED" << endl;
			return errorVec;
		}

		const CbmGlobalTrack* globalTrack = static_cast<const CbmGlobalTrack*>(globalTracks->At(globalTrackId));
		if (globalTrack == NULL) return errorVec;

		Int_t stsId = globalTrack->GetStsTrackIndex();
		if (stsId < 0) return errorVec;

		const FairTrackParam* pTrack = static_cast<const FairTrackParam*>(richProjections->At(stsId));
		if (pTrack == NULL) return errorVec;

		if (pTrack->GetX() == 0 && pTrack->GetY() == 0) return errorVec;

		Int_t richId = globalTrack->GetRichRingIndex();
		if (richId < 0) return errorVec;

		const CbmRichRing* richRing = static_cast<const CbmRichRing*>(richRings->At(richId));
		if (richRing == NULL) return errorVec;

		Double_t xRing = richRing->GetCenterX();
		Double_t yRing = richRing->GetCenterY();
		Double_t dx =  richRing->GetCenterX() - pTrack->GetX();
		Double_t dy = richRing->GetCenterY() - pTrack->GetY();

		Double_t dist = TMath::Sqrt( dx * dx + dy * dy );

		vector<Double_t> v = {dist, dx, dy};
	}
    
};

#endif