随风 Android10框架,build.py · 往事都随风/dlib-android - Gitee.com

#!/usr/bin/env python

# Copyright (c) 2017 Tzutalin

# Create by TzuTaLin

#

# Permission is hereby granted, free of charge, to any person obtaining a copy of

# this software and associated documentation files (the "Software"), to deal in

# the Software without restriction, including without limitation the rights to

# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of

# the Software, and to permit persons to whom the Software is furnished to do so,

# subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in all

# copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS

# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR

# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER

# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN

# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

import os

import sys

import argparse

import shutil

import subprocess

from subprocess import Popen, PIPE

DEVICE_ABI = 'armeabi-v7a'

class PrintColors:

HEADER = '\033[95m'

OKBLUE = '\033[94m'

OKGREEN = '\033[92m'

WARNING = '\033[93m'

FAIL = '\033[91m'

ENDC = '\033[0m'

BOLD = '\033[1m'

UNDERLINE = '\033[4m'

def parse_args():

"""

Parse input arguments

"""

parser = argparse.ArgumentParser(description='Build dlib-android')

parser.add_argument('--jobs', dest='jobs', help='Jobs to compile',

default=4, type=int)

parser.add_argument('--android_project', dest='android_project',

help='Android JNI folder path',

default=os.path.join('androidstudio-examples', 'dlib-android-app', 'dlib', 'src', 'main', 'jniLibs'), type=str)

parser.add_argument('--clean', action='store_true',

help='clean obj and binaries')

parser.add_argument('--test', action='store_true',

help='Push executable file to data/local/tmp/, and run them')

parser.add_argument('--debug', action='store_true',

help='Enable debug build')

args = parser.parse_args()

return args

def ndk_build(args):

jobs = str(args.jobs)

isDebug = '1' if args.debug else '0'

build_cmd = ['ndk-build',

'-j' + jobs,

'NDK_LOG=1',

'NDK_DEBUG=' + isDebug,

'V=0']

# Print the build command

print PrintColors.UNDERLINE + 'ndk build arguments:' + str(build_cmd) + PrintColors.ENDC

ret = subprocess.call(build_cmd)

if ret is not 0:

print PrintColors.FAIL + 'Build Error' + PrintColors.ENDC

os.sys.exit(1)

else:

print PrintColors.OKBLUE + 'Build Pass' + PrintColors.ENDC

def ndk_clean():

subprocess.call(['ndk-build', 'clean'])

def setDeviceABI():

global DEVICE_ABI

p = subprocess.Popen(

['adb', 'shell', 'getprop', 'ro.product.cpu.abi'], stdin=PIPE, stdout=PIPE, stderr=PIPE)

output, err = p.communicate(

b"input data that is passed to subprocess' stdin")

rc = p.returncode

if rc == 0:

if "x86" in output:

DEVICE_ABI = 'x86'

print PrintColors.OKBLUE + 'We will use ABI:' + DEVICE_ABI + ' binaries to test ' + PrintColors.ENDC

def test():

global DEVICE_ABI

# Test max_cost_assignment_ex daemon example

print '----max_cost_assignment_ex daemon test'

print '----Push svm_ex to phone device'

srcFolder = os.path.join('libs', DEVICE_ABI, 'max_cost_assignment_ex')

subprocess.call(['adb', 'push', srcFolder, '/data/local/tmp'])

print '----Execute /data/local/tmp/max_cost_assignment_ex'

subprocess.call(['adb', 'shell', './data/local/tmp/max_cost_assignment_ex'])

print '\n\n'

print 'Test dlib selective search'

print '----selective search algorithm'

srcFolder = os.path.join('data', 'lena.jpg')

print '----Push test image to data/local/tmp'

subprocess.call(['adb', 'push', srcFolder, '/data/local/tmp/lena.jpg'])

print '----Push daemon to /data/local/tmp'

srcFolder = os.path.join('libs', DEVICE_ABI, 'TestSelectiveSearch')

