From 1b4d2718a4864190a47d8b7863f36a9ae0b8055c Mon Sep 17 00:00:00 2001 From: houxiao <houxiao@454eff88-639b-444f-9e54-f578c98de674> Date: 星期五, 28 四月 2017 17:38:09 +0800 Subject: [PATCH] optimize --- VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNI.cpp | 126 ++++++++++++++++++++++++++++++++++++----- 1 files changed, 109 insertions(+), 17 deletions(-) diff --git a/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNI.cpp b/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNI.cpp index 439e16c..375c7eb 100644 --- a/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNI.cpp +++ b/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNI.cpp @@ -10,6 +10,13 @@ #include <pthread.h> +#include <ev_proto.h> +#include <face_daemon_proto.h> + +#include <stdlib.h> + +#include "DebugNetwork.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); @@ -22,7 +29,7 @@ void Java_cn_com_basic_face_util_RtspFaceNative_init(JNIEnv *env, jclass clazz) { - g_logger.set_level(VERBOSE); + g_logger.set_level(INFO); #ifdef USE_ST_SDK PipeLine::register_global_elem_creator("PL_SensetimeFaceTrack", create_PL_SensetimeFaceTrack); @@ -51,6 +58,24 @@ env->ReleaseStringUTFChars(ipaddr, utf8); } +void createHiddenSurface(JNIEnv *env, int cameraIdx) +{ + CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); + + //OutputSurface outputSurface = null; + //outputSurface = new OutputSurface(mWidth, mHeight); + //outputSurface.getSurface(); + //outputSurface.release(); + + jclass jcOutputSurface = env->FindClass("com/example/nativecodec/OutputSurface"); + jmethodID jmOutputSurface_ctor = env->GetMethodID(jcOutputSurface, "<init>", "(II)V"); + jmethodID jmOutputSurface_getSurface = env->GetMethodID(jcOutputSurface, "getSurface", "()Landroid/view/Surface;"); + + jobject jobjOutputSurface = env->NewObject(jcOutputSurface , jmOutputSurface_ctor, 1920, 1088); + jobject jobjSurface = env->CallObjectMethod(jobjOutputSurface, jmOutputSurface_getSurface); + cameraWrapper.windowDecode = ANativeWindow_fromSurface(env, jobjSurface); +} + // set the surface void Java_cn_com_basic_face_util_RtspFaceNative_setSurface(JNIEnv *env, jclass clazz, jint cameraIdx, jobject surface) { @@ -60,14 +85,17 @@ CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]); - // obtain a native window from a Java surface - if (cameraWrapper.window) + // obtain a native windowRender from a Java surface + if (cameraWrapper.windowRender) { - ANativeWindow_release((ANativeWindow*)(cameraWrapper.window)); - cameraWrapper.window = NULL; + ANativeWindow_release((ANativeWindow*)(cameraWrapper.windowRender)); + cameraWrapper.windowRender = NULL; } - cameraWrapper.window = ANativeWindow_fromSurface(env, surface); - LOGP(DEBUG, "@@@ setsurface %p", cameraWrapper.window); + cameraWrapper.windowRender = ANativeWindow_fromSurface(env, surface); + + //createHiddenSurface(env, cameraIdx); + + LOGP(DEBUG, "@@@ setsurface %p", cameraWrapper.windowRender); } jboolean Java_cn_com_basic_face_util_RtspFaceNative_createPlayer(JNIEnv* env, jclass clazz, jint cameraIdx, jstring uri) @@ -94,20 +122,23 @@ cameraWrapper.amcdConfig.ak_mime = "video/avc"; cameraWrapper.amcdConfig.ak_width = 640; cameraWrapper.amcdConfig.ak_height = 480; - cameraWrapper.amcdConfig.windowSurface = nullptr; // cameraWrapper.window - cameraWrapper.amcdConfig.releaseOutputBuffIdx = true; + cameraWrapper.amcdConfig.windowSurfaceDecode = nullptr; //cameraWrapper.windowDecode; + cameraWrapper.amcdConfig.windowSurfaceRender = nullptr; + cameraWrapper.amcdConfig.releaseOutputBuffIdx = true; + cameraWrapper.amcdConfig.releaseOutputBuffIdxInPay = false; cameraWrapper.amcdConfig.generateDecodedDataPerFrame = 1; - cameraWrapper.amcdConfig.renderFromOutputBuffIdx = false;//true - cameraWrapper.amcdConfig.renderFromOutputBuff = false; + #ifndef USE_ST_SDK cameraWrapper.amcdConfig.releaseOutputBuffIdxInPay = true; #endif } { - cameraWrapper.asvrConfig.windowSurface = cameraWrapper.window; + cameraWrapper.asvrConfig.windowSurface = cameraWrapper.windowRender; cameraWrapper.asvrConfig.outputOriginFrame = true; cameraWrapper.asvrConfig.outputRenderFrame = true; + cameraWrapper.asvrConfig.scaleToWidth = 640; + cameraWrapper.asvrConfig.scaleToHeight = 480; } { @@ -115,7 +146,9 @@ 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_feature = true; + cameraWrapper.sftConfig.explode_feature_rect_x = 30; + cameraWrapper.sftConfig.explode_feature_rect_y = 40; cameraWrapper.sftConfig.doTrackPerFrame = 50; } @@ -221,7 +254,7 @@ cameraWrapper.releaseFace(); } -jint Java_cn_com_basic_face_util_RtspFaceNative_getFaceList(JNIEnv* env, jclass clazz, jint cameraIdx, jobject faceListPb) +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); @@ -232,11 +265,12 @@ // Y channel of YUV420p, packed in protobuf uint8_t buffer[MAX_FACE * MAX_FACE_WIDTH * MAX_FACE_HEIGHT] = {0}; size_t buffSize = sizeof(buffer); - //const size_t arrSize = env->GetArrayLength(faceListPb); - bool ret = false; + bool ret = false; #ifdef USE_ST_SDK - ret = cameraWrapper.faceCache.getFaceListPb(buffer, buffSize); + 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) @@ -247,12 +281,38 @@ 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; } @@ -382,4 +442,36 @@ 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()); +} + } -- Gitblit v1.8.0