From 83c6b501faa7ff52cb4a317a4142725bf08ca39a Mon Sep 17 00:00:00 2001
From: nanguage <nanguage@yahoo.com>
Date: Tue, 17 Oct 2023 00:36:15 +0800
Subject: [PATCH] control kaibu view dataset with imjoy plugin

---
 public/plugins/viewer.imjoy.html |  8 ++++++++
 src/views/DatasetView.vue        | 14 +++++++-------
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/public/plugins/viewer.imjoy.html b/public/plugins/viewer.imjoy.html
index 7aed062..e1f4209 100644
--- a/public/plugins/viewer.imjoy.html
+++ b/public/plugins/viewer.imjoy.html
@@ -20,7 +20,9 @@
 </config>
 
 <script lang="python">
+import io
 import tifffile
+from imjoy import api
 
 class Plugin():
     async def setup(self):
@@ -29,6 +31,7 @@
             src="https://kaibu.org/#/app",
             window_id="kaibu-container"
         )
+        await self.viewer.open_sidebar(False)
         self.image = None
 
     def load_image_from_bytes(self, file_name, img_bytes):
@@ -37,12 +40,17 @@
         with tifffile.TiffFile(_file) as tif:
             image = tif.asarray()
         self.image = image
+        return image
 
     async def view_img_from_bytes(self, file_name, bytes_):
+        await self.viewer.clear_layers()
         image = self.load_image_from_bytes(file_name, bytes_)
         await self.viewer.view_image(image, name=file_name)
         return image.shape
 
+    async def add_points(self, points, name="points"):
+        await self.viewer.add_points(points, name=name)
+
     async def run(self, ctx):
         pass
 
diff --git a/src/views/DatasetView.vue b/src/views/DatasetView.vue
index 36ecb0e..afa1068 100644
--- a/src/views/DatasetView.vue
+++ b/src/views/DatasetView.vue
@@ -80,10 +80,9 @@ export default {
 
     async loadViewer() {
       if (imjoy !== null) {
-        const viewer = await imjoy.api.createWindow(
-          { src: "https://kaibu.org/#/app", window_id: "kaibu-container" })
+        const pluginUrl = window.location.origin + "/plugins/viewer.imjoy.html"
+        const viewer = await imjoy.api.loadPlugin({ src: pluginUrl })
         this.viewer = viewer
-        await viewer.set_mode("full")
       } else {
         setTimeout(() => this.loadViewer(), 1000)
       }
@@ -96,14 +95,15 @@ export default {
         const point = data.points[0];
         const imgName = point.text.split("<br>")[0].split(":</b> ")[1]
         console.log(`You clicked on point: ${imgName} at position (x: ${point.x}, y: ${point.y})`)
-        //const imgURL = this.imgBaseURL + imgName
-        //await this.viewer.view_image(imgURL, { name: imgName, type: 'itk-vtk' })
+        const imgURL = this.imgBaseURL + imgName
+        const imgBytes = await fetch(imgURL).then(resp => resp.arrayBuffer())
+        await this.viewer.view_img_from_bytes(imgName, imgBytes)
         const csvName = imgName.replace(".tif", ".csv")
         const csvURL = this.csvBaseURL + csvName
         const csvContent = await fetch(csvURL).then(resp => resp.text())
         const csvData = csv.toObjects(csvContent)
-        const points = csvData.map((d: any) => [d['axis-0'], d['axis-1']])
-        await this.viewer.add_points(points, { name: csvName })
+        const points = csvData.map((d: any) => [d['axis-1'], d['axis-0']])
+        await this.viewer.add_points(points, csvName)
       }
     },