319 lines
12 KiB
Python
319 lines
12 KiB
Python
import numpy as np
|
||
|
||
class make_inst(object):
|
||
|
||
def parse_instruction(self, instruction, labels, pc):
|
||
# 去掉所有逗号
|
||
instruction = instruction.replace(',', ' ')
|
||
parts = instruction.split()
|
||
opcode = parts[0].upper().strip()
|
||
|
||
if opcode.endswith(':'):
|
||
# 处理标签
|
||
label_name = opcode[:-1]
|
||
labels[label_name] = pc
|
||
return None
|
||
|
||
operands = [op.strip() for op in parts[1:]]
|
||
|
||
def parse_immediate(imm_str):
|
||
try:
|
||
if imm_str.startswith('0x') or imm_str.startswith('0X'):
|
||
return int(imm_str, 16)
|
||
elif imm_str.startswith('0b') or imm_str.startswith('0B'):
|
||
return int(imm_str, 2)
|
||
elif imm_str.startswith('-0x') or imm_str.startswith('-0X'):
|
||
return -int(imm_str[1:], 16)
|
||
elif imm_str.startswith('-0b') or imm_str.startswith('-0B'):
|
||
return -int(imm_str[1:], 2)
|
||
elif imm_str.startswith('-'):
|
||
return -int(imm_str[1:], 10)
|
||
else:
|
||
return int(imm_str, 10)
|
||
except ValueError:
|
||
raise ValueError(f"Invalid immediate value: {imm_str}")
|
||
|
||
if opcode == 'LUI':
|
||
rd, imm = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
return format((self.opcode_map[opcode]) | (imm & 0xFFFFF) << 12 | (rd << 7), '032b')
|
||
|
||
elif opcode == 'AUIPC':
|
||
rd, imm = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
return format((self.opcode_map[opcode]) | (imm & 0xFFFFF) << 12 | (rd << 7), '032b')
|
||
|
||
elif opcode == 'JAL':
|
||
rd, label_or_imm = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
if label_or_imm.lstrip('-').isdigit() or label_or_imm.startswith(('0x', '0X', '0b', '0B', '-0x', '-0X', '-0b', '-0B')):
|
||
imm = parse_immediate(label_or_imm)
|
||
else:
|
||
imm = labels.get(label_or_imm.upper().strip(), 0) - pc
|
||
imm_bits = (((imm >> 20) & 0x1) << 19) | (((imm >> 1) & 0x3FF) << 9) | (((imm >> 11) & 0x1) << 8) | ((imm >> 12) & 0xFF)
|
||
return format((self.opcode_map[opcode]) | (imm_bits) << 12 | (rd << 7), '032b')
|
||
|
||
elif opcode == 'JALR':
|
||
rd = operands[0]
|
||
operands[1] = operands[1].rstrip(')')
|
||
imm, rs1 = operands[1].split('(')
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
return format((self.opcode_map[opcode]) | (imm & 0xFFF) << 20 | (rs1 << 15) | (funct3 << 12) | (rd << 7), '032b')
|
||
|
||
elif opcode in ['BEQ', 'BNE', 'BLT', 'BGE', 'BLTU', 'BGEU']:
|
||
rs1, rs2, label_or_imm = operands
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
rs2 = int(rs2[1:]) # 去掉寄存器名称前的 'x'
|
||
if label_or_imm.lstrip('-').isdigit() or label_or_imm.startswith(('0x', '0X', '0b', '0B', '-0x', '-0X', '-0b', '-0B')):
|
||
imm = parse_immediate(label_or_imm)
|
||
else:
|
||
imm = labels.get(label_or_imm.upper().strip(), 0) - pc
|
||
imm_high_bits = (((imm >> 12) & 0x1) << 6) | (((imm >> 5 ) & 0x3F))
|
||
imm_low_bits = (((imm >> 1 ) & 0xF) << 1) | (((imm >> 11) & 0x1 ))
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
return format((self.opcode_map[opcode]) | (imm_high_bits << 25) | (rs2 << 20) | (rs1 << 15) | (funct3 << 12) | (imm_low_bits << 7), '032b')
|
||
|
||
elif opcode in ['LB', 'LH', 'LW', 'LBU', 'LHU']:
|
||
rd = operands[0]
|
||
operands[1] = operands[1].rstrip(')')
|
||
imm, rs1 = operands[1].split('(')
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
return format((self.opcode_map[opcode]) | ((imm & 0xFFF) << 20) | (rs1 << 15) | (funct3 << 12) | (rd << 7), '032b')
|
||
|
||
elif opcode in ['SB', 'SH', 'SW']:
|
||
rs2 = operands[0]
|
||
operands[1] = operands[1].rstrip(')')
|
||
imm, rs1 = operands[1].split('(')
|
||
rs2 = int(rs2[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
return format((self.opcode_map[opcode]) | (((imm >> 5) & 0x7F) << 25) | (rs2 << 20) | (rs1 << 15) | (funct3 << 12) | ((imm & 0x1F) << 7), '032b')
|
||
|
||
elif opcode in ['ADDI', 'SLTI', 'SLTIU', 'XORI', 'ORI', 'ANDI']:
|
||
rd, rs1, imm = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
return format((self.opcode_map[opcode]) | ((imm & 0xFFF) << 20) | (rs1 << 15) | (funct3 << 12) | (rd << 7), '032b')
|
||
|
||
elif opcode in ['SLLI', 'SRLI', 'SRAI']:
|
||
rd, rs1, shamt = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
shamt = parse_immediate(shamt)
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
funct7 = self.opcode_funct7_map[opcode]
|
||
return format((self.opcode_map[opcode]) | (funct7 << 25) | ((shamt & 0x1F) << 20) | (rs1 << 15) | (funct3 << 12) | (rd << 7), '032b')
|
||
|
||
elif opcode in ['ADD', 'SUB', 'SLL', 'SLT', 'SLTU', 'XOR', 'SRL', 'SRA', 'OR', 'AND']:
|
||
rd, rs1, rs2 = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
rs2 = int(rs2[1:]) # 去掉寄存器名称前的 'x'
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
funct7 = self.opcode_funct7_map[opcode]
|
||
return format((self.opcode_map[opcode]) | (funct7 << 25) | (rs2 << 20) | (rs1 << 15) | (funct3 << 12) | (rd << 7), '032b')
|
||
|
||
elif opcode in ['WAIT', 'SEND', 'SENDC']:
|
||
rd, rs1, imm = operands
|
||
rd = int(rd[1:]) # 去掉寄存器名称前的 'x'
|
||
rs1 = int(rs1[1:]) # 去掉寄存器名称前的 'x'
|
||
imm = parse_immediate(imm)
|
||
funct3 = self.opcode_funct3_map[opcode]
|
||
return format((self.opcode_map[opcode]) | ((imm & 0xFFF) << 20) | (rs1 << 15) | (funct3 << 12) | (rd << 7), '032b')
|
||
|
||
elif opcode in ['EXIT']:
|
||
return format((self.opcode_map[opcode]), '032b')
|
||
|
||
elif opcode in ['EXIT_IR']:
|
||
return '00000000000000000001000000101011'
|
||
|
||
else:
|
||
raise ValueError(f"Unsupported opcode: {opcode}")
|
||
|
||
def write(self, instructions, file_name, exaddr = 1, chip_id = 0, channel_id = 0, pc_start = 0, ard_flag = 0, show = False):
|
||
if instructions == "":
|
||
pass
|
||
else:
|
||
labels = {}
|
||
binary_instructions = []
|
||
|
||
# 将整段汇编代码拆成多条指令组成的字符串数组
|
||
inst_list = instructions.split('\n')
|
||
instructions = []
|
||
for this_inst in inst_list:
|
||
this_inst = this_inst.split('#')
|
||
if this_inst[0].strip() != '':
|
||
instructions.append(this_inst[0].strip())
|
||
|
||
# 第一遍扫描：记录标签位置
|
||
pc = pc_start
|
||
for instr in instructions:
|
||
binary = self.parse_instruction(instr, labels, pc)
|
||
if binary is not None:
|
||
binary_instructions.append(binary)
|
||
pc += 4
|
||
else:
|
||
# 如果是标签，不增加pc
|
||
pass
|
||
|
||
# 第二遍扫描：生成最终的二进制代码
|
||
pc = pc_start
|
||
final_binary_instructions = []
|
||
for instr in instructions:
|
||
binary = self.parse_instruction(instr, labels, pc)
|
||
if binary is not None:
|
||
final_binary_instructions.append(binary)
|
||
if show:
|
||
print(f"{instr}: {binary}")
|
||
pc += 4
|
||
else:
|
||
# 如果是标签，不增加pc
|
||
pass
|
||
|
||
with open(file_name, "a") as f:
|
||
base_addr = 0x010_0000 + pc_start + channel_id * 0x060_0000
|
||
length = np.size(final_binary_instructions)
|
||
f.write(f"{((ard_flag << 30) | (chip_id << 25) | (base_addr)):08x}\n")
|
||
f.write(f"{((exaddr << 20) | (length<<2)):08x}\n")
|
||
for binary_instr in final_binary_instructions:
|
||
f.write(f"{int(binary_instr,2):08x}\n")
|
||
f.write("\n")
|
||
|
||
return final_binary_instructions
|
||
|
||
opcode_map = {
|
||
'LUI': 0x37,
|
||
'AUIPC': 0x17,
|
||
'JAL': 0x6F,
|
||
'JALR': 0x67,
|
||
'BEQ': 0x63,
|
||
'BNE': 0x63,
|
||
'BLT': 0x63,
|
||
'BGE': 0x63,
|
||
'BLTU': 0x63,
|
||
'BGEU': 0x63,
|
||
'LB': 0x03,
|
||
'LH': 0x03,
|
||
'LW': 0x03,
|
||
'LBU': 0x03,
|
||
'LHU': 0x03,
|
||
'SB': 0x23,
|
||
'SH': 0x23,
|
||
'SW': 0x23,
|
||
'ADDI': 0x13,
|
||
'SLTI': 0x13,
|
||
'SLTIU': 0x13,
|
||
'XORI': 0x13,
|
||
'ORI': 0x13,
|
||
'ANDI': 0x13,
|
||
'SLLI': 0x13,
|
||
'SRLI': 0x13,
|
||
'SRAI': 0x13,
|
||
'ADD': 0x33,
|
||
'SUB': 0x33,
|
||
'SLL': 0x33,
|
||
'SLT': 0x33,
|
||
'SLTU': 0x33,
|
||
'XOR': 0x33,
|
||
'SRL': 0x33,
|
||
'SRA': 0x33,
|
||
'OR': 0x33,
|
||
'AND': 0x33,
|
||
'WAIT': 0x0B,
|
||
'SEND': 0x0B,
|
||
'SENDC': 0x0B,
|
||
'EXIT': 0x2B,
|
||
}
|
||
|
||
opcode_funct3_map = {
|
||
'JALR': 0x0,
|
||
'BEQ': 0x0,
|
||
'BNE': 0x1,
|
||
'BLT': 0x4,
|
||
'BGE': 0x5,
|
||
'BLTU': 0x6,
|
||
'BGEU': 0x7,
|
||
'LB': 0x0,
|
||
'LH': 0x1,
|
||
'LW': 0x2,
|
||
'LBU': 0x4,
|
||
'LHU': 0x5,
|
||
'SB': 0x0,
|
||
'SH': 0x1,
|
||
'SW': 0x2,
|
||
'ADDI': 0x0,
|
||
'SLTI': 0x2,
|
||
'SLTIU': 0x3,
|
||
'XORI': 0x4,
|
||
'ORI': 0x6,
|
||
'ANDI': 0x7,
|
||
'SLLI': 0x1,
|
||
'SRLI': 0x5,
|
||
'SRAI': 0x5,
|
||
'ADD': 0x0,
|
||
'SUB': 0x0,
|
||
'SLL': 0x1,
|
||
'SLT': 0x2,
|
||
'SLTU': 0x3,
|
||
'XOR': 0x4,
|
||
'SRL': 0x5,
|
||
'SRA': 0x5,
|
||
'OR': 0x6,
|
||
'AND': 0x7,
|
||
'WAIT': 0x0,
|
||
'SEND': 0x2,
|
||
'SENDC': 0x3,
|
||
'EXIT': 0x0,
|
||
}
|
||
|
||
opcode_funct7_map = {
|
||
'JALR': 0x00,
|
||
'BEQ': 0x00,
|
||
'BNE': 0x00,
|
||
'BLT': 0x00,
|
||
'BGE': 0x00,
|
||
'BLTU': 0x00,
|
||
'BGEU': 0x00,
|
||
'LB': 0x00,
|
||
'LH': 0x00,
|
||
'LW': 0x00,
|
||
'LBU': 0x00,
|
||
'LHU': 0x00,
|
||
'SB': 0x00,
|
||
'SH': 0x00,
|
||
'SW': 0x00,
|
||
'ADDI': 0x00,
|
||
'SLTI': 0x00,
|
||
'SLTIU': 0x00,
|
||
'XORI': 0x00,
|
||
'ORI': 0x00,
|
||
'ANDI': 0x00,
|
||
'SLLI': 0x00,
|
||
'SRLI': 0x00,
|
||
'SRAI': 0x20,
|
||
'ADD': 0x00,
|
||
'SUB': 0x20,
|
||
'SLL': 0x00,
|
||
'SLT': 0x00,
|
||
'SLTU': 0x00,
|
||
'XOR': 0x00,
|
||
'SRL': 0x00,
|
||
'SRA': 0x20,
|
||
'OR': 0x00,
|
||
'AND': 0x00,
|
||
'WAIT': 0x00,
|
||
'SEND': 0x00,
|
||
'SENDC': 0x00,
|
||
'EXIT': 0x00,
|
||
} |