#include "operator_ext_cylinder.h"
#include "operator_cylinder.h"
#include "engine_ext_cylinder.h"

Operator_Ext_Cylinder::Operator_Ext_Cylinder(Operator_Cylinder* op) : Operator_Extension(op)
{
	m_Op_Cyl = op;

	CC_R0_included=m_Op_Cyl->GetR0Included();
	CC_closedAlpha=m_Op_Cyl->GetClosedAlpha();

	vv_R0 = NULL;
	vi_R0 = NULL;
}

Operator_Ext_Cylinder::~Operator_Ext_Cylinder()
{
	delete[] vv_R0;vv_R0=NULL;
	delete[] vi_R0;vi_R0=NULL;
}

bool Operator_Ext_Cylinder::BuildExtension()
{
	delete[] vv_R0;vv_R0=NULL;
	delete[] vi_R0;vi_R0=NULL;

	//if r=0 is not included -> obviously no special treatment for r=0
	//if alpha direction is not closed, PEC-BC at r=0 necessary and already set...
	if ((CC_R0_included==false) || (CC_closedAlpha==false))
		return true;

	vv_R0 = new FDTD_FLOAT[m_Op->GetNumberOfLines(2)];
	vi_R0 = new FDTD_FLOAT[m_Op->GetNumberOfLines(2)];

	unsigned int pos[3];
	double inEC[4];
	double dT = m_Op->GetTimestep();
	pos[0]=0;
	for (pos[2]=0;pos[2]<m_Op->GetNumberOfLines(2);++pos[2])
	{
		double C=0;
		double G=0;
		for (pos[1]=0;pos[1]<m_Op->GetNumberOfLines(1)-1;++pos[1])
		{
			m_Op_Cyl->Calc_ECPos(2,pos,inEC);
			C+=inEC[0]*0.5;
			G+=inEC[1]*0.5;
		}
		m_Op->GetVV(2,0,0,pos[2]) = 1;
		vv_R0[pos[2]] = (1-dT*G/2/C)/(1+dT*G/2/C);
		vi_R0[pos[2]] = (dT/C)/(1+dT*G/2/C);
	}
	return true;
}

Engine_Extension* Operator_Ext_Cylinder::CreateEngineExtention()
{
	Engine_Ext_Cylinder* eng_ext = new Engine_Ext_Cylinder(this);
	return eng_ext;
}