From 080819a1118217ee88dc5a99d28b133f668835d1 Mon Sep 17 00:00:00 2001
From: Thinh Nguyen <thinh@vathes.com>
Date: Thu, 15 Feb 2024 15:27:38 -0600
Subject: [PATCH] feat(metadata): explicitly extract and store frame rate for
 video source

---
 aeon/dj_pipeline/utils/load_metadata.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/aeon/dj_pipeline/utils/load_metadata.py b/aeon/dj_pipeline/utils/load_metadata.py
index b2334516..18f0f690 100644
--- a/aeon/dj_pipeline/utils/load_metadata.py
+++ b/aeon/dj_pipeline/utils/load_metadata.py
@@ -186,6 +186,9 @@ def ingest_epoch_metadata(experiment_name, devices_schema, metadata_yml_filepath
 
     device_type_mapper, _ = get_device_mapper(devices_schema, metadata_yml_filepath)
 
+    # Retrieve video controller
+    video_controller = epoch_config["metadata"].pop("VideoController", {})
+
     # Insert into each device table
     epoch_device_types = []
     device_list = []
@@ -221,6 +224,14 @@ def ingest_epoch_metadata(experiment_name, devices_schema, metadata_yml_filepath
                 }
                 for attribute_name, attribute_value in device_config.items()
             ]
+            if "TriggerFrequency" in device_config:
+                table_attribute_entry.append(
+                    {
+                        **table_entry,
+                        "attribute_name": "SamplingFrequency",
+                        "attribute_value": video_controller[device_config["TriggerFrequency"]],
+                    }
+                )
 
             """Check if this device is currently installed. If the same device serial number is currently installed check for any changes in configuration. If not, skip this"""
             current_device_query = table - table.RemovalTime & experiment_key & device_key