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>LB;2b0;1bCH;1bFS;3b>WF;5b>TV;4b>PM;4b>NM;4b>SR',
    'UMC': '3b001#umc_parameter;3b>AG;4b>PW;6b>TT;2b>SB;6b>TF;1bCA;1bCB;1bCC;1bCD;4b>CHA;4b>CHB;4b>CHC;4b>CHD'
}

NEULIVE_PARAMETER = {
    'LB': 0,         # LOW BAND FREQEUNCY
    'CH': 0,         # CHOPPER
    'FS': 0,         # FAST SETTLE
    'WF': 1,         # WAVEFORM : POS, NEG, P2N...etc.
    'TV': 0,         # STIMULATION VOLTAGE
    'PM': 0,         # P CHANNEL SELECT
    'NM': 0,         # N CHANNEL SELECT
    'SR': 0,         # ADC SAMPLING RATE
    'AG': 5,         # AMP GAIN
    'PW': 0,         # PULSE WIDTH
    'TT': 9,         # STIMULATION TIMES
    'SB': 2,         # STIMULATION BEHAVIOR
    'TF': 1,         # STIMULATION FREQUENCY
    'CA': 1,         # AMP CHANNEL A ENABLE
    'CB': 1,         # AMP CHANNEL B ENABLE
    'CC': 1,         # AMP CHANNEL C ENABLE
    'CD': 1,         # AMP CHANNEL D ENABLE
    'CHA': 9,        # AMP CHANNEL A
    'CHB': 1,        # AMP CHANNEL B
    'CHC': 2,        # AMP CHANNEL C
    'CHD': 3,        # AMP CHANNEL D
}


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