#include "DetectorLayoutReader.h"

#include <fstream>

DetectorLayoutReader::DetectorLayoutReader():
	fGeometryFileName("data/sa_geometry.txt")
{

}

DetectorLayoutReader::~DetectorLayoutReader()
{

}

void DetectorLayoutReader::GetDetectorGeometryScal(
		LitDetectorLayoutMuonScal& layout) const
{
	GetDetectorGeometry<fscal>(layout);
}

void DetectorLayoutReader::GetDetectorGeometryVec(
		LitDetectorLayoutMuonVec& layout) const
{
	GetDetectorGeometry<fvec>(layout);
}

template<class T>
void DetectorLayoutReader::GetDetectorGeometry(
		LitDetectorLayoutMuon<T>& layout) const
{
	std::ifstream finLayout(fGeometryFileName.c_str(), std::ifstream::in);
	int nofStationGroups;
	finLayout >> nofStationGroups;
	layout.nofStationGroups = nofStationGroups;

	for (int isg = 0; isg < layout.nofStationGroups; isg++) {
		ReadStationGroup<T>(finLayout, layout.stationGroups[isg]);
	}

	finLayout.close();
}

template<class T>
void DetectorLayoutReader::ReadStationGroup(
		std::ifstream& fin,
		LitStationGroupMuon<T>& stationGroup) const
{
	int sgNo, nofStations;
	fin >> sgNo >> nofStations;
	stationGroup.nofStations = nofStations;
	for (int ist = 0; ist < stationGroup.nofStations; ist++) {
		ReadStation(fin, stationGroup.stations[ist]);
	}
	ReadAbsorber<T>(fin, stationGroup.absorber);
}

template<class T>
void DetectorLayoutReader::ReadStation(
		std::ifstream& fin,
		LitStationMuon<T>& station) const
{
	int stNo, nofSubstations;
	fin >> stNo >> nofSubstations;
	station.nofSubstations = nofSubstations;
	for (int iss = 0; iss < station.nofSubstations; iss++) {
		ReadSubstation<T>(fin, station.substations[iss]);
	}
}

template<class T>
void DetectorLayoutReader::ReadSubstation(
		std::ifstream& fin,
		LitSubstationMuon<T>& substation) const
{
	int ssNo;
	fin >> ssNo;
	fin >> substation.Z;
	ReadMaterial<T>(fin, substation.material);
	ReadFieldSlice<T>(fin, substation.fieldSlice);
}

template<class T>
void DetectorLayoutReader::ReadAbsorber(
		std::ifstream& fin,
		LitAbsorber<T>& absorber) const
{
	fin >> absorber.Z;
	ReadMaterial<T>(fin, absorber.material);
	ReadFieldSlice<T>(fin, absorber.fieldSliceFront);
	ReadFieldSlice<T>(fin, absorber.fieldSliceBack);
}

template<class T>
void DetectorLayoutReader::ReadMaterial(
		std::ifstream& fin,
		LitMaterialInfo<T>& material) const
{
	fin >> material.Thickness >> material.X0 >> material.Rho
		>> material.Z >> material.A >> material.Zpos >> material.I;
}

template<class T>
void DetectorLayoutReader::ReadFieldSlice(
		std::ifstream& fin,
		LitFieldSlice<T>& slice) const
{
	fin >> slice.Z;
	unsigned int N = slice.N;
	for(unsigned int i = 0; i < N; i++) fin >> slice.cx[i];
	for(unsigned int i = 0; i < N; i++) fin >> slice.cy[i];
	for(unsigned int i = 0; i < N; i++) fin >> slice.cz[i];
}