/*
*	Copyright (C) 2010 Thorsten Liebig (Thorsten.Liebig@gmx.de)
*
*	This program is free software: you can redistribute it and/or modify
*	it under the terms of the GNU General Public License as published by
*	the Free Software Foundation, either version 3 of the License, or
*	(at your option) any later version.
*
*	This program is distributed in the hope that it will be useful,
*	but WITHOUT ANY WARRANTY; without even the implied warranty of
*	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
*	GNU General Public License for more details.
*
*	You should have received a copy of the GNU General Public License
*	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "engine.h"
#include "engine_ext_cylinder.h"
#include "operator_ext_cylinder.h"

Engine_Ext_Cylinder::Engine_Ext_Cylinder(Operator_Ext_Cylinder* op_ext) : Engine_Extension(op_ext)
{
	cyl_Op = op_ext;

	CC_closedAlpha = op_ext->CC_closedAlpha;
	CC_R0_included = op_ext->CC_R0_included;

	for (int n=0;n<3;++n)
		numLines[n] = op_ext->m_Op->GetOriginalNumLines(n);
}

void Engine_Ext_Cylinder::Apply2Voltages()
{
	if (CC_closedAlpha==false) return;

	if (CC_R0_included)
	{
		unsigned int pos[3];
		pos[0] = 0;
		for (pos[2]=0;pos[2]<numLines[2];++pos[2])
		{
			m_Eng->GetVolt(2,0,0,pos[2]) *= cyl_Op->vv_R0[pos[2]];
			for (pos[1]=0;pos[1]<numLines[1]-1;++pos[1])
			{
					m_Eng->GetVolt(2,0,0,pos[2]) += cyl_Op->vi_R0[pos[2]] *  m_Eng->GetCurr(1,0,pos[1],pos[2]);
			}
		}
		for (pos[1]=0;pos[1]<numLines[1];++pos[1])
		{
			for (pos[2]=0;pos[2]<numLines[2];++pos[2])
			{
				m_Eng->GetVolt(1,0,pos[1],pos[2]) = 0; //no voltage in alpha-direction at r=0
				m_Eng->GetVolt(2,0,pos[1],pos[2]) = m_Eng->GetVolt(2,0,0,pos[2]);
			}
		}
	}

	//close alpha
	unsigned int pos[3];
	// copy voltages from last alpha-plane to first
	unsigned int last_A_Line = numLines[1]-1;
	for (pos[0]=0;pos[0]<numLines[0];++pos[0])
	{
		for (pos[2]=0;pos[2]<numLines[2];++pos[2])
		{
			m_Eng->GetVolt(0,pos[0],0,pos[2]) = m_Eng->GetVolt(0,pos[0],last_A_Line,pos[2]);
			m_Eng->GetVolt(1,pos[0],0,pos[2]) = m_Eng->GetVolt(1,pos[0],last_A_Line,pos[2]);
			m_Eng->GetVolt(2,pos[0],0,pos[2]) = m_Eng->GetVolt(2,pos[0],last_A_Line,pos[2]);
		}
	}
}

void Engine_Ext_Cylinder::Apply2Current()
{
	if (CC_closedAlpha==false) return;

	//close alpha
	unsigned int pos[3];
	// copy currents from first alpha-plane to last
	for (pos[0]=0;pos[0]<numLines[0]-1;++pos[0])
	{
		unsigned int last_A_Line = numLines[1]-1;
		for (pos[2]=0;pos[2]<numLines[2]-1;++pos[2])
		{
			m_Eng->GetCurr(0,pos[0],last_A_Line,pos[2]) = m_Eng->GetCurr(0,pos[0],0,pos[2]);
			m_Eng->GetCurr(1,pos[0],last_A_Line,pos[2]) = m_Eng->GetCurr(1,pos[0],0,pos[2]);
			m_Eng->GetCurr(2,pos[0],last_A_Line,pos[2]) = m_Eng->GetCurr(2,pos[0],0,pos[2]);
		}
	}
}