#include "RtspNativeCodecJNI.h"
|
#include "CameraWrapper.h"
|
#include <logger.h>
|
#include <Logger/src/logger.hpp>
|
|
//#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);
|
|
CameraWrapper g_CameraWrappers[CAMERA_COUNT];
|
|
extern "C"
|
{
|
|
void Java_cn_com_basic_face_util_RtspFaceNative_init(JNIEnv *env, jclass clazz)
|
{
|
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));
|
}
|
}
|
|
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<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.rtspURL = utf8Uri;
|
env->ReleaseStringUTFChars(uri, utf8Uri);
|
|
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 = 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_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<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();
|
//}
|
|
}
|
