189 lines
7.5 KiB
Python
189 lines
7.5 KiB
Python
|
import numpy as np
|
|||
|
|
import os
|
|||
|
|
import copy
|
|||
|
|
from FourChZreg_define import *
|
|||
|
|
|
|||
|
|
class FourChZChipConfig(object):
|
|||
|
|
CHANNEL_OFFSET = 0x00600000
|
|||
|
|
SCALE_FACTOR = 2 ** 31
|
|||
|
|
|
|||
|
|
def __init__(self, mk_instance, **kwargs):
|
|||
|
|
self.mk = mk_instance
|
|||
|
|
self.FolderName = kwargs.get('FolderName')
|
|||
|
|
self.config_file = kwargs.get('config_file')
|
|||
|
|
self._setup_environment()
|
|||
|
|
|
|||
|
|
def _setup_environment(self):
|
|||
|
|
if self.FolderName:
|
|||
|
|
os.makedirs(self.FolderName, exist_ok=True)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def write_registers_batch(self, register_values):
|
|||
|
|
"""批量写入寄存器"""
|
|||
|
|
if isinstance(register_values, dict):
|
|||
|
|
register_values = register_values.items()
|
|||
|
|
for address, value in register_values:
|
|||
|
|
self.mk.rw_once('w', address, value, self.config_file)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def write_register(self, address, value):
|
|||
|
|
self.mk.rw_once('w', address, value, self.config_file)
|
|||
|
|
|
|||
|
|
def _calculate_tc_coefficients(self, **kwargs):
|
|||
|
|
tc_coef_set = kwargs.get('tc_coef_set')
|
|||
|
|
coef_set = TC_COEFFICIENT_SETS[tc_coef_set]
|
|||
|
|
|
|||
|
|
amp_real = coef_set['amp_real']
|
|||
|
|
amp_imag = coef_set['amp_imag']
|
|||
|
|
time_real = coef_set['time_real']
|
|||
|
|
time_imag = coef_set['time_imag']
|
|||
|
|
|
|||
|
|
sampling_rate = 3e9
|
|||
|
|
coef1, coef2 = [], []
|
|||
|
|
for ar, ai, tr, ti in zip(amp_real, amp_imag, time_real, time_imag):
|
|||
|
|
amp_coef = ar + 1j * ai
|
|||
|
|
time_coef = tr + 1j * ti
|
|||
|
|
coef1.append(amp_coef * np.exp(1e9 / sampling_rate / 2 / (1 - amp_coef) * time_coef) / (1 - amp_coef))
|
|||
|
|
coef2.append(np.exp(1e9 / sampling_rate / (1 - amp_coef) * time_coef))
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
'alpha_re': [int(c.real * self.SCALE_FACTOR) for c in coef1],
|
|||
|
|
'alpha_im': [int(c.imag * self.SCALE_FACTOR) for c in coef1],
|
|||
|
|
'beta_re': [int(c.real * self.SCALE_FACTOR) for c in coef2],
|
|||
|
|
'beta_im': [int(c.imag * self.SCALE_FACTOR) for c in coef2]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def _general_reg_config(self, **kwargs):
|
|||
|
|
# 深拷贝 kwargs 保护原始字典
|
|||
|
|
kwargs = copy.deepcopy(kwargs)
|
|||
|
|
# 自定义寄存器配置支持
|
|||
|
|
if 'custom_registers' in kwargs:
|
|||
|
|
custom_registers = kwargs.pop('custom_registers')
|
|||
|
|
first_item = custom_registers[0]
|
|||
|
|
if isinstance(first_item, tuple):
|
|||
|
|
self.write_registers_batch(custom_registers)
|
|||
|
|
elif isinstance(first_item, dict):
|
|||
|
|
for item in custom_registers:
|
|||
|
|
self.write_register(item['addr'], item['values'])
|
|||
|
|
elif isinstance(first_item, list):
|
|||
|
|
for item in custom_registers:
|
|||
|
|
self.write_register(item[0], item[1])
|
|||
|
|
|
|||
|
|
# 芯片工作模式配置
|
|||
|
|
if 'chip_mode' in kwargs:
|
|||
|
|
chip_mode = kwargs.pop('chip_mode')
|
|||
|
|
if chip_mode == 'RAMP':
|
|||
|
|
self._ramp_config(**kwargs)
|
|||
|
|
elif chip_mode == 'AWG':
|
|||
|
|
self._awg_config(**kwargs)
|
|||
|
|
|
|||
|
|
def _ramp_config(self, **kwargs):
|
|||
|
|
# 统一使用相对基地址（动态适配当前 Channel）
|
|||
|
|
channel = kwargs.get('channel')
|
|||
|
|
ctrl_base = addr_base['CTRL0_BASE'] + channel * 0x600000
|
|||
|
|
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['MODDOTR'], 8) # 切换到RAMP输出
|
|||
|
|
ramp_ctrl = kwargs.get('ramp_ctrl')
|
|||
|
|
|
|||
|
|
if ramp_ctrl == 'MCU': # RAMP连到 mcu_regfile
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['SPI_RAMPENR'], 1 << 30)
|
|||
|
|
elif ramp_ctrl == 'SPI': # RAMP连到 ctrl_regfile
|
|||
|
|
ramp_spi_registers = []
|
|||
|
|
fixed_enable = kwargs.get('fixed_enable', False)
|
|||
|
|
if fixed_enable: # 固定值模式
|
|||
|
|
fixed_value = kwargs.get('fixed_value', 0)
|
|||
|
|
ramp_spi_registers += [1 << 15 | fixed_value << 16]
|
|||
|
|
# 写寄存器函数自带了通道偏移
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['SPI_RAMPFIXR'], ramp_spi_registers)
|
|||
|
|
else: # 斜坡模式
|
|||
|
|
height = kwargs.get('height', 0)
|
|||
|
|
length = kwargs.get('step_time', 0)
|
|||
|
|
ramp_spi_registers += [0 << 15]
|
|||
|
|
ramp_spi_registers += [height << 16]
|
|||
|
|
ramp_spi_registers += [length]
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['SPI_RAMPFIXR'], ramp_spi_registers)
|
|||
|
|
|
|||
|
|
# 配置使能
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['SPI_RAMPENR'], 1 << 31 | 0 << 30)
|
|||
|
|
|
|||
|
|
def _awg_config(self, **kwargs):
|
|||
|
|
channel = kwargs.get('channel', 0)
|
|||
|
|
ctrl_base = addr_base['CTRL0_BASE'] + channel*0x600000
|
|||
|
|
tcco_base = addr_base['TCCO0_BASE'] + channel*0x600000
|
|||
|
|
dtcm_base = addr_base['DTCM0_BASE'] + channel*0x600000
|
|||
|
|
|
|||
|
|
mode_map = {'nco': 6, 'nco_fm': 7, 'env': 4, 'mod': 5}
|
|||
|
|
mode = kwargs.get('mode', None)
|
|||
|
|
moddotr_val = mode_map.get(mode, 0)
|
|||
|
|
|
|||
|
|
# 1. DSP (拖尾矫正) 配置
|
|||
|
|
tail_en = kwargs.get('tail_en', False)
|
|||
|
|
if tail_en:
|
|||
|
|
coefficients = self._calculate_tc_coefficients(**kwargs)
|
|||
|
|
tc_coef_registers = (
|
|||
|
|
coefficients['alpha_re'] +
|
|||
|
|
coefficients['alpha_im'] +
|
|||
|
|
coefficients['beta_re'] +
|
|||
|
|
coefficients['beta_im']
|
|||
|
|
)
|
|||
|
|
self.write_register(tcco_base + FourChZreg_define['tc_reg']['TCPARR0'], tc_coef_registers)
|
|||
|
|
tccdr_addr = tcco_base + FourChZreg_define['tc_reg']['TCCDR']
|
|||
|
|
for i in range(8):
|
|||
|
|
self.write_register(tccdr_addr, 1 << i)
|
|||
|
|
moddotr_val &= ~4 # 把[2]清零
|
|||
|
|
|
|||
|
|
# 2. 数据选择寄存器 MODDOTR 配置
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['MODDOTR'], moddotr_val)
|
|||
|
|
|
|||
|
|
# 3. 调制使能寄存器 MODENR 配置
|
|||
|
|
amp_mod_enable = kwargs.get('amp_mod_enable', False)
|
|||
|
|
freq_mod_enable = kwargs.get('freq_mod_enable', False)
|
|||
|
|
bias_enable = kwargs.get('bias_enable', False)
|
|||
|
|
|
|||
|
|
mod_enable = (
|
|||
|
|
(int(not bias_enable) << 0)
|
|||
|
|
| (int(not freq_mod_enable) << 1)
|
|||
|
|
| (int(not amp_mod_enable) << 2)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
self.write_register(ctrl_base + FourChZreg_define['ctrl_reg']['MODENR'], mod_enable)
|
|||
|
|
|
|||
|
|
# 4. mcu_regfile 配置
|
|||
|
|
mcu_registers = []
|
|||
|
|
fcw_list = kwargs.get('fcw', [0])
|
|||
|
|
mcu_reg_clr = kwargs.get('mcu_reg_clr', False)
|
|||
|
|
pcw_list = kwargs.get('pcw', [0])
|
|||
|
|
|
|||
|
|
for fcw in fcw_list:
|
|||
|
|
mcu_registers.append(int(fcw / fs / 4 * 2 ** 32))
|
|||
|
|
|
|||
|
|
mcu_registers.append(1 << 31 if mcu_reg_clr else 0 << 31)
|
|||
|
|
|
|||
|
|
for pcw in pcw_list:
|
|||
|
|
mcu_registers.append(int(pcw / 360 * (2 ** 16 - 1)) << 16)
|
|||
|
|
|
|||
|
|
rz_pha = kwargs.get('rz_pha', 0)
|
|||
|
|
mcu_registers.append(int(rz_pha / 360 * (2 ** 16 - 1)))
|
|||
|
|
|
|||
|
|
ff_amp_list = kwargs.get('ff_amp', [0])
|
|||
|
|
fm_amp_list = kwargs.get('fm_amp', [0])
|
|||
|
|
for ff, fm in zip(ff_amp_list, fm_amp_list):
|
|||
|
|
mcu_registers.append(((ff & 0xFFFF) << 16) | (fm & 0xFFFF))
|
|||
|
|
|
|||
|
|
bias_list = kwargs.get('bias', [0])
|
|||
|
|
for bias in bias_list:
|
|||
|
|
mcu_registers.append(bias << 16)
|
|||
|
|
|
|||
|
|
self.write_register(dtcm_base + FourChZreg_define['mcu_reg']['CWFR0'], mcu_registers)
|
|||
|
|
|
|||
|
|
# 5. FMER 配置
|
|||
|
|
fm_en = kwargs.get('fm_en', False)
|
|||
|
|
self.write_register(dtcm_base + FourChZreg_define['mcu_reg']['FMER'], 1 << 31 if fm_en else 0)
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def config_chip_reg(mk_instance, **kwargs):
|
|||
|
|
config = FourChZChipConfig(mk_instance, **kwargs)
|
|||
|
|
config._general_reg_config(**kwargs)
|