From 972473f1af51d08437e597aa3f14b23acd3844ae Mon Sep 17 00:00:00 2001
From: caranicas <kendrick.caranicas@gmail.com>
Date: Thu, 15 Sep 2022 20:25:48 -0400
Subject: [PATCH] drawing in progress

---
 .../components/molecules/drawImage/index.tsx  | 91 +++++++++++++++++++
 .../organisms/displayPanel/index.tsx          |  2 +
 2 files changed, 93 insertions(+)
 create mode 100644 ui/frontend/build_src/src/components/molecules/drawImage/index.tsx

diff --git a/ui/frontend/build_src/src/components/molecules/drawImage/index.tsx b/ui/frontend/build_src/src/components/molecules/drawImage/index.tsx
new file mode 100644
index 00000000..6ba004d1
--- /dev/null
+++ b/ui/frontend/build_src/src/components/molecules/drawImage/index.tsx
@@ -0,0 +1,91 @@
+import React, {useRef, useEffect} from "react";
+
+// https://github.com/embiem/react-canvas-draw
+
+type DrawImageProps = {
+  imageData: string;
+};
+
+
+export default function DrawImage({imageData}: DrawImageProps) {
+
+  const canvasRef = useRef<HTMLCanvasElement>(null);
+
+  const draw = (ctx: CanvasRenderingContext2D) => {
+    ctx.fillStyle = "red";
+    ctx.fillRect(0, 0, 100, 100);
+  };
+
+  useEffect(() => {
+    const canvas = canvasRef.current;
+    if (canvas) {
+    
+      if(imageData){
+        const ctx = canvas.getContext("2d");
+        if (ctx) {
+          const img = new Image();
+          img.onload = () => {
+            ctx.drawImage(img, 0, 0);
+          };
+          img.src = imageData;
+        }
+      }
+      else {
+        const ctx = canvas.getContext("2d");
+        if (ctx) {
+          draw(ctx);
+        }
+      }
+    }
+    else {
+      console.log("canvas is null");
+    }
+  }, [imageData, draw]);
+
+
+  const _handleMouseDown = (e: React.MouseEvent<HTMLCanvasElement, MouseEvent>) => {
+    console.log("mouse down", e);
+    const canvas = canvasRef.current;
+    if (canvas) {
+      const ctx = canvas.getContext("2d");
+      ctx.strokeStyle = '#ff0000';
+      const {nativeEvent: {x, y}} = e;
+
+      console.log("x: " + x + " y: " + y);
+
+      ctx.moveTo(x,y);
+      ctx.lineTo(x+1,y+1);
+      ctx.stroke();
+    }
+  };
+
+  const _handleMouseUp = (e: React.MouseEvent<HTMLCanvasElement, MouseEvent>) => {
+    console.log("mouse up");
+    const canvas = canvasRef.current;
+    if (canvas) {
+      const ctx = canvas.getContext("2d");
+      // if (ctx) {
+      //   draw(ctx);
+      // }
+      const {nativeEvent: {x, y}} = e;
+
+      ctx.moveTo(x,y);
+      ctx.lineTo(x+1,y+1);
+      ctx.stroke();
+      ctx.closePath();
+    }
+  };
+
+
+  return (
+    <div>
+        <canvas
+          ref={canvasRef}
+          width={512}
+          height={512}
+          onMouseDown={_handleMouseDown}
+          onMouseUp={_handleMouseUp}
+        ></canvas>
+    </div>
+  );
+}
\ No newline at end of file
diff --git a/ui/frontend/build_src/src/components/organisms/displayPanel/index.tsx b/ui/frontend/build_src/src/components/organisms/displayPanel/index.tsx
index 0337520b..b752674c 100644
--- a/ui/frontend/build_src/src/components/organisms/displayPanel/index.tsx
+++ b/ui/frontend/build_src/src/components/organisms/displayPanel/index.tsx
@@ -10,6 +10,7 @@ import { doMakeImage, MakeImageKey } from "../../../api";
 import AudioDing from "./audioDing";
 
 import GeneratedImage from "../../molecules/generatedImage";
+import DrawImage from "../../molecules/drawImage";
 
 import {
   displayPanel,
@@ -109,6 +110,7 @@ export default function DisplayPanel() {
             />
 
             {/* TODO Put the in painting controls here */}
+            <DrawImage imageData={completedImages[0].data}></DrawImage>
           </div>
 
           <div className={previousImages}>