import frappy.core as fc

enabled_modules = [ 'TNMR', 'Razorbill', 'Capacitance', 'NetAnalyser' ] # default
#enabled_modules = [ 'Capacitance' ]

Node('uniaxial_nmr.psi.ch', 
    'The NMR system running the Scout and controlled with TNMR. Also contains a uniaxial cell (Razorbill RP100), a network analyser (ZVL), and a capacitance reader (for more direct measurements of the strain cell\'s status) (TSSOP16 with Arduino Nano V3)', 
    interface='tcp://5000')

# TNMR
if('TNMR' in enabled_modules):
    Mod('tnmr_otf_module', 'frappy_psi.tnmr.OTFModule.ProgrammedSequence', 'NMR Sequence')

# Razorbill RP100
if('Razorbill' in enabled_modules):
    Mod('io_razorbill',
        'frappy_psi.uniaxial_cell.RP100.RP100IO',
        'communication',
        uri='serial://COM3?baudrate=9600+bytesize=8+parity=none+stopbits=1')
    Mod('RP100Node_CH1', 'frappy_psi.uniaxial_cell.RP100.VoltageChannel', 'Razorbill RP100 PSU (CH1)', channel=1, io='io_razorbill')
    Mod('RP100Node_CH2', 'frappy_psi.uniaxial_cell.RP100.VoltageChannel', 'Razorbill RP100 PSU (CH2)', channel=2, io='io_razorbill')

# Capacitance readings (TSSOP16 with Arduino Nano V3)
if('Capacitance' in enabled_modules):
    Mod('io_capacitance',
        'frappy_psi.capacitance_readings.TSSOP16.TSSOP16_IO',
        'communication',
        uri='serial://COM5?baudrate=9600+bytesize=8+parity=none+stopbits=1')

    Mod('TSSOP16', 'frappy_psi.capacitance_readings.TSSOP16.TSSOP16', 'Capacitance-reading Arduino (with TSSOP16)', io='io_capacitance')

# ZVL Network Analyser
if('NetAnalyser' in enabled_modules):
    Mod('ZVLNode', 'frappy_psi.network_analysers.ZVL.ZVLNode.ZVLNode', 'ZVL Network Analyser', analyser_ip=Param('129.129.156.201')) # must be connected on the same ethernet network