// A basic clusterfinder for pixel detectors
#include "PndMvdTopixClusterFinder.h"
#include "TMath.h"

#include <iostream>

#include "FairMQLogger.h"

std::vector<std::vector<Int_t> > PndMvdTopixClusterFinder::GetClusters(std::vector<PndSdsDigiTopix4>& hits) {
//	fHits = hits;

//	LOG(INFO) << "HitsIn Cluster Finder!" << std::endl;
	std::vector<Int_t> posHits;
	for (UInt_t i = 0; i < hits.size(); i++){
//		LOG(INFO) << i << " : " << hits[i];
//		LOG(INFO) << i << " : " << hits[i].GetTimeStamp();
		posHits.push_back(i);
	}
	std::vector<std::vector<Int_t> > result;
	Int_t sizeTempHits = posHits.size();
	//Int_t actHit = 0;
	while (sizeTempHits != 0) {
		std::vector<Int_t> tempInt;
		tempInt.push_back(MoveHit(&posHits, 0));
		result.push_back(tempInt);
		Int_t sizeResultI = (result.end() - 1)->size(); // size of last vector in array
		for (Int_t i = 0; i < sizeResultI; i++) {
			sizeTempHits = posHits.size();
			for (Int_t j = 0; j < sizeTempHits; j++) {
				if (hits[(*(result.end() - 1))[i]].GetSensorID() == hits[posHits[j]].GetSensorID()) {
					if (IsInRange(&hits[(*(result.end() - 1))[i]], &hits[posHits[j]])) {
						(result.end() - 1)->push_back(MoveHit(&posHits, j));
						j--;
					}
				}
				sizeTempHits = posHits.size();
			}
			sizeResultI = (result.end() - 1)->size();
		}
	}
	return result;
}

Int_t PndMvdTopixClusterFinder::MoveHit(std::vector<Int_t>* hitVector, Int_t index) const
{
	Int_t result = -1;
	if (index < (Int_t)hitVector->size()){
		result = (*hitVector)[index];
		hitVector->erase(hitVector->begin()+index);
	}
	return result;
}

bool PndMvdTopixClusterFinder::IsInRange(PndSdsDigiPixel* hit1, PndSdsDigiPixel* hit2) const
{
	Double_t result1, result2;
	Int_t col1 = hit1->GetPixelColumn() + (Int_t)((hit1->GetFE()%10) * fcols);
	Int_t col2 = hit2->GetPixelColumn() + (Int_t)((hit2->GetFE()%10) * fcols);
	Int_t row1 = hit1->GetPixelRow()    + (Int_t)((hit1->GetFE()/10) * frows);
	Int_t row2 = hit2->GetPixelRow()    + (Int_t)((hit2->GetFE()/10) * frows);

	result1 = (col1-col2);
	result1 *= result1;
	result2 = (row1-row2);
	result2 *= result2;
	result1 += result2;

	return (TMath::Sqrt(result1) < fradius);
}