from devlib.util import print_instruction

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_NUM_INFINITE = 1000

STI_INSTRUCTION = {
    'GLOBAL': '3b001#header#;1b0;12bCURRENT',
    'LOCAL': '3b010#header;1b0;4b>CHANNEL;3b>STI_MODE;1b>PRECISION;12b>STI_FREQ;10b>STI_PW;10b>STI_PW_IPI;10b>STI_NUM'
}

STI_PARAMETER = {
    'CURRENT': 10,  # uA
    'CHANNEL': 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': STI_NUM_INFINITE,
}

print_instruction(STI_INSTRUCTION['GLOBAL'], STI_PARAMETER, c_style_uint8_array='buf')
print_instruction(STI_INSTRUCTION['LOCAL'], STI_PARAMETER, c_style_uint8_array='buf')