source: dev/BasicCompiler_Common/Enum.cpp@ 29

#include "common.h"

CEnumParent **ppobj_EnumParent;
int iEnumParentNum;

CEnumMember::CEnumMember(char *name,int value){
    m_name=(char *)HeapAlloc(hHeap,0,lstrlen(name)+1);
    lstrcpy(m_name,name);

    m_value=value;
}
CEnumMember::~CEnumMember(){
    if(m_name) HeapDefaultFree(m_name);
}


CEnumParent::CEnumParent(char *buffer,int NowLine){
    ppobj_EnumMember=(CEnumMember **)HeapAlloc(hHeap,0,1);
    iEnumMemberNum=0;

    int i=0,i2;

    if(!(buffer[i]==1&&buffer[i+1]==ESC_ENUM)) return;
    i+=2;

    //列挙体の名前を取得
    char temporary[VN_SIZE];
    for(i2=0;;i++,i2++){
        if(IsCommandDelimitation(buffer[i])){
            temporary[i2]=0;
            break;
        }
        if(!IsVariableChar(buffer[i])){
            SetError(1,NULL,i);
            break;
        }
        temporary[i2]=buffer[i];
    }

    TypeName=(char *)HeapAlloc(hHeap,0,lstrlen(temporary)+1);
    lstrcpy(TypeName,temporary);

    if(buffer[i]=='\0'){
        SetError(22,"Enum",NowLine);
        return;
    }

    int NextValue=0;
    while(1){
        i++;

        if(buffer[i]==1&&buffer[i+1]==ESC_ENDENUM) break;

        for(i2=0;;i2++,i++){
            if(IsCommandDelimitation(buffer[i])){
                temporary[i2]=0;
                break;
            }
            if(buffer[i]=='='){
                temporary[i2]=0;
                break;
            }
            temporary[i2]=buffer[i];
        }
        if(temporary[0]=='\0'){
            if(buffer[i]=='\0'){
                SetError(22,"Enum",NowLine);
                break;
            }
            continue;
        }

        if(buffer[i]!='='){
            NextValue++;
        }
        else{
            char temp2[VN_SIZE];
            for(i++,i2=0;;i2++,i++){
                if(IsCommandDelimitation(buffer[i])){
                    temp2[i2]=0;
                    break;
                }
                temp2[i2]=buffer[i];
            }
            NextValue=atoi(temp2);
        }

        //メンバを追加
        ppobj_EnumMember=(CEnumMember **)HeapReAlloc(hHeap,0,ppobj_EnumMember,(iEnumMemberNum+1)*sizeof(CEnumMember *));
        ppobj_EnumMember[iEnumMemberNum]=new CEnumMember(temporary,NextValue);
        iEnumMemberNum++;
    }
}
CEnumParent::~CEnumParent(){
    HeapDefaultFree(TypeName);

    int i;
    for(i=0;i<iEnumMemberNum;i++){
        delete ppobj_EnumMember[i];
    }
    HeapDefaultFree(ppobj_EnumMember);
}



void CEnumParent::InitEnum(void){
    ppobj_EnumParent=(CEnumParent **)HeapAlloc(hHeap,0,1);
    iEnumParentNum=0;

    extern char *basbuf;
    int i;
    for(i=0;;i++){
        if(basbuf[i]=='\0') break;

        if(basbuf[i]==1&&basbuf[i+1]==ESC_ENUM){
            if(i>=2){
                if(basbuf[i-2]==1&&basbuf[i-1]==ESC_CONST) continue;
            }
            ppobj_EnumParent=(CEnumParent **)HeapReAlloc(hHeap,0,ppobj_EnumParent,(iEnumParentNum+1)*sizeof(CEnumParent *));
            ppobj_EnumParent[iEnumParentNum]=new CEnumParent(basbuf+i,i);
            iEnumParentNum++;
        }
    }
}
void CEnumParent::DestroyEnum(void){
    int i;
    for(i=0;i<iEnumParentNum;i++){
        delete ppobj_EnumParent[i];
    }
    HeapDefaultFree(ppobj_EnumParent);
}

