From c47c2239cb26682b6d48022fbf81290f81e5b8ec Mon Sep 17 00:00:00 2001
From: nighca <nighca@live.cn>
Date: Fri, 10 Jan 2025 12:07:40 +0800
Subject: [PATCH] Optimize performance of ProjectRunnerV1 for svg assets

---
 .../project/runner/v1/ProjectRunnerV1.vue     | 20 ++++++-
 spx-gui/src/models/common/file.ts             | 20 ++-----
 spx-gui/src/utils/img.ts                      | 52 ++++++++++++++++---
 3 files changed, 67 insertions(+), 25 deletions(-)

diff --git a/spx-gui/src/components/project/runner/v1/ProjectRunnerV1.vue b/spx-gui/src/components/project/runner/v1/ProjectRunnerV1.vue
index 89a519585..3e1dbca69 100644
--- a/spx-gui/src/components/project/runner/v1/ProjectRunnerV1.vue
+++ b/spx-gui/src/components/project/runner/v1/ProjectRunnerV1.vue
@@ -8,9 +8,12 @@
 
 <script lang="ts" setup>
 import { onMounted, onUnmounted, ref } from 'vue'
+import JSZip from 'jszip'
 import { registerPlayer } from '@/utils/player-registry'
 import { until } from '@/utils/utils'
 import { useFileUrl } from '@/utils/file'
+import { toPng } from '@/utils/img'
+import { toNativeFile } from '@/models/common/file'
 import { Project } from '@/models/project'
 import { UIImg, UILoading } from '@/components/ui'
 import IframeDisplay, { preload } from './IframeDisplay.vue'
@@ -44,12 +47,25 @@ onUnmounted(() => {
   registered.onStopped()
 })
 
