feat: Create Choreographer (#11)

Creates `Choreographer`, a platform specific instance that fires
`onFrame` callbacks at the screen's refresh rate.

The `Choreographer` instance runs on the Thread that created it. We
could create such instances on the UI Thread or any other arbitrary
Thread using either react-native-reanimated or
react-native-worklets-core.

Currently you can create a `Choreographer` using the `FilamentProxy`:

```ts
const choreographer = FilamentProxy.createChoreographer()
choreographer.addOnFrameCallback((timestamp: number) => {
  console.log(`onFrame at ${timestamp}`)
})
choreographer.start()
setTimeout(() => {
  choreographer.stop()
}, 1500)
```

Create on the UI Thread:

```ts
runOnUI(() => {
  'worklet'
  const choreographer = FilamentProxy.createChoreographer()
  choreographer.addOnFrameCallback((timestamp: number) => {
    'worklet'
    // TODO: This is currently not possible since we don't have a Worklets integration. This func cannot be called probably.
    // Reanimated did work on something pretty interesting where they abstract all of this on the JS side, meaning there's
    // no change required from our end - we just receive a jsi::Function as normal - but it's not yet public.
    // So to make that work, we need to integrate react-native-reanimated or react-native-worklets-core in RNF for now.
  })
})()
```

Note: You need to keep a strong reference on the `Choreographer`,
otherwise it might get deleted by GC.

Author: mrousavy

package/android/CMakeLists.txt

@@ -22,6 +22,7 @@ add_library(
         ../cpp/FilamentProxy.cpp
         ../cpp/Surface.cpp
         ../cpp/SurfaceProvider.cpp
+        ../cpp/Choreographer.cpp
         ../cpp/Listener.cpp
         ../cpp/jsi/HybridObject.cpp
         ../cpp/jsi/Promise.cpp
@@ -34,6 +35,7 @@ add_library(
         src/main/cpp/JNISharedPtr.cpp
         src/main/cpp/FilamentInstaller.cpp
         src/main/cpp/java-bindings/JFilamentProxy.cpp
+        src/main/cpp/java-bindings/JChoreographer.cpp
         src/main/cpp/java-bindings/JFilamentView.cpp
         src/main/cpp/java-bindings/JSurfaceProvider.cpp
 )

package/android/src/main/cpp/AndroidFilamentProxy.cpp

@@ -25,4 +25,8 @@ std::shared_ptr<FilamentView> AndroidFilamentProxy::findFilamentView(int id) {
   return _proxy->cthis()->findFilamentView(id);
 }
 
+std::shared_ptr<Choreographer> AndroidFilamentProxy::createChoreographer() {
+  return _proxy->cthis()->createChoreographer();
+}
+
 } // namespace margelo

package/android/src/main/cpp/AndroidFilamentProxy.h

@@ -25,6 +25,7 @@ class AndroidFilamentProxy : public FilamentProxy {
   // TODO(hanno): implement
   int loadModel(std::string path) override;
   std::shared_ptr<FilamentView> findFilamentView(int id) override;
+  std::shared_ptr<Choreographer> createChoreographer() override;
 
 private:
   jni::global_ref<JFilamentProxy::javaobject> _proxy;

package/android/src/main/cpp/Filament.cpp

@@ -1,4 +1,5 @@
 #include "FilamentInstaller.h"
+#include "JChoreographer.h"
 #include "JFilamentProxy.h"
 #include "JFilamentView.h"
 #include "JSurfaceProvider.h"
@@ -11,5 +12,6 @@ JNIEXPORT jint JNICALL JNI_OnLoad(JavaVM* vm, void*) {
     margelo::JFilamentProxy::registerNatives();
     margelo::JSurfaceProvider::registerNatives();
     margelo::JFilamentView::registerNatives();
+    margelo::JChoreographer::registerNatives();
   });
 }

package/android/src/main/cpp/java-bindings/JChoreographer.cpp

@@ -0,0 +1,37 @@
+//
+// Created by Marc Rousavy on 23.02.24.
+//
+
+#include "JChoreographer.h"
+#include <android/choreographer.h>
+
+namespace margelo {
+
+void JChoreographer::registerNatives() {
+  registerHybrid({
+      makeNativeMethod("initHybrid", JChoreographer::initHybrid),
+      makeNativeMethod("onFrame", JChoreographer::onFrameLong),
+  });
+}
+
+JChoreographer::JChoreographer(const jni::alias_ref<jhybridobject>& javaThis) : _javaPart(jni::make_global(javaThis)) {}
+
+jni::local_ref<JChoreographer::jhybriddata> JChoreographer::initHybrid(jni::alias_ref<jhybridobject> javaThis) {
+  return makeCxxInstance(javaThis);
+}
+
+void JChoreographer::onFrameLong(jlong timestamp) {
+  onFrame(static_cast<double>(timestamp));
+}
+
+void JChoreographer::start() {
+  static const auto method = javaClassLocal()->getMethod<void()>("start");
+  method(_javaPart);
+}
+
+void JChoreographer::stop() {
+  static const auto method = javaClassLocal()->getMethod<void()>("stop");
+  method(_javaPart);
+}
+
+} // namespace margelo

