2011-09-06 23:30:28 +08:00
import sys , os , platform , xml , re , tempfile , glob , datetime , getpass
from optparse import OptionParser
from subprocess import Popen , PIPE
hostos = os . name # 'nt', 'posix'
hostmachine = platform . machine ( ) # 'x86', 'AMD64', 'x86_64'
2012-01-24 00:28:46 +08:00
SIMD_DETECTION_PROGRAM = """
#if __SSE5__
# error SSE5
#endif
#if __AVX2__
# error AVX2
#endif
#if __AVX__
# error AVX
#endif
#if __SSE4_2__
# error SSE4.2
#endif
#if __SSE4_1__
# error SSE4.1
#endif
#if __SSSE3__
# error SSSE3
#endif
#if __SSE3__
# error SSE3
#endif
#if __AES__
# error AES
#endif
#if __SSE2__
# error SSE2
#endif
#if __SSE__
# error SSE
#endif
#if __3dNOW__
# error 3dNOW
#endif
#if __MMX__
# error MMX
#endif
#if __ARM_NEON__
# error NEON
#endif
#error NOSIMD
"""
2011-09-06 23:30:28 +08:00
parse_patterns = (
2012-01-24 19:46:11 +08:00
{ ' name ' : " has_perf_tests " , ' default ' : " OFF " , ' pattern ' : re . compile ( " ^BUILD_PERF_TESTS:BOOL=(ON)$ " ) } ,
2012-04-05 20:53:25 +08:00
{ ' name ' : " has_accuracy_tests " , ' default ' : " OFF " , ' pattern ' : re . compile ( " ^BUILD_TESTS:BOOL=(ON)$ " ) } ,
2012-01-24 19:46:11 +08:00
{ ' name ' : " cmake_home " , ' default ' : None , ' pattern ' : re . compile ( " ^CMAKE_HOME_DIRECTORY:INTERNAL=(.+)$ " ) } ,
{ ' name ' : " opencv_home " , ' default ' : None , ' pattern ' : re . compile ( " ^OpenCV_SOURCE_DIR:STATIC=(.+)$ " ) } ,
{ ' name ' : " tests_dir " , ' default ' : None , ' pattern ' : re . compile ( " ^EXECUTABLE_OUTPUT_PATH:PATH=(.+)$ " ) } ,
{ ' name ' : " build_type " , ' default ' : " Release " , ' pattern ' : re . compile ( " ^CMAKE_BUILD_TYPE:STRING=(.*)$ " ) } ,
{ ' name ' : " svnversion_path " , ' default ' : None , ' pattern ' : re . compile ( " ^SVNVERSION_PATH:FILEPATH=(.*)$ " ) } ,
{ ' name ' : " cxx_flags " , ' default ' : " " , ' pattern ' : re . compile ( " ^CMAKE_CXX_FLAGS:STRING=(.*)$ " ) } ,
{ ' name ' : " cxx_flags_debug " , ' default ' : " " , ' pattern ' : re . compile ( " ^CMAKE_CXX_FLAGS_DEBUG:STRING=(.*)$ " ) } ,
{ ' name ' : " cxx_flags_release " , ' default ' : " " , ' pattern ' : re . compile ( " ^CMAKE_CXX_FLAGS_RELEASE:STRING=(.*)$ " ) } ,
{ ' name ' : " opencv_cxx_flags " , ' default ' : " " , ' pattern ' : re . compile ( " ^OPENCV_EXTRA_C_FLAGS:INTERNAL=(.*)$ " ) } ,
{ ' name ' : " opencv_cxx_flags_debug " , ' default ' : " " , ' pattern ' : re . compile ( " ^OPENCV_EXTRA_C_FLAGS_DEBUG:INTERNAL=(.*)$ " ) } ,
{ ' name ' : " opencv_cxx_flags_release " , ' default ' : " " , ' pattern ' : re . compile ( " ^OPENCV_EXTRA_C_FLAGS_RELEASE:INTERNAL=(.*)$ " ) } ,
{ ' name ' : " cxx_flags_android " , ' default ' : None , ' pattern ' : re . compile ( " ^ANDROID_CXX_FLAGS:INTERNAL=(.*)$ " ) } ,
{ ' name ' : " cxx_compiler_path " , ' default ' : None , ' pattern ' : re . compile ( " ^CMAKE_CXX_COMPILER:FILEPATH=(.*)$ " ) } ,
{ ' name ' : " ndk_path " , ' default ' : None , ' pattern ' : re . compile ( " ^(?:ANDROID_NDK|ANDROID_STANDALONE_TOOLCHAIN)?:PATH=(.*)$ " ) } ,
{ ' name ' : " android_abi " , ' default ' : None , ' pattern ' : re . compile ( " ^ANDROID_ABI:STRING=(.*)$ " ) } ,
{ ' name ' : " android_executable " , ' default ' : None , ' pattern ' : re . compile ( " ^ANDROID_EXECUTABLE:FILEPATH=(.*android.*)$ " ) } ,
{ ' name ' : " is_x64 " , ' default ' : " OFF " , ' pattern ' : re . compile ( " ^CUDA_64_BIT_DEVICE_CODE:BOOL=(ON)$ " ) } , #ugly(
{ ' name ' : " cmake_generator " , ' default ' : None , ' pattern ' : re . compile ( " ^CMAKE_GENERATOR:INTERNAL=(.+)$ " ) } ,
{ ' name ' : " cxx_compiler " , ' default ' : None , ' pattern ' : re . compile ( " ^CMAKE_CXX_COMPILER:FILEPATH=(.+)$ " ) } ,
{ ' name ' : " with_cuda " , ' default ' : " OFF " , ' pattern ' : re . compile ( " ^WITH_CUDA:BOOL=(ON)$ " ) } ,
{ ' name ' : " cuda_library " , ' default ' : None , ' pattern ' : re . compile ( " ^CUDA_CUDA_LIBRARY:FILEPATH=(.+)$ " ) } ,
{ ' name ' : " core_dependencies " , ' default ' : None , ' pattern ' : re . compile ( " ^opencv_core_LIB_DEPENDS:STATIC=(.+)$ " ) } ,
2011-09-06 23:30:28 +08:00
)
def query_yes_no ( stdout , question , default = " yes " ) :
valid = { " yes " : True , " y " : True , " ye " : True , " no " : False , " n " : False }
if default == None :
prompt = " [y/n] "
elif default == " yes " :
prompt = " [Y/n] "
elif default == " no " :
prompt = " [y/N] "
else :
raise ValueError ( " invalid default answer: ' %s ' " % default )
while True :
stdout . write ( os . linesep + question + prompt )
choice = raw_input ( ) . lower ( )
if default is not None and choice == ' ' :
return valid [ default ]
elif choice in valid :
return valid [ choice ]
else :
stdout . write ( " Please respond with ' yes ' or ' no ' " \
" (or ' y ' or ' n ' ). \n " )
2011-09-15 14:31:59 +08:00
2011-09-15 15:35:05 +08:00
def getRunningProcessExePathByName_win32 ( name ) :
2011-09-15 14:31:59 +08:00
from ctypes import windll , POINTER , pointer , Structure , sizeof
from ctypes import c_long , c_int , c_uint , c_char , c_ubyte , c_char_p , c_void_p
class PROCESSENTRY32 ( Structure ) :
_fields_ = [ ( ' dwSize ' , c_uint ) ,
( ' cntUsage ' , c_uint ) ,
( ' th32ProcessID ' , c_uint ) ,
( ' th32DefaultHeapID ' , c_uint ) ,
( ' th32ModuleID ' , c_uint ) ,
( ' cntThreads ' , c_uint ) ,
( ' th32ParentProcessID ' , c_uint ) ,
( ' pcPriClassBase ' , c_long ) ,
( ' dwFlags ' , c_uint ) ,
( ' szExeFile ' , c_char * 260 ) ,
( ' th32MemoryBase ' , c_long ) ,
( ' th32AccessKey ' , c_long ) ]
class MODULEENTRY32 ( Structure ) :
_fields_ = [ ( ' dwSize ' , c_long ) ,
( ' th32ModuleID ' , c_long ) ,
( ' th32ProcessID ' , c_long ) ,
( ' GlblcntUsage ' , c_long ) ,
( ' ProccntUsage ' , c_long ) ,
( ' modBaseAddr ' , c_long ) ,
( ' modBaseSize ' , c_long ) ,
( ' hModule ' , c_void_p ) ,
( ' szModule ' , c_char * 256 ) ,
( ' szExePath ' , c_char * 260 ) ]
TH32CS_SNAPPROCESS = 2
TH32CS_SNAPMODULE = 0x00000008
## CreateToolhelp32Snapshot
CreateToolhelp32Snapshot = windll . kernel32 . CreateToolhelp32Snapshot
CreateToolhelp32Snapshot . reltype = c_long
CreateToolhelp32Snapshot . argtypes = [ c_int , c_int ]
## Process32First
Process32First = windll . kernel32 . Process32First
Process32First . argtypes = [ c_void_p , POINTER ( PROCESSENTRY32 ) ]
Process32First . rettype = c_int
## Process32Next
Process32Next = windll . kernel32 . Process32Next
Process32Next . argtypes = [ c_void_p , POINTER ( PROCESSENTRY32 ) ]
Process32Next . rettype = c_int
## CloseHandle
CloseHandle = windll . kernel32 . CloseHandle
CloseHandle . argtypes = [ c_void_p ]
CloseHandle . rettype = c_int
## Module32First
Module32First = windll . kernel32 . Module32First
Module32First . argtypes = [ c_void_p , POINTER ( MODULEENTRY32 ) ]
Module32First . rettype = c_int
hProcessSnap = c_void_p ( 0 )
hProcessSnap = CreateToolhelp32Snapshot ( TH32CS_SNAPPROCESS , 0 )
pe32 = PROCESSENTRY32 ( )
pe32 . dwSize = sizeof ( PROCESSENTRY32 )
ret = Process32First ( hProcessSnap , pointer ( pe32 ) )
path = None
while ret :
if name + " .exe " == pe32 . szExeFile :
hModuleSnap = c_void_p ( 0 )
me32 = MODULEENTRY32 ( )
me32 . dwSize = sizeof ( MODULEENTRY32 )
hModuleSnap = CreateToolhelp32Snapshot ( TH32CS_SNAPMODULE , pe32 . th32ProcessID )
ret = Module32First ( hModuleSnap , pointer ( me32 ) )
path = me32 . szExePath
CloseHandle ( hModuleSnap )
if path :
break
ret = Process32Next ( hProcessSnap , pointer ( pe32 ) )
CloseHandle ( hProcessSnap )
return path
2011-09-15 15:35:05 +08:00
def getRunningProcessExePathByName_posix ( name ) :
pids = [ pid for pid in os . listdir ( ' /proc ' ) if pid . isdigit ( ) ]
for pid in pids :
try :
path = os . readlink ( os . path . join ( ' /proc ' , pid , ' exe ' ) )
if path and path . endswith ( name ) :
return path
except :
pass
2011-09-15 14:31:59 +08:00
def getRunningProcessExePathByName ( name ) :
try :
if hostos == " nt " :
return getRunningProcessExePathByName_win32 ( name )
2011-09-15 15:35:05 +08:00
elif hostos == " posix " :
return getRunningProcessExePathByName_posix ( name )
2011-09-15 14:31:59 +08:00
else :
return None
except :
return None
2011-09-06 23:30:28 +08:00
class RunInfo ( object ) :
2011-12-19 15:43:07 +08:00
def __init__ ( self , path , options ) :
2012-01-24 00:28:46 +08:00
self . options = options
2011-09-06 23:30:28 +08:00
self . path = path
self . error = None
2012-04-05 20:53:25 +08:00
self . nameprefix = " opencv_ " + options . mode + " _ "
2011-09-06 23:30:28 +08:00
for p in parse_patterns :
setattr ( self , p [ " name " ] , p [ " default " ] )
cachefile = open ( os . path . join ( path , " CMakeCache.txt " ) , " rt " )
try :
for l in cachefile . readlines ( ) :
ll = l . strip ( )
if not ll or ll . startswith ( " # " ) :
continue
for p in parse_patterns :
match = p [ " pattern " ] . match ( ll )
if match :
value = match . groups ( ) [ 0 ]
if value and not value . endswith ( " -NOTFOUND " ) :
setattr ( self , p [ " name " ] , value )
except :
pass
cachefile . close ( )
2011-09-08 19:01:45 +08:00
2011-09-06 23:30:28 +08:00
# fix empty tests dir
if not self . tests_dir :
self . tests_dir = self . path
# add path to adb
if self . android_executable :
self . adb = os . path . join ( os . path . dirname ( os . path . dirname ( self . android_executable ) ) , ( " platform-tools/adb " , " platform-tools/adb.exe " ) [ hostos == ' nt ' ] )
2011-09-15 14:31:59 +08:00
if not os . path . isfile ( self . adb ) or not os . access ( self . adb , os . X_OK ) :
self . adb = None
2011-09-06 23:30:28 +08:00
else :
self . adb = None
# detect target platform
2011-11-16 22:32:10 +08:00
if self . android_executable or self . android_abi or self . ndk_path :
2011-09-06 23:30:28 +08:00
self . targetos = " android "
2011-09-15 14:31:59 +08:00
else :
self . targetos = hostos
if self . targetos == " android " :
# fix adb tool location
if not self . adb :
self . adb = getRunningProcessExePathByName ( " adb " )
2011-09-06 23:30:28 +08:00
if not self . adb :
2011-12-19 15:43:07 +08:00
self . adb = " adb "
if options . adb_serial :
self . adb = [ self . adb , " -s " , options . adb_serial ]
else :
self . adb = [ self . adb ]
try :
output = Popen ( self . adb + [ " shell " , " ls " ] , stdout = PIPE , stderr = PIPE ) . communicate ( )
except OSError :
self . adb = [ ]
# remember current device serial. Needed if another device is connected while this script runs
if self . adb and not options . adb_serial :
adb_res = self . runAdb ( " devices " )
if not adb_res :
self . error = " Could not run adb command: %s (for %s ) " % ( self . error , self . path )
self . adb = [ ]
else :
connected_devices = re . findall ( r " ^[^ \ t]+[ \ t]+device \ r?$ " , adb_res , re . MULTILINE )
if len ( connected_devices ) != 1 :
self . error = " Too many ( %s ) devices are connected. Please specify single device using --serial option " % ( len ( connected_devices ) )
self . adb = [ ]
else :
adb_serial = connected_devices [ 0 ] . split ( " \t " ) [ 0 ]
self . adb = self . adb + [ " -s " , adb_serial ]
print " adb command: " , " " . join ( self . adb )
2011-09-06 23:30:28 +08:00
# fix has_perf_tests param
self . has_perf_tests = self . has_perf_tests == " ON "
2012-04-05 20:53:25 +08:00
self . has_accuracy_tests = self . has_accuracy_tests == " ON "
2011-09-06 23:30:28 +08:00
# fix is_x64 flag
self . is_x64 = self . is_x64 == " ON "
if not self . is_x64 and ( " X64 " in " %s %s %s " % ( self . cxx_flags , self . cxx_flags_release , self . cxx_flags_debug ) or " Win64 " in self . cmake_generator ) :
self . is_x64 = True
# fix test path
if " Visual Studio " in self . cmake_generator :
2011-12-19 15:43:07 +08:00
if options . configuration :
self . tests_dir = os . path . join ( self . tests_dir , options . configuration )
2011-09-08 14:27:46 +08:00
else :
self . tests_dir = os . path . join ( self . tests_dir , self . build_type )
2011-09-06 23:30:28 +08:00
elif not self . is_x64 and self . cxx_compiler :
#one more attempt to detect x64 compiler
try :
output = Popen ( [ self . cxx_compiler , " -v " ] , stdout = PIPE , stderr = PIPE ) . communicate ( )
if not output [ 0 ] and " x86_64 " in output [ 1 ] :
self . is_x64 = True
except OSError :
pass
# detect target arch
if self . targetos == " android " :
2012-01-24 00:28:46 +08:00
if " armeabi-v7a " in self . android_abi :
self . targetarch = " ARMv7a "
elif " armeabi-v6 " in self . android_abi :
self . targetarch = " ARMv6 "
elif " armeabi " in self . android_abi :
self . targetarch = " ARMv5te "
elif " x86 " in self . android_abi :
self . targetarch = " x86 "
else :
self . targetarch = " ARM "
2011-09-06 23:30:28 +08:00
elif self . is_x64 and hostmachine in [ " AMD64 " , " x86_64 " ] :
self . targetarch = " x64 "
elif hostmachine in [ " x86 " , " AMD64 " , " x86_64 " ] :
self . targetarch = " x86 "
else :
self . targetarch = " unknown "
2011-09-08 19:01:45 +08:00
# fix CUDA attributes
self . with_cuda = self . with_cuda == " ON "
if self . cuda_library and self . cuda_library . endswith ( " -NOTFOUND " ) :
self . cuda_library = None
self . has_cuda = self . with_cuda and self . cuda_library and self . targetarch in [ " x86 " , " x64 " ]
2011-09-06 23:30:28 +08:00
self . hardware = None
self . getSvnVersion ( self . cmake_home , " cmake_home_svn " )
if self . opencv_home == self . cmake_home :
self . opencv_home_svn = self . cmake_home_svn
else :
self . getSvnVersion ( self . opencv_home , " opencv_home_svn " )
self . tests = self . getAvailableTestApps ( )
def getSvnVersion ( self , path , name ) :
if not path :
setattr ( self , name , None )
return
if not self . svnversion_path and hostos == ' nt ' :
self . tryGetSvnVersionWithTortoise ( path , name )
else :
svnversion = self . svnversion_path
if not svnversion :
svnversion = " svnversion "
try :
output = Popen ( [ svnversion , " -n " , path ] , stdout = PIPE , stderr = PIPE ) . communicate ( )
if not output [ 1 ] :
setattr ( self , name , output [ 0 ] )
else :
setattr ( self , name , None )
except OSError :
setattr ( self , name , None )
def tryGetSvnVersionWithTortoise ( self , path , name ) :
try :
wcrev = " SubWCRev.exe "
dir = tempfile . mkdtemp ( )
#print dir
tmpfilename = os . path . join ( dir , " svn.tmp " )
tmpfilename2 = os . path . join ( dir , " svn_out.tmp " )
tmpfile = open ( tmpfilename , " w " )
tmpfile . write ( " $WCRANGE$$WCMODS?M:$ " )
tmpfile . close ( ) ;
output = Popen ( [ wcrev , path , tmpfilename , tmpfilename2 , " -f " ] , stdout = PIPE , stderr = PIPE ) . communicate ( )
if " is not a working copy " in output [ 0 ] :
version = " exported "
else :
tmpfile = open ( tmpfilename2 , " r " )
version = tmpfile . read ( )
tmpfile . close ( )
setattr ( self , name , version )
except :
setattr ( self , name , None )
finally :
if dir :
import shutil
shutil . rmtree ( dir )
def isTest ( self , fullpath ) :
if not os . path . isfile ( fullpath ) :
return False
if hostos == self . targetos :
return os . access ( fullpath , os . X_OK )
return True
def getAvailableTestApps ( self ) :
if self . tests_dir and os . path . isdir ( self . tests_dir ) :
2012-04-05 20:53:25 +08:00
files = glob . glob ( os . path . join ( self . tests_dir , self . nameprefix + " * " ) )
2011-09-06 23:30:28 +08:00
if self . targetos == hostos :
files = [ f for f in files if self . isTest ( f ) ]
return files
return [ ]
def getLogName ( self , app , timestamp ) :
app = os . path . basename ( app )
if app . endswith ( " .exe " ) :
app = app [ : - 4 ]
2012-04-05 20:53:25 +08:00
if app . startswith ( self . nameprefix ) :
app = app [ len ( self . nameprefix ) : ]
2012-01-24 00:28:46 +08:00
2011-09-06 23:30:28 +08:00
if self . cmake_home_svn :
if self . cmake_home_svn == self . opencv_home_svn :
rev = self . cmake_home_svn
elif self . opencv_home_svn :
rev = self . cmake_home_svn + " - " + self . opencv_home_svn
else :
rev = self . cmake_home_svn
else :
rev = None
if rev :
2012-01-24 00:28:46 +08:00
rev = rev . replace ( " : " , " to " )
2011-09-06 23:30:28 +08:00
else :
rev = " "
2012-01-24 00:28:46 +08:00
if self . options . useLongNames :
if not rev :
rev = " unknown "
tstamp = timestamp . strftime ( " % Y % m %d - % H % M % S " )
features = [ ]
#OS
_os = " "
if self . targetos == " android " :
_os = " Android " + self . runAdb ( " shell " , " getprop ro.build.version.release " ) . strip ( )
else :
mv = platform . mac_ver ( )
if mv [ 0 ] :
_os = " Darwin " + mv [ 0 ]
else :
wv = platform . win32_ver ( )
if wv [ 0 ] :
_os = " Windows " + wv [ 0 ]
else :
lv = platform . linux_distribution ( )
if lv [ 0 ] :
_os = lv [ 0 ] + lv [ 1 ]
else :
_os = self . targetos
features . append ( _os )
#HW(x86, x64, ARMv7a)
if self . targetarch :
features . append ( self . targetarch )
#TBB
if " ;tbb; " in self . core_dependencies :
features . append ( " TBB " )
#CUDA
if self . has_cuda :
#TODO: determine compute capability
features . append ( " CUDA " )
#SIMD
compiler_output = " "
try :
tmpfile = tempfile . mkstemp ( suffix = " .cpp " , text = True )
fd = os . fdopen ( tmpfile [ 0 ] , " w+b " )
fd . write ( SIMD_DETECTION_PROGRAM )
fd . close ( ) ;
options = [ self . cxx_compiler_path ]
2012-01-24 19:46:11 +08:00
cxx_flags = self . cxx_flags + " " + self . cxx_flags_release + " " + self . opencv_cxx_flags + " " + self . opencv_cxx_flags_release
2012-01-24 00:28:46 +08:00
if self . targetos == " android " :
cxx_flags = self . cxx_flags_android + " " + cxx_flags
prev_option = None
for opt in cxx_flags . split ( " " ) :
if opt . count ( ' \" ' ) % 2 == 1 :
if prev_option is None :
prev_option = opt
else :
options . append ( prev_option + " " + opt )
prev_option = None
elif prev_option is None :
options . append ( opt )
else :
prev_option = prev_option + " " + opt
options . append ( tmpfile [ 1 ] )
2012-01-24 19:46:11 +08:00
print options
2012-01-24 00:28:46 +08:00
output = Popen ( options , stdout = PIPE , stderr = PIPE ) . communicate ( )
compiler_output = output [ 1 ]
os . remove ( tmpfile [ 1 ] )
except OSError :
pass
if compiler_output :
m = re . search ( " #error \ W+( \ w+) " , compiler_output )
if m :
features . append ( m . group ( 1 ) )
#fin
return " %s __ %s __ %s __ %s .xml " % ( app , rev , tstamp , " _ " . join ( features ) )
2011-09-06 23:30:28 +08:00
else :
2012-01-24 00:28:46 +08:00
if rev :
rev = rev + " _ "
if self . hardware :
hw = str ( self . hardware ) . replace ( " " , " _ " ) + " _ "
elif self . has_cuda :
hw = " CUDA_ "
else :
hw = " "
tstamp = timestamp . strftime ( " % Y- % m- %d -- % H- % M- % S " )
return " %s _ %s _ %s _ %s %s %s .xml " % ( app , self . targetos , self . targetarch , hw , rev , tstamp )
2011-09-06 23:30:28 +08:00
def getTest ( self , name ) :
# full path
if self . isTest ( name ) :
return name
# name only
fullname = os . path . join ( self . tests_dir , name )
if self . isTest ( fullname ) :
return fullname
# name without extension
fullname + = " .exe "
if self . isTest ( fullname ) :
return fullname
# short name for OpenCV tests
for t in self . tests :
if t == name :
return t
fname = os . path . basename ( t )
if fname == name :
return t
if fname . endswith ( " .exe " ) :
fname = fname [ : - 4 ]
if fname == name :
return t
2012-04-05 20:53:25 +08:00
if fname . startswith ( self . nameprefix ) :
fname = fname [ len ( self . nameprefix ) : ]
2011-09-06 23:30:28 +08:00
if fname == name :
return t
return None
def runAdb ( self , * args ) :
2011-12-19 15:43:07 +08:00
cmd = self . adb [ : ]
2011-09-06 23:30:28 +08:00
cmd . extend ( args )
try :
output = Popen ( cmd , stdout = PIPE , stderr = PIPE ) . communicate ( )
if not output [ 1 ] :
return output [ 0 ]
self . error = output [ 1 ]
except OSError :
pass
return None
def isRunnable ( self ) :
2011-12-19 15:43:07 +08:00
if self . error :
return False
2011-09-06 23:30:28 +08:00
if self . targetarch == " x64 " and hostmachine == " x86 " :
self . error = " Target architecture is incompatible with current platform (at %s ) " % self . path
return False
if self . targetos == " android " :
2011-09-15 14:31:59 +08:00
if not self . adb :
self . error = " Could not find adb executable (for %s ) " % self . path
2011-09-06 23:30:28 +08:00
return False
2011-11-16 22:32:10 +08:00
if " armeabi-v7a " in self . android_abi :
2011-09-06 23:30:28 +08:00
adb_res = self . runAdb ( " shell " , " cat /proc/cpuinfo " )
if not adb_res :
2011-09-15 14:31:59 +08:00
self . error = " Could not get info about Android platform: %s (for %s ) " % ( self . error , self . path )
2011-09-06 23:30:28 +08:00
return False
if " ARMv7 " not in adb_res :
2011-09-15 14:31:59 +08:00
self . error = " Android device does not support ARMv7 commands, but tests are built for armeabi-v7a (for %s ) " % self . path
2011-09-06 23:30:28 +08:00
return False
2011-11-16 22:32:10 +08:00
if " NEON " in self . android_abi and " neon " not in adb_res :
self . error = " Android device has no NEON, but tests are built for %s (for %s ) " % ( self . android_abi , self . path )
2011-09-06 23:30:28 +08:00
return False
hw = re . search ( r " ^Hardware[ \ t]*:[ \ t]*(.*?)$ " , adb_res , re . MULTILINE )
if hw :
self . hardware = hw . groups ( ) [ 0 ] . strip ( )
return True
def runTest ( self , path , workingDir , _stdout , _stderr , args = [ ] ) :
if self . error :
return
args = args [ : ]
timestamp = datetime . datetime . now ( )
logfile = self . getLogName ( path , timestamp )
exe = os . path . abspath ( path )
userlog = [ a for a in args if a . startswith ( " --gtest_output= " ) ]
if len ( userlog ) == 0 :
args . append ( " --gtest_output=xml: " + logfile )
else :
2011-11-30 19:30:45 +08:00
logfile = userlog [ 0 ] [ userlog [ 0 ] . find ( " : " ) + 1 : ]
2011-09-06 23:30:28 +08:00
if self . targetos == " android " :
2012-03-15 16:31:57 +08:00
hostlogpath = " "
2011-09-06 23:30:28 +08:00
try :
2012-04-05 20:53:25 +08:00
andoidcwd = " /data/bin/ " + getpass . getuser ( ) . replace ( " " , " " ) + " _ " + self . options . mode + " / "
2011-09-06 23:30:28 +08:00
exename = os . path . basename ( exe )
androidexe = andoidcwd + exename
#upload
print >> _stderr , " Uploading " , exename , " to device... "
2011-12-19 15:43:07 +08:00
output = Popen ( self . adb + [ " push " , exe , androidexe ] , stdout = _stdout , stderr = _stderr ) . wait ( )
2011-09-06 23:30:28 +08:00
if output != 0 :
print >> _stderr , " adb finishes unexpectedly with error code " , output
return
#chmod
print >> _stderr , " Changing mode of " , androidexe
2011-12-19 15:43:07 +08:00
output = Popen ( self . adb + [ " shell " , " chmod 777 " + androidexe ] , stdout = _stdout , stderr = _stderr ) . wait ( )
2011-09-06 23:30:28 +08:00
if output != 0 :
print >> _stderr , " adb finishes unexpectedly with error code " , output
return
#run
2012-03-15 16:31:57 +08:00
if self . options . help :
command = exename + " --help "
else :
command = exename + " " + " " . join ( args )
2011-09-06 23:30:28 +08:00
print >> _stderr , " Running: " , command
2011-12-19 15:43:07 +08:00
Popen ( self . adb + [ " shell " , " export OPENCV_TEST_DATA_PATH= " + self . test_data_path + " && cd " + andoidcwd + " && ./ " + command ] , stdout = _stdout , stderr = _stderr ) . wait ( )
2011-09-06 23:30:28 +08:00
# try get log
2012-03-15 16:31:57 +08:00
if not self . options . help :
print >> _stderr , " Pulling " , logfile , " from device... "
hostlogpath = os . path . join ( workingDir , logfile )
output = Popen ( self . adb + [ " pull " , andoidcwd + logfile , hostlogpath ] , stdout = _stdout , stderr = _stderr ) . wait ( )
if output != 0 :
print >> _stderr , " adb finishes unexpectedly with error code " , output
return
#rm log
Popen ( self . adb + [ " shell " , " rm " + andoidcwd + logfile ] , stdout = _stdout , stderr = _stderr ) . wait ( )
2011-09-06 23:30:28 +08:00
except OSError :
pass
if os . path . isfile ( hostlogpath ) :
return hostlogpath
return None
else :
cmd = [ exe ]
2012-03-15 16:31:57 +08:00
if self . options . help :
cmd . append ( " --help " )
else :
cmd . extend ( args )
2011-09-06 23:30:28 +08:00
print >> _stderr , " Running: " , " " . join ( cmd )
try :
Popen ( cmd , stdout = _stdout , stderr = _stderr , cwd = workingDir ) . wait ( )
except OSError :
pass
logpath = os . path . join ( workingDir , logfile )
if os . path . isfile ( logpath ) :
return logpath
return None
def runTests ( self , tests , _stdout , _stderr , workingDir , args = [ ] ) :
if self . error :
return [ ]
if not tests :
tests = self . tests
logs = [ ]
for test in tests :
t = self . getTest ( test )
if t :
logfile = self . runTest ( t , workingDir , _stdout , _stderr , args )
if logfile :
logs . append ( os . path . relpath ( logfile , " . " ) )
else :
print >> _stderr , " Error: Test \" %s \" is not found in %s " % ( test , self . tests_dir )
return logs
if __name__ == " __main__ " :
test_args = [ a for a in sys . argv if a . startswith ( " --perf_ " ) or a . startswith ( " --gtest_ " ) ]
argv = [ a for a in sys . argv if not ( a . startswith ( " --perf_ " ) or a . startswith ( " --gtest_ " ) ) ]
parser = OptionParser ( )
parser . add_option ( " -t " , " --tests " , dest = " tests " , help = " comma-separated list of modules to test " , metavar = " SUITS " , default = " " )
parser . add_option ( " -w " , " --cwd " , dest = " cwd " , help = " working directory for tests " , metavar = " PATH " , default = " . " )
2012-04-05 20:53:25 +08:00
parser . add_option ( " -a " , " --accuracy " , dest = " accuracy " , help = " look for accuracy tests instead of performance tests " , action = " store_true " , default = False )
2012-01-24 00:28:46 +08:00
parser . add_option ( " -l " , " --longname " , dest = " useLongNames " , action = " store_true " , help = " generate log files with long names " , default = False )
2011-09-06 23:30:28 +08:00
parser . add_option ( " " , " --android_test_data_path " , dest = " test_data_path " , help = " OPENCV_TEST_DATA_PATH for Android run " , metavar = " PATH " , default = " /sdcard/opencv_testdata/ " )
2011-09-08 14:27:46 +08:00
parser . add_option ( " " , " --configuration " , dest = " configuration " , help = " force Debug or Release donfiguration " , metavar = " CFG " , default = " " )
2011-12-19 15:43:07 +08:00
parser . add_option ( " " , " --serial " , dest = " adb_serial " , help = " Android: directs command to the USB device or emulator with the given serial number " , metavar = " serial number " , default = " " )
2012-03-15 16:31:57 +08:00
parser . add_option ( " " , " --help-tests " , dest = " help " , help = " Show help for test executable " , action = " store_true " , default = False )
2011-09-06 23:30:28 +08:00
( options , args ) = parser . parse_args ( argv )
2012-04-05 20:53:25 +08:00
if options . accuracy :
options . mode = " test "
else :
options . mode = " perf "
2011-09-06 23:30:28 +08:00
run_args = [ ]
for path in args :
path = os . path . abspath ( path )
while ( True ) :
if os . path . isdir ( path ) and os . path . isfile ( os . path . join ( path , " CMakeCache.txt " ) ) :
run_args . append ( path )
break
npath = os . path . dirname ( path )
if npath == path :
break
path = npath
if len ( run_args ) == 0 :
print >> sys . stderr , " Usage: \n " , os . path . basename ( sys . argv [ 0 ] ) , " <build_path> "
exit ( 1 )
tests = [ s . strip ( ) for s in options . tests . split ( " , " ) if s ]
if len ( tests ) != 1 or len ( run_args ) != 1 :
#remove --gtest_output from params
test_args = [ a for a in test_args if not a . startswith ( " --gtest_output= " ) ]
logs = [ ]
for path in run_args :
2011-12-19 15:43:07 +08:00
info = RunInfo ( path , options )
2011-09-06 23:30:28 +08:00
#print vars(info),"\n"
if not info . isRunnable ( ) :
print >> sys . stderr , " Error: " , info . error
else :
info . test_data_path = options . test_data_path
logs . extend ( info . runTests ( tests , sys . stdout , sys . stderr , options . cwd , test_args ) )
if logs :
print >> sys . stderr , " Collected: " , " " . join ( logs )