source: dev/trunk/abdev/BasicCompiler_Common/LexicalScoping.cpp@ 150

#include "common.h"
#ifdef _AMD64_
#include "../BasicCompiler64/opcode.h"
#else
#include "../BasicCompiler32/opcode.h"
#endif


CLexicalScopes obj_LexScopes;


CScope::CScope( int level, int addr, SCOPE_TYPE TypeOfStatement ){
    this->level = level;
    this->StartAddress = addr;
    this->TypeOfStatement = TypeOfStatement;

    pBreakSchedule = (DWORD *)malloc( 1 );
    nBreakSchedule = 0;
}
CScope::~CScope(){
    free( pBreakSchedule );
}

int CScope::GetStartAddress(){
    return StartAddress;
}
SCOPE_TYPE CScope::GetTypeOfStatement(){
    return TypeOfStatement;
}

void CScope::Break(){
    //未解放のローカルオブジェクトを解放する
    obj_LexScopes.CallDestructorsOfReturn( level );

    //jmp ...(Next addr)
    OpBuffer[obp++]=(char)0xE9;

    pBreakSchedule=(DWORD *)realloc( pBreakSchedule, ( nBreakSchedule + 1 ) * sizeof(DWORD) );
    pBreakSchedule[nBreakSchedule]=obp;
    nBreakSchedule++;

    obp+=sizeof(long);
}
void CScope::RunScheduleOfBreak(){
    for(int i=0;i<nBreakSchedule;i++){
        *((long *)(OpBuffer+pBreakSchedule[i]))=obp-(pBreakSchedule[i]+sizeof(long));
    }
}


CScope *CLexicalScopes::SearchScope( SCOPE_TYPE TypeOfStatement ){
    for( int i = level; i>=0; i-- ){
        if( ppScopes[i]->GetTypeOfStatement() == TypeOfStatement ){
            return ppScopes[i];
        }
    }
    return NULL;
}

CLexicalScopes::CLexicalScopes(){
    ppScopes = (CScope **)malloc( 1 );
    level=0;
}
CLexicalScopes::~CLexicalScopes(){
    free( ppScopes );
}
void CLexicalScopes::Init(int addr){
    // TODO: エラーチェック

    level = -1;
    Start( addr, SCOPE_TYPE_BASE );
}
void CLexicalScopes::Start( int addr, SCOPE_TYPE TypeOfStatement ){
    level++;
    ppScopes = (CScope **)realloc( ppScopes, ( level + 1 ) * sizeof( CScope * ) );
    ppScopes[level] = new CScope( level, addr, TypeOfStatement );
}
void CLexicalScopes::End(){
    if( level <= 0 ){
        SetError(300,NULL,cp);
        return;
    }

    //デストラクタを呼ぶ
    CallDestructorsOfScopeEnd();

    Variables &vars = UserProc::IsGlobalAreaCompiling()?
        globalVars :
        UserProc::CompilingUserProc().localVars;

    //使用済みローカル変数の生存チェックを外す
    foreach( Variable *pVar, vars ){
        if(pVar->bLiving&&pVar->ScopeLevel==level){
            pVar->bLiving=0;
            extern int obp;
            pVar->ScopeEndAddress=obp;
        }
    }


    //スコープ抜け出しスケジュール
    ppScopes[level]->RunScheduleOfBreak();


    //スコープレベルを下げる
    delete ppScopes[level];
    level--;
}

void CLexicalScopes::ExitFor(){
    CScope *pScope = SearchScope( SCOPE_TYPE_FOR );
    if( !pScope ){
        SetError(12,"Exit For",cp);
        return;
    }

    pScope->Break();
}
void CLexicalScopes::ExitWhile(){
    CScope *pScope = SearchScope( SCOPE_TYPE_WHILE );
    if( !pScope ){
        SetError(12,"Exit While",cp);
        return;
    }

    pScope->Break();
}
void CLexicalScopes::ExitDo(){
    CScope *pScope = SearchScope( SCOPE_TYPE_DO );
    if( !pScope ){
        SetError(12,"Exit Do",cp);
        return;
    }

    pScope->Break();
}

int CLexicalScopes::GetNowLevel(){
    return level;
}
void CLexicalScopes::SetNowLevel( int level ){
    this->level = level;
}
int CLexicalScopes::GetStartAddress(){
    return ppScopes[level]->GetStartAddress();
}


// スコープ終了時のデストラクタ呼び出し
void CLexicalScopes::CallDestructorsOfScopeEnd(){

    Variables &vars = UserProc::IsGlobalAreaCompiling()?
        globalVars :
        UserProc::CompilingUserProc().localVars;


    int i3;
    int indexSystemGC=-1;
    for( i3 = (int)vars.size() - 1; i3 >= 0; i3-- ){        //確保したのと逆順序で解放するため、バックサーチにする

        Variable *pVar = vars[i3];

        if( UserProc::IsGlobalAreaCompiling() && GetNowLevel() == 0 ){
            if( pVar->GetName() == "_System_GC" ){
                indexSystemGC=i3;
                continue;
            }
        }

        //同一レベルのレキシカルスコープのみを検知
        if(!pVar->bLiving) continue;
        if( pVar->ScopeLevel != GetNowLevel() ) continue;

        if( pVar->IsStruct() && pVar->IsParameter() ){
            //構造体パラメータを持つとき

            //メモリを解放する

#ifdef _AMD64_
            //x64ビットコード

            //mov rcx,qword ptr[rsp+offset]
            op_mov_RM(sizeof(_int64),REG_RCX,REG_RSP,
                -pVar->offset,
                MOD_BASE_DISP32);
            obp-=sizeof(long);
            AddLocalVarAddrSchedule();
            obp+=sizeof(long);
#else
            //x86コード

            //mov ecx,dword ptr[ebp+offset]
            op_mov_RM(sizeof(long),REG_ECX,REG_EBP,-pVar->offset,MOD_BASE_DISP32);
            obp-=sizeof(long);
            AddLocalVarAddrSchedule();
            obp+=sizeof(long);

            //push ecx
            op_push(REG_ECX);
#endif

            //call free
            extern UserProc *pSub_free;
            op_call(pSub_free);


            if( UserProc::IsGlobalAreaCompiling() ){
                //ここには来ないハズ
                SetError(300,NULL,cp);
            }
        }
    }

    if(indexSystemGC!=-1){
        //_System_GCオブジェクトのデストラクタの呼び出し処理
        const CMethod *method = vars[indexSystemGC]->GetClass().GetDestructorMethod();
        if( method ){
            Opcode_CallProc("",method->pUserProc,0,vars[indexSystemGC]->GetName().c_str(),DEF_OBJECT);
        }
    }
}

// Returnステートメントで発行されるデストラクタを生成
void CLexicalScopes::CallDestructorsOfReturn( int BaseLevel ){
    //現在のスコープレベルを退避
    int backupScopeLevel = GetNowLevel();

    for( int i = GetNowLevel(); i >= BaseLevel; i-- ){
        SetNowLevel( i );

        CallDestructorsOfScopeEnd();
    }

    //現在のスコープレベルを復元
    SetNowLevel( backupScopeLevel );
}