| | |
| | | #include "RtspNativeCodecJNI.h" |
| | | #include <PipeLine.h> |
| | | #include <PL_RTSPClient.h> |
| | | #include <PL_AndroidMediaCodecDecoder.h> |
| | | #include <PL_AndroidSurfaceViewSink.h> |
| | | #include "CameraWrapper.h" |
| | | #include <logger.h> |
| | | #include <Logger/src/logger.hpp> |
| | | #include <logger.h> |
| | | |
| | | #include <jni.h> |
| | | #include "looper.h" |
| | | #include "media/NdkMediaCodec.h" |
| | | #include "media/NdkMediaExtractor.h" |
| | | //#include "looper.h" |
| | | #include <android/native_window_jni.h> |
| | | #include <media/NdkMediaCodec.h> |
| | | #include <media/NdkMediaExtractor.h> |
| | | |
| | | #include <pthread.h> |
| | | |
| | | #include <ev_proto.h> |
| | | #include <face_daemon_proto.h> |
| | | |
| | | #include <stdlib.h> |
| | | |
| | | #include "DebugNetwork.h" |
| | | |
| | | //#include <mediastreamer2/include/mediastreamer2/msjava.h> |
| | | |
| | | std::stringstream logss; |
| | | //std::fstream logss("/storage/sdcard/log.log", std::fstream::out); |
| | | //std::fstream logss("/storage/emulated/0/log.log", std::fstream::out); |
| | | Logger g_logger(logss); |
| | | |
| | | class CameraWrapper |
| | | { |
| | | PipeLine* pipeLine; |
| | | |
| | | PL_RTSPClient_Config rtspConfig; |
| | | PL_AndroidMediaCodecDecoder_Config amcdConfig; |
| | | PL_AndroidSurfaceViewSink asvsConfig; |
| | | |
| | | jmethodID faceCallback; |
| | | |
| | | CameraWrapper() : pipeLine(nullptr), rtspConfig(), amcdConfig(), asvsConfig(), faceCallback(0) |
| | | { |
| | | } |
| | | |
| | | ~CameraWrapper() |
| | | { |
| | | delete pipeLine; |
| | | } |
| | | |
| | | bool start() |
| | | { |
| | | } |
| | | |
| | | void stop() |
| | | { |
| | | } |
| | | }; |
| | | |
| | | CameraWrapper g_CameraWrappers[CAMERA_COUNT]; |
| | | |
| | | extern "C" |
| | | { |
| | | |
| | | void Java_RtspNativeCodec_init(JNIEnv *env, jclass clazz) |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_init(JNIEnv *env, jclass clazz) |
| | | { |
| | | g_logger.set_level(VERBOSE); |
| | | |
| | | PipeLine::register_global_elem_creator("PL_RTSPClient", create_PL_RTSPClient); |
| | | PipeLine::register_global_elem_creator("PL_AndroidMediaCodecDecoder", create_PL_AndroidMediaCodecDecoder); |
| | | PipeLine::register_global_elem_creator("PL_AndroidSurfaceViewSink", create_PL_AndroidSurfaceViewSink); |
| | | g_logger.set_level(INFO); |
| | | |
| | | #ifdef USE_ST_SDK |
| | | PipeLine::register_global_elem_creator("PL_SensetimeFaceTrack", create_PL_SensetimeFaceTrack); |
| | | #endif |
| | | |
| | | for (size_t i = 0; i < CAMERA_COUNT; i++) |
| | | { |
| | | g_CameraWrappers[i].cameraIdx = i + 1; |
| | | |
| | | g_CameraWrappers[i].pipeLine = new PipeLine; |
| | | PipeLine& pipeLine(*(g_CameraWrappers[i].pipeLine)); |
| | | } |
| | | } |
| | | |
| | | // set the surface |
| | | void Java_RtspNativeCodec_setSurface(JNIEnv *env, jclass clazz, jint cameraIdx, jobject surface) |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_setLocalIP(JNIEnv *env, jclass clazz, jstring ipaddr) |
| | | { |
| | | LOGV("@@@ Java_RtspNativeCodec_setSurface"); |
| | | assert(cameraIdx <= CAMERA_COUNT); |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_setLocalIP" << LOG_ENDL; |
| | | |
| | | // obtain a native window from a Java surface |
| | | if (asvsConfig.window) |
| | | { |
| | | ANativeWindow_release((ANativeWindow)(asvsConfig.window)); |
| | | asvsConfig.window = NULL; |
| | | const char *utf8 = env->GetStringUTFChars(ipaddr, NULL); |
| | | for (size_t i = 0; i < CAMERA_COUNT; i++) |
| | | { |
| | | g_CameraWrappers[i].rtspConfig.receivingInterfaceAddr = utf8; |
| | | g_CameraWrappers[i].rtspConfig.desiredPortNum = 10056 + i * 4; |
| | | g_CameraWrappers[i].rtspConfig.requestStreamingOverTcp = true; |
| | | } |
| | | asvsConfig.window = ANativeWindow_fromSurface(env, surface); |
| | | LOGV("@@@ setsurface %p", asvsConfig.window); |
| | | env->ReleaseStringUTFChars(ipaddr, utf8); |
| | | } |
| | | |
| | | jboolean Java_RtspNativeCodec_createPlayer(JNIEnv* env, jclass clazz, jobject assetMgr, jint cameraIdx, jstring uri) |
| | | static jclass _jcOutputSurface = nullptr; |
| | | static jmethodID _jmOutputSurface_ctor = nullptr; |
| | | static jmethodID _jmOutputSurface_getSurface = nullptr; |
| | | static jmethodID _jmOutputSurface_readPixels = nullptr; |
| | | static jmethodID _jmOutputSurface_awaitNewImage = nullptr; |
| | | static jmethodID _jmOutputSurface_drawImage = nullptr; |
| | | static jmethodID _jmOutputSurface_makeCurrent = nullptr; |
| | | |
| | | static jobject _jobjOutputSurface = nullptr; |
| | | static jobject _jobjSurface = nullptr; |
| | | static JNIEnv *_env = nullptr; |
| | | |
| | | void readPixels(void* surface, uint8_t* buf, size_t& maxSize) |
| | | { |
| | | LOGV("@@@ Java_RtspNativeCodec_createPlayer"); |
| | | assert(cameraIdx <= CAMERA_COUNT); |
| | | //ANativeWindow* window = (ANativeWindow*)surface; |
| | | |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[0]); |
| | | |
| | | int getEnvStat = cameraWrapper.javaVM->GetEnv((void **)&(cameraWrapper.javaEnv), JNI_VERSION_1_6); |
| | | if (getEnvStat == JNI_EDETACHED) |
| | | { |
| | | //LOG_WARN << "GetEnv: not attached" << std::endl; |
| | | if (cameraWrapper.javaVM->AttachCurrentThread(&(cameraWrapper.javaEnv), NULL) != 0) |
| | | LOG_WARN << "Failed to attach" << LOG_ENDL; |
| | | else |
| | | getEnvStat = JNI_OK; |
| | | } |
| | | else if (getEnvStat == JNI_OK) |
| | | { |
| | | } |
| | | else if (getEnvStat == JNI_EVERSION) |
| | | LOG_WARN << "GetEnv: version not supported" << LOG_ENDL; |
| | | else if (getEnvStat == JNI_ERR) |
| | | LOG_WARN << "GetEnv: JNI_ERR" << LOG_ENDL; |
| | | |
| | | if (maxSize==0) |
| | | { |
| | | //cameraWrapper.javaEnv->CallVoidMethod(_jobjOutputSurface, _jmOutputSurface_makeCurrent); |
| | | } |
| | | else |
| | | { |
| | | //cameraWrapper.javaEnv->CallVoidMethod(_jobjOutputSurface, _jmOutputSurface_awaitNewImage); |
| | | //cameraWrapper.javaEnv->CallVoidMethod(_jobjOutputSurface, _jmOutputSurface_drawImage); |
| | | // |
| | | //jbyteArray jbarrBuffer = (jbyteArray)cameraWrapper.javaEnv->CallObjectMethod(_jobjOutputSurface, _jmOutputSurface_readPixels); |
| | | //if (jbarrBuffer != nullptr) |
| | | // cameraWrapper.javaEnv->GetByteArrayRegion(jbarrBuffer, 0, 640 * 480 * 2, (jbyte*)buf); |
| | | } |
| | | |
| | | if (cameraWrapper.javaEnv->ExceptionCheck()) |
| | | cameraWrapper.javaEnv->ExceptionDescribe(); |
| | | |
| | | cameraWrapper.javaVM->DetachCurrentThread(); |
| | | } |
| | | |
| | | |
| | | /* |
| | | static void* createDecodeSurface() |
| | | { |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[0]); |
| | | |
| | | //OutputSurface outputSurface = null; |
| | | //outputSurface = new OutputSurface(mWidth, mHeight); |
| | | //outputSurface.getSurface(); |
| | | //outputSurface.release(); |
| | | |
| | | |
| | | //int getEnvStat = cameraWrapper.javaVM->GetEnv((void **)&(cameraWrapper.javaEnv), JNI_VERSION_1_6); |
| | | //if (getEnvStat == JNI_EDETACHED) |
| | | //{ |
| | | // //LOG_WARN << "GetEnv: not attached" << std::endl; |
| | | // if (cameraWrapper.javaVM->AttachCurrentThread(&(cameraWrapper.javaEnv), NULL) != 0) |
| | | // LOG_WARN << "Failed to attach" << LOG_ENDL; |
| | | // else |
| | | // getEnvStat = JNI_OK; |
| | | //} |
| | | //else if (getEnvStat == JNI_OK) |
| | | //{ |
| | | //} |
| | | //else if (getEnvStat == JNI_EVERSION) |
| | | // LOG_WARN << "GetEnv: version not supported" << LOG_ENDL; |
| | | //else if (getEnvStat == JNI_ERR) |
| | | // LOG_WARN << "GetEnv: JNI_ERR" << LOG_ENDL; |
| | | |
| | | //void* ret = nullptr; |
| | | //{ |
| | | // jclass jcOutputSurface = _jcOutputSurface = cameraWrapper.javaEnv->FindClass("com/example/nativecodec/OutputSurface"); |
| | | // _jcOutputSurface = static_cast<jclass>( cameraWrapper.javaEnv->NewGlobalRef( jcOutputSurface )); |
| | | // |
| | | // jmethodID jmOutputSurface_ctor = _jmOutputSurface_ctor = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "<init>", "(II)V"); |
| | | // jmethodID jmOutputSurface_getSurface = _jmOutputSurface_getSurface = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "getSurface", "()Landroid/view/Surface;"); |
| | | // jmethodID jmOutputSurface_readPixels = _jmOutputSurface_readPixels = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "readPixels", "()[B"); |
| | | // |
| | | // jmethodID jmOutputSurface_awaitNewImage = _jmOutputSurface_awaitNewImage = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "awaitNewImage", "()V"); |
| | | // jmethodID jmOutputSurface_drawImage = _jmOutputSurface_drawImage = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "drawImage", "()V"); |
| | | // jmethodID jmOutputSurface_makeCurrent = _jmOutputSurface_makeCurrent = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "makeCurrent", "()V"); |
| | | // |
| | | // jobject jobjOutputSurface = _jobjOutputSurface = cameraWrapper.javaEnv->NewObject(jcOutputSurface , jmOutputSurface_ctor, 1920, 1088); |
| | | // _jobjOutputSurface = static_cast<jobject>( cameraWrapper.javaEnv->NewGlobalRef( jobjOutputSurface )); |
| | | // |
| | | // jobject jobjSurface = _jobjSurface = cameraWrapper.javaEnv->CallObjectMethod(jobjOutputSurface, jmOutputSurface_getSurface); |
| | | // ret = ANativeWindow_fromSurface(cameraWrapper.javaEnv, jobjSurface); |
| | | // cameraWrapper.amcdConfig.readPixels_callback = readPixels; |
| | | //} |
| | | |
| | | //if (cameraWrapper.javaEnv->ExceptionCheck()) |
| | | // cameraWrapper.javaEnv->ExceptionDescribe(); |
| | | // |
| | | //cameraWrapper.javaVM->DetachCurrentThread(); |
| | | |
| | | //return ret; |
| | | } |
| | | */ |
| | | |
| | | static void* getSurfaceHolderGetSurface(JNIEnv* env, jobject jobjSurfaceHolder) |
| | | { |
| | | jclass jcSurfaceHolder = env->FindClass("android/view/SurfaceHolder"); |
| | | jmethodID jmSurfaceHolder_getSurface = env->GetMethodID(jcSurfaceHolder, "getSurface", "()Landroid/view/Surface;"); |
| | | jobject surface = env->CallObjectMethod(jobjSurfaceHolder, jmSurfaceHolder_getSurface); |
| | | return surface; |
| | | } |
| | | |
| | | // set the surface |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_setSurface(JNIEnv *env, jclass clazz, jint cameraIdx, jobject surfaceRender) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_setSurface" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | // obtain a native windowRender from a Java surface |
| | | //if (cameraWrapper.windowDecode != nullptr) |
| | | //{ |
| | | // ANativeWindow_release((ANativeWindow*)(cameraWrapper.windowDecode)); |
| | | // cameraWrapper.windowDecode = NULL; |
| | | //} |
| | | //cameraWrapper.windowDecode = ANativeWindow_fromSurface(env, surfaceDecoder); |
| | | //cameraWrapper.windowDecode = getSurfaceHolderGetSurface(env, surfaceDecoder); |
| | | //cameraWrapper.windowDecode = ANativeWindow_fromSurface(env, (jobject)getSurfaceHolderGetSurface(env, surfaceDecoder)); |
| | | |
| | | if (cameraWrapper.windowRender != nullptr) |
| | | { |
| | | ANativeWindow_release((ANativeWindow*)(cameraWrapper.windowRender)); |
| | | cameraWrapper.windowRender = NULL; |
| | | } |
| | | cameraWrapper.windowRender = ANativeWindow_fromSurface(env, surfaceRender); |
| | | //cameraWrapper.windowRender = ANativeWindow_fromSurface(env, (jobject)getSurfaceHolderGetSurface(env, surfaceRender)); |
| | | |
| | | LOGP(DEBUG, "@@@ setsurface %p,%p", cameraWrapper.windowDecode, cameraWrapper.windowRender); |
| | | } |
| | | |
| | | jboolean Java_cn_com_basic_face_util_RtspFaceNative_createPlayer(JNIEnv* env, jclass clazz, jint cameraIdx, jstring uri) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_createPlayer" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | { |
| | | const char *utf8Uri = env->GetStringUTFChars(uri, NULL); |
| | | cameraWrapper.rtspConfig = utf8Uri; |
| | | cameraWrapper.rtspConfig.rtspURL = utf8Uri; |
| | | env->ReleaseStringUTFChars(uri, utf8Uri); |
| | | |
| | | cameraWrapper.rtspConfig.progName = "RtspNativeCodec"; |
| | | cameraWrapper.rtspConfig.rtspURL = utf8; |
| | | cameraWrapper.rtspConfig.aux = true; // ffmpeg need aux, but live555 not |
| | | cameraWrapper.rtspConfig.verbosityLevel = 1; |
| | | cameraWrapper.rtspConfig.tunnelOverHTTPPortNum = 0; |
| | | cameraWrapper.rtspConfig.args = nullptr; |
| | | |
| | | PL_RTSPClient* rtspClient = (PL_RTSPClient*)cameraWrapper.pipeLine.push_elem("PL_RTSPClient"); |
| | | bool ret = rtspClient->init(&(cameraWrapper.rtspConfig)); |
| | | if (!ret) |
| | | { |
| | | LOG_ERROR << "rtspClient.init error" << std::endl; |
| | | return JNI_FALSE; |
| | | } |
| | | } |
| | | |
| | | { |
| | | PL_AndroidMediaCodecDecoder* amcDecoder = (PL_AndroidMediaCodecDecoder*)cameraWrapper.pipeLine.push_elem("PL_AndroidMediaCodecDecoder"); |
| | | cameraWrapper.amcdConfig.ak_mime = ""; |
| | | cameraWrapper.amcdConfig.ak_width = 1920; |
| | | cameraWrapper.amcdConfig.ak_height = 1080; |
| | | bool ret = amcDecoder->init(&(cameraWrapper.amcdConfig)); |
| | | if (!ret) |
| | | { |
| | | LOG_ERROR << "amcDecoder.init error" << std::endl; |
| | | return JNI_FALSE; |
| | | } |
| | | } |
| | | |
| | | { |
| | | PL_AndroidSurfaceViewSink* asvSink = (PL_AndroidSurfaceViewSink*)cameraWrapper.pipeLine.push_elem("PL_AndroidSurfaceViewSink"); |
| | | cameraWrapper.asvsConfig. |
| | | bool ret = asvSink->init(&(cameraWrapper.asvsConfig)); |
| | | if (!ret) |
| | | { |
| | | LOG_ERROR << "asvSink.init error" << std::endl; |
| | | return JNI_FALSE; |
| | | } |
| | | } |
| | | |
| | | return (cameraWrapper.start() ? JNI_TRUE : JNI_FALSE); |
| | | { |
| | | cameraWrapper.amcdConfig.ak_mime = "video/avc"; |
| | | cameraWrapper.amcdConfig.ak_width = 1920; |
| | | cameraWrapper.amcdConfig.ak_height = 1080; // nvida yuv420p,1088; Amlogic yuv420p,1080 |
| | | //cameraWrapper.amcdConfig.windowSurfaceDecode = cameraWrapper.windowDecode; //cameraWrapper.windowDecode; |
| | | //cameraWrapper.amcdConfig.windowSurfaceRender = cameraWrapper.windowRender; //cameraWrapper.windowRender; |
| | | cameraWrapper.amcdConfig.releaseOutputBuffIdx = true; |
| | | cameraWrapper.amcdConfig.releaseOutputBuffIdxInPay = false; |
| | | cameraWrapper.amcdConfig.generateDecodedDataPerFrame = 1; |
| | | |
| | | //cameraWrapper.amcdConfig.createSurface_callback = createDecodeSurface; |
| | | //cameraWrapper.amcdConfig.readPixels_callback = readPixels; |
| | | |
| | | #ifndef USE_ST_SDK |
| | | cameraWrapper.amcdConfig.releaseOutputBuffIdxInPay = true;//#todo |
| | | #endif |
| | | } |
| | | |
| | | { |
| | | cameraWrapper.asvrConfig.windowSurface = cameraWrapper.windowRender; |
| | | cameraWrapper.asvrConfig.outputOriginFrame = true; |
| | | cameraWrapper.asvrConfig.outputRenderFrame = true; |
| | | cameraWrapper.asvrConfig.scaleToWidth = 640; |
| | | cameraWrapper.asvrConfig.scaleToHeight = 480; |
| | | } |
| | | |
| | | { |
| | | cameraWrapper.sftConfig.point_size = 21; |
| | | cameraWrapper.sftConfig.detect_face_cnt_limit = MAX_FACE; |
| | | cameraWrapper.sftConfig.draw_face_rect = false; |
| | | cameraWrapper.sftConfig.draw_face_feature_point = false; |
| | | cameraWrapper.sftConfig.generate_face_feature = true; |
| | | cameraWrapper.sftConfig.explode_feature_rect_x = 30; |
| | | cameraWrapper.sftConfig.explode_feature_rect_y = 40; |
| | | cameraWrapper.sftConfig.doTrackPerFrame = 50; |
| | | } |
| | | |
| | | bool ret = cameraWrapper.initPl(); |
| | | if (ret) |
| | | return (cameraWrapper.start() ? JNI_TRUE : JNI_FALSE); |
| | | else |
| | | return JNI_FALSE; |
| | | } |
| | | |
| | | // shut down the native media system |
| | | void Java_RtspNativeCodec_shutdown(JNIEnv* env, jclass clazz, jint cameraIdx) |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_shutdown(JNIEnv* env, jclass clazz, jint cameraIdx) |
| | | { |
| | | LOGV("@@@ Java_RtspNativeCodec_shutdown"); |
| | | assert(cameraIdx <= CAMERA_COUNT); |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_shutdown" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | cameraWrapper.stop(); |
| | | } |
| | | |
| | | jboolean Java_RtspNativeCodec_setFaceCallback(JNIEnv* env, jclass clazz, jobject assetMgr, jint cameraIdx, jstring func) |
| | | { |
| | | LOGV("@@@ Java_RtspNativeCodec_setFaceCallback"); |
| | | assert(cameraIdx <= CAMERA_COUNT); |
| | | |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | LOGE("jni callback (0)"); |
| | | jclass cls = env->GetObjectClass(clazz); |
| | | cameraWrapper.faceCallback = env->GetMethodID(cls, func, "(II)V"); // Java_FaceCallback_func |
| | | |
| | | // call: |
| | | //env->CallVoidMethod(obj, callback, 5 , 10); |
| | | |
| | | //#todo remove |
| | | //see: https://github.com/BelledonneCommunications/mediastreamer2/blob/master/src/android/android_mediacodec.cpp |
| | | static int handle_java_exception(JNIEnv *env) { |
| | | if (env->ExceptionCheck()) { |
| | | env->ExceptionDescribe(); |
| | | env->ExceptionClear(); |
| | | return -1; |
| | | } |
| | | return 0; |
| | | } |
| | | //#todo remove |
| | | static bool _loadClass(JNIEnv *env, const char *className, jclass *_class) { |
| | | *_class = env->FindClass(className); |
| | | if(handle_java_exception(env) == -1 || *_class == NULL) { |
| | | LOGP(ERROR, "Could not load Java class [%s]", className); |
| | | return false; |
| | | } |
| | | return true; |
| | | } |
| | | |
| | | jboolean Java_RtspNativeCodec_getFaceList(JNIEnv* env, jclass clazz, jobject assetMgr, jint cameraIdx, jbyteArray faceListPb) |
| | | { |
| | | assert(cameraIdx <= CAMERA_COUNT); |
| | | //#todo remove |
| | | static bool _getStaticMethodID(JNIEnv *env, jclass _class, const char *name, const char *sig, jmethodID *method) { |
| | | *method = env->GetStaticMethodID(_class, name, sig); |
| | | if(handle_java_exception(env) == -1 || *method == NULL) { |
| | | LOGP(ERROR, "Could not get static method %s[%s]", name, sig); |
| | | return false; |
| | | } |
| | | return true; |
| | | } |
| | | |
| | | jboolean Java_cn_com_basic_face_util_RtspFaceNative_setFaceCallback(JNIEnv* env, jclass clazz, jint cameraIdx, jstring className, jstring funcName) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_setFaceCallback" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | //jclass cls = env->GetObjectClass(clazz); |
| | | |
| | | // set for multithread callback |
| | | env->GetJavaVM(&(cameraWrapper.javaVM)); |
| | | cameraWrapper.javaEnv = env; |
| | | |
| | | //ms_set_jvm(cameraWrapper.javaVM); |
| | | |
| | | std::string _className; |
| | | { |
| | | const char *utfFunc = env->GetStringUTFChars(className, NULL); |
| | | _className = utfFunc; |
| | | env->ReleaseStringUTFChars(className, utfFunc); |
| | | } |
| | | std::string _funcName; |
| | | { |
| | | const char *utfFunc = env->GetStringUTFChars(funcName, NULL); |
| | | _funcName = utfFunc; |
| | | env->ReleaseStringUTFChars(funcName, utfFunc); |
| | | } |
| | | |
| | | _loadClass(env, _className.c_str(), &(cameraWrapper.faceCallbackClazz)); |
| | | |
| | | cameraWrapper.faceCallbackClazz = static_cast<jclass>( env->NewGlobalRef( cameraWrapper.faceCallbackClazz )); //#todo need release? |
| | | |
| | | _getStaticMethodID(env, cameraWrapper.faceCallbackClazz, _funcName.c_str(), "(II)V", &(cameraWrapper.faceCallbackFunc)); |
| | | |
| | | // call test (sync) |
| | | cameraWrapper.javaEnv->CallStaticVoidMethod(cameraWrapper.faceCallbackClazz, cameraWrapper.faceCallbackFunc, cameraWrapper.cameraIdx, 0); |
| | | } |
| | | |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_lockFace(JNIEnv* env, jclass clazz, jint cameraIdx) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_lockFace" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | cameraWrapper.lockFace(); |
| | | } |
| | | |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_releaseFace(JNIEnv* env, jclass clazz, jint cameraIdx) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_releaseFace" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | cameraWrapper.releaseFace(); |
| | | } |
| | | |
| | | jint Java_cn_com_basic_face_util_RtspFaceNative_getFaceList(JNIEnv* env, jclass clazz, jint cameraIdx, jint dbId, jobject faceListPb) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_getFaceList" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | // #todo optimize not copy data here, set data to jbyteArray directly |
| | | // Y channel of YUV420p, packed in protobuf |
| | | uint8_t buffer[MAX_FACE * MAX_FACE_WIDTH * MAX_FACE_HEIGHT] = {0}; |
| | | size_t buffSize = sizeof(buffer); |
| | | |
| | | bool ret = false; |
| | | #ifdef USE_ST_SDK |
| | | uint8_t * pBufPB = buffer + sizeof(EVPHeader) + sizeof(FDP_FaceDetectPB); |
| | | buffSize = sizeof(buffer) - sizeof(EVPHeader) - sizeof(FDP_FaceDetectPB); |
| | | ret = cameraWrapper.faceCache.getFaceListPb(pBufPB, buffSize); |
| | | #endif |
| | | |
| | | if (!ret) |
| | | { |
| | | LOG_INFO << "No face captured" << LOG_ENDL; |
| | | return 0; |
| | | } |
| | | |
| | | LOG_INFO << "Face captured " << LOG_ENDL; |
| | | |
| | | EVPHeader* evpHeader = new (buffer) EVPHeader; |
| | | evpHeader->proto = EVPProto::EVPP_PROTOBUF; |
| | | evpHeader->cmd = FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_PB; |
| | | evpHeader->size = sizeof(EVPHeader) + sizeof(FDP_FaceDetectPB) + buffSize; |
| | | buffSize = evpHeader->size; |
| | | |
| | | FDP_FaceDetectPB* fdpFaceDetectPB = new (buffer + sizeof(EVPHeader)) FDP_FaceDetectPB(dbId); |
| | | |
| | | evpHeader->hton(); |
| | | fdpFaceDetectPB->hton(); |
| | | |
| | | //{ |
| | | // static size_t f = 0; |
| | | // char fname[50]; |
| | | // sprintf(fname, "/sdcard/facelist-%d-%d-%u.pb", dbId, cameraIdx, ++f); |
| | | // FILE *pFile = fopen(fname, "wb"); |
| | | // fwrite(buffer, 1, buffSize, pFile); |
| | | // fclose(pFile); |
| | | // if (f > 10)exit(0); |
| | | //} |
| | | |
| | | jclass jcRefByteArray = env->FindClass("cn/com/basic/face/util/RefByteArray"); |
| | | jfieldID jfRefByteArray_arr = env->GetFieldID(jcRefByteArray, "arr","[B"); |
| | | |
| | | jbyteArray jbaFaceListPb = env->NewByteArray(buffSize); |
| | | env->SetByteArrayRegion(jbaFaceListPb, 0, buffSize, (const jbyte*)buffer); |
| | | env->SetObjectField(faceListPb, jfRefByteArray_arr, jbaFaceListPb); |
| | | |
| | | #ifdef ENABLE_DEBUG_SENDFACEDETECT |
| | | dbgSendFaceDetect(buffer, buffSize); |
| | | #endif |
| | | |
| | | return buffSize; |
| | | } |
| | | |
| | | jobject _Java_cn_com_basic_face_util_RtspFaceNative_getFaceImages(JNIEnv* env, jclass clazz, jint cameraIdx, jobject faceImages) |
| | | { |
| | | LOG_DEBUG << "@@@ _Java_cn_com_basic_face_util_RtspFaceNative_getFaceImages" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | std::vector<NativeImgIdx> imgIdxes; |
| | | size_t _faceImagesSize = MAX_FACE * MAX_FACE_WIDTH * MAX_FACE_HEIGHT * 3; |
| | | uint8_t* _faceImages = new uint8_t[_faceImagesSize]; // android stack is small |
| | | memset(_faceImages, 0, _faceImagesSize); |
| | | |
| | | bool ret = false; |
| | | |
| | | #ifdef USE_ST_SDK |
| | | ret = cameraWrapper.faceCache.getFaceListImage(imgIdxes, _faceImages, _faceImagesSize); |
| | | #endif |
| | | |
| | | if (imgIdxes.size() == 0) |
| | | { |
| | | LOG_INFO << "No face image captured" << LOG_ENDL; |
| | | delete[] _faceImages; |
| | | return nullptr; |
| | | } |
| | | |
| | | jclass jcRefByteArray = env->FindClass("cn/com/basic/face/util/RefByteArray"); |
| | | jfieldID jfRefByteArray_arr = env->GetFieldID(jcRefByteArray, "arr","[B"); |
| | | |
| | | jbyteArray jbaFaceImages = env->NewByteArray(_faceImagesSize); |
| | | LOG_DEBUG << "_faceImagesSize=" << _faceImagesSize << LOG_ENDL; |
| | | env->SetByteArrayRegion(jbaFaceImages, 0, _faceImagesSize, (const jbyte*)_faceImages); |
| | | env->SetObjectField(faceImages, jfRefByteArray_arr, jbaFaceImages); |
| | | delete[] _faceImages; |
| | | |
| | | jclass jcArrayList = env->FindClass("java/util/ArrayList"); |
| | | jmethodID jmArrayList_ctor = env->GetMethodID(jcArrayList, "<init>", "()V"); |
| | | jmethodID jmArrayList_add = env->GetMethodID(jcArrayList, "add", "(Ljava/lang/Object;)Z"); |
| | | |
| | | jclass jcNativeImgIdx = env->FindClass("cn/com/basic/face/util/NativeImgIdx"); |
| | | jmethodID jmNativeImgIdx_ctor = env->GetMethodID(jcNativeImgIdx, "<init>", "()V"); |
| | | jfieldID jfNativeImgIdx_offset = env->GetFieldID(jcNativeImgIdx, "offset","I"); |
| | | jfieldID jfNativeImgIdx_size = env->GetFieldID(jcNativeImgIdx, "size","I"); |
| | | jfieldID jfNativeImgIdx_type = env->GetFieldID(jcNativeImgIdx, "type","I"); |
| | | jfieldID jfNativeImgIdx_width = env->GetFieldID(jcNativeImgIdx, "width","I"); |
| | | jfieldID jfNativeImgIdx_height = env->GetFieldID(jcNativeImgIdx, "height","I"); |
| | | |
| | | jobject jobjArrayList_ImgIdxes = env->NewObject(jcArrayList , jmArrayList_ctor); |
| | | |
| | | for(size_t i = 0 ; i < imgIdxes.size(); i++) |
| | | { |
| | | jobject jobjNativeImgIdx = env->NewObject(jcNativeImgIdx , jmNativeImgIdx_ctor); |
| | | env->SetIntField(jobjNativeImgIdx, jfNativeImgIdx_offset, imgIdxes[i].offset); |
| | | env->SetIntField(jobjNativeImgIdx, jfNativeImgIdx_size, imgIdxes[i].size); |
| | | env->SetIntField(jobjNativeImgIdx, jfNativeImgIdx_type, imgIdxes[i].type); |
| | | env->SetIntField(jobjNativeImgIdx, jfNativeImgIdx_width, imgIdxes[i].width); |
| | | env->SetIntField(jobjNativeImgIdx, jfNativeImgIdx_height, imgIdxes[i].height); |
| | | |
| | | env->CallBooleanMethod(jobjArrayList_ImgIdxes , jmArrayList_add , jobjNativeImgIdx); |
| | | |
| | | LOG_DEBUG << "imgIdx " << i << ":" << std::string(imgIdxes[i]) << LOG_ENDL; |
| | | } |
| | | |
| | | return jobjArrayList_ImgIdxes; |
| | | } |
| | | |
| | | jobject Java_cn_com_basic_face_util_RtspFaceNative_getFaceImages(JNIEnv* env, jclass clazz, jint cameraIdx) |
| | | { |
| | | LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_getFaceImages" << LOG_ENDL; |
| | | assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); |
| | | cameraIdx -= 1; |
| | | CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); |
| | | |
| | | std::vector<NativeImgIdx> imgIdxes; |
| | | size_t _faceImagesSize = MAX_FACE * MAX_FACE_WIDTH * MAX_FACE_HEIGHT * 3; |
| | | uint8_t* _faceImages = new uint8_t[_faceImagesSize]; // android stack is small |
| | | memset(_faceImages, 0, _faceImagesSize); |
| | | |
| | | bool ret = false; |
| | | |
| | | #ifdef USE_ST_SDK |
| | | ret = cameraWrapper.faceCache.getFaceListImage(imgIdxes, _faceImages, _faceImagesSize); |
| | | #endif |
| | | |
| | | if (imgIdxes.size() == 0) |
| | | { |
| | | LOG_INFO << "No face image captured" << LOG_ENDL; |
| | | delete[] _faceImages; |
| | | return nullptr; |
| | | } |
| | | |
| | | jclass jcArrayList = env->FindClass("java/util/ArrayList"); |
| | | jmethodID jmArrayList_ctor = env->GetMethodID(jcArrayList, "<init>", "()V"); |
| | | jmethodID jmArrayList_add = env->GetMethodID(jcArrayList, "add", "(Ljava/lang/Object;)Z"); |
| | | |
| | | jclass jcNativeImg = env->FindClass("cn/com/basic/face/util/NativeImg"); |
| | | jmethodID jmNativeImg_ctor = env->GetMethodID(jcNativeImg, "<init>", "()V"); |
| | | jfieldID jfNativeImg_size = env->GetFieldID(jcNativeImg, "size","I"); |
| | | jfieldID jfNativeImg_type = env->GetFieldID(jcNativeImg, "type","I"); |
| | | jfieldID jfNativeImg_width = env->GetFieldID(jcNativeImg, "width","I"); |
| | | jfieldID jfNativeImg_height = env->GetFieldID(jcNativeImg, "height","I"); |
| | | jfieldID jfNativeImg_image = env->GetFieldID(jcNativeImg, "image","[B"); |
| | | |
| | | jobject jobjArrayList_Imgs = env->NewObject(jcArrayList , jmArrayList_ctor); |
| | | |
| | | for(size_t i = 0 ; i < imgIdxes.size(); i++) |
| | | { |
| | | jobject jobjNativeImg = env->NewObject(jcNativeImg , jmNativeImg_ctor); |
| | | env->SetIntField(jobjNativeImg, jfNativeImg_size, imgIdxes[i].size); |
| | | env->SetIntField(jobjNativeImg, jfNativeImg_type, imgIdxes[i].type); |
| | | env->SetIntField(jobjNativeImg, jfNativeImg_width, imgIdxes[i].width); |
| | | env->SetIntField(jobjNativeImg, jfNativeImg_height, imgIdxes[i].height); |
| | | |
| | | jbyteArray jbaFaceImage = env->NewByteArray(imgIdxes[i].size); |
| | | env->SetByteArrayRegion(jbaFaceImage, 0, imgIdxes[i].size, (const jbyte*)(_faceImages + imgIdxes[i].offset)); |
| | | env->SetObjectField(jobjNativeImg, jfNativeImg_image, jbaFaceImage); |
| | | |
| | | env->CallBooleanMethod(jobjArrayList_Imgs, jmArrayList_add , jobjNativeImg); |
| | | |
| | | LOG_DEBUG << "imgIdx " << i << ":" << std::string(imgIdxes[i]) << LOG_ENDL; |
| | | } |
| | | |
| | | delete[] _faceImages; |
| | | _faceImages = nullptr; |
| | | return jobjArrayList_Imgs; |
| | | } |
| | | |
| | | void Java_cn_com_basic_face_util_RtspFaceNative_dbgFaceAdd(JNIEnv* env, jclass clazz, jobject faceImage) |
| | | { |
| | | jclass jcNativeImg = env->FindClass("cn/com/basic/face/util/NativeImg"); |
| | | jmethodID jmNativeImg_ctor = env->GetMethodID(jcNativeImg, "<init>", "()V"); |
| | | jfieldID jfNativeImg_size = env->GetFieldID(jcNativeImg, "size","I"); |
| | | jfieldID jfNativeImg_type = env->GetFieldID(jcNativeImg, "type","I"); |
| | | jfieldID jfNativeImg_width = env->GetFieldID(jcNativeImg, "width","I"); |
| | | jfieldID jfNativeImg_height = env->GetFieldID(jcNativeImg, "height","I"); |
| | | jfieldID jfNativeImg_image = env->GetFieldID(jcNativeImg, "image","[B"); |
| | | |
| | | NativeImgIdx imgidx; |
| | | imgidx.size = env->GetIntField(faceImage, jfNativeImg_size); |
| | | imgidx.type = env->GetIntField(faceImage, jfNativeImg_type); |
| | | imgidx.width = env->GetIntField(faceImage, jfNativeImg_width); |
| | | imgidx.height = env->GetIntField(faceImage, jfNativeImg_height); |
| | | |
| | | uint8_t buffer[MAX_FACE * MAX_FACE_WIDTH * MAX_FACE_HEIGHT * 3]; |
| | | size_t buffSize = 0; |
| | | |
| | | jbyteArray jbaFaceImage = (jbyteArray)env->GetObjectField(faceImage, jfNativeImg_image); |
| | | env->GetByteArrayRegion(jbaFaceImage, 0, imgidx.size, (jbyte*)buffer); |
| | | |
| | | #ifdef ENABLE_DEBUG_SENDFACEADD |
| | | dbgSendFaceAdd(imgidx, buffer); |
| | | #endif |
| | | } |
| | | |
| | | jstring Java_cn_com_basic_face_util_RtspFaceNative_dbgGetLog(JNIEnv *env, jclass clazz) |
| | | { |
| | | return env->NewStringUTF(g_dbgLog.c_str()); |
| | | } |
| | | |
| | | //void Java_cn_com_basic_face_util_RtspFaceNative_dbgReadTexture(JNIEnv* env, jclass clazz) |
| | | //{ |
| | | // dbgReadTexture(); |
| | | //} |
| | | |
| | | } |