Update PAWSStandard for adding VO2 max samples

StanfordBDHG · Feb 4, 2024 · 2e1180a · 2e1180a
1 parent f3e8f16
commit 2e1180a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/PAWS/ECGRecordings/ECGModule.swift b/PAWS/ECGRecordings/ECGModule.swift
@@ -13,6 +13,7 @@ import Spezi
 @Observable
 class ECGModule: Module, DefaultInitializable, EnvironmentAccessible {
     var hkElectrocardiograms: [HKElectrocardiogram] = []
+    var vo2Samples: [HKQuantitySample] = []
 
 
     /// Creates an instance of a ``MockWebService``.

diff --git a/PAWS/ECGRecordings/ECGRecording.swift b/PAWS/ECGRecordings/ECGRecording.swift
@@ -12,6 +12,7 @@ import SwiftUI
 
 struct ECGRecording: View {
     let hkElectrocardiogram: HKElectrocardiogram
+    // let vo2Samples: [HKQuantitySample]
     @State var symptoms: HKElectrocardiogram.Symptoms = [:]
 
 

diff --git a/PAWS/PAWSStandard.swift b/PAWS/PAWSStandard.swift
@@ -70,6 +70,9 @@ actor PAWSStandard: Standard, EnvironmentAccessible, HealthKitConstraint, Onboar
     func add(sample: HKSample) async {
         if let hkElectrocardiogram = sample as? HKElectrocardiogram {
             ecgStorage.hkElectrocardiograms.append(hkElectrocardiogram)
+        } else if let vo2Sample = sample as? HKQuantitySample,
+                  vo2Sample.quantityType.identifier == HKQuantityTypeIdentifier.vo2Max.rawValue {
+            ecgStorage.vo2Samples.append(vo2Sample)
         }
 
         if let mockWebService {