Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 1 | # Copyright 2016 The Android Open Source Project |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | |
| 15 | import os |
| 16 | import re |
| 17 | import subprocess |
| 18 | import sys |
| 19 | import time |
Leslie Shaw | d73c934 | 2020-09-01 00:56:38 +0000 | [diff] [blame] | 20 | |
Clemenz Portmann | d440895 | 2018-10-12 08:51:49 -0700 | [diff] [blame] | 21 | import its.cv2image |
Clemenz Portmann | 9b752ed | 2018-03-21 12:35:15 -0700 | [diff] [blame] | 22 | import numpy as np |
| 23 | |
Clemenz Portmann | 7a3f7eb | 2019-08-01 13:08:50 -0700 | [diff] [blame] | 24 | LOAD_SCENE_DELAY = 2 # seconds |
| 25 | |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 26 | |
| 27 | def main(): |
| 28 | """Load charts on device and display.""" |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 29 | scene = None |
| 30 | for s in sys.argv[1:]: |
| 31 | if s[:6] == 'scene=' and len(s) > 6: |
| 32 | scene = s[6:] |
| 33 | elif s[:7] == 'screen=' and len(s) > 7: |
| 34 | screen_id = s[7:] |
Clemenz Portmann | 8f07042 | 2017-11-27 16:06:33 -0800 | [diff] [blame] | 35 | elif s[:5] == 'dist=' and len(s) > 5: |
| 36 | chart_distance = float(re.sub('cm', '', s[5:])) |
| 37 | elif s[:4] == 'fov=' and len(s) > 4: |
| 38 | camera_fov = float(s[4:]) |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 39 | |
| 40 | cmd = ('adb -s %s shell am force-stop com.google.android.apps.docs' % |
| 41 | screen_id) |
| 42 | subprocess.Popen(cmd.split()) |
Leslie Shaw | d73c934 | 2020-09-01 00:56:38 +0000 | [diff] [blame] | 43 | |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 44 | if not scene: |
| 45 | print 'Error: need to specify which scene to load' |
| 46 | assert False |
| 47 | |
Clemenz Portmann | a9afdd2 | 2016-09-21 07:52:57 -0700 | [diff] [blame] | 48 | if not screen_id: |
| 49 | print 'Error: need to specify screen serial' |
| 50 | assert False |
| 51 | |
Clemenz Portmann | d440895 | 2018-10-12 08:51:49 -0700 | [diff] [blame] | 52 | src_scene_path = os.path.join(os.environ['CAMERA_ITS_TOP'], 'tests', scene) |
| 53 | dst_scene_file = '/sdcard/Download/%s.pdf' % scene |
| 54 | chart_scaling = its.cv2image.calc_chart_scaling(chart_distance, camera_fov) |
| 55 | if np.isclose(chart_scaling, its.cv2image.SCALE_TELE_IN_WFOV_BOX, atol=0.01): |
Clemenz Portmann | 31ea19b | 2019-04-10 13:55:20 -0700 | [diff] [blame] | 56 | file_name = '%s_%sx_scaled.pdf' % ( |
Clemenz Portmann | d440895 | 2018-10-12 08:51:49 -0700 | [diff] [blame] | 57 | scene, str(its.cv2image.SCALE_TELE_IN_WFOV_BOX)) |
| 58 | elif np.isclose(chart_scaling, its.cv2image.SCALE_RFOV_IN_WFOV_BOX, atol=0.01): |
Clemenz Portmann | 31ea19b | 2019-04-10 13:55:20 -0700 | [diff] [blame] | 59 | file_name = '%s_%sx_scaled.pdf' % ( |
Clemenz Portmann | b1c43b0 | 2018-11-20 13:49:05 -0800 | [diff] [blame] | 60 | scene, str(its.cv2image.SCALE_RFOV_IN_WFOV_BOX)) |
Clemenz Portmann | 8f07042 | 2017-11-27 16:06:33 -0800 | [diff] [blame] | 61 | else: |
Clemenz Portmann | d440895 | 2018-10-12 08:51:49 -0700 | [diff] [blame] | 62 | file_name = '%s.pdf' % scene |
| 63 | src_scene_file = os.path.join(src_scene_path, file_name) |
| 64 | print 'Loading %s on %s' % (src_scene_file, screen_id) |
| 65 | cmd = 'adb -s %s push %s /mnt%s' % (screen_id, src_scene_file, |
| 66 | dst_scene_file) |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 67 | subprocess.Popen(cmd.split()) |
Clemenz Portmann | 7a3f7eb | 2019-08-01 13:08:50 -0700 | [diff] [blame] | 68 | time.sleep(LOAD_SCENE_DELAY) # wait-for-device doesn't always seem to work |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 69 | # The intent require PDF viewing app be installed on device. |
| 70 | # Also the first time such app is opened it might request some permission, |
| 71 | # so it's better to grant those permissions before using this script |
| 72 | cmd = ("adb -s %s wait-for-device shell am start -d 'file://%s'" |
Clemenz Portmann | d440895 | 2018-10-12 08:51:49 -0700 | [diff] [blame] | 73 | " -a android.intent.action.VIEW" % (screen_id, dst_scene_file)) |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 74 | subprocess.Popen(cmd.split()) |
| 75 | |
Clemenz Portmann | f7f23ee | 2016-08-18 13:00:59 -0700 | [diff] [blame] | 76 | if __name__ == '__main__': |
| 77 | main() |