Include attribute to create devices with equivalent data frames

Author: aacuevas

OpenEphys.Onix1/ConfigurePersistentHeartbeat.cs

@@ -74,6 +74,7 @@ public override IObservable<ContextTask> Process(IObservable<ContextTask> source
         }
     }
 
+    [EquivalentDataSource(typeof(Heartbeat))]
     static class PersistentHeartbeat
     {
         public const int ID = 34;

OpenEphys.Onix1/ContextHelper.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Linq;
 using System.Reflection;
 using oni;
 
@@ -68,5 +69,14 @@ static void ThrowInvalidDeviceException(Type expectedType, uint address)
         {
             throw new InvalidOperationException($"Invalid device ID. The device found at address {address} is not a '{expectedType.Name}' device.");
         }
+
+        internal static bool CheckDeviceType(Type deviceType, Type targetType)
+        {
+            if (deviceType == targetType) return true;
+
+            var equivalentTypes = deviceType.GetCustomAttributes(typeof(EquivalentDataSource), false).Cast<EquivalentDataSource>();
+
+            return equivalentTypes.Any(t => t.BaseType == targetType);
+        }
     }
 }

OpenEphys.Onix1/DeviceInfo.cs

@@ -24,7 +24,7 @@ public DeviceInfo(ContextTask context, Type deviceType, uint deviceAddress)
 
         public void AssertType(Type expectedType)
         {
-            if (DeviceType != expectedType)
+            if (!ContextHelper.CheckDeviceType(DeviceType, expectedType))
             {
                 throw new InvalidOperationException(
                     $"Expected device of type {expectedType.Name}. Actual type is {DeviceType.Name}."

OpenEphys.Onix1/DeviceNameConverter.cs

@@ -112,7 +112,7 @@ from element in SelectContextElements(level)
                                      let factory = ExpressionBuilder.GetWorkflowElement(element) as DeviceFactory
                                      where factory != null
                                      from device in factory.GetDevices()
-                                     where device.DeviceType == targetType
+                                     where ContextHelper.CheckDeviceType(device.DeviceType,targetType)
                                      select device.DeviceName)
                                      .Distinct()
                                      .ToList();