ImageBlurDetection.cpp 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. //
  2. // Created by 阳坤 on 2022/2/23.
  3. //
  4. #include "ImageBlurDetection.h"
  5. static jint Java_JNI_OpenCV_ImagePathBlurDetection(JNIEnv *env, jobject obj, jstring filepath) {
  6. const char *image_path = env->GetStringUTFChars(filepath, 0);
  7. int ret = image_blur_detection(image_path);
  8. env->ReleaseStringUTFChars(filepath, image_path);
  9. return ret;
  10. }
  11. static jstring Java_JNI_OpenCV_GetVideoSimilarityLists(JNIEnv *env, jobject obj, jstring filepath) {
  12. const char *image_path = env->GetStringUTFChars(filepath, 0);
  13. const char *ret_json = get_video_similarity_list(image_path);
  14. env->ReleaseStringUTFChars(filepath, image_path);
  15. if (ret_json) {
  16. jstring str = env->NewStringUTF(ret_json);
  17. free((void *) ret_json);
  18. ret_json = NULL;
  19. return str;
  20. }
  21. return NULL;
  22. }
  23. static jint
  24. Java_JNI_OpenCV_ImageBytesBlurDetection(JNIEnv *env, jobject obj, jbyteArray data) {
  25. jbyte *image_bytes = env->GetByteArrayElements(data, 0);
  26. int ret = bytes_blur_detection((uint8_t *) image_bytes, env->GetArrayLength(data));
  27. env->ReleaseByteArrayElements(data, image_bytes, 0);
  28. return ret;
  29. }
  30. static const JNINativeMethod configMethods[] = {
  31. {"blurDetectionFromImagePath", "(Ljava/lang/String;)I", (void **) Java_JNI_OpenCV_ImagePathBlurDetection},
  32. {"blurDetectionFromImageBytes", "([B)I", (void **) Java_JNI_OpenCV_ImageBytesBlurDetection},
  33. {"getVideoSimilarityLists", "(Ljava/lang/String;)Ljava/lang/String;", (void **) Java_JNI_OpenCV_GetVideoSimilarityLists},
  34. };
  35. /**
  36. * System.load(xxx.so) ִ
  37. * @param vm
  38. * @param pVoid
  39. * @return
  40. */
  41. int JNI_OnLoad(JavaVM *vm, void *pVoid) {
  42. JNIEnv *jniEnv;
  43. if (vm->GetEnv(reinterpret_cast<void **>(&jniEnv), JNI_VERSION_1_6) != JNI_OK) {
  44. printf(" JNI_OnLoad 1 error.\n");
  45. return JNI_ERR;
  46. }
  47. jclass jcls = jniEnv->FindClass(JAVA_JNI_CLASS_PATH);
  48. if (jniEnv->RegisterNatives(jcls, configMethods, NELEM(configMethods)) != JNI_OK) {
  49. printf(" JNI_OnLoad error.\n");
  50. return JNI_ERR;
  51. }
  52. printf(" JNI_OnLoad ok.\n");
  53. setupBreakpad("/datalog");
  54. return JNI_VERSION_1_6;
  55. };