feat(Canvas): add support for displaying output in Jupyter Notebook (#60

)
DjDeveloperr · Oct 7, 2023 · a6a2219 · a6a2219
1 parent 57b8e31
commit a6a2219
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/canvas.ts b/src/canvas.ts
@@ -209,6 +209,12 @@ export class Canvas {
   flush() {
     if (this[_gpu]) sk_canvas_flush(this[_ptr]);
   }
+
+  [Symbol.for("Jupyter.display")]() {
+    return {
+      "image/png": encodeBase64(this.encode("png")),
+    };
+  }
 }
 
 /**

diff --git a/src/svgcanvas.ts b/src/svgcanvas.ts
@@ -114,6 +114,13 @@ export class SvgCanvas {
     );
     const text = new TextDecoder().decode(buffer);
     sk_data_free(skdata);
+    Object.defineProperty(text, Symbol.for("Jupyter.display"), {
+      value: function (this: string) {
+        return {
+          "image/svg+xml": this,
+        };
+      },
+    });
     return text;
   }