From 521a1e28582f4e315ec8e5b806704cca9953bf98 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Sun, 9 Oct 2022 13:01:35 +0800
Subject: [PATCH] press Alt + mouse move, change brush size

---
 .../app/src/components/Editor/Editor.tsx      | 41 ++++++++++++++++++-
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/lama_cleaner/app/src/components/Editor/Editor.tsx b/lama_cleaner/app/src/components/Editor/Editor.tsx
index fd2f9f6..ce79e7c 100644
--- a/lama_cleaner/app/src/components/Editor/Editor.tsx
+++ b/lama_cleaner/app/src/components/Editor/Editor.tsx
@@ -48,6 +48,8 @@ import emitter, { EVENT_PROMPT } from '../../event'
 import FileSelect from '../FileSelect/FileSelect'
 
 const TOOLBAR_SIZE = 200
+const MIN_BRUSH_SIZE = 10
+const MAX_BRUSH_SIZE = 200
 const BRUSH_COLOR = '#ffcc00bb'
 
 interface Line {
@@ -108,6 +110,14 @@ export default function Editor() {
   const [showBrush, setShowBrush] = useState(false)
   const [showRefBrush, setShowRefBrush] = useState(false)
   const [isPanning, setIsPanning] = useState<boolean>(false)
+  const [isChangingBrushSizeByMouse, setIsChangingBrushSizeByMouse] =
+    useState<boolean>(false)
+  const [changeBrushSizeByMouseInit, setChangeBrushSizeByMouseInit] = useState({
+    x: -1,
+    y: -1,
+    brushSize: 20,
+  })
+
   const [showOriginal, setShowOriginal] = useState(false)
   const [scale, setScale] = useState<number>(1)
   const [panned, setPanned] = useState<boolean>(false)
@@ -486,6 +496,15 @@ export default function Editor() {
   }
 
   const onMouseDrag = (ev: SyntheticEvent) => {
+    if (isChangingBrushSizeByMouse) {
+      const initX = changeBrushSizeByMouseInit.x
+      // move right: increase brush size
+      const newSize = changeBrushSizeByMouseInit.brushSize + (x - initX)
+      if (newSize <= MAX_BRUSH_SIZE && newSize >= MIN_BRUSH_SIZE) {
+        setBrushSize(newSize)
+      }
+      return
+    }
     if (isPanning) {
       return
     }
@@ -552,6 +571,9 @@ export default function Editor() {
   }
 
   const onMouseDown = (ev: SyntheticEvent) => {
+    if (isChangingBrushSizeByMouse) {
+      return
+    }
     if (isPanning) {
       return
     }
@@ -892,6 +914,21 @@ export default function Editor() {
     }
   )
 
+  useKeyPressEvent(
+    'Alt',
+    ev => {
+      ev?.preventDefault()
+      ev?.stopPropagation()
+      setIsChangingBrushSizeByMouse(true)
+      setChangeBrushSizeByMouseInit({ x, y, brushSize })
+    },
+    ev => {
+      ev?.preventDefault()
+      ev?.stopPropagation()
+      setIsChangingBrushSizeByMouse(false)
+    }
+  )
+
   const getCurScale = (): number => {
     let s = minScale
     if (viewportRef.current?.state.scale !== undefined) {
@@ -1070,8 +1107,8 @@ export default function Editor() {
         )}
         <Slider
           label="Brush"
-          min={10}
-          max={150}
+          min={MIN_BRUSH_SIZE}
+          max={MAX_BRUSH_SIZE}
           value={brushSize}
           onChange={handleSliderChange}
           onClick={() => setShowRefBrush(false)}