source: dev/trunk/abdev/BasicCompiler32/OperatorProc.cpp@ 228

#include "stdafx.h"

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

#include <Compiler.h>

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

void FreeTempObject(int reg,const CClass *pobj_c){
    if(!IsSafeReg(reg)) SetError(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 *bUseHeap,int &sp){
    //オーバーロードされたオペレータ関数を呼び出す
    CClass *pobj_c;
    pobj_c=(CClass *)index_stack[sp-2];

    std::vector<const UserProc *> subs;
    pobj_c->GetMethods().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( "", Type( type_stack[sp-1], index_stack[sp-1] ) ) );
    }

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

    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]->GetBasicType(),
            pUserProc->Params()[i]->GetIndex(),
            params[i]->GetBasicType(),
            params[i]->GetIndex(),
            "",
            i);
    }

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

    int right_side_size = GetTypeSize(type_stack[sp-1],index_stack[sp-1]);

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


    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(bUseHeap[sp-1]){
        //mov esi,eax
        compiler.codeGenerator.op_mov_RR(REG_ESI,REG_EAX);
    }
    if(bUseHeap[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(bUseHeap[sp-1]){
        FreeTempObject(REG_ESI,(CClass *)index_stack[sp-1]);
    }
    if(bUseHeap[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する必要あり
        bUseHeap[sp-1]=1;
    }
    else bUseHeap[sp-1]=0;

    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){
        //未解放のインスタンスが存在する旨を示す警告
        SetError(-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;
    }

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

    Opcode_CallProc(Parameter,subs[0],0,ObjectName,DEF_OBJECT);
    resultType = subs[0]->ReturnType();
}