123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343 |
- #!/usr/bin/python3
- import argparse
- import glob
- import os
- import subprocess
- import sys
- from typing import Any, NoReturn
- # The gearings below were picked based on various tests by the ClockworkPi devs.
- # The maximum-performance maximum-power gearing is present for completeness, but
- # shouldn't be needed for most uses.
- #
- # You can customise the gearings by editing the list below. The valid freqencies
- # for CPU <N> can be looked up here (substituting for <N>):
- # /sys/devices/system/cpu/cpu<N>/cpufreq/scaling_available_frequencies
- #
- # The valid GPU frequencies can be looked up here:
- # /sys/devices/platform/ff9a0000.gpu/devfreq/ff9a0000.gpu/available_frequencies
- #
- # Gears are numbered in-order, starting from 1.
- # It's up to you to ensure that they are sorted by performance :)
- GPU_GOV_SIMPLE = "simple_ondemand"
- GPU_GOV_PERF = "performance"
- def gears() -> list[dict[str, Any]]:
- return [
- gear(little=(600000,), use="simple writing tasks with long battery life"),
- gear(little=(800000,) * 2, use="browsing most websites with long battery life"),
- gear(little=(1008000,) * 4, gpu_freq=400000000, use="most 2D games and emulators"),
- gear(big=(1008000,) * 2, gpu_freq=400000000, use="playing videos and 3D games"),
- gear(big=(1200000,) * 2, gpu_freq=400000000, use="performance-first tasks"),
- gear(
- little=(1416000,) * 4,
- big=(1800000,) * 2,
- gpu_freq=800000000,
- use="max performance, max power (usage)",
- ),
- ]
- def gear(
- little=(0, 0, 0, 0),
- big=(0, 0),
- gpu_freq=200000000,
- gpu_gov=GPU_GOV_SIMPLE,
- use="",
- ) -> dict[str, Any]:
- """Helper to convert the concise gear format above into a full description.
- `little` and `big` define the number of A53 and A72 CPU cores to enable, and
- their maximum frequencies (in kHz). Cores that are omitted or set to zero are
- :param little: A53 Core.
- :param big: A72 Core
- :param gpu_freq: Gpu frequency in kHz
- :param gpu_gov: Gpu Governor
- :param use: Description of the gear.
- :return: A dictionary of the parameters.
- """
- # Extend to 4 little and 2 big cores (matching the A06).
- assert len(little) <= 4
- assert len(big) <= 2
- cpu = little + (0,) * (4 - len(little)) + big + (0,) * (2 - len(big))
- # At least one CPU must be enabled
- assert sum(cpu) > 0
- return {
- "cpu": cpu,
- "gpu_freq": gpu_freq,
- "gpu_gov": gpu_gov,
- "use": use,
- }
- # We placed gears() at the top of the file to make it easier to find and edit.
- # Now that we've defined the helpers it needs, evaluate the gears.
- gears = gears()
- def load_gear(gear: int) -> dict[str, Any]:
- return gears[gear - 1]
- cur_stat = [
- "+-----------------------------------+-----------------+-----------+",
- "| Cortex-A53 | Cortex-A72 | Mali-T860 |",
- "+--------+--------+--------+--------+--------+--------+-----------+",
- "| CPU 0 | CPU 1 | CPU 2 | CPU 3 | CPU 4 | CPU 5 | GPU |",
- "+--------+--------+--------+--------+--------+--------+-----------+",
- "| 600 MHz| OFF | OFF | OFF | OFF | OFF | 400 MHz |",
- "+--------+--------+--------+--------+--------+--------+-----------+",
- ]
- class A06:
- """A06 Module class."""
- cpus = []
- cpu_scaling_governor: str = "schedutil"
- gear = load_gear(1) # 1-5
- null_out: str = "2>/dev/null"
- def __init__(self):
- self.cpus = []
- self.init_cpu_infos()
- self.cpu_total_count = len(self.cpus)
- def init_cpu_infos(self):
- self.cpus = glob.glob("/sys/devices/system/cpu/cpu[0-9]")
- self.cpus.sort()
- @property
- def cpu_gov(self) -> str:
- if self.gear["cpu"][0] > 0:
- cpu_gov_path = "/sys/devices/system/cpu/cpufreq/policy0/scaling_governor"
- else:
- cpu_gov_path = "/sys/devices/system/cpu/cpufreq/policy4/scaling_governor"
- with open(cpu_gov_path, "r") as f:
- return f.read().strip()
- @property
- def gpu_gov(self) -> str:
- with open(
- "/sys/devices/platform/ff9a0000.gpu/devfreq/ff9a0000.gpu/governor", "r"
- ) as gov_file:
- return gov_file.read().strip()
- def set_cpu_gov0(self, gov):
- cpu_gov_path = "/sys/devices/system/cpu/cpufreq/policy0/scaling_governor"
- try:
- subprocess.run(
- "echo %s | sudo tee %s " % (gov, cpu_gov_path),
- shell=True,
- stdout=subprocess.DEVNULL,
- )
- except Exception:
- print("set cpu governor failed")
- def set_cpu_gov4(self, gov):
- cpu_gov_path = "/sys/devices/system/cpu/cpufreq/policy4/scaling_governor"
- try:
- subprocess.run(
- "echo %s | sudo tee %s" % (gov, cpu_gov_path),
- shell=True,
- stdout=subprocess.DEVNULL,
- )
- except Exception:
- print("set cpu governor failed")
- def get_cpu_on_off(self, cpu_num):
- cpu_onoff_file = "/sys/devices/system/cpu/cpu%d/online" % cpu_num
- with open(cpu_onoff_file, "r") as f:
- onoff = f.read().strip()
- if onoff == "1":
- cpu_max_freq_file = "/sys/devices/system/cpu/cpu%d/cpufreq/scaling_max_freq" % cpu_num
- with open(cpu_max_freq_file, "r") as f:
- max_freq = f.read().strip()
- mhz = int(max_freq) // 1000
- return f"{mhz} MHz"
- return "OFF"
- def set_cpu_on_off(self, cpu_num, onoff):
- cpu_onoff_file = "/sys/devices/system/cpu/cpu%d/online" % cpu_num
- try:
- subprocess.run(
- "echo %d | sudo tee %s" % (onoff, cpu_onoff_file),
- shell=True,
- stdout=subprocess.DEVNULL,
- )
- except Exception:
- print("set cpu %d on off failed" % cpu_num)
- def set_cpu_max_freq(self, cpu_num, max_freq):
- cpu_max_freq_file = "/sys/devices/system/cpu/cpu%d/cpufreq/scaling_max_freq" % cpu_num
- try:
- subprocess.run(
- "echo %d | sudo tee %s" % (max_freq, cpu_max_freq_file),
- shell=True,
- stdout=subprocess.DEVNULL,
- )
- except Exception:
- print("set cpu %d max freq failed" % cpu_num)
- def get_gpu_freq(self):
- gpu_sys_path = "/sys/devices/platform/ff9a0000.gpu/devfreq/ff9a0000.gpu"
- gpu_freq_path = os.path.join(gpu_sys_path, "max_freq")
- with open(gpu_freq_path, "r") as f:
- freq = f.read().strip()
- mhz = int(freq) // 1000000
- return f"{mhz} MHz"
- def set_gpu(self, gov, hz):
- gpu_sys_path = "/sys/devices/platform/ff9a0000.gpu/devfreq/ff9a0000.gpu"
- gpu_gov_path = os.path.join(gpu_sys_path, "governor")
- gpu_freq_path = os.path.join(gpu_sys_path, "max_freq")
- try:
- subprocess.run(
- "echo %s | sudo tee %s" % (gov, gpu_gov_path), shell=True, stdout=subprocess.DEVNULL
- )
- subprocess.run(
- "echo %d | sudo tee %s" % (hz, gpu_freq_path), shell=True, stdout=subprocess.DEVNULL
- )
- except Exception:
- print("set gpu failed")
- def print_cpu_gpu_gov(self):
- print(f"CPU Governor: {self.cpu_gov} GPU Governor: {self.gpu_gov}")
- def print_cur_status(self):
- global cur_stat
- stat_str = "|%s|%s|%s|%s|%s|%s|%s|"
- cpu0 = self.get_cpu_on_off(0).center(8)[:8]
- cpu1 = self.get_cpu_on_off(1).center(8)[:8]
- cpu2 = self.get_cpu_on_off(2).center(8)[:8]
- cpu3 = self.get_cpu_on_off(3).center(8)[:8]
- cpu4 = self.get_cpu_on_off(4).center(8)[:8]
- cpu5 = self.get_cpu_on_off(5).center(8)[:8]
- gpu = self.get_gpu_freq().center(11)[:11]
- table_str = stat_str % (cpu0, cpu1, cpu2, cpu3, cpu4, cpu5, gpu)
- print("\nCurrent Status:")
- for idx, val in enumerate(cur_stat):
- if idx == 5:
- print(table_str)
- else:
- print(val)
- self.print_cpu_gpu_gov()
- def set_gear(self, g):
- self.gear = load_gear(g)
- if g > 3:
- for (cpu, freq) in reversed(list(enumerate(self.gear["cpu"]))):
- enabled = freq > 0
- self.set_cpu_on_off(cpu, int(enabled))
- if enabled:
- self.set_cpu_max_freq(cpu, freq)
- else:
- for (cpu, freq) in enumerate(self.gear["cpu"]):
- enabled = freq > 0
- self.set_cpu_on_off(cpu, int(enabled))
- if enabled:
- self.set_cpu_max_freq(cpu, freq)
- self.set_gpu(self.gear["gpu_gov"], self.gear["gpu_freq"])
- # TODO: Generalise this
- if self.gear["cpu"][0] > 0:
- self.set_cpu_gov0(self.cpu_scaling_governor)
- else:
- self.set_cpu_gov4(self.cpu_scaling_governor)
- def print_gear_map(gear: int) -> NoReturn:
- print(
- " +-----------------------------------+-----------------+-----------+\n"
- " | Cortex-A53 | Cortex-A72 | Mali-T860 |\n"
- " +--------+--------+--------+--------+--------+--------+-----------+\n"
- " | CPU 0 | CPU 1 | CPU 2 | CPU 3 | CPU 4 | CPU 5 | GPU |\n"
- "+---+--------+--------+--------+--------+--------+--------+-----------+"
- )
- def freq(khz: int) -> str:
- mhz = khz // 1000
- if mhz >= 1000:
- return f"{mhz} MHz"
- elif mhz > 0:
- return f" {mhz} MHz"
- else:
- return " OFF "
- for idx, val in enumerate(gears):
- g = idx + 1
- selected = g == gear
- print(
- "|%s|%s| %s |%s"
- % (
- ("*%s*" if selected else " %s ") % g,
- "|".join([freq(cpu) for cpu in val["cpu"]]),
- freq(val["gpu_freq"] // 1000),
- " <===" if selected else "",
- )
- )
- print("+---+--------+--------+--------+--------+--------+--------+-----------+")
- def print_help_msg() -> NoReturn:
- print("Usage: devterm-a06-gearbox [OPTION]...")
- print(
- "Show or set the CPU operating frequency,online status and GPU operating frequency for DevTerm A06.\n"
- )
- print(f" -s, --set [n] set a speed mode between the number 1-{len(gears)}:")
- for (i, _) in enumerate(gears):
- print(" %d for %s." % (i + 1, gears[i]["use"]))
- print()
- print("Examples:")
- # TODO: Generate this example
- print("Set to mode 1, single LITTLE core @600MHz(max), GPU@200MHz.")
- print(" $ devterm-a06-gearbox -s 1")
- def is_root() -> bool:
- return os.geteuid() == 0
- def main() -> SystemExit:
- devterm = A06()
- parser = argparse.ArgumentParser(add_help=False)
- parser.add_argument("-s", "--set", type=int)
- parser.add_argument("-h", "--help", action="store_true")
- args = parser.parse_args()
- if args.set:
- gear = args.set
- if gear not in range(1, len(gears) + 1):
- print(f"Illegal input: mode range 1-{len(gears)}")
- sys.exit(-1)
- if is_root():
- devterm.set_gear(gear)
- print_gear_map(gear)
- devterm.print_cpu_gpu_gov()
- else:
- print("Requires super user privilege to set mode, try running it with sudo.")
- sys.exit(1)
- elif args.help:
- print_help_msg()
- sys.exit()
- else:
- if len(sys.argv) == 1:
- devterm.print_cur_status()
- sys.exit()
- if __name__ == "__main__":
- main()
|