commit a7424bf2addc12b4186eb3a462692ea73b4b9fbc
author Moja Hsu <mojahsu@google.com> Thu Nov 25 10:24:46 2021 +0800
committer Commit Bot <commit-bot@chromium.org> Fri Nov 26 01:36:11 2021 +0000
tree bb888ed59caed27574251b297bdcc8f99e20cf53
parent 972c2ed97d80484e9d78032ba1ab279bdfa60b1b

autotest: Add --display_level argument

Add --display_level argument to control the display level when display
chart.

BUG=b:203608939
TEST=python2 display_chart.py <Chart file> --display_level=50

Change-Id: I268319057befe14161942d401e6e1bb5dc60aadd
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/third_party/autotest/+/3301205
Reviewed-by: Inker Kuo <inker@chromium.org>
Tested-by: Hsu Wei-Cheng <mojahsu@chromium.org>
Commit-Queue: Hsu Wei-Cheng <mojahsu@chromium.org>

client/bin/display_chart.py

diff --git a/client/bin/display_chart.py b/client/bin/display_chart.py
index fc7818f..6b2f191 100755
--- a/client/bin/display_chart.py
+++ b/client/bin/display_chart.py
@@ -26,6 +26,8 @@
 from autotest_lib.client.cros.multimedia import facade_resource
 from autotest_lib.client.common_lib.cros import chrome
 
+DEFAULT_DISPLAY_LEVEL = 96.0
+
 
 @contextlib.contextmanager
 def set_display_brightness(display_level):
@@ -44,9 +46,8 @@
     utils.system(SET_BRIGHTNESS_CMD % original_display_level)
 
 
-def display(filepath):
+def display(filepath, display_level):
     """Display chart with filepath on device by using telemetry."""
-    DISPLAY_LEVEL = 96.0
     DISPLAY_ORIENTATION = 90
 
     assert os.path.isfile(filepath), 'filepath %r not found.' % filepath
@@ -66,7 +67,7 @@
             extension_paths=[constants.DISPLAY_TEST_EXTENSION],
             autotest_ext=True,
             init_network_controller=True) as cr, set_display_brightness(
-                    DISPLAY_LEVEL):
+                    display_level):
         logging.info('Set fullscreen.')
         facade = facade_resource.FacadeResource(cr)
         display_facade = display_facade_native.DisplayFacadeNative(facade)
@@ -98,6 +99,12 @@
             description='Display chart file on chrome by using telemetry.'
             ' Send SIGINT or keyboard interrupt to stop displaying.')
     argparser.add_argument('filepath', help='Path of displayed chart file.')
+    argparser.add_argument(
+            '--display_level',
+            type=float,
+            default=DEFAULT_DISPLAY_LEVEL,
+            help=
+            'Set brightness as linearly-calculated percent in [0.0, 100.0].')
 
     args = argparser.parse_args()
-    display(args.filepath)
+    display(args.filepath, args.display_level)