#!/usr/bin/python
# COPYRIGHT (C) 2006 Matthew Garman
# matthew (dot) garman (at) gmail (dot) com
# License: MIT <http://www.opensource.org/licenses/mit-license.php>
import sys, time, getopt, os, re
class tpPowerMonitorDataSource:
# The name of the file to parse for needed data (data_source_file) OR
# the name of a command whose output will be read as a file
# (os.popen(shell_command)).
#
# Typically, data_source_file will be something from /proc (or
# possibly /sys), and shell_command will be a utility+arguments such
# as "iwconfig eth1 power".
#
# Note that you should set EITHER data_source_file OR shell_command,
# but not both.
data_source_file = None
shell_command = None
# The line number of the data file containing our data OR the string
# that marks the beginning of the line in which we're interested
# (startswith_string) OR a regular expression to key the data for
# which we're looking (regexp_pattern).
line_number = None
startswith_string = None
regexp_pattern = None
# The string that delimits the line containing our data. This will be
# used as the parameter to split(). Note that you can leave this as
# None to split on whitespace.
split_string = None
# A mapping of data names to indices in the split string.
name_index_map = dict()
def __init__(self, file, cmd, line, swstr, pat, splstr, map):
self.data_source_file = file
self.shell_command = cmd
self.line_number = line
self.startswith_string = swstr
self.regexp_pattern = pat
self.split_string = splstr
self.name_index_map = map
def printMembers(self):
print "\t" + 'self.data_source_file = ' + str(self.data_source_file)
print "\t" + 'self.shell_command = ' + str(self.shell_command)
print "\t" + 'self.line_number = ' + str(self.line_number)
print "\t" + 'self.startswith_string = ' + str(self.startswith_string)
print "\t" + 'self.regexp_pattern = ' + str(self.regexp_pattern)
print "\t" + 'self.split_string = ' + str(self.split_string)
print "\t" + 'self.name_index_map = ' + str(self.name_index_map)
def readData(self, d):
# open the data source file or run the command that will produce
# the data
file = None
if self.data_source_file:
file = open(self.data_source_file, 'r')
elif self.shell_command:
file = os.popen(self.shell_command)
else:
print 'error: no data source defined'
self.printMembers()
# read the contents of the file into a list and close the file
lines = None
if file:
lines = file.readlines()
file.close()
else:
print 'error: data file not opened'
self.printMembers()
# now get the line in the file with our data
line = None
if lines:
if None != self.line_number:
line = lines[self.line_number]
elif self.startswith_string:
for l in lines:
if l.startswith(self.startswith_string):
line = l
break
elif self.regexp_pattern:
for l in lines:
if re.compile(self.regexp_pattern).match(l):
line = l
break
else:
print 'error: no lines in data file'
self.printMembers()
# now get the data we want from the line itself
if line:
fields = line.split(self.split_string)
for key in self.name_index_map.keys():
d[key] = fields[self.name_index_map[key]].strip()
else:
print 'error: could not find line with data in file'
self.printMembers()
# END --- class tpPowerMonitorDataSource
tp_data_sources = [
tpPowerMonitorDataSource(
'/proc/cpuinfo', # data file
None, # shell command
None, # line index
'cpu MHz', # startswith() string
None, # regexp pattern
':', # split string
{'proc_cpuinfo_cpu_mhz': 1} # name-to-index map
),
# http://thinkwiki.org/wiki/Ipw2200#Power_Management
tpPowerMonitorDataSource(
None, # data file
'/sbin/iwpriv eth1 get_power', # shell command
0, # line index
None, # startswith() string
None, # regexp pattern
':', # split string
{'iwpriv_get_power': 2} # name-to-index map
),
# http://forums.gentoo.org/viewtopic-t-447841.html
# see post from ruben on Wed Mar 29, 2006 4:09 am
tpPowerMonitorDataSource(
None,
'/usr/bin/sudo /sbin/hdparm -C /dev/sda',
None,
' drive state is',
None,
':',
{'hard_drive_power_state': 1}
),
# http://thinkwiki.org/wiki/Thermal_sensors
tpPowerMonitorDataSource(
'/proc/acpi/ibm/thermal',
None,
0,
None,
None,
None,
{'ibm_thermal_cpu': 1,
'ibm_thermal_hdaps': 2,
'ibm_thermal_pmcia': 3,
'ibm_thermal_gpu': 4,
'ibm_thermal_batt_fl': 5,
'ibm_thermal_batt_br': 7 }
),
# http://mailman.linux-thinkpad.org/pipermail/linux-thinkpad/2006-July/034738.html
tpPowerMonitorDataSource(
'/proc/acpi/processor/CPU/power',
None,
None,
'bus master activity',
None,
':',
{'bus_master_activity': 1}
),
tpPowerMonitorDataSource(
'/proc/acpi/battery/BAT0/state',
None,
None,
'present rate:',
None,
None,
{'battery0_state_present_rate': 2}
),
tpPowerMonitorDataSource(
None,
'/usr/bin/sudo /usr/sbin/radeontool dac',
0,
None,
None,
None,
{'radeontool_dac_ext_vga': -1}
),
tpPowerMonitorDataSource(
None,
'/usr/bin/sudo /usr/sbin/radeontool light',
0,
None,
None,
None,
{'radeontool_light_lcd': -1}
),
# http://forums.gentoo.org/viewtopic-t-343029-highlight-rovclock.html
tpPowerMonitorDataSource(
None,
'/usr/bin/sudo /usr/sbin/rovclock -i',
None,
'Core: ',
None,
None,
{'rovclock_gpu_clock': 1,
'rovclock_mem_clock': 4 }
),
tpPowerMonitorDataSource(
None,
'/sbin/iwconfig eth1',
None,
None,
'.*Power Management.*',
':',
{'wireless_power_mgmt_state': 1}
),
tpPowerMonitorDataSource(
'/proc/loadavg',
None,
0,
None,
None,
None,
{'proc_loadavg_1min': 0,
'proc_loadavg_5min': 1,
'proc_loadavg_15min': 2 }
),
]
log_data = list()
poll_freq_hz = 1
run_time_sec = 60*60
logfile = None
def collectPowerData():
global run_time_sec
global poll_freq_hz
global log_data
global logfile
global tp_data_sources
log = None
if logfile:
log = open(logfile, 'w')
log.write("time, data\n")
end_t = time.time()+run_time_sec
while time.time() < end_t:
datum = dict()
for dsrc in tp_data_sources:
dsrc.readData(datum)
log_data.append(datum)
if log: log.write(str(time.time()) + ', ' + str(datum) + "\n")
time.sleep(1.0/float(poll_freq_hz))
if log: log.close()
def createStats():
global log_data
global logfile
log = None
if logfile:
log = open(logfile, 'a')
if log: log.close()
def usage():
print 'Usage: ' + sys.argv[0] + \
'[-h] [-f freq_hz] [-t time_min] [-l logfile]'
print "\t-h Display this help"
print "\t-f freq_hz The polling frequency in Hertz, default=" \
+ str(poll_freq_hz)
print "\t-t time_min Total poll time in minutes, default=" \
+ str(run_time_sec/60)
print "\t-l logfile Write poll data to indicated log file"
def main():
global poll_freq_hz
global run_time_sec
global logfile
# parse options
# see: http://docs.python.org/lib/module-getopt.html
try:
opts, args = getopt.getopt(sys.argv[1:], "hf:t:l:")
except getopt.GetoptError:
usage()
sys.exit(2)
for o, a in opts:
if '-h' == o:
usage()
sys.exit()
if '-f' == o:
poll_freq_hz = int(a)
if '-t' == o:
run_time_sec = int(a)*60
if '-l' == o:
logfile = a
# do stuff
print 'running for ' + str(run_time_sec/60) + ' minutes'
print 'poll rate: ' + str(poll_freq_hz) + ' Hz'
print 'logfile: ' + str(logfile)
collectPowerData()
print '...done!'
print 'Data points collected: ' + str(len(log_data))
sum = 0
for d in log_data:
sum += int(d['battery0_state_present_rate'])
avg = float(sum) / float(len(log_data))
print 'Average power draw: ' + str(avg)
if logfile: log = open(logfile, 'a')
if log:
log.write('Average power draw: ' + str(avg))
log.close()
if __name__ == '__main__':
main()
