refactor: use picture recorder backend

Author: Brooooooklyn

skia-c/skia_c.cpp

@@ -61,6 +61,38 @@ extern "C"
     SkGraphics::PurgeAllCaches();
   }
 
+  void skiac_create_picture_recorder(uint32_t width, uint32_t height, skiac_picture_recorder_created *pr)
+  {
+    auto picture_recorder = new SkPictureRecorder();
+    auto canvas = picture_recorder->beginRecording((SkScalar)width, (SkScalar)height);
+    auto surface = canvas->getSurface();
+    pr->recorder = reinterpret_cast<skiac_picture_recorder *>(picture_recorder);
+    pr->canvas = reinterpret_cast<skiac_canvas *>(canvas);
+    pr->surface = reinterpret_cast<skiac_surface *>(surface);
+  }
+
+  void skiac_picture_recorder_destroy(skiac_picture_recorder *pr)
+  {
+    delete reinterpret_cast<SkPictureRecorder *>(pr);
+  }
+
+  skiac_picture *skiac_picture_recorder_finish_as_picture(skiac_picture_recorder *pr)
+  {
+    auto picture_recorder = reinterpret_cast<SkPictureRecorder *>(pr);
+    auto picture = picture_recorder->finishRecordingAsPicture().release();
+    return reinterpret_cast<skiac_picture *>(picture);
+  }
+
+  void skiac_picture_ref(skiac_picture *picture)
+  {
+    SkSafeRef(reinterpret_cast<SkPicture *>(picture));
+  }
+
+  void skiac_picture_destroy(skiac_picture *picture)
+  {
+    SkSafeUnref(reinterpret_cast<SkPicture *>(picture));
+  }
+
   // Surface
 
   static SkSurface *skiac_surface_create(int width, int height, SkAlphaType alphaType, uint8_t cs)
@@ -133,7 +165,7 @@ extern "C"
   void skiac_surface_destroy(skiac_surface *c_surface)
   {
     // SkSurface is ref counted.
-    SURFACE_CAST->unref();
+    SkSafeUnref(SURFACE_CAST);
   }
 
   skiac_surface *skiac_surface_copy_rgba(
@@ -356,6 +388,16 @@ extern "C"
     CANVAS_CAST->drawImageRect(image, src, dst, sampling, nullptr, SkCanvas::kFast_SrcRectConstraint);
   }
 
+  void skiac_canvas_draw_picture(
+      skiac_canvas *c_canvas,
+      skiac_picture *c_picture,
+      skiac_matrix *c_matrix,
+      skiac_paint *c_paint)
+  {
+    auto picture = reinterpret_cast<SkPicture *>(c_picture);
+    CANVAS_CAST->drawPicture(picture, MATRIX_CAST, PAINT_CAST);
+  }
+
   void skiac_canvas_get_line_metrics_or_draw_text(
       const char *text,
       size_t text_len,

skia-c/skia_c.hpp

@@ -5,6 +5,7 @@
 #include <include/core/SkPicture.h>
 #include <include/core/SkSamplingOptions.h>
 #include <include/core/SkString.h>
+#include <include/core/SkPictureRecorder.h>
 #include <include/effects/SkImageFilters.h>
 #include <include/pathops/SkPathOps.h>
 #include <include/utils/SkParsePath.h>
@@ -47,10 +48,12 @@
 
 using namespace skia::textlayout;
 
+typedef struct skiac_picture_recorder skiac_picture_recorder;
 typedef struct skiac_surface skiac_surface;
 typedef struct skiac_canvas skiac_canvas;
 typedef struct skiac_paint skiac_paint;
 typedef struct skiac_path skiac_path;
+typedef struct skiac_picture skiac_picture;
 typedef struct skiac_shader skiac_shader;
 typedef struct skiac_path_effect skiac_path_effect;
 typedef struct skiac_matrix skiac_matrix;
@@ -137,6 +140,13 @@ struct skiac_font_collection
   }
 };
 
