#include "stdafx.h"

#include <jenga/include/smoothie/Smoothie.h>

#include <Compiler.h>

#include "../BasicCompiler_Common/common.h"
#include "Opcode.h"

void Call_DebugSys_SaveContext(){
	//call _System_GetEip
	extern const UserProc *pSub_System_GetEip;
	compiler.codeGenerator.op_call(pSub_System_GetEip);

	//push eax
	compiler.codeGenerator.op_push(REG_EAX);

	//push ebp
	compiler.codeGenerator.op_push(REG_EBP);

	//call _DebugSys_SaveContext
	extern const UserProc *pSub_DebugSys_SaveContext;
	compiler.codeGenerator.op_call(pSub_DebugSys_SaveContext);
}

bool Opcode_CallProcPtr( const char *variable, const char *lpszParms,ProcPointer *pProcPointer){

	extern BOOL bDebugCompile;
	extern BOOL bDebugSupportProc;
	if(bDebugCompile&&bDebugSupportProc==0)
		Call_DebugSys_SaveContext();


	////////////////////////
	// パラメータのセット
	////////////////////////

	//パラメータオブジェクトを生成
	ParamImpl *pobj_parameter=0;
	pobj_parameter=new ParamImpl(lpszParms);

	// デフォルト引数を適用
	pobj_parameter->ApplyDefaultParameters( pProcPointer->Params() );

	//エラーチェック
	if( !pobj_parameter->ErrorCheck(variable,pProcPointer->Params() ) ){
		//パラメータにエラーがあるときは処理を終える
		return false;
	}

	//一時オブジェクトを生成
	pobj_parameter->NewTempParameters( variable,pProcPointer->Params() );

	//レジスタ、スタックフレームにセット
	pobj_parameter->SetParameter(variable,pProcPointer->Params() );



	////////////////////////
	// call
	////////////////////////
	RELATIVE_VAR RelativeVar;
	GetVarOffsetReadOnly(variable,&RelativeVar,Type());
	SetVarPtrToEax(&RelativeVar);

	//mov eax,dword ptr[eax]
	compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EAX, REG_EAX, 0, MOD_BASE );

	//call eax
	compiler.codeGenerator.op_call_R( REG_EAX );



	//一時オブジェクトを破棄
	pobj_parameter->DeleteTempParameters();

	//パラメータオブジェクトを破棄
	delete pobj_parameter;

	return true;
}

