Revert "Add message in case of Arduino disconnect at any time"
This reverts commit ff20725b0f
.
Help i broke it
This commit is contained in:
parent
ff20725b0f
commit
d8572bb331
1 changed files with 56 additions and 69 deletions
|
@ -1,11 +1,8 @@
|
||||||
|
import os
|
||||||
|
import GPUtil
|
||||||
import time
|
import time
|
||||||
|
from pySerialTransfer import pySerialTransfer as txfer
|
||||||
from subprocess import getoutput
|
from subprocess import getoutput
|
||||||
try:
|
|
||||||
import GPUtil
|
|
||||||
from pySerialTransfer import pySerialTransfer as txfer
|
|
||||||
from serial import SerialException
|
|
||||||
except ImportError:
|
|
||||||
print('Make sure you have installed the required libraries: GPUtil, pySerialTransfer')
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# ---
|
# ---
|
||||||
|
@ -15,80 +12,70 @@ try:
|
||||||
# Then, just remove "/dev/", and the number, from the port, and type it into the variable `port`
|
# Then, just remove "/dev/", and the number, from the port, and type it into the variable `port`
|
||||||
# ---
|
# ---
|
||||||
|
|
||||||
def get_port():
|
port = 'ttyACM'
|
||||||
port = 'ttyACM'
|
while True:
|
||||||
while True:
|
ports = getoutput(f'ls /dev | grep {port}').split('\n')
|
||||||
ports = getoutput(f'ls /dev | grep {port}').split('\n')
|
if ports != ['']:
|
||||||
if ports != ['']:
|
ports.sort()
|
||||||
ports.sort()
|
port = ports[len(ports) - 1]
|
||||||
port = ports[len(ports) - 1]
|
break
|
||||||
break
|
print('Waiting for Arduino...')
|
||||||
print('Waiting for Arduino...')
|
time.sleep(3)
|
||||||
time.sleep(3)
|
|
||||||
|
|
||||||
return port
|
link = txfer.SerialTransfer(port, 115200, timeout=.1)
|
||||||
|
|
||||||
link = txfer.SerialTransfer(get_port(), 115200, timeout=.1)
|
|
||||||
link.open()
|
link.open()
|
||||||
|
time.sleep(2)
|
||||||
|
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
|
time.sleep(1)
|
||||||
|
gpus = GPUtil.getGPUs()
|
||||||
|
temp = max(gpu.temperature for gpu in gpus)
|
||||||
|
|
||||||
try:
|
speed = int(((temp - 40) / 40) * 100)
|
||||||
time.sleep(1)
|
min_speed = 0
|
||||||
gpus = GPUtil.getGPUs()
|
max_speed = 100
|
||||||
temp = max(gpu.temperature for gpu in gpus)
|
if speed < 0:
|
||||||
|
speed = min_speed
|
||||||
|
elif speed > 100:
|
||||||
|
speed = max_speed
|
||||||
|
|
||||||
speed = int(((temp - 40) / 40) * 100)
|
print(f'Temp: {temp}')
|
||||||
min_speed = 34
|
print(f'Speed: {speed}')
|
||||||
max_speed = 100
|
|
||||||
if speed < 0:
|
|
||||||
speed = min_speed
|
|
||||||
elif speed > 100:
|
|
||||||
speed = max_speed
|
|
||||||
|
|
||||||
print(f'Temp: {temp}')
|
################################################################################################################
|
||||||
print(f'Speed: {speed}')
|
# Send data to the Arduino
|
||||||
|
################################################################################################################
|
||||||
|
|
||||||
################################################################################################################
|
# send_size will be increased when data is added to payload
|
||||||
# Send data to the Arduino
|
send_size = 0
|
||||||
################################################################################################################
|
|
||||||
|
|
||||||
# send_size will be increased when data is added to payload
|
# Adds data to payload
|
||||||
send_size = 0
|
int_size = link.tx_obj(speed, send_size) - send_size
|
||||||
|
send_size += int_size
|
||||||
|
|
||||||
# Adds data to payload
|
# Sends data to Arduino
|
||||||
int_size = link.tx_obj(speed, send_size) - send_size
|
link.send(send_size)
|
||||||
send_size += int_size
|
|
||||||
|
|
||||||
# Sends data to Arduino
|
# Waits for response from Arduino, and reports errors while receiving packets
|
||||||
link.send(send_size)
|
while not link.available():
|
||||||
|
if link.status < 0:
|
||||||
|
if link.status == txfer.CRC_ERROR:
|
||||||
|
print('Error: CRC_ERROR')
|
||||||
|
elif link.status == txfer.PAYLOAD_ERROR:
|
||||||
|
print('Error: PAYLOAD_ERROR')
|
||||||
|
elif link.status == txfer.STOP_BYTE_ERROR:
|
||||||
|
print('Error: STOP_BYTE_ERROR')
|
||||||
|
else:
|
||||||
|
print('Error: {}'.format(link.status))
|
||||||
|
|
||||||
# Waits for response from Arduino, and reports errors while receiving packets
|
# Parse response from Arduino
|
||||||
while not link.available():
|
rec_int = link.rx_obj(obj_type=int, obj_byte_size=int_size, start_pos=(send_size - int_size))
|
||||||
if link.status < 0:
|
|
||||||
if link.status == txfer.CRC_ERROR:
|
|
||||||
print('Error: CRC_ERROR')
|
|
||||||
elif link.status == txfer.PAYLOAD_ERROR:
|
|
||||||
print('Error: PAYLOAD_ERROR')
|
|
||||||
elif link.status == txfer.STOP_BYTE_ERROR:
|
|
||||||
print('Error: STOP_BYTE_ERROR')
|
|
||||||
else:
|
|
||||||
print('Error: {}'.format(link.status))
|
|
||||||
|
|
||||||
# Parse response from Arduino
|
|
||||||
rec_int = link.rx_obj(obj_type=int, obj_byte_size=int_size, start_pos=(send_size - int_size))
|
|
||||||
|
|
||||||
# Evaluate for comm. errors
|
|
||||||
if speed != rec_int:
|
|
||||||
print(f'Error: sent {speed}, received {rec_int}')
|
|
||||||
except OSError or SerialException:
|
|
||||||
try:
|
|
||||||
link.close()
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
link = txfer.SerialTransfer(get_port(), 115200, timeout=.1)
|
|
||||||
link.open()
|
|
||||||
|
|
||||||
|
# Evaluate for comm. errors
|
||||||
|
if speed != rec_int:
|
||||||
|
print(f'Error: sent {speed}, received {rec_int}')
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Reference in a new issue