+struct skiac_picture_recorder_created
+{
+  skiac_picture_recorder *recorder;
+  skiac_canvas *canvas;
+  skiac_surface *surface;
+};
+
 struct skiac_line_metrics
 {
   float ascent;
@@ -212,6 +222,13 @@ struct skiac_mapped_point
 extern "C"
 {
   void skiac_clear_all_cache();
+  // PictureRecorder
+  void skiac_create_picture_recorder(uint32_t width, uint32_t height, skiac_picture_recorder_created *pr);
+  void skiac_picture_recorder_destroy(skiac_picture_recorder *pr);
+  // Picture
+  skiac_picture *skiac_picture_recorder_finish_as_picture(skiac_picture_recorder *pr);
+  void skiac_picture_ref(skiac_picture *picture);
+  void skiac_picture_destroy(skiac_picture *picture);
   // Surface
   skiac_surface *skiac_surface_create_rgba_premultiplied(int width, int height, uint8_t cs);
   void skiac_surface_create_svg(skiac_svg_surface *c_surface, int width, int height, int alphaType, uint32_t flag, uint8_t cs);
@@ -284,6 +301,11 @@ extern "C"
       float dw,
       float dh,
       int filter_quality);
+  void skiac_canvas_draw_picture(
+      skiac_canvas *c_canvas,
+      skiac_picture *c_picture,
+      skiac_matrix *c_matrix,
+      skiac_paint *c_paint);
   void skiac_canvas_get_line_metrics_or_draw_text(
       const char *text,
       size_t text_len,

src/ctx.rs

@@ -43,13 +43,26 @@ pub struct Context {
   pub alpha: bool,
   pub(crate) states: Vec<Context2dRenderingState>,
   state: Context2dRenderingState,
-  pub width: u32,
-  pub height: u32,
   pub color_space: ColorSpace,
   pub stream: Option<SkWMemoryStream>,
 }
 
 impl Context {
+  pub fn new_recorder(width: u32, height: u32, color_space: ColorSpace) -> Result<Self> {
+    let surface = Surface::new_picture_recorder(width, height).ok_or_else(|| {
+      Error::from_reason("Create skia picture recorder surface failed".to_owned())
+    })?;
+    Ok(Self {
+      surface,
+      alpha: true,
+      path: SkPath::new(),
+      states: vec![],
+      state: Context2dRenderingState::default(),
+      color_space,
+      stream: None,
+    })
+  }
+
   pub fn new_svg(
     width: u32,
     height: u32,
@@ -70,8 +83,6 @@ impl Context {
       path: SkPath::new(),
       states: vec![],
       state: Context2dRenderingState::default(),
-      width,
-      height,
       color_space,
       stream: Some(stream),
     })
@@ -86,8 +97,6 @@ impl Context {
       path: SkPath::new(),
       states: vec![],
       state: Context2dRenderingState::default(),
-      width,
-      height,
       color_space,
       stream: None,
     })
@@ -659,6 +668,7 @@ impl Context {
     paint: &Paint,
   ) -> result::Result<(), SkError> {
     let state = &self.state;
+    let width = self.surface.width;
     let weight = state.font_style.weight;
     let stretch = state.font_style.stretch;
     let slant = state.font_style.style;
@@ -671,7 +681,7 @@ impl Context {
         x,
         y,
         max_width,
-        self.width as f32,
+        width as f32,
         weight,
         stretch as i32,
         slant,
@@ -691,7 +701,7 @@ impl Context {
       x,
       y,
       max_width,
-      self.width as f32,
+      self.surface.width as f32,
       weight,
       stretch as i32,
       slant,
@@ -785,7 +795,9 @@ pub struct CanvasRenderingContext2D {
 
 impl ObjectFinalize for CanvasRenderingContext2D {
   fn finalize(self, mut env: Env) -> Result<()> {
-    env.adjust_external_memory(-((self.context.width * self.context.height * 4) as i64))?;
+    env.adjust_external_memory(
+      -((self.context.surface.width * self.context.surface.height * 4) as i64),
+    )?;
     Ok(())
   }
 }
@@ -803,7 +815,7 @@ impl CanvasRenderingContext2D {
     let context = if let Some(flag) = flag {
       Context::new_svg(width, height, flag.into(), color_space)?
     } else {
-      Context::new(width, height, color_space)?
+      Context::new_recorder(width, height, color_space)?
     };
     Ok(Self { context })
   }

src/lib.rs

@@ -251,7 +251,12 @@ impl CanvasElement {
       "png" => ContextData::Png(surface_ref),
       "avif" => {
         let cfg = AvifConfig::from(&quality_or_config);
-        ContextData::Avif(surface_ref, cfg.into(), ctx2d.width, ctx2d.height)
+        ContextData::Avif(
+          surface_ref,
+          cfg.into(),
+          ctx2d.surface.width,
+          ctx2d.surface.height,
+        )
       }
       _ => {
         return Err(Error::new(
@@ -313,8 +318,8 @@ fn get_data_ref(
       let config = AvifConfig::from(quality_or_config).into();
       let output = avif::encode(
         unsafe { slice::from_raw_parts(data, size) },
-        ctx2d.width,
-        ctx2d.height,
+        ctx2d.surface.width,
+        ctx2d.surface.height,
         &config,
       )
       .map_err(|e| Error::new(Status::GenericFailure, format!("{}", e)))?;
@@ -462,7 +467,10 @@ impl SVGCanvas {
   #[napi]
   pub fn get_content(&self, env: Env) -> Result<JsBuffer> {
     let svg_data_stream = self.ctx.context.stream.as_ref().unwrap();
-    let svg_data = svg_data_stream.data(self.ctx.context.width, self.ctx.context.height);
+    let svg_data = svg_data_stream.data(
+      self.ctx.context.surface.width,
+      self.ctx.context.surface.height,
+    );
     unsafe {
       env
         .create_buffer_with_borrowed_data(svg_data.0.ptr, svg_data.0.size, svg_data, |d, _| {