package/android/src/main/cpp/java-bindings/JChoreographer.h

@@ -0,0 +1,33 @@
+//
+// Created by Marc Rousavy on 23.02.24.
+//
+
+#pragma once
+
+#include "Choreographer.h"
+#include <fbjni/fbjni.h>
+
+namespace margelo {
+
+using namespace facebook;
+
+class JChoreographer : public jni::HybridClass<JChoreographer>, public Choreographer {
+public:
+  static void registerNatives();
+
+  void start() override;
+  void stop() override;
+  void onFrameLong(jlong timestamp);
+
+private:
+  friend HybridBase;
+  jni::global_ref<JChoreographer::javaobject> _javaPart;
+  static auto constexpr TAG = "JChoreographer";
+  static auto constexpr kJavaDescriptor = "Lcom/margelo/filament/FilamentChoreographer;";
+
+private:
+  explicit JChoreographer(const jni::alias_ref<jhybridobject>& javaThis);
+  static jni::local_ref<jhybriddata> initHybrid(jni::alias_ref<jhybridobject> javaThis);
+};
+
+} // namespace margelo

package/android/src/main/cpp/java-bindings/JFilamentProxy.cpp

@@ -3,6 +3,7 @@
 //
 
 #include "JFilamentProxy.h"
+#include "JChoreographer.h"
 #include "JFilamentView.h"
 #include "JNISharedPtr.h"
 #include <fbjni/fbjni.h>
@@ -32,6 +33,14 @@ std::shared_ptr<FilamentView> JFilamentProxy::findFilamentView(int id) {
   return std::static_pointer_cast<FilamentView>(sharedRef);
 }
 
+std::shared_ptr<Choreographer> JFilamentProxy::createChoreographer() {
+  static const auto method = javaClassLocal()->getMethod<jni::alias_ref<JChoreographer::javaobject>()>("createChoreographer");
+  jni::local_ref<JChoreographer::javaobject> choreographer = method(_javaPart);
+  jni::global_ref<JChoreographer::javaobject> globalRef = jni::make_global(choreographer);
+  std::shared_ptr<JChoreographer> sharedRef = JNISharedPtr::make_shared_from_jni<JChoreographer>(globalRef);
+  return std::static_pointer_cast<Choreographer>(sharedRef);
+}
+
 jsi::Runtime& JFilamentProxy::getRuntime() {
   if (_runtime == nullptr) {
     [[unlikely]];

package/android/src/main/cpp/java-bindings/JFilamentProxy.h

@@ -4,6 +4,7 @@
 
 #pragma once
 
+#include "Choreographer.h"
 #include "FilamentView.h"
 #include <ReactCommon/CallInvokerHolder.h>
 #include <fbjni/fbjni.h>
@@ -25,6 +26,7 @@ class JFilamentProxy : public jni::HybridClass<JFilamentProxy> {
   // TODO(hanno): implement
   int loadModel(const std::string& path);
   std::shared_ptr<FilamentView> findFilamentView(int id);
+  std::shared_ptr<Choreographer> createChoreographer();
 
   jsi::Runtime& getRuntime();
 

package/android/src/main/java/com/margelo/filament/FilamentChoreographer.java

@@ -0,0 +1,52 @@
+package com.margelo.filament;
+
+import android.view.Choreographer;
+
+import androidx.annotation.Keep;
+
+import com.facebook.jni.HybridData;
+import com.facebook.proguard.annotations.DoNotStrip;
+
+/** @noinspection JavaJniMissingFunction*/
+public class FilamentChoreographer {
+    /** @noinspection unused, FieldCanBeLocal */
+    @DoNotStrip
+    @Keep
+    private final HybridData mHybridData;
+    private final Choreographer choreographer;
+    private boolean isRunning;
+
+    public FilamentChoreographer() {
+        mHybridData = initHybrid();
+        choreographer = Choreographer.getInstance();
+    }
+
+    private void onFrameCallback(long timestamp) {
+        if (!isRunning) return;
+        onFrame(timestamp);
+        choreographer.postFrameCallback(this::onFrameCallback);
+    }
+
+    /** @noinspection unused */
+    @DoNotStrip
+    @Keep
+    private synchronized void start() {
+        if (!isRunning) {
+            isRunning = true;
+            choreographer.postFrameCallback(this::onFrameCallback);
+        }
+    }
+
+    /** @noinspection unused */
+    @DoNotStrip
+    @Keep
+    private synchronized void stop() {
+        if (isRunning) {
+            isRunning = false;
+            choreographer.removeFrameCallback(this::onFrameCallback);
+        }
+    }
+
+    private native HybridData initHybrid();
+    private native void onFrame(long timestamp);
+}

package/android/src/main/java/com/margelo/filament/FilamentProxy.java

@@ -36,6 +36,13 @@ class FilamentProxy {
         reactContext = context;
     }
 
+    /** @noinspection unused*/
+    @DoNotStrip
+    @Keep
+    FilamentChoreographer createChoreographer() {
+        return new FilamentChoreographer();
+    }
+
     /** @noinspection unused*/
     @DoNotStrip
     @Keep