android_media_ImageReader.cpp revision 0c79884076405bc36c0fb4f1bce27f883b97d64c
1212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He/* 2212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * Copyright 2013 The Android Open Source Project 3212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * 4212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * Licensed under the Apache License, Version 2.0 (the "License"); 5212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * you may not use this file except in compliance with the License. 6212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * You may obtain a copy of the License at 7212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * 8212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * http://www.apache.org/licenses/LICENSE-2.0 9212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * 10212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * Unless required by applicable law or agreed to in writing, software 11212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * distributed under the License is distributed on an "AS IS" BASIS, 12212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * See the License for the specific language governing permissions and 14212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He * limitations under the License. 15212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He */ 16212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 17212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He//#define LOG_NDEBUG 0 18212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#define LOG_TAG "ImageReader_JNI" 19212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <utils/Log.h> 20212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <utils/misc.h> 21212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <utils/List.h> 22ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown#include <utils/String8.h> 23212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 24212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <cstdio> 25212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 26212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <gui/CpuConsumer.h> 27212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <gui/Surface.h> 28534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He#include <camera3.h> 29212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 30212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <android_runtime/AndroidRuntime.h> 31212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <android_runtime/android_view_Surface.h> 32212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 33212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <jni.h> 34212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#include <JNIHelp.h> 35212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 365096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin#include <stdint.h> 375096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin#include <inttypes.h> 385096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin 39212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#define ALIGN(x, mask) ( ((x) + (mask) - 1) & ~((mask) - 1) ) 40212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 41212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#define ANDROID_MEDIA_IMAGEREADER_CTX_JNI_ID "mNativeContext" 42212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#define ANDROID_MEDIA_SURFACEIMAGE_BUFFER_JNI_ID "mLockedBuffer" 43212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He#define ANDROID_MEDIA_SURFACEIMAGE_TS_JNI_ID "mTimestamp" 44212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 45212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// ---------------------------------------------------------------------------- 46212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 47212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Heusing namespace android; 48212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 49212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Heenum { 50212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He IMAGE_READER_MAX_NUM_PLANES = 3, 51212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He}; 52212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 53e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkinenum { 54e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin ACQUIRE_SUCCESS = 0, 55e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin ACQUIRE_NO_BUFFERS = 1, 56e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin ACQUIRE_MAX_IMAGES = 2, 57e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin}; 58e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin 59ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brownstatic struct { 60ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown jfieldID mNativeContext; 61ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown jmethodID postEventFromNative; 62ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown} gImageReaderClassInfo; 63ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown 64ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brownstatic struct { 65ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown jfieldID mLockedBuffer; 66ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown jfieldID mTimestamp; 67ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown} gSurfaceImageClassInfo; 68212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 69ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brownstatic struct { 70212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jclass clazz; 71212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jmethodID ctor; 72ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown} gSurfacePlaneClassInfo; 73212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 74212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// ---------------------------------------------------------------------------- 75212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 76212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Heclass JNIImageReaderContext : public CpuConsumer::FrameAvailableListener 77212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 78212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hepublic: 79212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext(JNIEnv* env, jobject weakThiz, jclass clazz, int maxImages); 80212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 81212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He virtual ~JNIImageReaderContext(); 82212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 83212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He virtual void onFrameAvailable(); 84212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 85212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer* getLockedBuffer(); 86212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 87212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He void returnLockedBuffer(CpuConsumer::LockedBuffer* buffer); 88212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 8952a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian void setCpuConsumer(const sp<CpuConsumer>& consumer) { mConsumer = consumer; } 90212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer* getCpuConsumer() { return mConsumer.get(); } 91212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 925b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza void setProducer(const sp<IGraphicBufferProducer>& producer) { mProducer = producer; } 935b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza IGraphicBufferProducer* getProducer() { return mProducer.get(); } 94212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 95212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He void setBufferFormat(int format) { mFormat = format; } 96212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int getBufferFormat() { return mFormat; } 97212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 98212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He void setBufferWidth(int width) { mWidth = width; } 99212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int getBufferWidth() { return mWidth; } 100212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 101212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He void setBufferHeight(int height) { mHeight = height; } 102212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int getBufferHeight() { return mHeight; } 103212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 104212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Heprivate: 105212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He static JNIEnv* getJNIEnv(bool* needsDetach); 106212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He static void detachJNI(); 107212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 108212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He List<CpuConsumer::LockedBuffer*> mBuffers; 109212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He sp<CpuConsumer> mConsumer; 1105b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza sp<IGraphicBufferProducer> mProducer; 111212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jobject mWeakThiz; 112212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jclass mClazz; 113212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int mFormat; 114212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int mWidth; 115212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int mHeight; 116212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He}; 117212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 118212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun HeJNIImageReaderContext::JNIImageReaderContext(JNIEnv* env, 119212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jobject weakThiz, jclass clazz, int maxImages) : 120212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mWeakThiz(env->NewGlobalRef(weakThiz)), 121212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mClazz((jclass)env->NewGlobalRef(clazz)) { 122212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He for (int i = 0; i < maxImages; i++) { 123212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer *buffer = new CpuConsumer::LockedBuffer; 124212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mBuffers.push_back(buffer); 125212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 126212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 127212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 128212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun HeJNIEnv* JNIImageReaderContext::getJNIEnv(bool* needsDetach) { 129212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(needsDetach == NULL, "needsDetach is null!!!"); 130212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He *needsDetach = false; 131212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIEnv* env = AndroidRuntime::getJNIEnv(); 132212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (env == NULL) { 133212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JavaVMAttachArgs args = {JNI_VERSION_1_4, NULL, NULL}; 134212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JavaVM* vm = AndroidRuntime::getJavaVM(); 135212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int result = vm->AttachCurrentThread(&env, (void*) &args); 136212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (result != JNI_OK) { 137212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGE("thread attach failed: %#x", result); 138212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return NULL; 139212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 140212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He *needsDetach = true; 141212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 142212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return env; 143212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 144212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 145212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hevoid JNIImageReaderContext::detachJNI() { 146212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JavaVM* vm = AndroidRuntime::getJavaVM(); 147212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int result = vm->DetachCurrentThread(); 148212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (result != JNI_OK) { 149212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGE("thread detach failed: %#x", result); 150212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 151212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 152212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 153212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun HeCpuConsumer::LockedBuffer* JNIImageReaderContext::getLockedBuffer() { 154212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (mBuffers.empty()) { 155212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return NULL; 156212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 157212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Return a LockedBuffer pointer and remove it from the list 158212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He List<CpuConsumer::LockedBuffer*>::iterator it = mBuffers.begin(); 159212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer* buffer = *it; 160212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mBuffers.erase(it); 161212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return buffer; 162212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 163212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 164d901c033756b01f5bd3c697fb3802331e9b45ad0Igor Murashkinvoid JNIImageReaderContext::returnLockedBuffer(CpuConsumer::LockedBuffer* buffer) { 165212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mBuffers.push_back(buffer); 166212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 167212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 168212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun HeJNIImageReaderContext::~JNIImageReaderContext() { 169212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He bool needsDetach = false; 170212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIEnv* env = getJNIEnv(&needsDetach); 171212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (env != NULL) { 172212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He env->DeleteGlobalRef(mWeakThiz); 173212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He env->DeleteGlobalRef(mClazz); 174212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } else { 175212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGW("leaking JNI object references"); 176212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 177212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (needsDetach) { 178212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He detachJNI(); 179212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 180212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 181212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Delete LockedBuffers 182212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He for (List<CpuConsumer::LockedBuffer *>::iterator it = mBuffers.begin(); 183212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He it != mBuffers.end(); it++) { 184212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He delete *it; 185212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 186212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mBuffers.clear(); 187212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He mConsumer.clear(); 188212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 189212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 190212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hevoid JNIImageReaderContext::onFrameAvailable() 191212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 192212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: frame available", __FUNCTION__); 193212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He bool needsDetach = false; 194212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIEnv* env = getJNIEnv(&needsDetach); 195212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (env != NULL) { 196ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown env->CallStaticVoidMethod(mClazz, gImageReaderClassInfo.postEventFromNative, mWeakThiz); 197212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } else { 198212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGW("onFrameAvailable event will not posted"); 199212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 200212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (needsDetach) { 201212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He detachJNI(); 202212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 203212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 204212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 205212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// ---------------------------------------------------------------------------- 206212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 207212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Heextern "C" { 208212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 209212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic JNIImageReaderContext* ImageReader_getContext(JNIEnv* env, jobject thiz) 210212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 211212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext *ctx; 212212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx = reinterpret_cast<JNIImageReaderContext *> 213ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown (env->GetLongField(thiz, gImageReaderClassInfo.mNativeContext)); 214212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return ctx; 215212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 216212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 217212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic CpuConsumer* ImageReader_getCpuConsumer(JNIEnv* env, jobject thiz) 218212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 219212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s:", __FUNCTION__); 220212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext* const ctx = ImageReader_getContext(env, thiz); 221212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (ctx == NULL) { 222212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowRuntimeException(env, "ImageReaderContext is not initialized"); 223212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return NULL; 224212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 225212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return ctx->getCpuConsumer(); 226212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 227212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 2285b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stozastatic IGraphicBufferProducer* ImageReader_getProducer(JNIEnv* env, jobject thiz) 22952a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian{ 23052a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian ALOGV("%s:", __FUNCTION__); 23152a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian JNIImageReaderContext* const ctx = ImageReader_getContext(env, thiz); 23252a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian if (ctx == NULL) { 23352a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian jniThrowRuntimeException(env, "ImageReaderContext is not initialized"); 23452a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian return NULL; 23552a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian } 2365b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza return ctx->getProducer(); 23752a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian} 23852a9a10b6b8c7b7a9f97777541841b94d4fd9754Mathias Agopian 239212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void ImageReader_setNativeContext(JNIEnv* env, 240212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jobject thiz, sp<JNIImageReaderContext> ctx) 241212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 242212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s:", __FUNCTION__); 243212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext* const p = ImageReader_getContext(env, thiz); 244212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (ctx != 0) { 245212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx->incStrong((void*)ImageReader_setNativeContext); 246212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 247212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (p) { 248212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He p->decStrong((void*)ImageReader_setNativeContext); 249212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 250ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown env->SetLongField(thiz, gImageReaderClassInfo.mNativeContext, 251ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown reinterpret_cast<jlong>(ctx.get())); 252212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 253212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 254212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic CpuConsumer::LockedBuffer* Image_getLockedBuffer(JNIEnv* env, jobject image) 255212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 256ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown return reinterpret_cast<CpuConsumer::LockedBuffer*>( 257ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown env->GetLongField(image, gSurfaceImageClassInfo.mLockedBuffer)); 258212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 259212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 260212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void Image_setBuffer(JNIEnv* env, jobject thiz, 261212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He const CpuConsumer::LockedBuffer* buffer) 262212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 263ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown env->SetLongField(thiz, gSurfaceImageClassInfo.mLockedBuffer, reinterpret_cast<jlong>(buffer)); 264212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 265212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 266212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// Some formats like JPEG defined with different values between android.graphics.ImageFormat and 267212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// graphics.h, need convert to the one defined in graphics.h here. 268212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic int Image_getPixelFormat(JNIEnv* env, int format) 269212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 270d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He int jpegFormat; 271212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jfieldID fid; 272212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 273212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: format = 0x%x", __FUNCTION__, format); 274212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 275212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jclass imageFormatClazz = env->FindClass("android/graphics/ImageFormat"); 276212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(imageFormatClazz != NULL); 277212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 278212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He fid = env->GetStaticFieldID(imageFormatClazz, "JPEG", "I"); 279212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jpegFormat = env->GetStaticIntField(imageFormatClazz, fid); 280212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 281a5e1528f80f0472701dcbfc42d2f45ffc4a16466Zhijun He // Translate the JPEG to BLOB for camera purpose. 282212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (format == jpegFormat) { 283212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He format = HAL_PIXEL_FORMAT_BLOB; 284212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 285212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 286212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return format; 287212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 288212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 28931798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunkstatic uint32_t Image_getJpegSize(CpuConsumer::LockedBuffer* buffer, bool usingRGBAOverride) 290534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He{ 291534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He ALOG_ASSERT(buffer != NULL, "Input buffer is NULL!!!"); 292534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He uint32_t size = 0; 293534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He uint32_t width = buffer->width; 294534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He uint8_t* jpegBuffer = buffer->data; 295534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He 29631798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk if (usingRGBAOverride) { 2970c79884076405bc36c0fb4f1bce27f883b97d64cRuben Brunk width = (buffer->width + buffer->stride * (buffer->height - 1)) * 4; 29831798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk } 29931798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk 300534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He // First check for JPEG transport header at the end of the buffer 301534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He uint8_t* header = jpegBuffer + (width - sizeof(struct camera3_jpeg_blob)); 302534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He struct camera3_jpeg_blob *blob = (struct camera3_jpeg_blob*)(header); 303534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He if (blob->jpeg_blob_id == CAMERA3_JPEG_BLOB_ID) { 304534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He size = blob->jpeg_size; 305534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He ALOGV("%s: Jpeg size = %d", __FUNCTION__, size); 306534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He } 307534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He 308534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He // failed to find size, default to whole buffer 309534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He if (size == 0) { 3105096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin /* 3115096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin * This is a problem because not including the JPEG header 3125096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin * means that in certain rare situations a regular JPEG blob 3135096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin * will be misidentified as having a header, in which case 3145096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin * we will get a garbage size value. 3155096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin */ 3165096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin ALOGW("%s: No JPEG header detected, defaulting to size=width=%d", 3175096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin __FUNCTION__, width); 318534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He size = width; 319534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He } 320534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He 321534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He return size; 322534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He} 323534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He 32431798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunkstatic bool usingRGBAToJpegOverride(int32_t bufferFormat, int32_t readerCtxFormat) { 32531798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk return readerCtxFormat == HAL_PIXEL_FORMAT_BLOB && bufferFormat == HAL_PIXEL_FORMAT_RGBA_8888; 32631798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk} 32731798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk 3280fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunkstatic int32_t applyFormatOverrides(int32_t bufferFormat, int32_t readerCtxFormat) 3290fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk{ 3300fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk // Using HAL_PIXEL_FORMAT_RGBA_8888 gralloc buffers containing JPEGs to get around SW 3310fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk // write limitations for some platforms (b/17379185). 33231798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk if (usingRGBAToJpegOverride(bufferFormat, readerCtxFormat)) { 3330fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk return HAL_PIXEL_FORMAT_BLOB; 3340fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk } 3350fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk return bufferFormat; 3360fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk} 3370fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 338212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void Image_getLockedBufferInfo(JNIEnv* env, CpuConsumer::LockedBuffer* buffer, int idx, 3390fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk uint8_t **base, uint32_t *size, int32_t readerFormat) 340212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 341212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(buffer != NULL, "Input buffer is NULL!!!"); 342212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(base != NULL, "base is NULL!!!"); 343212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(size != NULL, "size is NULL!!!"); 344212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT((idx < IMAGE_READER_MAX_NUM_PLANES) && (idx >= 0)); 345212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 3467f4d3147d1851d2f0c544e45390c139bda9fd9aaZhijun He ALOGV("%s: buffer: %p", __FUNCTION__, buffer); 347212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 348212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He uint32_t dataSize, ySize, cSize, cStride; 349212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He uint8_t *cb, *cr; 350212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He uint8_t *pData = NULL; 351708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He int bytesPerPixel = 0; 352212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 353212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He dataSize = ySize = cSize = cStride = 0; 354212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int32_t fmt = buffer->format; 3550fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 35631798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk bool usingRGBAOverride = usingRGBAToJpegOverride(fmt, readerFormat); 3570fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk fmt = applyFormatOverrides(fmt, readerFormat); 358212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He switch (fmt) { 359212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YCbCr_420_888: 360212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = 361212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (idx == 0) ? 362212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He buffer->data : 363212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (idx == 1) ? 364212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He buffer->dataCb : 365212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He buffer->dataCr; 366212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (idx == 0) { 367212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He dataSize = buffer->stride * buffer->height; 368212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } else { 369212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He dataSize = buffer->chromaStride * buffer->height / 2; 370212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 371212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 372212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // NV21 373212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YCrCb_420_SP: 374212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cr = buffer->data + (buffer->stride * buffer->height); 375212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cb = cr + 1; 376212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ySize = buffer->width * buffer->height; 377212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cSize = buffer->width * buffer->height / 2; 378212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 379212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = 380212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (idx == 0) ? 381212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He buffer->data : 382212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (idx == 1) ? 383212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cb: 384212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cr; 385212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 386212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He dataSize = (idx == 0) ? ySize : cSize; 387212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 388212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YV12: 389212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Y and C stride need to be 16 pixel aligned. 390212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(buffer->stride % 16, 391212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Stride is not 16 pixel aligned %d", buffer->stride); 392212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 393212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ySize = buffer->stride * buffer->height; 394212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cStride = ALIGN(buffer->stride / 2, 16); 395212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cr = buffer->data + ySize; 396212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cSize = cStride * buffer->height / 2; 397212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cb = cr + cSize; 398212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 399212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = 400212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (idx == 0) ? 401212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He buffer->data : 402212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (idx == 1) ? 403212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cb : 404212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He cr; 405212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He dataSize = (idx == 0) ? ySize : cSize; 406212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 407212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_Y8: 408212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Single plane, 8bpp. 409212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 410212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 411212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = buffer->data; 412212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He dataSize = buffer->stride * buffer->height; 413212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 414212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_Y16: 415977ad8d26e701ed3902247852986d474e0e07f4dZhijun He bytesPerPixel = 2; 416212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Single plane, 16bpp, strides are specified in pixels, not in bytes 417212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 418212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 419212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = buffer->data; 420977ad8d26e701ed3902247852986d474e0e07f4dZhijun He dataSize = buffer->stride * buffer->height * bytesPerPixel; 421212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 422212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_BLOB: 423212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Used for JPEG data, height must be 1, width == size, single plane. 424212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 425212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(buffer->height == 1, "JPEG should has height value %d", buffer->height); 426212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 427212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = buffer->data; 42831798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk dataSize = Image_getJpegSize(buffer, usingRGBAOverride); 429212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 430212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_RAW_SENSOR: 431212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Single plane 16bpp bayer data. 432977ad8d26e701ed3902247852986d474e0e07f4dZhijun He bytesPerPixel = 2; 433212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 434212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pData = buffer->data; 435977ad8d26e701ed3902247852986d474e0e07f4dZhijun He dataSize = buffer->stride * buffer->height * bytesPerPixel; 436212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 437d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He case HAL_PIXEL_FORMAT_RAW10: 438d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He // Single plane 10bpp bayer data. 439d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 440d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He LOG_ALWAYS_FATAL_IF(buffer->width % 4, 441d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He "Width is not multiple of 4 %d", buffer->width); 442d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He LOG_ALWAYS_FATAL_IF(buffer->height % 2, 443d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He "Height is not even %d", buffer->height); 4444c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He LOG_ALWAYS_FATAL_IF(buffer->stride < (buffer->width * 10 / 8), 4454c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He "stride (%d) should be at least %d", 4464c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He buffer->stride, buffer->width * 10 / 8); 447d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He pData = buffer->data; 4484c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He dataSize = buffer->stride * buffer->height; 449d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He break; 450708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGBA_8888: 451708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGBX_8888: 452708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He // Single plane, 32bpp. 453708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He bytesPerPixel = 4; 454708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 455708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He pData = buffer->data; 456708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He dataSize = buffer->stride * buffer->height * bytesPerPixel; 457708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 458708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGB_565: 459708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He // Single plane, 16bpp. 460708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He bytesPerPixel = 2; 461708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 462708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He pData = buffer->data; 463708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He dataSize = buffer->stride * buffer->height * bytesPerPixel; 464708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 465708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGB_888: 466708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He // Single plane, 24bpp. 467708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He bytesPerPixel = 3; 468708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 469708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He pData = buffer->data; 470708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He dataSize = buffer->stride * buffer->height * bytesPerPixel; 471708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 472212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He default: 473212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowExceptionFmt(env, "java/lang/UnsupportedOperationException", 474212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Pixel format: 0x%x is unsupported", fmt); 475212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 476212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 477212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 478212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He *base = pData; 479212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He *size = dataSize; 480212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 481212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 4820fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunkstatic jint Image_imageGetPixelStride(JNIEnv* env, CpuConsumer::LockedBuffer* buffer, int idx, 4830fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk int32_t readerFormat) 484212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 485212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: buffer index: %d", __FUNCTION__, idx); 486212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT((idx < IMAGE_READER_MAX_NUM_PLANES) && (idx >= 0), "Index is out of range:%d", idx); 487212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 488212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int pixelStride = 0; 489212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(buffer != NULL, "buffer is NULL"); 490212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 491212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int32_t fmt = buffer->format; 4920fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 4930fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk fmt = applyFormatOverrides(fmt, readerFormat); 4940fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 495212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He switch (fmt) { 496212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YCbCr_420_888: 497212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pixelStride = (idx == 0) ? 1 : buffer->chromaStep; 498212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 499212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YCrCb_420_SP: 500212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pixelStride = (idx == 0) ? 1 : 2; 501212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 502212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_Y8: 503212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Single plane 8bpp data. 504212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 505212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pixelStride; 506212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 507212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YV12: 508212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pixelStride = 1; 509212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 510212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_BLOB: 511d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He case HAL_PIXEL_FORMAT_RAW10: 512d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He // Blob is used for JPEG data, RAW10 is used for 10-bit raw data, they are 513d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He // single plane, row and pixel strides are 0. 514212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 515212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pixelStride = 0; 516212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 517212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_Y16: 518212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_RAW_SENSOR: 519708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGB_565: 520212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Single plane 16bpp data. 521212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 522212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He pixelStride = 2; 523212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 524708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGBA_8888: 525708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGBX_8888: 526708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 527708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He pixelStride = 4; 528708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 529708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGB_888: 530708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He // Single plane, 24bpp. 531708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 532708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He pixelStride = 3; 533708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 534212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He default: 535212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowExceptionFmt(env, "java/lang/UnsupportedOperationException", 536212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Pixel format: 0x%x is unsupported", fmt); 537212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 538212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 539212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 540212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return pixelStride; 541212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 542212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 5430fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunkstatic jint Image_imageGetRowStride(JNIEnv* env, CpuConsumer::LockedBuffer* buffer, int idx, 5440fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk int32_t readerFormat) 545212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 546212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: buffer index: %d", __FUNCTION__, idx); 547212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT((idx < IMAGE_READER_MAX_NUM_PLANES) && (idx >= 0)); 548212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 549212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int rowStride = 0; 550212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(buffer != NULL, "buffer is NULL"); 551212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 552212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int32_t fmt = buffer->format; 553212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 5540fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk fmt = applyFormatOverrides(fmt, readerFormat); 5550fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 556212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He switch (fmt) { 557212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YCbCr_420_888: 558212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He rowStride = (idx == 0) ? buffer->stride : buffer->chromaStride; 559212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 560212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YCrCb_420_SP: 561212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He rowStride = buffer->width; 562212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 563212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_YV12: 564212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(buffer->stride % 16, 565212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Stride is not 16 pixel aligned %d", buffer->stride); 566212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He rowStride = (idx == 0) ? buffer->stride : ALIGN(buffer->stride / 2, 16); 567212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 568212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_BLOB: 569d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He // Blob is used for JPEG data, RAW10 is used for 10-bit raw data, they are 570d1988a98ed69db8c33b77b5c085ab91d22ef3bbcZhijun He // single plane, row and pixel strides are 0. 571212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 572212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He rowStride = 0; 573212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 5744c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He case HAL_PIXEL_FORMAT_RAW10: 5754c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 5764c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He rowStride = buffer->stride; 5774c4064fa66f551f49b7e1017b1ebe65a05f9df21Zhijun He break; 578212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_Y8: 579212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 580212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(buffer->stride % 16, 581212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Stride is not 16 pixel aligned %d", buffer->stride); 582212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He rowStride = buffer->stride; 583212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 584212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_Y16: 585212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He case HAL_PIXEL_FORMAT_RAW_SENSOR: 586212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // In native side, strides are specified in pixels, not in bytes. 587212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Single plane 16bpp bayer data. even width/height, 588212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // row stride multiple of 16 pixels (32 bytes) 589212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 590212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(buffer->stride % 16, 591212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Stride is not 16 pixel aligned %d", buffer->stride); 592212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He rowStride = buffer->stride * 2; 593212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 594708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGB_565: 595708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 596708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He rowStride = buffer->stride * 2; 597708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 598708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGBA_8888: 599708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGBX_8888: 600708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 601708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He rowStride = buffer->stride * 4; 602708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 603708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He case HAL_PIXEL_FORMAT_RGB_888: 604708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He // Single plane, 24bpp. 605708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He ALOG_ASSERT(idx == 0, "Wrong index: %d", idx); 606708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He rowStride = buffer->stride * 3; 607708e3595031fa15f4ac26c5675a53c1ed495b895Zhijun He break; 608212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He default: 609212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGE("%s Pixel format: 0x%x is unsupported", __FUNCTION__, fmt); 610212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowException(env, "java/lang/UnsupportedOperationException", 611212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "unsupported buffer format"); 612212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He break; 613212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 614212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 615212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return rowStride; 616212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 617212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 618212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// ---------------------------------------------------------------------------- 619212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 620212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void ImageReader_classInit(JNIEnv* env, jclass clazz) 621212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 622212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s:", __FUNCTION__); 623212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 624212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jclass imageClazz = env->FindClass("android/media/ImageReader$SurfaceImage"); 625212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(imageClazz == NULL, 626212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "can't find android/graphics/ImageReader$SurfaceImage"); 627ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gSurfaceImageClassInfo.mLockedBuffer = env->GetFieldID( 628ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown imageClazz, ANDROID_MEDIA_SURFACEIMAGE_BUFFER_JNI_ID, "J"); 629ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown LOG_ALWAYS_FATAL_IF(gSurfaceImageClassInfo.mLockedBuffer == NULL, 630212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "can't find android/graphics/ImageReader.%s", 631212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ANDROID_MEDIA_SURFACEIMAGE_BUFFER_JNI_ID); 632212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 633ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gSurfaceImageClassInfo.mTimestamp = env->GetFieldID( 634ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown imageClazz, ANDROID_MEDIA_SURFACEIMAGE_TS_JNI_ID, "J"); 635ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown LOG_ALWAYS_FATAL_IF(gSurfaceImageClassInfo.mTimestamp == NULL, 636212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "can't find android/graphics/ImageReader.%s", 637212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ANDROID_MEDIA_SURFACEIMAGE_TS_JNI_ID); 638212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 639ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gImageReaderClassInfo.mNativeContext = env->GetFieldID( 640ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown clazz, ANDROID_MEDIA_IMAGEREADER_CTX_JNI_ID, "J"); 641ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown LOG_ALWAYS_FATAL_IF(gImageReaderClassInfo.mNativeContext == NULL, 642212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "can't find android/graphics/ImageReader.%s", 643212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ANDROID_MEDIA_IMAGEREADER_CTX_JNI_ID); 644212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 645ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gImageReaderClassInfo.postEventFromNative = env->GetStaticMethodID( 646ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown clazz, "postEventFromNative", "(Ljava/lang/Object;)V"); 647ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown LOG_ALWAYS_FATAL_IF(gImageReaderClassInfo.postEventFromNative == NULL, 648212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "can't find android/graphics/ImageReader.postEventFromNative"); 649212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 650212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jclass planeClazz = env->FindClass("android/media/ImageReader$SurfaceImage$SurfacePlane"); 651212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He LOG_ALWAYS_FATAL_IF(planeClazz == NULL, "Can not find SurfacePlane class"); 652212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // FindClass only gives a local reference of jclass object. 653ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gSurfacePlaneClassInfo.clazz = (jclass) env->NewGlobalRef(planeClazz); 654ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gSurfacePlaneClassInfo.ctor = env->GetMethodID(gSurfacePlaneClassInfo.clazz, "<init>", 655ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown "(Landroid/media/ImageReader$SurfaceImage;III)V"); 656ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown LOG_ALWAYS_FATAL_IF(gSurfacePlaneClassInfo.ctor == NULL, 657ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown "Can not find SurfacePlane constructor"); 658212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 659212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 660212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void ImageReader_init(JNIEnv* env, jobject thiz, jobject weakThiz, 661212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jint width, jint height, jint format, jint maxImages) 662212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 663212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He status_t res; 664212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int nativeFormat; 665212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 666212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: width:%d, height: %d, format: 0x%x, maxImages:%d", 667212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He __FUNCTION__, width, height, format, maxImages); 668212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 669212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He nativeFormat = Image_getPixelFormat(env, format); 670212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 6715b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza sp<IGraphicBufferProducer> gbProducer; 6725b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza sp<IGraphicBufferConsumer> gbConsumer; 6735b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza BufferQueue::createBufferQueue(&gbProducer, &gbConsumer); 6745b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza sp<CpuConsumer> consumer = new CpuConsumer(gbConsumer, maxImages, 675dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin /*controlledByApp*/true); 676212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // TODO: throw dvm exOutOfMemoryError? 677212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (consumer == NULL) { 678212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowRuntimeException(env, "Failed to allocate native CpuConsumer"); 679212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return; 680212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 681212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 682212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jclass clazz = env->GetObjectClass(thiz); 683212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (clazz == NULL) { 684212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowRuntimeException(env, "Can't find android/graphics/ImageReader"); 685212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return; 686212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 687212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He sp<JNIImageReaderContext> ctx(new JNIImageReaderContext(env, weakThiz, clazz, maxImages)); 688212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx->setCpuConsumer(consumer); 6895b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza ctx->setProducer(gbProducer); 690212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He consumer->setFrameAvailableListener(ctx); 691212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ImageReader_setNativeContext(env, thiz, ctx); 692212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx->setBufferFormat(nativeFormat); 693212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx->setBufferWidth(width); 694212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx->setBufferHeight(height); 695212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 696212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Set the width/height/format to the CpuConsumer 697212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He res = consumer->setDefaultBufferSize(width, height); 698212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (res != OK) { 699212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowException(env, "java/lang/IllegalStateException", 700212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Failed to set CpuConsumer buffer size"); 701212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return; 702212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 703212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He res = consumer->setDefaultBufferFormat(nativeFormat); 704212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (res != OK) { 705212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowException(env, "java/lang/IllegalStateException", 706212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "Failed to set CpuConsumer buffer format"); 707212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 708212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 709212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 710212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void ImageReader_close(JNIEnv* env, jobject thiz) 711212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 712212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s:", __FUNCTION__); 713212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 714212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext* const ctx = ImageReader_getContext(env, thiz); 715212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (ctx == NULL) { 716212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // ImageReader is already closed. 717212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return; 718212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 719212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 720212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer* consumer = ImageReader_getCpuConsumer(env, thiz); 721212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (consumer != NULL) { 722212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He consumer->abandon(); 723212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He consumer->setFrameAvailableListener(NULL); 724212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 725212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ImageReader_setNativeContext(env, thiz, NULL); 726212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 727212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 728212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic void ImageReader_imageRelease(JNIEnv* env, jobject thiz, jobject image) 729212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 730212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s:", __FUNCTION__); 731212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext* ctx = ImageReader_getContext(env, thiz); 732212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (ctx == NULL) { 733212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGW("ImageReader#close called before Image#close, consider calling Image#close first"); 734212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return; 735212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 736212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 737212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer* consumer = ctx->getCpuConsumer(); 738212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer* buffer = Image_getLockedBuffer(env, image); 739212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (!buffer) { 740212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGW("Image already released!!!"); 741212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return; 742212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 743212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He consumer->unlockBuffer(*buffer); 744212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He Image_setBuffer(env, image, NULL); 745212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ctx->returnLockedBuffer(buffer); 746212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 747212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 748e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkinstatic jint ImageReader_imageSetup(JNIEnv* env, jobject thiz, 749212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jobject image) 750212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 751212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s:", __FUNCTION__); 752212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He JNIImageReaderContext* ctx = ImageReader_getContext(env, thiz); 753212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (ctx == NULL) { 754212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowRuntimeException(env, "ImageReaderContext is not initialized"); 755e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return -1; 756212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 757212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 758212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer* consumer = ctx->getCpuConsumer(); 759212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer* buffer = ctx->getLockedBuffer(); 760212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (buffer == NULL) { 761d901c033756b01f5bd3c697fb3802331e9b45ad0Igor Murashkin ALOGW("Unable to acquire a lockedBuffer, very likely client tries to lock more than" 762d901c033756b01f5bd3c697fb3802331e9b45ad0Igor Murashkin " maxImages buffers"); 763e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return ACQUIRE_MAX_IMAGES; 764212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 765212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He status_t res = consumer->lockNextBuffer(buffer); 766212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (res != NO_ERROR) { 767334778981dc01b5f8f4e137eb3f78745e214cf4dlina.x.pi ctx->returnLockedBuffer(buffer); 768f724c277d3362dbc8099fcbf8674609a424cd2eeJeff Brown if (res != BAD_VALUE /*no buffers*/) { 769dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin if (res == NOT_ENOUGH_DATA) { 770e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return ACQUIRE_MAX_IMAGES; 771dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin } else { 772dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin ALOGE("%s Fail to lockNextBuffer with error: %d ", 773dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin __FUNCTION__, res); 774e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin jniThrowExceptionFmt(env, "java/lang/AssertionError", 775dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin "Unknown error (%d) when we tried to lock buffer.", 776dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin res); 777dd0643202de80cc4ced37d1844e722c8a5e89154Igor Murashkin } 778f724c277d3362dbc8099fcbf8674609a424cd2eeJeff Brown } 779e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return ACQUIRE_NO_BUFFERS; 780212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 781212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 7824eda9f5359347c11914e47f477535c9533674d32Zhijun He if (buffer->format == HAL_PIXEL_FORMAT_YCrCb_420_SP) { 7834eda9f5359347c11914e47f477535c9533674d32Zhijun He jniThrowException(env, "java/lang/UnsupportedOperationException", 7844eda9f5359347c11914e47f477535c9533674d32Zhijun He "NV21 format is not supported by ImageReader"); 7854eda9f5359347c11914e47f477535c9533674d32Zhijun He return -1; 7864eda9f5359347c11914e47f477535c9533674d32Zhijun He } 7874eda9f5359347c11914e47f477535c9533674d32Zhijun He 788212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Check if the left-top corner of the crop rect is origin, we currently assume this point is 789212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // zero, will revist this once this assumption turns out problematic. 790212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He Point lt = buffer->crop.leftTop(); 791212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (lt.x != 0 || lt.y != 0) { 7924eda9f5359347c11914e47f477535c9533674d32Zhijun He jniThrowExceptionFmt(env, "java/lang/UnsupportedOperationException", 7934eda9f5359347c11914e47f477535c9533674d32Zhijun He "crop left top corner [%d, %d] need to be at origin", lt.x, lt.y); 794e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return -1; 795212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 796212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 797212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Check if the producer buffer configurations match what ImageReader configured. 798212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // We want to fail for the very first image because this case is too bad. 799534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He int outputWidth = buffer->width; 800534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He int outputHeight = buffer->height; 801534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He 80237682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He // Correct width/height when crop is set. 8039e6d073a999d7934aa3f22a5877c6e8e2ce15766Zhijun He if (!buffer->crop.isEmpty()) { 804cfa553369fd8ef68db751fe1052cd17fd5763965Zhijun He outputWidth = buffer->crop.getWidth(); 805cfa553369fd8ef68db751fe1052cd17fd5763965Zhijun He outputHeight = buffer->crop.getHeight(); 806534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He } 807534046d2b12fd13776ad782b982649cb0bea9b79Zhijun He 8080fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk int imgReaderFmt = ctx->getBufferFormat(); 809212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int imageReaderWidth = ctx->getBufferWidth(); 810212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int imageReaderHeight = ctx->getBufferHeight(); 8110fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk if ((buffer->format != HAL_PIXEL_FORMAT_BLOB) && (imgReaderFmt != HAL_PIXEL_FORMAT_BLOB) && 81237682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He (imageReaderWidth != outputWidth || imageReaderHeight > outputHeight)) { 81337682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He /** 81437682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He * For video decoder, the buffer height is actually the vertical stride, 81537682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He * which is always >= actual image height. For future, decoder need provide 81637682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He * right crop rectangle to CpuConsumer to indicate the actual image height, 81737682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He * see bug 9563986. After this bug is fixed, we can enforce the height equal 81837682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He * check. Right now, only make sure buffer height is no less than ImageReader 81937682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He * height. 82037682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He */ 82137682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He jniThrowExceptionFmt(env, "java/lang/IllegalStateException", 82237682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He "Producer buffer size: %dx%d, doesn't match ImageReader configured size: %dx%d", 82337682135da2fd90e7bc6a89a418862d1f4ca15fdZhijun He outputWidth, outputHeight, imageReaderWidth, imageReaderHeight); 824e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return -1; 825212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 826212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 827feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk int bufFmt = buffer->format; 828feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk if (imgReaderFmt != bufFmt) { 8290fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 83091b9aabc9fa0c058ecc4a8b3f486540c28fe1cc0Ruben Brunk if (imgReaderFmt == HAL_PIXEL_FORMAT_YCbCr_420_888 && (bufFmt == 83191b9aabc9fa0c058ecc4a8b3f486540c28fe1cc0Ruben Brunk HAL_PIXEL_FORMAT_YCrCb_420_SP || bufFmt == HAL_PIXEL_FORMAT_YV12)) { 8320fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk // Special casing for when producer switches to a format compatible with flexible YUV 8330fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk // (HAL_PIXEL_FORMAT_YCbCr_420_888). 83491b9aabc9fa0c058ecc4a8b3f486540c28fe1cc0Ruben Brunk ctx->setBufferFormat(bufFmt); 8350fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk ALOGD("%s: Overriding buffer format YUV_420_888 to %x.", __FUNCTION__, bufFmt); 8360fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk } else if (imgReaderFmt == HAL_PIXEL_FORMAT_BLOB && bufFmt == HAL_PIXEL_FORMAT_RGBA_8888) { 8370fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk // Using HAL_PIXEL_FORMAT_RGBA_8888 gralloc buffers containing JPEGs to get around SW 8380fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk // write limitations for (b/17379185). 8390fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk ALOGD("%s: Receiving JPEG in HAL_PIXEL_FORMAT_RGBA_8888 buffer.", __FUNCTION__); 840feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk } else { 841feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk // Return the buffer to the queue. 842feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk consumer->unlockBuffer(*buffer); 843feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk ctx->returnLockedBuffer(buffer); 844feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk 845feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk // Throw exception 846feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk ALOGE("Producer output buffer format: 0x%x, ImageReader configured format: 0x%x", 847feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk buffer->format, ctx->getBufferFormat()); 848feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk String8 msg; 849feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk msg.appendFormat("The producer output buffer format 0x%x doesn't " 850feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk "match the ImageReader's configured buffer format 0x%x.", 851feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk buffer->format, ctx->getBufferFormat()); 852feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk jniThrowException(env, "java/lang/UnsupportedOperationException", 853feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk msg.string()); 854feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk return -1; 855feb50af361e4305a25758966b6b5df2738c00259Ruben Brunk } 856212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 857212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Set SurfaceImage instance member variables 858212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He Image_setBuffer(env, image, buffer); 859ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown env->SetLongField(image, gSurfaceImageClassInfo.mTimestamp, 860ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown static_cast<jlong>(buffer->timestamp)); 861212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 862e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin return ACQUIRE_SUCCESS; 863212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 864212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 865212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic jobject ImageReader_getSurface(JNIEnv* env, jobject thiz) 866212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 867212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: ", __FUNCTION__); 868212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 8695b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza IGraphicBufferProducer* gbp = ImageReader_getProducer(env, thiz); 8705b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza if (gbp == NULL) { 871212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowRuntimeException(env, "CpuConsumer is uninitialized"); 872212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return NULL; 873212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 874212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 875212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Wrap the IGBP in a Java-language Surface. 8765b3c7c1e575a7fe1bc765e5d14690e1bfd971ffbDan Stoza return android_view_Surface_createFromIGraphicBufferProducer(env, gbp); 877212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 878212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 8790fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunkstatic jobject Image_createSurfacePlane(JNIEnv* env, jobject thiz, int idx, int readerFormat) 880212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 881212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int rowStride, pixelStride; 882212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: buffer index: %d", __FUNCTION__, idx); 883212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 884212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer* buffer = Image_getLockedBuffer(env, thiz); 885212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 886212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOG_ASSERT(buffer != NULL); 887212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (buffer == NULL) { 888212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowException(env, "java/lang/IllegalStateException", "Image was released"); 889212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 8900fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 8910fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk readerFormat = Image_getPixelFormat(env, readerFormat); 8920fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 8930fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk rowStride = Image_imageGetRowStride(env, buffer, idx, readerFormat); 8940fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk pixelStride = Image_imageGetPixelStride(env, buffer, idx, readerFormat); 895212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 896ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown jobject surfPlaneObj = env->NewObject(gSurfacePlaneClassInfo.clazz, 897ef961215599b1c154130d4e64e46a401d6bfef67Jeff Brown gSurfacePlaneClassInfo.ctor, thiz, idx, rowStride, pixelStride); 898212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 899212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return surfPlaneObj; 900212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 901212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 9020fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunkstatic jobject Image_getByteBuffer(JNIEnv* env, jobject thiz, int idx, int readerFormat) 903212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He{ 904212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He uint8_t *base = NULL; 905212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He uint32_t size = 0; 906212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jobject byteBuffer; 907212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 908212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He ALOGV("%s: buffer index: %d", __FUNCTION__, idx); 909212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 910212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He CpuConsumer::LockedBuffer* buffer = Image_getLockedBuffer(env, thiz); 911212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 912212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if (buffer == NULL) { 913212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowException(env, "java/lang/IllegalStateException", "Image was released"); 914212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 915212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 9160fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk readerFormat = Image_getPixelFormat(env, readerFormat); 9170fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk 918212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // Create byteBuffer from native buffer 9190fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk Image_getLockedBufferInfo(env, buffer, idx, &base, &size, readerFormat); 9205096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin 9215096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin if (size > static_cast<uint32_t>(INT32_MAX)) { 9225096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin // Byte buffer have 'int capacity', so check the range 9235096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin jniThrowExceptionFmt(env, "java/lang/IllegalStateException", 92431798f33184fd59dd3d3cc55e6373d9f91d220b6Ruben Brunk "Size too large for bytebuffer capacity %" PRIu32, size); 9255096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin return NULL; 9265096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin } 9275096defdaa4716ce81047a855d6e5ce3f8263600Igor Murashkin 928212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He byteBuffer = env->NewDirectByteBuffer(base, size); 929212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He // TODO: throw dvm exOutOfMemoryError? 930212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He if ((byteBuffer == NULL) && (env->ExceptionCheck() == false)) { 931212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He jniThrowException(env, "java/lang/IllegalStateException", "Failed to allocate ByteBuffer"); 932212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He } 933212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 934212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return byteBuffer; 935212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 936212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 937212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} // extern "C" 938212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 939212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He// ---------------------------------------------------------------------------- 940212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 941212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic JNINativeMethod gImageReaderMethods[] = { 942212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He {"nativeClassInit", "()V", (void*)ImageReader_classInit }, 943212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He {"nativeInit", "(Ljava/lang/Object;IIII)V", (void*)ImageReader_init }, 944212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He {"nativeClose", "()V", (void*)ImageReader_close }, 945212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He {"nativeReleaseImage", "(Landroid/media/Image;)V", (void*)ImageReader_imageRelease }, 946e3351f1942bfe86682389b278e7ff128a72ea671Igor Murashkin {"nativeImageSetup", "(Landroid/media/Image;)I", (void*)ImageReader_imageSetup }, 947212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He {"nativeGetSurface", "()Landroid/view/Surface;", (void*)ImageReader_getSurface }, 948212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He}; 949212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 950212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Hestatic JNINativeMethod gImageMethods[] = { 9510fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk {"nativeImageGetBuffer", "(II)Ljava/nio/ByteBuffer;", (void*)Image_getByteBuffer }, 9520fd198ad89ec9c600bb1761b10d938146c28bb98Ruben Brunk {"nativeCreatePlane", "(II)Landroid/media/ImageReader$SurfaceImage$SurfacePlane;", 953212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He (void*)Image_createSurfacePlane }, 954212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He}; 955212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 956212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun Heint register_android_media_ImageReader(JNIEnv *env) { 957212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 958212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int ret1 = AndroidRuntime::registerNativeMethods(env, 959212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "android/media/ImageReader", gImageReaderMethods, NELEM(gImageReaderMethods)); 960212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 961212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He int ret2 = AndroidRuntime::registerNativeMethods(env, 962212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He "android/media/ImageReader$SurfaceImage", gImageMethods, NELEM(gImageMethods)); 963212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He 964212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He return (ret1 || ret2); 965212e78df9eb3bfff069de01aa7820cf4201c5f82Zhijun He} 966