123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316 |
- #!/usr/bin/python3
- import glob
- import os
- import sys,getopt
- import subprocess
- import time
- # 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 :)
- def gears():
- 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)"),
- ]
- GPU_GOV_SIMPLE = "simple_ondemand"
- GPU_GOV_PERF = "performance"
- # 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
- # disabled.
- def gear(
- little=(0, 0, 0, 0),
- big=(0, 0),
- gpu_freq=200000000,
- gpu_gov=GPU_GOV_SIMPLE,
- use="",
- ):
- # 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):
- return gears[gear - 1]
- cur_stat = []
- cur_stat.append("+-----------------------------------+-----------------+-----------+")
- cur_stat.append("| Cortex-A53 | Cortex-A72 | Mali-T860 |")
- cur_stat.append("+--------+--------+--------+--------+--------+--------+-----------+")
- cur_stat.append("| CPU 0 | CPU 1 | CPU 2 | CPU 3 | CPU 4 | CPU 5 | GPU |")
- cur_stat.append("+--------+--------+--------+--------+--------+--------+-----------+")
- cur_stat.append("| 600MHz | OFF | OFF | OFF | OFF | OFF | 400MHz |") #5
- cur_stat.append("+--------+--------+--------+--------+--------+--------+-----------+")
- def isDigit(x):
- try:
- float(x)
- return True
- except ValueError:
- return False
- class A06:
- cpus = []
- cpu_scaling_governor= "schedutil"
- gear = load_gear(1) # 1-5
- null_out = "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()
-
- def get_cpu_gov(self):
- 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"
- gov = ""
- with open(cpu_gov_path,"r") as f: gov = f.read().strip()
- return gov
- 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:
- 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:
- 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
- onoff = "0"
- max_freq = "0"
- 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 "%dMhz" % 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:
- #print("echo %d | sudo tee %s" %(onoff,cpu_onoff_file) )
- subprocess.run( "echo %d | sudo tee %s" %(onoff,cpu_onoff_file),shell=True,stdout=subprocess.DEVNULL)
- except:
- 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:
- 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")
- freq = ""
- with open(gpu_freq_path,"r") as f: freq = f.read().strip()
- mhz = int(freq)/1000000
- return "%dMHz" % 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:
- print("set gpu failed")
- def print_cpu_gpu_gov(self):
- print("CPU Governor: %s GPU Governor: %s" % (self.get_cpu_gov(), self.gear["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)
- 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):
- print(" +-----------------------------------+-----------------+-----------+")
- print(" | Cortex-A53 | Cortex-A72 | Mali-T860 |")
- print(" +--------+--------+--------+--------+--------+--------+-----------+")
- print(" | CPU 0 | CPU 1 | CPU 2 | CPU 3 | CPU 4 | CPU 5 | GPU |")
- div = "+---+--------+--------+--------+--------+--------+--------+-----------+"
- print(div)
- def freq(khz):
- mhz = khz/1000
- if mhz >= 1000:
- return "%d MHz" % mhz
- elif mhz > 0:
- return " %d 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(div)
- def print_help_msg():
- print("Usage: devterm-a06-gearbox [OPTION]...")
- print("Show or set the CPU operating frequency,online status and GPU operating frequency for DevTerm A06.")
- print()
- print(" -s, --set [n] set a speed mode between the number 1-%d:" % 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():
- return os.geteuid() == 0
- def main(argv):
- gear = 1
- try:
- opts, args = getopt.getopt(argv,"hs:",["set="])
- except getopt.GetoptError:
- print_help_msg()
- sys.exit(2)
- for opt, arg in opts:
- if opt == '-h':
- print_help_msg()
- sys.exit()
- elif opt in ("-s","--set"):
- if(isDigit(arg)):
- gear = int(arg)
- if gear not in range(1, len(gears) + 1):
- print("illegal input: mode range 1-%d" % len(gears))
- sys.exit(-1)
-
-
- DT = A06()
-
- if len(argv) == 0:
- DT.print_cur_status()
- sys.exit(0)
- DT = A06()
- if is_root():
- DT.set_gear(gear)
- print_gear_map(gear)
- DT.print_cpu_gpu_gov()
- else:
- print("Require super user privilege to set mode,try run it with sudo")
- sys.exit(1)
- if __name__ == "__main__":
- main(sys.argv[1:])
-
|