char *CEnumParent::GenerateCodes(void){
    int i,i2;

    char *buffer;
    int MaxSize,length;
    MaxSize=65535;
    buffer=(char *)HeapAlloc(hHeap,0,MaxSize+65535);
    length=0;

    buffer[0]=0;

    for(i=0;i<iEnumParentNum;i++){
        CEnumParent *parent;
        parent=ppobj_EnumParent[i];

        sprintf(buffer+length,"Class %s\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tInherits EnumBase\n");
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"Public\n");
        length+=lstrlen(buffer+length);
        sprintf(buffer+length,"\tSub %s()\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        if(parent->iEnumMemberNum){
            sprintf(buffer+length,"\t\tEnumBase(%d)\n",
                parent->ppobj_EnumMember[0]->m_value);
            length+=lstrlen(buffer+length);
        }
        lstrcpy(buffer+length,"\tEnd Sub\n");
        length+=lstrlen(buffer+length);
        sprintf(buffer+length,"\tSub %s(value As Long)\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\t\tEnumBase(value)\n");
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tEnd Sub\n");
        length+=lstrlen(buffer+length);
        sprintf(buffer+length,"\tSub ~%s()\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tEnd Sub\n");
        length+=lstrlen(buffer+length);

        for(i2=0;i2<parent->iEnumMemberNum;i2++){
            CEnumMember *member;
            member=parent->ppobj_EnumMember[i2];

            sprintf(buffer+length,"\tStatic %s As %s(%d)\n",
                member->m_name,
                parent->TypeName,
                member->m_value);
            length+=lstrlen(buffer+length);
        }

        sprintf(buffer+length,"\tOverride Function ToString() As String\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\t\tSelect Case m_Value\n");
        length+=lstrlen(buffer+length);
        for(i2=0;i2<parent->iEnumMemberNum;i2++){
            CEnumMember *member;
            member=parent->ppobj_EnumMember[i2];

            sprintf(buffer+length,"\t\t\tCase %d\n",member->m_value);
            length+=lstrlen(buffer+length);
            sprintf(buffer+length,"\t\t\t\tReturn \"%s\"\n",member->m_name);
            length+=lstrlen(buffer+length);
        }
        lstrcpy(buffer+length,"\t\tEnd Select\n");
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tEnd Function\n");
        length+=lstrlen(buffer+length);

        sprintf(buffer+length,"\tSub Operator= (ByRef value As %s)\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\t\tThis.Copy(ByVal VarPtr(value))\n");
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tEnd Sub\n");
        length+=lstrlen(buffer+length);

        /*sprintf(buffer+length,"\tSub Operator= (ByRef value As String)\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\t\tSelect Case value\n");
        length+=lstrlen(buffer+length);
        for(i2=0;i2<parent->iEnumMemberNum;i2++){
            CEnumMember *member;
            member=parent->ppobj_EnumMember[i2];

            sprintf(buffer+length,"\t\t\tCase \"%s\"\n",member->m_name);
            length+=lstrlen(buffer+length);
            sprintf(buffer+length,"\t\t\t\tThis=%s.%s\n",parent->TypeName,member->m_name);
            length+=lstrlen(buffer+length);
        }
        lstrcpy(buffer+length,"\t\tEnd Select\n");
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tEnd Sub\n");
        length+=lstrlen(buffer+length);*/

        sprintf(buffer+length,"\tSub Operator= (value As Long)\n",parent->TypeName);
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\t\tm_Value=value\n");
        length+=lstrlen(buffer+length);
        lstrcpy(buffer+length,"\tEnd Sub\n");
        length+=lstrlen(buffer+length);

        lstrcpy(buffer+length,"End Class\n");
        length+=lstrlen(buffer+length);


        //バッファ領域が足りなくなった場合はバッファを増量する
        if(length>MaxSize){
            MaxSize+=65535;
            buffer=(char *)HeapReAlloc(hHeap,0,buffer,MaxSize+65535);
        }
    }

    return buffer;
}