source: dev/trunk/abdev/BasicCompiler_Common/src/LexicalScopingImpl.cpp@ 228

#include "stdafx.h"

#include <LexicalScopingImpl.h>
#include <Compiler.h>

#include "../common.h"

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


void ScopeImpl::Break(){
    //未解放のローカルオブジェクトを解放する
    GetLexicalScopes().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 ScopeImpl::RunScheduleOfBreak(){
    for(int i=0;i<nBreakSchedule;i++){
        *((long *)(OpBuffer+pBreakSchedule[i]))=obp-(pBreakSchedule[i]+sizeof(long));
    }
}

void LexicalScopesImpl::End(){
    if( level <= 0 ){
        SetError();
        return;
    }

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

    Variables &vars = UserProc::IsGlobalAreaCompiling() ?
        compiler.GetMeta().GetGlobalVars() :
        UserProc::CompilingUserProc().GetLocalVars();

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


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


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

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

    Variables &vars = UserProc::IsGlobalAreaCompiling() ?
        compiler.GetMeta().GetGlobalVars() :
        UserProc::CompilingUserProc().GetLocalVars();


    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->GetScopeLevel() != GetNowLevel() ) continue;

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

            //メモリを解放する

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

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

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

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

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


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

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

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

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

        CallDestructorsOfScopeEnd();
    }

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

LexicalScopesImpl &GetLexicalScopes()
{
    static LexicalScopesImpl *pTemp = NULL;
    if( !pTemp )
    {
        pTemp = (LexicalScopesImpl *)Smoothie::Temp::pLexicalScopes;
    }
    return *pTemp;
}