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
15import os
16import re
17import subprocess
18import sys
19import time
20
21import its.cv2image
22import numpy as np
23
24
25def main():
26    """Load charts on device and display."""
27    scene = None
28    for s in sys.argv[1:]:
29        if s[:6] == 'scene=' and len(s) > 6:
30            scene = s[6:]
31        elif s[:7] == 'screen=' and len(s) > 7:
32            screen_id = s[7:]
33        elif s[:5] == 'dist=' and len(s) > 5:
34            chart_distance = float(re.sub('cm', '', s[5:]))
35        elif s[:4] == 'fov=' and len(s) > 4:
36            camera_fov = float(s[4:])
37
38    cmd = ('adb -s %s shell am force-stop com.google.android.apps.docs' %
39           screen_id)
40    subprocess.Popen(cmd.split())
41
42    if not scene:
43        print 'Error: need to specify which scene to load'
44        assert False
45
46    if not screen_id:
47        print 'Error: need to specify screen serial'
48        assert False
49
50    src_scene_path = os.path.join(os.environ['CAMERA_ITS_TOP'], 'tests', scene)
51    dst_scene_file = '/sdcard/Download/%s.pdf' % scene
52    chart_scaling = its.cv2image.calc_chart_scaling(chart_distance, camera_fov)
53    if np.isclose(chart_scaling, its.cv2image.SCALE_TELE_IN_WFOV_BOX, atol=0.01):
54        file_name = '%s_%s_scaled.pdf' % (
55                scene, str(its.cv2image.SCALE_TELE_IN_WFOV_BOX))
56    elif np.isclose(chart_scaling, its.cv2image.SCALE_RFOV_IN_WFOV_BOX, atol=0.01):
57        file_name = '%s_%s_scaled.pdf' % (
58                scene, str(its.cv2image.SCALE_RFOV_IN_WFOV_BOX))
59    else:
60        file_name = '%s.pdf' % scene
61    src_scene_file = os.path.join(src_scene_path, file_name)
62    print 'Loading %s on %s' % (src_scene_file, screen_id)
63    cmd = 'adb -s %s push %s /mnt%s' % (screen_id, src_scene_file,
64                                        dst_scene_file)
65    subprocess.Popen(cmd.split())
66    time.sleep(1)  # wait-for-device doesn't always seem to work...
67    # The intent require PDF viewing app be installed on device.
68    # Also the first time such app is opened it might request some permission,
69    # so it's  better to grant those permissions before using this script
70    cmd = ("adb -s %s wait-for-device shell am start -d 'file://%s'"
71           " -a android.intent.action.VIEW" % (screen_id, dst_scene_file))
72    subprocess.Popen(cmd.split())
73
74if __name__ == '__main__':
75    main()
76