@@ -74,6 +74,18 @@ class FlowViewController:
7474 var isRecording = false
7575 var timer = Timer ( )
7676
77+ var deviceUUID : String = " "
78+ var sessionDateTime : String = " "
79+ var getParams = [ " String " : " String " ]
80+
81+ func getDeviceUUID( ) {
82+ if UserDefaults . standard. object ( forKey: " phoneUUID " ) == nil {
83+ UserDefaults . standard. set ( UUID ( ) . uuidString, forKey: " phoneUUID " )
84+ }
85+ deviceUUID = String ( UserDefaults . standard. object ( forKey: " phoneUUID " ) as! String )
86+ getParams [ " deviceUUID " ] = deviceUUID
87+ }
88+
7789 override func viewDidLoad( ) {
7890 super. viewDidLoad ( )
7991
@@ -113,6 +125,8 @@ class FlowViewController:
113125 EDALabel . text = " "
114126 flexLabel. text = " "
115127 }
128+
129+ getDeviceUUID ( )
116130
117131 // Do any additional setup after loading the view.
118132 }
@@ -214,11 +228,15 @@ class FlowViewController:
214228 dreamButton. setTitle ( " Cancel " , for: . normal)
215229 dreamButton. setTitleColor ( UIColor . red, for: . normal)
216230 dreamLabel. text = " Enjoy your dreams :) "
231+
217232 currentStatus = " CALIBRATING "
218233
219234 self . detectSleepTimer. invalidate ( )
220235
221- SleepAPI . apiGet ( endpoint: " init " )
236+ SleepAPI . apiGet ( endpoint: " init " , params: getParams, onSuccess: { json in
237+ self . sessionDateTime = json [ " datetime " ] as! String
238+ self . getParams [ " datetime " ] = self . sessionDateTime
239+ } )
222240 self . calibrateStart ( )
223241 self . numOnsets = 0
224242
@@ -231,7 +249,7 @@ class FlowViewController:
231249 self . currentStatus = " RUNNING "
232250 self . calibrateEnd ( )
233251
234- SleepAPI . apiGet ( endpoint: " train " )
252+ SleepAPI . apiGet ( endpoint: " train " , params : self . getParams )
235253
236254 self . detectSleepTimerPause = false
237255 self . detectSleepTimer = Timer . scheduledTimer ( timeInterval: 3 , target: self , selector: #selector( self . detectSleep ( sender: ) ) , userInfo: nil , repeats: true )
@@ -252,33 +270,46 @@ class FlowViewController:
252270 }
253271
254272 @objc func detectSleep( sender: Timer ) {
255- SleepAPI . apiGet ( endpoint: " predict " , onSuccess: { json in
273+ SleepAPI . apiGet ( endpoint: " predict " , params: getParams, onSuccess: { json in
274+ var onsetTrigger : OnsetTrigger ?
275+
256276 let score = Int ( ( json [ " max_sleep " ] as! NSNumber ) . floatValue. rounded ( ) )
257277 if ( !self . detectSleepTimerPause && self . numOnsets == 0 ) {
258278 if ( self . dreamDetectorControl. selectedSegmentIndex == 0 && score >= ( UserDefaults . standard. object ( forKey: " deltaHBOSS " ) as! Int ) ) {
259279 DispatchQueue . main. async {
260- self . sleepDetected ( )
280+ self . sleepDetected ( trigger : OnsetTrigger . HBOSS )
261281 }
262282 } else if ( self . dreamDetectorControl. selectedSegmentIndex == 1 && abs ( self . lastHR - self . meanHR) >= ( UserDefaults . standard. object ( forKey: " deltaHR " ) as! Int ) ) {
263283 DispatchQueue . main. async {
264- self . sleepDetected ( )
284+ self . sleepDetected ( trigger : OnsetTrigger . HR )
265285 }
266286 } else if ( self . dreamDetectorControl. selectedSegmentIndex == 2 && abs ( self . lastEDA - self . meanEDA) >= ( UserDefaults . standard. object ( forKey: " deltaEDA " ) as! Int ) ) {
267287 DispatchQueue . main. async {
268- self . sleepDetected ( )
288+ self . sleepDetected ( trigger : OnsetTrigger . EDA )
269289 }
270290 } else if ( self . dreamDetectorControl. selectedSegmentIndex == 3 && abs ( self . lastFlex - self . meanFlex) >= ( UserDefaults . standard. object ( forKey: " deltaFlex " ) as! Int ) ) {
271291 DispatchQueue . main. async {
272- self . sleepDetected ( )
292+ self . sleepDetected ( trigger : OnsetTrigger . FLEX )
273293 }
274294 }
275295 }
276296 } )
277297 }
278298
279- func sleepDetected( ) {
299+ func sleepDetected( trigger : OnsetTrigger ) {
280300 self . timer. invalidate ( )
281301 print ( " Sleep! " )
302+
303+ print ( " Sleep! " )
304+ print ( " TRIGGER WAS " , String ( describing: trigger) )
305+
306+ let json : [ String : Any ] = [ " trigger " : String ( describing: trigger) ,
307+ " currDateTime " : Date ( ) . timeIntervalSince1970,
308+ " legitimate " : true ,
309+ " deviceUUID " : deviceUUID,
310+ " datetime " : sessionDateTime]
311+ SleepAPI . apiPost ( endpoint: " reportTrigger " , json: json)
312+
282313 if ( !self . playedAudio) {
283314 self . playedAudio = true
284315 self . detectSleepTimerPause = true
@@ -305,7 +336,7 @@ class FlowViewController:
305336
306337 self . timer = Timer . scheduledTimer ( withTimeInterval: Double ( UserDefaults . standard. object ( forKey: " waitForOnsetTime " ) as! Int ) , repeats: false , block: {
307338 t in
308- self . sleepDetected ( )
339+ self . sleepDetected ( trigger : OnsetTrigger . TIMER )
309340 } )
310341 } )
311342 }
@@ -356,7 +387,10 @@ class FlowViewController:
356387 // send buffer to server
357388 let json : [ String : Any ] = [ " flex " : flexBuffer,
358389 " eda " : edaBuffer,
359- " ecg " : hrBuffer]
390+ " ecg " : hrBuffer,
391+ " deviceUUID " : getParams [ " deviceUUID " ] ,
392+ " datetime " : getParams [ " datetime " ]
393+ ]
360394 SleepAPI . apiPost ( endpoint: " upload " , json: json)
361395
362396 lastEDA = Int ( Float ( edaBuffer. reduce ( 0 , + ) ) / Float( edaBuffer. count) )
0 commit comments