from devlib.util import print_instruction

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,
    'CHANNEL':    0,
    'STI_MODE':   3,
    'PRECISION':  0,
    'STI_FREQ':   10,
    'STI_PW':     10,
    'STI_PW_IPI': 10,
    'STI_NUM':    10,
}

NEULIVE_INSTRUCTION = {
    'RIS': '3b101#header#;2b0;3b>LOW_FREQ_BAND;4b0;1bCHOPPER;1bFAST_SETTLE;3b>STI_WAVEFORM;5b>STI_VOLT;4b>STI_PMOS;4b>STI_NMOS;4b>SAMPLE_RATE',
    'UMC': '3b001#umc_parameter#;3b>AMP_GAIN;4b>PULSE_WIDTH;6b>STI_TIMES;2b>STI_BEHAVIOR;6b>STI_FREQ;1bCA;1bCB;1bCC;1bCD;4b>CHA;4b>CHB;4b>CHC;4b>CHD'
}

NEULIVE_PARAMETER = {
    'LOW_FREQ_BAND' : 0,
    'CHOPPER'       : 0,
    'FAST_SETTLE'   : 0,
    'STI_WAVEFORM'  : 0,
    'STI_VOLT'      : 0,
    'STI_PMOS'      : 15,
    'STI_NMOS'      : 0,
    'SAMPLE_RATE'   : 0,
    'AMP_GAIN'      : 5,
    'PULSE_WIDTH'   : 0,
    'STI_TIMES'     : 0,
    'STI_BEHAVIOR'  : 0,
    'STI_FREQ'      : 3,
}


print_instruction(NEULIVE_INSTRUCTION['RIS'],NEULIVE_PARAMETER)
print_instruction(NEULIVE_INSTRUCTION['UMC'],NEULIVE_PARAMETER)

print_instruction(STI_INSTRUCTION['GLOBAL'], STI_PARAMETER)
print_instruction(STI_INSTRUCTION['LOCAL'], STI_PARAMETER)