bool Opcode_CallProc(const char *Parameter,const UserProc *pUserProc,DWORD dwFlags,const char *ObjectName ){
	int i2;

	if( pUserProc->IsMacro() ){
		if( lstrcmpi( pUserProc->GetName().c_str(), "Print" ) == 0 ){
			Opcode_Print(Parameter,0);
			return true;
		}
		if( lstrcmpi( pUserProc->GetName().c_str(), "Input" ) == 0 ){
			Opcode_Input(Parameter);
			return true;
		}
		if( lstrcmpi( pUserProc->GetName().c_str(), "Write" ) == 0 ){
			Opcode_Print(Parameter,1);
			return true;
		}
	}

	pUserProc->Using();

	bool isStatic = false;
	const CClass *pobj_c = NULL;
	const CMethod *pMethod = NULL;
	Type leftType;
	bool isFixedClass = false;
	if( pUserProc->GetParentClassPtr() ){
		//クラスのメンバ関数を呼び出す場合はアクセスチェックを行う
		if(ObjectName[0] && (dwFlags&PROCFLAG_NEW)==0)
		{
			if(lstrcmpi(ObjectName,"Super")==0)
			{
				//クラスメンバ関数内から基底クラスの呼び出し
				pobj_c=&compiler.pCompilingClass->GetSuperClass();

				isFixedClass = true;
			}
			else
			{
				//"->"によってオブジェクトを指定する通常のメンバ関数呼び出し
				Type varType;
				GetVarType( ObjectName, varType, false );
				if( NATURAL_TYPE( varType.GetBasicType() ) == DEF_OBJECT )
				{
					pobj_c = &varType.GetClass();
					leftType = varType;
				}
				else
				{
					pobj_c=compiler.GetObjectModule().meta.GetClasses().Find(ObjectName);
					if( pobj_c ){
						isStatic = true;
					}
					else{
						SetError(300,NULL,cp);
					}
				}
			}
		}
		else{
			if(dwFlags&PROCFLAG_NEW){
				GetVarType( ObjectName, leftType, false );

				//New演算子によるコンストラクタ呼び出し
				pobj_c=pUserProc->GetParentClassPtr();
			}
			else{
				//クラスメンバ関数内から同一クラスのメンバ関数の呼び出し
				pobj_c=compiler.pCompilingClass;
			}
		}


		/////////////////////////////////
		// メソッド情報を取得
		/////////////////////////////////
		pMethod = NULL;
		if( ! isStatic ) pMethod = pobj_c->GetMethods().GetMethodPtr( pUserProc );
		if( ! pMethod ){
			//動的メソッドが取得できなかったときは静的メソッドを当たる
			pMethod = pobj_c->GetStaticMethods().GetMethodPtr( pUserProc );
			if( !pMethod ){
				SetError(300,NULL,cp);
				return false;
			}

			//静的メンバ
			isStatic = true;
		}


		//////////////////////////////
		// アクセスエラーチェック
		//////////////////////////////

		if(ObjectName[0]){
			//外部からの呼び出し
			if(pobj_c==compiler.pCompilingClass){
				//同一クラスオブジェクトの場合はプライベートアクセスを容認する
				if( pMethod->IsNoneAccess() ){
					SetError(109,pUserProc->GetName(),cp);
					return false;
				}
			}
			else{
				if( pMethod->IsPrivate()
					|| pMethod->IsNoneAccess() ){
					SetError(109,pUserProc->GetName(),cp);
					return false;
				}
				if( !pMethod->GetUserProc().GetParentClass().IsEqualsOrSubClass( pobj_c ) && pMethod->IsProtected() ){
					SetError(110,pUserProc->GetName(),cp);
					return false;
				}
			}
		}
		else{
			//クラス内部からの呼び出し（継承によるACCESS_NONのみをエラーとする）
			if( pMethod->IsNoneAccess() ){
				SetError(109,pUserProc->GetName(),cp);
				return false;
			}
		}
	}


	///////////////////////////////////////////////////////////////
	// _System_LocalThisのダミーをセット
	///////////////////////////////////////////////////////////////

	char temporary[VN_SIZE]={0};
	if( pUserProc->GetParentClassPtr() && isStatic == false ){
		//_System_LocalThis（第一パラメータ）のダミーを作成
		lstrcpy(temporary,"0,");
	}

	if(Parameter[0]=='\0'&&temporary[0])
		temporary[lstrlen(temporary)-1]=0;
	else lstrcat(temporary,Parameter);


	////////////////////////
	// パラメータをセット
	////////////////////////

	//パラメータオブジェクトを生成
	ParamImpl *pobj_parameter=0;
	pobj_parameter=new ParamImpl(temporary);

	// デフォルト引数を適用
	pobj_parameter->ApplyDefaultParameters( pUserProc->RealParams() );

	// 型パラメータを適用
	pobj_parameter->SetLeftType( leftType );

	//エラーチェック
	if( !pobj_parameter->ErrorCheck(pUserProc->GetName(),pUserProc->RealParams(),pUserProc->GetSecondParmNum() ) ){
		//パラメータにエラーがあるときは処理を終える
		return false;
	}

	if(pUserProc->IsMacro()){
		//マクロ関数の場合は、パラメータ省略を考慮する
		pobj_parameter->MacroParameterSupport( pUserProc->RealParams() );
	}

	//一時オブジェクトを生成
	int tempSize = pobj_parameter->NewTempParameters( pUserProc->GetName(),pUserProc->RealParams(),pUserProc->GetRealSecondParmNum() );

	//レジスタ、スタックフレームにセット
	int ParmSize = pobj_parameter->SetParameter(pUserProc->GetName(),pUserProc->RealParams(),pUserProc->GetRealSecondParmNum(), pUserProc );

	if(pUserProc->ReturnType().IsStruct() ){
		//////////////////////////////////////////////////////
		// 戻り値に構造体インスタンスを持つ場合
		// ※ByRef _System_ReturnValue パラメータをセット
		//////////////////////////////////////////////////////

		int object_size = pUserProc->ReturnType().GetClass().GetSize();

		//push object_size
		compiler.codeGenerator.op_push_V(object_size);

		//call calloc
		extern const UserProc *pSub_calloc;
		compiler.codeGenerator.op_call(pSub_calloc);

		//push eax
		compiler.codeGenerator.op_push(REG_EAX);

		ParmSize += PTR_SIZE;
	}


	if( pUserProc->GetParentClassPtr() && isStatic == false ){
		//////////////////////////////////////////////////////
		// メンバ関数の場合
		// ※_System_LocalThis パラメータをセット
		//////////////////////////////////////////////////////

		if(ObjectName[0] && (dwFlags&PROCFLAG_NEW)==0){
			if(lstrcmpi(ObjectName,"Super")==0) goto InClassMember;
			else{
				RELATIVE_VAR RelativeVar;
				if( pMethod->IsConst() ){
					//Constアクセスが可能なメソッドの場合
					if( !GetVarOffsetReadOnly( ObjectName, &RelativeVar, Type() ) ){
						return false;
					}
				}
				else{
					//Constアクセスが不可能なメソッドの場合
					if( !GetVarOffsetReadWrite( ObjectName, &RelativeVar, Type() ) ){
						return false;
					}
				}

				SetVarPtrToEax(&RelativeVar);

				// 参照を実体ポインタにする
				compiler.codeGenerator.op_mov_RM( sizeof(long), REG_ECX, REG_EAX, 0, MOD_BASE );
			}
		}
		else{
InClassMember:
			if(dwFlags&PROCFLAG_NEW){
				//New演算子によるコンストラクタ呼び出しの場合

				//mov ecx,dword ptr[esp+ParmSize]
				compiler.codeGenerator.op_mov_RM( sizeof(long), REG_ECX, REG_ESP, ParmSize + tempSize, MOD_BASE_DISP32 );
			}
			else{
				//Thisポインタをecxにコピー
				SetThisPtrToReg(REG_ECX);
			}
		}

		//push ecx
		compiler.codeGenerator.op_push(REG_ECX);
	}

	if( pUserProc->IsVirtual() && !isFixedClass ){
		//仮想関数（オブジェクトメソッド）呼び出し
		//pObj->func_table->func1
		//                ->func2
		//                ->func3

		//mov edx,dword ptr[ecx]
		compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EDX, REG_ECX, 0, MOD_BASE );

		i2 = pobj_c->GetFuncNumInVtbl( pUserProc );

		//call dword ptr[edx+func_index]
		if(i2*PTR_SIZE<=0x7F){
			compiler.codeGenerator.PutOld(
				(char)0xFF,
				(char)0x52,
				(char)(i2*PTR_SIZE)
			);
		}
		else{
			compiler.codeGenerator.PutOld(
				(char)0xFF,
				(char)0x92
			);
			compiler.codeGenerator.PutOld( (long)(i2*PTR_SIZE), Schedule::None );
		}
	}
	else{
		//通常呼び出し

		//call ProcAddr
		compiler.codeGenerator.op_call(pUserProc);
	}

	if(pUserProc->IsCdecl()){
		//add esp,ParmSize
		compiler.codeGenerator.op_add_esp(ParmSize);
	}

	//一時オブジェクトを破棄
	pobj_parameter->DeleteTempParameters();

	//パラメータオブジェクトを破棄
	delete pobj_parameter;

	return true;
}

