pkg.py

from config import args
from enum import Enum, IntEnum, unique, auto

# Indentation
INDENT = "  "

# block id
BLOCK_ID = 0

# program info in header
PROGRAM_INFO = []

def INFO_logging(info):
  PROGRAM_INFO.append(info)

def Get_BLOCKID():
    global BLOCK_ID
    BLOCK_ID += 1
    return BLOCK_ID

# data types
@unique
class VarTypes(Enum):
    # VOID = "void"
    # BOOL = "bool"
    # UCHAR = "unsigned_char"
    # CHAR = "char"
    # USHORT = "unsigned_short"
    # SHORT = "short"
    # UINT = "unsigned_int"
    # INT = "int"
    # ULONG = "unsigned_long"
    # LONG = "long"
    # ULONGLONG = "unsigned_long_long"
    # LONGLONG = "long_long"
    # FLOAT = "float"
    # DOUBLE = "double"
    UINT8 = "uint8_t"
    INT8 = "int8_t"
    UINT16 = "uint16_t"
    INT16 = "int16_t"
    UINT32 = "uint32_t"
    INT32 = "int32_t"
    UINT64 = "uint64_t"
    INT64 = "int64_t"
  
'''
def GetVarTypesBit(stype,platform="linux",bit=64):
  if platform=="linux" and bit==64:
    VarType2Bit = {
        VarTypes.CHAR:1,
        VarTypes.UCHAR:1,
        VarTypes.SHORT:2,
        VarTypes.USHORT:2,
        VarTypes.INT:4,
        VarTypes.UINT:4,
        VarTypes.LONG:8,
        VarTypes.ULONG:8,
        VarTypes.LONGLONG:8,
        VarTypes.ULONGLONG:8
    }
    return VarType2Bit[stype]
  else:
    raise NotImplementedError()

VarTypesUSign = [
  VarTypes.UCHAR,
  VarTypes.USHORT,
  VarTypes.UINT,
  VarTypes.ULONG,
  VarTypes.ULONGLONG
]
'''

# Arithmetic operation
@unique
class MathematicalTypes(Enum):
    ADD = "+"
    SUB = "-"
    MUL = "*"
    DIV = "/"
    REM = "%"
    # ADDS = "++"
    # SUBS = "--"
    # bits 
    # AND = "&"
    # OR = "|"
    # XOR = "^"
    # # ROC = "~"
    # SRL = ">>"
    # SLL = "<<"

# Assignment operation
@unique
class AssignementTypes(Enum):
    ASSIGNE = "="
    # ADDASSIGN = "+="
    # SUBASSIGN = "-="
    # MULASSIGNE = "*="
    # DIVASSIGNE = "/="
    # REMASSIGNE = "%="
    # ANDASSIGNE = "&="
    # ORASSIGNE = "|="
    # XORASSIGNE = "^="
    # # ROCASSIGNE = "~="
    # SRLASSIGNE = ">>="
    # SLLASSIGNE = "<<="

class StatementsTypes(Enum):
    Assignement = auto()

# Data Range between different types
class TypeRanges:
    BOOL_MIN = 0
    BOOL_MAX = 1
    INT8_MIN = -128
    INT8_MAX = 127
    UINT8_MIN = 0
    UINT8_MAX = 255
    UINT16_MIN = 0
    UINT16_MAX = 65535
    INT16_MIN = -32768
    INT16_MAX = 32767
    UINT32_MIN = 0
    UINT32_MAX = 4294967295  # 1 << 32
    INT32_MIN = -2147483648
    INT32_MAX = 2147483647
    INT64_MIN = - (1 << 63)
    INT64_MAX = (1 << 63) - 1
    UINT64_MIN = 0
    UINT64_MAX = (1 << 64) - 1


'''
VarType2CType = {
  VarTypes.UCHAR:"c_ubyte",
  VarTypes.CHAR:"c_byte",
  VarTypes.USHORT:"c_ushort",
  VarTypes.SHORT:"c_short",
  VarTypes.UINT:"c_uint",
  VarTypes.INT:"c_int",
  VarTypes.ULONG:"c_ulong",
  VarTypes.LONG:"c_long",
  VarTypes.ULONGLONG:"c_ulonglong",
  VarTypes.LONGLONG:"c_longlong"
}

MathematicalTypes2FuncOp = {
  MathematicalTypes.ADD:"add",
  MathematicalTypes.SUB:"sub",
  MathematicalTypes.MUL:"mul",
  MathematicalTypes.DIV:"div",
  MathematicalTypes.REM:"rem"
}

VarTypeCalPriority = {
    VarTypes.CHAR: 0,
    VarTypes.UCHAR: 0,
    VarTypes.SHORT: 0,
    VarTypes.USHORT: 0,
    VarTypes.INT: 1,
    VarTypes.UINT: 2,
    VarTypes.LONG: 3,
    VarTypes.ULONG: 4,
    VarTypes.LONGLONG: 5,
    VarTypes.ULONGLONG: 6,
}

VarPriority2Type = {
    1: VarTypes.INT,
    2: VarTypes.UINT,
    3: VarTypes.LONG,
    4: VarTypes.ULONG,
    5: VarTypes.LONGLONG,
    6: VarTypes.ULONGLONG,
}


def get_return_type(type1:VarTypes, type2:VarTypes):
  # 参考这个 https://wiki.sei.cmu.edu/confluence/display/c/INT02-C.+Understand+integer+conversion+rules
  # 默认c11标准
  # 只适合linux64,不同平台位宽参考如下https://img.jbzj.com/file_images/article/202111/2021111010230228.png
  # 短类型需先转为int
  type1 = VarPriority2Type[max(1,VarTypeCalPriority[type1])]
  type2 = VarPriority2Type[max(1,VarTypeCalPriority[type2])]
  types = [type1,type2]
  if type1==type2:
    return type1
  elif type1 in VarTypesUSign and type2 in VarTypesUSign or type1 not in VarTypesUSign and type2 not in VarTypesUSign:
    return VarPriority2Type[max(VarTypeCalPriority[type1],VarTypeCalPriority[type2])]
  else:
    stype = type1 if type1 not in VarTypesUSign else type2
    utype = type1 if type1 in VarTypesUSign else type2
    if VarTypeCalPriority[utype]>VarTypeCalPriority[stype]:
      return utype
    elif GetVarTypesBit(stype)>GetVarTypesBit(utype):# 要求有符号能完全表达无符号所有值，则必须位宽更大
      return stype
    else:
      return VarPriority2Type[VarTypeCalPriority[stype]+1]
    
if __name__=="__main__":
  result = get_return_type(VarTypes.ULONG, VarTypes.LONGLONG)
  print(result)

'''