power_Draw/Idle/LoadTest: Support temperature sensing for x86

client/cros/power_status.py

 import glob, logging, os, re, time
 from autotest_lib.client.bin import utils
 from autotest_lib.client.common_lib import error
 
 
 class DevStat(object):
     """
     Device power status. This class implements generic status initialization
     and parsing routines.
     """
@@ skipping 23 matching lines @@
             return field_type(0)
 
 
     def read_all_vals(self):
         for field, prop in self.fields.iteritems():
             if prop[0]:
                 val = self.read_val(prop[0], prop[1])
                 setattr(self, field, val)
 
 
+class ThermalStat(DevStat):
+    """
+    Thermal status.
+
+    Fields:
+    (All temperatures are in degrees Celsius times 1000.)
+
+    str   enabled:            Whether thermal zone is enabled
+    int   temp:               Current temperature
+    str   type:               Thermal zone type
+    int   num_trip_points:    Number of thermal trip points that activate
+                                cooling devices
+    int   num_points_tripped: Temperature is above this many trip points
+    str   trip_point_N_type:  Trip point #N's type
+    int   trip_point_N_temp:  Trip point #N's temperature value
+    int   cdevX_trip_point:   Trip point o cooling device #X (index)
+    """
+
+    MAX_TRIP_POINTS = 20
+
+    thermal_fields = {
+        'enabled':              ['enabled', str],
+        'temp':                 ['temp', int],
+        'type':                 ['type', str],
+        'num_points_tripped':   ['', '']
+        }
+    def __init__(self, path=None):
+        # Dynamically generate the rest of the fields
+        thermal_point_strings = [ 'trip_point_%d_type',
+                                  'trip_point_%d_temp',
+                                  'cdev%d_trip_point' ]
+        thermal_point_types = [ str, int, int ]
+        self.num_trip_points = 0
+
+        for i in range(MAX_TRIP_POINTS):
+            for j in range(len(thermal_point_strings)):
+                file = thermal_point_strings[j] % i
+
+                if os.path.exists(path + file):
+                    self.thermal_fields[file] = [file, thermal_point_types[j]]
+                    if j == 0:
+                        num_trip_points += 1
+
+        super(ThermalStat, self).__init__(self.thermal_fields, path)
+        self.update()
+
+    def update(self):
+        if not os.path.exists(self.path):
+            return
+
+        self.read_all_vals()
+        self.num_points_tripped = 0
+
+        for i in range(self.num_trip_points):

[Sameer Nanda, 2011/04/06 18:13:40]

probably better to just save the names of all trip point temp files in the loop
in __init__ and just loop through them.  That way there is no requirement for
temp files names to be numerically contiguous.

[Simon Que, 2011/04/07 20:59:47]

Done.

+            if self.temp > self.read_val('trip_point_%d_temp' % i, int):
+                self.num_points_tripped += 1
+
+
+
 class BatteryStat(DevStat):
     """
     Battery status.
 
     Fields:
 
     float charge_full:        Last full capacity reached [Ah]
     float charge_full_design: Full capacity by design [Ah]
     float charge_now:         Remaining charge [Ah]
     float current_now:        Battery discharge rate [A]
@@ skipping 74 matching lines @@
         self.online = self.is_online == 1
 
 
 class SysStat(object):
     """
     System power status for a given host.
 
     Fields:
 
     battery:   A list of BatteryStat objects.
-    linepower: A list of LineStat opbjects.
+    linepower: A list of LineStat objects.
     """
 
     def __init__(self):
         power_supply_path = '/sys/class/power_supply/*'
         self.battery = None
         self.linepower = None
+        self.thermal = None
         battery_path = None
         linepower_path = None
+        thermal_path = '/sys/class/thermal/thermal_zone0'

[Sameer Nanda, 2011/04/06 18:13:40]

do you want to use thermal_zone* here instead?  I think if you switch to glob,
this may just work naturally.

[Simon Que, 2011/04/07 20:59:47]

Done.

+
         power_supplies = glob.glob(power_supply_path)
         for path in power_supplies:
             type_path = os.path.join(path,'type')
             if not os.path.exists(type_path):
                 continue
             type = utils.read_one_line(type_path)
             if type == 'Battery':
                 battery_path = path
             elif type == 'Mains':
                 linepower_path = path
         if battery_path and linepower_path:
             self.battery_path = battery_path
             self.linepower_path = linepower_path
         else:
             raise error.TestError('Battery or Linepower path not found')
+        self.thermal_path = thermal_path
 
+        self.min_temp = 999999999
+        self.max_temp = -999999999
 
     def refresh(self):
         """
         Initialize device power status objects for a single battery and a
         single power line by parsing the output of devkit-power -d.
         """
         self.battery = [ BatteryStat(self.battery_path) ]
         self.linepower = [ LineStat(self.linepower_path) ]
+        self.thermal = [ ThermalStat(self.thermal_path) ]
+
+        try:

[Sameer Nanda, 2011/04/06 18:13:40]

lets add a logging.info for the temperature that was just read.  This way we
have the data to plot out a temperature curve over the duration of a test.

[Simon Que, 2011/04/07 20:59:47]

This isn't trivial, as there may be other log messages between the refresh()
calls.  Also, the SysStat object does not know the current time, so it can only
record a sequence of temp values.

[Simon Que, 2011/04/07 23:30:51]

Done.

+            if self.thermal[0].temp < self.min_temp:
+                self.min_temp = self.thermal[0].temp
+            if self.thermal[0].temp > self.max_temp:
+                self.max_temp = self.thermal[0].temp
+        except:
+            logging.error('Could not read temperature, skipping.')
 
 
 def get_status():
     """
     Return a new power status object (SysStat). A new power status snapshot
     for a given host can be obtained by either calling this routine again and
     constructing a new SysStat object, or by using the refresh method of the
     SysStat object.
     """
     status = SysStat()
@@ skipping 208 matching lines @@
 
             active = int(utils.read_file(active_duration_path))
             connected = int(utils.read_file(connected_duration_path))
             logging.debug('device %s active for %.2f%%',
                           path, active * 100.0 / connected)
 
             total_active += active
             total_connected += connected
 
         return total_active, total_connected