source: dev/trunk/abdev/BasicCompiler_Common/Subroutine.cpp@ 206

#include "stdafx.h"

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

#include <Compiler.h>
#include <Procedure.h>
#include <NamespaceSupporter.h>

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

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

int GetCallProcName(char *buffer,char *name){
    int i2,i3,IsStr=0;

    for(i2=0;;i2++){
        if(buffer[i2]=='\"') IsStr^=1;
        if(IsDBCSLeadByte(buffer[i2])){
            name[i2]=buffer[i2];
            i2++;
            name[i2]=buffer[i2];
            continue;
        }
        if(buffer[i2]=='['&&IsStr==0){
            i3=GetStringInBracket(name+i2,buffer+i2);
            i2+=i3-1;
            continue;
        }
        if(buffer[i2]=='('&&IsStr==0){
            name[i2]=0;
            break;
        }
        if(buffer[i2]=='='&&IsStr==0){
            name[i2]=0;
            break;
        }

        name[i2]=buffer[i2];
        if(buffer[i2]=='\0') break;
    }
    return i2;
}

int GetProc(char *name,void **ppInfo){

    //ユーザー定義関数
    *ppInfo=(void *)GetSubHash(name);
    if(*ppInfo) return PROC_DEFAULT;

    //DLL関数
    *ppInfo=(void *)GetDeclareHash(name);
    if(*ppInfo) return PROC_DLL;

    //コンパイラ埋め込み型
    *ppInfo=(void *)(_int64)GetFunctionFromName(name);
    if(*ppInfo) return PROC_BUILTIN;

    //関数ポインタ
    Type type;
    if( !GetVarType( name, type, false ) ){
        return 0;
    }
    if( type.IsProcPtr() ){
        return PROC_PTR;
    }

    return 0;
}

void SplitObjectName(const char *name,char *ObjectName,int *pRefType){
    int i4;
    for(i4=lstrlen(name)-1;i4>=0;i4--){
        if(name[i4]=='.'||(name[i4]==1&&name[i4+1]==ESC_PSMEM))
            break;
    }
    if(i4==-1) ObjectName[0]=0;
    else{
        //参照タイプを判別
        if(name[i4]=='.') *pRefType=DEF_OBJECT;
        else *pRefType=DEF_PTR_OBJECT;

        if(i4==0) GetWithName(ObjectName);
        else{
            memcpy(ObjectName,name,i4);
            ObjectName[i4]=0;
        }
    }
}

bool CallProc( int kind, const void *pProc, const char *fullCallName, const char *lpszParms, Type &resultType, bool isCallOn ){

    //GetSubHash内でエラー提示が行われた場合
    if(pProc==(Procedure *)-1){
        return false;
    }

    if(kind==PROC_DEFAULT){
        /////////////////////
        // ユーザー定義関数
        /////////////////////

        const UserProc *pUserProc = (const UserProc *)pProc;

        //オブジェクト名を取得
        char ObjectName[VN_SIZE];
        int RefType;
        SplitObjectName(fullCallName,ObjectName,&RefType);


        ////////////////////////
        // オーバーロードを解決
        ////////////////////////

        std::vector<const UserProc *> subs;
        GetOverloadSubHash(fullCallName,subs);
        if(subs.size()){
            //オーバーロードを解決
            pUserProc=OverloadSolutionWithStrParam(fullCallName,subs,lpszParms,ObjectName);

            if(!pUserProc){
                return false;
            }
        }

        resultType = pUserProc->ReturnType();

        if( isCallOn ){
            if( !Opcode_CallProc(lpszParms,pUserProc,0,ObjectName,RefType) ){
                return false;
            }
        }
    }
    else if(kind==PROC_DLL){
        /////////////////////////
        // DLL関数
        /////////////////////////
        DllProc *pDllProc = (DllProc *)pProc;

        resultType = pDllProc->ReturnType();

        if( isCallOn ){
            if( !Opcode_CallDllProc(lpszParms,pDllProc) ){
                return false;
            }
        }
    }
    else if(kind==PROC_BUILTIN){
        /////////////////////////
        // 組み込み関数
        /////////////////////////
        int FuncId = (int)(_int64)pProc;

        if( !Opcode_CallFunc( lpszParms, FuncId, resultType, isCallOn ) ){
            return false;
        }
    }
    else if(kind==PROC_PTR){
        /////////////////
        // 関数ポインタ
        /////////////////

        Type type;
        GetVarType(fullCallName,type,false);

        ProcPointer *pProcPtr = compiler.GetMeta().GetProcPointers()[type.GetIndex()];
        resultType = pProcPtr->ReturnType();

        if( isCallOn ){
            if( !Opcode_CallProcPtr(fullCallName,lpszParms,pProcPtr) ){
                return false;
            }
        }
    }
    else{
        return false;
    }

    return true;
}
bool CallPropertyMethod( const char *variable, const char *rightSide, Type &resultType){
    //プロパティ用のメソッドを呼び出す

    //配列要素を取得
    char VarName[VN_SIZE],ArrayElements[VN_SIZE];
    GetArrayElement(variable,VarName,ArrayElements);

    //オブジェクト名を取得
    char ObjectName[VN_SIZE];
    int RefType;
    SplitObjectName(VarName,ObjectName,&RefType);

    //オーバーロード用の関数リストを作成
    std::vector<const UserProc *> subs;
    GetOverloadSubHash(VarName,subs);
    if(subs.size()==0){
        return false;
    }

    //パラメータを整備
    char *Parameter;
    Parameter=(char *)HeapAlloc(hHeap,0,lstrlen(ArrayElements)+lstrlen(rightSide)+32);
    lstrcpy(Parameter,ArrayElements);
    if(rightSide){
        if(Parameter[0]&&rightSide[0]) lstrcat(Parameter,",");
        lstrcat(Parameter,rightSide);
    }

    //オーバーロードを解決
    const UserProc *pUserProc = OverloadSolutionWithStrParam(VarName,subs,Parameter,ObjectName);

    if(pUserProc){
        //呼び出し
        Opcode_CallProc(Parameter,pUserProc,0,ObjectName,RefType);

        resultType = pUserProc->ReturnType();
    }

    HeapDefaultFree(Parameter);

    return true;
}

