version:1.6.0

fix:
update:使用ocr进行坐标点击

app/src/main/AndroidManifest.xml

@@ -41,7 +41,12 @@
     <uses-permission android:name="android.permission.WRITE_CALL_LOG" />
     <uses-permission android:name="android.permission.SYSTEM_ALERT_WINDOW" />
     <uses-permission android:name="android.permission.READ_SMS" />
-    <uses-permission android:name="android.permission.RECEIVE_SMS" /><!-- 接收短信权限 -->
+    <!-- 接收短信权限 -->
+    <uses-permission android:name="android.permission.RECEIVE_SMS" />
+
+    <uses-permission android:name="android.permission.READ_FRAME_BUFFER" />
+    <uses-permission android:name="android.permission.CAPTURE_VIDEO_OUTPUT" />
+
 
     <!-- 允许访问网络，必选权限 -->
     <uses-permission android:name="android.permission.INTERNET" />
@@ -348,6 +353,7 @@
 
         <service
             android:name=".service.WeAccessibilityService"
+            android:foregroundServiceType="mediaProjection"
             android:label="@string/accessibility_service_label"
             android:permission="android.permission.BIND_ACCESSIBILITY_SERVICE">
             <intent-filter>

app/src/main/cpp/CMakeLists.txt

@@ -0,0 +1,117 @@
+# For more information about using CMake with Android Studio, read the
+# documentation: https://d.android.com/studio/projects/add-native-code.html
+
+# Sets the minimum version of CMake required to build the native library.
+
+cmake_minimum_required(VERSION 3.4.1)
+
+# Creates and names a library, sets it as either STATIC or SHARED, and provides
+# the relative paths to its source code. You can define multiple libraries, and
+# CMake builds them for you. Gradle automatically packages shared libraries with
+# your APK.
+
+set(PaddleLite_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../../../PaddleLite")
+include_directories(${PaddleLite_DIR}/cxx/include)
+
+set(OpenCV_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../../../OpenCV/sdk/native/jni")
+message(STATUS "opencv dir: ${OpenCV_DIR}")
+find_package(OpenCV REQUIRED)
+message(STATUS "OpenCV libraries: ${OpenCV_LIBS}")
+include_directories(${OpenCV_INCLUDE_DIRS})
+aux_source_directory(. SOURCES)
+set(CMAKE_CXX_FLAGS
+        "${CMAKE_CXX_FLAGS} -ffast-math -Ofast -Os"
+        )
+set(CMAKE_CXX_FLAGS
+        "${CMAKE_CXX_FLAGS} -fvisibility=hidden -fvisibility-inlines-hidden -fdata-sections -ffunction-sections"
+        )
+set(CMAKE_SHARED_LINKER_FLAGS
+        "${CMAKE_SHARED_LINKER_FLAGS} -Wl,--gc-sections -Wl,-z,nocopyreloc")
+
+add_library(
+        # Sets the name of the library.
+        Native
+        # Sets the library as a shared library.
+        SHARED
+        # Provides a relative path to your source file(s).
+        ${SOURCES})
+
+find_library(
+        # Sets the name of the path variable.
+        log-lib
+        # Specifies the name of the NDK library that you want CMake to locate.
+        log)
+
+add_library(
+        # Sets the name of the library.
+        paddle_light_api_shared
+        # Sets the library as a shared library.
+        SHARED
+        # Provides a relative path to your source file(s).
+        IMPORTED)
+
+set_target_properties(
+        # Specifies the target library.
+        paddle_light_api_shared
+        # Specifies the parameter you want to define.
+        PROPERTIES
+        IMPORTED_LOCATION
+        ${PaddleLite_DIR}/cxx/libs/${ANDROID_ABI}/libpaddle_light_api_shared.so
+        # Provides the path to the library you want to import.
+)
+
+
+# Specifies libraries CMake should link to your target library. You can link
+# multiple libraries, such as libraries you define in this build script,
+# prebuilt third-party libraries, or system libraries.
+
+target_link_libraries(
+        # Specifies the target library.
+        Native
+        paddle_light_api_shared
+        ${OpenCV_LIBS}
+        GLESv2
+        EGL
+        jnigraphics
+        ${log-lib}
+)
+
+add_custom_command(
+        TARGET Native
+        POST_BUILD
+        COMMAND
+        ${CMAKE_COMMAND} -E copy
+        ${PaddleLite_DIR}/cxx/libs/${ANDROID_ABI}/libc++_shared.so
+        ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/libc++_shared.so)
+
+add_custom_command(
+        TARGET Native
+        POST_BUILD
+        COMMAND
+        ${CMAKE_COMMAND} -E copy
+        ${PaddleLite_DIR}/cxx/libs/${ANDROID_ABI}/libpaddle_light_api_shared.so
+        ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/libpaddle_light_api_shared.so)
+
+add_custom_command(
+        TARGET Native
+        POST_BUILD
+        COMMAND
+        ${CMAKE_COMMAND} -E copy
+        ${PaddleLite_DIR}/cxx/libs/${ANDROID_ABI}/libhiai.so
+        ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/libhiai.so)
+
+add_custom_command(
+        TARGET Native
+        POST_BUILD
+        COMMAND
+        ${CMAKE_COMMAND} -E copy
+        ${PaddleLite_DIR}/cxx/libs/${ANDROID_ABI}/libhiai_ir.so
+        ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/libhiai_ir.so)
+
+add_custom_command(
+        TARGET Native
+        POST_BUILD
+        COMMAND
+        ${CMAKE_COMMAND} -E copy
+        ${PaddleLite_DIR}/cxx/libs/${ANDROID_ABI}/libhiai_ir_build.so
+        ${CMAKE_LIBRARY_OUTPUT_DIRECTORY}/libhiai_ir_build.so)

app/src/main/cpp/common.h

@@ -0,0 +1,37 @@
+//
+// Created by fu on 4/25/18.
+//
+
+#pragma once
+#import <numeric>
+#import <vector>
+
+#ifdef __ANDROID__
+
+#include <android/log.h>
+
+#define LOG_TAG "OCR_NDK"
+
+#define LOGI(...) __android_log_print(ANDROID_LOG_INFO, LOG_TAG, __VA_ARGS__)
+#define LOGW(...) __android_log_print(ANDROID_LOG_WARN, LOG_TAG, __VA_ARGS__)
+#define LOGE(...) __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, __VA_ARGS__)
+#else
+#include <stdio.h>
+#define LOGI(format, ...)                                                      \
+  fprintf(stdout, "[" LOG_TAG "]" format "\n", ##__VA_ARGS__)
+#define LOGW(format, ...)                                                      \
+  fprintf(stdout, "[" LOG_TAG "]" format "\n", ##__VA_ARGS__)
+#define LOGE(format, ...)                                                      \
+  fprintf(stderr, "[" LOG_TAG "]Error: " format "\n", ##__VA_ARGS__)
+#endif
+
+enum RETURN_CODE { RETURN_OK = 0 };
+
+enum NET_TYPE { NET_OCR = 900100, NET_OCR_INTERNAL = 991008 };
+
+template <typename T> inline T product(const std::vector<T> &vec) {
+  if (vec.empty()) {
+    return 0;
+  }
+  return std::accumulate(vec.begin(), vec.end(), 1, std::multiplies<T>());
+}

app/src/main/cpp/native.cpp

@@ -0,0 +1,120 @@
+//
+// Created by fujiayi on 2020/7/5.
+//
+
+#include "native.h"
+#include "ocr_ppredictor.h"
+#include <algorithm>
+#include <paddle_api.h>
+#include <string>
+
+static paddle::lite_api::PowerMode str_to_cpu_mode(const std::string &cpu_mode);
+
+extern "C" JNIEXPORT jlong JNICALL
+Java_com_xxpatx_os_ocr_paddle_OCRPredictorNative_init(
+    JNIEnv *env, jobject thiz, jstring j_det_model_path,
+    jstring j_rec_model_path, jstring j_cls_model_path, jint j_use_opencl,
+    jint j_thread_num, jstring j_cpu_mode) {
+  std::string det_model_path = jstring_to_cpp_string(env, j_det_model_path);
+  std::string rec_model_path = jstring_to_cpp_string(env, j_rec_model_path);
+  std::string cls_model_path = jstring_to_cpp_string(env, j_cls_model_path);
+  int thread_num = j_thread_num;
+  std::string cpu_mode = jstring_to_cpp_string(env, j_cpu_mode);
+  ppredictor::OCR_Config conf;
+  conf.use_opencl = j_use_opencl;
+  conf.thread_num = thread_num;
+  conf.mode = str_to_cpu_mode(cpu_mode);
+  ppredictor::OCR_PPredictor *orc_predictor =
+      new ppredictor::OCR_PPredictor{conf};
+  orc_predictor->init_from_file(det_model_path, rec_model_path, cls_model_path);
+  return reinterpret_cast<jlong>(orc_predictor);
+}
+
+/**
+ * "LITE_POWER_HIGH" convert to paddle::lite_api::LITE_POWER_HIGH
+ * @param cpu_mode
+ * @return
+ */
+static paddle::lite_api::PowerMode
+str_to_cpu_mode(const std::string &cpu_mode) {
+  static std::map<std::string, paddle::lite_api::PowerMode> cpu_mode_map{
+      {"LITE_POWER_HIGH", paddle::lite_api::LITE_POWER_HIGH},
+      {"LITE_POWER_LOW", paddle::lite_api::LITE_POWER_HIGH},
+      {"LITE_POWER_FULL", paddle::lite_api::LITE_POWER_FULL},
+      {"LITE_POWER_NO_BIND", paddle::lite_api::LITE_POWER_NO_BIND},
+      {"LITE_POWER_RAND_HIGH", paddle::lite_api::LITE_POWER_RAND_HIGH},
+      {"LITE_POWER_RAND_LOW", paddle::lite_api::LITE_POWER_RAND_LOW}};
+  std::string upper_key;
+  std::transform(cpu_mode.cbegin(), cpu_mode.cend(), upper_key.begin(),
+                 ::toupper);
+  auto index = cpu_mode_map.find(upper_key.c_str());
+  if (index == cpu_mode_map.end()) {
+    LOGE("cpu_mode not found %s", upper_key.c_str());
+    return paddle::lite_api::LITE_POWER_HIGH;
+  } else {
+    return index->second;
+  }
+}
+
+extern "C" JNIEXPORT jfloatArray JNICALL
+Java_com_xxpatx_os_ocr_paddle_OCRPredictorNative_forward(
+    JNIEnv *env, jobject thiz, jlong java_pointer, jobject original_image,
+    jint j_max_size_len, jint j_run_det, jint j_run_cls, jint j_run_rec) {
+  LOGI("begin to run native forward");
+  if (java_pointer == 0) {
+    LOGE("JAVA pointer is NULL");
+    return cpp_array_to_jfloatarray(env, nullptr, 0);
+  }
+
+  cv::Mat origin = bitmap_to_cv_mat(env, original_image);
+  if (origin.size == 0) {
+    LOGE("origin bitmap cannot convert to CV Mat");
+    return cpp_array_to_jfloatarray(env, nullptr, 0);
+  }
+
+  int max_size_len = j_max_size_len;
+  int run_det = j_run_det;
+  int run_cls = j_run_cls;
+  int run_rec = j_run_rec;
+
+  ppredictor::OCR_PPredictor *ppredictor =
+      (ppredictor::OCR_PPredictor *)java_pointer;
+  std::vector<int64_t> dims_arr;
+  std::vector<ppredictor::OCRPredictResult> results =
+      ppredictor->infer_ocr(origin, max_size_len, run_det, run_cls, run_rec);
+  LOGI("infer_ocr finished with boxes %ld", results.size());
+
+  // 这里将std::vector<ppredictor::OCRPredictResult> 序列化成
+  // float数组，传输到java层再反序列化
+  std::vector<float> float_arr;
+  for (const ppredictor::OCRPredictResult &r : results) {
+    float_arr.push_back(r.points.size());
+    float_arr.push_back(r.word_index.size());
+    float_arr.push_back(r.score);
+    // add det point
+    for (const std::vector<int> &point : r.points) {
+      float_arr.push_back(point.at(0));
+      float_arr.push_back(point.at(1));
+    }
+    // add rec word idx
+    for (int index : r.word_index) {
+      float_arr.push_back(index);
+    }
+    // add cls result
+    float_arr.push_back(r.cls_label);
+    float_arr.push_back(r.cls_score);
+  }
+  return cpp_array_to_jfloatarray(env, float_arr.data(), float_arr.size());
+}
+
+extern "C" JNIEXPORT void JNICALL
+Java_com_xxpatx_os_ocr_paddle_OCRPredictorNative_release(
+    JNIEnv *env, jobject thiz, jlong java_pointer) {
+  if (java_pointer == 0) {
+    LOGE("JAVA pointer is NULL");
+    return;
+  }
+  ppredictor::OCR_PPredictor *ppredictor =
+      (ppredictor::OCR_PPredictor *)java_pointer;
+  delete ppredictor;
+}

app/src/main/cpp/native.h

@@ -0,0 +1,137 @@
+//
+// Created by fujiayi on 2020/7/5.
+//
+
+#pragma once
+
+#include "common.h"
+#include <android/bitmap.h>
+#include <jni.h>
+#include <opencv2/opencv.hpp>
+#include <string>
+#include <vector>
+
+inline std::string jstring_to_cpp_string(JNIEnv *env, jstring jstr) {
+  // In java, a unicode char will be encoded using 2 bytes (utf16).
+  // so jstring will contain characters utf16. std::string in c++ is
+  // essentially a string of bytes, not characters, so if we want to
+  // pass jstring from JNI to c++, we have convert utf16 to bytes.
+  if (!jstr) {
+    return "";
+  }
+  const jclass stringClass = env->GetObjectClass(jstr);
+  const jmethodID getBytes =
+      env->GetMethodID(stringClass, "getBytes", "(Ljava/lang/String;)[B");
+  const jbyteArray stringJbytes = (jbyteArray)env->CallObjectMethod(
+      jstr, getBytes, env->NewStringUTF("UTF-8"));
+
+  size_t length = (size_t)env->GetArrayLength(stringJbytes);
+  jbyte *pBytes = env->GetByteArrayElements(stringJbytes, NULL);
+
+  std::string ret = std::string(reinterpret_cast<char *>(pBytes), length);
+  env->ReleaseByteArrayElements(stringJbytes, pBytes, JNI_ABORT);
+
+  env->DeleteLocalRef(stringJbytes);
+  env->DeleteLocalRef(stringClass);
+  return ret;
+}
+
+inline jstring cpp_string_to_jstring(JNIEnv *env, std::string str) {
+  auto *data = str.c_str();
+  jclass strClass = env->FindClass("java/lang/String");
+  jmethodID strClassInitMethodID =
+      env->GetMethodID(strClass, "<init>", "([BLjava/lang/String;)V");
+
+  jbyteArray bytes = env->NewByteArray(strlen(data));
+  env->SetByteArrayRegion(bytes, 0, strlen(data),
+                          reinterpret_cast<const jbyte *>(data));
+
+  jstring encoding = env->NewStringUTF("UTF-8");
+  jstring res = (jstring)(env->NewObject(strClass, strClassInitMethodID, bytes,
+                                         encoding));
+
+  env->DeleteLocalRef(strClass);
+  env->DeleteLocalRef(encoding);
+  env->DeleteLocalRef(bytes);
+
+  return res;
+}
+
+inline jfloatArray cpp_array_to_jfloatarray(JNIEnv *env, const float *buf,
+                                            int64_t len) {
+  if (len == 0) {
+    return env->NewFloatArray(0);
+  }
+  jfloatArray result = env->NewFloatArray(len);
+  env->SetFloatArrayRegion(result, 0, len, buf);
+  return result;
+}
+
+inline jintArray cpp_array_to_jintarray(JNIEnv *env, const int *buf,
+                                        int64_t len) {
+  jintArray result = env->NewIntArray(len);
+  env->SetIntArrayRegion(result, 0, len, buf);
+  return result;
+}
+
+inline jbyteArray cpp_array_to_jbytearray(JNIEnv *env, const int8_t *buf,
+                                          int64_t len) {
+  jbyteArray result = env->NewByteArray(len);
+  env->SetByteArrayRegion(result, 0, len, buf);
+  return result;
+}
+
+inline jlongArray int64_vector_to_jlongarray(JNIEnv *env,
+                                             const std::vector<int64_t> &vec) {
+  jlongArray result = env->NewLongArray(vec.size());
+  jlong *buf = new jlong[vec.size()];
+  for (size_t i = 0; i < vec.size(); ++i) {
+    buf[i] = (jlong)vec[i];
+  }
+  env->SetLongArrayRegion(result, 0, vec.size(), buf);
+  delete[] buf;
+  return result;
+}
+
+inline std::vector<int64_t> jlongarray_to_int64_vector(JNIEnv *env,
+                                                       jlongArray data) {
+  int data_size = env->GetArrayLength(data);
+  jlong *data_ptr = env->GetLongArrayElements(data, nullptr);
+  std::vector<int64_t> data_vec(data_ptr, data_ptr + data_size);
+  env->ReleaseLongArrayElements(data, data_ptr, 0);
+  return data_vec;
+}
+
+inline std::vector<float> jfloatarray_to_float_vector(JNIEnv *env,
+                                                      jfloatArray data) {
+  int data_size = env->GetArrayLength(data);
+  jfloat *data_ptr = env->GetFloatArrayElements(data, nullptr);
+  std::vector<float> data_vec(data_ptr, data_ptr + data_size);
+  env->ReleaseFloatArrayElements(data, data_ptr, 0);
+  return data_vec;
+}
+
+inline cv::Mat bitmap_to_cv_mat(JNIEnv *env, jobject bitmap) {
+  AndroidBitmapInfo info;
+  int result = AndroidBitmap_getInfo(env, bitmap, &info);
+  if (result != ANDROID_BITMAP_RESULT_SUCCESS) {
+    LOGE("AndroidBitmap_getInfo failed, result: %d", result);
+    return cv::Mat{};
+  }
+  if (info.format != ANDROID_BITMAP_FORMAT_RGBA_8888) {
+    LOGE("Bitmap format is not RGBA_8888 !");
+    return cv::Mat{};
+  }
+  unsigned char *srcData = NULL;
+  AndroidBitmap_lockPixels(env, bitmap, (void **)&srcData);
+  cv::Mat mat = cv::Mat::zeros(info.height, info.width, CV_8UC4);
+  memcpy(mat.data, srcData, info.height * info.width * 4);
+  AndroidBitmap_unlockPixels(env, bitmap);
+  cv::cvtColor(mat, mat, cv::COLOR_RGBA2BGR);
+  /**
+  if (!cv::imwrite("/sdcard/1/copy.jpg", mat)){
+      LOGE("Write image failed " );
+  }
+   */
+  return mat;
+}

app/src/main/cpp/ocr_clipper.hpp

@@ -0,0 +1,540 @@
+/*******************************************************************************
+ *                                                                              *
+ * Author    :  Angus Johnson * Version   :  6.4.2 * Date      :  27 February
+ *2017                                                * Website   :
+ *http://www.angusj.com                                           * Copyright :
+ *Angus Johnson 2010-2017                                         *
+ *                                                                              *
+ * License: * Use, modification & distribution is subject to Boost Software
+ *License Ver 1. * http://www.boost.org/LICENSE_1_0.txt *
+ *                                                                              *
+ * Attributions: * The code in this library is an extension of Bala Vatti's
+ *clipping algorithm: * "A generic solution to polygon clipping" *
+ * Communications of the ACM, Vol 35, Issue 7 (July 1992) pp 56-63. *
+ * http://portal.acm.org/citation.cfm?id=129906 *
+ *                                                                              *
+ * Computer graphics and geometric modeling: implementation and algorithms * By
+ *Max K. Agoston                                                            *
+ * Springer; 1 edition (January 4, 2005) *
+ * http://books.google.com/books?q=vatti+clipping+agoston *
+ *                                                                              *
+ * See also: * "Polygon Offsetting by Computing Winding Numbers" * Paper no.
+ *DETC2005-85513 pp. 565-575                                         * ASME 2005
+ *International Design Engineering Technical Conferences             * and
+ *Computers and Information in Engineering Conference (IDETC/CIE2005)      *
+ * September 24-28, 2005 , Long Beach, California, USA *
+ * http://www.me.berkeley.edu/~mcmains/pubs/DAC05OffsetPolygon.pdf *
+ *                                                                              *
+ *******************************************************************************/
+
+#ifndef clipper_hpp
+#define clipper_hpp
+
+#define CLIPPER_VERSION "6.4.2"
+
+// use_int32: When enabled 32bit ints are used instead of 64bit ints. This
+// improve performance but coordinate values are limited to the range +/- 46340
+//#define use_int32
+
+// use_xyz: adds a Z member to IntPoint. Adds a minor cost to performance.
+//#define use_xyz
+
+// use_lines: Enables line clipping. Adds a very minor cost to performance.
+#define use_lines
+
+// use_deprecated: Enables temporary support for the obsolete functions
+//#define use_deprecated
+
+#include <cstdlib>
+#include <cstring>
+#include <functional>
+#include <list>
+#include <ostream>
+#include <queue>
+#include <set>
+#include <stdexcept>
+#include <vector>
+
+namespace ClipperLib {
+
+enum ClipType { ctIntersection, ctUnion, ctDifference, ctXor };
+enum PolyType { ptSubject, ptClip };
+// By far the most widely used winding rules for polygon filling are
+// EvenOdd & NonZero (GDI, GDI+, XLib, OpenGL, Cairo, AGG, Quartz, SVG, Gr32)
+// Others rules include Positive, Negative and ABS_GTR_EQ_TWO (only in OpenGL)
+// see http://glprogramming.com/red/chapter11.html
+enum PolyFillType { pftEvenOdd, pftNonZero, pftPositive, pftNegative };
+
+#ifdef use_int32
+typedef int cInt;
+static cInt const loRange = 0x7FFF;
+static cInt const hiRange = 0x7FFF;
+#else
+typedef signed long long cInt;
+static cInt const loRange = 0x3FFFFFFF;
+static cInt const hiRange = 0x3FFFFFFFFFFFFFFFLL;
+typedef signed long long long64; // used by Int128 class
+typedef unsigned long long ulong64;
+
+#endif
+
+struct IntPoint {
+  cInt X;
+  cInt Y;
+#ifdef use_xyz
+  cInt Z;
+  IntPoint(cInt x = 0, cInt y = 0, cInt z = 0) : X(x), Y(y), Z(z){};
+#else
+
+  IntPoint(cInt x = 0, cInt y = 0) : X(x), Y(y){};
+#endif
+
+  friend inline bool operator==(const IntPoint &a, const IntPoint &b) {
+    return a.X == b.X && a.Y == b.Y;
+  }
+
+  friend inline bool operator!=(const IntPoint &a, const IntPoint &b) {
+    return a.X != b.X || a.Y != b.Y;
+  }
+};
+//------------------------------------------------------------------------------
+
+typedef std::vector<IntPoint> Path;
+typedef std::vector<Path> Paths;
+
+inline Path &operator<<(Path &poly, const IntPoint &p) {
+  poly.push_back(p);
+  return poly;
+}
+
+inline Paths &operator<<(Paths &polys, const Path &p) {
+  polys.push_back(p);
+  return polys;
+}
+
+std::ostream &operator<<(std::ostream &s, const IntPoint &p);
+
+std::ostream &operator<<(std::ostream &s, const Path &p);
+
+std::ostream &operator<<(std::ostream &s, const Paths &p);
+
+struct DoublePoint {
+  double X;
+  double Y;
+
+  DoublePoint(double x = 0, double y = 0) : X(x), Y(y) {}
+
+  DoublePoint(IntPoint ip) : X((double)ip.X), Y((double)ip.Y) {}
+};
+//------------------------------------------------------------------------------
+
+#ifdef use_xyz
+typedef void (*ZFillCallback)(IntPoint &e1bot, IntPoint &e1top, IntPoint &e2bot,
+                              IntPoint &e2top, IntPoint &pt);
+#endif
+
+enum InitOptions {
+  ioReverseSolution = 1,
+  ioStrictlySimple = 2,
+  ioPreserveCollinear = 4
+};
+enum JoinType { jtSquare, jtRound, jtMiter };
+enum EndType {
+  etClosedPolygon,
+  etClosedLine,
+  etOpenButt,
+  etOpenSquare,
+  etOpenRound
+};
+
+class PolyNode;
+
+typedef std::vector<PolyNode *> PolyNodes;
+
+class PolyNode {
+public:
+  PolyNode();
+
+  virtual ~PolyNode(){};
+  Path Contour;
+  PolyNodes Children;
+  PolyNode *Parent;
+
+  PolyNode *GetNext() const;
+
+  bool IsHole() const;
+
+  bool IsOpen() const;
+
+  int ChildCount() const;
+
+private:
+  // PolyNode& operator =(PolyNode& other);
+  unsigned Index; // node index in Parent.Children
+  bool m_IsOpen;
+  JoinType m_jointype;
+  EndType m_endtype;
+
+  PolyNode *GetNextSiblingUp() const;
+
+  void AddChild(PolyNode &child);
+
+  friend class Clipper; // to access Index
+  friend class ClipperOffset;
+};
+
+class PolyTree : public PolyNode {
+public:
+  ~PolyTree() { Clear(); };
+
+  PolyNode *GetFirst() const;
+
+  void Clear();
+
+  int Total() const;
+
+private:
+  // PolyTree& operator =(PolyTree& other);
+  PolyNodes AllNodes;
+
+  friend class Clipper; // to access AllNodes
+};
+
+bool Orientation(const Path &poly);
+
+double Area(const Path &poly);
+
+int PointInPolygon(const IntPoint &pt, const Path &path);
+
+void SimplifyPolygon(const Path &in_poly, Paths &out_polys,
+                     PolyFillType fillType = pftEvenOdd);
+
+void SimplifyPolygons(const Paths &in_polys, Paths &out_polys,
+                      PolyFillType fillType = pftEvenOdd);
+
+void SimplifyPolygons(Paths &polys, PolyFillType fillType = pftEvenOdd);
+
+void CleanPolygon(const Path &in_poly, Path &out_poly, double distance = 1.415);
+
+void CleanPolygon(Path &poly, double distance = 1.415);
+
+void CleanPolygons(const Paths &in_polys, Paths &out_polys,
+                   double distance = 1.415);
+
+void CleanPolygons(Paths &polys, double distance = 1.415);
+
+void MinkowskiSum(const Path &pattern, const Path &path, Paths &solution,
+                  bool pathIsClosed);
+
+void MinkowskiSum(const Path &pattern, const Paths &paths, Paths &solution,
+                  bool pathIsClosed);
+
+void MinkowskiDiff(const Path &poly1, const Path &poly2, Paths &solution);
+
+void PolyTreeToPaths(const PolyTree &polytree, Paths &paths);
+
+void ClosedPathsFromPolyTree(const PolyTree &polytree, Paths &paths);
+
+void OpenPathsFromPolyTree(PolyTree &polytree, Paths &paths);
+
+void ReversePath(Path &p);
+
+void ReversePaths(Paths &p);
+
+struct IntRect {
+  cInt left;
+  cInt top;
+  cInt right;
+  cInt bottom;
+};
+
+// enums that are used internally ...
+enum EdgeSide { esLeft = 1, esRight = 2 };
+
+// forward declarations (for stuff used internally) ...
+struct TEdge;
+struct IntersectNode;
+struct LocalMinimum;
+struct OutPt;
+struct OutRec;
+struct Join;
+
+typedef std::vector<OutRec *> PolyOutList;
+typedef std::vector<TEdge *> EdgeList;
+typedef std::vector<Join *> JoinList;
+typedef std::vector<IntersectNode *> IntersectList;
+
+//------------------------------------------------------------------------------
+
+// ClipperBase is the ancestor to the Clipper class. It should not be
+// instantiated directly. This class simply abstracts the conversion of sets of
+// polygon coordinates into edge objects that are stored in a LocalMinima list.
+class ClipperBase {
+public:
+  ClipperBase();
+
+  virtual ~ClipperBase();
+
+  virtual bool AddPath(const Path &pg, PolyType PolyTyp, bool Closed);
+
+  bool AddPaths(const Paths &ppg, PolyType PolyTyp, bool Closed);
+
+  virtual void Clear();
+
+  IntRect GetBounds();
+
+  bool PreserveCollinear() { return m_PreserveCollinear; };
+
+  void PreserveCollinear(bool value) { m_PreserveCollinear = value; };
+
+protected:
+  void DisposeLocalMinimaList();
+
+  TEdge *AddBoundsToLML(TEdge *e, bool IsClosed);
+
+  virtual void Reset();
+
+  TEdge *ProcessBound(TEdge *E, bool IsClockwise);
+
+  void InsertScanbeam(const cInt Y);
+
+  bool PopScanbeam(cInt &Y);
+
+  bool LocalMinimaPending();
+
+  bool PopLocalMinima(cInt Y, const LocalMinimum *&locMin);
+
+  OutRec *CreateOutRec();
+
+  void DisposeAllOutRecs();
+
+  void DisposeOutRec(PolyOutList::size_type index);
+
+  void SwapPositionsInAEL(TEdge *edge1, TEdge *edge2);
+
+  void DeleteFromAEL(TEdge *e);
+
+  void UpdateEdgeIntoAEL(TEdge *&e);
+
+  typedef std::vector<LocalMinimum> MinimaList;
+  MinimaList::iterator m_CurrentLM;
+  MinimaList m_MinimaList;
+
+  bool m_UseFullRange;
+  EdgeList m_edges;
+  bool m_PreserveCollinear;
+  bool m_HasOpenPaths;
+  PolyOutList m_PolyOuts;
+  TEdge *m_ActiveEdges;
+
+  typedef std::priority_queue<cInt> ScanbeamList;
+  ScanbeamList m_Scanbeam;
+};
+//------------------------------------------------------------------------------
+
+class Clipper : public virtual ClipperBase {
+public:
+  Clipper(int initOptions = 0);
+
+  bool Execute(ClipType clipType, Paths &solution,
+               PolyFillType fillType = pftEvenOdd);
+
+  bool Execute(ClipType clipType, Paths &solution, PolyFillType subjFillType,
+               PolyFillType clipFillType);
+
+  bool Execute(ClipType clipType, PolyTree &polytree,
+               PolyFillType fillType = pftEvenOdd);
+
+  bool Execute(ClipType clipType, PolyTree &polytree, PolyFillType subjFillType,
+               PolyFillType clipFillType);
+
+  bool ReverseSolution() { return m_ReverseOutput; };
+
+  void ReverseSolution(bool value) { m_ReverseOutput = value; };
+
+  bool StrictlySimple() { return m_StrictSimple; };
+
+  void StrictlySimple(bool value) { m_StrictSimple = value; };
+// set the callback function for z value filling on intersections (otherwise Z
+// is 0)
+#ifdef use_xyz
+  void ZFillFunction(ZFillCallback zFillFunc);
+#endif
+protected:
+  virtual bool ExecuteInternal();
+
+private:
+  JoinList m_Joins;
+  JoinList m_GhostJoins;
+  IntersectList m_IntersectList;
+  ClipType m_ClipType;
+  typedef std::list<cInt> MaximaList;
+  MaximaList m_Maxima;
+  TEdge *m_SortedEdges;
+  bool m_ExecuteLocked;
+  PolyFillType m_ClipFillType;
+  PolyFillType m_SubjFillType;
+  bool m_ReverseOutput;
+  bool m_UsingPolyTree;
+  bool m_StrictSimple;
+#ifdef use_xyz
+  ZFillCallback m_ZFill; // custom callback
+#endif
+
+  void SetWindingCount(TEdge &edge);
+
+  bool IsEvenOddFillType(const TEdge &edge) const;
+
+  bool IsEvenOddAltFillType(const TEdge &edge) const;
+
+  void InsertLocalMinimaIntoAEL(const cInt botY);
+
+  void InsertEdgeIntoAEL(TEdge *edge, TEdge *startEdge);
+
+  void AddEdgeToSEL(TEdge *edge);
+
+  bool PopEdgeFromSEL(TEdge *&edge);
+
+  void CopyAELToSEL();
+
+  void DeleteFromSEL(TEdge *e);
+
+  void SwapPositionsInSEL(TEdge *edge1, TEdge *edge2);
+
+  bool IsContributing(const TEdge &edge) const;
+
+  bool IsTopHorz(const cInt XPos);
+
+  void DoMaxima(TEdge *e);
+
+  void ProcessHorizontals();
+
+  void ProcessHorizontal(TEdge *horzEdge);
+
+  void AddLocalMaxPoly(TEdge *e1, TEdge *e2, const IntPoint &pt);
+
+  OutPt *AddLocalMinPoly(TEdge *e1, TEdge *e2, const IntPoint &pt);
+
+  OutRec *GetOutRec(int idx);
+
+  void AppendPolygon(TEdge *e1, TEdge *e2);
+
+  void IntersectEdges(TEdge *e1, TEdge *e2, IntPoint &pt);
+
+  OutPt *AddOutPt(TEdge *e, const IntPoint &pt);
+
+  OutPt *GetLastOutPt(TEdge *e);
+
+  bool ProcessIntersections(const cInt topY);
+
+  void BuildIntersectList(const cInt topY);
+
+  void ProcessIntersectList();
+
+  void ProcessEdgesAtTopOfScanbeam(const cInt topY);
+
+  void BuildResult(Paths &polys);
+
+  void BuildResult2(PolyTree &polytree);
+
+  void SetHoleState(TEdge *e, OutRec *outrec);
+
+  void DisposeIntersectNodes();
+
+  bool FixupIntersectionOrder();
+
+  void FixupOutPolygon(OutRec &outrec);
+
+  void FixupOutPolyline(OutRec &outrec);
+
+  bool IsHole(TEdge *e);
+
+  bool FindOwnerFromSplitRecs(OutRec &outRec, OutRec *&currOrfl);
+
+  void FixHoleLinkage(OutRec &outrec);
+
+  void AddJoin(OutPt *op1, OutPt *op2, const IntPoint offPt);
+
+  void ClearJoins();
+
+  void ClearGhostJoins();
+
+  void AddGhostJoin(OutPt *op, const IntPoint offPt);
+
+  bool JoinPoints(Join *j, OutRec *outRec1, OutRec *outRec2);
+
+  void JoinCommonEdges();
+
+  void DoSimplePolygons();
+
+  void FixupFirstLefts1(OutRec *OldOutRec, OutRec *NewOutRec);
+
+  void FixupFirstLefts2(OutRec *InnerOutRec, OutRec *OuterOutRec);
+
+  void FixupFirstLefts3(OutRec *OldOutRec, OutRec *NewOutRec);
+
+#ifdef use_xyz
+  void SetZ(IntPoint &pt, TEdge &e1, TEdge &e2);
+#endif
+};
+//------------------------------------------------------------------------------
+
+class ClipperOffset {
+public:
+  ClipperOffset(double miterLimit = 2.0, double roundPrecision = 0.25);
+
+  ~ClipperOffset();
+
+  void AddPath(const Path &path, JoinType joinType, EndType endType);
+
+  void AddPaths(const Paths &paths, JoinType joinType, EndType endType);
+
+  void Execute(Paths &solution, double delta);
+
+  void Execute(PolyTree &solution, double delta);
+
+  void Clear();
+
+  double MiterLimit;
+  double ArcTolerance;
+
+private:
+  Paths m_destPolys;
+  Path m_srcPoly;
+  Path m_destPoly;
+  std::vector<DoublePoint> m_normals;
+  double m_delta, m_sinA, m_sin, m_cos;
+  double m_miterLim, m_StepsPerRad;
+  IntPoint m_lowest;
+  PolyNode m_polyNodes;
+
+  void FixOrientations();
+
+  void DoOffset(double delta);
+
+  void OffsetPoint(int j, int &k, JoinType jointype);
+
+  void DoSquare(int j, int k);
+
+  void DoMiter(int j, int k, double r);
+
+  void DoRound(int j, int k);
+};
+//------------------------------------------------------------------------------
+
+class clipperException : public std::exception {
+public:
+  clipperException(const char *description) : m_descr(description) {}
+
+  virtual ~clipperException() throw() {}
+
+  virtual const char *what() const throw() { return m_descr.c_str(); }
+
+private:
+  std::string m_descr;
+};
+//------------------------------------------------------------------------------
+
+} // namespace ClipperLib
+
+#endif // clipper_hpp

app/src/main/cpp/ocr_cls_process.cpp

@@ -0,0 +1,45 @@
+// Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "ocr_cls_process.h"
+#include <cmath>
+#include <cstring>
+#include <fstream>
+#include <iostream>
+#include <vector>
+
+const std::vector<int> CLS_IMAGE_SHAPE = {3, 48, 192};
+
+cv::Mat cls_resize_img(const cv::Mat &img) {
+  int imgC = CLS_IMAGE_SHAPE[0];
+  int imgW = CLS_IMAGE_SHAPE[2];
+  int imgH = CLS_IMAGE_SHAPE[1];
+
+  float ratio = float(img.cols) / float(img.rows);
+  int resize_w = 0;
+  if (ceilf(imgH * ratio) > imgW)
+    resize_w = imgW;
+  else
+    resize_w = int(ceilf(imgH * ratio));
+
+  cv::Mat resize_img;
+  cv::resize(img, resize_img, cv::Size(resize_w, imgH), 0.f, 0.f,
+             cv::INTER_CUBIC);
+
+  if (resize_w < imgW) {
+    cv::copyMakeBorder(resize_img, resize_img, 0, 0, 0, int(imgW - resize_w),
+                       cv::BORDER_CONSTANT, {0, 0, 0});
+  }
+  return resize_img;
+}

app/src/main/cpp/ocr_cls_process.h

@@ -0,0 +1,23 @@
+// Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+
+#include "common.h"
+#include <opencv2/opencv.hpp>
+#include <vector>
+
+extern const std::vector<int> CLS_IMAGE_SHAPE;
+
+cv::Mat cls_resize_img(const cv::Mat &img);

app/src/main/cpp/ocr_crnn_process.cpp

@@ -0,0 +1,141 @@
+// Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "ocr_crnn_process.h"
+#include <cmath>
+#include <cstring>
+#include <fstream>
+#include <iostream>
+#include <vector>
+
+const std::string CHARACTER_TYPE = "ch";
+const int MAX_DICT_LENGTH = 6624;
+const std::vector<int> REC_IMAGE_SHAPE = {3, 32, 320};
+
+static cv::Mat crnn_resize_norm_img(cv::Mat img, float wh_ratio) {
+  int imgC = REC_IMAGE_SHAPE[0];
+  int imgW = REC_IMAGE_SHAPE[2];
+  int imgH = REC_IMAGE_SHAPE[1];
+
+  if (CHARACTER_TYPE == "ch")
+    imgW = int(32 * wh_ratio);
+
+  float ratio = float(img.cols) / float(img.rows);
+  int resize_w = 0;
+  if (ceilf(imgH * ratio) > imgW)
+    resize_w = imgW;
+  else
+    resize_w = int(ceilf(imgH * ratio));
+  cv::Mat resize_img;
+  cv::resize(img, resize_img, cv::Size(resize_w, imgH), 0.f, 0.f,
+             cv::INTER_CUBIC);
+
+  resize_img.convertTo(resize_img, CV_32FC3, 1 / 255.f);
+
+  for (int h = 0; h < resize_img.rows; h++) {
+    for (int w = 0; w < resize_img.cols; w++) {
+      resize_img.at<cv::Vec3f>(h, w)[0] =
+          (resize_img.at<cv::Vec3f>(h, w)[0] - 0.5) * 2;
+      resize_img.at<cv::Vec3f>(h, w)[1] =
+          (resize_img.at<cv::Vec3f>(h, w)[1] - 0.5) * 2;
+      resize_img.at<cv::Vec3f>(h, w)[2] =
+          (resize_img.at<cv::Vec3f>(h, w)[2] - 0.5) * 2;
+    }
+  }
+
+  cv::Mat dist;
+  cv::copyMakeBorder(resize_img, dist, 0, 0, 0, int(imgW - resize_w),
+                     cv::BORDER_CONSTANT, {0, 0, 0});
+
+  return dist;
+}
+
+cv::Mat crnn_resize_img(const cv::Mat &img, float wh_ratio) {
+  int imgC = REC_IMAGE_SHAPE[0];
+  int imgW = REC_IMAGE_SHAPE[2];
+  int imgH = REC_IMAGE_SHAPE[1];
+
+  if (CHARACTER_TYPE == "ch") {
+    imgW = int(32 * wh_ratio);
+  }
+
+  float ratio = float(img.cols) / float(img.rows);
+  int resize_w = 0;
+  if (ceilf(imgH * ratio) > imgW)
+    resize_w = imgW;
+  else
+    resize_w = int(ceilf(imgH * ratio));
+  cv::Mat resize_img;
+  cv::resize(img, resize_img, cv::Size(resize_w, imgH));
+  return resize_img;
+}
+
+cv::Mat get_rotate_crop_image(const cv::Mat &srcimage,
+                              const std::vector<std::vector<int>> &box) {
+
+  std::vector<std::vector<int>> points = box;
+
+  int x_collect[4] = {box[0][0], box[1][0], box[2][0], box[3][0]};
+  int y_collect[4] = {box[0][1], box[1][1], box[2][1], box[3][1]};
+  int left = int(*std::min_element(x_collect, x_collect + 4));
+  int right = int(*std::max_element(x_collect, x_collect + 4));
+  int top = int(*std::min_element(y_collect, y_collect + 4));
+  int bottom = int(*std::max_element(y_collect, y_collect + 4));
+
+  cv::Mat img_crop;
+  srcimage(cv::Rect(left, top, right - left, bottom - top)).copyTo(img_crop);
+
+  for (int i = 0; i < points.size(); i++) {
+    points[i][0] -= left;
+    points[i][1] -= top;
+  }
+
+  int img_crop_width = int(sqrt(pow(points[0][0] - points[1][0], 2) +
+                                pow(points[0][1] - points[1][1], 2)));
+  int img_crop_height = int(sqrt(pow(points[0][0] - points[3][0], 2) +
+                                 pow(points[0][1] - points[3][1], 2)));
+
+  cv::Point2f pts_std[4];
+  pts_std[0] = cv::Point2f(0., 0.);
+  pts_std[1] = cv::Point2f(img_crop_width, 0.);
+  pts_std[2] = cv::Point2f(img_crop_width, img_crop_height);
+  pts_std[3] = cv::Point2f(0.f, img_crop_height);
+
+  cv::Point2f pointsf[4];
+  pointsf[0] = cv::Point2f(points[0][0], points[0][1]);
+  pointsf[1] = cv::Point2f(points[1][0], points[1][1]);
+  pointsf[2] = cv::Point2f(points[2][0], points[2][1]);
+  pointsf[3] = cv::Point2f(points[3][0], points[3][1]);
+
+  cv::Mat M = cv::getPerspectiveTransform(pointsf, pts_std);
+
+  cv::Mat dst_img;
+  cv::warpPerspective(img_crop, dst_img, M,
+                      cv::Size(img_crop_width, img_crop_height),
+                      cv::BORDER_REPLICATE);
+
+  if (float(dst_img.rows) >= float(dst_img.cols) * 1.5) {
+    /*
+    cv::Mat srcCopy = cv::Mat(dst_img.rows, dst_img.cols, dst_img.depth());
+    cv::transpose(dst_img, srcCopy);
+    cv::flip(srcCopy, srcCopy, 0);
+    return srcCopy;
+    */
+    cv::transpose(dst_img, dst_img);
+    cv::flip(dst_img, dst_img, 0);
+    return dst_img;
+  } else {
+    return dst_img;
+  }
+}

app/src/main/cpp/ocr_crnn_process.h

@@ -0,0 +1,20 @@
+//
+// Created by fujiayi on 2020/7/3.
+//
+#pragma once
+
+#include "common.h"
+#include <opencv2/opencv.hpp>
+#include <vector>
+
+extern const std::vector<int> REC_IMAGE_SHAPE;
+
+cv::Mat get_rotate_crop_image(const cv::Mat &srcimage,
+                              const std::vector<std::vector<int>> &box);
+
+cv::Mat crnn_resize_img(const cv::Mat &img, float wh_ratio);
+
+template <class ForwardIterator>
+inline size_t argmax(ForwardIterator first, ForwardIterator last) {
+  return std::distance(first, std::max_element(first, last));
+}

app/src/main/cpp/ocr_db_post_process.cpp

@@ -0,0 +1,342 @@
+// Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "ocr_clipper.hpp"
+#include "opencv2/core.hpp"
+#include "opencv2/imgcodecs.hpp"
+#include "opencv2/imgproc.hpp"
+#include <iostream>
+#include <math.h>
+#include <vector>
+
+static void getcontourarea(float **box, float unclip_ratio, float &distance) {
+  int pts_num = 4;
+  float area = 0.0f;
+  float dist = 0.0f;
+  for (int i = 0; i < pts_num; i++) {
+    area += box[i][0] * box[(i + 1) % pts_num][1] -
+            box[i][1] * box[(i + 1) % pts_num][0];
+    dist += sqrtf((box[i][0] - box[(i + 1) % pts_num][0]) *
+                      (box[i][0] - box[(i + 1) % pts_num][0]) +
+                  (box[i][1] - box[(i + 1) % pts_num][1]) *
+                      (box[i][1] - box[(i + 1) % pts_num][1]));
+  }
+  area = fabs(float(area / 2.0));
+
+  distance = area * unclip_ratio / dist;
+}
+
+static cv::RotatedRect unclip(float **box) {
+  float unclip_ratio = 2.0;
+  float distance = 1.0;
+
+  getcontourarea(box, unclip_ratio, distance);
+
+  ClipperLib::ClipperOffset offset;
+  ClipperLib::Path p;
+  p << ClipperLib::IntPoint(int(box[0][0]), int(box[0][1]))
+    << ClipperLib::IntPoint(int(box[1][0]), int(box[1][1]))
+    << ClipperLib::IntPoint(int(box[2][0]), int(box[2][1]))
+    << ClipperLib::IntPoint(int(box[3][0]), int(box[3][1]));
+  offset.AddPath(p, ClipperLib::jtRound, ClipperLib::etClosedPolygon);
+
+  ClipperLib::Paths soln;
+  offset.Execute(soln, distance);
+  std::vector<cv::Point2f> points;
+
+  for (int j = 0; j < soln.size(); j++) {
+    for (int i = 0; i < soln[soln.size() - 1].size(); i++) {
+      points.emplace_back(soln[j][i].X, soln[j][i].Y);
+    }
+  }
+  cv::RotatedRect res = cv::minAreaRect(points);
+
+  return res;
+}
+
+static float **Mat2Vec(cv::Mat mat) {
+  auto **array = new float *[mat.rows];
+  for (int i = 0; i < mat.rows; ++i) {
+    array[i] = new float[mat.cols];
+  }
+  for (int i = 0; i < mat.rows; ++i) {
+    for (int j = 0; j < mat.cols; ++j) {
+      array[i][j] = mat.at<float>(i, j);
+    }
+  }
+
+  return array;
+}
+
+static void quickSort(float **s, int l, int r) {
+  if (l < r) {
+    int i = l, j = r;
+    float x = s[l][0];
+    float *xp = s[l];
+    while (i < j) {
+      while (i < j && s[j][0] >= x) {
+        j--;
+      }
+      if (i < j) {
+        std::swap(s[i++], s[j]);
+      }
+      while (i < j && s[i][0] < x) {
+        i++;
+      }
+      if (i < j) {
+        std::swap(s[j--], s[i]);
+      }
+    }
+    s[i] = xp;
+    quickSort(s, l, i - 1);
+    quickSort(s, i + 1, r);
+  }
+}
+
+static void quickSort_vector(std::vector<std::vector<int>> &box, int l, int r,
+                             int axis) {
+  if (l < r) {
+    int i = l, j = r;
+    int x = box[l][axis];
+    std::vector<int> xp(box[l]);
+    while (i < j) {
+      while (i < j && box[j][axis] >= x) {
+        j--;
+      }
+      if (i < j) {
+        std::swap(box[i++], box[j]);
+      }
+      while (i < j && box[i][axis] < x) {
+        i++;
+      }
+      if (i < j) {
+        std::swap(box[j--], box[i]);
+      }
+    }
+    box[i] = xp;
+    quickSort_vector(box, l, i - 1, axis);
+    quickSort_vector(box, i + 1, r, axis);
+  }
+}
+
+static std::vector<std::vector<int>>
+order_points_clockwise(std::vector<std::vector<int>> pts) {
+  std::vector<std::vector<int>> box = pts;
+  quickSort_vector(box, 0, int(box.size() - 1), 0);
+  std::vector<std::vector<int>> leftmost = {box[0], box[1]};
+  std::vector<std::vector<int>> rightmost = {box[2], box[3]};
+
+  if (leftmost[0][1] > leftmost[1][1]) {
+    std::swap(leftmost[0], leftmost[1]);
+  }
+
+  if (rightmost[0][1] > rightmost[1][1]) {
+    std::swap(rightmost[0], rightmost[1]);
+  }
+
+  std::vector<std::vector<int>> rect = {leftmost[0], rightmost[0], rightmost[1],
+                                        leftmost[1]};
+  return rect;
+}
+
+static float **get_mini_boxes(cv::RotatedRect box, float &ssid) {
+  ssid = box.size.width >= box.size.height ? box.size.height : box.size.width;
+
+  cv::Mat points;
+  cv::boxPoints(box, points);
+  // sorted box points
+  auto array = Mat2Vec(points);
+  quickSort(array, 0, 3);
+
+  float *idx1 = array[0], *idx2 = array[1], *idx3 = array[2], *idx4 = array[3];
+  if (array[3][1] <= array[2][1]) {
+    idx2 = array[3];
+    idx3 = array[2];
+  } else {
+    idx2 = array[2];
+    idx3 = array[3];
+  }
+  if (array[1][1] <= array[0][1]) {
+    idx1 = array[1];
+    idx4 = array[0];
+  } else {
+    idx1 = array[0];
+    idx4 = array[1];
+  }
+
+  array[0] = idx1;
+  array[1] = idx2;
+  array[2] = idx3;
+  array[3] = idx4;
+
+  return array;
+}
+
+template <class T> T clamp(T x, T min, T max) {
+  if (x > max) {
+    return max;
+  }
+  if (x < min) {
+    return min;
+  }
+  return x;
+}
+
+static float clampf(float x, float min, float max) {
+  if (x > max)
+    return max;
+  if (x < min)
+    return min;
+  return x;
+}
+
+float box_score_fast(float **box_array, cv::Mat pred) {
+  auto array = box_array;
+  int width = pred.cols;
+  int height = pred.rows;
+
+  float box_x[4] = {array[0][0], array[1][0], array[2][0], array[3][0]};
+  float box_y[4] = {array[0][1], array[1][1], array[2][1], array[3][1]};
+
+  int xmin = clamp(int(std::floorf(*(std::min_element(box_x, box_x + 4)))), 0,
+                   width - 1);
+  int xmax = clamp(int(std::ceilf(*(std::max_element(box_x, box_x + 4)))), 0,
+                   width - 1);
+  int ymin = clamp(int(std::floorf(*(std::min_element(box_y, box_y + 4)))), 0,
+                   height - 1);
+  int ymax = clamp(int(std::ceilf(*(std::max_element(box_y, box_y + 4)))), 0,
+                   height - 1);
+
+  cv::Mat mask;
+  mask = cv::Mat::zeros(ymax - ymin + 1, xmax - xmin + 1, CV_8UC1);
+
+  cv::Point root_point[4];
+  root_point[0] = cv::Point(int(array[0][0]) - xmin, int(array[0][1]) - ymin);
+  root_point[1] = cv::Point(int(array[1][0]) - xmin, int(array[1][1]) - ymin);
+  root_point[2] = cv::Point(int(array[2][0]) - xmin, int(array[2][1]) - ymin);
+  root_point[3] = cv::Point(int(array[3][0]) - xmin, int(array[3][1]) - ymin);
+  const cv::Point *ppt[1] = {root_point};
+  int npt[] = {4};
+  cv::fillPoly(mask, ppt, npt, 1, cv::Scalar(1));
+
+  cv::Mat croppedImg;
+  pred(cv::Rect(xmin, ymin, xmax - xmin + 1, ymax - ymin + 1))
+      .copyTo(croppedImg);
+
+  auto score = cv::mean(croppedImg, mask)[0];
+  return score;
+}
+
+std::vector<std::vector<std::vector<int>>>
+boxes_from_bitmap(const cv::Mat &pred, const cv::Mat &bitmap) {
+  const int min_size = 3;
+  const int max_candidates = 1000;
+  const float box_thresh = 0.5;
+
+  int width = bitmap.cols;
+  int height = bitmap.rows;
+
+  std::vector<std::vector<cv::Point>> contours;
+  std::vector<cv::Vec4i> hierarchy;
+
+  cv::findContours(bitmap, contours, hierarchy, cv::RETR_LIST,
+                   cv::CHAIN_APPROX_SIMPLE);
+
+  int num_contours =
+      contours.size() >= max_candidates ? max_candidates : contours.size();
+
+  std::vector<std::vector<std::vector<int>>> boxes;
+
+  for (int _i = 0; _i < num_contours; _i++) {
+    float ssid;
+    cv::RotatedRect box = cv::minAreaRect(contours[_i]);
+    auto array = get_mini_boxes(box, ssid);
+
+    auto box_for_unclip = array;
+    // end get_mini_box
+
+    if (ssid < min_size) {
+      continue;
+    }
+
+    float score;
+    score = box_score_fast(array, pred);
+    // end box_score_fast
+    if (score < box_thresh) {
+      continue;
+    }
+
+    // start for unclip
+    cv::RotatedRect points = unclip(box_for_unclip);
+    // end for unclip
+
+    cv::RotatedRect clipbox = points;
+    auto cliparray = get_mini_boxes(clipbox, ssid);
+
+    if (ssid < min_size + 2)
+      continue;
+
+    int dest_width = pred.cols;
+    int dest_height = pred.rows;
+    std::vector<std::vector<int>> intcliparray;
+
+    for (int num_pt = 0; num_pt < 4; num_pt++) {
+      std::vector<int> a{int(clampf(roundf(cliparray[num_pt][0] / float(width) *
+                                           float(dest_width)),
+                                    0, float(dest_width))),
+                         int(clampf(roundf(cliparray[num_pt][1] /
+                                           float(height) * float(dest_height)),
+                                    0, float(dest_height)))};
+      intcliparray.emplace_back(std::move(a));
+    }
+    boxes.emplace_back(std::move(intcliparray));
+
+  } // end for
+  return boxes;
+}
+
+int _max(int a, int b) { return a >= b ? a : b; }
+
+int _min(int a, int b) { return a >= b ? b : a; }
+
+std::vector<std::vector<std::vector<int>>>
+filter_tag_det_res(const std::vector<std::vector<std::vector<int>>> &o_boxes,
+                   float ratio_h, float ratio_w, const cv::Mat &srcimg) {
+  int oriimg_h = srcimg.rows;
+  int oriimg_w = srcimg.cols;
+  std::vector<std::vector<std::vector<int>>> boxes{o_boxes};
+  std::vector<std::vector<std::vector<int>>> root_points;
+  for (int n = 0; n < boxes.size(); n++) {
+    boxes[n] = order_points_clockwise(boxes[n]);
+    for (int m = 0; m < boxes[0].size(); m++) {
+      boxes[n][m][0] /= ratio_w;
+      boxes[n][m][1] /= ratio_h;
+
+      boxes[n][m][0] = int(_min(_max(boxes[n][m][0], 0), oriimg_w - 1));
+      boxes[n][m][1] = int(_min(_max(boxes[n][m][1], 0), oriimg_h - 1));
+    }
+  }
+
+  for (int n = 0; n < boxes.size(); n++) {
+    int rect_width, rect_height;
+    rect_width = int(sqrt(pow(boxes[n][0][0] - boxes[n][1][0], 2) +
+                          pow(boxes[n][0][1] - boxes[n][1][1], 2)));
+    rect_height = int(sqrt(pow(boxes[n][0][0] - boxes[n][3][0], 2) +
+                           pow(boxes[n][0][1] - boxes[n][3][1], 2)));
+    if (rect_width <= 10 || rect_height <= 10)
+      continue;
+    root_points.push_back(boxes[n]);
+  }
+  return root_points;
+}

app/src/main/cpp/ocr_db_post_process.h

@@ -0,0 +1,13 @@
+//
+// Created by fujiayi on 2020/7/2.
+//
+#pragma once
+#include <opencv2/opencv.hpp>
+#include <vector>
+
+std::vector<std::vector<std::vector<int>>>
+boxes_from_bitmap(const cv::Mat &pred, const cv::Mat &bitmap);
+
+std::vector<std::vector<std::vector<int>>>
+filter_tag_det_res(const std::vector<std::vector<std::vector<int>>> &o_boxes,
+                   float ratio_h, float ratio_w, const cv::Mat &srcimg);

app/src/main/cpp/ocr_ppredictor.cpp

@@ -0,0 +1,354 @@
+//
+// Created by fujiayi on 2020/7/1.
+//
+
+#include "ocr_ppredictor.h"
+#include "common.h"
+#include "ocr_cls_process.h"
+#include "ocr_crnn_process.h"
+#include "ocr_db_post_process.h"
+#include "preprocess.h"
+
+namespace ppredictor {
+
+OCR_PPredictor::OCR_PPredictor(const OCR_Config &config) : _config(config) {}
+
+int OCR_PPredictor::init(const std::string &det_model_content,
+                         const std::string &rec_model_content,
+                         const std::string &cls_model_content) {
+  _det_predictor = std::unique_ptr<PPredictor>(new PPredictor{
+      _config.use_opencl, _config.thread_num, NET_OCR, _config.mode});
+  _det_predictor->init_nb(det_model_content);
+
+  _rec_predictor = std::unique_ptr<PPredictor>(new PPredictor{
+      _config.use_opencl, _config.thread_num, NET_OCR_INTERNAL, _config.mode});
+  _rec_predictor->init_nb(rec_model_content);
+
+  _cls_predictor = std::unique_ptr<PPredictor>(new PPredictor{
+      _config.use_opencl, _config.thread_num, NET_OCR_INTERNAL, _config.mode});
+  _cls_predictor->init_nb(cls_model_content);
+  return RETURN_OK;
+}
+
+int OCR_PPredictor::init_from_file(const std::string &det_model_path,
+                                   const std::string &rec_model_path,
+                                   const std::string &cls_model_path) {
+  _det_predictor = std::unique_ptr<PPredictor>(new PPredictor{
+      _config.use_opencl, _config.thread_num, NET_OCR, _config.mode});
+  _det_predictor->init_from_file(det_model_path);
+
+  _rec_predictor = std::unique_ptr<PPredictor>(new PPredictor{
+      _config.use_opencl, _config.thread_num, NET_OCR_INTERNAL, _config.mode});
+  _rec_predictor->init_from_file(rec_model_path);
+
+  _cls_predictor = std::unique_ptr<PPredictor>(new PPredictor{
+      _config.use_opencl, _config.thread_num, NET_OCR_INTERNAL, _config.mode});
+  _cls_predictor->init_from_file(cls_model_path);
+  return RETURN_OK;
+}
+/**
+ * for debug use, show result of First Step
+ * @param filter_boxes
+ * @param boxes
+ * @param srcimg
+ */
+static void
+visual_img(const std::vector<std::vector<std::vector<int>>> &filter_boxes,
+           const std::vector<std::vector<std::vector<int>>> &boxes,
+           const cv::Mat &srcimg) {
+  // visualization
+  cv::Point rook_points[filter_boxes.size()][4];
+  for (int n = 0; n < filter_boxes.size(); n++) {
+    for (int m = 0; m < filter_boxes[0].size(); m++) {
+      rook_points[n][m] =
+          cv::Point(int(filter_boxes[n][m][0]), int(filter_boxes[n][m][1]));
+    }
+  }
+
+  cv::Mat img_vis;
+  srcimg.copyTo(img_vis);
+  for (int n = 0; n < boxes.size(); n++) {
+    const cv::Point *ppt[1] = {rook_points[n]};
+    int npt[] = {4};
+    cv::polylines(img_vis, ppt, npt, 1, 1, CV_RGB(0, 255, 0), 2, 8, 0);
+  }
+  // 调试用，自行替换需要修改的路径
+  cv::imwrite("/sdcard/1/vis.png", img_vis);
+}
+
+std::vector<OCRPredictResult>
+OCR_PPredictor::infer_ocr(cv::Mat &origin, int max_size_len, int run_det,
+                          int run_cls, int run_rec) {
+  LOGI("ocr cpp start *****************");
+  LOGI("ocr cpp det: %d, cls: %d, rec: %d", run_det, run_cls, run_rec);
+  std::vector<OCRPredictResult> ocr_results;
+  if (run_det) {
+    infer_det(origin, max_size_len, ocr_results);
+  }
+  if (run_rec) {
+    if (ocr_results.size() == 0) {
+      OCRPredictResult res;
+      ocr_results.emplace_back(std::move(res));
+    }
+    for (int i = 0; i < ocr_results.size(); i++) {
+      infer_rec(origin, run_cls, ocr_results[i]);
+    }
+  } else if (run_cls) {
+    ClsPredictResult cls_res = infer_cls(origin);
+    OCRPredictResult res;
+    res.cls_score = cls_res.cls_score;
+    res.cls_label = cls_res.cls_label;
+    ocr_results.push_back(res);
+  }
+
+  LOGI("ocr cpp end *****************");
+  return ocr_results;
+}
+
+cv::Mat DetResizeImg(const cv::Mat img, int max_size_len,
+                     std::vector<float> &ratio_hw) {
+  int w = img.cols;
+  int h = img.rows;
+
+  float ratio = 1.f;
+  int max_wh = w >= h ? w : h;
+  if (max_wh > max_size_len) {
+    if (h > w) {
+      ratio = static_cast<float>(max_size_len) / static_cast<float>(h);
+    } else {
+      ratio = static_cast<float>(max_size_len) / static_cast<float>(w);
+    }
+  }
+
+  int resize_h = static_cast<int>(float(h) * ratio);
+  int resize_w = static_cast<int>(float(w) * ratio);
+  if (resize_h % 32 == 0)
+    resize_h = resize_h;
+  else if (resize_h / 32 < 1 + 1e-5)
+    resize_h = 32;
+  else
+    resize_h = (resize_h / 32 - 1) * 32;
+
+  if (resize_w % 32 == 0)
+    resize_w = resize_w;
+  else if (resize_w / 32 < 1 + 1e-5)
+    resize_w = 32;
+  else
+    resize_w = (resize_w / 32 - 1) * 32;
+
+  cv::Mat resize_img;
+  cv::resize(img, resize_img, cv::Size(resize_w, resize_h));
+
+  ratio_hw.push_back(static_cast<float>(resize_h) / static_cast<float>(h));
+  ratio_hw.push_back(static_cast<float>(resize_w) / static_cast<float>(w));
+  return resize_img;
+}
+
+void OCR_PPredictor::infer_det(cv::Mat &origin, int max_size_len,
+                               std::vector<OCRPredictResult> &ocr_results) {
+  std::vector<float> mean = {0.485f, 0.456f, 0.406f};
+  std::vector<float> scale = {1 / 0.229f, 1 / 0.224f, 1 / 0.225f};
+
+  PredictorInput input = _det_predictor->get_first_input();
+
+  std::vector<float> ratio_hw;
+  cv::Mat input_image = DetResizeImg(origin, max_size_len, ratio_hw);
+  input_image.convertTo(input_image, CV_32FC3, 1 / 255.0f);
+  const float *dimg = reinterpret_cast<const float *>(input_image.data);
+  int input_size = input_image.rows * input_image.cols;
+
+  input.set_dims({1, 3, input_image.rows, input_image.cols});
+
+  neon_mean_scale(dimg, input.get_mutable_float_data(), input_size, mean,
+                  scale);
+  LOGI("ocr cpp det shape %d,%d", input_image.rows, input_image.cols);
+  std::vector<PredictorOutput> results = _det_predictor->infer();
+  PredictorOutput &res = results.at(0);
+  std::vector<std::vector<std::vector<int>>> filtered_box =
+      calc_filtered_boxes(res.get_float_data(), res.get_size(),
+                          input_image.rows, input_image.cols, origin);
+  LOGI("ocr cpp det Filter_box size %ld", filtered_box.size());
+
+  for (int i = 0; i < filtered_box.size(); i++) {
+    LOGI("ocr cpp box  %d,%d,%d,%d,%d,%d,%d,%d", filtered_box[i][0][0],
+         filtered_box[i][0][1], filtered_box[i][1][0], filtered_box[i][1][1],
+         filtered_box[i][2][0], filtered_box[i][2][1], filtered_box[i][3][0],
+         filtered_box[i][3][1]);
+    OCRPredictResult res;
+    res.points = filtered_box[i];
+    ocr_results.push_back(res);
+  }
+}
+
+void OCR_PPredictor::infer_rec(const cv::Mat &origin_img, int run_cls,
+                               OCRPredictResult &ocr_result) {
+  std::vector<float> mean = {0.5f, 0.5f, 0.5f};
+  std::vector<float> scale = {1 / 0.5f, 1 / 0.5f, 1 / 0.5f};
+  std::vector<int64_t> dims = {1, 3, 0, 0};
+
+  PredictorInput input = _rec_predictor->get_first_input();
+
+  const std::vector<std::vector<int>> &box = ocr_result.points;
+  cv::Mat crop_img;
+  if (box.size() > 0) {
+    crop_img = get_rotate_crop_image(origin_img, box);
+  } else {
+    crop_img = origin_img;
+  }
+
+  if (run_cls) {
+    ClsPredictResult cls_res = infer_cls(crop_img);
+    crop_img = cls_res.img;
+    ocr_result.cls_score = cls_res.cls_score;
+    ocr_result.cls_label = cls_res.cls_label;
+  }
+
+  float wh_ratio = float(crop_img.cols) / float(crop_img.rows);
+  cv::Mat input_image = crnn_resize_img(crop_img, wh_ratio);
+  input_image.convertTo(input_image, CV_32FC3, 1 / 255.0f);
+  const float *dimg = reinterpret_cast<const float *>(input_image.data);
+  int input_size = input_image.rows * input_image.cols;
+
+  dims[2] = input_image.rows;
+  dims[3] = input_image.cols;
+  input.set_dims(dims);
+
+  neon_mean_scale(dimg, input.get_mutable_float_data(), input_size, mean,
+                  scale);
+
+  std::vector<PredictorOutput> results = _rec_predictor->infer();
+  const float *predict_batch = results.at(0).get_float_data();
+  const std::vector<int64_t> predict_shape = results.at(0).get_shape();
+
+  // ctc decode
+  int argmax_idx;
+  int last_index = 0;
+  float score = 0.f;
+  int count = 0;
+  float max_value = 0.0f;
+
+  for (int n = 0; n < predict_shape[1]; n++) {
+    argmax_idx = int(argmax(&predict_batch[n * predict_shape[2]],
+                            &predict_batch[(n + 1) * predict_shape[2]]));
+    max_value =
+        float(*std::max_element(&predict_batch[n * predict_shape[2]],
+                                &predict_batch[(n + 1) * predict_shape[2]]));
+    if (argmax_idx > 0 && (!(n > 0 && argmax_idx == last_index))) {
+      score += max_value;
+      count += 1;
+      ocr_result.word_index.push_back(argmax_idx);
+    }
+    last_index = argmax_idx;
+  }
+  score /= count;
+  ocr_result.score = score;
+  LOGI("ocr cpp rec word size %ld", count);
+}
+
+ClsPredictResult OCR_PPredictor::infer_cls(const cv::Mat &img, float thresh) {
+  std::vector<float> mean = {0.5f, 0.5f, 0.5f};
+  std::vector<float> scale = {1 / 0.5f, 1 / 0.5f, 1 / 0.5f};
+  std::vector<int64_t> dims = {1, 3, 0, 0};
+
+  PredictorInput input = _cls_predictor->get_first_input();
+
+  cv::Mat input_image = cls_resize_img(img);
+  input_image.convertTo(input_image, CV_32FC3, 1 / 255.0f);
+  const float *dimg = reinterpret_cast<const float *>(input_image.data);
+  int input_size = input_image.rows * input_image.cols;
+
+  dims[2] = input_image.rows;
+  dims[3] = input_image.cols;
+  input.set_dims(dims);
+
+  neon_mean_scale(dimg, input.get_mutable_float_data(), input_size, mean,
+                  scale);
+
+  std::vector<PredictorOutput> results = _cls_predictor->infer();
+
+  const float *scores = results.at(0).get_float_data();
+  float score = 0;
+  int label = 0;
+  for (int64_t i = 0; i < results.at(0).get_size(); i++) {
+    LOGI("ocr cpp cls output scores [%f]", scores[i]);
+    if (scores[i] > score) {
+      score = scores[i];
+      label = i;
+    }
+  }
+  cv::Mat srcimg;
+  img.copyTo(srcimg);
+  if (label % 2 == 1 && score > thresh) {
+    cv::rotate(srcimg, srcimg, 1);
+  }
+  ClsPredictResult res;
+  res.cls_label = label;
+  res.cls_score = score;
+  res.img = srcimg;
+  LOGI("ocr cpp cls word cls %ld, %f", label, score);
+  return res;
+}
+
+std::vector<std::vector<std::vector<int>>>
+OCR_PPredictor::calc_filtered_boxes(const float *pred, int pred_size,
+                                    int output_height, int output_width,
+                                    const cv::Mat &origin) {
+  const double threshold = 0.3;
+  const double maxvalue = 1;
+
+  cv::Mat pred_map = cv::Mat::zeros(output_height, output_width, CV_32F);
+  memcpy(pred_map.data, pred, pred_size * sizeof(float));
+  cv::Mat cbuf_map;
+  pred_map.convertTo(cbuf_map, CV_8UC1);
+
+  cv::Mat bit_map;
+  cv::threshold(cbuf_map, bit_map, threshold, maxvalue, cv::THRESH_BINARY);
+
+  std::vector<std::vector<std::vector<int>>> boxes =
+      boxes_from_bitmap(pred_map, bit_map);
+  float ratio_h = output_height * 1.0f / origin.rows;
+  float ratio_w = output_width * 1.0f / origin.cols;
+  std::vector<std::vector<std::vector<int>>> filter_boxes =
+      filter_tag_det_res(boxes, ratio_h, ratio_w, origin);
+  return filter_boxes;
+}
+
+std::vector<int>
+OCR_PPredictor::postprocess_rec_word_index(const PredictorOutput &res) {
+  const int *rec_idx = res.get_int_data();
+  const std::vector<std::vector<uint64_t>> rec_idx_lod = res.get_lod();
+
+  std::vector<int> pred_idx;
+  for (int n = int(rec_idx_lod[0][0]); n < int(rec_idx_lod[0][1] * 2); n += 2) {
+    pred_idx.emplace_back(rec_idx[n]);
+  }
+  return pred_idx;
+}
+
+float OCR_PPredictor::postprocess_rec_score(const PredictorOutput &res) {
+  const float *predict_batch = res.get_float_data();
+  const std::vector<int64_t> predict_shape = res.get_shape();
+  const std::vector<std::vector<uint64_t>> predict_lod = res.get_lod();
+  int blank = predict_shape[1];
+  float score = 0.f;
+  int count = 0;
+  for (int n = predict_lod[0][0]; n < predict_lod[0][1] - 1; n++) {
+    int argmax_idx = argmax(predict_batch + n * predict_shape[1],
+                            predict_batch + (n + 1) * predict_shape[1]);
+    float max_value = predict_batch[n * predict_shape[1] + argmax_idx];
+    if (blank - 1 - argmax_idx > 1e-5) {
+      score += max_value;
+      count += 1;
+    }
+  }
+  if (count == 0) {
+    LOGE("calc score count 0");
+  } else {
+    score /= count;
+  }
+  LOGI("calc score: %f", score);
+  return score;
+}
+
+NET_TYPE OCR_PPredictor::get_net_flag() const { return NET_OCR; }
+} // namespace ppredictor

app/src/main/cpp/ocr_ppredictor.h

@@ -0,0 +1,131 @@
+//
+// Created by fujiayi on 2020/7/1.
+//
+
+#pragma once
+
+#include "ppredictor.h"
+#include <opencv2/opencv.hpp>
+#include <paddle_api.h>
+#include <string>
+
+namespace ppredictor {
+
+/**
+ * Config
+ */
+struct OCR_Config {
+  int use_opencl = 0;
+  int thread_num = 4; // Thread num
+  paddle::lite_api::PowerMode mode =
+      paddle::lite_api::LITE_POWER_HIGH; // PaddleLite Mode
+};
+
+/**
+ * Polygons Result
+ */
+struct OCRPredictResult {
+  std::vector<int> word_index;
+  std::vector<std::vector<int>> points;
+  float score;
+  float cls_score;
+  int cls_label = -1;
+};
+
+struct ClsPredictResult {
+  float cls_score;
+  int cls_label = -1;
+  cv::Mat img;
+};
+/**
+ * OCR there are 2 models
+ * 1. First model（det），select polygons to show where are the texts
+ * 2. crop from the origin images, use these polygons to infer
+ */
+class OCR_PPredictor : public PPredictor_Interface {
+public:
+  OCR_PPredictor(const OCR_Config &config);
+
+  virtual ~OCR_PPredictor() {}
+
+  /**
+   * 初始化二个模型的Predictor
+   * @param det_model_content
+   * @param rec_model_content
+   * @return
+   */
+  int init(const std::string &det_model_content,
+           const std::string &rec_model_content,
+           const std::string &cls_model_content);
+  int init_from_file(const std::string &det_model_path,
+                     const std::string &rec_model_path,
+                     const std::string &cls_model_path);
+  /**
+   * Return OCR result
+   * @param dims
+   * @param input_data
+   * @param input_len
+   * @param net_flag
+   * @param origin
+   * @return
+   */
+  virtual std::vector<OCRPredictResult> infer_ocr(cv::Mat &origin,
+                                                  int max_size_len, int run_det,
+                                                  int run_cls, int run_rec);
+
+  virtual NET_TYPE get_net_flag() const;
+
+private:
+  /**
+   * calculate polygons from the result image of first model
+   * @param pred
+   * @param output_height
+   * @param output_width
+   * @param origin
+   * @return
+   */
+  std::vector<std::vector<std::vector<int>>>
+  calc_filtered_boxes(const float *pred, int pred_size, int output_height,
+                      int output_width, const cv::Mat &origin);
+
+  void infer_det(cv::Mat &origin, int max_side_len,
+                 std::vector<OCRPredictResult> &ocr_results);
+  /**
+   * infer for rec model
+   *
+   * @param boxes
+   * @param origin
+   * @return
+   */
+  void infer_rec(const cv::Mat &origin, int run_cls,
+                 OCRPredictResult &ocr_result);
+
+  /**
+   * infer for cls model
+   *
+   * @param boxes
+   * @param origin
+   * @return
+   */
+  ClsPredictResult infer_cls(const cv::Mat &origin, float thresh = 0.9);
+
+  /**
+   * Postprocess or second model to extract text
+   * @param res
+   * @return
+   */
+  std::vector<int> postprocess_rec_word_index(const PredictorOutput &res);
+
+  /**
+   * calculate confidence of second model text result
+   * @param res
+   * @return
+   */
+  float postprocess_rec_score(const PredictorOutput &res);
+
+  std::unique_ptr<PPredictor> _det_predictor;
+  std::unique_ptr<PPredictor> _rec_predictor;
+  std::unique_ptr<PPredictor> _cls_predictor;
+  OCR_Config _config;
+};
+} // namespace ppredictor

app/src/main/cpp/ppredictor.cpp

@@ -0,0 +1,99 @@
+#include "ppredictor.h"
+#include "common.h"
+
+namespace ppredictor {
+PPredictor::PPredictor(int use_opencl, int thread_num, int net_flag,
+                       paddle::lite_api::PowerMode mode)
+    : _use_opencl(use_opencl), _thread_num(thread_num), _net_flag(net_flag),
+      _mode(mode) {}
+
+int PPredictor::init_nb(const std::string &model_content) {
+  paddle::lite_api::MobileConfig config;
+  config.set_model_from_buffer(model_content);
+  return _init(config);
+}
+
+int PPredictor::init_from_file(const std::string &model_content) {
+  paddle::lite_api::MobileConfig config;
+  config.set_model_from_file(model_content);
+  return _init(config);
+}
+
+template <typename ConfigT> int PPredictor::_init(ConfigT &config) {
+  bool is_opencl_backend_valid =
+      paddle::lite_api::IsOpenCLBackendValid(/*check_fp16_valid = false*/);
+  if (is_opencl_backend_valid) {
+    if (_use_opencl != 0) {
+      // Make sure you have write permission of the binary path.
+      // We strongly recommend each model has a unique binary name.
+      const std::string bin_path = "/data/local/tmp/";
+      const std::string bin_name = "lite_opencl_kernel.bin";
+      config.set_opencl_binary_path_name(bin_path, bin_name);
+
+      // opencl tune option
+      // CL_TUNE_NONE: 0
+      // CL_TUNE_RAPID: 1
+      // CL_TUNE_NORMAL: 2
+      // CL_TUNE_EXHAUSTIVE: 3
+      const std::string tuned_path = "/data/local/tmp/";
+      const std::string tuned_name = "lite_opencl_tuned.bin";
+      config.set_opencl_tune(paddle::lite_api::CL_TUNE_NORMAL, tuned_path,
+                             tuned_name);
+
+      // opencl precision option
+      // CL_PRECISION_AUTO: 0, first fp16 if valid, default
+      // CL_PRECISION_FP32: 1, force fp32
+      // CL_PRECISION_FP16: 2, force fp16
+      config.set_opencl_precision(paddle::lite_api::CL_PRECISION_FP32);
+      LOGI("ocr cpp device: running on gpu.");
+    }
+  } else {
+    LOGI("ocr cpp device: running on cpu.");
+    // you can give backup cpu nb model instead
+    // config.set_model_from_file(cpu_nb_model_dir);
+  }
+  config.set_threads(_thread_num);
+  config.set_power_mode(_mode);
+  _predictor = paddle::lite_api::CreatePaddlePredictor(config);
+  LOGI("ocr cpp paddle instance created");
+  return RETURN_OK;
+}
+
+PredictorInput PPredictor::get_input(int index) {
+  PredictorInput input{_predictor->GetInput(index), index, _net_flag};
+  _is_input_get = true;
+  return input;
+}
+
+std::vector<PredictorInput> PPredictor::get_inputs(int num) {
+  std::vector<PredictorInput> results;
+  for (int i = 0; i < num; i++) {
+    results.emplace_back(get_input(i));
+  }
+  return results;
+}
+
+PredictorInput PPredictor::get_first_input() { return get_input(0); }
+
+std::vector<PredictorOutput> PPredictor::infer() {
+  LOGI("ocr cpp infer Run start %d", _net_flag);
+  std::vector<PredictorOutput> results;
+  if (!_is_input_get) {
+    return results;
+  }
+  _predictor->Run();
+  LOGI("ocr cpp infer Run end");
+
+  for (int i = 0; i < _predictor->GetOutputNames().size(); i++) {
+    std::unique_ptr<const paddle::lite_api::Tensor> output_tensor =
+        _predictor->GetOutput(i);
+    LOGI("ocr cpp output tensor[%d] size %ld", i,
+         product(output_tensor->shape()));
+    PredictorOutput result{std::move(output_tensor), i, _net_flag};
+    results.emplace_back(std::move(result));
+  }
+  return results;
+}
+
+NET_TYPE PPredictor::get_net_flag() const { return (NET_TYPE)_net_flag; }
+} // namespace ppredictor

app/src/main/cpp/ppredictor.h

@@ -0,0 +1,64 @@
+#pragma once
+
+#include "paddle_api.h"
+#include "predictor_input.h"
+#include "predictor_output.h"
+
+namespace ppredictor {
+
+/**
+ * PaddleLite Preditor Common Interface
+ */
+class PPredictor_Interface {
+public:
+  virtual ~PPredictor_Interface() {}
+
+  virtual NET_TYPE get_net_flag() const = 0;
+};
+
+/**
+ * Common Predictor
+ */
+class PPredictor : public PPredictor_Interface {
+public:
+  PPredictor(
+      int use_opencl, int thread_num, int net_flag = 0,
+      paddle::lite_api::PowerMode mode = paddle::lite_api::LITE_POWER_HIGH);
+
+  virtual ~PPredictor() {}
+
+  /**
+   * init paddlitelite opt model，nb format ，or use ini_paddle
+   * @param model_content
+   * @return 0
+   */
+  virtual int init_nb(const std::string &model_content);
+
+  virtual int init_from_file(const std::string &model_content);
+
+  std::vector<PredictorOutput> infer();
+
+  std::shared_ptr<paddle::lite_api::PaddlePredictor> get_predictor() {
+    return _predictor;
+  }
+
+  virtual std::vector<PredictorInput> get_inputs(int num);
+
+  virtual PredictorInput get_input(int index);
+
+  virtual PredictorInput get_first_input();
+
+  virtual NET_TYPE get_net_flag() const;
+
+protected:
+  template <typename ConfigT> int _init(ConfigT &config);
+
+private:
+  int _use_opencl;
+  int _thread_num;
+  paddle::lite_api::PowerMode _mode;
+  std::shared_ptr<paddle::lite_api::PaddlePredictor> _predictor;
+  bool _is_input_get = false;
+  int _net_flag;
+};
+} // namespace ppredictor

app/src/main/cpp/predictor_input.cpp

@@ -0,0 +1,28 @@
+#include "predictor_input.h"
+
+namespace ppredictor {
+
+void PredictorInput::set_dims(std::vector<int64_t> dims) {
+  // yolov3
+  if (_net_flag == 101 && _index == 1) {
+    _tensor->Resize({1, 2});
+    _tensor->mutable_data<int>()[0] = (int)dims.at(2);
+    _tensor->mutable_data<int>()[1] = (int)dims.at(3);
+  } else {
+    _tensor->Resize(dims);
+  }
+  _is_dims_set = true;
+}
+
+float *PredictorInput::get_mutable_float_data() {
+  if (!_is_dims_set) {
+    LOGE("PredictorInput::set_dims is not called");
+  }
+  return _tensor->mutable_data<float>();
+}
+
+void PredictorInput::set_data(const float *input_data, int input_float_len) {
+  float *input_raw_data = get_mutable_float_data();
+  memcpy(input_raw_data, input_data, input_float_len * sizeof(float));
+}
+} // namespace ppredictor

app/src/main/cpp/predictor_input.h

@@ -0,0 +1,26 @@
+#pragma once
+
+#include "common.h"
+#include <paddle_api.h>
+#include <vector>
+
+namespace ppredictor {
+class PredictorInput {
+public:
+  PredictorInput(std::unique_ptr<paddle::lite_api::Tensor> &&tensor, int index,
+                 int net_flag)
+      : _tensor(std::move(tensor)), _index(index), _net_flag(net_flag) {}
+
+  void set_dims(std::vector<int64_t> dims);
+
+  float *get_mutable_float_data();
+
+  void set_data(const float *input_data, int input_float_len);
+
+private:
+  std::unique_ptr<paddle::lite_api::Tensor> _tensor;
+  bool _is_dims_set = false;
+  int _index;
+  int _net_flag;
+};
+} // namespace ppredictor

app/src/main/cpp/predictor_output.cpp

@@ -0,0 +1,26 @@
+#include "predictor_output.h"
+namespace ppredictor {
+const float *PredictorOutput::get_float_data() const {
+  return _tensor->data<float>();
+}
+
+const int *PredictorOutput::get_int_data() const {
+  return _tensor->data<int>();
+}
+
+const std::vector<std::vector<uint64_t>> PredictorOutput::get_lod() const {
+  return _tensor->lod();
+}
+
+int64_t PredictorOutput::get_size() const {
+  if (_net_flag == NET_OCR) {
+    return _tensor->shape().at(2) * _tensor->shape().at(3);
+  } else {
+    return product(_tensor->shape());
+  }
+}
+
+const std::vector<int64_t> PredictorOutput::get_shape() const {
+  return _tensor->shape();
+}
+} // namespace ppredictor

app/src/main/cpp/predictor_output.h

@@ -0,0 +1,31 @@
+#pragma once
+
+#include "common.h"
+#include <paddle_api.h>
+#include <vector>
+
+namespace ppredictor {
+class PredictorOutput {
+public:
+  PredictorOutput() {}
+  PredictorOutput(std::unique_ptr<const paddle::lite_api::Tensor> &&tensor,
+                  int index, int net_flag)
+      : _tensor(std::move(tensor)), _index(index), _net_flag(net_flag) {}
+
+  const float *get_float_data() const;
+  const int *get_int_data() const;
+  int64_t get_size() const;
+  const std::vector<std::vector<uint64_t>> get_lod() const;
+  const std::vector<int64_t> get_shape() const;
+
+  std::vector<float> data;    // return float, or use data_int
+  std::vector<int> data_int;  // several layers return int ，or use data
+  std::vector<int64_t> shape; // PaddleLite output shape
+  std::vector<std::vector<uint64_t>> lod; // PaddleLite output lod
+
+private:
+  std::unique_ptr<const paddle::lite_api::Tensor> _tensor;
+  int _index;
+  int _net_flag;
+};
+} // namespace ppredictor

app/src/main/cpp/preprocess.cpp

@@ -0,0 +1,82 @@
+#include "preprocess.h"
+#include <android/bitmap.h>
+
+cv::Mat bitmap_to_cv_mat(JNIEnv *env, jobject bitmap) {
+  AndroidBitmapInfo info;
+  int result = AndroidBitmap_getInfo(env, bitmap, &info);
+  if (result != ANDROID_BITMAP_RESULT_SUCCESS) {
+    LOGE("AndroidBitmap_getInfo failed, result: %d", result);
+    return cv::Mat{};
+  }
+  if (info.format != ANDROID_BITMAP_FORMAT_RGBA_8888) {
+    LOGE("Bitmap format is not RGBA_8888 !");
+    return cv::Mat{};
+  }
+  unsigned char *srcData = NULL;
+  AndroidBitmap_lockPixels(env, bitmap, (void **)&srcData);
+  cv::Mat mat = cv::Mat::zeros(info.height, info.width, CV_8UC4);
+  memcpy(mat.data, srcData, info.height * info.width * 4);
+  AndroidBitmap_unlockPixels(env, bitmap);
+  cv::cvtColor(mat, mat, cv::COLOR_RGBA2BGR);
+  /**
+  if (!cv::imwrite("/sdcard/1/copy.jpg", mat)){
+      LOGE("Write image failed " );
+  }
+   */
+
+  return mat;
+}
+
+cv::Mat resize_img(const cv::Mat &img, int height, int width) {
+  if (img.rows == height && img.cols == width) {
+    return img;
+  }
+  cv::Mat new_img;
+  cv::resize(img, new_img, cv::Size(height, width));
+  return new_img;
+}
+
+// fill tensor with mean and scale and trans layout: nhwc -> nchw, neon speed up
+void neon_mean_scale(const float *din, float *dout, int size,
+                     const std::vector<float> &mean,
+                     const std::vector<float> &scale) {
+  if (mean.size() != 3 || scale.size() != 3) {
+    LOGE("[ERROR] mean or scale size must equal to 3");
+    return;
+  }
+
+  float32x4_t vmean0 = vdupq_n_f32(mean[0]);
+  float32x4_t vmean1 = vdupq_n_f32(mean[1]);
+  float32x4_t vmean2 = vdupq_n_f32(mean[2]);
+  float32x4_t vscale0 = vdupq_n_f32(scale[0]);
+  float32x4_t vscale1 = vdupq_n_f32(scale[1]);
+  float32x4_t vscale2 = vdupq_n_f32(scale[2]);
+
+  float *dout_c0 = dout;
+  float *dout_c1 = dout + size;
+  float *dout_c2 = dout + size * 2;
+
+  int i = 0;
+  for (; i < size - 3; i += 4) {
+    float32x4x3_t vin3 = vld3q_f32(din);
+    float32x4_t vsub0 = vsubq_f32(vin3.val[0], vmean0);
+    float32x4_t vsub1 = vsubq_f32(vin3.val[1], vmean1);
+    float32x4_t vsub2 = vsubq_f32(vin3.val[2], vmean2);
+    float32x4_t vs0 = vmulq_f32(vsub0, vscale0);
+    float32x4_t vs1 = vmulq_f32(vsub1, vscale1);
+    float32x4_t vs2 = vmulq_f32(vsub2, vscale2);
+    vst1q_f32(dout_c0, vs0);
+    vst1q_f32(dout_c1, vs1);
+    vst1q_f32(dout_c2, vs2);
+
+    din += 12;
+    dout_c0 += 4;
+    dout_c1 += 4;
+    dout_c2 += 4;
+  }
+  for (; i < size; i++) {
+    *(dout_c0++) = (*(din++) - mean[0]) * scale[0];
+    *(dout_c1++) = (*(din++) - mean[1]) * scale[1];
+    *(dout_c2++) = (*(din++) - mean[2]) * scale[2];
+  }
+}

app/src/main/cpp/preprocess.h

@@ -0,0 +1,12 @@
+#pragma once
+
+#include "common.h"
+#include <jni.h>
+#include <opencv2/opencv.hpp>
+cv::Mat bitmap_to_cv_mat(JNIEnv *env, jobject bitmap);
+
+cv::Mat resize_img(const cv::Mat &img, int height, int width);
+
+void neon_mean_scale(const float *din, float *dout, int size,
+                     const std::vector<float> &mean,
+                     const std::vector<float> &scale);

app/src/main/java/com/xxpatx/os/activity/callwechat/CallWechatActivity.java

@@ -1,9 +1,13 @@
 package com.xxpatx.os.activity.callwechat;
 
+import android.content.Context;
 import android.content.Intent;
+import android.media.projection.MediaProjectionManager;
 import android.net.Uri;
+import android.os.Build;
 import android.provider.Settings;
 import android.text.TextUtils;
+import android.util.Log;
 import android.view.Gravity;
 import android.view.View;
 import android.widget.Toast;
@@ -18,7 +22,12 @@ import com.xxpatx.os.utils.AccessibilityUtils;
 
 public class CallWechatActivity extends BaseMvvmActivity<CallWechatViewModel, ActivityWechatCallBinding> {
 
+    private static final String TAG = "CallWechatActivity";
+
     private Contact mContact;
+    private int mCallType = 0;
+
+    private static final int REQUEST_CODE_SCREEN_CAPTURE = 1874;
 
     @Override
     public boolean setfitWindow() {
@@ -67,6 +76,29 @@ public class CallWechatActivity extends BaseMvvmActivity<CallWechatViewModel, Ac
         }
     }
 
+    @Override
+    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
+        super.onActivityResult(requestCode, resultCode, data);
+        Log.e(TAG, "onActivityResult: requestCode = " + requestCode);
+        Log.e(TAG, "onActivityResult: resultCode = " + resultCode);
+
+        if (requestCode == REQUEST_CODE_SCREEN_CAPTURE && resultCode == RESULT_OK) {
+            // 启动屏幕捕获服务
+            Intent intent = new Intent(CallWechatActivity.this, WeAccessibilityService.class);
+            intent.putExtra("WechatInfo", mContact);
+            intent.putExtra("call_type", mCallType);
+            intent.putExtra("resultCode", resultCode);
+            intent.putExtra("resultData", data);
+
+            if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
+                startForegroundService(intent);
+            } else {
+                startService(intent);
+            }
+            finish();
+        }
+    }
+
     private boolean checkSettings() {
         boolean accessibility = AccessibilityUtils.isAccessibilitySettingsOn(this);
         if (!accessibility) {
@@ -89,29 +121,39 @@ public class CallWechatActivity extends BaseMvvmActivity<CallWechatViewModel, Ac
         public void callWechatVideo(View view) {
             if (TextUtils.isEmpty(mContact.getTag())) {
                 Toaster.show("没有设置标签，无法拨打微信视频");
+                finish();
                 return;
             }
             if (checkSettings()) {
-                Intent intent = new Intent(CallWechatActivity.this, WeAccessibilityService.class);
-                intent.putExtra("WechatInfo", mContact);
-                intent.putExtra("call_type", WeAccessibilityService.TYPE_VIDEO);
-                startService(intent);
+                mCallType = WeAccessibilityService.TYPE_VIDEO;
+
+                // 1. 获取 MediaProjectionManager 实例
+                MediaProjectionManager projectionManager = (MediaProjectionManager)
+                        getSystemService(Context.MEDIA_PROJECTION_SERVICE);
+                Intent captureIntent = projectionManager.createScreenCaptureIntent();
+                startActivityForResult(captureIntent, REQUEST_CODE_SCREEN_CAPTURE);
+
+//                // 2. 系统应用可通过反射绕过用户确认
+//                Intent intent = new Intent("android.media.action.GET_SCREEN_CAPTURE");
+//                intent.putExtra("extra_screen_capture_allowed", true);
+//                startActivityForResult(intent, REQUEST_CODE_SCREEN_CAPTURE);
             }
-            finish();
         }
 
         public void callWechatVoice(View view) {
             if (TextUtils.isEmpty(mContact.getTag())) {
                 Toaster.show("没有设置标签，无法拨打微信语音");
+                finish();
                 return;
             }
             if (checkSettings()) {
-                Intent intent = new Intent(CallWechatActivity.this, WeAccessibilityService.class);
-                intent.putExtra("WechatInfo", mContact);
-                intent.putExtra("call_type", WeAccessibilityService.TYPE_VOICE);
-                startService(intent);
+                mCallType = WeAccessibilityService.TYPE_VOICE;
+                MediaProjectionManager projectionManager = (MediaProjectionManager)
+                        getSystemService(Context.MEDIA_PROJECTION_SERVICE);
+                Intent captureIntent = projectionManager.createScreenCaptureIntent();
+                startActivityForResult(captureIntent, REQUEST_CODE_SCREEN_CAPTURE);
+
             }
-            finish();
         }
 
         public void exit(View view) {

app/src/main/java/com/xxpatx/os/activity/main/MainActivity.java

@@ -74,6 +74,7 @@ import com.xxpatx.os.fragment.settings.SettingsFragment;
 import com.xxpatx.os.manager.DesktopIconManager;
 import com.xxpatx.os.manager.SpeechManager;
 import com.xxpatx.os.service.NotificationService;
+import com.xxpatx.os.service.WeAccessibilityService;
 import com.xxpatx.os.utils.ApkUtils;
 import com.xxpatx.os.utils.AppUsedTimeUtils;
 import com.xxpatx.os.utils.ContactsUtils;
@@ -832,7 +833,7 @@ public class MainActivity extends BaseMvvmActivity<MainViewModel, ActivityMainBi
             // 去开启 监听通知权限
             startActivity(new Intent("android.settings.ACTION_NOTIFICATION_LISTENER_SETTINGS"));
         }
-
+        sendBroadcast(new Intent(WeAccessibilityService.STOP_RECORD_ACTION));
         setDockApp();
     }
 

app/src/main/java/com/xxpatx/os/ocr/paddle/OCRPredictorNative.java

@@ -0,0 +1,105 @@
+package com.xxpatx.os.ocr.paddle;
+
+import android.graphics.Bitmap;
+import android.util.Log;
+
+import java.util.ArrayList;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+public class OCRPredictorNative {
+
+    private static final AtomicBoolean isSOLoaded = new AtomicBoolean();
+
+    public static void loadLibrary() throws RuntimeException {
+        if (!isSOLoaded.get() && isSOLoaded.compareAndSet(false, true)) {
+            try {
+                System.loadLibrary("Native");
+            } catch (Throwable e) {
+                RuntimeException exception = new RuntimeException(
+                        "Load libNative.so failed, please check it exists in apk file.", e);
+                throw exception;
+            }
+        }
+    }
+
+    private Config config;
+
+    private long nativePointer = 0;
+
+    public OCRPredictorNative(Config config) {
+        this.config = config;
+        loadLibrary();
+        nativePointer = init(config.detModelFilename, config.recModelFilename, config.clsModelFilename, config.useOpencl,
+                config.cpuThreadNum, config.cpuPower);
+        Log.i("OCRPredictorNative", "load success " + nativePointer);
+
+    }
+
+
+    public ArrayList<OcrResultModel> runImage(Bitmap originalImage, int max_size_len, int run_det, int run_cls, int run_rec) {
+        Log.i("OCRPredictorNative", "begin to run image ");
+        float[] rawResults = forward(nativePointer, originalImage, max_size_len, run_det, run_cls, run_rec);
+        ArrayList<OcrResultModel> results = postprocess(rawResults);
+        return results;
+    }
+
+    public static class Config {
+        public int useOpencl;
+        public int cpuThreadNum;
+        public String cpuPower;
+        public String detModelFilename;
+        public String recModelFilename;
+        public String clsModelFilename;
+
+    }
+
+    public void destroy() {
+        if (nativePointer != 0) {
+            release(nativePointer);
+            nativePointer = 0;
+        }
+    }
+
+    protected native long init(String detModelPath, String recModelPath, String clsModelPath, int useOpencl, int threadNum, String cpuMode);
+
+    protected native float[] forward(long pointer, Bitmap originalImage,int max_size_len, int run_det, int run_cls, int run_rec);
+
+    protected native void release(long pointer);
+
+    private ArrayList<OcrResultModel> postprocess(float[] raw) {
+        ArrayList<OcrResultModel> results = new ArrayList<OcrResultModel>();
+        int begin = 0;
+
+        while (begin < raw.length) {
+            int point_num = Math.round(raw[begin]);
+            int word_num = Math.round(raw[begin + 1]);
+            OcrResultModel res = parse(raw, begin + 2, point_num, word_num);
+            begin += 2 + 1 + point_num * 2 + word_num + 2;
+            results.add(res);
+        }
+
+        return results;
+    }
+
+    private OcrResultModel parse(float[] raw, int begin, int pointNum, int wordNum) {
+        int current = begin;
+        OcrResultModel res = new OcrResultModel();
+        res.setConfidence(raw[current]);
+        current++;
+        for (int i = 0; i < pointNum; i++) {
+            res.addPoints(Math.round(raw[current + i * 2]), Math.round(raw[current + i * 2 + 1]));
+        }
+        current += (pointNum * 2);
+        for (int i = 0; i < wordNum; i++) {
+            int index = Math.round(raw[current + i]);
+            res.addWordIndex(index);
+        }
+        current += wordNum;
+        res.setClsIdx(raw[current]);
+        res.setClsConfidence(raw[current + 1]);
+        Log.i("OCRPredictorNative", "word finished " + wordNum);
+        return res;
+    }
+
+
+}

app/src/main/java/com/xxpatx/os/ocr/paddle/OcrResultModel.java

@@ -0,0 +1,79 @@
+package com.xxpatx.os.ocr.paddle;
+
+import android.graphics.Point;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class OcrResultModel {
+    private List<Point> points;
+    private List<Integer> wordIndex;
+    private String label;
+    private float confidence;
+    private float cls_idx;
+    private String cls_label;
+    private float cls_confidence;
+
+    public OcrResultModel() {
+        super();
+        points = new ArrayList<>();
+        wordIndex = new ArrayList<>();
+    }
+
+    public void addPoints(int x, int y) {
+        Point point = new Point(x, y);
+        points.add(point);
+    }
+
+    public void addWordIndex(int index) {
+        wordIndex.add(index);
+    }
+
+    public List<Point> getPoints() {
+        return points;
+    }
+
+    public List<Integer> getWordIndex() {
+        return wordIndex;
+    }
+
+    public String getLabel() {
+        return label;
+    }
+
+    public void setLabel(String label) {
+        this.label = label;
+    }
+
+    public float getConfidence() {
+        return confidence;
+    }
+
+    public void setConfidence(float confidence) {
+        this.confidence = confidence;
+    }
+
+    public float getClsIdx() {
+        return cls_idx;
+    }
+
+    public void setClsIdx(float idx) {
+        this.cls_idx = idx;
+    }
+
+    public String getClsLabel() {
+        return cls_label;
+    }
+
+    public void setClsLabel(String label) {
+        this.cls_label = label;
+    }
+
+    public float getClsConfidence() {
+        return cls_confidence;
+    }
+
+    public void setClsConfidence(float confidence) {
+        this.cls_confidence = confidence;
+    }
+}

app/src/main/java/com/xxpatx/os/ocr/paddle/Predictor.java

@@ -0,0 +1,283 @@
+package com.xxpatx.os.ocr.paddle;
+
+import android.content.Context;
+import android.graphics.Bitmap;
+import android.graphics.Canvas;
+import android.graphics.Color;
+import android.graphics.Paint;
+import android.graphics.Path;
+import android.graphics.Point;
+import android.util.Log;
+
+import java.io.File;
+import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+import java.util.Vector;
+
+public class Predictor {
+    private static final String TAG = Predictor.class.getSimpleName();
+    public boolean isLoaded = false;
+    public int warmupIterNum = 1;
+    public int inferIterNum = 1;
+    public int cpuThreadNum = 6;
+    public String cpuPowerMode = "LITE_POWER_HIGH";
+    public String modelPath = "";
+    public String modelName = "";
+    protected OCRPredictorNative paddlePredictor = null;
+    protected float inferenceTime = 0;
+    // Only for object detection
+    protected Vector<String> wordLabels = new Vector<String>();
+    protected int detLongSize = 960;
+    protected float scoreThreshold = 0.1f;
+    protected Bitmap inputImage = null;
+    protected Bitmap outputImage = null;
+    protected volatile String outputResult = "";
+    protected float postprocessTime = 0;
+    protected ArrayList<OcrResultModel> mOcrResultModels;
+
+    public Predictor() {
+    }
+
+    public boolean init(Context appCtx, String modelPath, String labelPath, int useOpencl, int cpuThreadNum, String cpuPowerMode) {
+        isLoaded = loadModel(appCtx, modelPath, useOpencl, cpuThreadNum, cpuPowerMode);
+        if (!isLoaded) {
+            return false;
+        }
+        isLoaded = loadLabel(appCtx, labelPath);
+        return isLoaded;
+    }
+
+
+    public boolean init(Context appCtx, String modelPath, String labelPath, int useOpencl, int cpuThreadNum, String cpuPowerMode,
+                        int detLongSize, float scoreThreshold) {
+        boolean isLoaded = init(appCtx, modelPath, labelPath, useOpencl, cpuThreadNum, cpuPowerMode);
+        if (!isLoaded) {
+            return false;
+        }
+        this.detLongSize = detLongSize;
+        this.scoreThreshold = scoreThreshold;
+        return true;
+    }
+
+    protected boolean loadModel(Context appCtx, String modelPath, int useOpencl, int cpuThreadNum, String cpuPowerMode) {
+        // Release model if exists
+        releaseModel();
+
+        // Load model
+        if (modelPath.isEmpty()) {
+            return false;
+        }
+        String realPath = modelPath;
+        if (!modelPath.substring(0, 1).equals("/")) {
+            // Read model files from custom path if the first character of mode path is '/'
+            // otherwise copy model to cache from assets
+            realPath = appCtx.getCacheDir() + "/" + modelPath;
+            Utils.copyDirectoryFromAssets(appCtx, modelPath, realPath);
+        }
+        if (realPath.isEmpty()) {
+            return false;
+        }
+
+        OCRPredictorNative.Config config = new OCRPredictorNative.Config();
+        config.useOpencl = useOpencl;
+        config.cpuThreadNum = cpuThreadNum;
+        config.cpuPower = cpuPowerMode;
+        config.detModelFilename = realPath + File.separator + "det_db.nb";
+        config.recModelFilename = realPath + File.separator + "rec_crnn.nb";
+        config.clsModelFilename = realPath + File.separator + "cls.nb";
+        Log.i("Predictor", "model path" + config.detModelFilename + " ; " + config.recModelFilename + ";" + config.clsModelFilename);
+        paddlePredictor = new OCRPredictorNative(config);
+
+        this.cpuThreadNum = cpuThreadNum;
+        this.cpuPowerMode = cpuPowerMode;
+        this.modelPath = realPath;
+        this.modelName = realPath.substring(realPath.lastIndexOf("/") + 1);
+        return true;
+    }
+
+    public void releaseModel() {
+        if (paddlePredictor != null) {
+            paddlePredictor.destroy();
+            paddlePredictor = null;
+        }
+        isLoaded = false;
+        cpuThreadNum = 1;
+        cpuPowerMode = "LITE_POWER_HIGH";
+        modelPath = "";
+        modelName = "";
+    }
+
+    protected boolean loadLabel(Context appCtx, String labelPath) {
+        wordLabels.clear();
+        wordLabels.add("black");
+        // Load word labels from file
+        try {
+            InputStream assetsInputStream = appCtx.getAssets().open(labelPath);
+            int available = assetsInputStream.available();
+            byte[] lines = new byte[available];
+            assetsInputStream.read(lines);
+            assetsInputStream.close();
+            String words = new String(lines);
+            String[] contents = words.split("\n");
+            for (String content : contents) {
+                wordLabels.add(content);
+            }
+            wordLabels.add(" ");
+            Log.i(TAG, "Word label size: " + wordLabels.size());
+        } catch (Exception e) {
+            Log.e(TAG, e.getMessage());
+            return false;
+        }
+        return true;
+    }
+
+
+    public boolean runModel(int run_det, int run_cls, int run_rec) {
+        if (inputImage == null || !isLoaded()) {
+            return false;
+        }
+        Log.d(TAG, "runModel: " + inputImage.getWidth() + "x" + inputImage.getHeight());
+        // Warm up
+        for (int i = 0; i < warmupIterNum; i++) {
+            paddlePredictor.runImage(inputImage, detLongSize, run_det, run_cls, run_rec);
+        }
+        warmupIterNum = 0; // do not need warm
+        // Run inference
+        Date start = new Date();
+        ArrayList<OcrResultModel> results = paddlePredictor.runImage(inputImage, detLongSize, run_det, run_cls, run_rec);
+        Date end = new Date();
+        inferenceTime = (end.getTime() - start.getTime()) / (float) inferIterNum;
+
+        results = postprocess(results);
+        mOcrResultModels = results;
+        Log.i(TAG, "[stat] Inference Time: " + inferenceTime + " ;Box Size " + results.size());
+        drawResults(results);
+
+        return true;
+    }
+
+    public ArrayList<OcrResultModel> getOcrResultModels() {
+        return mOcrResultModels;
+    }
+
+    public boolean isLoaded() {
+        return paddlePredictor != null && isLoaded;
+    }
+
+    public String modelPath() {
+        return modelPath;
+    }
+
+    public String modelName() {
+        return modelName;
+    }
+
+    public int cpuThreadNum() {
+        return cpuThreadNum;
+    }
+
+    public String cpuPowerMode() {
+        return cpuPowerMode;
+    }
+
+    public float inferenceTime() {
+        return inferenceTime;
+    }
+
+    public Bitmap inputImage() {
+        return inputImage;
+    }
+
+    public Bitmap outputImage() {
+        return outputImage;
+    }
+
+    public String outputResult() {
+        return outputResult;
+    }
+
+    public float postprocessTime() {
+        return postprocessTime;
+    }
+
+
+    public void setInputImage(Bitmap image) {
+        if (image == null) {
+            return;
+        }
+        this.inputImage = image.copy(Bitmap.Config.ARGB_8888, true);
+    }
+
+    private ArrayList<OcrResultModel> postprocess(ArrayList<OcrResultModel> results) {
+        for (OcrResultModel r : results) {
+            StringBuffer word = new StringBuffer();
+            for (int index : r.getWordIndex()) {
+                if (index >= 0 && index < wordLabels.size()) {
+                    word.append(wordLabels.get(index));
+                } else {
+                    Log.e(TAG, "Word index is not in label list:" + index);
+                    word.append("×");
+                }
+            }
+            r.setLabel(word.toString());
+            r.setClsLabel(r.getClsIdx() == 1 ? "180" : "0");
+        }
+        return results;
+    }
+
+    private void drawResults(ArrayList<OcrResultModel> results) {
+        StringBuffer outputResultSb = new StringBuffer("");
+        for (int i = 0; i < results.size(); i++) {
+            OcrResultModel result = results.get(i);
+            StringBuilder sb = new StringBuilder("");
+            if (result.getPoints().size() > 0) {
+                sb.append("Det: ");
+                for (Point p : result.getPoints()) {
+                    sb.append("(").append(p.x).append(",").append(p.y).append(") ");
+                }
+            }
+            if (result.getLabel().length() > 0) {
+                sb.append("\n Rec: ").append(result.getLabel());
+                sb.append(",").append(result.getConfidence());
+            }
+            if (result.getClsIdx() != -1) {
+                sb.append(" Cls: ").append(result.getClsLabel());
+                sb.append(",").append(result.getClsConfidence());
+            }
+            Log.i(TAG, sb.toString()); // show LOG in Logcat panel
+            outputResultSb.append(i + 1).append(": ").append(sb.toString()).append("\n");
+        }
+        outputResult = outputResultSb.toString();
+        outputImage = inputImage;
+        Canvas canvas = new Canvas(outputImage);
+        Paint paintFillAlpha = new Paint();
+        paintFillAlpha.setStyle(Paint.Style.FILL);
+        paintFillAlpha.setColor(Color.parseColor("#3B85F5"));
+        paintFillAlpha.setAlpha(50);
+
+        Paint paint = new Paint();
+        paint.setColor(Color.parseColor("#3B85F5"));
+        paint.setStrokeWidth(5);
+        paint.setStyle(Paint.Style.STROKE);
+
+        for (OcrResultModel result : results) {
+            Path path = new Path();
+            List<Point> points = result.getPoints();
+            if (points.size() == 0) {
+                continue;
+            }
+            path.moveTo(points.get(0).x, points.get(0).y);
+            for (int i = points.size() - 1; i >= 0; i--) {
+                Point p = points.get(i);
+                path.lineTo(p.x, p.y);
+            }
+            canvas.drawPath(path, paint);
+            canvas.drawPath(path, paintFillAlpha);
+        }
+        Log.d(TAG, "drawResults: " + outputImage.getWidth() + "x" + outputImage.getHeight());
+
+    }
+
+}

app/src/main/java/com/xxpatx/os/ocr/paddle/Utils.java

@@ -0,0 +1,166 @@
+package com.xxpatx.os.ocr.paddle;
+
+import android.content.Context;
+import android.graphics.Bitmap;
+import android.graphics.Matrix;
+import android.media.ExifInterface;
+import android.os.Environment;
+
+import java.io.BufferedInputStream;
+import java.io.BufferedOutputStream;
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+
+public class Utils {
+    private static final String TAG = Utils.class.getSimpleName();
+
+    public static void copyFileFromAssets(Context appCtx, String srcPath, String dstPath) {
+        if (srcPath.isEmpty() || dstPath.isEmpty()) {
+            return;
+        }
+        InputStream is = null;
+        OutputStream os = null;
+        try {
+            is = new BufferedInputStream(appCtx.getAssets().open(srcPath));
+            os = new BufferedOutputStream(new FileOutputStream(new File(dstPath)));
+            byte[] buffer = new byte[1024];
+            int length = 0;
+            while ((length = is.read(buffer)) != -1) {
+                os.write(buffer, 0, length);
+            }
+        } catch (FileNotFoundException e) {
+            e.printStackTrace();
+        } catch (IOException e) {
+            e.printStackTrace();
+        } finally {
+            try {
+                os.close();
+                is.close();
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
+    public static void copyDirectoryFromAssets(Context appCtx, String srcDir, String dstDir) {
+        if (srcDir.isEmpty() || dstDir.isEmpty()) {
+            return;
+        }
+        try {
+            if (!new File(dstDir).exists()) {
+                new File(dstDir).mkdirs();
+            }
+            for (String fileName : appCtx.getAssets().list(srcDir)) {
+                String srcSubPath = srcDir + File.separator + fileName;
+                String dstSubPath = dstDir + File.separator + fileName;
+                if (new File(srcSubPath).isDirectory()) {
+                    copyDirectoryFromAssets(appCtx, srcSubPath, dstSubPath);
+                } else {
+                    copyFileFromAssets(appCtx, srcSubPath, dstSubPath);
+                }
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    public static float[] parseFloatsFromString(String string, String delimiter) {
+        String[] pieces = string.trim().toLowerCase().split(delimiter);
+        float[] floats = new float[pieces.length];
+        for (int i = 0; i < pieces.length; i++) {
+            floats[i] = Float.parseFloat(pieces[i].trim());
+        }
+        return floats;
+    }
+
+    public static long[] parseLongsFromString(String string, String delimiter) {
+        String[] pieces = string.trim().toLowerCase().split(delimiter);
+        long[] longs = new long[pieces.length];
+        for (int i = 0; i < pieces.length; i++) {
+            longs[i] = Long.parseLong(pieces[i].trim());
+        }
+        return longs;
+    }
+
+    public static String getSDCardDirectory() {
+        return Environment.getExternalStorageDirectory().getAbsolutePath();
+    }
+
+    public static boolean isSupportedNPU() {
+        return false;
+        // String hardware = android.os.Build.HARDWARE;
+        // return hardware.equalsIgnoreCase("kirin810") || hardware.equalsIgnoreCase("kirin990");
+    }
+
+    public static Bitmap resizeWithStep(Bitmap bitmap, int maxLength, int step) {
+        int width = bitmap.getWidth();
+        int height = bitmap.getHeight();
+        int maxWH = Math.max(width, height);
+        float ratio = 1;
+        int newWidth = width;
+        int newHeight = height;
+        if (maxWH > maxLength) {
+            ratio = maxLength * 1.0f / maxWH;
+            newWidth = (int) Math.floor(ratio * width);
+            newHeight = (int) Math.floor(ratio * height);
+        }
+
+        newWidth = newWidth - newWidth % step;
+        if (newWidth == 0) {
+            newWidth = step;
+        }
+        newHeight = newHeight - newHeight % step;
+        if (newHeight == 0) {
+            newHeight = step;
+        }
+        return Bitmap.createScaledBitmap(bitmap, newWidth, newHeight, true);
+    }
+
+    public static Bitmap rotateBitmap(Bitmap bitmap, int orientation) {
+
+        Matrix matrix = new Matrix();
+        switch (orientation) {
+            case ExifInterface.ORIENTATION_NORMAL:
+                return bitmap;
+            case ExifInterface.ORIENTATION_FLIP_HORIZONTAL:
+                matrix.setScale(-1, 1);
+                break;
+            case ExifInterface.ORIENTATION_ROTATE_180:
+                matrix.setRotate(180);
+                break;
+            case ExifInterface.ORIENTATION_FLIP_VERTICAL:
+                matrix.setRotate(180);
+                matrix.postScale(-1, 1);
+                break;
+            case ExifInterface.ORIENTATION_TRANSPOSE:
+                matrix.setRotate(90);
+                matrix.postScale(-1, 1);
+                break;
+            case ExifInterface.ORIENTATION_ROTATE_90:
+                matrix.setRotate(90);
+                break;
+            case ExifInterface.ORIENTATION_TRANSVERSE:
+                matrix.setRotate(-90);
+                matrix.postScale(-1, 1);
+                break;
+            case ExifInterface.ORIENTATION_ROTATE_270:
+                matrix.setRotate(-90);
+                break;
+            default:
+                return bitmap;
+        }
+        try {
+            Bitmap bmRotated = Bitmap.createBitmap(bitmap, 0, 0, bitmap.getWidth(), bitmap.getHeight(), matrix, true);
+            bitmap.recycle();
+            return bmRotated;
+        }
+        catch (OutOfMemoryError e) {
+            e.printStackTrace();
+            return null;
+        }
+    }
+}

app/src/main/java/com/xxpatx/os/service/WeAccessibilityService.java

@@ -2,15 +2,29 @@ package com.xxpatx.os.service;
 
 import android.accessibilityservice.AccessibilityService;
 import android.accessibilityservice.GestureDescription;
+import android.app.Notification;
+import android.app.NotificationChannel;
+import android.app.NotificationManager;
+import android.app.PendingIntent;
 import android.content.BroadcastReceiver;
 import android.content.ComponentName;
 import android.content.Context;
 import android.content.Intent;
 import android.content.IntentFilter;
+import android.graphics.Bitmap;
 import android.graphics.Path;
+import android.graphics.PixelFormat;
 import android.graphics.Point;
 import android.graphics.Rect;
+import android.hardware.display.DisplayManager;
+import android.hardware.display.VirtualDisplay;
+import android.media.Image;
+import android.media.ImageReader;
+import android.media.projection.MediaProjection;
+import android.media.projection.MediaProjectionManager;
+import android.os.Build;
 import android.os.Handler;
+import android.os.Looper;
 import android.text.TextUtils;
 import android.util.DisplayMetrics;
 import android.util.Log;
@@ -20,13 +34,35 @@ import android.view.accessibility.AccessibilityNodeInfo;
 import android.view.accessibility.AccessibilityWindowInfo;
 import android.widget.Toast;
 
+import androidx.core.app.NotificationCompat;
+
 import com.hjq.toast.Toaster;
 import com.tencent.mmkv.MMKV;
+import com.xxpatx.os.R;
+import com.xxpatx.os.activity.main.MainActivity;
 import com.xxpatx.os.bean.Contact;
 import com.xxpatx.os.config.CommonConfig;
+import com.xxpatx.os.ocr.paddle.OcrResultModel;
+import com.xxpatx.os.ocr.paddle.Predictor;
 import com.xxpatx.os.utils.ForegroundAppUtil;
 
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
+import java.util.function.Function;
+import java.util.function.Predicate;
+import java.util.stream.Collectors;
+
+import io.reactivex.rxjava3.android.schedulers.AndroidSchedulers;
+import io.reactivex.rxjava3.annotations.NonNull;
+import io.reactivex.rxjava3.core.Observable;
+import io.reactivex.rxjava3.core.ObservableEmitter;
+import io.reactivex.rxjava3.core.ObservableOnSubscribe;
+import io.reactivex.rxjava3.core.Observer;
+import io.reactivex.rxjava3.disposables.Disposable;
+import io.reactivex.rxjava3.schedulers.Schedulers;
+
+import static android.app.Activity.RESULT_OK;
 
 /**
  * 通过微信标签最高支持8.0.49，8.0.50 获取不到数据
@@ -37,6 +73,29 @@ public class WeAccessibilityService extends AccessibilityService {
 
     private MMKV mMMKV = MMKV.mmkvWithID(CommonConfig.MMKV_ID, MMKV.MULTI_PROCESS_MODE);
 
+    private static final int NOTIFICATION_ID = 12345;
+    private static final String CHANNEL_ID = "ScreenCaptureChannel";
+
+    private MediaProjectionManager mediaProjectionManager;
+    private MediaProjection mediaProjection;
+    private ImageReader imageReader;
+    private VirtualDisplay virtualDisplay;
+    private int screenWidth, screenHeight, screenDensity;
+
+    // Model settings of ocr
+    protected String modelPath = "models/ch_PP-OCRv2";
+    protected String labelPath = "labels/ppocr_keys_v1.txt";
+    protected String imagePath = "";
+    protected int cpuThreadNum = 6;
+    protected String cpuPowerMode = "LITE_POWER_HIGH";
+    protected int detLongSize = 960;
+    protected float scoreThreshold = 0.1f;
+    protected int useOpencl = 0;
+
+    protected Predictor predictor = new Predictor();
+
+    private String mCurrentPackageName = "";
+    private String mCurrentClassName = "";
 
     private static final String DIALER_TEXT = "音视频通话";
     private static final String CONTACT_TEXT = "通讯录";
@@ -77,14 +136,69 @@ public class WeAccessibilityService extends AccessibilityService {
         }
     };
 
+    @Override
+    public void onAccessibilityEvent(AccessibilityEvent event) {
+        Log.v(TAG, "onAccessibilityEvent: event = " + event.toString());
+        if (event.getEventType() == AccessibilityEvent.TYPE_WINDOW_STATE_CHANGED) {
+            ComponentName componentName = new ComponentName(
+                    event.getPackageName().toString(),
+                    event.getClassName().toString()
+            );
+            mCurrentPackageName = componentName.getPackageName();
+            mCurrentClassName = componentName.getClassName();
+        }
+
+        if (finished) {
+            finished = false;
+        } else {
+            Log.v(TAG, "bounce");
+            handler.removeCallbacks(runnable);
+        }
+        input = event;
+        handler.postDelayed(runnable, WAIT_TIME);
+    }
+
     @Override
     public void onCreate() {
         super.onCreate();
         Log.e(TAG, "onCreate: ");
         registerSettingReceiver();
+        registerStopReceiver();
         mAutoAccept = mMMKV.decodeBool(CommonConfig.WECHAT_CALL_AUTO_ACCEPT, false);
         handler = new Handler();
 
+        mediaProjectionManager = (MediaProjectionManager) getSystemService(Context.MEDIA_PROJECTION_SERVICE);
+
+        // 获取屏幕尺寸和密度
+        WindowManager windowManager = (WindowManager) getSystemService(Context.WINDOW_SERVICE);
+        DisplayMetrics metrics = new DisplayMetrics();
+        windowManager.getDefaultDisplay().getMetrics(metrics);
+        screenDensity = metrics.densityDpi;
+        Log.e(TAG, "onCreate: screenDensity = " + screenDensity);
+
+        Point realSize = new Point();
+        windowManager.getDefaultDisplay().getRealSize(realSize);
+
+        screenWidth = realSize.x;
+        screenHeight = realSize.y;
+        Log.e(TAG, "onCreate: screenWidth = " + screenWidth);
+        Log.e(TAG, "onCreate: screenHeight = " + screenHeight);
+        int widthPixels = metrics.widthPixels;
+        int heightPixels = metrics.heightPixels;
+        Log.e(TAG, "onCreate: widthPixels = " + widthPixels);
+        Log.e(TAG, "onCreate: heightPixels = " + heightPixels);
+
+        // 创建通知渠道（针对 Android 8.0 及以上版本）
+        createNotificationChannel();
+
+        // 启动前台服务
+        startForeground(NOTIFICATION_ID, createNotification());
+
+        if (onLoadModel()) {
+            Log.e(TAG, "onCreate: load model successful");
+        } else {
+            Log.e(TAG, "onCreate: load model failed");
+        }
     }
 
     @Override
@@ -103,6 +217,13 @@ public class WeAccessibilityService extends AccessibilityService {
             }
             mCurrentStep = Step.CLICK_CONTACT;
             launchWeChat();
+
+            convertImage();
+
+            int resultCode = intent.getIntExtra("resultCode", RESULT_OK);
+            Intent resultData = intent.getParcelableExtra("resultData");
+            // 开始屏幕录制
+            startScreenCapture(resultCode, resultData);
         }
         return super.onStartCommand(intent, flags, startId);
     }
@@ -114,21 +235,539 @@ public class WeAccessibilityService extends AccessibilityService {
         if (mSettingBroadcastReceiver != null) {
             unregisterReceiver(mSettingBroadcastReceiver);
         }
-    }
-
-    @Override
-    public void onAccessibilityEvent(AccessibilityEvent event) {
-        Log.v(TAG, "onAccessibilityEvent: event = " + event.toString());
-        if (finished) {
-            finished = false;
-        } else {
-            Log.v(TAG, "bounce");
-            handler.removeCallbacks(runnable);
+        if (mStopBroadcastReceiver != null) {
+            unregisterReceiver(mStopBroadcastReceiver);
         }
-        input = event;
-        handler.postDelayed(runnable, WAIT_TIME);
+        releaseDisplay();
     }
 
+    private void releaseDisplay() {
+        Log.e(TAG, "releaseDisplay: ");
+        // 释放资源
+        if (virtualDisplay != null) {
+            virtualDisplay.release();
+            virtualDisplay = null;
+        }
+
+        if (imageReader != null) {
+            imageReader.close();
+            imageReader = null;
+        }
+
+        if (mediaProjection != null) {
+            mediaProjection.stop();
+            mediaProjection = null;
+        }
+    }
+
+    private void startScreenCapture(int resultCode, Intent resultData) {
+        // 创建 MediaProjection
+        mediaProjection = mediaProjectionManager.getMediaProjection(resultCode, resultData);
+
+        // 创建 ImageReader 用于获取屏幕图像
+        imageReader = ImageReader.newInstance(
+                screenWidth,
+                screenHeight,
+                PixelFormat.RGBA_8888,
+                3
+        );
+
+        // 创建虚拟显示器
+        virtualDisplay = mediaProjection.createVirtualDisplay(
+                "ScreenCapture",
+                screenWidth,
+                screenHeight,
+                screenDensity,
+                DisplayManager.VIRTUAL_DISPLAY_FLAG_AUTO_MIRROR | DisplayManager.VIRTUAL_DISPLAY_FLAG_PUBLIC,
+                imageReader.getSurface(),
+                null,
+                new Handler()
+        );
+
+        // 设置图像可用时的回调
+        imageReader.setOnImageAvailableListener(reader -> {
+            Log.e(TAG, "startScreenCapture: ");
+            Image image = reader.acquireLatestImage();
+            if (completed) {
+                completed = false;
+                mProcessImage.onImage(image);
+            } else {
+                Log.e(TAG, "startScreenCapture: Processing Data");
+                if (image != null) {
+                    image.close();
+                }
+            }
+        }, new Handler());
+    }
+
+    private Bitmap convertImageToBitmap(Image image) {
+        Image.Plane[] planes = image.getPlanes();
+        int width = image.getWidth();
+        int height = image.getHeight();
+
+        // 创建 Bitmap
+        Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
+
+        // 处理图像数据
+        int pixelStride = planes[0].getPixelStride();
+        int rowStride = planes[0].getRowStride();
+        int rowPadding = rowStride - pixelStride * width;
+
+        // 复制图像数据到 Bitmap
+        byte[] buffer = new byte[width * height * 4];
+        java.nio.ByteBuffer byteBuffer = planes[0].getBuffer();
+        byteBuffer.get(buffer);
+
+        // 调整图像数据
+        int offset = 0;
+        for (int y = 0; y < height; y++) {
+            for (int x = 0; x < width; x++) {
+                int pixel = 0;
+                pixel |= (buffer[offset + 3] & 0xff) << 24; // alpha
+                pixel |= (buffer[offset + 0] & 0xff) << 16; // red
+                pixel |= (buffer[offset + 1] & 0xff) << 8;  // green
+                pixel |= (buffer[offset + 2] & 0xff);       // blue
+                bitmap.setPixel(x, y, pixel);
+                offset += pixelStride;
+            }
+            offset += rowPadding;
+        }
+
+        return bitmap;
+    }
+
+    private processImage mProcessImage;
+
+    public interface processImage {
+        void onImage(Image image);
+    }
+
+    private boolean completed = true;
+
+    private void convertImage() {
+        Observable.create(new ObservableOnSubscribe<ArrayList<OcrResultModel>>() {
+            @Override
+            public void subscribe(@NonNull ObservableEmitter<ArrayList<OcrResultModel>> emitter) throws Throwable {
+                mProcessImage = new processImage() {
+                    @Override
+                    public void onImage(Image image) {
+                        Log.e(TAG, "onImage: time = " + System.currentTimeMillis());
+                        // 获取最新的图像
+                        if (image != null) {
+                            Log.e(TAG, "startScreenCapture: image width = " + image.getWidth());
+                            Log.e(TAG, "startScreenCapture: image height = " + image.getHeight());
+                            Image.Plane[] planes = image.getPlanes();
+                            if (planes.length > 0) {
+                                // 将 Image 转换为 Bitmap
+                                Bitmap bitmap = convertImageToBitmap(image);
+                                // 处理获取到的屏幕图像
+                                if (bitmap != null) {
+                                    Log.e(TAG, "startScreenCapture: bitmap width = " + bitmap.getWidth());
+                                    Log.e(TAG, "startScreenCapture: bitmap height = " + bitmap.getHeight());
+                                    predictor.setInputImage(bitmap);
+                                    if (onRunModel()) {
+                                        Bitmap outputImage = predictor.outputImage();
+                                        Log.e(TAG, "startScreenCapture: outputImage width = " + outputImage.getWidth());
+                                        Log.e(TAG, "startScreenCapture: outputImage height = " + outputImage.getHeight());
+                                        ArrayList<OcrResultModel> ocrResultModels = predictor.getOcrResultModels();
+                                        emitter.onNext(ocrResultModels);
+                                        Log.e(TAG, "convertImage: Inference time: " + predictor.inferenceTime() + " ms");
+                                    } else {
+                                        Log.e(TAG, "convertImage: failed");
+                                        completed = true;
+                                    }
+                                }
+                            }
+                            // 释放资源
+                            image.close();
+                        } else {
+                            Log.e(TAG, "onImage: image is null");
+                            completed = true;
+                        }
+                    }
+                };
+            }
+        })
+//                .throttleFirst(800, TimeUnit.MILLISECONDS)
+                .subscribeOn(Schedulers.computation())
+                .observeOn(AndroidSchedulers.mainThread())
+                .subscribe(new Observer<ArrayList<OcrResultModel>>() {
+                    @Override
+                    public void onSubscribe(Disposable d) {
+                        Log.e("convertImage", "onSubscribe: ");
+                    }
+
+                    @Override
+                    public void onNext(ArrayList<OcrResultModel> ocrResultModels) {
+                        Log.e(TAG, "onNext: ocrResultList size = " + ocrResultModels.size());
+                        Log.v(TAG, "onNext: " + ocrResultModels);
+                        analysisUi(ocrResultModels);
+                        completed = true;
+                    }
+
+                    @Override
+                    public void onError(Throwable e) {
+                        Log.e("convertImage", "onError: " + e.getMessage());
+                        completed = true;
+                    }
+
+                    @Override
+                    public void onComplete() {
+                        Log.e("convertImage", "onComplete: ");
+                        completed = true;
+                    }
+                });
+    }
+
+
+    private void analysisUi(ArrayList<OcrResultModel> ocrResultModels) {
+        String topAppClassName = ForegroundAppUtil.getForegroundActivityName(WeAccessibilityService.this);
+        Log.e(TAG, "analysisUi: topAppClassName = " + topAppClassName);
+        Log.e(TAG, "analysisUi: mCurrentClassName = " + mCurrentClassName);
+        Log.e(TAG, "analysisUi: mCurrentPackageName = " + mCurrentPackageName);
+        if (!"com.tencent.mm".equals(mCurrentPackageName)) {
+            completed = true; // 非微信应用，直接允许处理下一张
+            return;
+        }
+        if (ocrResultModels == null || ocrResultModels.isEmpty()) {
+            Log.e(TAG, "analysisUi: ocrResultModels empty");
+            completed = true;
+            return;
+        }
+
+        List<String> stringList = ocrResultModels.stream().map(new Function<OcrResultModel, String>() {
+            @Override
+            public String apply(OcrResultModel ocrResultModel) {
+                return ocrResultModel.getLabel();
+            }
+        }).collect(Collectors.toList());
+
+        switch (topAppClassName) {
+            case "com.tencent.mm.ui.LauncherUI"://主页或者聊天页面
+                int weixinSize = countSubstringInListStream(stringList, "微信");
+                long contactSize = stringList.stream().filter(new Predicate<String>() {
+                    @Override
+                    public boolean test(String s) {
+                        return "通讯录".equals(s);
+                    }
+                }).count();
+                //两个微信说明在微信页面
+                if (weixinSize >= 2) {
+                    if (contactSize >= 2) {
+                        equalsViewTouch(ocrResultModels, topAppClassName, "标签");
+                    } else {
+                        equalsViewTouch(ocrResultModels, topAppClassName, "通讯录");
+                    }
+                } else {
+                    if (stringList.contains("通讯录")) {//在通讯录页面
+                        if (contactSize >= 2) {
+                            equalsViewTouch(ocrResultModels, topAppClassName, "标签");
+                        }
+                    }
+                }
+
+                break;
+            case "com.tencent.mm.plugin.label.ui.ContactLabelManagerUI"://标签页面
+                if (stringList.contains("通讯录标签")) {//在标签里面
+                    containsViewTouch(ocrResultModels, topAppClassName, mTagName);
+                }
+                break;
+            case "com.tencent.mm.ui.mvvm.MvvmContactListUI"://标签联系人列表
+                if (stringList.contains(mName) && findString(stringList, mTagName)) {
+                    equalsViewTouch(ocrResultModels, topAppClassName, mName);
+                } else {
+                    //滑动啥的
+                    scrollDown();
+                }
+                break;
+            case "com.tencent.mm.plugin.profile.ui.ContactInfoUI"://个人信息页面
+                if (findString(stringList, mName) && findString(stringList, DIALER_TEXT)) {
+                    if (findString(stringList, CALL_TEXT) && findString(stringList, VIDEO_TEXT)) {
+                        String callName;
+                        if (mCallType == TYPE_VIDEO) {
+                            callName = VIDEO_TEXT;
+                        } else {
+                            callName = CALL_TEXT;
+                        }
+                        boolean successful = containsViewTouch(ocrResultModels, topAppClassName, callName);
+                        Log.e(TAG, "analysisUi: successful = " + successful);
+                    } else {
+                        new Handler(Looper.getMainLooper()).postDelayed(new Runnable() {
+                            @Override
+                            public void run() {
+                                containsViewTouch(ocrResultModels, topAppClassName, DIALER_TEXT);
+                            }
+                        }, 2000);
+                    }
+                } else if (findString(stringList, CALL_TEXT) || findString(stringList, VIDEO_TEXT)) {
+                    String callName;
+                    if (mCallType == TYPE_VIDEO) {
+                        callName = VIDEO_TEXT;
+                    } else {
+                        callName = CALL_TEXT;
+                    }
+                    boolean successful = containsViewTouch(ocrResultModels, topAppClassName, callName);
+                    Log.e(TAG, "analysisUi: successful = " + successful);
+//                    if (successful) {//有可能太快了点击不生效
+//                        Toaster.showLong("拨打完成");
+//                        releaseDisplay();
+//                    }
+                }
+
+                break;
+            case "com.tencent.mm.plugin.voip.ui.VideoActivity"://视频或语音通话
+                Toaster.showLong("拨打完成");
+                releaseDisplay();
+                break;
+            case "com.tencent.mm.plugin.brandservice.ui.flutter.BizFlutterTLFlutterViewActivity"://公众号
+            case "com.tencent.mm.plugin.brandservice.ui.timeline.preload.ui.TmplWebViewMMUI"://文章
+            case "com.tencent.mm.plugin.finder.ui.FinderShareFeedRelUI"://视频
+            case "com.tencent.mm.plugin.subapp.ui.friend.FMessageConversationUI "://新的朋友
+            case "com.tencent.mm.ui.contact.OnlyChatContactMgrUI"://仅聊天朋友
+            case "com.tencent.mm.ui.contact.ChatroomContactUI"://群聊
+            case "com.tencent.mm.plugin.brandservice.ui.BrandServiceIndexUI"://公众号列表 服务号
+                launchWeChat();//重新打开微信
+                break;
+            case "com.tencent.mm.plugin.finder.feed.ui.FinderProfileUI":
+            case "com.tencent.mm.plugin.sns.ui.SnsUserUI":
+            case "com.tencent.mm.plugin.profile.ui.ContactMoreInfoUI":
+            case "com.tencent.mm.plugin.profile.ui.PermissionSettingUI":
+                performGlobalAction(GLOBAL_ACTION_BACK);
+                break;
+            case "com.tencent.mm.plugin.account.ui.WelcomeActivity":
+            case "com.tencent.mm.plugin.account.ui.LoginPasswordUI":
+                Toaster.showLong("请先登录微信");
+                releaseDisplay();
+                break;
+            default:
+        }
+
+//        if (stringList.contains("微信")) {//在主页
+//
+//        } else if (stringList.contains("通讯录")) {//在通讯录页面
+//            long contactSize = stringList.stream().filter(new Predicate<String>() {
+//                @Override
+//                public boolean test(String s) {
+//                    return "通讯录".equals(s);
+//                }
+//            }).count();
+//            if (contactSize >= 2) {
+//                equalsViewTouch(ocrResultModels, "标签");
+//            }
+//        } else if (stringList.contains("标签") && stringList.contains("通讯录")) {//在通讯录页面
+//            equalsViewTouch(ocrResultModels, "标签");
+//        } else if (stringList.contains("通讯录标签")) {//在标签里面
+//            equalsViewTouch(ocrResultModels, mTagName);
+//        } else if (stringList.contains(mName) && findString(stringList, DIALER_TEXT)) {
+//            if ("com.tencent.mm.plugin.profile.ui.ContactInfoUI".equals(mCurrentClassName)) {//防止dialog弹出来 去点击音视频通话
+//                containsViewTouch(ocrResultModels, DIALER_TEXT);
+//            }
+//        } else if (stringList.contains(mName)) {
+////            "com.tencent.mm.ui.mvvm.MvvmContactListUI"
+//            if (!stringList.contains("微信") && !stringList.contains("通讯录")) {
+//                equalsViewTouch(ocrResultModels, mName);
+//            } else {
+//                Log.e(TAG, "analysisUi: in home wechat fragment");
+//            }
+//        } else if (findString(stringList, DIALER_TEXT)) {
+//            if ("com.tencent.mm.plugin.profile.ui.ContactInfoUI".equals(mCurrentClassName)) {//防止dialog弹出来 去点击音视频通话
+//                containsViewTouch(ocrResultModels, DIALER_TEXT);
+//            }
+//        } else if (findString(stringList, VIDEO_TEXT) || findString(stringList, CALL_TEXT)) {
+//            Log.e(TAG, "analysisUi:  com.tencent.mm.ui.widget.dialog.w3");
+//            String callName;
+//            if (mCallType == TYPE_VIDEO) {
+//                callName = VIDEO_TEXT;
+//            } else {
+//                callName = CALL_TEXT;
+//            }
+//            boolean s = equalsViewTouch(ocrResultModels, callName);
+//            if (s) {
+//                Toaster.showLong("拨打完成");
+//                releaseDisplay();
+//            }
+//        } else {
+//            completed = true;
+//        }
+    }
+
+    private boolean clickByPoint(int x, int y, String className, Runnable onComplete) {
+        Log.e(TAG, "clickByNode: x = " + x);
+        Log.e(TAG, "clickByNode: y = " + y);
+
+        Point point = new Point(x, y);
+        Path path = new Path();
+        path.moveTo(point.x, point.y);
+        GestureDescription.Builder builder = new GestureDescription.Builder();
+        builder.addStroke(new GestureDescription.StrokeDescription(path, 0, 200));
+        GestureDescription gesture = builder.build();
+        String topAppClassName = ForegroundAppUtil.getForegroundActivityName(WeAccessibilityService.this);
+        if (topAppClassName.equals(className)) {
+            boolean dispatched = dispatchGesture(gesture, new GestureResultCallback() {
+                @Override
+                public void onCompleted(GestureDescription gestureDescription) {
+                    super.onCompleted(gestureDescription);
+                    Log.e("clickByNode", "onCompleted: ");
+                    if (onComplete != null) {
+                        onComplete.run();
+                    }
+                }
+
+                @Override
+                public void onCancelled(GestureDescription gestureDescription) {
+                    super.onCancelled(gestureDescription);
+                    Log.e("clickByNode", "onCompleted: ");
+                    if (onComplete != null) {
+                        onComplete.run();
+                    }
+                }
+            }, null);
+            Log.e(TAG, "clickByPoint: dispatched = " + dispatched);
+            return dispatched;
+        }
+        return false;
+    }
+
+
+    private boolean findString(List<String> strings, String text) {
+        boolean found = strings.stream()
+                .anyMatch(s -> s.contains(text));
+        return found;
+    }
+
+    public static int countSubstringInListStream(List<String> list, String target) {
+        return list.stream()
+                .mapToInt(s -> {
+                    int count = 0;
+                    int index = 0;
+                    while ((index = s.indexOf(target, index)) != -1) {
+                        count++;
+                        index += target.length();
+                    }
+                    return count;
+                })
+                .sum();  // 对每个元素的统计结果求和[6](@ref)
+    }
+
+    private boolean equalsViewTouch(ArrayList<OcrResultModel> ocrResultModels, String className, String text) {
+        Optional<OcrResultModel> ocrResultOptional = ocrResultModels.stream().filter(new Predicate<OcrResultModel>() {
+            @Override
+            public boolean test(OcrResultModel ocrResultModel) {
+                return text.equals(ocrResultModel.getLabel());
+            }
+        }).findAny();
+        if (ocrResultOptional.isPresent()) {
+            OcrResultModel ocrResult = ocrResultOptional.get();
+            List<Point> points = ocrResult.getPoints();
+            Log.e(TAG, "analysisUi: points = " + points);
+            Point point = getCenterPoint(points);
+            Log.e(TAG, "analysisUi: CenterPoint = " + point);
+            return clickByPoint(point.x, point.y, className, new Runnable() {
+                @Override
+                public void run() {
+                    completed = true;
+                }
+            });
+        }
+        return false;
+    }
+
+    private boolean containsViewTouch(ArrayList<OcrResultModel> ocrResultModels, String className, String text) {
+        Optional<OcrResultModel> ocrResultOptional = ocrResultModels.stream().filter(new Predicate<OcrResultModel>() {
+            @Override
+            public boolean test(OcrResultModel ocrResultModel) {
+                return ocrResultModel.getLabel().contains(text);
+            }
+        }).findAny();
+        if (ocrResultOptional.isPresent()) {
+            OcrResultModel ocrResult = ocrResultOptional.get();
+            List<Point> points = ocrResult.getPoints();
+            Log.e(TAG, "analysisUi: points = " + points);
+            Point point = getCenterPoint(points);
+            Log.e(TAG, "analysisUi: CenterPoint = " + point);
+            return clickByPoint(point.x, point.y, className, new Runnable() {
+                @Override
+                public void run() {
+                    completed = true;
+                }
+            });
+        }
+        return false;
+    }
+
+    /**
+     * Rec: 通讯录,0.9980758
+     * 5: Det: (250,1291) (304,1291) (304,1335) (250,1335)
+     *
+     * @param points
+     * @return
+     */
+    private Point getCenterPoint(List<Point> points) {
+        if (points.size() != 4) {
+            Log.e(TAG, "getCenterPoint: ");
+        } else {
+            int x1 = points.get(0).x;
+            int x2 = points.get(2).x;
+            int y1 = points.get(0).y;
+            int y2 = points.get(2).y;
+            int x = (x1 + x2) / 2;
+            int y = (y1 + y2) / 2;
+            return new Point(x, y);
+        }
+        return new Point();
+    }
+
+    private void processScreenImage(Bitmap bitmap) {
+        // 在这里处理获取到的屏幕图像
+        // 例如，可以发送广播或通过回调传递图像数据
+        Log.d(TAG, "Screen image captured: " + bitmap.getWidth() + "x" + bitmap.getHeight());
+        // 示例：保存图像到文件
+        // saveBitmapToFile(bitmap);
+
+    }
+
+    private Notification createNotification() {
+        Intent notificationIntent = new Intent(this, MainActivity.class);
+        PendingIntent pendingIntent = PendingIntent.getActivity(
+                this,
+                0,
+                notificationIntent,
+                PendingIntent.FLAG_IMMUTABLE
+        );
+
+        return new NotificationCompat.Builder(this, CHANNEL_ID)
+                .setContentTitle("孝心平安通讯OS服务")
+                .setContentText("正在运行中")
+                .setSmallIcon(R.mipmap.ic_launcher)
+                .setContentIntent(pendingIntent)
+                .build();
+    }
+
+    private void createNotificationChannel() {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
+            NotificationChannel channel = new NotificationChannel(
+                    CHANNEL_ID,
+                    "孝心平安通讯OS服务",
+                    NotificationManager.IMPORTANCE_DEFAULT
+            );
+            NotificationManager manager = getSystemService(NotificationManager.class);
+            manager.createNotificationChannel(channel);
+        }
+    }
+
+
+    public boolean onLoadModel() {
+        if (predictor.isLoaded()) {
+            predictor.releaseModel();
+        }
+        return predictor.init(getApplication(), modelPath, labelPath, useOpencl, cpuThreadNum,
+                cpuPowerMode, detLongSize, scoreThreshold);
+    }
+
+    public boolean onRunModel() {
+        /*检测 分类 识别*/
+        return predictor.isLoaded() && predictor.runModel(1, 0, 1);
+    }
 
     /**
      * 1.在微信页面直接找到联系人拨打电话
@@ -137,6 +776,7 @@ public class WeAccessibilityService extends AccessibilityService {
      *
      * @param event
      */
+    @Deprecated
     private void _onAccessibilityEvent(AccessibilityEvent event) {
         Log.e(TAG, "_onAccessibilityEvent: " + mCurrentStep);
         switch (mCurrentStep) {
@@ -148,87 +788,95 @@ public class WeAccessibilityService extends AccessibilityService {
                 break;
             default:
                 if (!TextUtils.isEmpty(event.getClassName())) {
-                    if ("com.tencent.mm.ui.LauncherUI".contentEquals(event.getClassName())) {
+                    String className = event.getClassName().toString();
+                    Log.e(TAG, "_onAccessibilityEvent: className = " + className);
+                    if (className.startsWith("com.")) {
+//                        mCurrentClassName = className;
+                    }
+                    if ("com.tencent.mm.ui.LauncherUI".equals(className)) {
                         if (mCurrentStep != Step.FIND_CONTACT) {
                             mCurrentStep = Step.CLICK_CONTACT;
                         }
+                    } else if ("com.tencent.mm.plugin.account.ui.WelcomeActivity".contentEquals(event.getClassName())) {
+                        mCurrentStep = Step.WAITING;
+                        Toaster.showLong("请先登录微信");
                     } else if ("com.tencent.mm.plugin.label.ui.ContactLabelManagerUI".contentEquals(event.getClassName())) {
 
                     }
                 }
         }
-        switch (mCurrentStep) {
-            case WAITING:
-                mAutoAccept = mMMKV.decodeBool(CommonConfig.WECHAT_CALL_AUTO_ACCEPT, false);
-                Log.e(TAG, "_onAccessibilityEvent: mAutoAccept = " + mAutoAccept);
-                if (!mAutoAccept) {
-                    return;
-                }
-                if (stepAnswer(Property.DESCRIPTION, RECEIVE_DESCRIPTION)) {
-                    mCurrentStep = Step.WECHAT_HANDS_FREE;
-                    Toast.makeText(this, "已自动接听视频/语音", Toast.LENGTH_LONG).show();
-                } else {
-                    mCurrentStep = Step.WAITING;
-//                    clickAnswer();
-                }
-                break;
-            case WECHAT_HANDS_FREE:
-                handsFree(Property.DESCRIPTION, HANDS_FREE_TEXT);
-                break;
-            case DIALER_HANDS_FREE:
-                if (findHandsFree(Property.DESCRIPTION, DIALER_HANDS_FREE_CLOSE_TEXT)) {
-                    dialerHandsFree(Property.TEXT, DIALER_HANDS_FREE_TEXT);
-                } else {
-                    mCurrentStep = Step.WAITING;
-                }
-                break;
-            case CLICK_HOME://主页能找到直接点击进去更多
-                stepHome(Property.TEXT, mName);
-                break;
-            case CLICK_QUICK_WECHAT_CALL://点击更多页面
-                step(Property.DESCRIPTION, MORE_NAME, Step.CLICK_TARGET);
-                break;
-            case CLICK_TARGET://点击视频通话
-                stepCall(Property.TEXT, PARENT_VIDEO_TEXT);
-                break;
-
-            case CLICK_CONTACT://进入通讯录界面
-                if (stepHome(Property.TEXT, CONTACT_TEXT, Step.FIND_TAG)) {
-                    Log.e(TAG, "_onAccessibilityEvent: enter contact");
-                } else {
-                    touchContact();
-                }
-                break;
-            case FIND_CONTACT://模拟滑动找到联系人
-                findContact(Property.TEXT, mName, Step.CLICK_NAME);
-                break;
-            case FIND_TAG:
-                step(Property.TEXT, TAG_TEXT, Step.CLICK_TAG);
-                break;
-            case CLICK_TAG:
-                step(Property.TEXT, mTagName, Step.CLICK_NAME);
-                break;
-            case CLICK_NAME://点击item
-                step(Property.TEXT, mName, Step.CLICK_INFO);
-                break;
-            case CLICK_INFO://进入个人信息页面
-                stepCallDialog(Property.TEXT, DIALER_TEXT, Step.CLICK_CALL);
-                break;
-            case CLICK_CALL://打视频或者电话
-                if (mCallType == TYPE_VIDEO) {
-                    step(Property.TEXT, VIDEO_TEXT, Step.WAITING);
-                } else if (mCallType == TYPE_VOICE) {
-                    step(Property.TEXT, CALL_TEXT, Step.WAITING);
-                }
-                break;
-//            case CLICK_VIDEO_CALL:
-//                if (step(Property.TEXT, VIDEO_TEXT)) {
-//                    Log.d(TAG, "finish, now: " + mCurrentStep);
-//                    Toast.makeText(this, "成功发起视频聊天", Toast.LENGTH_LONG).show();
+//        switch (mCurrentStep) {
+//            case WAITING:
+//                mAutoAccept = mMMKV.decodeBool(CommonConfig.WECHAT_CALL_AUTO_ACCEPT, false);
+//                Log.e(TAG, "_onAccessibilityEvent: mAutoAccept = " + mAutoAccept);
+//                if (!mAutoAccept) {
+//                    return;
+//                }
+//                if (stepAnswer(Property.DESCRIPTION, RECEIVE_DESCRIPTION)) {
+//                    mCurrentStep = Step.WECHAT_HANDS_FREE;
+//                    Toast.makeText(this, "已自动接听视频/语音", Toast.LENGTH_LONG).show();
+//                } else {
+//                    mCurrentStep = Step.WAITING;
+////                    clickAnswer();
 //                }
 //                break;
-            default:
-        }
+//            case WECHAT_HANDS_FREE:
+//                handsFree(Property.DESCRIPTION, HANDS_FREE_TEXT);
+//                break;
+//            case DIALER_HANDS_FREE:
+//                if (findHandsFree(Property.DESCRIPTION, DIALER_HANDS_FREE_CLOSE_TEXT)) {
+//                    dialerHandsFree(Property.TEXT, DIALER_HANDS_FREE_TEXT);
+//                } else {
+//                    mCurrentStep = Step.WAITING;
+//                }
+//                break;
+//            case CLICK_HOME://主页能找到直接点击进去更多
+//                stepHome(Property.TEXT, mName);
+//                break;
+//            case CLICK_QUICK_WECHAT_CALL://点击更多页面
+//                step(Property.DESCRIPTION, MORE_NAME, Step.CLICK_TARGET);
+//                break;
+//            case CLICK_TARGET://点击视频通话
+//                stepCall(Property.TEXT, PARENT_VIDEO_TEXT);
+//                break;
+//
+//            case CLICK_CONTACT://进入通讯录界面
+//                if (stepHome(Property.TEXT, CONTACT_TEXT, Step.FIND_TAG)) {
+//                    Log.e(TAG, "_onAccessibilityEvent: enter contact");
+//                } else {
+//                    touchContact();
+//                }
+//                break;
+//            case FIND_CONTACT://模拟滑动找到联系人
+//                findContact(Property.TEXT, mName, Step.CLICK_NAME);
+//                break;
+//            case FIND_TAG:
+//                step(Property.TEXT, TAG_TEXT, Step.CLICK_TAG);
+//                break;
+//            case CLICK_TAG:
+//                step(Property.TEXT, mTagName, Step.CLICK_NAME);
+//                break;
+//            case CLICK_NAME://点击item
+//                step(Property.TEXT, mName, Step.CLICK_INFO);
+//                break;
+//            case CLICK_INFO://进入个人信息页面
+//                stepCallDialog(Property.TEXT, DIALER_TEXT, Step.CLICK_CALL);
+//                break;
+//            case CLICK_CALL://打视频或者电话
+//                if (mCallType == TYPE_VIDEO) {
+//                    step(Property.TEXT, VIDEO_TEXT, Step.WAITING);
+//                } else if (mCallType == TYPE_VOICE) {
+//                    step(Property.TEXT, CALL_TEXT, Step.WAITING);
+//                }
+//                break;
+////            case CLICK_VIDEO_CALL:
+////                if (step(Property.TEXT, VIDEO_TEXT)) {
+////                    Log.d(TAG, "finish, now: " + mCurrentStep);
+////                    Toast.makeText(this, "成功发起视频聊天", Toast.LENGTH_LONG).show();
+////                }
+////                break;
+//            default:
+//        }
     }
 
     @Override
@@ -738,5 +1386,30 @@ public class WeAccessibilityService extends AccessibilityService {
         }
     }
 
+    public static final String STOP_RECORD_ACTION = "stop_screen_record";
+
+    private void registerStopReceiver() {
+        if (mStopBroadcastReceiver == null) {
+            mStopBroadcastReceiver = new StopBroadcastReceiver();
+        }
+        IntentFilter filter = new IntentFilter();
+        filter.addAction(STOP_RECORD_ACTION);
+        registerReceiver(mStopBroadcastReceiver, filter);
+    }
+
+    private StopBroadcastReceiver mStopBroadcastReceiver;
+
+    private class StopBroadcastReceiver extends BroadcastReceiver {
+
+        @Override
+        public void onReceive(Context context, Intent intent) {
+            Log.e("StopBroadcastReceiver", "onReceive: " + intent.getAction());
+            if (STOP_RECORD_ACTION.equals(intent.getAction())) {
+                mCurrentPackageName = "";
+                mCurrentClassName = "";
+                releaseDisplay();
+            }
+        }
+    }
 
 }

app/src/main/java/com/xxpatx/os/utils/Utils.java

@@ -66,7 +66,7 @@ public class Utils {
             Log.e("e", "读取设备序列号异常：" + e.toString());
         }
         if (BuildConfig.DEBUG) {
-//            return "TYZ185P5250100275";
+//            return "PATX1202407AC0020";
         }
         return serial;
     }

app/src/main/res/xml/accessibility_service_config.xml

@@ -1,9 +1,11 @@
 <accessibility-service xmlns:android="http://schemas.android.com/apk/res/android"
     android:accessibilityEventTypes="typeAllMask|typeViewClicked|typeViewFocused|typeWindowStateChanged|typeWindowsChanged|typeContextClicked|typeWindowContentChanged"
     android:accessibilityFeedbackType="feedbackAllMask|feedbackGeneric|feedbackSpoken"
-    android:accessibilityFlags="flagDefault|flagRetrieveInteractiveWindows|flagIncludeNotImportantViews"
+    android:accessibilityFlags="flagRetrieveInteractiveWindows|flagReportViewIds|flagRequestEnhancedWebAccessibility|flagIncludeNotImportantViews|flagDefault"
     android:canPerformGestures="true"
+    android:canRequestEnhancedWebAccessibility="true"
+    android:canRequestFilterKeyEvents="true"
     android:canRetrieveWindowContent="true"
     android:description="@string/accessibility_service_description"
-    android:notificationTimeout="100"
+    android:notificationTimeout="1000"
     android:packageNames="com.tencent.mm,com.android.dialer" />