source: dev/trunk/ab5.0/abdev/BasicCompiler32/OperatorProc.cpp @ 465

#include "stdafx.h"

#include <Compiler.h>

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

void FreeTempObject(int reg,const CClass *pobj_c){
    if(!IsSafeReg(reg)) compiler.errorMessenger.Output(300,NULL,cp);

    const CMethod *method = pobj_c->GetDestructorMethod();
    if( method ){
        //push reg
        compiler.codeGenerator.op_push(reg);

        //call DestructorProcAddr
        compiler.codeGenerator.op_call( &method->GetUserProc() );
    }

    //push reg
    compiler.codeGenerator.op_push(reg);

    //call free
    extern const UserProc *pSub_free;
    compiler.codeGenerator.op_call(pSub_free);
}

int CallOperatorProc(int idCalc, const Type &baseType, int *type_stack, LONG_PTR *index_stack,bool isNeedHeapFreeStructureStack[],int &sp)
{
    Type leftType( type_stack[sp-2], index_stack[sp-2] );
    Type rightType( type_stack[sp-1] & (~FLAG_CAST), index_stack[sp-1] );

    //オーバーロードされたオペレータ関数を呼び出す
    const CClass *pobj_c = &leftType.GetClass();

    std::vector<const UserProc *> subs;
    pobj_c->GetDynamicMethods().Enum( idCalc, subs );
    if( subs.size() == 0 ){
        return 0;
    }


    //項の数
    BOOL bTwoTerm=1;
    if(idCalc==CALC_AS) bTwoTerm=0;


    /////////////////////////////////////////////
    // オーバーロード解決用のパラメータを設定
    /////////////////////////////////////////////

    Parameters params;

    if(bTwoTerm){
        params.push_back( new Parameter( "", rightType ) );
    }

    //オーバーロードを解決
    char temporary[255];
    if(idCalc==CALC_EQUAL) lstrcpy(temporary,"==");
    else GetCalcName(idCalc,temporary);
    const UserProc *pUserProc = OverloadSolution( temporary, subs, params, baseType, leftType );

    if(!pUserProc){
        if(bTwoTerm){
            delete params[0];
        }
        return -1;
    }
    else{
        //オーバーロードされていないが、パラメータ個数が一致しないとき
        if(params.size()!=pUserProc->Params().size()){
            if(bTwoTerm){
                delete params[0];
            }
            return -1;
        }
    }

    for(int i=0;i<(int)params.size();i++){
        CheckDifferentType(
            *pUserProc->Params()[i],
            *params[i],
            NULL,
            i);
    }

    if(bTwoTerm){
        delete params[0];
    }

    int right_side_size = rightType.GetSize();

    if(bTwoTerm){
        if( pUserProc->RealParams()[1]->IsStruct() &&pUserProc->RealParams()[1]->IsRef() == false ){
            //一時オブジェクトはメソッド内で破棄される
            isNeedHeapFreeStructureStack[sp-1] = false;
        }
    }


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

        //mov ebx,eax
        compiler.codeGenerator.op_mov_RR(REG_EBX,REG_EAX);
    }


    //2つの項を取り出す
    if(bTwoTerm){
        if(right_side_size==sizeof(_int64)){
            //pop eax
            compiler.codeGenerator.op_pop(REG_EAX);

            //pop edx
            compiler.codeGenerator.op_pop(REG_EDX);
        }
        else{
            //pop eax
            compiler.codeGenerator.op_pop(REG_EAX);
        }
    }

    //pop ecx
    compiler.codeGenerator.op_pop(REG_ECX);


    //ヒープ解放用に退避
    if(isNeedHeapFreeStructureStack[sp-1]){
        //mov esi,eax
        compiler.codeGenerator.op_mov_RR(REG_ESI,REG_EAX);
    }
    if(isNeedHeapFreeStructureStack[sp-2]){
        //mov edi,ecx
        compiler.codeGenerator.op_mov_RR(REG_EDI,REG_ECX);
    }



    if(bTwoTerm){
        if(right_side_size==sizeof(_int64)){
            //push edx
            compiler.codeGenerator.op_push(REG_EDX);

            //push eax
            compiler.codeGenerator.op_push(REG_EAX);
        }
        else{
            //push eax
            compiler.codeGenerator.op_push(REG_EAX);
        }

        if( pUserProc->RealParams()[1]->IsRef() ){
            //一時参照を作成

            //mov eax,esp
            compiler.codeGenerator.op_mov_RR( REG_EAX, REG_ESP );

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

    if( pUserProc->ReturnType().IsStruct() ){
        //push ebx
        compiler.codeGenerator.op_push(REG_EBX);
    }

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

    //call operator_proc
    compiler.codeGenerator.op_call(pUserProc);

    if(bTwoTerm){
        if( pUserProc->RealParams()[1]->IsRef() ){
            //一時参照を破棄
            compiler.codeGenerator.op_pop( REG_NON );
        }
    }

    if( !pUserProc->ReturnType().IsNull() ){
        //スタックへプッシュ
        PushReturnValue(pUserProc->ReturnType().GetBasicType());
    }

    if(isNeedHeapFreeStructureStack[sp-1]){
        FreeTempObject(REG_ESI,(CClass *)index_stack[sp-1]);
    }
    if(isNeedHeapFreeStructureStack[sp-2]){
        FreeTempObject(REG_EDI,(CClass *)index_stack[sp-2]);
    }

    sp--;
    type_stack[sp-1]=pUserProc->ReturnType().GetBasicType();
    index_stack[sp-1]=pUserProc->ReturnType().GetIndex();

    if( pUserProc->ReturnType().IsStruct() )
    {
        //構造体が戻ったときはヒープ領域にインスタンスが格納されている
        //※後にfreeする必要あり
        isNeedHeapFreeStructureStack[sp-1] = true;
    }
    else
    {
        isNeedHeapFreeStructureStack[sp-1] = false;
    }

    return 1;
}

void CallCastOperatorProc(Type &calcType,BOOL bCalcUseHeap,const Type &toType){
    int type_stack[10];
    LONG_PTR index_stack[10];
    bool array_bUseHeap[10];
    int sp=2;

    if(bCalcUseHeap){
        //未解放のインスタンスが存在する旨を示す警告
        compiler.errorMessenger.Output(-105,NULL,cp);
    }

    //左辺
    type_stack[0]=calcType.GetBasicType();
    index_stack[0]=calcType.GetIndex();
    array_bUseHeap[0]=0;
    type_stack[1]=toType.GetBasicType();
    index_stack[1]=toType.GetIndex();
    array_bUseHeap[1]=0;

    int iRet = CallOperatorProc(CALC_AS,toType,type_stack,index_stack,array_bUseHeap,sp);

    if(iRet==1){
        //成功したとき
        calcType.SetType( type_stack[0], index_stack[0] );
        return;
    }
    else if(iRet==-1){
        //エラーが発行されたとき
        return;
    }

    //エラーを発行
    compiler.errorMessenger.Output(-1,"キャスト演算子がオーバーロードされていません。",cp);
}
void CallIndexerGetterProc( const Type &classType, const char *ObjectName, char *Parameter,Type &resultType, DWORD dwProcFlags ){
    std::vector<const UserProc *> subs;
    classType.GetClass().GetDynamicMethods().Enum( CALC_ARRAY_GET, subs );
    if( subs.size() == 0 ){
        return;
    }

    const UserProc *pUserProc = subs[0];

    Opcode_CallProc(Parameter,pUserProc,dwProcFlags,ObjectName);
    resultType = pUserProc->ReturnType();

    // 型パラメータを解決
    ResolveFormalGenericTypeParameter( resultType, classType, pUserProc );
}