Add temp and speed vars
This commit is contained in:
parent
c3de89f678
commit
d5bd3acd8a
1 changed files with 82 additions and 0 deletions
82
main.py
Normal file
82
main.py
Normal file
|
@ -0,0 +1,82 @@
|
|||
import GPUtil
|
||||
import time
|
||||
|
||||
from pySerialTransfer import pySerialTransfer as txfer
|
||||
|
||||
try:
|
||||
# ---
|
||||
# You will need to customize port to your own computer; to find your port:
|
||||
# With your Arduino plugged in, open Arduino IDE, and click "Select Board". Then, you can see the port.
|
||||
# On Linux, it will be "/dev/ttyXXX#", like "/dev/ttyACM0" or "/dev/ttyUSB0". I think macOS is the same.
|
||||
# On Windows, it will be "COM#", like "COM3"
|
||||
# ---
|
||||
port = '/dev/ttyACM0'
|
||||
link = txfer.SerialTransfer(port, 115200, timeout=.1)
|
||||
|
||||
link.open()
|
||||
time.sleep(2)
|
||||
|
||||
# GPUtil setup
|
||||
gpus = GPUtil.getGPUs()
|
||||
|
||||
while True:
|
||||
time.sleep(1)
|
||||
temp = max(gpu.temperature for gpu in gpus)
|
||||
|
||||
speed = int(((temp - 40) / 40) * 100)
|
||||
if speed < 40:
|
||||
speed = 40
|
||||
elif speed > 100:
|
||||
speed = 100
|
||||
|
||||
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_size = 0
|
||||
|
||||
# Adds data to payload
|
||||
int_size = link.tx_obj(speed, send_size) - send_size
|
||||
send_size += int_size
|
||||
|
||||
# Sends data to Arduino
|
||||
link.send(send_size)
|
||||
|
||||
# Waits for response from Arduino, and reports errors while receiving packets
|
||||
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))
|
||||
|
||||
# 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 KeyboardInterrupt:
|
||||
try:
|
||||
link.close()
|
||||
except:
|
||||
pass
|
||||
|
||||
except:
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
|
||||
try:
|
||||
link.close()
|
||||
except:
|
||||
pass
|
Loading…
Reference in a new issue