From c6968ba2fbd5e9f4e32fd5f62fd587f9a7d2ebc1 Mon Sep 17 00:00:00 2001 From: houxiao <houxiao@454eff88-639b-444f-9e54-f578c98de674> Date: 星期五, 11 八月 2017 18:52:31 +0800 Subject: [PATCH] bug fix for rtsp server --- VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNIClient.cpp | 193 +++++++---------------------------------------- 1 files changed, 31 insertions(+), 162 deletions(-) diff --git a/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNIClient.cpp b/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNIClient.cpp index 72f5046..e4890ea 100644 --- a/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNIClient.cpp +++ b/VisitFace/RtspNativeCodec/app/src/main/cpp/RtspNativeCodecJNIClient.cpp @@ -10,7 +10,6 @@ #include <media/NdkMediaExtractor.h> #include <pthread.h> - #include <ev_proto.h> #include <face_daemon_proto.h> @@ -20,7 +19,6 @@ #include "TeleWrapper.h" #include "cpu_sched_test.h" - //#include <mediastreamer2/include/mediastreamer2/msjava.h> std::stringstream logss; @@ -36,30 +34,32 @@ TeleWrapper g_TeleWrapper; +const size_t _faceImagesSize = MAX_FACE * MAX_FACE_WIDTH * MAX_FACE_HEIGHT * 3; +uint8_t _faceImages[_faceImagesSize]; // android stack is small +void remoteFireFaceCountListener(const RtspFaceDetect::FireFaceCountListenerParams::Reader& params, RtspFaceDetect::FireFaceCountListenerResults::Builder& results){ + int count = params.getFaceCount(); + int index = params.getCameraIndex(); + g_CameraWrappers[index].fireFaceCount(count); +} extern "C" { void Java_cn_com_basic_face_util_RtspFaceNative_init(JNIEnv *env, jclass clazz) { - g_logger.set_level(INFO); +g_logger.set_level(INFO); - cpu_sched(); +cpu_sched(); -#ifdef USE_ST_SDK - PipeLine::register_global_elem_creator("PL_SensetimeFaceTrackMitiTrd", create_PL_SensetimeFaceTrackMultiTrd); -#endif +for (size_t i = 0; i < CAMERA_COUNT; i++) +{ + g_CameraWrappers[i].cameraIdx = i + 1; - for (size_t i = 0; i < CAMERA_COUNT; i++) - { - g_CameraWrappers[i].cameraIdx = i + 1; + g_CameraWrappers[i].pipeLineDecoderDetector = new PipeLine; + g_CameraWrappers[i].pipeLineRender = new PipeLine; + //PipeLine& pipeLine(*(g_CameraWrappers[i].pipeLineDecoderDetector)); +} - g_CameraWrappers[i].pipeLineDecoderDetector = new PipeLine; - g_CameraWrappers[i].pipeLineAnalyzer = new PipeLine; - g_CameraWrappers[i].pipeLineRender = new PipeLine; - //PipeLine& pipeLine(*(g_CameraWrappers[i].pipeLineDecoderDetector)); - } - - g_TeleWrapper.start(); +g_TeleWrapper.start(); } void @@ -75,126 +75,6 @@ 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<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 @@ -324,22 +204,6 @@ cameraWrapper.asvrConfig.directlyDisplay = true; } - { - 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 = 30.0f; - cameraWrapper.sftConfig.score_min = 0.4f; - } - cameraWrapper.fontPath = "/data/msyh.ttc"; bool ret = cameraWrapper.initPl(); @@ -444,6 +308,17 @@ cameraWrapper.cameraIdx, 0); } +JNIEXPORT jboolean JNICALL +Java_cn_com_basic_face_util_RtspFaceNative_addFaceCountListener(JNIEnv *env, jclass type, + jint cameraIdx, jstring host_, + jint port) { + const char *host = env->GetStringUTFChars(host_, 0); + + //TODO + + env->ReleaseStringUTFChars(host_, host); +} + void Java_cn_com_basic_face_util_RtspFaceNative_lockFace(JNIEnv *env, jclass clazz, jint cameraIdx) { @@ -540,20 +415,19 @@ 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); + size_t faceImagesSize =_faceImagesSize; + ret = cameraWrapper.faceCache.getFaceListImage(imgIdxes, _faceImages,faceImagesSize); #endif if (imgIdxes.size() == 0) { LOG_INFO << "No face image captured" << LOG_ENDL; - delete[] _faceImages; return nullptr; } @@ -564,7 +438,6 @@ 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"); @@ -619,7 +492,6 @@ if (imgIdxes.size() == 0) { LOG_INFO << "No face image captured" << LOG_ENDL; - delete[] _faceImages; return nullptr; } @@ -656,9 +528,6 @@ LOG_DEBUG << "imgIdx " << i << ":" << std::string(imgIdxes[i]) << LOG_ENDL; } - - delete[] _faceImages; - _faceImages = nullptr; return jobjArrayList_Imgs; } -- Gitblit v1.8.0