ENH: Add Enums to MATLAB

BrownBiomechanics · Jul 27, 2023 · 511ef47 · 511ef47
1 parent ebcd91d
commit 511ef47
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 7 deletions.
diff --git a/scripts/matlab/AutoscoperConnection.m b/scripts/matlab/AutoscoperConnection.m
@@ -343,14 +343,25 @@ function optimizeFrame(obj, volNum, frameNum, repeats, max_itr, min_lim, max_lim
                 dframe = 1;
             end
             if nargin < 10
-                opt_method = 0;
+                opt_method = OptimizationMethod.PARTICLE_SWARM_OPTIMIZATION;
             end
             if nargin < 11
-                cf_model = 0;
+                cf_model = CostFunction.NORMALIZED_CROSS_CORRELATION;
             end
             if nargin < 12
-                opt_init_heuristic = 0;
+                opt_init_heuristic = OptimizationInitializationHeuristic.PREVIOUS_FRAME;
             end
+
+            if ~isenum(opt_method)
+                opt_method = OptimizationMethod(opt_method);
+            end
+            if ~isenum(cf_model)
+                cf_model = CostFunction(cf_model);
+            end
+            if ~isenum(opt_init_heuristic)
+                opt_init_heuristic = OptimizationInitializationHeuristic(opt_init_heuristic);
+            end
+
             fwrite(obj.socket_descriptor, [ ...
                                            11 ...
                                            typecast(int32(volNum), 'uint8') ...
@@ -421,13 +432,23 @@ function trackingDialog(obj, volNum, startframe, endframe, repeats, max_itr, min
                 dframe = 1;
             end
             if nargin < 11
-                opt_method = 0;
+                opt_method = OptimizationMethod.PARTICLE_SWARM_OPTIMIZATION;
             end
             if nargin < 12
-                cf_model = 0;
+                cf_model = CostFunction.NORMALIZED_CROSS_CORRELATION;
             end
             if nargin < 13
-                opt_init_heuristic = 0;
+                opt_init_heuristic = OptimizationInitializationHeuristic.PREVIOUS_FRAME;
+            end
+
+            if ~isenum(opt_method)
+                opt_method = OptimizationMethod(opt_method);
+            end
+            if ~isenum(cf_model)
+                cf_model = CostFunction(cf_model);
+            end
+            if ~isenum(opt_init_heuristic)
+                opt_init_heuristic = OptimizationInitializationHeuristic(opt_init_heuristic);
             end
 
             for i = startframe:endframe
@@ -494,4 +515,4 @@ function trackingDialog(obj, volNum, startframe, endframe, repeats, max_itr, min
         end
 
     end
-end
+end
diff --git a/scripts/matlab/CostFunction.m b/scripts/matlab/CostFunction.m
@@ -0,0 +1,6 @@
+classdef CostFunction < int32
+    enumeration
+        NORMALIZED_CROSS_CORRELATION (0)
+        SUM_OF_ABSOLUTE_DIFFERENCES (1)
+    end
+end
diff --git a/scripts/matlab/OptimizationInitializationHeuristic.m b/scripts/matlab/OptimizationInitializationHeuristic.m
@@ -0,0 +1,8 @@
+classdef OptimizationInitializationHeuristic < int32
+    enumeration
+        CURRENT_FRAME (0)
+        PREVIOUS_FRAME (1)
+        LINEAR_EXTRAPOLATION (2)
+        SPLINE_INTERPOLATION (3)
+    end
+end
diff --git a/scripts/matlab/OptimizationMethod.m b/scripts/matlab/OptimizationMethod.m
@@ -0,0 +1,6 @@
+classdef OptimizationMethod < int32
+    enumeration
+        PARTICLE_SWARM_OPTIMIZATION (0)
+        DOWNHILL_SIMPLEX (1)
+    end
+end