opencv/platforms/android/build-tests/test_cmake_build.py

155 lines
5.7 KiB
Python
Raw Normal View History

2015-08-10 23:02:41 +08:00
#!/usr/bin/env python
import unittest
import os, sys, subprocess, argparse, shutil, re
2016-01-18 19:11:02 +08:00
import logging as log
log.basicConfig(format='%(message)s', level=log.DEBUG)
2015-08-10 23:02:41 +08:00
CMAKE_TEMPLATE='''\
CMAKE_MINIMUM_REQUIRED(VERSION 2.8)
2018-03-26 20:21:45 +08:00
# Enable C++11
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_CXX_STANDARD_REQUIRED TRUE)
2015-08-10 23:02:41 +08:00
SET(PROJECT_NAME hello-android)
PROJECT(${PROJECT_NAME})
2018-03-26 20:21:45 +08:00
2015-08-10 23:02:41 +08:00
FIND_PACKAGE(OpenCV REQUIRED %(libset)s)
FILE(GLOB srcs "*.cpp")
2018-03-26 20:21:45 +08:00
2015-08-10 23:02:41 +08:00
ADD_EXECUTABLE(${PROJECT_NAME} ${srcs})
TARGET_LINK_LIBRARIES(${PROJECT_NAME} ${OpenCV_LIBS} dl z)
'''
CPP_TEMPLATE = '''\
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
using namespace cv;
const char* message = "Hello Android!";
int main(int argc, char* argv[])
{
(void)argc; (void)argv;
printf("%s\\n", message);
2018-03-26 20:21:45 +08:00
Size textsize = getTextSize(message, FONT_HERSHEY_COMPLEX, 3, 5, 0);
2015-08-10 23:02:41 +08:00
Mat img(textsize.height + 20, textsize.width + 20, CV_32FC1, Scalar(230,230,230));
2018-03-26 20:21:45 +08:00
putText(img, message, Point(10, img.rows - 10), FONT_HERSHEY_COMPLEX, 3, Scalar(0, 0, 0), 5);
2015-08-10 23:02:41 +08:00
imwrite("/mnt/sdcard/HelloAndroid.png", img);
return 0;
}
'''
#===================================================================================================
class TestCmakeBuild(unittest.TestCase):
def __init__(self, libset, abi, cmake_vars, opencv_cmake_path, workdir, *args, **kwargs):
2015-08-10 23:02:41 +08:00
unittest.TestCase.__init__(self, *args, **kwargs)
self.libset = libset
self.abi = abi
self.cmake_vars = cmake_vars
2015-08-10 23:02:41 +08:00
self.opencv_cmake_path = opencv_cmake_path
self.workdir = workdir
self.srcdir = os.path.join(self.workdir, "src")
self.bindir = os.path.join(self.workdir, "build")
def shortDescription(self):
return "ABI: %s, LIBSET: %s" % (self.abi, self.libset)
def getCMakeToolchain(self):
if True:
toolchain = os.path.join(os.environ['ANDROID_NDK'], 'build', 'cmake', 'android.toolchain.cmake')
if os.path.exists(toolchain):
return toolchain
toolchain = os.path.join(self.opencv_cmake_path, "android.toolchain.cmake")
if os.path.exists(toolchain):
return toolchain
else:
raise Exception("Can't find toolchain")
2015-08-10 23:02:41 +08:00
def gen_cmakelists(self):
return CMAKE_TEMPLATE % {"libset": self.libset}
def gen_code(self):
return CPP_TEMPLATE
def write_src_file(self, fname, content):
with open(os.path.join(self.srcdir, fname), "w") as f:
f.write(content)
def setUp(self):
if os.path.exists(self.workdir):
shutil.rmtree(self.workdir)
os.mkdir(self.workdir)
os.mkdir(self.srcdir)
os.mkdir(self.bindir)
self.write_src_file("CMakeLists.txt", self.gen_cmakelists())
self.write_src_file("main.cpp", self.gen_code())
os.chdir(self.bindir)
def tearDown(self):
pass
#if os.path.exists(self.workdir):
# shutil.rmtree(self.workdir)
2015-08-10 23:02:41 +08:00
def runTest(self):
cmd = [
"cmake",
"-GNinja",
"-DOpenCV_DIR=%s" % self.opencv_cmake_path,
"-DCMAKE_TOOLCHAIN_FILE=%s" % self.getCMakeToolchain(),
2015-08-10 23:02:41 +08:00
self.srcdir
] + [ "-D{}={}".format(key, value) for key, value in self.cmake_vars.items() ]
2016-01-18 19:11:02 +08:00
log.info("Executing: %s" % cmd)
2015-08-10 23:02:41 +08:00
retcode = subprocess.call(cmd)
self.assertEqual(retcode, 0, "cmake failed")
cmd = ["ninja", "-v"]
2016-01-18 19:11:02 +08:00
log.info("Executing: %s" % cmd)
2015-08-10 23:02:41 +08:00
retcode = subprocess.call(cmd)
self.assertEqual(retcode, 0, "make failed")
def suite(workdir, opencv_cmake_path):
abis = {
"armeabi-v7a": { "ANDROID_ABI": "armeabi-v7a", "ANDROID_TOOLCHAIN": "clang", "ANDROID_STL": "c++_shared", 'ANDROID_NATIVE_API_LEVEL': "21" },
"arm64-v8a": { "ANDROID_ABI": "arm64-v8a", "ANDROID_TOOLCHAIN": "clang", "ANDROID_STL": "c++_shared", 'ANDROID_NATIVE_API_LEVEL': "21" },
"x86": { "ANDROID_ABI": "x86", "ANDROID_TOOLCHAIN": "clang", "ANDROID_STL": "c++_shared", 'ANDROID_NATIVE_API_LEVEL': "21" },
"x86_64": { "ANDROID_ABI": "x86_64", "ANDROID_TOOLCHAIN": "clang", "ANDROID_STL": "c++_shared", 'ANDROID_NATIVE_API_LEVEL': "21" },
2015-08-10 23:02:41 +08:00
}
suite = unittest.TestSuite()
for libset in ["", "opencv_java"]:
for abi, cmake_vars in abis.items():
suite.addTest(TestCmakeBuild(libset, abi, cmake_vars, opencv_cmake_path,
os.path.join(workdir, "{}-{}".format(abi, "static" if libset == "" else "shared"))))
2015-08-10 23:02:41 +08:00
return suite
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Test OpenCV for Android SDK with cmake')
parser.add_argument('--sdk_path', help="Path to Android SDK to use for build")
parser.add_argument('--ndk_path', help="Path to Android NDK to use for build")
parser.add_argument("--workdir", default="testspace", help="Working directory (and output)")
parser.add_argument("opencv_cmake_path", help="Path to folder with OpenCVConfig.cmake and android.toolchain.cmake (usually <SDK>/sdk/native/jni/")
args = parser.parse_args()
if args.sdk_path is not None:
os.environ["ANDROID_SDK"] = os.path.abspath(args.sdk_path)
if args.ndk_path is not None:
os.environ["ANDROID_NDK"] = os.path.abspath(args.ndk_path)
if not 'ANDROID_HOME' in os.environ and 'ANDROID_SDK' in os.environ:
os.environ['ANDROID_HOME'] = os.environ["ANDROID_SDK"]
2015-08-10 23:02:41 +08:00
print("Using SDK: %s" % os.environ["ANDROID_SDK"])
print("Using NDK: %s" % os.environ["ANDROID_NDK"])
workdir = os.path.abspath(args.workdir)
if not os.path.exists(workdir):
os.mkdir(workdir)
res = unittest.TextTestRunner(verbosity=3).run(suite(workdir, os.path.abspath(args.opencv_cmake_path)))
2015-08-10 23:02:41 +08:00
if not res.wasSuccessful():
sys.exit(res)