1 /*
2  * Copyright (C) 2010 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #undef LOG_TAG
18 #define LOG_TAG "HardwareBufferRenderer"
19 #define ATRACE_TAG ATRACE_TAG_VIEW
20 
21 #include <GraphicsJNI.h>
22 #include <RootRenderNode.h>
23 #include <TreeInfo.h>
24 #include <android-base/unique_fd.h>
25 #include <android/native_window.h>
26 #include <nativehelper/JNIPlatformHelp.h>
27 #include <renderthread/CanvasContext.h>
28 #include <renderthread/RenderProxy.h>
29 #include <renderthread/RenderThread.h>
30 
31 #include "HardwareBufferHelpers.h"
32 #include "JvmErrorReporter.h"
33 
34 namespace android {
35 
36 using namespace android::uirenderer;
37 using namespace android::uirenderer::renderthread;
38 
39 struct {
40     jclass clazz;
41     jmethodID invokeRenderCallback;
42 } gHardwareBufferRendererClassInfo;
43 
createRenderCallback(JNIEnv * env,jobject releaseCallback)44 static RenderCallback createRenderCallback(JNIEnv* env, jobject releaseCallback) {
45     if (releaseCallback == nullptr) return nullptr;
46 
47     JavaVM* vm = nullptr;
48     LOG_ALWAYS_FATAL_IF(env->GetJavaVM(&vm) != JNI_OK, "Unable to get Java VM");
49     auto globalCallbackRef =
50             std::make_shared<JGlobalRefHolder>(vm, env->NewGlobalRef(releaseCallback));
51     return [globalCallbackRef](android::base::unique_fd&& fd, int status) {
52         globalCallbackRef->env()->CallStaticVoidMethod(
53                 gHardwareBufferRendererClassInfo.clazz,
54                 gHardwareBufferRendererClassInfo.invokeRenderCallback, globalCallbackRef->object(),
55                 reinterpret_cast<jint>(fd.release()), reinterpret_cast<jint>(status));
56     };
57 }
58 
android_graphics_HardwareBufferRenderer_createRootNode(JNIEnv * env,jobject)59 static long android_graphics_HardwareBufferRenderer_createRootNode(JNIEnv* env, jobject) {
60     auto* node = new RootRenderNode(std::make_unique<JvmErrorReporter>(env));
61     node->incStrong(nullptr);
62     node->setName("RootRenderNode");
63     return reinterpret_cast<jlong>(node);
64 }
65 
android_graphics_hardwareBufferRenderer_destroyRootNode(JNIEnv *,jobject,jlong renderNodePtr)66 static void android_graphics_hardwareBufferRenderer_destroyRootNode(JNIEnv*, jobject,
67                                                                     jlong renderNodePtr) {
68     auto* node = reinterpret_cast<RootRenderNode*>(renderNodePtr);
69     node->destroy();
70 }
71 
android_graphics_HardwareBufferRenderer_create(JNIEnv * env,jobject,jobject buffer,jlong renderNodePtr)72 static long android_graphics_HardwareBufferRenderer_create(JNIEnv* env, jobject, jobject buffer,
73                                                            jlong renderNodePtr) {
74     auto* hardwareBuffer = HardwareBufferHelpers::AHardwareBuffer_fromHardwareBuffer(env, buffer);
75     auto* rootRenderNode = reinterpret_cast<RootRenderNode*>(renderNodePtr);
76     ContextFactoryImpl factory(rootRenderNode);
77     auto* proxy = new RenderProxy(false, rootRenderNode, &factory);
78     proxy->setHardwareBuffer(hardwareBuffer);
79     return (jlong)proxy;
80 }
81 
HardwareBufferRenderer_destroy(jlong renderProxy)82 static void HardwareBufferRenderer_destroy(jlong renderProxy) {
83     auto* proxy = reinterpret_cast<RenderProxy*>(renderProxy);
84     delete proxy;
85 }
86 
createMatrixFromBufferTransform(SkScalar width,SkScalar height,int transform)87 static SkMatrix createMatrixFromBufferTransform(SkScalar width, SkScalar height, int transform) {
88     switch (transform) {
89         case ANATIVEWINDOW_TRANSFORM_ROTATE_90:
90             return SkMatrix::MakeAll(0, -1, height, 1, 0, 0, 0, 0, 1);
91         case ANATIVEWINDOW_TRANSFORM_ROTATE_180:
92             return SkMatrix::MakeAll(-1, 0, width, 0, -1, height, 0, 0, 1);
93         case ANATIVEWINDOW_TRANSFORM_ROTATE_270:
94             return SkMatrix::MakeAll(0, 1, 0, -1, 0, width, 0, 0, 1);
95         default:
96             ALOGE("Invalid transform provided. Transform should be validated from"
97                   "the java side. Leveraging identity transform as a fallback");
98             [[fallthrough]];
99         case ANATIVEWINDOW_TRANSFORM_IDENTITY:
100             return SkMatrix::I();
101     }
102 }
103 
android_graphics_HardwareBufferRenderer_render(JNIEnv * env,jobject,jlong renderProxy,jint transform,jint width,jint height,jlong colorspacePtr,jobject consumer)104 static int android_graphics_HardwareBufferRenderer_render(JNIEnv* env, jobject, jlong renderProxy,
105                                                           jint transform, jint width, jint height,
106                                                           jlong colorspacePtr, jobject consumer) {
107     auto* proxy = reinterpret_cast<RenderProxy*>(renderProxy);
108     auto skWidth = static_cast<SkScalar>(width);
109     auto skHeight = static_cast<SkScalar>(height);
110     auto matrix = createMatrixFromBufferTransform(skWidth, skHeight, transform);
111     auto colorSpace = GraphicsJNI::getNativeColorSpace(colorspacePtr);
112     proxy->setHardwareBufferRenderParams(HardwareBufferRenderParams(
113             width, height, matrix, colorSpace, createRenderCallback(env, consumer)));
114     nsecs_t vsync = systemTime(SYSTEM_TIME_MONOTONIC);
115     UiFrameInfoBuilder(proxy->frameInfo())
116                 .setVsync(vsync, vsync, UiFrameInfoBuilder::INVALID_VSYNC_ID,
117                     UiFrameInfoBuilder::UNKNOWN_DEADLINE,
118                     UiFrameInfoBuilder::UNKNOWN_FRAME_INTERVAL)
119                 .addFlag(FrameInfoFlags::SurfaceCanvas);
120     return proxy->syncAndDrawFrame();
121 }
122 
android_graphics_HardwareBufferRenderer_setLightGeometry(JNIEnv *,jobject,jlong renderProxyPtr,jfloat lightX,jfloat lightY,jfloat lightZ,jfloat lightRadius)123 static void android_graphics_HardwareBufferRenderer_setLightGeometry(JNIEnv*, jobject,
124                                                                      jlong renderProxyPtr,
125                                                                      jfloat lightX, jfloat lightY,
126                                                                      jfloat lightZ,
127                                                                      jfloat lightRadius) {
128     auto* proxy = reinterpret_cast<RenderProxy*>(renderProxyPtr);
129     proxy->setLightGeometry((Vector3){lightX, lightY, lightZ}, lightRadius);
130 }
131 
android_graphics_HardwareBufferRenderer_setLightAlpha(JNIEnv * env,jobject,jlong renderProxyPtr,jfloat ambientShadowAlpha,jfloat spotShadowAlpha)132 static void android_graphics_HardwareBufferRenderer_setLightAlpha(JNIEnv* env, jobject,
133                                                                   jlong renderProxyPtr,
134                                                                   jfloat ambientShadowAlpha,
135                                                                   jfloat spotShadowAlpha) {
136     auto* proxy = reinterpret_cast<RenderProxy*>(renderProxyPtr);
137     proxy->setLightAlpha((uint8_t)(255 * ambientShadowAlpha), (uint8_t)(255 * spotShadowAlpha));
138 }
139 
android_graphics_HardwareBufferRenderer_getFinalizer()140 static jlong android_graphics_HardwareBufferRenderer_getFinalizer() {
141     return static_cast<jlong>(reinterpret_cast<uintptr_t>(&HardwareBufferRenderer_destroy));
142 }
143 
144 // ----------------------------------------------------------------------------
145 // JNI Glue
146 // ----------------------------------------------------------------------------
147 
148 const char* const kClassPathName = "android/graphics/HardwareBufferRenderer";
149 
150 static const JNINativeMethod gMethods[] = {
151         {"nCreateHardwareBufferRenderer", "(Landroid/hardware/HardwareBuffer;J)J",
152          (void*)android_graphics_HardwareBufferRenderer_create},
153         {"nRender", "(JIIIJLjava/util/function/Consumer;)I",
154          (void*)android_graphics_HardwareBufferRenderer_render},
155         {"nCreateRootRenderNode", "()J",
156          (void*)android_graphics_HardwareBufferRenderer_createRootNode},
157         {"nSetLightGeometry", "(JFFFF)V",
158          (void*)android_graphics_HardwareBufferRenderer_setLightGeometry},
159         {"nSetLightAlpha", "(JFF)V", (void*)android_graphics_HardwareBufferRenderer_setLightAlpha},
160         {"nGetFinalizer", "()J", (void*)android_graphics_HardwareBufferRenderer_getFinalizer},
161         {"nDestroyRootRenderNode", "(J)V",
162          (void*)android_graphics_hardwareBufferRenderer_destroyRootNode}};
163 
register_android_graphics_HardwareBufferRenderer(JNIEnv * env)164 int register_android_graphics_HardwareBufferRenderer(JNIEnv* env) {
165     jclass hardwareBufferRendererClazz =
166             FindClassOrDie(env, "android/graphics/HardwareBufferRenderer");
167     gHardwareBufferRendererClassInfo.clazz =
168             reinterpret_cast<jclass>(env->NewGlobalRef(hardwareBufferRendererClazz));
169     gHardwareBufferRendererClassInfo.invokeRenderCallback =
170             GetStaticMethodIDOrDie(env, hardwareBufferRendererClazz, "invokeRenderCallback",
171                                    "(Ljava/util/function/Consumer;II)V");
172     HardwareBufferHelpers::init();
173     return RegisterMethodsOrDie(env, kClassPathName, gMethods, NELEM(gMethods));
174 }
175 
176 }  // namespace android