Curve Fit - Add spikey data sets for testing.

Author: david-cattermole

lib/rust/mmscenegraph/tests/data/example_curves.py

@@ -27,6 +27,24 @@ def _write_data(name, data, output_directory):
     return
 
 
+def _generate_variance_data(raw_data, variance):
+    data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+    assert len(raw_data) == len(data)
+    return data
+
+
+def _generate_spike_data(raw_data, variance, spike_probability):
+    data = []
+    temp_data = [(f, random.uniform(v - variance, v + variance)) for f, v in raw_data]
+    for (raw_f, raw_v), (spike_f, spike_v) in zip(raw_data, temp_data):
+        if random.random() < spike_probability:
+            data.append((spike_f, spike_v))
+        else:
+            data.append((raw_f, raw_v))
+    assert len(raw_data) == len(data)
+    return data
+
+
 output_directory = (
     '${USERPROFILE}/dev/mayaMatchMoveSolver/lib/rust/mmscenegraph/tests/data'
 )
@@ -39,46 +57,66 @@ def _write_data(name, data, output_directory):
 variance_high = 1.0
 variance_very_high = 10.0
 
+spike_probability_low = 0.01
+spike_probability_medium = 0.1
+spike_probability_high = 0.25
 
 # Identity (all zero values)
 raw_data = _sample_attr(start_frame, end_frame, 'identity.translateY')
 _write_data('identity_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('identity_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('identity_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('identity_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('identity_variance4', data, output_directory)
 
+variance = variance_low
+spike_probability = spike_probability_medium
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('identity_spike1', data, output_directory)
+
+variance = variance_medium
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('identity_spike2', data, output_directory)
+
+variance = variance_high
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('identity_spike3', data, output_directory)
+
+variance = variance_very_high
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('identity_spike4', data, output_directory)
+
 
 # Degrees 45 - Up
 raw_data = _sample_attr(start_frame, end_frame, 'degree_45_up.translateY')
 _write_data('degree_45_up_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_up_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_up_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_up_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_up_variance4', data, output_directory)
 
 
@@ -87,19 +125,19 @@ def _write_data(name, data, output_directory):
 _write_data('degree_45_down_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_down_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_down_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_down_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('degree_45_down_variance4', data, output_directory)
 
 
@@ -108,19 +146,19 @@ def _write_data(name, data, output_directory):
 _write_data('linear_2_point_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_2_point_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_2_point_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_2_point_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_2_point_variance4', data, output_directory)
 
 
@@ -129,40 +167,57 @@ def _write_data(name, data, output_directory):
 _write_data('linear_3_point_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_3_point_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_3_point_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_3_point_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_3_point_variance4', data, output_directory)
 
+variance = variance_low
+spike_probability = spike_probability_medium
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('linear_3_point_spike1', data, output_directory)
+
+variance = variance_medium
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('linear_3_point_spike2', data, output_directory)
+
+variance = variance_high
+data = _generate_spike_data(raw_data, variance, spike_probability)
+_write_data('linear_3_point_spike3', data, output_directory)
+
+variance = variance_very_high
+_generate_spike_data(raw_data, variance, spike_probability)
+data = _write_data('linear_3_point_spike4', data, output_directory)
+
 
 # Linear (4-point)
 raw_data = _sample_attr(start_frame, end_frame, 'linear_4_point.translateY')
 _write_data('linear_4_point_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_4_point_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_4_point_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_4_point_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('linear_4_point_variance4', data, output_directory)
 
 
@@ -171,19 +226,19 @@ def _write_data(name, data, output_directory):
 _write_data('up_down_up_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('up_down_up_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('up_down_up_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('up_down_up_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('up_down_up_variance4', data, output_directory)
 
 
@@ -192,19 +247,19 @@ def _write_data(name, data, output_directory):
 _write_data('down_up_down_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('down_up_down_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('down_up_down_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('down_up_down_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('down_up_down_variance4', data, output_directory)
 
 
@@ -213,17 +268,17 @@ def _write_data(name, data, output_directory):
 _write_data('bounce_5_up_down_raw', raw_data, output_directory)
 
 variance = variance_low
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('bounce_5_up_down_variance1', data, output_directory)
 
 variance = variance_medium
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('bounce_5_up_down_variance2', data, output_directory)
 
 variance = variance_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('bounce_5_up_down_variance3', data, output_directory)
 
 variance = variance_very_high
-data = [(f, (random.uniform(v - variance, v + variance))) for f, v in raw_data]
+data = _generate_variance_data(raw_data, variance)
 _write_data('bounce_5_up_down_variance4', data, output_directory)

lib/rust/mmscenegraph/tests/data/identity_spike1.chan

@@ -0,0 +1,101 @@
+1001 0.0
+1002 0.0
+1003 0.0
+1004 0.0
+1005 0.0
+1006 0.0
+1007 0.0
+1008 0.0
+1009 0.0
+1010 0.0
+1011 0.0
+1012 0.0
+1013 0.0
+1014 0.0
+1015 0.0
+1016 0.0
+1017 -0.00350741872282
+1018 0.0
+1019 -0.000707299254751
+1020 0.0
+1021 0.00786208728069
+1022 0.0
+1023 0.0
+1024 0.0
+1025 0.0
+1026 0.0
+1027 0.0
+1028 0.0
+1029 0.0
+1030 0.0
+1031 0.0
+1032 0.0
+1033 0.0
+1034 0.0
+1035 0.0
+1036 0.0
+1037 0.0
+1038 0.0
+1039 0.0
+1040 0.0
+1041 0.0
+1042 0.0
+1043 0.0
+1044 0.0
+1045 0.0
+1046 0.0
+1047 0.0
+1048 0.0
+1049 2.57698817654e-05
+1050 0.0
+1051 0.0
+1052 0.0
+1053 0.0
+1054 0.0
+1055 0.0
+1056 0.0
+1057 0.0
+1058 0.0
+1059 0.0
+1060 0.0
+1061 0.0
+1062 0.0
+1063 0.0
+1064 0.0
+1065 0.0
+1066 0.0023426780206
+1067 0.0
+1068 -0.0010250596413
+1069 0.0
+1070 0.0
+1071 0.0
+1072 0.00124821264131
+1073 0.0
+1074 0.0
+1075 0.0
+1076 0.0
+1077 0.0
+1078 0.0
+1079 0.0
+1080 0.0
+1081 0.0
+1082 0.0
+1083 0.0
+1084 0.0
+1085 0.0
+1086 0.0
+1087 0.0
+1088 0.0
+1089 0.0
+1090 0.0
+1091 0.0
+1092 0.0
+1093 0.0
+1094 0.0
+1095 0.0
+1096 -0.0071282343961
+1097 0.0
+1098 0.0
+1099 0.0
+1100 0.0
+1101 0.0