bool GetReturnTypeOfPropertyMethod( const char *variable, const char *rightSide, Type &resultType ){
    //プロパティ用のメソッドを呼び出す

    //配列要素を取得
    char VarName[VN_SIZE],ArrayElements[VN_SIZE];
    GetArrayElement(variable,VarName,ArrayElements);

    //オブジェクト名を取得
    char ObjectName[VN_SIZE];
    int RefType;
    SplitObjectName(VarName,ObjectName,&RefType);

    //オーバーロード用の関数リストを作成
    std::vector<const UserProc *> subs;
    GetOverloadSubHash(VarName,subs);
    if(subs.size()==0){
        return 0;
    }

    //パラメータを整備
    char *Parameter;
    Parameter=(char *)HeapAlloc(hHeap,0,lstrlen(ArrayElements)+lstrlen(rightSide)+32);
    lstrcpy(Parameter,ArrayElements);
    if(rightSide){
        if(Parameter[0]&&rightSide[0]) lstrcat(Parameter,",");
        lstrcat(Parameter,rightSide);
    }

    //オーバーロードを解決
    const UserProc *pUserProc = OverloadSolutionWithStrParam(VarName,subs,Parameter,ObjectName);

    if(pUserProc){
        resultType = pUserProc->ReturnType();
    }

    return 1;
}

//インデクサ（getter）の戻り値を取得
bool GetReturnTypeOfIndexerGetterProc( const CClass &objClass, Type &resultType ){
    vector<const UserProc *> subs;
    objClass.GetMethods().Enum( CALC_ARRAY_GET, subs );
    if( subs.size() == 0 ){
        return false;
    }

    resultType = subs[0]->ReturnType();

    return true;
}


