opencv

1
#!/usr/bin/env python
2
"""
3
This script runs OpenCV.framework tests for iOS.
4
"""
5

6
from __future__ import print_function
7
import glob, re, os, os.path, shutil, string, sys, argparse, traceback, multiprocessing
8
from subprocess import check_call, check_output, CalledProcessError
9

10
IPHONEOS_DEPLOYMENT_TARGET='9.0'  # default, can be changed via command line options or environment variable
11

12
def execute(cmd, cwd = None):
13
    print("Executing: %s in %s" % (cmd, cwd), file=sys.stderr)
14
    print('Executing: ' + ' '.join(cmd))
15
    retcode = check_call(cmd, cwd = cwd)
16
    if retcode != 0:
17
        raise Exception("Child returned:", retcode)
18

19
class TestRunner:
20
    def __init__(self, script_dir, tests_dir, build_dir, framework_dir, framework_name, arch, target, platform):
21
        self.script_dir = script_dir
22
        self.tests_dir = tests_dir
23
        self.build_dir = build_dir
24
        self.framework_dir = framework_dir
25
        self.framework_name = framework_name
26
        self.arch = arch
27
        self.target = target
28
        self.platform = platform
29

30
    def _run(self):
31
        if not os.path.isdir(self.build_dir):
32
            os.makedirs(self.build_dir)
33

34
        self.runTest()
35

36
    def run(self):
37
        try:
38
            self._run()
39
        except Exception as e:
40
            print("="*60, file=sys.stderr)
41
            print("ERROR: %s" % e, file=sys.stderr)
42
            print("="*60, file=sys.stderr)
43
            traceback.print_exc(file=sys.stderr)
44
            sys.exit(1)
45

46
    def getToolchain(self):
47
        return None
48

49
    def getCMakeArgs(self):
50
        args = [
51
            "cmake",
52
            "-GXcode",
53
            "-DFRAMEWORK_DIR=%s" % self.framework_dir,
54
            "-DFRAMEWORK_NAME=%s" % self.framework_name,
55
        ]
56
        return args
57

58
    def makeCMakeCmd(self):
59
        toolchain = self.getToolchain()
60
        cmakecmd = self.getCMakeArgs() + \
61
            (["-DCMAKE_TOOLCHAIN_FILE=%s" % toolchain] if toolchain is not None else []) + \
62
            ["-DCMAKE_INSTALL_NAME_TOOL=install_name_tool"]
63
        cmakecmd.append(self.tests_dir)
64
        return cmakecmd
65

66
    def runTest(self):
67
        cmakecmd = self.makeCMakeCmd()
68
        execute(cmakecmd, cwd = self.build_dir)
69
        buildcmd = self.getTestCommand()
70
        execute(buildcmd, cwd = self.build_dir)
71

72
    def getTestCommand(self):
73
        testcmd = [
74
            "xcodebuild",
75
            "test",
76
            "-project", "OpenCVTest.xcodeproj",
77
            "-scheme", "OpenCVTestTests",
78
            "-destination", "platform=%s" % self.platform
79
        ]
80
        return testcmd
81

82
class iOSTestRunner(TestRunner):
83

84
    def getToolchain(self):
85
        toolchain = os.path.join(self.script_dir, "cmake", "Toolchains", "Toolchain-%s_Xcode.cmake" % self.target)
86
        return toolchain
87

88
    def getCMakeArgs(self):
89
        args = TestRunner.getCMakeArgs(self)
90
        args = args + [
91
            "-DIOS_ARCH=%s" % self.arch,
92
            "-DIPHONEOS_DEPLOYMENT_TARGET=%s" % os.environ['IPHONEOS_DEPLOYMENT_TARGET'],
93
        ]
94
        return args
95

96
if __name__ == "__main__":
97
    script_dir = os.path.abspath(os.path.dirname(sys.argv[0]))
98
    parser = argparse.ArgumentParser(description='The script builds OpenCV.framework for iOS.')
99
    parser.add_argument('tests_dir', metavar='TEST_DIR', help='folder where test files are located')
100
    parser.add_argument('--build_dir', default=None, help='folder where test will be built (default is "../test_build" relative to tests_dir)')
101
    parser.add_argument('--framework_dir', default=None, help='folder where OpenCV framework is located')
102
    parser.add_argument('--framework_name', default='opencv2', help='Name of OpenCV framework (default: opencv2, will change to OpenCV in future version)')
103
    parser.add_argument('--iphoneos_deployment_target', default=os.environ.get('IPHONEOS_DEPLOYMENT_TARGET', IPHONEOS_DEPLOYMENT_TARGET), help='specify IPHONEOS_DEPLOYMENT_TARGET')
104
    parser.add_argument('--platform', default='iOS Simulator,name=iPhone 11', help='xcodebuild platform parameter (default is iOS 11 simulator)')
105
    args = parser.parse_args()
106

107
    os.environ['IPHONEOS_DEPLOYMENT_TARGET'] = args.iphoneos_deployment_target
108
    print('Using IPHONEOS_DEPLOYMENT_TARGET=' + os.environ['IPHONEOS_DEPLOYMENT_TARGET'])
109
    arch = "x86_64"
110
    target = "iPhoneSimulator"
111
    print('Using iPhoneSimulator ARCH=' + arch)
112

113
    r = iOSTestRunner(script_dir, args.tests_dir, args.build_dir if args.build_dir else os.path.join(args.tests_dir, "../test_build"), args.framework_dir, args.framework_name, arch, target, args.platform)
114
    r.run()
115