Fix a bug where arduino would crash on incorrectly sized i2c payload

.gitignore

@@ -5,3 +5,7 @@ __pycache__
 *.stl
 *.FCStd1
 
+# Artifacts
+*.png
+*.jpg
+*.wav

README.md

@@ -0,0 +1,7 @@
+# To install on robot
+
+```
+git clone <url>
+cd lewis-crawler/crawler_software/raspberry_pi
+make install?
+```

crawler_software/arduino/crawler_slave/crawler_slave.ino

@@ -0,0 +1,63 @@
+/* Crawler Slave
+ *  
+ *  This code runs on the crawler i2c network
+ *  and provides a cleaner, less CPU intensive control over PWM devices.
+ */
+
+#include <Wire.h>
+#include <Servo.h>
+
+// This servo is used to wipe and clean the camera lens
+Servo windowWiperServo;
+
+// Variables populated over i2c from master
+int id;
+int val;
+
+void setup() {
+  // For debugging
+  //Serial.begin(115200);
+  
+  // Attach the wiper servo to pin 9
+  windowWiperServo.attach(9);
+
+  // This is the address the pi will speak to us at
+  Wire.begin(0x4);
+  
+  // Call receiveEvent when data received                
+  Wire.onReceive(receiveEvent);
+
+  // Setup LED
+  pinMode(LED_BUILTIN, OUTPUT);
+  digitalWrite(LED_BUILTIN, LOW);
+
+  //Serial.println("Started");
+}
+
+// Just loop to keep the running code alive, and wait for events to happen.
+void loop() {
+  delay(50);
+}
+
+// This method runs when we receive a message
+void receiveEvent(int n) {
+  Wire.read(); // Remove smbus trash
+  if (true) { // Dont do anything if this is not true
+    id  = Wire.read(); // ID of the servo/device to access
+    val = Wire.read(); // Value to assign
+
+    //Serial.println(id);
+    //Serial.println(val);
+  
+    switch(id) {
+      case 1:
+        windowWiperServo.write(val);
+        break;
+    }
+  }
+
+  // Prevents a bug where if bytes are left in buffer, arduino crashes.
+  while (Wire.available()) {
+    Wire.read();
+  }
+}

crawler_software/raspberry_pi/config/hardware.yaml

@@ -0,0 +1,5 @@
+wiper_servo:
+  pin: 17
+  min_pulse: 0.000544
+  max_pulse: 0.0024
+

crawler_software/raspberry_pi/crawler.service

@@ -0,0 +1,11 @@
+[Unit]
+Description=Crawler service overseer, manages running main crawler software
+After=multi-user.target
+
+[Service]
+Type=simple
+Restart=always
+ExecStart=/usr/bin/python /srv/crawler/crawler.py
+
+[Install]
+WantedBy=multi-user.target

crawler_software/raspberry_pi/requirements.txt

@@ -0,0 +1,2 @@
+PySSTV
+picamera

crawler_software/raspberry_pi/sstv.py

@@ -1,9 +1,15 @@
 from shutil import copyfile
 from PIL import Image, ImageFont, ImageDraw
 from pysstv import color
-from picamera import PiCamera
+
 import logging as log
 
+try:
+    from picamera import PiCamera
+except ModuleNotFoundError:
+    log.info("Running in simulator mode")
+
+
 def take_photo():
     # This def is meant to take a photograph from the robot, 
     # it should include all steps and error checking to raise the mast
@@ -16,9 +22,12 @@ def take_photo():
     # Software to take the photo should be here
     #copyfile('photos/camera_latest.jpg', 'working/working.jpg')
     log.debug('Initalizing camera.')
+    try:
         camera = PiCamera()
         log.info('Saving photo.')
         camera.capture('working/working.jpg')
+    except NameError:
+        log.info("Running in simulator mode, not replacing test pattern")
 
 def mark_photo():
     log.info('Opening photo for viewing.')

crawler_software/raspberry_pi/tests/discordbot.py

