/*
 * PndSTTStrawMap.cxx
 *
 *  Created on: Apr 16, 2013
 *      Author: mertens
 */


#include "PndSttStrawMap.h"
#include "TClonesArray.h"
#include "PndSttTube.h"
#include "TMath.h"

#include <iostream>
#include <vector>

using std::cout;
using std::endl;
using std::vector;

PndSttStrawMap::PndSttStrawMap()
: fStrawMapInitialized(false)
{
	fSectorStart.push_back(1.57);
	fSectorEnd.push_back(2.62);
	fSectorStart.push_back(2.62);
	fSectorEnd.push_back(-2.62+2*TMath::Pi());
	fSectorStart.push_back(-2.62+2*TMath::Pi());
	fSectorEnd.push_back(-1.57+2*TMath::Pi());
	fSectorStart.push_back(-1.57+2*TMath::Pi());
	fSectorEnd.push_back(-0.52+2*TMath::Pi());
	fSectorStart.push_back(-0.52+2*TMath::Pi());
	fSectorEnd.push_back(0.52);
	fSectorStart.push_back(0.52);
	fSectorEnd.push_back(1.57);
}

void PndSttStrawMap::GenerateStrawMap(const TClonesArray* const stt_tube_array)
{
	vector<int> currentRow;
	int sector = 0;
	int lastsector = 0;
	int row = 0;
	fSectorOfStraw.push_back(-1);
	fRowOfStraw.push_back(-1);
	cout << "Generating straw map for " << stt_tube_array->GetEntriesFast() << " straws." << endl;
	for (int i = 1; i < stt_tube_array->GetEntriesFast(); i++) {
		cout << "Finding tube." << endl;
		PndSttTube* tube = (PndSttTube*)(stt_tube_array->At(i));
		cout << "Tube address: " << tube << endl;
		double phi = tube->GetPosition().Phi();
		cout << "Phi: " << phi << endl;
		if (phi < 0) phi += 2*TMath::Pi();
		cout << "Checking sector." << endl;
		while ( !( (phi > fSectorStart.at(sector)) && (phi < fSectorEnd.at(sector)) )
				&& !( (sector == 4) && ( (phi > fSectorStart.at(4)) || (phi < fSectorEnd.at(4)) ) ) ) {
			sector++;
			sector %= 6;
		}
		if (sector != lastsector) {
			fStrawIndex[lastsector].push_back(currentRow);
			currentRow.clear();
			cout << "SECTOR COMPLETE: Row " << row << " added to sector " << lastsector << endl;
		}
		if (sector < lastsector) row++;
		lastsector = sector;
		currentRow.push_back(i);
		fSectorOfStraw.push_back(sector);
		fRowOfStraw.push_back(row);
		cout << "Straw " << i << " added to " << sector << ", " << row << endl;
	}
	fStrawIndex[lastsector].push_back(currentRow);
	cout << "STT COMPLETE. SECTOR COMPLETE: Row " << row << " added to sector " << lastsector << endl;

	fStrawMapInitialized = true;
}

int PndSttStrawMap::FindPhiSector(double phi) const
{
	int sector = 0;
	if (phi < 0) phi += 2*TMath::Pi();
	while ( !( (phi > fSectorStart.at(sector)) && (phi < fSectorEnd.at(sector)) )
			&& !( (sector == 4) && ( (phi > fSectorStart.at(4)) || (phi < fSectorEnd.at(4)) ) ) ) {
		sector++;
		sector %= 6;
	}
	return sector;
}

bool PndSttStrawMap::IsEdgeStraw(int strawindex) const
{
	if (GetRow(strawindex) == 0) return true;
	int endstraw = 0; //there is no straw index 0
	switch (GetSector(strawindex)) {
	case 0: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).front();
	break;
	case 2: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).back();
	break;
	case 3: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).front();
	break;
	case 5: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).back();
	break;
	default: return false;
	};
	return (strawindex == endstraw);
}

int PndSttStrawMap::IsSectorBorderStraw(int strawindex) const
{
	int endstrawcw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).front();
	int endstrawccw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).back();
	if (strawindex == endstrawcw) return -1;
	if (strawindex == endstrawccw) return 1;
	return 0;
}

ClassImp(PndSttStrawMap)