1# Copyright 2016 Google Inc.
2#
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6from _hardware import HardwareException, Expectation
7from _hardware_android import HardwareAndroid
8
9CPU_CLOCK_RATE = 1728000
10GPU_CLOCK_RATE = 510000000
11
12class HardwareNexus6P(HardwareAndroid):
13  def __init__(self, adb):
14    HardwareAndroid.__init__(self, adb)
15
16  def __enter__(self):
17    self._lock_clocks()
18    return HardwareAndroid.__enter__(self)
19
20  def __exit__(self, exception_type, exception_value, exception_traceback):
21    HardwareAndroid.__exit__(self, exception_type,
22                             exception_value, exception_traceback)
23    self._unlock_clocks()
24
25  def _lock_clocks(self):
26    if not self._adb.is_root():
27      return
28
29    self._adb.shell('''\
30      stop thermal-engine
31      stop thermald
32      stop perfd
33      stop mpdecision''')
34
35    # enable and lock 3 of 4 big cores.
36    self._adb.shell('''\
37      for N in 4 5 6; do
38        echo 1 > /sys/devices/system/cpu/cpu$N/online
39        echo userspace > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_governor
40        echo %i > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_max_freq
41        echo %i > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_min_freq
42        echo %i > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_setspeed
43      done''' % tuple(CPU_CLOCK_RATE for _ in range(3)))
44
45    # turn off all other cores
46    self._adb.shell('''\
47      for N in 0 1 2 3 7; do
48        echo 0 > /sys/devices/system/cpu/cpu$N/online
49      done''')
50
51    # gpu/ddr perf commands from
52    # https://developer.qualcomm.com/qfile/28823/lm80-p0436-11_adb_commands.pdf
53    self._adb.shell('''\
54      echo 0 > /sys/class/kgsl/kgsl-3d0/bus_split
55      echo 1 > /sys/class/kgsl/kgsl-3d0/force_bus_on
56      echo 1 > /sys/class/kgsl/kgsl-3d0/force_rail_on
57      echo 1 > /sys/class/kgsl/kgsl-3d0/force_clk_on
58      echo 1000000 > /sys/class/kgsl/kgsl-3d0/idle_timer
59      echo performance > /sys/class/kgsl/kgsl-3d0/devfreq/governor
60      echo %i > /sys/class/kgsl/kgsl-3d0/devfreq/max_freq
61      echo %i > /sys/class/kgsl/kgsl-3d0/devfreq/min_freq
62      echo %i > /sys/class/kgsl/kgsl-3d0/gpuclk''' %
63      tuple(GPU_CLOCK_RATE for _ in range(3)))
64
65    # ddr perf commands from
66    # https://developer.qualcomm.com/qfile/28823/lm80-p0436-11_adb_commands.pdf
67    self._adb.shell('''\
68      echo performance > /sys/class/devfreq/qcom,cpubw.32/governor
69      echo 9887 > /sys/class/devfreq/qcom,cpubw.32/max_freq
70      echo 9887 > /sys/class/devfreq/qcom,cpubw.32/min_freq
71      echo performance > /sys/class/devfreq/qcom,gpubw.70/governor
72      echo 9887 > /sys/class/devfreq/qcom,gpubw.70/max_freq
73      echo 9887 > /sys/class/devfreq/qcom,gpubw.70/min_freq''')
74
75  def _unlock_clocks(self):
76    if not self._adb.is_root():
77      return
78
79    # restore ddr settings to default.
80    self._adb.shell('''\
81      echo 1525 > /sys/class/devfreq/qcom,cpubw.32/min_freq
82      echo 9887 > /sys/class/devfreq/qcom,cpubw.32/max_freq
83      echo bw_hwmon > /sys/class/devfreq/qcom,cpubw.32/governor
84      echo 1525 > /sys/class/devfreq/qcom,gpubw.70/min_freq
85      echo 9887 > /sys/class/devfreq/qcom,gpubw.70/max_freq
86      echo bw_hwmon > /sys/class/devfreq/qcom,gpubw.70/governor''')
87
88    # restore gpu settings to default.
89    self._adb.shell('''\
90      echo 180000000 > /sys/class/kgsl/kgsl-3d0/devfreq/min_freq
91      echo 600000000 > /sys/class/kgsl/kgsl-3d0/devfreq/max_freq
92      echo 180000000 > /sys/class/kgsl/kgsl-3d0/gpuclk
93      echo msm-adreno-tz > /sys/class/kgsl/kgsl-3d0/devfreq/governor
94      echo 0 > /sys/class/kgsl/kgsl-3d0/idle_timer
95      echo 0 > /sys/class/kgsl/kgsl-3d0/force_clk_on
96      echo 0 > /sys/class/kgsl/kgsl-3d0/force_rail_on
97      echo 0 > /sys/class/kgsl/kgsl-3d0/force_bus_on
98      echo 1 > /sys/class/kgsl/kgsl-3d0/bus_split''')
99
100    # turn the disabled cores back on.
101    self._adb.shell('''\
102      for N in 7 3 2 1 0; do
103        echo 1 > /sys/devices/system/cpu/cpu$N/online
104      done''')
105
106    # unlock the 3 enabled big cores.
107    self._adb.shell('''\
108      for N in 6 5 4; do
109        echo 633600 > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_min_freq
110        echo 1958400 > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_max_freq
111        echo 0 > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_setspeed
112        echo interactive > /sys/devices/system/cpu/cpu$N/cpufreq/scaling_governor
113      done''')
114
115    self._adb.shell('''\
116      start mpdecision
117      start perfd
118      start thermald
119      start thermal-engine''')
120
121  def sanity_check(self):
122    HardwareAndroid.sanity_check(self)
123
124    if not self._adb.is_root():
125      return
126
127    result = self._adb.check('''\
128      cat /sys/class/power_supply/battery/capacity \
129          /sys/devices/system/cpu/online \
130          /sys/class/thermal/thermal_zone14/temp \
131          /sys/class/thermal/thermal_zone15/temp \
132          /sys/kernel/debug/clk/oxili_gfx3d_clk/measure \
133          /sys/kernel/debug/clk/bimc_clk/measure
134      for N in 4 5 6; do
135        cat /sys/devices/system/cpu/cpu$N/cpufreq/scaling_cur_freq
136      done''')
137
138    expectations = \
139      [Expectation(int, min_value=30, name='battery', sleeptime=30*60),
140       Expectation(str, exact_value='4-6', name='online cpus'),
141       Expectation(int, max_value=88, name='tsens_tz_sensor13'),
142       Expectation(int, max_value=88, name='tsens_tz_sensor14'),
143       Expectation(long, min_value=(GPU_CLOCK_RATE - 5000),
144                   max_value=(GPU_CLOCK_RATE + 5000), name='gpu clock rate'),
145       Expectation(long, min_value=647995000, max_value=648007500,
146                   name='ddr clock rate', sleeptime=10)] + \
147      [Expectation(int, exact_value=CPU_CLOCK_RATE, name='cpu_%i clock rate' %i)
148       for i in range(4, 7)]
149
150    Expectation.check_all(expectations, result.splitlines())
151
152  def sleep(self, sleeptime):
153    self._unlock_clocks()
154    HardwareAndroid.sleep(self, sleeptime)
155    self._lock_clocks()
156