source: dev/trunk/abdev/BasicCompiler_Common/src/Method.cpp@ 349

#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);
}