|
| 1 | +#include "JRiveWorkletDispatcher.hpp" |
| 2 | +#include <android/log.h> |
| 3 | + |
| 4 | +namespace margelo::nitro::rive { |
| 5 | + |
| 6 | +using namespace facebook; |
| 7 | + |
| 8 | +JRiveWorkletDispatcher::JRiveWorkletDispatcher( |
| 9 | + jni::alias_ref<JRiveWorkletDispatcher::jhybridobject> jThis) |
| 10 | + : _javaPart(jni::make_global(jThis)) {} |
| 11 | + |
| 12 | +jni::local_ref<JRiveWorkletDispatcher::jhybriddata> JRiveWorkletDispatcher::initHybrid( |
| 13 | + jni::alias_ref<jhybridobject> jThis) { |
| 14 | + return makeCxxInstance(jThis); |
| 15 | +} |
| 16 | + |
| 17 | +jni::local_ref<JRiveWorkletDispatcher::javaobject> JRiveWorkletDispatcher::create() { |
| 18 | + return newObjectJavaArgs(); |
| 19 | +} |
| 20 | + |
| 21 | +void JRiveWorkletDispatcher::trigger() { |
| 22 | + std::unique_lock lock(_mutex); |
| 23 | + while (!_jobs.empty()) { |
| 24 | + auto job = std::move(_jobs.front()); |
| 25 | + _jobs.pop(); |
| 26 | + lock.unlock(); |
| 27 | + job(); |
| 28 | + lock.lock(); |
| 29 | + } |
| 30 | +} |
| 31 | + |
| 32 | +void JRiveWorkletDispatcher::scheduleTrigger() { |
| 33 | + static const auto method = _javaPart->getClass()->getMethod<void()>("scheduleTrigger"); |
| 34 | + method(_javaPart.get()); |
| 35 | +} |
| 36 | + |
| 37 | +void JRiveWorkletDispatcher::runAsync(std::function<void()>&& function) { |
| 38 | + std::unique_lock lock(_mutex); |
| 39 | + _jobs.push(std::move(function)); |
| 40 | + lock.unlock(); |
| 41 | + scheduleTrigger(); |
| 42 | +} |
| 43 | + |
| 44 | +void JRiveWorkletDispatcher::runSync(std::function<void()>&& function) { |
| 45 | + std::mutex mtx; |
| 46 | + std::condition_variable cv; |
| 47 | + bool done = false; |
| 48 | + |
| 49 | + runAsync([&]() { |
| 50 | + function(); |
| 51 | + { |
| 52 | + std::lock_guard<std::mutex> lock(mtx); |
| 53 | + done = true; |
| 54 | + } |
| 55 | + cv.notify_one(); |
| 56 | + }); |
| 57 | + |
| 58 | + std::unique_lock<std::mutex> lock(mtx); |
| 59 | + cv.wait(lock, [&]{ return done; }); |
| 60 | +} |
| 61 | + |
| 62 | +void JRiveWorkletDispatcher::registerNatives() { |
| 63 | + registerHybrid({ |
| 64 | + makeNativeMethod("initHybrid", JRiveWorkletDispatcher::initHybrid), |
| 65 | + makeNativeMethod("trigger", JRiveWorkletDispatcher::trigger), |
| 66 | + }); |
| 67 | +} |
| 68 | + |
| 69 | +AndroidMainThreadDispatcher::AndroidMainThreadDispatcher( |
| 70 | + jni::local_ref<JRiveWorkletDispatcher::javaobject> javaDispatcher) |
| 71 | + : _javaDispatcher(jni::make_global(javaDispatcher)) {} |
| 72 | + |
| 73 | +void AndroidMainThreadDispatcher::runAsync(std::function<void()>&& function) { |
| 74 | + _javaDispatcher->cthis()->runAsync(std::move(function)); |
| 75 | +} |
| 76 | + |
| 77 | +void AndroidMainThreadDispatcher::runSync(std::function<void()>&& function) { |
| 78 | + _javaDispatcher->cthis()->runSync(std::move(function)); |
| 79 | +} |
| 80 | + |
| 81 | +} // namespace margelo::nitro::rive |
0 commit comments