source: dev/trunk/abdev/BasicCompiler_Common/include/CodeGenerator.h@ 247

#pragma once

#include <NativeCode.h>

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


void ReallocNativeCodeBuffer();

//Goto未知ラベル
class GotoLabelSchedule
{
    std::string name;
    int line;
    int nativeCodePos;
    int sourceCodePos;
public:
    GotoLabelSchedule( const std::string &name, int nativeCodePos, int sourceCodePos )
        : name( name )
        , line( -1 )
        , nativeCodePos( nativeCodePos )
        , sourceCodePos( sourceCodePos )
    {
    }
    GotoLabelSchedule( int line, int nativeCodePos, int sourceCodePos )
        : line( line )
        , nativeCodePos( nativeCodePos )
        , sourceCodePos( sourceCodePos )
    {
    }
    const std::string &GetName() const
    {
        return name;
    }
    int GetLineNum() const
    {
        return line;
    }
    int GetNativeCodePos() const
    {
        return nativeCodePos;
    }
    int GetSourceCodePos() const
    {
        return sourceCodePos;
    }
};

class CodeGenerator
{
    NativeCode *pNativeCode;

public:

    // コード生成時の部分的なスケジューリング
    class PertialSchedule
    {
        int codePos;    // バッファ位置
        int typeSize;   // 対象サイズ（一般的には8bit/32bit）

        int _obpOld;    // 未完成
    public:
        PertialSchedule( int codePos, int typeSize )
            : codePos( codePos )
            , typeSize( typeSize )
        {
            extern int obp;
            _obpOld = obp;
        }
        ~PertialSchedule()
        {
        }

        int GetCodePos() const
        {
            return codePos;
        }
        int GetTypeSize() const
        {
            return typeSize;
        }
        int GetObpOld() const
        {
            return _obpOld;
        }
    };

private:
    // 部分スケジュールの管理
    typedef std::vector<const PertialSchedule *> PertialSchedules;
    PertialSchedules pertialSchedules;

    // Continue用のコード位置情報の管理
    std::vector<long> continueCodePositions;
    std::vector<long> _continueCodePositions_ObpOld;

public:

    // Exit Subスケジュールの管理
    std::vector<long> exitSubCodePositions;
    std::vector<int> _exitSubCodePositions_ObpOld;

    // Gotoスケジュールの管理
    std::vector<GotoLabelSchedule> gotoLabelSchedules;

    CodeGenerator()
        : pNativeCode( 0 )
    {
    }
    ~CodeGenerator()
    {
        if( pNativeCode )
        {
            CheckUnresolveSchedule();
        }
    }

    void Select( NativeCode &nativeCode )
    {
        if( pNativeCode )
        {
            CheckUnresolveSchedule();
        }
        pNativeCode = &nativeCode;
    }

    long GetContinueCodePos() const
    {
        if( continueCodePositions.size() == 0 )
        {
            return -1;
        }
        return continueCodePositions[continueCodePositions.size()-1];
    }
    void ClearContinueArea()
    {
        continueCodePositions.clear();
        _continueCodePositions_ObpOld.clear();
    }
    void ContinueAreaBegin()
    {
        continueCodePositions.push_back( pNativeCode->GetSize() );

        extern int obp;
        _continueCodePositions_ObpOld.push_back( obp );
    }
    void ContinueAreaEnd()
    {
        continueCodePositions.pop_back();
        _continueCodePositions_ObpOld.pop_back();
    }
    long GetContinueCodePosOld() const
    {
        if( _continueCodePositions_ObpOld.size() == 0 )
        {
            return -1;
        }
        return _continueCodePositions_ObpOld[_continueCodePositions_ObpOld.size()-1];
    }
    
    void ResolveExitSubSchedule();

    void CheckUnresolveSchedule();

    void opfix_JmpPertialSchedule( const PertialSchedule *pPertialSchedule );


