Home | History | Annotate | Download | only in scripts
      1 # -*- coding: utf-8 -*-
      2 
      3 #-------------------------------------------------------------------------
      4 # drawElements Quality Program utilities
      5 # --------------------------------------
      6 #
      7 # Copyright 2015 The Android Open Source Project
      8 #
      9 # Licensed under the Apache License, Version 2.0 (the "License");
     10 # you may not use this file except in compliance with the License.
     11 # You may obtain a copy of the License at
     12 #
     13 #      http://www.apache.org/licenses/LICENSE-2.0
     14 #
     15 # Unless required by applicable law or agreed to in writing, software
     16 # distributed under the License is distributed on an "AS IS" BASIS,
     17 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     18 # See the License for the specific language governing permissions and
     19 # limitations under the License.
     20 #
     21 #-------------------------------------------------------------------------
     22 
     23 import sys
     24 import os
     25 import time
     26 import string
     27 import argparse
     28 
     29 import common
     30 
     31 def install (buildRoot, extraArgs = [], printPrefix=""):
     32 	print printPrefix + "Removing old dEQP Package...\n",
     33 	common.execArgsInDirectory([common.ADB_BIN] + extraArgs + [
     34 			'uninstall',
     35 			'com.drawelements.deqp'
     36 		], buildRoot, printPrefix, failOnNonZeroExit=False)
     37 	print printPrefix + "Remove complete\n",
     38 
     39 	print printPrefix + "Installing dEQP Package from %s...\n" %(buildRoot),
     40 	common.execArgsInDirectory([common.ADB_BIN] + extraArgs + [
     41 			'install',
     42 			'-r',
     43 			'package/bin/dEQP-debug.apk'
     44 		], buildRoot, printPrefix)
     45 	print printPrefix + "Install complete\n",
     46 
     47 def installToDevice (device, buildRoot, printPrefix=""):
     48 	if len(printPrefix) == 0:
     49 		print "Installing to %s (%s)...\n" % (device.serial, device.model),
     50 	else:
     51 		print printPrefix + "Installing to %s\n" % device.serial,
     52 
     53 	install(buildRoot, ['-s', device.serial], printPrefix)
     54 
     55 def installToDevices (devices, doParallel, buildRoot):
     56 	padLen = max([len(device.model) for device in devices])+1
     57 	if doParallel:
     58 		common.parallelApply(installToDevice, [(device, buildRoot, ("(%s):%s" % (device.model, ' ' * (padLen - len(device.model))))) for device in devices]);
     59 	else:
     60 		common.serialApply(installToDevice, [(device, buildRoot) for device in devices]);
     61 
     62 def installToAllDevices (doParallel, buildRoot):
     63 	devices = common.getDevices(common.ADB_BIN)
     64 	installToDevices(devices, doParallel, buildRoot)
     65 
     66 if __name__ == "__main__":
     67 	parser = argparse.ArgumentParser()
     68 	parser.add_argument('-p', '--parallel', dest='doParallel', action="store_true", help="Install package in parallel.")
     69 	parser.add_argument('-s', '--serial', dest='serial', type=str, nargs='+', help="Install package to device with serial number.")
     70 	parser.add_argument('-a', '--all', dest='all', action="store_true", help="Install to all devices.")
     71 	parser.add_argument('-b', '--build-root', dest='buildRoot', default=common.ANDROID_DIR, help="Root directory from which to pick up APK. Generally, build root specified in build.py")
     72 
     73 	args = parser.parse_args()
     74 	absBuildRoot = os.path.abspath(args.buildRoot)
     75 
     76 	if args.all:
     77 		installToAllDevices(args.doParallel, absBuildRoot)
     78 	else:
     79 		if args.serial == None:
     80 			devices = common.getDevices(common.ADB_BIN)
     81 			if len(devices) == 0:
     82 				common.die('No devices connected')
     83 			elif len(devices) == 1:
     84 				installToDevice(devices[0], absBuildRoot)
     85 			else:
     86 				print "More than one device connected:"
     87 				for i in range(0, len(devices)):
     88 					print "%3d: %16s %s" % ((i+1), devices[i].serial, devices[i].model)
     89 
     90 				deviceNdx = int(raw_input("Choose device (1-%d): " % len(devices)))
     91 				installToDevice(devices[deviceNdx-1], absBuildRoot)
     92 		else:
     93 			devices = common.getDevices(common.ADB_BIN)
     94 
     95 			devices = [dev for dev in devices if dev.serial in args.serial]
     96 			devSerials = [dev.serial for dev in devices]
     97 			notFounds = [serial for serial in args.serial if not serial in devSerials]
     98 
     99 			for notFound in notFounds:
    100 				print("Couldn't find device matching serial '%s'" % notFound)
    101 
    102 			installToDevices(devices, args.doParallel, absBuildRoot)
    103