serial_comm.py

import serial
import time

import argparse

from vars import *

def pack_channel(gain=3,
                 filt=3,
                 dcOffset=0,
                 testEnable=0,
                 outputCoupling=0,
                 outputBuffer=0):

    '''
    Contructs the word to set a certain channel
    '''

    ret = \
    ( ( ( gain & SG_M )          << SG_S  ) | 
    ( ( filt & ST_M )          << ST_S  ) | 
    ( ( dcOffset & SNC_M )       << SNC_S ) | 
    ( ( testEnable & STS_M )     << STS_S ) |
    ( ( outputCoupling & SDC_M ) << SDC_S ) |
    ( ( outputBuffer & SDF_M )   << SDF_S ) )

    return ret

def get_word(plane='induction'):
    '''
    Returns the channel word for the induction or collection plane
    '''

    word = pack_channel(dcOffset=0 if plane == 'induction' else 1)

    word = hex(word)

    word = word.upper()[2:]

    print('Word is', word)

    return word


def set_channels(ser):
    '''
    Sets all the 480 channels.
    Use this if all 10 boards are connected.
    '''

    print('Setting induction plane')

    word = get_word(plane='induction')

    cmd = f'edit {word} 0 {480/2-1:.0f}\r'

    ser.write(cmd.encode())
    printlines(ser)


    print('Setting collection plane')

    word = get_word(plane='collection')

    cmd = f'edit {word} {480/2:.0f} {480-1:.0f}\r'

    ser.write(cmd.encode())
    printlines(ser)


def set_channels_oneboard(ser):
    '''
    Sets 48 channels in a single board.
    Use this if only one board is connected.
    '''

    print('Setting channels for a single board')

    word = get_word(plane='collection')

    cmd = f'edit {word} 0 {48-1:.0f}\r'

    ser.write(cmd.encode())
    printlines(ser)


def set_test_channel(ser, ch):
    '''
    Sets channel ch to be a test channel
    '''

    # Get default word
    word = get_word(plane='collection')

    # str to int
    word = int(word, 16)

    # Set test bit to 1
    word |= (1 << STS_S)

    # To hex str
    word = hex(word)

    word = word.upper()[2:]

    print('Updated word is', word)

    cmd = f'edit {word} {ch} {ch}\r'

    ser.write(cmd.encode())
    printlines(ser)


def set_all_test_channels(ser):
    '''
    Sets all channels to be test channels
    '''

    # Get default word
    word = get_word(plane='collection')

    # str to int
    word = int(word, 16)

    # Set test bit to 1
    word |= (1 << STS_S)

    # To hex str
    word = hex(word)

    word = word.upper()[2:]

    print('Updated word is', word)

    cmd = f'edit {word} 0 {48-1:.0f}\r'

    ser.write(cmd.encode())
    printlines(ser)




def printlines(ser):
    '''
    Prints all lines from the serial chain
    '''

    lines = ser.readlines()

    for line in lines:
        print(str(line))


def serial_test():
    '''
    Runs a simple test
    '''

    print('Opening serial...')
    # ser = serial.Serial('/dev/tty.usbmodem123451', 19200, timeout=1)
    ser = serial.Serial('/dev/ttyACM0', 19200, timeout=1)
    print('...serial is open.')


    print('HELP:')
    ser.write(b'help\r')
    printlines(ser)

    print()

    print('Is TEST on?')
    ser.write(b'test\r')
    printlines(ser)



def test_all_channels():
    print('Opening serial...')
    # ser = serial.Serial('/dev/tty.usbmodem123451', 19200, timeout=1)
    ser = serial.Serial('/dev/ttyACM0', 19200, timeout=1)
    print('...serial is open.')

    print('HELP:')
    ser.write(b'help\r')
    printlines(ser)

    print()

    print('Is TEST on?')
    ser.write(b'test\r')
    printlines(ser)


    print()

    print('Current configuration:')
    ser.write(b'print\r')
    printlines(ser)

    set_channels_oneboard(ser)
    set_all_test_channels(ser)

    # Have to shift twice to make it work

    ser.write(b'shift\r')
    printlines(ser)

    ser.write(b'shift\r')
    printlines(ser)

    
    print('Reading back from the serial chain:')
    ser.write(b'readback\r')
    printlines(ser)

    print('Starting test pulser')
    ser.write(b'teston 200\r')
    printlines(ser)

if __name__ == "__main__":

    # serial_test()
    test_all_channels()