Skip to content

Commit

Permalink
Merge pull request #66 from emotional-cities/schema-cleanup
Browse files Browse the repository at this point in the history 
Update outdoor and VR schemas to latest wearable configuration
  • Loading branch information
@glopesdev
glopesdev authored Oct 18, 2024
2 parents 47ed2ca + dbe4289 commit d983e72
Show file tree
Hide file tree
Showing 2 changed files with 80 additions and 133 deletions.
106 changes: 49 additions & 57 deletions pluma/schema/outdoor.py
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from dotmap import DotMap
from typing import Union

from pluma.stream.ecg import EcgStream
from pluma.stream.harp import HarpStream
from pluma.stream.accelerometer import AccelerometerStream
from pluma.stream.empatica import EmpaticaStream
Expand All @@ -27,89 +28,80 @@ def build_schema(root: Union[str, ComplexPath], parent_dataset=None, autoload: b
kwargs = dict(root=root, autoload=autoload, parent_dataset=parent_dataset)
streams = DotMap()

# Pupil streams
streams.PupilLabs.Counter.DecodedFrames = HarpStream(209, device="PupilLabs", streamlabel="Counter_DecodedFrames", **kwargs)
streams.PupilLabs.Counter.RawFrames = HarpStream(210, device="PupilLabs", streamlabel="Counter_RawFrames", **kwargs)
streams.PupilLabs.Data.RawFrames = PupilWorldCameraStream(device="PupilLabs", streamlabel="WorldCamera.Data", **kwargs)
streams.PupilLabs.Counter.IMU = HarpStream(211, device="PupilLabs", streamlabel="Counter_IMU", **kwargs)
streams.PupilLabs.Counter.Gaze = HarpStream(212, device="PupilLabs", streamlabel="Counter_Gaze", **kwargs)
streams.PupilLabs.Data.Gaze = PupilGazeStream(device="PupilLabs", streamlabel="Gaze.Data", **kwargs)
streams.PupilLabs.Counter.Audio = HarpStream(213, device="PupilLabs", streamlabel="Counter_Audio", **kwargs)
streams.PupilLabs.Counter.Key = HarpStream(214, device="PupilLabs", streamlabel="Counter_Key", **kwargs)

# BioData streams
streams.BioData.EnableStreams = HarpStream(32, device="BioData", streamlabel="EnableStreams", **kwargs)
streams.BioData.DisableStreams = HarpStream(33, device="BioData", streamlabel="DisableStreams", **kwargs)
streams.BioData.ECG = HarpStream(35, device="BioData", streamlabel="ECG", **kwargs)
streams.BioData.GSR = HarpStream(36, device="BioData", streamlabel="GSR", **kwargs)
streams.BioData.Accelerometer = HarpStream(37, device="BioData", streamlabel="Accelerometer", **kwargs)
streams.BioData.DigitalIn = HarpStream(38, device="BioData", streamlabel="DigitalIn", **kwargs)
streams.BioData.Set = HarpStream(39, device="BioData", streamlabel="Set", **kwargs)
streams.BioData.Clear = HarpStream(40, device="BioData", streamlabel="Clear", **kwargs)

# TinkerForge streams
streams.TK.AmbientLight.AmbientLight = HarpStream(223, device="TK", streamlabel="AmbientLight.AmbientLight", **kwargs)
streams.BioData.EnableStreams = HarpStream(32, device="BioData", streamlabel="EnableStreams", **kwargs)
streams.BioData.DisableStreams = HarpStream(33, device="BioData", streamlabel="DisableStreams", **kwargs)
streams.BioData.ECG = EcgStream (35, device="BioData", streamlabel="ECG", **kwargs)
streams.BioData.GSR = HarpStream(36, device="BioData", streamlabel="GSR", **kwargs)
streams.BioData.Accelerometer = HarpStream(37, device="BioData", streamlabel="Accelerometer", **kwargs)
streams.BioData.DigitalIn = HarpStream(38, device="BioData", streamlabel="DigitalIn", **kwargs)
streams.BioData.Set = HarpStream(39, device="BioData", streamlabel="Set", **kwargs)
streams.BioData.Clear = HarpStream(40, device="BioData", streamlabel="Clear", **kwargs)

streams.TK.CO2V2.CO2Conc = HarpStream(224, device="TK", streamlabel="CO2V2.CO2Conc", **kwargs)
streams.TK.CO2V2.Temperature = HarpStream(225, device="TK", streamlabel="CO2V2.Temperature", **kwargs)
streams.TK.CO2V2.Humidity = HarpStream(226, device="TK", streamlabel="CO2V2.Humidity", **kwargs)
# Pupil streams
streams.PupilLabs.DecodedFrames = HarpStream(209, device="PupilLabs", streamlabel="Pupil_RawFrames", **kwargs)
streams.PupilLabs.RawFrames = PupilWorldCameraStream(210, device="PupilLabs", streamlabel="Pupil_RawFrames", **kwargs)
streams.PupilLabs.Counter.IMU = HarpStream(211, device="PupilLabs", streamlabel="Counter_IMU", **kwargs)
streams.PupilLabs.PupilGaze = PupilGazeStream(212, device="PupilLabs", streamlabel="Pupil_Gaze", **kwargs)
streams.PupilLabs.Counter.Audio = HarpStream(213, device="PupilLabs", streamlabel="Counter_Audio", **kwargs)
streams.PupilLabs.Counter.Key = HarpStream(214, device="PupilLabs", streamlabel="Counter_Key", **kwargs)

streams.TK.GPS.Latitude = HarpStream(227, device="TK", streamlabel="GPS.Latitude", **kwargs)
streams.TK.GPS.Longitude = HarpStream(228, device="TK", streamlabel="GPS.Longitude", **kwargs)
streams.TK.GPS.Altitude = HarpStream(229, device="TK", streamlabel="GPS.Altitude", **kwargs)
streams.TK.GPS.Data = HarpStream(230, device="TK", streamlabel="GPS.Data", **kwargs)
streams.TK.GPS.Time = HarpStream(231, device="TK", streamlabel="GPS.Time", **kwargs)
streams.TK.GPS.HasFix = HarpStream(232, device="TK", streamlabel="GPS.HasFix", **kwargs)
# TinkerForge streams
streams.TK.GPS.Latitude = HarpStream(227, device="TK", streamlabel="GPS_Latitude", **kwargs)
streams.TK.GPS.Longitude = HarpStream(228, device="TK", streamlabel="GPS_Longitude", **kwargs)
streams.TK.GPS.Altitude = HarpStream(229, device="TK", streamlabel="GPS_Altitude", **kwargs)
streams.TK.GPS.Data = HarpStream(230, device="TK", streamlabel="GPS_Data", **kwargs)
streams.TK.GPS.Time = HarpStream(231, device="TK", streamlabel="GPS_Time", **kwargs)
streams.TK.GPS.HasFix = HarpStream(232, device="TK", streamlabel="GPS_HasFix", **kwargs)

streams.TK.AirQuality.IAQIndex = HarpStream(233, device="TK", streamlabel="AirQuality.IAQIndex", **kwargs)
streams.TK.AirQuality.Temperature = HarpStream(234, device="TK", streamlabel="AirQuality.Temperature", **kwargs)
streams.TK.AirQuality.Humidity = HarpStream(235, device="TK", streamlabel="AirQuality.Humidity", **kwargs)
streams.TK.AirQuality.AirPressure = HarpStream(236, device="TK", streamlabel="AirQuality.AirPressure", **kwargs)
streams.TK.AirQuality.IAQIndex = HarpStream(233, device="TK", streamlabel="AirQuality_IAQIndex", **kwargs)
streams.TK.AirQuality.Temperature = HarpStream(234, device="TK", streamlabel="AirQuality_Temperature", **kwargs)
streams.TK.AirQuality.Humidity = HarpStream(235, device="TK", streamlabel="AirQuality_Humidity", **kwargs)
streams.TK.AirQuality.AirPressure = HarpStream(236, device="TK", streamlabel="AirQuality_AirPressure", **kwargs)

streams.TK.SoundPressureLevel.SPL = HarpStream(237, device="TK", streamlabel="SoundPressureLevel.SPL", **kwargs)
streams.TK.SoundPressureLevel.SPL = HarpStream(237, device="TK", streamlabel="SoundPressureLevel_SPL", **kwargs)

streams.TK.Humidity.Humidity = HarpStream(238, device="TK", streamlabel="Humidity.Humidity", **kwargs)
streams.TK.Humidity.Humidity = HarpStream(238, device="TK", streamlabel="Humidity_Humidity", **kwargs)

streams.TK.AnalogIn.Voltage = HarpStream(239, device="TK", streamlabel="AnalogIn.Voltage", **kwargs)
streams.TK.AnalogIn.Voltage = HarpStream(239, device="TK", streamlabel="AnalogIn_Voltage", **kwargs)

streams.TK.ParticulateMatter.PM1_0 = HarpStream(240, device="TK", streamlabel="ParticulateMatter.PM1_0", **kwargs)
streams.TK.ParticulateMatter.PM2_5 = HarpStream(241, device="TK", streamlabel="ParticulateMatter.PM2_5", **kwargs)
streams.TK.ParticulateMatter.PM10_0 = HarpStream(242, device="TK", streamlabel="ParticulateMatter.PM10_0", **kwargs)
streams.TK.ParticulateMatter.PM1_0 = HarpStream(240, device="TK", streamlabel="ParticulateMatter_PM1_0", **kwargs)
streams.TK.ParticulateMatter.PM2_5 = HarpStream(241, device="TK", streamlabel="ParticulateMatter_PM2_5", **kwargs)
streams.TK.ParticulateMatter.PM10_0 = HarpStream(242, device="TK", streamlabel="ParticulateMatter_PM10_0", **kwargs)

streams.TK.Dual0_20mA.SolarLight = HarpStream(243, device="TK", streamlabel="Dual0_20mA.SolarLight", **kwargs)
streams.TK.Dual0_20mA.SolarLight = HarpStream(243, device="TK", streamlabel="Dual0_20mA_SolarLight", **kwargs)

streams.TK.Thermocouple.Temperature = HarpStream(244, device="TK", streamlabel="Thermocouple.Temperature", **kwargs)
streams.TK.Thermocouple.Temperature = HarpStream(244, device="TK", streamlabel="Thermocouple_Temperature", **kwargs)

streams.TK.PTC.AirTemp = HarpStream(245, device="TK", streamlabel="PTC.AirTemp", **kwargs)
streams.TK.PTC.AirTemp = HarpStream(245, device="TK", streamlabel="PTC_AirTemp", **kwargs)

# ATMOS streams
streams.Atmos.NorthWind = HarpStream(246, device="Atmos", streamlabel="NorthWind", **kwargs)
streams.Atmos.EastWind = HarpStream(247, device="Atmos", streamlabel="EastWind", **kwargs)
streams.Atmos.GustWind = HarpStream(248, device="Atmos", streamlabel="GustWind", **kwargs)
streams.Atmos.AirTemperature = HarpStream(249, device="Atmos", streamlabel="AirTemperature", **kwargs)
streams.Atmos.XOrientation = HarpStream(250, device="Atmos", streamlabel="XOrientation", **kwargs)
streams.Atmos.YOrientation = HarpStream(251, device="Atmos", streamlabel="YOrientation", **kwargs)
streams.Atmos.NullValue = HarpStream(252, device="Atmos", streamlabel="NullValue", **kwargs)
streams.Atmos.NorthWind = HarpStream(246, device="Atmos", streamlabel="NorthWind", **kwargs)
streams.Atmos.EastWind = HarpStream(247, device="Atmos", streamlabel="EastWind", **kwargs)
streams.Atmos.GustWind = HarpStream(248, device="Atmos", streamlabel="GustWind", **kwargs)
streams.Atmos.AirTemperature = HarpStream(249, device="Atmos", streamlabel="AirTemperature", **kwargs)
streams.Atmos.XOrientation = HarpStream(250, device="Atmos", streamlabel="XOrientation", **kwargs)
streams.Atmos.YOrientation = HarpStream(251, device="Atmos", streamlabel="YOrientation", **kwargs)
streams.Atmos.NullValue = HarpStream(252, device="Atmos", streamlabel="NullValue", **kwargs)

# Accelerometer streams
streams.Accelerometer = AccelerometerStream(device="Accelerometer", streamlabel="Accelerometer", **kwargs)
streams.Accelerometer = AccelerometerStream(device="Accelerometer", streamlabel="Accelerometer", **kwargs)

# Empatica streams
streams.Empatica = EmpaticaStream(device="Empatica", streamlabel="Empatica", **kwargs)
streams.Empatica = EmpaticaStream(device="Empatica", streamlabel="Empatica", **kwargs)

# Microphone streams
streams.Microphone.Audio = MicrophoneStream(device="Microphone", streamlabel="Audio", **kwargs)
streams.Microphone.BufferIndex = HarpStream(222, device="Microphone", streamlabel="BufferIndex", **kwargs)
streams.Microphone.Audio = MicrophoneStream(device="Microphone", streamlabel="Audio", **kwargs)
streams.Microphone.BufferIndex = HarpStream(222, device="Microphone", streamlabel="BufferIndex", **kwargs)

# UBX streams
streams.UBX = UbxStream(device="UBX", streamlabel="UBX", **kwargs,
streams.UBX = UbxStream(device="UBX", streamlabel="UBX", **kwargs,
autoload_messages=[
_UBX_MSGIDS.NAV_HPPOSLLH,
_UBX_MSGIDS.TIM_TM2,
_UBX_MSGIDS.TIM_TP,
])

# EEG stream
streams.EEG = EegStream(device="Enobio", streamlabel="EEG", **kwargs)

streams.EEG = EegStream(device="Enobio", streamlabel="EEG", **kwargs)
return streams
Loading

0 comments on commit d983e72

Please sign in to comment.