Merge remote-tracking branch 'origin/dev' into visdf

codecs/wp2/enc/wp2_enc.cpp

@@ -0,0 +1,69 @@
+#include <emscripten/bind.h>
+#include <emscripten/threading.h>
+#include <emscripten/val.h>
+#include <cstdio>
+#include "src/wp2/encode.h"
+
+using namespace emscripten;
+
+thread_local const val Uint8Array = val::global("Uint8Array");
+
+struct WP2Options {
+  float quality;
+  float alpha_quality;
+  int effort;
+  int pass;
+  int uv_mode;
+  float sns;
+  int csp_type;
+  int error_diffusion;
+  bool use_random_matrix;
+};
+
+val encode(std::string image_in, int image_width, int image_height, WP2Options options) {
+  WP2::EncoderConfig config = {};
+
+  config.quality = options.quality;
+  config.alpha_quality = options.alpha_quality;
+  config.effort = options.effort;
+  config.pass = options.pass;
+  config.uv_mode = static_cast<WP2::EncoderConfig::UVMode>(options.uv_mode);
+  config.csp_type = static_cast<WP2::Csp>(options.csp_type);
+  config.sns = options.sns;
+  config.error_diffusion = options.error_diffusion;
+  config.use_random_matrix = options.use_random_matrix;
+
+  uint8_t* image_buffer = (uint8_t*)image_in.c_str();
+  WP2::ArgbBuffer src = WP2::ArgbBuffer();
+  WP2Status status =
+      src.Import(WP2_rgbA_32,  // Format. WP2_RGBA_32 is the same but NOT premultiplied alpha
+                 image_width, image_height, image_buffer, 4 * image_width);
+  if (status != WP2_STATUS_OK) {
+    return val::null();
+  }
+
+  WP2::MemoryWriter memory_writer;
+  // In WebP2, thread_level is number of *extra* threads to use (0 for no multithreading).
+  config.thread_level = emscripten_num_logical_cores() - 1;
+  status = WP2::Encode(src, &memory_writer, config);
+  if (status != WP2_STATUS_OK) {
+    return val::null();
+  }
+
+  return Uint8Array.new_(typed_memory_view(memory_writer.size_, memory_writer.mem_));
+}
+
+EMSCRIPTEN_BINDINGS(my_module) {
+  value_object<WP2Options>("WP2Options")
+      .field("quality", &WP2Options::quality)
+      .field("alpha_quality", &WP2Options::alpha_quality)
+      .field("effort", &WP2Options::effort)
+      .field("pass", &WP2Options::pass)
+      .field("uv_mode", &WP2Options::uv_mode)
+      .field("csp_type", &WP2Options::csp_type)
+      .field("error_diffusion", &WP2Options::error_diffusion)
+      .field("use_random_matrix", &WP2Options::use_random_matrix)
+      .field("sns", &WP2Options::sns);
+
+  function("encode", &encode);
+}