fix: Merge pull request #244 from UniversalDataTool/disable-hotkeys

Allow disable hotkeys mode for image classification (as prop)
UniversalDataTool · Aug 8, 2020 · 60e0a57 · 60e0a57
2 parents 2257a60 + e00cdaf
commit 60e0a57
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/components/ImageClassification/index.js b/src/components/ImageClassification/index.js
@@ -83,6 +83,8 @@ export default ({
     iface.labels = iface.availableLabels
   }
 
+  const disableHotkeys = containerProps.disableHotkeys
+
   if (!iface.labels)
     throw new Error("No labels defined. Add some labels in Setup to continue.")
   const [sampleIndex, changeSampleIndex] = useState(0)
@@ -154,6 +156,7 @@ export default ({
   }, [sampleIndex, globalSampleIndex, samples])
 
   const [hotkeyMap, labelKeyMap] = useMemo(() => {
+    if (disableHotkeys) return [{}, {}]
     const hotkeyMap = {
       " ": onNext,
       backspace: onPrev,
@@ -172,9 +175,10 @@ export default ({
       labelKeyMap[label.id] = nextAvailableLetter
     }
     return [hotkeyMap, labelKeyMap]
-  }, [labels, onClickLabel, onDone, onNext, onPrev])
+  }, [labels, onClickLabel, onDone, onNext, onPrev, disableHotkeys])
 
   useEffect(() => {
+    if (disableHotkeys) return
     const onKeyDown = (e) => {
       const key = e.key.toLowerCase()
       if (hotkeyMap[key]) {
@@ -185,7 +189,7 @@ export default ({
     return () => {
       window.removeEventListener("keydown", onKeyDown)
     }
-  }, [hotkeyMap])
+  }, [hotkeyMap, disableHotkeys])
 
   return (
     <WorkspaceContainer

diff --git a/src/components/UniversalDataViewer/index.js b/src/components/UniversalDataViewer/index.js
@@ -20,6 +20,7 @@ export const UniversalDataViewer = ({
   onExit,
   hideHeader,
   hideDescription,
+  disableHotkeys = false,
   datasetName,
   requireCompleteToPressNext,
   onSaveTaskOutputItem,
@@ -36,6 +37,7 @@ export const UniversalDataViewer = ({
       requireCompleteToPressNext,
       onExit,
       height,
+      disableHotkeys,
     }),
     [
       hideHeader,
@@ -44,6 +46,7 @@ export const UniversalDataViewer = ({
       datasetName,
       height,
       onExit,
+      disableHotkeys,
     ]
   )