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
23import sys
24import os
25import time
26import string
27import argparse
28
29import common
30
31def install (extraArgs = [], printPrefix=""):
32	print printPrefix + "Removing old dEQP Package...\n",
33	common.execArgsInDirectory([common.ADB_BIN] + extraArgs + [
34			'uninstall',
35			'com.drawelements.deqp'
36		], common.ANDROID_DIR, printPrefix)
37	print printPrefix + "Remove complete\n",
38
39	print printPrefix + "Installing dEQP Package...\n",
40	common.execArgsInDirectory([common.ADB_BIN] + extraArgs + [
41			'install',
42			'-r',
43			'package/bin/dEQP-debug.apk'
44		], common.ANDROID_DIR, printPrefix)
45	print printPrefix + "Install complete\n",
46
47def installToDevice (device, 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(['-s', device.serial], printPrefix)
54
55def installToDevices (devices, doParallel):
56	padLen = max([len(device.model) for device in devices])+1
57	if doParallel:
58		common.parallelApply(installToDevice, [(device, ("(%s):%s" % (device.model, ' ' * (padLen - len(device.model))))) for device in devices]);
59	else:
60		common.serialApply(installToDevice, [(device, ) for device in devices]);
61
62def installToAllDevices (doParallel):
63	devices = common.getDevices(common.ADB_BIN)
64	installToDevices(devices, doParallel)
65
66if __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
72	args = parser.parse_args()
73
74	if args.all:
75		installToAllDevices(args.doParallel)
76	else:
77		if args.serial == None:
78			devices = common.getDevices(common.ADB_BIN)
79			if len(devices) == 0:
80				common.die('No devices connected')
81			elif len(devices) == 1:
82				installToDevice(devices[0])
83			else:
84				print "More than one device connected:"
85				for i in range(0, len(devices)):
86					print "%3d: %16s %s" % ((i+1), devices[i].serial, devices[i].model)
87
88				deviceNdx = int(raw_input("Choose device (1-%d): " % len(devices)))
89				installToDevice(devices[deviceNdx-1])
90		else:
91			devices = common.getDevices(common.ADB_BIN)
92
93			devices = [dev for dev in devices if dev.serial in args.serial]
94			devSerials = [dev.serial for dev in devices]
95			notFounds = [serial for serial in args.serial if not serial in devSerials]
96
97			for notFound in notFounds:
98				print("Couldn't find device matching serial '%s'" % notFound)
99
100			installToDevices(devices, args.doParallel)
101