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

save calibration by default

Browse Source
This commit is contained in:
Brendan Haines 2020-03-02 22:18:32 -07:00
parent 5ae0707162
commit e4cc932d1a
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Software/python/hwtest.py
View File

@ -38,8 +38,14 @@ servos[1].throttle = 0
servos[2].throttle = 0 servos[2].throttle = 0
# Initialize calibration # Initialize calibration
# TODO: save cal and load from file by default try:
white_cal = np.loadtxt('cal_white.txt')
except FileNotFoundError:
white_cal = [0]*8 white_cal = [0]*8
try:
black_cal = np.loadtxt('cal_black.txt')
except FileNotFoundError:
black_cal = [5]*8 black_cal = [5]*8
def get_reflectivity(chan): def get_reflectivity(chan):
@ -157,11 +163,13 @@ def update_plots(attrname=None, old=None, new=None):
def cal_white(attrname=None, old=None, new=None): def cal_white(attrname=None, old=None, new=None):
global white_cal global white_cal
white_cal = [get_reflectivity(c) for c in range(8)] white_cal = [get_reflectivity(c) for c in range(8)]
np.savetxt('cal_white.txt', white_cal)
update_plots() update_plots()
def cal_black(attrname=None, old=None, new=None): def cal_black(attrname=None, old=None, new=None):
global black_cal global black_cal
black_cal = [get_reflectivity(c) for c in range(8)] black_cal = [get_reflectivity(c) for c in range(8)]
np.savetxt('cal_black.txt', black_cal)
update_plots() update_plots()
def start_controller(attrname=None, old=None, new=None): def start_controller(attrname=None, old=None, new=None):