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

#include "stdafx.h"

#include <Class.h>


bool DynamicMethod::Override( const UserProc *pUserProc, Prototype::Accessibility accessibility, bool isOverrideModifier )
{
    if( this->IsVirtual()
        && !this->IsAbstract()
        && isOverrideModifier == false )
    {
        // Override修飾子が無い状況で基底クラスの実体メソッドをオーバーライドしようとした
        SetError(127,NULL,cp);
    }

    //メンバ関数を上書き
    this->SetUserProcPtr( pUserProc );
    this->SetAbstractMark( false );

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

    return true;
}


StaticMethod::StaticMethod( const StaticMethod &staticMethod )
{
    // 静的メソッドがコピーコンストラトされることは想定しない
    SetError();
}

Methods::Methods()
{
}

// コピーコンストラクタ
Methods::Methods( const Methods &methods )
{
    BOOST_FOREACH( CMethod *pMethod, methods )
    {
        this->push_back( new DynamicMethod( dynamic_cast<DynamicMethod &>(*pMethod) ) );
    }
}

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

CMethod *Methods::FindForOverride( const UserProc *pUserProc )
{
    //メソッドのオーバーライド
    Methods &methods = *this;
    BOOST_FOREACH( CMethod *pMethod, methods )
    {
        if( !pMethod->IsNotUse() && pMethod->GetUserProc().IsEqualForOverride( pUserProc ) )
        {
            return pMethod;
        }
    }
    return NULL;
}

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 && methods[i]->IsNotUse() == false ){
            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);
}