@@ -0,0 +1,73 @@
+from discord_webhook import DiscordWebhook
+from picamera import PiCamera
+from time import sleep
+from gps import *
+from smbus import SMBus
+import time
+
+addr = 0x4 # bus address
+bus = SMBus(1) # indicates /dev/ic2-1
+
+numb = 1
+
+def writeData(value):
+    byteValue = StringToBytes(value)    
+    bus.write_i2c_block_data(addr,0x00,byteValue) #first byte is 0=command byte.. just is.
+    return -1
+
+
+def StringToBytes(val):
+        retVal = []
+        for c in val:
+                retVal.append(ord(c))
+        return retVal
+
+try:
+    for _x in range (0, 2):
+            for i in range(78, 130):
+                    writeData("WIPE-" + str(i))
+                    time.sleep(0.02)
+            for i in range(130, 78, -1):
+                    writeData("WIPE-" + str(i))
+                    time.sleep(0.02)
+except OSError:
+    print("Could not speak to ardujmemo")
+
+def get_uptime():
+    with open('/proc/uptime', 'r') as f:
+        uptime_seconds = float(f.readline().split()[0])
+
+    return uptime_seconds
+
+def getPositionData(gps):
+    location = [None]
+    while(location[0] == None):
+        print("Trying again")
+        nx = gpsd.next()
+        # For a list of all supported classes and fields refer to:
+        # https://gpsd.gitlab.io/gpsd/gpsd_json.html
+        if nx['class'] == 'TPV':
+            latitude = getattr(nx,'lat', "Unknown")
+            longitude = getattr(nx,'lon', "Unknown")
+            #print "Your position: lon = " + str(longitude) + ", lat = " + str(latitude)
+            location = [latitude, longitude]
+    return location
+
+gpsd = gps(mode=WATCH_ENABLE|WATCH_NEWSTYLE)
+
+loc = getPositionData(gpsd)
+
+webhookURL = "https://discord.com/api/webhooks/856609966404534272/TR9tnLq2sIGZoOeADNswmGRNlzBcqM5aKihfU6snVTP9WhSSoVVvi7nT6i-ZfZS7Hcqm"
+
+print(loc[0])
+print(loc[1])
+webhook = DiscordWebhook(url=webhookURL, content="Uptime: " + str( round( ((get_uptime() / 60) / 60 ), 2 )) + " hours. Lat is " + str(loc[0]) + ", long is " + str(loc[1]))
+
+camera = PiCamera()
+sleep(3) # let iso settle out
+camera.capture('still.jpg')
+
+with open("still.jpg", "rb") as f:
+	webhook.add_file(file=f.read(), filename='still.jpg')
+response = webhook.execute() # Hit send
+

crawler_software/raspberry_pi/tests/servo_test.py

@@ -0,0 +1,19 @@
+import RPi.GPIO as GPIO
+import time
+
+servoPIN = 17
+GPIO.setmode(GPIO.BCM)
+GPIO.setup(servoPIN, GPIO.OUT)
+
+p = GPIO.PWM(servoPIN, 50) # GPIO 17 for PWM with 50Hz
+p.start(2.5) # Initialization
+try:
+  while True:
+    p.ChangeDutyCycle(10)
+    time.sleep(2)
+    p.ChangeDutyCycle(2.5)
+    time.sleep(2)
+except KeyboardInterrupt:
+  p.stop()
+  GPIO.cleanup()
+

crawler_software/raspberry_pi/tests/stack_overflow.py

@@ -0,0 +1,20 @@
+import smbus
+import time
+import struct
+
+# for RPI version 1, use bus = smbus.SMBus(0)
+bus = smbus.SMBus(1)
+
+# This is the address we setup in the Arduino Program
+address = 0x04
+
+try:
+    for _x in range (0, 2):
+            for i in range(78, 130):
+                    bus.write_i2c_block_data(address, 0, [1, i])
+                    time.sleep(0.02)
+            for i in range(130, 78, -1):
+                    bus.write_i2c_block_data(address, 0, [1, i])
+                    time.sleep(0.02)
+except OSError:
+    print("Could not speak to ardujmemo")

crawler_software/raspberry_pi/tests/test_i2c.py

@@ -0,0 +1,35 @@
+#  Raspberry Pi Master for Arduino Slave
+#  i2c_master_pi.py
+#  Connects to Arduino via I2C
+  
+#  DroneBot Workshop 2019
+#  https://dronebotworkshop.com
+
+from smbus import SMBus
+import time
+
+addr = 0x8 # bus address
+bus = SMBus(1) # indicates /dev/ic2-1
+
+numb = 1
+
+print ("Enter num")
+
+for _x in range (0, 4):
+	for i in range(76, 130):
+		bus.write_byte(addr, i)
+		time.sleep(0.02)
+	for i in range(130, 76, -1):
+                bus.write_byte(addr, i)
+                time.sleep(0.02)
+
+#while numb == 1:
+#
+#	ledstate = input(">>>>   ")
+#
+#	if ledstate == "1":
+#		bus.write_byte(addr, 0x1) # switch it on
+#	elif ledstate == "0":
+#		bus.write_byte(addr, 0x0) # switch it on
+#	else:
+#		numb = 0

crawler_software/raspberry_pi/tests/test_tx.py

@@ -0,0 +1,12 @@
+import RPi.GPIO as GPIO # Import Raspberry Pi GPIO library
+from time import sleep # Import the sleep function from the time module
+
+GPIO.setwarnings(False) # Ignore warning for now
+GPIO.setmode(GPIO.BOARD) # Use physical pin numbering
+GPIO.setup(12, GPIO.OUT, initial=GPIO.LOW) # Set pin 8 to be an output pin and set initial value to low (off)
+
+while True: # Run forever
+ GPIO.output(12, GPIO.HIGH) # Turn on
+ sleep(1) # Sleep for 1 second
+ GPIO.output(12, GPIO.LOW) # Turn off
+ sleep(1) # Sleep for 1 second

requirements.txt

@@ -1,2 +0,0 @@
-PySSTV
-picamera