/*
 * PndMQHitsEventBuilder.cxx
 *
 *  Created on: 21.10.2015
 *      Author: Stockmanns
 */

#include <PndMQHitsEventBuilder.h>

#include "FairMQLogger.h"

PndMQHitsEventBuilder::PndMQHitsEventBuilder(int nChannels) : fNChannels(nChannels), fEventCounter(0), fSensorsInEvent(nChannels, 0) {
	 fInputData.resize(nChannels);

}

PndMQHitsEventBuilder::~PndMQHitsEventBuilder() {
	// TODO Auto-generated destructor stub
}

void PndMQHitsEventBuilder::AddData(vector<vector<vector<PndSdsHit> > >& data)
{
	for (int channelNr = 0; channelNr < data.size(); channelNr++){
		//LOG(INFO) << "PndMQHitsEventBuilder::AddData from Channel " << channelNr << " size " << data[channelNr].size();
		fInputData[channelNr].insert(fInputData[channelNr].end(), data[channelNr].begin(), data[channelNr].end());
	}
}

vector<vector<PndSdsHit> > PndMQHitsEventBuilder::GetEvents()
{
	vector<vector<PndSdsHit> > result;
	while (HasData()){
		result.push_back(GetNextEvent());
	}
	return result;
}

bool PndMQHitsEventBuilder::HasData()
{
	bool result = true;
	for (int channelIter = 0; channelIter < fInputData.size(); channelIter++){
		if (fInputData[channelIter].size() == 0){
			result = false;
		}
	}
//	if (result == false)
//		LOG(INFO) << "HasData " << result;
	return result;
}

vector<PndSdsHit> PndMQHitsEventBuilder::GetNextEvent()
{
	vector<PndSdsHit> result;
//	if (fEventCounter % 1000 == 0){
//		LOG(INFO) << fEventCounter;
//		for (int channelNr = 0; channelNr < fNChannels; channelNr++){
//			LOG(INFO) << channelNr << " : " << fInputData[channelNr].size();
//		}
//	}
//	fEventCounter++;
//	LOG(INFO) << "Data in channels";
//	for (auto channelIter: fInputData){
//		LOG(INFO) << "Channel size: " << channelIter.size();
//		for (auto eventIter : channelIter)
//			for (auto dataIter : eventIter)
//				LOG(INFO) << dataIter.GetSensorID() << " " << dataIter.GetTimeStamp();
//	}

	vector<vector<PndSdsHit> > firstDataInChannels;
	firstDataInChannels.resize(fNChannels);
	for (int channelNr = 0; channelNr < fInputData.size(); channelNr++){
		if (fInputData[channelNr].size() > 0){
			if (fInputData[channelNr][0].size() > 0)
				firstDataInChannels[channelNr] = fInputData[channelNr][0];
			else
				return result;
		}
	}
//	LOG(INFO) << "FirstDataInChannel:";
//	for (int channel = 0; channel < firstDataInChannels.size(); channel++){
//		//LOG(INFO) << "Channel: " << channel << " : ";
//		for (auto data : firstDataInChannels[channel]){
//			LOG(INFO) << channel << " : " << TString::Format("%12.0f", data.GetTimeStamp()).Data();
//		}
//	}
	vector<bool> channelInEvent = GetChannelsInEvent(firstDataInChannels);

	int nSensorsInEvent = 0;
//	LOG(INFO) << "ChannelMatch:";
	for (auto inEvent : channelInEvent){
		//LOG(INFO) << inEvent;
		nSensorsInEvent += inEvent;
	}

	if (nSensorsInEvent > 0)
		fSensorsInEvent[nSensorsInEvent - 1]++;

//	LOG(INFO) << "channelInEvent.size " << channelInEvent.size();
	for(int channelNr = 0; channelNr < fNChannels; channelNr++){
		if (channelInEvent[channelNr] == true){
			//LOG(INFO) << "Add Data to result from: " << channelNr;
			result.insert(result.end(), firstDataInChannels[channelNr].begin(), firstDataInChannels[channelNr].end());
			//LOG(INFO) << "DataInserted: " << result.size();
			fInputData[channelNr].erase(fInputData[channelNr].begin());
//			LOG(INFO) << "Delete data from InputData " << channelNr << " " << fInputData[channelNr].size();
		}
	}
	//if (nChannelsInEvent > 2){
	//	LOG(INFO) << "Data in Result: ";
	//	for (auto data : result)
	//		LOG(INFO) << data.GetSensorID() << " : " << data.GetTimeStamp();
	//}
	return result;
}

vector<bool> PndMQHitsEventBuilder::GetChannelsInEvent(vector<vector<PndSdsHit> >& eventData)
{
	double offset = 50;
	vector<bool> result(fNChannels, false);

	int firstChannel = FindFirstChannel(eventData);
//	LOG(INFO) << "FirstChannel: " << firstChannel;
	if (firstChannel > -1){
		double firstTimeStamp = eventData[firstChannel][0].GetTimeStamp();
		double lastTimeStamp = eventData[firstChannel].rbegin()->GetTimeStamp() + offset;
		result[firstChannel] = true;

		for (int channelNr = 0; channelNr < eventData.size(); channelNr++){
			if (channelNr != firstChannel){
				double actualFirstTS = eventData[channelNr].begin()->GetTimeStamp();
				double actualLastTS = eventData[channelNr].rbegin()->GetTimeStamp();
				if (actualFirstTS == firstTimeStamp)
					result[channelNr] = true;
				if (actualFirstTS < firstTimeStamp){
					if (actualLastTS >= firstTimeStamp){
						result[channelNr] = true;
					}
				} else if (actualFirstTS <= lastTimeStamp) {
					result[channelNr] = true;
				}
			}
		}
	}
	return result;
}

int PndMQHitsEventBuilder::FindFirstChannel(vector<vector<PndSdsHit> >& eventData)
{
	int result = -1;
	double oldTimeStamp = -1;
	for (int channelNr = 0; channelNr < eventData.size(); channelNr++){
		double actualTimeStamp = eventData[channelNr].front().GetTimeStamp();
		if (actualTimeStamp < oldTimeStamp || oldTimeStamp < 0){
			result = channelNr;
			oldTimeStamp = actualTimeStamp;
		}
	}
	return result;
}