void AddDeclareData(const NamespaceScopes &namespaceScopes, char *buffer,int nowLine){
    extern HANDLE hHeap;
    int i2;

    int i=0;

    //Sub/Function
    Procedure::Kind kind = Procedure::Sub;
    if(buffer[i]==ESC_SUB){
    }
    else if(buffer[i]==ESC_FUNCTION){
        kind = Procedure::Function;
    }
    else{
        SetError(1,NULL,nowLine);
        return;
    }
    i++;

    //プロシージャ名
    char procName[VN_SIZE];
    bool isCdecl = false;
    for(i2=0;;i++,i2++){
        if(buffer[i]==1&&buffer[i+1]==ESC_CDECL){
            isCdecl = true;

            i+=2;
            procName[i2]=0;
            break;
        }
        if(buffer[i]==','){
            procName[i2]=0;
            break;
        }
        if(buffer[i]=='\0'){
            SetError(1,NULL,nowLine);
            return;
        }
        procName[i2]=buffer[i];
    }
    i++;

    //ユーザー定義関数との重複チェック
    if(GetSubHash(procName)){
        SetError(15,procName,nowLine);
        return;
    }


    //ライブラリ
    char dllFileName[MAX_PATH];
    i = GetOneParameter( buffer, i, dllFileName );
    Type resultType;
    _int64 i64data;
    if( !StaticCalculation( true, dllFileName, 0, &i64data, resultType ) ){
        return;
    }
    if( resultType.GetBasicType() != typeOfPtrChar ){
        SetError(1,NULL,nowLine);
        return;
    }
    lstrcpy( dllFileName, (char *)i64data );
    CharUpper(dllFileName);
    if(!strstr(dllFileName,".")){
        lstrcat(dllFileName,".DLL");
        if(lstrlen(dllFileName)>=16){
            SetError(7,NULL,nowLine);
            return;
        }
    }

    //エイリアス
    char alias[VN_SIZE];
    i = GetOneParameter( buffer, i, alias );
    if( alias[0] ){
        if( !StaticCalculation( true, alias, 0, &i64data, resultType ) ){
            return;
        }
        if( resultType.GetBasicType() != typeOfPtrChar ){
            SetError(1,NULL,nowLine);
            return;
        }
        lstrcpy( alias, (char *)i64data );
    }
    else{
        //省略されたときは関数名
        lstrcpy( alias, procName );
    }


    // オブジェクトを生成
    DllProc *pDllProc = new DllProc( namespaceScopes, procName, kind, isCdecl, dllFileName, alias );

    // パラメータを解析
    // ※第1パラメータにに指定するデータの例："( s As String ) As String"
    pDllProc->SetParamsAndReturnType( buffer + i, nowLine );

    // パラメータのエラーチェック
    BOOST_FOREACH( const Parameter *pParam, pDllProc->Params() ){
        if( pParam->IsObject() ){
            SetError(25,pParam->GetVarName(),nowLine);
        }
        if( !pParam->IsRef() ){
            if( pParam->IsStruct() ){
                SetError(28,pParam->GetVarName(),nowLine);
            }
        }
    }

    //戻り値のエラーチェック
    if( pDllProc->IsFunction() ){
        // Function定義

        if( pDllProc->ReturnType().IsObject() ){
            // DLL関数ではオブジェクトを戻り値にできない
            SetError(40,pDllProc->GetName(),nowLine);
        }
    }


    /////////////////////////////////
    // 格納位置を計算してppDeclareHashにセット
    /////////////////////////////////

    //ハッシュ値を取得
    int key;
    key=hash_default(procName);

    extern DllProc **ppDeclareHash;
    if(ppDeclareHash[key]){
        DllProc *pTempProc = ppDeclareHash[key];
        while(1){
            if( pTempProc->IsEqualSymbol( pDllProc->GetNamespaceScopes(), pDllProc->GetName() ) ){
                //重複エラー
                SetError(15,procName,nowLine);
                return;
            }

            if(pTempProc->pNextData==0){
                pTempProc->pNextData=pDllProc;
                break;
            }
            pTempProc=pTempProc->pNextData;
        }
        pTempProc=pTempProc->pNextData;
    }
    else{
        ppDeclareHash[key]=pDllProc;
    }
}

