arduino.py

#!/usr/bin/env python3
from grabPorts import grabPorts
from text import printme, waitForEnter
from failing import errorloc
from saving_data import writeSingleValue

# OTHER'S CODE
import time
import serial
import numpy as np
import struct
import os
import keyboard

class ArdUIno(grabPorts):
    def __init__(
        self, winPort = None, num_ards = 1, usbname = None, name="Arduino"
    ):

        self.ports = grabPorts()
        self.usbname = usbname
        self.name = name
        self.ports.arduinoPort(winPort, num_ards, self.usbname)
        print(str(self.ports.arduino_ports))

        if num_ards == 1:
            try:
                self.arduino = serial.Serial(
                    self.ports.arduino_ports[0], 9600, timeout=1
                )
            except IndexError:
                print("I cannot find any arduino boards!")
        elif num_ards > 1:
            self.arduino1 = serial.Serial(self.ports.arduino_ports[0], 9600, timeout=1)
            self.arduino2 = serial.Serial(self.ports.arduino_ports[1], 9600, timeout=1)

        self.arduino.flushInput()

    def readData(self, dataParser=float):
        self.read_parsed = None

        try:
            self.read = self.arduino.readline()
            self.read_parsed = dataParser(self.read)

        except Exception as e:
            print(f"Exception from arduino readData method {e}")

################################################################################
############################# FUNCTIONS #########################################
################################################################################


def reLoad(arduino, up = 5, down = 6):
    os.system("clear")
    was_pressed = False
    printme('Position syringe pusher ("d" for down / "u" for up / "e" to move on)')
    while True:
        if keyboard.is_pressed("e"):
            break

        elif keyboard.is_pressed("d"):
            if not was_pressed:
                try:
                    arduino.arduino.write(struct.pack(">B", down))
                except Exception as e:
                    errorloc(e)
                was_pressed = True

        elif keyboard.is_pressed("u"):
            if not was_pressed:
                try:
                    arduino.arduino.write(struct.pack(">B", up))
                except Exception as e:
                    errorloc(e)
                was_pressed = True
        else:
            was_pressed = False


def shakeShutter(arduino, times, open = 1, close = 0):
    for i in np.arange(times):
        arduino.arduino.write(struct.pack(">B", open))
        printme("Open shutter")

        time.sleep(0.2)

        arduino.arduino.write(struct.pack(">B", close))

        printme("Close shutter")
        time.sleep(0.2)


def tryexceptArduino(arduino, signal):
    try:
        arduino.arduino.write(struct.pack(">B", signal))
        print(f"TALKING TO {arduino.name}")
        time.sleep(0.1)

    except Exception as e:
        os.system("clear")
        errorloc(e)
        waitForEnter(f"\n\n Press enter when {arduino.name} is fixed...")
        arduino = ArdUIno(usb_port=arduino.usb_port, n_modem=arduino.n_modem)
        arduino.arduino.flushInput()
        time.sleep(1)
        arduino.arduino.write(struct.pack(">B", signal))
        time.sleep(0.1)

    if signal == 6 and arduino.name == "syringe":
        file_path = "./data/"
        file_name = "pusher_counter"
        file = open((file_path + file_name))
        old_value = int(file.read())
        old_value += 1
        writeSingleValue(file_path, file_name, old_value)


def movePanTilt(arduino, trio_array, keydelay = 0.15, trigger_move=8):
    printme(
        f"Sending to PanTilt x: {trio_array[0]}, y: {trio_array[1]}, z: {trio_array[2]}"
    )
    try:
        arduino.arduino.write(struct.pack(">B", trigger_move))
        time.sleep(keydelay)
        arduino.arduino.write(
            struct.pack(">BBB", trio_array[0], trio_array[1], trio_array[2])
        )
    except Exception as e:
        os.system("clear")
        errorloc(e)
        waitForEnter(f"\n\n Press enter when Arduino PanTilt is fixed...")
        arduino = ArdUIno(
            usb_port=arduino.usb_port, n_modem = arduino.modem_port
        )
        arduino.arduino.flushInput()
        time.sleep(1)
        arduino.arduino.write(struct.pack(">B", trigger_move))
        time.sleep(keydelay)
        arduino.arduino.write(
            struct.pack(">BBB", trio_array[0], trio_array[1], trio_array[2])
        )

    print("TALKING TO PANTILT")