bool Opcode_CallDllProc( const char *lpszParms, const DllProc *pDllProc ){

	extern BOOL bDebugCompile;
	extern BOOL bDebugSupportProc;
	if(bDebugCompile&&bDebugSupportProc==0&& pDllProc->IsEqualSymbol( "DebugBreak" ) ){
		Call_DebugSys_SaveContext();
	}


	////////////////////////
	// パラメータのセット
	////////////////////////

	//パラメータオブジェクトを生成
	ParamImpl *pobj_parameter=0;
	pobj_parameter=new ParamImpl(lpszParms);

	// デフォルト引数を適用
	pobj_parameter->ApplyDefaultParameters( pDllProc->Params() );

	//エラーチェック
	if( !pobj_parameter->ErrorCheck( pDllProc->GetName(), pDllProc->Params() ) ){
		//パラメータにエラーがあるときは処理を終える
		return false;
	}

	//一時オブジェクトを生成
	pobj_parameter->NewTempParameters( pDllProc->GetName(), pDllProc->Params() );

	//レジスタ、スタックフレームにセット
	int ParmSize = pobj_parameter->SetParameter(pDllProc->GetName(), pDllProc->Params() );


	//動的リンクされたプロシージャの呼び出し

	//call dword ptr[LookupTable]
	compiler.codeGenerator.op_call( pDllProc );

	if(pDllProc->IsCdecl()){
		//add esp,ParmSize
		compiler.codeGenerator.op_add_esp(ParmSize);
	}

	//一時オブジェクトを破棄
	pobj_parameter->DeleteTempParameters();

	//パラメータオブジェクトを破棄
	delete pobj_parameter;

	return true;
}