#include "stdafx.h"

#include <Class.h>


Methods::Methods()
{
}
Methods::~Methods()
{
	Methods &methods = *this;
	BOOST_FOREACH( CMethod *pMethod, methods ){
		delete pMethod;
	}
}

void Methods::Add( UserProc *pUserProc,Prototype::Accessibility accessibility, bool isConst, bool isAbstract, bool isVirtual ){
	CMethod *pMethod = new DynamicMethod( pUserProc, accessibility, isAbstract, isVirtual, isConst );
	this->push_back( pMethod );
	pUserProc->SetMethod( pMethod );
}
void Methods::AddStatic(UserProc *pUserProc, Prototype::Accessibility accessibility ){
	CMethod *pMethod = new StaticMethod( pUserProc, accessibility );
	this->push_back( pMethod );
	pUserProc->SetMethod( pMethod );
}

bool Methods::Override( UserProc *pUserProc, Prototype::Accessibility accessibility )
{
	//メソッドのオーバーライド
	Methods &methods = *this;
	BOOST_FOREACH( CMethod *pMethod, methods )
	{
		if( pMethod->GetUserProc().GetName() == pUserProc->GetName() )
		{
			if( pMethod->GetUserProc().Params().Equals( pUserProc->Params() )
				&& pMethod->GetUserProc().ReturnType().Equals( pUserProc->ReturnType() ) )
			{
				//メンバ関数を上書き
				pMethod->SetUserProcPtr( pUserProc );
				pMethod->Override();

				if( !pMethod->IsVirtual() )
				{
					// オーバーライドミス
					SetError(136,NULL,cp);
				}
				if(pMethod->GetAccessibility() != accessibility )
				{
					SetError(128,NULL,cp);
				}

				pUserProc->SetMethod( pMethod );
				return true;
			}
		}
	}
	return false;
}

const CMethod *Methods::GetMethodPtr( const UserProc *pUserProc ) const
{
	const Methods &methods = *this;
	for( int i=(int)methods.size()-1; i>=0; i-- ){
		if( pUserProc == &methods[i]->GetUserProc() ){
			return methods[i];
		}
	}
	return NULL;
}
bool Methods::IsExist( const char *name ) const
{
	const Methods &methods = *this;
	BOOST_FOREACH( const CMethod *pMethod, methods ){
		if( pMethod->GetUserProc().GetName() == name ) return true;
	}
	return false;
}
void Methods::Enum( const char *methodName, vector<const UserProc *> &subs ) const
{
	//オブジェクトのメンバ関数の場合
	//※オーバーライドされた関数を先にサーチする必要があるため、バックサーチを行う
	const Methods &methods = *this;
	for( int i=(int)methods.size()-1; i>=0; i-- ){
		if( methods[i]->GetUserProc().GetName() == methodName ){
			subs.push_back( &methods[i]->GetUserProc() );
		}
	}
}
void Methods::Enum( BYTE idOperatorCalc, vector<const UserProc *> &subs ) const
{
	//オブジェクトのメンバ関数の場合
	//※オーバーライドされた関数を先にサーチする必要があるため、バックサーチを行う
	const Methods &methods = *this;
	for( int i=(int)methods.size()-1; i>=0; i-- ){
		const UserProc &userProc = methods[i]->GetUserProc();
		const char *temp = userProc.GetName().c_str();
		if(temp[0]==1&&temp[1]==ESC_OPERATOR){
			if((BYTE)temp[2]==idOperatorCalc){
				subs.push_back( &userProc );
			}
		}
	}
}

int Methods::GetVtblNum() const
{
	int count = 0;
	const Methods &methods = *this;
	BOOST_FOREACH( const CMethod *pMethod, methods )
	{
		if( pMethod->IsVirtual() )
		{
			count++;
		}
	}
	return count;
}

void Methods::GenerateVTablePart( long &vtableDataTableOffset ) const
{
	const UserProc **ppsi = (const UserProc **)malloc(GetVtblNum()*sizeof(UserProc *));

	//関数テーブルに値をセット
	int i2 = 0;
	const Methods &methods = *this;
	BOOST_FOREACH( const CMethod *pMethod, methods ){
		if(pMethod->IsVirtual()){
			if( !pMethod->GetUserProc().IsUsing() )
			{
				ts((char *)pMethod->GetUserProc().GetFullName().c_str());
			}
			pMethod->GetUserProc().Using();

			if(pMethod->IsAbstract()){
				SetError();

				ppsi[i2]=0;
			}
			else{
				ppsi[i2]=&pMethod->GetUserProc();
			}
			i2++;
		}
	}

	vtableDataTableOffset = compiler.GetObjectModule().dataTable.AddBinary( (void *)ppsi, GetVtblNum()*sizeof(LONG_PTR) );

	for( int i=0; i < GetVtblNum(); i++ ){
		pobj_Reloc->AddSchedule_DataSection(static_cast<DWORD>(vtableDataTableOffset+i*sizeof(LONG_PTR)));
	}

	free(ppsi);
}