void UserProcs::CollectUserProcs( const BasicSource &source, UserProcs &userProcs )
{
    extern HANDLE hHeap;
    int i,i2,i3;
    char temporary[8192];

    //Declare（DLL関数）情報を初期化
    extern DllProc **ppDeclareHash;
    ppDeclareHash=(DllProc **)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,MAX_HASH*sizeof(DllProc *));

    //サブルーチン（ユーザー定義）情報を初期化
    userProcs.Clear();

    // 名前空間管理
    NamespaceScopes &namespaceScopes = compiler.GetNamespaceSupporter().GetLivingNamespaceScopes();
    namespaceScopes.clear();

    // Importsされた名前空間の管理
    NamespaceScopesCollection &importedNamespaces = compiler.GetNamespaceSupporter().GetImportedNamespaces();
    importedNamespaces.clear();

    i=-1;
    while(1){
        i++;

        if(source[i]==1&&(source[i+1]==ESC_CLASS||source[i+1]==ESC_INTERFACE)){
            /*  Class ～ End Class
                Interface ～ End Interface
                を飛び越す           */
            i3=GetEndXXXCommand(source[i+1]);
            for(i+=2,i2=0;;i++,i2++){
                if(source[i]=='\0') break;
                if(source[i]==1&&source[i+1]==(char)i3){
                    i++;
                    break;
                }
            }
            if(source[i]=='\0') break;
            continue;
        }

        if( source[i] == 1 && source[i+1] == ESC_NAMESPACE ){
            for(i+=2,i2=0;;i2++,i++){
                if( IsCommandDelimitation( source[i] ) ){
                    temporary[i2]=0;
                    break;
                }
                temporary[i2]=source[i];
            }
            namespaceScopes.push_back( temporary );

            continue;
        }
        else if( source[i] == 1 && source[i+1] == ESC_ENDNAMESPACE ){
            if( namespaceScopes.size() <= 0 ){
                SetError(12, "End Namespace", i );
            }
            else{
                namespaceScopes.pop_back();
            }

            i += 2;
            continue;
        }
        else if( source[i] == 1 && source[i+1] == ESC_IMPORTS ){
            for(i+=2,i2=0;;i2++,i++){
                if( IsCommandDelimitation( source[i] ) ){
                    temporary[i2]=0;
                    break;
                }
                temporary[i2]=source[i];
            }
            if( !compiler.GetNamespaceSupporter().ImportsNamespace( temporary ) )
            {
                SetError(64,temporary,cp );
            }

            continue;
        }
        else if( source[i] == 1 && source[i+1] == ESC_CLEARNAMESPACEIMPORTED ){
            importedNamespaces.clear();
            continue;
        }

        if(source[i]==1&&source[i+1]==ESC_DECLARE){
            for(i+=2,i2=0;;i2++,i++){
                if(source[i]=='\n'){
                    temporary[i2]=0;
                    break;
                }
                temporary[i2]=source[i];
                if(source[i]=='\0') break;
            }
            AddDeclareData(namespaceScopes,temporary,i);

            continue;
        }
        if(source[i]==1&&(source[i+1]==ESC_SUB||source[i+1]==ESC_FUNCTION||source[i+1]==ESC_MACRO)){
            char statementChar = source[i+1];

            for(i2=0;;i2++,i++){
                if(IsCommandDelimitation(source[i])){
                    temporary[i2]=0;
                    break;
                }
                temporary[i2]=source[i];
                if(source[i]=='\0') break;
            }
            userProcs.Add(namespaceScopes, importedNamespaces, temporary,i,false,NULL,false);

            /*  Sub ～ End Sub
                Function ～ End Function
                Macro ～ End Macro
                を飛び越す           */
            char endStatementChar = GetEndXXXCommand( statementChar );
            for(i2=0;;i++,i2++){
                if( source[i] == '\0' ) break;
                if( source[i] == 1 && source[i+1] == endStatementChar ){
                    i++;
                    break;
                }
            }
            if(source[i]=='\0') break;
            continue;
        }

        //次の行
        for(;;i++){
            if(IsCommandDelimitation(source[i])) break;
        }
        if(source[i]=='\0') break;
    }

    ////////////
    // 特殊関数
    ////////////
    namespaceScopes.clear();
    importedNamespaces.clear();

    sprintf(temporary,"%c%c_allrem()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_aullrem()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_allmul()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_alldiv()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_aulldiv()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_allshl()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_allshr()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_aullshr()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);

    sprintf(temporary,"%c%c_System_InitStaticLocalVariables()",1,ESC_SUB);
    userProcs.Add( namespaceScopes, importedNamespaces, temporary,0,false,NULL,false);
}
void Delete_di(DllProc *pDllProc){
    if(pDllProc->pNextData) Delete_di(pDllProc->pNextData);

    delete pDllProc;
}
void DeleteDeclareInfo(void){
    //DLL情報を解放
    extern DllProc **ppDeclareHash;
    int i;
    for(i=0;i<MAX_HASH;i++){
        if(!ppDeclareHash[i]) continue;

        Delete_di(ppDeclareHash[i]);
    }
    HeapDefaultFree(ppDeclareHash);
}

bool IsNeedProcCompile(){
    compiler.GetMeta().GetUserProcs().Iterator_Reset();
    while( compiler.GetMeta().GetUserProcs().Iterator_HasNext() )
    {
        UserProc *pUserProc = compiler.GetMeta().GetUserProcs().Iterator_GetNext();
        if( pUserProc->IsUsing() && pUserProc->IsCompiled() == false ){
            return true;
        }
    }
    return false;
}