Decision mask in arbter kernel output schema

The downstream analysis need to know the decision of kernels.

@hammerd

src/calng/FrameSelectionArbiter.py

@@ -402,6 +402,7 @@ class AdvancedFrameSelectionArbiter(BaseFrameSelectionArbiter):
                 except Exception as ex:
                     warn(f"Kernel for {prefix} failed: {ex}")
                     res[selection_name] = np.ones(num_frames, dtype=bool)
+                out_hash[f"{selection_name}.mask"] = list(map(bool, res[selection_name]))
         return res
 
     def _configure_decision(self, decision_string):
@@ -480,6 +481,17 @@ class AdvancedFrameSelectionArbiter(BaseFrameSelectionArbiter):
                 kernel_class.extend_device_schema(
                     schema_update, kernel_prefix
                 )
+                (
+	            NODE_ELEMENT(output_schema_update)
+	            .key(selection_name)
+                    .commit(),
+
+                    VECTOR_BOOL_ELEMENT(output_schema_update)
+                    .key(f"{selection_name}.mask")
+                    .assignmentOptional()
+                    .defaultValue([])
+                    .commit(),
+                )
                 kernel_class.extend_output_schema(output_schema_update, selection_name)
                 new_selection_steps[selection_name] = (kernel_class, preselection)
 

[Egor Sobolev]

This line actually writes... Or do I misunderstand you?

[David Hammer]

Oh you're right of course. Not sure how I missed this line.