    /////////////////////////////////////////////////////////////////
    // 32bit/64bit共通 機械語生成
    /////////////////////////////////////////////////////////////////

private:
    const PertialSchedule *__jmp_op_format( char opcode, long offset, int op_size, bool isPertialSchedule = false, bool isSelfOpcodeOffset = false );
public:
    const PertialSchedule *op_jle( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jbe( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jge( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jae( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jl( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jb( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jg( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_ja( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jne( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_je( long offset, int op_size = sizeof(char), bool isPertialSchedule = false );
    const PertialSchedule *op_jmp( long offset, int op_size = sizeof(char), bool isPertialSchedule = false, bool isSelfOpcodeOffset = false );
    void op_jmp_continue();
    void op_jmp_exitsub();
    void op_jmp_goto_schedule( const GotoLabelSchedule &gotoLabelSchedule );


#ifdef _AMD64_
    /////////////////////////////////////////////////////////////////
    // 64ビット機械語生成
    /////////////////////////////////////////////////////////////////
private:
    void set_rex(int op_size,int reg,int index_reg,int base_reg);
    void set_mod_rm_sib_disp(char mod,int reg,int scale,int index_reg,int base_reg,long disp, Schedule::Type scheduleType = Schedule::None );
    void __op_format(int op_size,char op_prefix,char opcode1,char opcode2,int reg,int base_reg,long offset,char mod, Schedule::Type scheduleType = Schedule::None );
public:
    void op_mov_RV                  (int op_size,int reg,long i32data, Schedule::Type scheduleType = Schedule::None );
    void op_mov_RV64                (int reg,_int64 i64data);
    void op_mov_RM                  (int op_size,int reg,int base_reg,long offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_mov_RM_ex               (int op_size,int reg,int base_reg1,int base_reg2,long offset,BOOL bUseOffset, Schedule::Type scheduleType = Schedule::None );
    void op_mov_MR                  (int op_size,int reg,int base_reg,long offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_mov_MR_ex               (int op_size,int reg,int base_reg1,int base_reg2,long offset,BOOL bUseOffset, Schedule::Type scheduleType = Schedule::None );
    void op_mov_MV                  (int op_size,int base_reg,int offset, Schedule::Type offsetScheduleType, BOOL bUseOffset,long i32data);
    void op_mov_RR                  (int reg1,int reg2);
    void op_movsxd                  (int reg64,int reg32);
    void op_movsx64_FromReg16       (int reg64,int reg16);
    void op_movsx64_FromReg8        (int reg64,int reg8);
    void op_movsx32_FromReg16       (int reg32,int reg16);
    void op_movsx32_FromReg8        (int reg32,int reg8);
    void op_movsx16_FromReg8        (int reg32,int reg8);
    void op_cqo                     ();
    void op_inc                     (int reg);
    void op_dec                     (int reg);
    void op_add_RM                  (int op_size,int reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_add_RV                  (int reg,long offset, Schedule::Type scheduleType = Schedule::None );
    void op_add_RR              (int reg1,int reg2);
    void op_add32_reg               (int reg1,int reg2);
    void op_sub_RV                  (int op_size,int reg,long i32data);
    void op_sub64_reg               (int reg1,int reg2);
    void op_sub32_reg               (int reg1,int reg2);
    void op_sbb_RR                  ( int op_size, int reg1, int reg2 );
    void op_imul_RR                 (int op_size,int reg1,int reg2);
    void op_imul_RV                 (int op_size,int reg,long i32data);
    void op_div64_reg               (int reg);
    void op_idiv64_reg              (int reg);
    void op_shl_reg                 (int op_size,int reg);
    void op_sar_reg                 (int op_size,int reg);
    void op_shr_reg                 (int op_size,int reg);
    void op_and_reg                 (int op_size,int reg1,int reg2);
    void op_and64_value             (int reg,long offset);
    void op_and32_value             (int reg,long offset);
    void op_or_reg                  (int op_size,int reg1,int reg2);
    void op_xor_reg                 (int op_size,int reg1,int reg2);
    void op_not_reg                 (int op_size,int reg);
    void op_neg                     ( int reg );
    void op_test                    (int reg1,int reg2);
    void op_cmp_reg                 (int op_size,int reg1,int reg2);
    void op_cmp_value               (int op_size,int reg,char byte_data);
    void op_setne                   (int reg);
    void op_movlpd_MR               (int xmm_reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_movlpd_RM               (int xmm_reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_movsd_RR                (int xmm_reg1,int xmm_reg2);
    void op_movsd_MR                (int xmm_reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_movss_RR                (int xmm_reg1,int xmm_reg2);
    void op_movss_RM                (int xmm_reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_movss_MR                (int xmm_reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_movd_RX                 (int reg,int xmm_reg);
    void op_cvtsd2ss                (int xmm_reg1,int xmm_reg2);
    void op_cvtss2sd                (int xmm_reg1,int xmm_reg2);
    void op_cvttsd2si_xmm           (int op_size,int reg,int xmm_reg);
    void op_cvttss2si_xmm           (int op_size,int reg,int xmm_reg);
    void op_cvtsi2sd_reg            (int op_size,int xmm_reg,int reg);
    void op_cvtsi2ss_reg            (int op_size,int xmm_reg,int reg);
    void op_comisd                  (int xmm_reg1,int xmm_reg2);
    void op_comiss                  (int xmm_reg1,int xmm_reg2);
    void op_rep_movs                (int op_size);
    void op_add_rsp(long num);
    void op_sub_rsp(long num);
    void op_add_esp(long num);
    void op_sub_esp(long num);
    void op_fld_ptr_esp(int type);
    void op_zero_reg(int reg);
    void op_call( const UserProc *pUserProc );
    void op_call( const DllProc *pDllProc );
    void op_ret();

#else
    /////////////////////////////////////////////////////////////////
    // 32ビット機械語生成
    /////////////////////////////////////////////////////////////////
private:
    void set_mod_rm_sib_disp(char mod,int reg,int scale,int index_reg,int base_reg,long disp, Schedule::Type scheduleType = Schedule::None );
    void __op_format(char op_prefix,char opcode,int reg);
    void __op_format(char op_prefix,char opcode1,char opcode2,int reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
public:
    void op_mov_MV          ( int op_size, int base_reg, long offset, Schedule::Type offsetScheduleType, long value, Schedule::Type valueScheduleType = Schedule::None );
    void op_mov_RV          (int reg,long offset, Schedule::Type scheduleType = Schedule::None );
    void op_mov_RR          (int reg1,int reg2);
    void op_mov_RM          (int op_size,int reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_mov_RM_ex       (int op_size,int reg,int base_reg1,int base_reg2,long offset,BOOL bUseOffset, Schedule::Type scheduleType = Schedule::None );
    void op_mov_MR          (int op_size,int reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_mov_MR_ex       (int op_size,int reg,int base_reg1,int base_reg2,long offset,BOOL bUseOffset, Schedule::Type scheduleType = Schedule::None );
    void op_movsx_R32R16    (int reg32,int reg16 = REG_NON);
    void op_movsx_R32R8     (int reg32,int reg8 = REG_NON);
    void op_movsx_R16R8     (int reg16,int reg8 = REG_NON);
    void op_lea_RM          ( int reg, int base_reg, long offset, char mod, Schedule::Type scheduleType = Schedule::None );
    void op_inc             (int reg);
    void op_dec             (int reg);
    void op_add_RV8         (int reg,char cValue);
    void op_add_RV          ( int reg, long offset, Schedule::Type scheduleType = Schedule::None );
    void op_add_RR          ( int reg1, int reg2 );
    void op_add_RM          (int op_size,int reg,int base_reg,int offset,char mod, Schedule::Type scheduleType = Schedule::None );
    void op_adc_RV8         (int reg,char cValue);
    void op_adc_RR          ( int reg1, int reg2 );
    void op_sub_RV8         (int reg,char cValue);
    void op_sub_RR          ( int reg1, int reg2 );
    void op_sbb_RV8         (int reg,char cValue);
    void op_sbb_RR          ( int reg1, int reg2 );
    void op_imul_RR         (int reg1,int reg2);
    void op_imul_RV         (int reg,long i32data);
    void op_imul_RV8        (int reg,char cValue);
    void op_div_R           ( int reg );
    void op_idiv_R          ( int reg );
    void op_and_RV          (int reg,long value);
    void op_and_RR          ( int reg1, int reg2 );
    void op_or_RR           ( int op_size, int reg1, int reg2 );
    void op_xor_RR          ( int reg1, int reg2 = REG_NON );
    void op_neg             ( int reg );
    void op_cdq             ();

    void op_rep_movs        (int op_size);

    void op_push(int reg);
    void op_push_V( long data, Schedule::Type scheduleType = Schedule::None );
    void op_push_M( int base_reg );
    void op_push_M( int base_reg, long offset, Schedule::Type scheduleType = Schedule::None );
    void op_pop(int reg = REG_NON);
    void op_add_esp(long num);
    void op_sub_esp(long num);
    void op_cmp_RR( int reg1, int reg2 );
    void op_cmp_value(int op_size,int reg,char byte_data);
    void op_setne( int reg );
    void op_test(int reg1,int reg2);
    void op_test_ah( char cValue );
    void op_fld_ptr_esp(int type);
    void op_fld_basereg         (int type,int base_reg);
    void op_fld_base_offset     (int type,int base_reg,long offset, Schedule::Type scheduleType = Schedule::None );
    void op_fld_base_offset_ex  (int type,int base_reg1,int base_reg2,long offset,BOOL bUseOffset, Schedule::Type scheduleType = Schedule::None);
    void op_fstp_basereg        (int type,int base_reg);
    void op_fstp_base_offset    (int type,int base_reg,long offset, Schedule::Type scheduleType = Schedule::None);
    void op_fstp_base_offset_ex (int type,int base_reg1,int base_reg2,long offset,BOOL bUseOffset, Schedule::Type scheduleType = Schedule::None);
    void op_fistp_ptr_esp       ( int typeSize );
    void op_fstp_push           ( Type &type );
    void op_fcompp();
    void op_fnstsw_ax();
    void op_zero_reg(int reg);
    void fpu_cast();
    void fpu_cast_end();

    void op_call_R( int reg );
    void op_call(const UserProc *pUserProc);
    void op_ret();
    void op_ret( short stackFrameSize );
    void op_addressof( int reg, const UserProc *pUserProc );
#endif



    void PutOld( long l, Schedule::Type scheduleType )
    {
        pNativeCode->Put( l, scheduleType );
    }

    void PutOld( char c1, char c2 )
    {
        pNativeCode->Put( c1 );
        pNativeCode->Put( c2 );
    }
    void PutOld( char c1, char c2, char c3 )
    {
        pNativeCode->Put( c1 );
        pNativeCode->Put( c2 );
        pNativeCode->Put( c3 );
    }
    void PutOld( char c1, char c2, char c3, char c4 )
    {
        pNativeCode->Put( c1 );
        pNativeCode->Put( c2 );
        pNativeCode->Put( c3 );
        pNativeCode->Put( c4 );
    }
    void PutOld( char c1, char c2, char c3, long l )
    {
        pNativeCode->Put( c1 );
        pNativeCode->Put( c2 );
        pNativeCode->Put( c3 );
        pNativeCode->Put( l );
    }
    void PutOld( char c1, char c2, char c3, char c4, char c5 )
    {
        pNativeCode->Put( c1 );
        pNativeCode->Put( c2 );
        pNativeCode->Put( c3 );
        pNativeCode->Put( c4 );
        pNativeCode->Put( c5 );
    }
    void PutOld( char c1, char c2, char c3, char c4, char c5, char c6 )
    {
        pNativeCode->Put( c1 );
        pNativeCode->Put( c2 );
        pNativeCode->Put( c3 );
        pNativeCode->Put( c4 );
        pNativeCode->Put( c5 );
        pNativeCode->Put( c6 );
    }
};