mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
257 lines
12 KiB
257 lines
12 KiB
#!/usr/bin/env python |
|
|
|
import argparse |
|
import json |
|
from os import path |
|
import os |
|
import shutil |
|
import subprocess |
|
|
|
from build_java_shared_aar import cleanup, fill_template, get_compiled_aar_path, get_opencv_version |
|
|
|
|
|
ANDROID_PROJECT_TEMPLATE_DIR = path.join(path.dirname(__file__), "aar-template") |
|
TEMP_DIR = "build_static" |
|
ANDROID_PROJECT_DIR = path.join(TEMP_DIR, "AndroidProject") |
|
COMPILED_AAR_PATH_1 = path.join(ANDROID_PROJECT_DIR, "OpenCV/build/outputs/aar/OpenCV-release.aar") # original package name |
|
COMPILED_AAR_PATH_2 = path.join(ANDROID_PROJECT_DIR, "OpenCV/build/outputs/aar/opencv-release.aar") # lower case package name |
|
AAR_UNZIPPED_DIR = path.join(TEMP_DIR, "aar_unzipped") |
|
FINAL_AAR_PATH_TEMPLATE = "outputs/opencv_static_<OPENCV_VERSION>.aar" |
|
FINAL_REPO_PATH = "outputs/maven_repo" |
|
MAVEN_PACKAGE_NAME = "opencv-static" |
|
|
|
|
|
def get_list_of_opencv_libs(sdk_dir): |
|
files = os.listdir(path.join(sdk_dir, "sdk/native/staticlibs/arm64-v8a")) |
|
libs = [f[3:-2] for f in files if f[:3] == "lib" and f[-2:] == ".a"] |
|
return libs |
|
|
|
def get_list_of_3rdparty_libs(sdk_dir, abis): |
|
libs = [] |
|
for abi in abis: |
|
files = os.listdir(path.join(sdk_dir, "sdk/native/3rdparty/libs/" + abi)) |
|
cur_libs = [f[3:-2] for f in files if f[:3] == "lib" and f[-2:] == ".a"] |
|
for lib in cur_libs: |
|
if lib not in libs: |
|
libs.append(lib) |
|
return libs |
|
|
|
def add_printing_linked_libs(sdk_dir, opencv_libs): |
|
""" |
|
Modifies CMakeLists.txt file in Android project, so it prints linked libraries for each OpenCV library" |
|
""" |
|
sdk_jni_dir = sdk_dir + "/sdk/native/jni" |
|
with open(path.join(ANDROID_PROJECT_DIR, "OpenCV/src/main/cpp/CMakeLists.txt"), "a") as f: |
|
f.write('\nset(OpenCV_DIR "' + sdk_jni_dir + '")\n') |
|
f.write('find_package(OpenCV REQUIRED)\n') |
|
for lib_name in opencv_libs: |
|
output_filename_prefix = "linkedlibs." + lib_name + "." |
|
f.write('get_target_property(OUT "' + lib_name + '" INTERFACE_LINK_LIBRARIES)\n') |
|
f.write('file(WRITE "' + output_filename_prefix + '${ANDROID_ABI}.txt" "${OUT}")\n') |
|
|
|
def read_linked_libs(lib_name, abis): |
|
""" |
|
Reads linked libs for each OpenCV library from files, that was generated by gradle. See add_printing_linked_libs() |
|
""" |
|
deps_lists = [] |
|
for abi in abis: |
|
with open(path.join(ANDROID_PROJECT_DIR, "OpenCV/src/main/cpp", f"linkedlibs.{lib_name}.{abi}.txt")) as f: |
|
text = f.read() |
|
linked_libs = text.split(";") |
|
linked_libs = [x.replace("$<LINK_ONLY:", "").replace(">", "") for x in linked_libs] |
|
deps_lists.append(linked_libs) |
|
|
|
return merge_dependencies_lists(deps_lists) |
|
|
|
def merge_dependencies_lists(deps_lists): |
|
""" |
|
One library may have different dependencies for different ABIS. |
|
We need to merge them into one list with all the dependencies preserving the order. |
|
""" |
|
result = [] |
|
for d_list in deps_lists: |
|
for i in range(len(d_list)): |
|
if d_list[i] not in result: |
|
if i == 0: |
|
result.append(d_list[i]) |
|
else: |
|
index = result.index(d_list[i-1]) |
|
result = result[:index + 1] + [d_list[i]] + result[index + 1:] |
|
|
|
return result |
|
|
|
def convert_deps_list_to_prefab(linked_libs, opencv_libs, external_libs): |
|
""" |
|
Converting list of dependencies into prefab format. |
|
""" |
|
prefab_linked_libs = [] |
|
for lib in linked_libs: |
|
if (lib in opencv_libs) or (lib in external_libs): |
|
prefab_linked_libs.append(":" + lib) |
|
elif (lib[:3] == "lib" and lib[3:] in external_libs): |
|
prefab_linked_libs.append(":" + lib[3:]) |
|
elif lib == "ocv.3rdparty.android_mediandk": |
|
prefab_linked_libs += ["-landroid", "-llog", "-lmediandk"] |
|
print("Warning: manualy handled ocv.3rdparty.android_mediandk dependency") |
|
elif lib == "ocv.3rdparty.flatbuffers": |
|
print("Warning: manualy handled ocv.3rdparty.flatbuffers dependency") |
|
elif lib.startswith("ocv.3rdparty"): |
|
raise Exception("Unknown lib " + lib) |
|
else: |
|
prefab_linked_libs.append("-l" + lib) |
|
return prefab_linked_libs |
|
|
|
def main(args): |
|
opencv_version = get_opencv_version(args.opencv_sdk_path) |
|
abis = os.listdir(path.join(args.opencv_sdk_path, "sdk/native/libs")) |
|
final_aar_path = FINAL_AAR_PATH_TEMPLATE.replace("<OPENCV_VERSION>", opencv_version) |
|
sdk_dir = args.opencv_sdk_path |
|
|
|
print("Removing data from previous runs...") |
|
cleanup([TEMP_DIR, final_aar_path, path.join(FINAL_REPO_PATH, "org/opencv", MAVEN_PACKAGE_NAME)]) |
|
|
|
print("Preparing Android project...") |
|
# ANDROID_PROJECT_TEMPLATE_DIR contains an Android project template that creates AAR |
|
shutil.copytree(ANDROID_PROJECT_TEMPLATE_DIR, ANDROID_PROJECT_DIR) |
|
|
|
# Configuring the Android project to static C++ libs version |
|
fill_template(path.join(ANDROID_PROJECT_DIR, "OpenCV/build.gradle.template"), |
|
path.join(ANDROID_PROJECT_DIR, "OpenCV/build.gradle"), |
|
{"LIB_NAME": "templib", |
|
"LIB_TYPE": "c++_static", |
|
"PACKAGE_NAME": MAVEN_PACKAGE_NAME, |
|
"OPENCV_VERSION": opencv_version, |
|
"COMPILE_SDK": args.android_compile_sdk, |
|
"MIN_SDK": args.android_min_sdk, |
|
"TARGET_SDK": args.android_target_sdk, |
|
"ABI_FILTERS": ", ".join(['"' + x + '"' for x in abis]), |
|
"JAVA_VERSION": args.java_version, |
|
}) |
|
fill_template(path.join(ANDROID_PROJECT_DIR, "OpenCV/src/main/cpp/CMakeLists.txt.template"), |
|
path.join(ANDROID_PROJECT_DIR, "OpenCV/src/main/cpp/CMakeLists.txt"), |
|
{"LIB_NAME": "templib", "LIB_TYPE": "STATIC"}) |
|
|
|
local_props = "" |
|
if args.ndk_location: |
|
local_props += "ndk.dir=" + args.ndk_location + "\n" |
|
if args.cmake_location: |
|
local_props += "cmake.dir=" + args.cmake_location + "\n" |
|
|
|
if local_props: |
|
with open(path.join(ANDROID_PROJECT_DIR, "local.properties"), "wt") as f: |
|
f.write(local_props) |
|
|
|
opencv_libs = get_list_of_opencv_libs(sdk_dir) |
|
external_libs = get_list_of_3rdparty_libs(sdk_dir, abis) |
|
|
|
add_printing_linked_libs(sdk_dir, opencv_libs) |
|
|
|
print("Running gradle assembleRelease...") |
|
cmd = ["./gradlew", "assembleRelease"] |
|
if args.offline: |
|
cmd = cmd + ["--offline"] |
|
# Running gradle to build the Android project |
|
subprocess.run(cmd, shell=False, cwd=ANDROID_PROJECT_DIR, check=True) |
|
|
|
# The created AAR package contains only one empty libtemplib.a library. |
|
# We need to add OpenCV libraries manually. |
|
# AAR package is just a zip archive |
|
complied_aar_path = get_compiled_aar_path(COMPILED_AAR_PATH_1, COMPILED_AAR_PATH_2) # two possible paths |
|
shutil.unpack_archive(complied_aar_path, AAR_UNZIPPED_DIR, "zip") |
|
|
|
print("Adding libs to AAR...") |
|
|
|
# Copying 3rdparty libs from SDK into the AAR |
|
for lib in external_libs: |
|
for abi in abis: |
|
os.makedirs(path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi)) |
|
if path.exists(path.join(sdk_dir, "sdk/native/3rdparty/libs/" + abi, "lib" + lib + ".a")): |
|
shutil.copy(path.join(sdk_dir, "sdk/native/3rdparty/libs/" + abi, "lib" + lib + ".a"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi, "lib" + lib + ".a")) |
|
else: |
|
# One OpenCV library may have different dependency lists for different ABIs, but we can write only one |
|
# full dependency list for all ABIs. So we just add empty .a library if this ABI doesn't have this dependency. |
|
shutil.copy(path.join(AAR_UNZIPPED_DIR, "prefab/modules/templib/libs/android." + abi, "libtemplib.a"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi, "lib" + lib + ".a")) |
|
shutil.copy(path.join(AAR_UNZIPPED_DIR, "prefab/modules/templib/libs/android." + abi + "/abi.json"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi + "/abi.json")) |
|
shutil.copy(path.join(AAR_UNZIPPED_DIR, "prefab/modules/templib/module.json"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/module.json")) |
|
|
|
# Copying OpenV libs from SDK into the AAR |
|
for lib in opencv_libs: |
|
for abi in abis: |
|
os.makedirs(path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi)) |
|
shutil.copy(path.join(sdk_dir, "sdk/native/staticlibs/" + abi, "lib" + lib + ".a"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi, "lib" + lib + ".a")) |
|
shutil.copy(path.join(AAR_UNZIPPED_DIR, "prefab/modules/templib/libs/android." + abi + "/abi.json"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/libs/android." + abi + "/abi.json")) |
|
os.makedirs(path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/include/opencv2")) |
|
shutil.copy(path.join(sdk_dir, "sdk/native/jni/include/opencv2/" + lib.replace("opencv_", "") + ".hpp"), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/include/opencv2/" + lib.replace("opencv_", "") + ".hpp")) |
|
module_include_folder = path.join(sdk_dir, "sdk/native/jni/include/opencv2/" + lib.replace("opencv_", "")) |
|
if os.path.exists(module_include_folder): |
|
shutil.copytree(module_include_folder, |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/include/opencv2/" + lib.replace("opencv_", ""))) |
|
|
|
# Adding dependencies list |
|
module_json_text = { |
|
"export_libraries": convert_deps_list_to_prefab(read_linked_libs(lib, abis), opencv_libs, external_libs), |
|
"android": {}, |
|
} |
|
with open(path.join(AAR_UNZIPPED_DIR, "prefab/modules/" + lib + "/module.json"), "w") as f: |
|
json.dump(module_json_text, f) |
|
|
|
for h_file in ("cvconfig.h", "opencv.hpp", "opencv_modules.hpp"): |
|
shutil.copy(path.join(sdk_dir, "sdk/native/jni/include/opencv2/" + h_file), |
|
path.join(AAR_UNZIPPED_DIR, "prefab/modules/opencv_core/include/opencv2/" + h_file)) |
|
|
|
|
|
shutil.rmtree(path.join(AAR_UNZIPPED_DIR, "prefab/modules/templib")) |
|
|
|
# Creating final AAR zip archive |
|
os.makedirs("outputs", exist_ok=True) |
|
shutil.make_archive(final_aar_path, "zip", AAR_UNZIPPED_DIR, ".") |
|
os.rename(final_aar_path + ".zip", final_aar_path) |
|
|
|
print("Creating local maven repo...") |
|
|
|
shutil.copy(final_aar_path, path.join(ANDROID_PROJECT_DIR, "OpenCV/opencv-release.aar")) |
|
|
|
print("Creating a maven repo from project sources (with sources jar and javadoc jar)...") |
|
cmd = ["./gradlew", "publishReleasePublicationToMyrepoRepository"] |
|
if args.offline: |
|
cmd = cmd + ["--offline"] |
|
subprocess.run(cmd, shell=False, cwd=ANDROID_PROJECT_DIR, check=True) |
|
|
|
os.makedirs(path.join(FINAL_REPO_PATH, "org/opencv"), exist_ok=True) |
|
shutil.move(path.join(ANDROID_PROJECT_DIR, "OpenCV/build/repo/org/opencv", MAVEN_PACKAGE_NAME), |
|
path.join(FINAL_REPO_PATH, "org/opencv", MAVEN_PACKAGE_NAME)) |
|
|
|
print("Creating a maven repo from modified AAR (with cpp libraries)...") |
|
cmd = ["./gradlew", "publishModifiedPublicationToMyrepoRepository"] |
|
if args.offline: |
|
cmd = cmd + ["--offline"] |
|
subprocess.run(cmd, shell=False, cwd=ANDROID_PROJECT_DIR, check=True) |
|
|
|
# Replacing AAR from the first maven repo with modified AAR from the second maven repo |
|
shutil.copytree(path.join(ANDROID_PROJECT_DIR, "OpenCV/build/repo/org/opencv", MAVEN_PACKAGE_NAME), |
|
path.join(FINAL_REPO_PATH, "org/opencv", MAVEN_PACKAGE_NAME), |
|
dirs_exist_ok=True) |
|
|
|
print("Done") |
|
|
|
|
|
if __name__ == "__main__": |
|
parser = argparse.ArgumentParser(description="Builds AAR with static C++ libs from OpenCV SDK") |
|
parser.add_argument('opencv_sdk_path') |
|
parser.add_argument('--android_compile_sdk', default="31") |
|
parser.add_argument('--android_min_sdk', default="21") |
|
parser.add_argument('--android_target_sdk', default="31") |
|
parser.add_argument('--java_version', default="1_8") |
|
parser.add_argument('--ndk_location', default="") |
|
parser.add_argument('--cmake_location', default="") |
|
parser.add_argument('--offline', action="store_true", help="Force Gradle use offline mode") |
|
args = parser.parse_args() |
|
|
|
main(args)
|
|
|