brendanhaines/ECEN5458
1
1
Fork 0
mirror of https://github.com/brendanhaines/ECEN5458_Project.git synced 2024-12-27 03:17:06 -07:00
Code Issues Packages Projects Releases Wiki Activity

WIP: remove if main

Browse Source
This commit is contained in:
Brendan Haines 2020-03-02 15:59:59 -07:00
parent 62952cfd5e
commit 7e7ffffd64
1 changed files with 70 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

141
Software/python/hwtest.py
View File

@ -13,89 +13,88 @@ from bokeh.layouts import column, row
from bokeh.models import ColumnDataSource, Slider, TextInput, Button from bokeh.models import ColumnDataSource, Slider, TextInput, Button
from bokeh.plotting import figure from bokeh.plotting import figure
if __name__ == "__main__": mux_io = [None] * 4
mux_io = [None] * 4 mux_io[0] = digitalio.DigitalInOut(board.D17)
mux_io[0] = digitalio.DigitalInOut(board.D17) mux_io[1] = digitalio.DigitalInOut(board.D27)
mux_io[1] = digitalio.DigitalInOut(board.D27) mux_io[2] = digitalio.DigitalInOut(board.D22)
mux_io[2] = digitalio.DigitalInOut(board.D22) mux_io[3] = digitalio.DigitalInOut(board.D23)
mux_io[3] = digitalio.DigitalInOut(board.D23)
for ii, io in enumerate(mux_io):
io.switch_to_output()
i2c = busio.I2C(board.SCL, board.SDA)
adc = ADS.ADS1015(i2c)
adc_mux = AnalogIn(adc, ADS.P0)
white_cal = [0]*8
black_cal = [5]*8
def get_reflectivity(chan):
chan = int(chan)
global mux_io
global adc_mux
mux = 1-np.array(list(f"{chan:04b}"), dtype=int)
for ii, io in enumerate(mux_io): for ii, io in enumerate(mux_io):
io.switch_to_output() io.value = mux[ii]
return adc_mux.voltage
i2c = busio.I2C(board.SCL, board.SDA) def get_normalized_reflectivity(chan):
adc = ADS.ADS1015(i2c) global white_cal
adc_mux = AnalogIn(adc, ADS.P0) global black_cal
white_cal = [0]*8
black_cal = [5]*8 return (get_reflectivity(chan) - black_cal[chan]) / (white_cal[chan] - black_cal[chan])
def get_reflectivity(chan): brightness_idx = np.arange(8)
chan = int(chan) brightness = [get_normalized_reflectivity(c) for c in range(8)]
global mux_io
global adc_mux
mux = 1-np.array(list(f"{chan:04b}"), dtype=int)
for ii, io in enumerate(mux_io):
io.value = mux[ii]
return adc_mux.voltage
def get_normalized_reflectivity(chan): plt_source = ColumnDataSource(data=dict(x=brightness_idx, y=brightness))
global white_cal
global black_cal
return (get_reflectivity(chan) - black_cal[chan]) / (white_cal[chan] - black_cal[chan])
brightness_idx = np.arange(8) # Set up plot
plot = figure(plot_height=400, plot_width=400, title="my sine wave",
tools="save",
x_range=[0, 7], y_range=[0, 5])
plot.line('x', 'y', source=plt_source, line_width=3, line_alpha=0.6)
def update_data(*args, **kwargs):
brightness = [get_normalized_reflectivity(c) for c in range(8)] brightness = [get_normalized_reflectivity(c) for c in range(8)]
plt_source.data = dict(x=brightness_idx, y=brightness)
plt_source = ColumnDataSource(data=dict(x=brightness_idx, y=brightness)) def cal_white(*args, **kwargs):
global white_cal
white_cal = [get_reflectivity(c) for c in range(8)]
update_data()
# Set up plot def cal_black(*args, **kwargs):
plot = figure(plot_height=400, plot_width=400, title="my sine wave", global black_cal
tools="save", black_cal = [get_reflectivity(c) for c in range(8)]
x_range=[0, 7], y_range=[0, 5]) update_data()
plot.line('x', 'y', source=plt_source, line_width=3, line_alpha=0.6) cal_white_button = Button(label="Cal White")
cal_white_button.on_click(cal_white)
cal_black_button = Button(label="Cal Black")
cal_black_button.on_click(cal_black)
def update_data(*args, **kwargs): controls = column(cal_white_button, cal_black_button)
brightness = [get_normalized_reflectivity(c) for c in range(8)]
plt_source.data = dict(x=brightness_idx, y=brightness)
# def cal_white(*args, **kwargs): curdoc().add_root(row(controls, plot, width=800))
# global white_cal # curdoc().add_root(row(plot, width=800))
# white_cal = [get_reflectivity(c) for c in range(8)] curdoc().title = "test"
# update_data()
# def cal_black(*args, **kwargs): # while True:
# global black_cal # time.sleep(0.1)
# black_cal = [get_reflectivity(c) for c in range(8)] # update_data()
# update_data()
# cal_white_button = Button(label="Cal White")
# cal_white_button.on_click(cal_white)
# cal_black_button = Button(label="Cal Black")
# cal_black_button.on_click(cal_black)
# controls = column(cal_white_button, cal_black_button)
# curdoc().add_root(row(controls, plot, width=800))
curdoc().add_root(row(plot, width=800))
curdoc().title = "test"
# while True:
# time.sleep(0.1)
# update_data()
# servos = ServoKit(channels=16).continuous_servo # servos = ServoKit(channels=16).continuous_servo
# servos[0].throttle = 0 # servos[0].throttle = 0
# servos[1].throttle = 0 # servos[1].throttle = 0
# servos[2].throttle = 0 # servos[2].throttle = 0
# time.sleep(1) # time.sleep(1)
# servos[0].throttle = 20 # servos[0].throttle = 20
# servos[1].throttle = 20 # servos[1].throttle = 20
# servos[2].throttle = 20 # servos[2].throttle = 20
# time.sleep(1) # time.sleep(1)
# servos[0].throttle = 0 # servos[0].throttle = 0
# servos[1].throttle = 0 # servos[1].throttle = 0
# servos[2].throttle = 0 # servos[2].throttle = 0