#include "RtspNativeCodecJNI.h" #include "CameraWrapper.h" #include "CaptureCamera.h" #include #include //#include "looper.h" #include #include #include #include #include #include #include #include "DebugNetwork.h" #include "cpu_sched_test.h" //#include 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); static std::string g_stface_license_str; CameraWrapper g_CameraWrappers[CAMERA_COUNT]; CaptureCamera g_CaptureCamera; extern "C" { void Java_cn_com_basic_face_util_RtspFaceNative_init(JNIEnv *env, jclass clazz) { g_logger.set_level(INFO); cpu_sched(); #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].pipeLineDecoderDetector = new PipeLine; g_CameraWrappers[i].pipeLineAnalyzer = new PipeLine; g_CameraWrappers[i].pipeLineRender = new PipeLine; //PipeLine& pipeLine(*(g_CameraWrappers[i].pipeLineDecoderDetector)); } } void Java_cn_com_basic_face_util_RtspFaceNative_setLocalIP(JNIEnv *env, jclass clazz, jstring ipaddr) { LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_setLocalIP" << LOG_ENDL; 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; } env->ReleaseStringUTFChars(ipaddr, utf8); } //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) { //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( cameraWrapper.javaEnv->NewGlobalRef( jcOutputSurface )); // // jmethodID jmOutputSurface_ctor = _jmOutputSurface_ctor = cameraWrapper.javaEnv->GetMethodID(jcOutputSurface, "", "(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( 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, surfaceRender); //cameraWrapper.windowDecode = getSurfaceHolderGetSurface(env, surfaceRender); //cameraWrapper.windowDecode = ANativeWindow_fromSurface(env, (jobject)getSurfaceHolderGetSurface(env, surfaceRender)); if (cameraWrapper.windowRender != nullptr) { ANativeWindow_release((ANativeWindow *) (cameraWrapper.windowRender)); cameraWrapper.windowRender = NULL; } cameraWrapper.windowRender = ANativeWindow_fromSurface(env, surfaceRender); cameraWrapper.asvrConfig.windowSurface = cameraWrapper.windowRender; // support reset //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]); int w = 0, h = 0;//#todo { const char *utf8Uri = env->GetStringUTFChars(uri, NULL); cameraWrapper.rtspConfig.rtspURL = utf8Uri; env->ReleaseStringUTFChars(uri, utf8Uri); if (cameraWrapper.rtspConfig.rtspURL.find("/main/") != std::string::npos || cameraWrapper.rtspConfig.rtspURL.find("subtype=0") != std::string::npos)//#todo temp use { w = 1920; h = 1080; } else if (cameraWrapper.rtspConfig.rtspURL.find("/h264/") != std::string::npos && cameraWrapper.rtspConfig.rtspURL.find("/sub/") != std::string::npos) { w = 640; h = 480; } else if (cameraWrapper.rtspConfig.rtspURL.find("/h264_2") != std::string::npos) { w = 800; h = 600; } else if (cameraWrapper.rtspConfig.rtspURL.find("/h264") != std::string::npos) { w = 1920; h = 1080; } else if (cameraWrapper.rtspConfig.rtspURL.find("subtype=2") != std::string::npos) { w = 1280; h = 720; } else { w = 640; h = 480; } cameraWrapper.rtspConfig.progName = "RtspNativeCodec"; cameraWrapper.rtspConfig.aux = true; // ffmpeg need aux, but live555 not cameraWrapper.rtspConfig.verbosityLevel = 1; cameraWrapper.rtspConfig.tunnelOverHTTPPortNum = 0; cameraWrapper.rtspConfig.args = nullptr; } { cameraWrapper.amcdConfig.ak_mime = "video/avc"; cameraWrapper.amcdConfig.ak_width = w; cameraWrapper.amcdConfig.ak_height = h; // nvida yuv420p,1088; Amlogic yuv420p,1080; RK3288 NV12 //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 = 0; cameraWrapper.asvrConfig.scaleToHeight = 0; } { 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.generate_face_point = false; cameraWrapper.sftConfig.explode_feature_rect_x = 30; cameraWrapper.sftConfig.explode_feature_rect_y = 40; cameraWrapper.sftConfig.clamp_feature_rect = true; cameraWrapper.sftConfig.doTrackPerFrame = 1; //cameraWrapper.sftConfig.license_str = g_stface_license_str; cameraWrapper.sftConfig.visionConeAngle = 35.0f; cameraWrapper.sftConfig.score_min = 0.2f; } 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_cn_com_basic_face_util_RtspFaceNative_shutdown(JNIEnv *env, jclass clazz, jint cameraIdx) { 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(); } //#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; } //#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( 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, cameraIdx + 1); 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 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, "", "()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, "", "()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 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, "", "()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, "", "()V"); jfieldID jfNativeImg_st_track_id = env->GetFieldID(jcNativeImg, "st_track_id", "I"); 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_st_track_id, imgIdxes[i].st_track_id); 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, "", "()V"); jfieldID jfNativeImg_st_track_id = env->GetFieldID(jcNativeImg, "st_track_id", "I"); 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.st_track_id = env->GetIntField(faceImage, jfNativeImg_st_track_id); 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(); //} void Java_cn_com_basic_face_util_RtspFaceNative_dbgSetStfaceLicense(JNIEnv *env, jclass clazz, jstring lic) { LOG_WARN << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_dbgSetStfaceLicense" << LOG_ENDL; const char *utfLic = env->GetStringUTFChars(lic, NULL); g_stface_license_str = utfLic; env->ReleaseStringUTFChars(lic, utfLic); //LOG_ERROR << g_stface_license_str << LOG_ERROR; } void Java_cn_com_basic_face_util_RtspFaceNative_setCaptureSurface(JNIEnv *env, jclass clazz, jobject surfaceCapture) { if (g_CaptureCamera.windowRender != nullptr) { ANativeWindow_release((ANativeWindow *) (g_CaptureCamera.windowRender)); g_CaptureCamera.windowRender = NULL; } g_CaptureCamera.windowRender = ANativeWindow_fromSurface(env, surfaceCapture); } bool Java_cn_com_basic_face_util_RtspFaceNative_startCapturePlayer(JNIEnv *env, jclass clazz, jstring uri) { std::string _uri; { const char *utfFunc = env->GetStringUTFChars(uri, NULL); _uri = utfFunc; env->ReleaseStringUTFChars(uri, utfFunc); } if (_uri.empty()) return false; if (g_CaptureCamera.windowRender == nullptr) return false; g_CaptureCamera.cameraUri = _uri; return g_CaptureCamera.start(); } void Java_cn_com_basic_face_util_RtspFaceNative_stopCapturePlayer(JNIEnv *env, jclass clazz) { } jobject Java_cn_com_basic_face_util_RtspFaceNative_getCaptureImages(JNIEnv *env, jclass clazz) { } void Java_cn_com_basic_face_util_RtspFaceNative_setFaceLabel(JNIEnv *env, jclass clazz, jint cameraIdx, jint stTrackId, jstring label) { LOG_DEBUG << "@@@ Java_cn_com_basic_face_util_RtspFaceNative_setFaceLabel" << LOG_ENDL; assert(cameraIdx > 0 && cameraIdx <= CAMERA_COUNT); cameraIdx -= 1; CameraWrapper &cameraWrapper(g_CameraWrappers[cameraIdx]); const char *utfChars = env->GetStringUTFChars(label, NULL); std::string labelstr(utfChars); std::wstring wlabelstr; wlabelstr.assign(labelstr.begin(), labelstr.end()); env->ReleaseStringUTFChars(label, utfChars); cameraWrapper.setFaceLabel(stTrackId, wlabelstr); } } // extern C