1cb93a386Sopenharmony_ci# Copyright 2017 Google Inc.
2cb93a386Sopenharmony_ci#
3cb93a386Sopenharmony_ci# Use of this source code is governed by a BSD-style license that can be
4cb93a386Sopenharmony_ci# found in the LICENSE file.
5cb93a386Sopenharmony_ci
6cb93a386Sopenharmony_cifrom _hardware import Expectation
7cb93a386Sopenharmony_cifrom _hardware_android import HardwareAndroid
8cb93a386Sopenharmony_ci
9cb93a386Sopenharmony_ciCPU_CLOCK_RATE = 1670400
10cb93a386Sopenharmony_ciGPU_CLOCK_RATE = 315000000
11cb93a386Sopenharmony_ci
12cb93a386Sopenharmony_ciclass HardwarePixel(HardwareAndroid):
13cb93a386Sopenharmony_ci  def __init__(self, adb):
14cb93a386Sopenharmony_ci    HardwareAndroid.__init__(self, adb)
15cb93a386Sopenharmony_ci
16cb93a386Sopenharmony_ci  def __enter__(self):
17cb93a386Sopenharmony_ci    HardwareAndroid.__enter__(self)
18cb93a386Sopenharmony_ci    if not self._adb.is_root():
19cb93a386Sopenharmony_ci      return self
20cb93a386Sopenharmony_ci
21cb93a386Sopenharmony_ci    self._adb.shell('\n'.join([
22cb93a386Sopenharmony_ci      # enable and lock the two fast cores.
23cb93a386Sopenharmony_ci      '''
24cb93a386Sopenharmony_ci      stop thermal-engine
25cb93a386Sopenharmony_ci      stop perfd
26cb93a386Sopenharmony_ci
27cb93a386Sopenharmony_ci      for N in 3 2; do
28cb93a386Sopenharmony_ci        echo 1 > /sys/devices/system/cpu/cpu$N/online
29cb93a386Sopenharmony_ci        echo userspace > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_governor
30cb93a386Sopenharmony_ci        echo %i > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_max_freq
31cb93a386Sopenharmony_ci        echo %i > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_min_freq
32cb93a386Sopenharmony_ci        echo %i > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_setspeed
33cb93a386Sopenharmony_ci      done''' % tuple(CPU_CLOCK_RATE for _ in range(3)),
34cb93a386Sopenharmony_ci
35cb93a386Sopenharmony_ci      # turn off the two slow cores
36cb93a386Sopenharmony_ci      '''
37cb93a386Sopenharmony_ci      for N in 1 0; do
38cb93a386Sopenharmony_ci        echo 0 > /sys/devices/system/cpu/cpu$N/online
39cb93a386Sopenharmony_ci      done''',
40cb93a386Sopenharmony_ci
41cb93a386Sopenharmony_ci      # pylint: disable=line-too-long
42cb93a386Sopenharmony_ci
43cb93a386Sopenharmony_ci      # Set GPU bus and idle timer
44cb93a386Sopenharmony_ci      # Set DDR frequency to max
45cb93a386Sopenharmony_ci      # Set GPU to performance mode, 315 MHZ
46cb93a386Sopenharmony_ci      # See https://android.googlesource.com/platform/frameworks/base/+/master/libs/hwui/tests/scripts/prep_marlfish.sh
47cb93a386Sopenharmony_ci      '''
48cb93a386Sopenharmony_ci      echo 0 > /sys/class/kgsl/kgsl-3d0/bus_split
49cb93a386Sopenharmony_ci      echo 1 > /sys/class/kgsl/kgsl-3d0/force_clk_on
50cb93a386Sopenharmony_ci      echo 10000 > /sys/class/kgsl/kgsl-3d0/idle_timer
51cb93a386Sopenharmony_ci
52cb93a386Sopenharmony_ci      echo 13763 > /sys/class/devfreq/soc:qcom,gpubw/min_freq
53cb93a386Sopenharmony_ci
54cb93a386Sopenharmony_ci      echo performance > /sys/class/kgsl/kgsl-3d0/devfreq/governor
55cb93a386Sopenharmony_ci      echo %i > /sys/class/kgsl/kgsl-3d0/devfreq/max_freq
56cb93a386Sopenharmony_ci      echo %i > /sys/class/kgsl/kgsl-3d0/devfreq/min_freq
57cb93a386Sopenharmony_ci
58cb93a386Sopenharmony_ci      echo 4 > /sys/class/kgsl/kgsl-3d0/max_pwrlevel
59cb93a386Sopenharmony_ci      echo 4 > /sys/class/kgsl/kgsl-3d0/min_pwrlevel''' %
60cb93a386Sopenharmony_ci      tuple(GPU_CLOCK_RATE for _ in range(2))]))
61cb93a386Sopenharmony_ci
62cb93a386Sopenharmony_ci    return self
63cb93a386Sopenharmony_ci
64cb93a386Sopenharmony_ci  def sanity_check(self):
65cb93a386Sopenharmony_ci    HardwareAndroid.sanity_check(self)
66cb93a386Sopenharmony_ci
67cb93a386Sopenharmony_ci    if not self._adb.is_root():
68cb93a386Sopenharmony_ci      return
69cb93a386Sopenharmony_ci
70cb93a386Sopenharmony_ci    result = self._adb.check(' '.join(
71cb93a386Sopenharmony_ci      ['cat',
72cb93a386Sopenharmony_ci       '/sys/class/power_supply/battery/capacity',
73cb93a386Sopenharmony_ci       '/sys/devices/system/cpu/online'] + \
74cb93a386Sopenharmony_ci      ['/sys/devices/system/cpu/cpu%i/cpufreq/scaling_cur_freq' % i
75cb93a386Sopenharmony_ci       for i in range(2, 4)] + \
76cb93a386Sopenharmony_ci      ['/sys/kernel/debug/clk/bimc_clk/measure',
77cb93a386Sopenharmony_ci       '/sys/class/thermal/thermal_zone22/temp',
78cb93a386Sopenharmony_ci       '/sys/class/thermal/thermal_zone23/temp']))
79cb93a386Sopenharmony_ci
80cb93a386Sopenharmony_ci    expectations = \
81cb93a386Sopenharmony_ci      [Expectation(int, min_value=30, name='battery', sleeptime=30*60),
82cb93a386Sopenharmony_ci       Expectation(str, exact_value='2-3', name='online cpus')] + \
83cb93a386Sopenharmony_ci      [Expectation(int, exact_value=CPU_CLOCK_RATE, name='cpu_%i clock rate' %i)
84cb93a386Sopenharmony_ci       for i in range(2, 4)] + \
85cb93a386Sopenharmony_ci       [Expectation(long, min_value=902390000, max_value=902409999,
86cb93a386Sopenharmony_ci                   name='measured ddr clock', sleeptime=10),
87cb93a386Sopenharmony_ci       Expectation(int, max_value=41000, name='pm8994_tz temperature'),
88cb93a386Sopenharmony_ci       Expectation(int, max_value=40, name='msm_therm temperature')]
89cb93a386Sopenharmony_ci
90cb93a386Sopenharmony_ci    Expectation.check_all(expectations, result.splitlines())
91