Ta-Shun Su
116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
from typing import Dict, Any
|
|
|
|
from devlib.util import print_instruction, DeviceInstruction
|
|
|
|
CHANNEL_STI_0 = 0
|
|
CHANNEL_STI_1 = 1
|
|
|
|
STI_WAVEFORM_POS = 0x00
|
|
STI_WAVEFORM_NEG = 0x01
|
|
STI_WAVEFORM_P2N = 0x02
|
|
STI_WAVEFORM_N2P = 0x03
|
|
|
|
STI_PRECISION_10 = 0
|
|
STI_PRECISION_01 = 1
|
|
|
|
STI_INSTRUCTION = {
|
|
'GLOBAL': '5b11111;2b0#GV;12bCURRENT',
|
|
'LOCAL': '5b11111;2b01#LV;3b>CHANNEL;1b>PRECISION;3b>STI_MODE;12b>STI_FREQ;10b>STI_PW;10b>STI_PW_IPI;10b>STI_NUM',
|
|
'CHANNEL': '5b11111;2b10#CE;ba0;ba1;ba2;ba3;ba4;ba5;ba6;ba7;ba8;ba9;baa;bab;bac;bad;bae;baf'
|
|
}
|
|
|
|
DEFAULT_PARAMETER = {
|
|
'CURRENT': 10, # uA
|
|
'CHANNEL': CHANNEL_STI_0,
|
|
'STI_MODE': STI_WAVEFORM_POS,
|
|
'PRECISION': STI_PRECISION_10,
|
|
'STI_FREQ': 1000, # Hz
|
|
'STI_PW': 100, # us
|
|
'STI_PW_IPI': 50, # us
|
|
'STI_NUM': 900,
|
|
}
|
|
|
|
|
|
def validate_frequency_parameter(parameter: Dict[str, Any]) -> bool:
|
|
p = int(parameter['PRECISION'])
|
|
f = int(parameter['STI_FREQ'])
|
|
w = int(parameter['STI_PW'])
|
|
i = int(parameter['STI_PW_IPI'])
|
|
return 1e6 * (1 if p == 0 else 10) / f >= 2 * w + i
|
|
|
|
|
|
def get_channel_enable_parameter(*channel: int):
|
|
return {
|
|
'a0': 1 if 0 in channel else 0,
|
|
'a1': 1 if 1 in channel else 0,
|
|
'a2': 1 if 2 in channel else 0,
|
|
'a3': 1 if 3 in channel else 0,
|
|
'a4': 1 if 4 in channel else 0,
|
|
'a5': 1 if 5 in channel else 0,
|
|
'a6': 1 if 6 in channel else 0,
|
|
'a7': 1 if 7 in channel else 0,
|
|
'a8': 1 if 8 in channel else 0,
|
|
'a9': 1 if 9 in channel else 0,
|
|
'aa': 1 if 10 in channel else 0,
|
|
'ab': 1 if 11 in channel else 0,
|
|
'ac': 1 if 12 in channel else 0,
|
|
'ad': 1 if 13 in channel else 0,
|
|
'ae': 1 if 14 in channel else 0,
|
|
'af': 1 if 15 in channel else 0,
|
|
}
|
|
|
|
|
|
# -------------------------------------------------------------------------------------------------------------------- #
|
|
|
|
STI_PARAMETER_0 = {
|
|
'CURRENT': 10, # uA
|
|
'CHANNEL': CHANNEL_STI_0,
|
|
'STI_MODE': STI_WAVEFORM_POS,
|
|
'PRECISION': STI_PRECISION_10,
|
|
'STI_FREQ': 10, # Hz
|
|
'STI_PW': 10, # us
|
|
'STI_PW_IPI': 10, # us
|
|
'STI_NUM': 900,
|
|
}
|
|
|
|
STI_PARAMETER_1 = {
|
|
'CURRENT': 10, # uA
|
|
'CHANNEL': CHANNEL_STI_1,
|
|
'STI_MODE': STI_WAVEFORM_NEG,
|
|
'PRECISION': STI_PRECISION_10,
|
|
'STI_FREQ': 10, # Hz
|
|
'STI_PW': 10, # us
|
|
'STI_PW_IPI': 10, # us
|
|
'STI_NUM': 900,
|
|
}
|
|
|
|
print_instruction(STI_INSTRUCTION['GLOBAL'], STI_PARAMETER_0, append_ris_type=False)
|
|
|
|
valid_channel_0 = validate_frequency_parameter(STI_PARAMETER_0)
|
|
valid_channel_1 = validate_frequency_parameter(STI_PARAMETER_1)
|
|
enable_channel = []
|
|
|
|
if not valid_channel_0:
|
|
print('invalidate parameter')
|
|
else:
|
|
print('----')
|
|
print_instruction(STI_INSTRUCTION['LOCAL'], STI_PARAMETER_0, append_ris_type=False)
|
|
enable_channel.append(0)
|
|
|
|
if not valid_channel_1:
|
|
print('invalidate parameter')
|
|
else:
|
|
print('----')
|
|
print_instruction(STI_INSTRUCTION['LOCAL'], STI_PARAMETER_1, append_ris_type=False)
|
|
enable_channel.append(1)
|
|
|
|
if valid_channel_0 or valid_channel_1:
|
|
print('----')
|
|
print_instruction(STI_INSTRUCTION['CHANNEL'], get_channel_enable_parameter(*enable_channel), append_ris_type=False)
|
|
|
|
print('--sti--')
|
|
print_instruction(DeviceInstruction.VIS_STI, {})
|
|
|
|
print('--int--')
|
|
print_instruction(DeviceInstruction.VIS_INT, {})
|