1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
|
#include "bionic_fg/session.hpp"
#include "framegen_context.hpp"
#include "logging.hpp"
#include <jni.h>
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
namespace bionic_fg {
static jstring toJStr(JNIEnv* env, const std::string& s) {
return env->NewStringUTF(s.c_str());
}
static Session* sessionFromHandle(jlong h) {
return reinterpret_cast<Session*>(static_cast<uintptr_t>(h));
}
static jlong sessionToHandle(Session* p) {
return static_cast<jlong>(reinterpret_cast<uintptr_t>(p));
}
static FramegenContext* ctxFromHandle(jlong h) {
return reinterpret_cast<FramegenContext*>(static_cast<uintptr_t>(h));
}
static jlong ctxToHandle(FramegenContext* p) {
return static_cast<jlong>(reinterpret_cast<uintptr_t>(p));
}
} // namespace bionic_fg
// ─── Session bootstrap JNI ───────────────────────────────────────────────────
extern "C" JNIEXPORT jboolean JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeIsReady(
JNIEnv*, jclass) {
using namespace bionic_fg;
return embedded::kShaderRegistry.size() > 0 &&
Session::EmbeddedShaderCount() == Session::ValidEmbeddedShaderCount();
}
extern "C" JNIEXPORT jint JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeShaderCount(JNIEnv*, jclass) {
return static_cast<jint>(bionic_fg::Session::EmbeddedShaderCount());
}
extern "C" JNIEXPORT jint JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeValidShaderCount(JNIEnv*, jclass) {
return static_cast<jint>(bionic_fg::Session::ValidEmbeddedShaderCount());
}
extern "C" JNIEXPORT jstring JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeDescribeBundle(JNIEnv* env, jclass) {
return bionic_fg::toJStr(env, bionic_fg::Session::DescribeEmbeddedBundle());
}
extern "C" JNIEXPORT jlong JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeCreateSession(
JNIEnv*, jclass, jint width, jint height, jint multiplier,
jfloat flowScale, jint model) {
bionic_fg::Config cfg;
cfg.width = static_cast<uint32_t>(width > 0 ? width : 0);
cfg.height = static_cast<uint32_t>(height > 0 ? height : 0);
cfg.multiplier = static_cast<uint32_t>(multiplier > 0 ? multiplier : 2);
cfg.flowScale = flowScale;
cfg.model = static_cast<uint32_t>(model >= 0 ? model : 0);
auto* s = new bionic_fg::Session(cfg);
return bionic_fg::sessionToHandle(s);
}
extern "C" JNIEXPORT void JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeDestroySession(
JNIEnv*, jclass, jlong h) {
delete bionic_fg::sessionFromHandle(h);
}
extern "C" JNIEXPORT jboolean JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeUpdateSessionConfig(
JNIEnv*, jclass, jlong h, jint w, jint ht, jint mult, jfloat fs, jint model) {
auto* s = bionic_fg::sessionFromHandle(h);
if (!s) return JNI_FALSE;
bionic_fg::Config cfg;
cfg.width = static_cast<uint32_t>(w > 0 ? w : 0);
cfg.height = static_cast<uint32_t>(ht > 0 ? ht : 0);
cfg.multiplier = static_cast<uint32_t>(mult > 0 ? mult : 2);
cfg.flowScale = fs;
cfg.model = static_cast<uint32_t>(model >= 0 ? model : 0);
s->updateConfig(cfg);
return JNI_TRUE;
}
extern "C" JNIEXPORT jstring JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeDescribeSession(
JNIEnv* env, jclass, jlong h) {
auto* s = bionic_fg::sessionFromHandle(h);
if (!s) return bionic_fg::toJStr(env, "BionicFGSession{null}");
return bionic_fg::toJStr(env, s->describe());
}
// ─── FramegenContext JNI (AHB path) ─────────────────────────────────────────
extern "C" JNIEXPORT jlong JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeCreateContext(
JNIEnv* env, jclass,
jobject prevAhb, jobject currAhb,
jobjectArray outputAhbs,
jint width, jint height,
jint multiplier, jfloat flowScale, jint model) {
#ifdef __ANDROID__
auto* prevBuf = static_cast<AHardwareBuffer*>(env->GetDirectBufferAddress(prevAhb));
auto* currBuf = static_cast<AHardwareBuffer*>(env->GetDirectBufferAddress(currAhb));
if (!prevBuf || !currBuf) {
BFG_LOGE("nativeCreateContext: null AHB (not a direct buffer?)");
return 0L;
}
jsize outCount = env->GetArrayLength(outputAhbs);
std::vector<AHardwareBuffer*> outBufs;
outBufs.reserve(static_cast<size_t>(outCount));
for (jsize i = 0; i < outCount; ++i) {
auto* buf = static_cast<AHardwareBuffer*>(
env->GetDirectBufferAddress(env->GetObjectArrayElement(outputAhbs, i)));
if (!buf) { BFG_LOGE("nativeCreateContext: null output AHB at index %d", i); return 0L; }
outBufs.push_back(buf);
}
bionic_fg::Config cfg;
cfg.width = static_cast<uint32_t>(width);
cfg.height = static_cast<uint32_t>(height);
cfg.multiplier = static_cast<uint32_t>(multiplier);
cfg.flowScale = flowScale;
cfg.model = static_cast<uint32_t>(model);
VkExtent2D extent;
extent.width = static_cast<uint32_t>(width);
extent.height = static_cast<uint32_t>(height);
// Standalone JNI test path: create our own device (owned). The Vulkan-layer
// path instead passes the application's wrapped device (single-device mode).
auto ctx = bionic_fg::FramegenContext::create(
bionic_fg::vk::Device::create(),
prevBuf, currBuf, outBufs, extent, VK_FORMAT_R8G8B8A8_UNORM, cfg);
if (!ctx) { BFG_LOGE("nativeCreateContext: FramegenContext::create failed"); return 0L; }
return bionic_fg::ctxToHandle(ctx.release());
#else
(void)env; (void)prevAhb; (void)currAhb; (void)outputAhbs;
(void)width; (void)height; (void)multiplier; (void)flowScale; (void)model;
return 0L;
#endif
}
extern "C" JNIEXPORT void JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeDestroyContext(
JNIEnv*, jclass, jlong h) {
auto* ctx = bionic_fg::ctxFromHandle(h);
if (ctx) { ctx->destroy(); delete ctx; }
}
extern "C" JNIEXPORT jboolean JNICALL
Java_io_github_bionicfg_BionicFGNative_nativePresent(
JNIEnv* env, jclass, jlong h, jobject prevAhb, jobject currAhb) {
#ifdef __ANDROID__
auto* ctx = bionic_fg::ctxFromHandle(h);
if (!ctx) return JNI_FALSE;
auto* prev = static_cast<AHardwareBuffer*>(env->GetDirectBufferAddress(prevAhb));
auto* curr = static_cast<AHardwareBuffer*>(env->GetDirectBufferAddress(currAhb));
ctx->present(prev, curr);
return JNI_TRUE;
#else
(void)env; (void)h; (void)prevAhb; (void)currAhb;
return JNI_FALSE;
#endif
}
extern "C" JNIEXPORT jstring JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeDescribeContext(
JNIEnv* env, jclass, jlong h) {
auto* ctx = bionic_fg::ctxFromHandle(h);
if (!ctx) return bionic_fg::toJStr(env, "FramegenContext{null}");
return bionic_fg::toJStr(env, ctx->describe());
}
extern "C" JNIEXPORT void JNICALL
Java_io_github_bionicfg_BionicFGNative_nativeContextUpdateConfig(
JNIEnv*, jclass, jlong h, jint mult, jfloat flowScale, jint model) {
auto* ctx = bionic_fg::ctxFromHandle(h);
if (!ctx) return;
bionic_fg::Config cfg = ctx->valid()
? bionic_fg::Config{0, 0,
static_cast<uint32_t>(mult),
flowScale,
static_cast<uint32_t>(model)}
: bionic_fg::Config{};
ctx->updateConfig(cfg);
}
|