+async function getProjectZipData() {
+  const zip = new JSZip()
+  const [, files] = await props.project.export()
+  Object.entries(files).forEach(([path, file]) => {
+    if (file == null) return
+    const f = toNativeFile(file).then(nf => {
+      if (nf.type === 'image/svg+xml') return toPng(nf)
+      return nf
+    })
+    if (file != null) zip.file(path, f)
+  })
+  return zip.generateAsync({ type: 'arraybuffer' })
+}
+
 defineExpose({
   async run() {
     loading.value = true
     registered.onStart()
-    const gbpFile = await props.project.exportGbpFile()
-    zipData.value = await gbpFile.arrayBuffer()
+    zipData.value = await getProjectZipData()
     await until(() => !loading.value)
   },
   stop() {
diff --git a/spx-gui/src/models/common/file.ts b/spx-gui/src/models/common/file.ts
index b692fc96b..832b35a14 100644
--- a/spx-gui/src/models/common/file.ts
+++ b/spx-gui/src/models/common/file.ts
@@ -6,6 +6,7 @@
 
 import { markRaw } from 'vue'
 import { getMimeFromExt } from '@/utils/file'
+import { getSVGSize } from '@/utils/img'
 import { extname } from '@/utils/path'
 import { Disposable, type Disposer } from '@/utils/disposable'
 import { Cancelled } from '@/utils/exception'
@@ -122,20 +123,6 @@ export async function toConfig(file: File) {
   return JSON.parse(text) as unknown
 }
 
-async function getSVGImageSize(svgFile: File) {
-  const svgText = await toText(svgFile)
-  const parser = new DOMParser()
-  const svg = parser.parseFromString(svgText, 'image/svg+xml').documentElement
-  if (!(svg instanceof SVGSVGElement)) throw new Error(`invalid svg: ${svgFile.name}`)
-  // Keep consistent with spx, for details see:
-  // * https://github.com/goplus/spx/blob/15b2e572746f3aaea519c2d9c0027188b50b62c8/internal/svgr/svg.go#L39
-  // * https://github.com/qiniu/oksvg/blob/917f53935572252ba3da8909ca4fbedec418bde1/svgd.go#L1015-L1049
-  let { width, height } = svg.viewBox.baseVal
-  if (width === 0) width = svg.width.baseVal.value
-  if (height === 0) height = svg.height.baseVal.value
-  return { width, height }
-}
-
 async function getBitmapImageSize(bitmapImgFile: File) {
   const d = new Disposable()
   const imgUrl = await bitmapImgFile.url((fn) => d.addDisposer(fn))
@@ -157,7 +144,10 @@ async function getBitmapImageSize(bitmapImgFile: File) {
 }
 
 export async function getImageSize(file: File) {
-  if (file.type === 'image/svg+xml') return getSVGImageSize(file)
+  if (file.type === 'image/svg+xml') {
+    const svgText = await toText(file)
+    return getSVGSize(svgText)
+  }
   return getBitmapImageSize(file)
 }
 
diff --git a/spx-gui/src/utils/img.ts b/spx-gui/src/utils/img.ts
index 688d81b35..2728c2507 100644
--- a/spx-gui/src/utils/img.ts
+++ b/spx-gui/src/utils/img.ts
@@ -1,28 +1,64 @@
 import { Disposable } from '@/utils/disposable'
 
-/** Convert arbitrary-type (supported by current browser) image content to type-`image/jpeg` content. */
-export async function toJpeg(blob: Blob) {
+/** Convert arbitrary-type (supported by current browser) image content to another type. */
+export function convertImg(
+  /** Input image */
+  input: Blob,
+  /** Mime type of the output image, see details in https://developer.mozilla.org/en-US/docs/Web/API/HTMLCanvasElement/toBlob#type */
+  type: string
+) {
   const d = new Disposable()
   return new Promise<Blob>((resolve, reject) => {
     const img = new Image()
-    img.onload = () => {
+    img.onload = async () => {
       const canvas = document.createElement('canvas')
-      canvas.width = img.naturalWidth
-      canvas.height = img.naturalHeight
-      canvas.getContext('2d')?.drawImage(img, 0, 0)
+      if (input.type === 'image/svg+xml') {
+        const svgText = await input.text()
+        const { width, height } = await getSVGSize(svgText)
+        canvas.width = width
+        canvas.height = height
+      } else {
+        canvas.width = img.naturalWidth
+        canvas.height = img.naturalHeight
+      }
+      canvas.getContext('2d')?.drawImage(img, 0, 0, canvas.width, canvas.height)
       canvas.toBlob((newBlob) => {
         if (newBlob == null) {
           reject(new Error('toBlob failed'))
           return
         }
         resolve(newBlob)
-      }, 'image/jpeg')
+      }, type)
     }
     img.onerror = (e) => reject(new Error(`load image failed: ${e.toString()}`))
-    const url = URL.createObjectURL(blob)
+    const url = URL.createObjectURL(input)
     d.addDisposer(() => URL.revokeObjectURL(url))
     img.src = url
   }).finally(() => {
     d.dispose()
   })
 }
+
+/** Convert arbitrary-type (supported by current browser) image content to type-`image/jpeg` content. */
+export function toJpeg(blob: Blob) {
+  return convertImg(blob, 'image/jpeg')
+}
+
+/** Convert arbitrary-type (supported by current browser) image content to type-`image/png` content. */
+export async function toPng(blob: Blob) {
+  return convertImg(blob, 'image/png')
+}
+
+/** Get the size of the SVG image, keeping consistent with spx. */
+export async function getSVGSize(svgText: string) {
+  const parser = new DOMParser()
+  const svg = parser.parseFromString(svgText, 'image/svg+xml').documentElement
+  if (!(svg instanceof SVGSVGElement)) throw new Error('invalid svg')
+  // Keep consistent with spx, for details see:
+  // * https://github.com/goplus/spx/blob/15b2e572746f3aaea519c2d9c0027188b50b62c8/internal/svgr/svg.go#L39
+  // * https://github.com/qiniu/oksvg/blob/917f53935572252ba3da8909ca4fbedec418bde1/svgd.go#L1015-L1049
+  let { width, height } = svg.viewBox.baseVal
+  if (width === 0) width = svg.width.baseVal.value
+  if (height === 0) height = svg.height.baseVal.value
+  return { width, height }
+}