react-native-wgpu 0.2.10 → 0.3.1

package/cpp/rnwgpu/async/AsyncTaskHandle.h

@@ -0,0 +1,55 @@
+#pragma once
+
+#include <functional>
+#include <memory>
+#include <mutex>
+#include <optional>
+#include <string>
+
+#include <jsi/jsi.h>
+
+#include "AsyncDispatcher.h"
+
+namespace margelo {
+class Promise;
+}
+
+namespace rnwgpu::async {
+
+class AsyncRunner;
+
+/**
+ * Represents a pending asynchronous WebGPU operation that can be converted into
+ * a JavaScript Promise.
+ */
+class AsyncTaskHandle {
+public:
+  struct State;
+
+  using ValueFactory =
+      std::function<facebook::jsi::Value(facebook::jsi::Runtime &)>;
+  using ResolveFunction = std::function<void(ValueFactory)>;
+  using RejectFunction = std::function<void(std::string)>;
+
+  AsyncTaskHandle();
+
+  /**
+   * Internal constructor used by AsyncRunner.
+   */
+  explicit AsyncTaskHandle(std::shared_ptr<State> state);
+
+  bool valid() const;
+
+  ResolveFunction createResolveFunction() const;
+  RejectFunction createRejectFunction() const;
+
+  void attachPromise(const std::shared_ptr<margelo::Promise> &promise) const;
+
+  static AsyncTaskHandle create(const std::shared_ptr<AsyncRunner> &runner,
+                                bool keepPumping);
+
+private:
+  std::shared_ptr<State> _state;
+};
+
+} // namespace rnwgpu::async

package/cpp/rnwgpu/async/JSIMicrotaskDispatcher.cpp

@@ -0,0 +1,23 @@
+#include "JSIMicrotaskDispatcher.h"
+
+#include <utility>
+
+namespace rnwgpu::async {
+
+JSIMicrotaskDispatcher::JSIMicrotaskDispatcher(jsi::Runtime &runtime)
+    : _runtime(runtime) {}
+
+void JSIMicrotaskDispatcher::post(Work work) {
+  auto microtask = jsi::Function::createFromHostFunction(
+      _runtime, jsi::PropNameID::forAscii(_runtime, "AsyncMicrotask"), 0,
+      [work = std::move(work)](
+          jsi::Runtime &runtime, const jsi::Value & /*thisValue*/,
+          const jsi::Value * /*args*/, size_t /*count*/) -> jsi::Value {
+        work(runtime);
+        return jsi::Value::undefined();
+      });
+
+  _runtime.queueMicrotask(std::move(microtask));
+}
+
+} // namespace rnwgpu::async

package/cpp/rnwgpu/async/JSIMicrotaskDispatcher.h

@@ -0,0 +1,22 @@
+#pragma once
+
+#include "AsyncDispatcher.h"
+
+namespace rnwgpu::async {
+
+/**
+ * Dispatcher implementation backed by `jsi::Runtime::queueMicrotask`.
+ */
+class JSIMicrotaskDispatcher final
+    : public AsyncDispatcher,
+      public std::enable_shared_from_this<JSIMicrotaskDispatcher> {
+public:
+  explicit JSIMicrotaskDispatcher(jsi::Runtime &runtime);
+
+  void post(Work work) override;
+
+private:
+  jsi::Runtime &_runtime;
+};
+
+} // namespace rnwgpu::async