Added support for boolean parameters in annotations actions (#8798)

cvat-ai · Dec 10, 2024 · 17016de · 17016de
1 parent 15fd273
commit 17016de
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/changelog.d/20241209_110126_sekachev.bs_support_boolean.md b/changelog.d/20241209_110126_sekachev.bs_support_boolean.md
@@ -0,0 +1,4 @@
+### Added
+
+- Support for boolean parameters in annotations actions
+  (<https://github.com/cvat-ai/cvat/pull/8798>)
diff --git a/cvat-core/src/annotations-actions/base-action.ts b/cvat-core/src/annotations-actions/base-action.ts
@@ -9,6 +9,7 @@ import { Job, Task } from '../session';
 export enum ActionParameterType {
     SELECT = 'select',
     NUMBER = 'number',
+    CHECKBOX = 'checkbox',
 }
 
 // For SELECT values should be a list of possible options

diff --git a/cvat-ui/src/components/annotation-page/annotations-actions/annotations-actions-modal.tsx b/cvat-ui/src/components/annotation-page/annotations-actions/annotations-actions-modal.tsx
@@ -25,10 +25,12 @@ import { getCVATStore } from 'cvat-store';
 import {
     BaseCollectionAction, BaseAction, Job, getCore,
     ObjectState,
+    ActionParameterType,
 } from 'cvat-core-wrapper';
 import { Canvas } from 'cvat-canvas-wrapper';
 import { fetchAnnotationsAsync } from 'actions/annotation-actions';
 import { clamp } from 'utils/math';
+import { Switch } from 'antd/lib';
 
 const core = getCore();
 
@@ -248,7 +250,7 @@ function ActionParameterComponent(props: ActionParameterProps & { onChange: (val
 
     const computedValues = typeof values === 'function' ? values({ instance: job }) : values;
 
-    if (type === 'select') {
+    if (type === ActionParameterType.SELECT) {
         return (
             <Select value={value} onChange={setValue}>
                 {computedValues.map((_value: string) => (
@@ -258,8 +260,18 @@ function ActionParameterComponent(props: ActionParameterProps & { onChange: (val
         );
     }
 
-    const [min, max, step] = computedValues.map((val) => +val);
+    if (type === ActionParameterType.CHECKBOX) {
+        return (
+            <Switch
+                checked={value.toLowerCase() === 'true'}
+                onChange={(val: boolean) => {
+                    setValue(String(val));
+                }}
+            />
+        );
+    }
 
+    const [min, max, step] = computedValues.map((val) => +val);
     return (
         <InputNumber
             value={+value}