subprocess.call(['adb', 'push', srcFolder, '/data/local/tmp'])

print '----Execute /data/local/tmp/TestSelectiveSearch'

subprocess.call(

['adb', 'shell', './data/local/tmp/TestSelectiveSearch', '/data/local/tmp/lena.jpg'])

print '\n\n'

print 'Test face landmark'

srcFolder = os.path.join('data', 'lena.bmp')

subprocess.call(['adb', 'push', srcFolder, '/data/local/tmp'])

srcFolder = os.path.join('data', 'shape_predictor_68_face_landmarks.dat')

subprocess.call(['adb', 'push', srcFolder, '/data/local/tmp'])

srcFolder = os.path.join('libs', DEVICE_ABI, 'face_landmark')

subprocess.call(['adb', 'push', srcFolder, '/data/local/tmp'])

print '----Execute /data/local/tmp/face_lanmark'

subprocess.call(['adb', 'shell', './data/local/tmp/face_landmark',

'/data/local/tmp/shape_predictor_68_face_landmarks.dat', '/data/local/tmp/lena.bmp'])

def copytree(src, dst, symlinks=False, ignore=None):

if not os.path.exists(dst):

os.makedirs(dst)

for item in os.listdir(src):

s = os.path.join(src, item)

d = os.path.join(dst, item)

if os.path.isdir(s):

copytree(s, d, symlinks, ignore)

else:

if not os.path.exists(d) or os.stat(s).st_mtime - os.stat(d).st_mtime > 1:

shutil.copy2(s, d)

def copySoToAS(src, dst):

for root, subdirs, files in os.walk(src):

for f in files:

if f.endswith('.so'):

path = os.path.join(root, f)

abis = os.path.basename(os.path.dirname(path))

shutil.copy(path, os.path.join(dst, abis))

# TODO : Refactor

def test_cmake():

curr = os.getcwd()

buildFolder = 'build'

if os.path.exists(buildFolder):

shutil.rmtree(buildFolder)

os.mkdir(buildFolder)

os.chdir(buildFolder)

cmake_cmd = ['cmake']

build_cmd = ['make']

ABI = 'arm64-v8a'

# ABI = 'x86'

# TODO

NDK_PATH = '/home/darrenl/tools/android-ndk-r10e'

cmake_cmd = cmake_cmd + ['-DCMAKE_SYSTEM_NAME=Android', '-DCMAKE_SYSTEM_VERSION=21',

'-DCMAKE_ANDROID_ARCH_ABI=' + ABI, '-DCMAKE_ANDROID_STL_TYPE=gnustl_static',

'-DCMAKE_ANDROID_NDK=' + NDK_PATH]

cmake_cmd = cmake_cmd + ['-D', 'CMAKE_INSTALL_PREFIX=.']

cmake_cmd = cmake_cmd + ['..']

# Print the build command

print PrintColors.UNDERLINE + 'cmake build arguments:' + str(cmake_cmd) + PrintColors.ENDC

ret = subprocess.call(cmake_cmd)

ret = subprocess.call(build_cmd)

if ret is not 0:

print PrintColors.FAIL + 'Build Error' + PrintColors.ENDC

os.sys.exit(1)

else:

print PrintColors.OKBLUE + 'Build Pass' + PrintColors.ENDC

if __name__ == '__main__':

# Move to top-level

ROOT = os.path.dirname(os.path.abspath(__file__))

os.chdir(ROOT)

libs_dir = os.path.join(ROOT, 'libs')

obj_dir = os.path.join(ROOT, 'obj')

setDeviceABI()

args = parse_args()

if args.test:

os.environ['Test'] = "true"

if args.clean:

ndk_clean()

else:

ndk_build(args)

if args.android_project and os.path.exists(args.android_project):

srcFolder = os.path.join('libs')

copySoToAS(srcFolder, args.android_project)

if args.test:

test()

os.sys.exit(0)

一键复制

编辑

Web IDE

原始数据

按行查看

历史