Last active
January 29, 2024 20:47
-
-
Save brandoncurtis/33a67d9d402973face8d to your computer and use it in GitHub Desktop.
Realtime Data Acquisition and Plotting with Arduino and Python
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/* | |
AnalogReadSerial | |
Reads an analog input on pin 0, prints the result to the serial monitor. | |
Attach the center pin of a potentiometer to pin A0, and the outside pins to +5V and ground. | |
This example code is in the public domain. | |
Upload this to the Arduino using the Arduino IDE! | |
*/ | |
// the setup routine runs once when you press reset: | |
void setup() { | |
// initialize serial communication at 9600 bits per second: | |
Serial.begin(115200); | |
} | |
// the loop routine runs over and over again forever: | |
void loop() { | |
// read the input on analog pin 0: | |
int sensorValue = analogRead(A0); | |
// print out the value you read: | |
Serial.println(sensorValue); | |
delay(50); // delay in between reads for stability | |
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
%matplotlib notebook | |
# run this in a Jupyter (IPython) Notebook! | |
# modified from http://www.lebsanft.org/?p=48 | |
# http://pyserial.readthedocs.org/en/latest/pyserial_api.html | |
import serial | |
import numpy as np | |
from matplotlib import pyplot as plt | |
from time import time | |
# If you're not using Linux, you'll need to change this | |
# check the Arduino IDE to see what serial port it's attached to | |
ser = serial.Serial('/dev/ttyACM0', 115200) | |
# set plot to animated | |
plt.ion() | |
start_time = time() | |
timepoints = [] | |
ydata = [] | |
yrange = [-0.1,5.1] | |
view_time = 4 # seconds of data to view at once | |
duration = 24 # total seconds to collect data | |
fig1 = plt.figure() | |
# http://matplotlib.org/users/text_props.html | |
fig1.suptitle('live updated data', fontsize='18', fontweight='bold') | |
plt.xlabel('time, seconds', fontsize='14', fontstyle='italic') | |
plt.ylabel('potential, volts', fontsize='14', fontstyle='italic') | |
plt.axes().grid(True) | |
line1, = plt.plot(ydata,marker='o',markersize=4,linestyle='none',markerfacecolor='red') | |
plt.ylim(yrange) | |
plt.xlim([0,view_time]) | |
# flush any junk left in the serial buffer | |
ser.flushInput() | |
# ser.reset_input_buffer() # for pyserial 3.0+ | |
run = True | |
# collect the data and plot a moving frame | |
while run: | |
ser.reset_input_buffer() | |
data = ser.readline().split(' ') | |
# sometimes the incoming data is garbage, so just 'try' to do this | |
try: | |
# store the entire dataset for later | |
ydata.append(float(data[0])*5.0/1024) | |
timepoints.append(time()-start_time) | |
current_time = timepoints[-1] | |
# update the plotted data | |
line1.set_xdata(timepoints) | |
line1.set_ydata(ydata) | |
# slide the viewing frame along | |
if current_time > view_time: | |
plt.xlim([current_time-view_time,current_time]) | |
# when time's up, kill the collect+plot loop | |
if timepoints[-1] > duration: run=False | |
# if the try statement throws an error, just do nothing | |
except: pass | |
# update the plot | |
fig1.canvas.draw() | |
# plot all of the data you collected | |
fig2 = plt.figure() | |
# http://matplotlib.org/users/text_props.html | |
fig2.suptitle('complete data trace', fontsize='18', fontweight='bold') | |
plt.xlabel('time, seconds', fontsize='14', fontstyle='italic') | |
plt.ylabel('potential, volts', fontsize='14', fontstyle='italic') | |
plt.axes().grid(True) | |
plt.plot(timepoints, ydata,marker='o',markersize=4,linestyle='none',markerfacecolor='red') | |
plt.ylim(yrange) | |
fig2.show() | |
ser.close() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
don't work, just empty window