##### Libraries #####
from sense_hat import SenseHat
from datetime import datetime
from time import sleep
from threading import Thread
##### Logging Settings ######
FILENAME = ""
WRITE_FREQUENCY = 1
TEMP_H=False
TEMP_P=True
HUMIDITY=True
PRESSURE=True
ORIENTATION=True
ACCELERATION=True
MAG=True
GYRO=True
DELAY=60
##### Functions #####
def log_data():
output_string = ",".join(str(value) for value in sense_data)
batch_data.append(output_string)
def file_setup(filename):
header =[]
if TEMP_H:
header.append("temp_h")
if TEMP_P:
header.append("temp_p")
if HUMIDITY:
header.append("humidity")
if PRESSURE:
header.append("pressure")
if ORIENTATION:
header.extend(["pitch","roll","yaw"])
if MAG:
header.extend(["mag_x","mag_y","mag_z"])
if ACCELERATION:
header.extend(["accel_x","accel_y","accel_z"])
if GYRO:
header.extend(["gyro_x","gyro_y","gyro_z"])
header.append("timestamp")
with open(filename, "w") as f:
f.write(",".join(str(value) for value in header)+ "\n")
def get_sense_data():
sense_data=[]
if TEMP_H:
sense_data.append(sense.get_temperature_from_humidity())
if TEMP_P:
sense_data.append(sense.get_temperature_from_pressure())
if HUMIDITY:
sense_data.append(sense.get_humidity())
if PRESSURE:
sense_data.append(sense.get_pressure())
if ORIENTATION:
o = sense.get_orientation()
yaw = o["yaw"]
pitch = o["pitch"]
roll = o["roll"]
sense_data.extend([pitch,roll,yaw])
if MAG:
mag = sense.get_compass_raw()
mag_x = mag["x"]
mag_y = mag["y"]
mag_z = mag["z"]
sense_data.extend([mag_x,mag_y,mag_z])
if ACCELERATION:
acc = sense.get_accelerometer_raw()
x = acc["x"]
y = acc["y"]
z = acc["z"]
sense_data.extend([x,y,z])
if GYRO:
gyro = sense.get_gyroscope_raw()
gyro_x = ["x"]
gyro_y = ["y"]
gyro_z = ["z"]
sense_data.extend([gyro_x,gyro_y,gyro_z])
sense_data.append(datetime.now())
return sense_data
def timed_log():
while True:
log_data()
sleep(DELAY)
#### Main Program ####
sense = SenseHat()
batch_data= []
if FILENAME == "":
filename = "SenseLog-"+str(datetime.now())+".csv"
else:
filename = FILENAME+"-"+str(datetime.now())+".csv"
file_setup(filename)
if DELAY > 0:
sense_data = get_sense_data()
Thread(target= timed_log).start()
while True:
sense_data = get_sense_data()
if DELAY == 0:
log_data()
if len(batch_data) >= WRITE_FREQUENCY:
print("Writing to file..")
with open(filename, "a") as f:
for line in batch_data:
f.write(line + "\n")
batch_data = []
