mirror of
https://github.com/ARM-software/devlib.git
synced 2025-09-09 21:41:53 +01:00
141 lines
4.3 KiB
Python
141 lines
4.3 KiB
Python
# Copyright 2015 ARM Limited
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
import re
|
|
from collections import defaultdict
|
|
|
|
from devlib.module import Module
|
|
from devlib.utils.types import integer
|
|
|
|
|
|
HWMON_ROOT = '/sys/class/hwmon'
|
|
HWMON_FILE_REGEX = re.compile(r'(?P<kind>\w+?)(?P<number>\d+)_(?P<item>\w+)')
|
|
|
|
|
|
class HwmonSensor(object):
|
|
|
|
def __init__(self, device, path, kind, number):
|
|
self.device = device
|
|
self.path = path
|
|
self.kind = kind
|
|
self.number = number
|
|
self.target = self.device.target
|
|
self.name = '{}/{}{}'.format(self.device.name, self.kind, self.number)
|
|
self.label = self.name
|
|
self.items = set()
|
|
|
|
def add(self, item):
|
|
self.items.add(item)
|
|
if item == 'label':
|
|
self.label = self.get('label')
|
|
|
|
def get(self, item):
|
|
path = self.get_file(item)
|
|
value = self.target.read_value(path)
|
|
try:
|
|
return integer(value)
|
|
except (TypeError, ValueError):
|
|
return value
|
|
|
|
def set(self, item, value):
|
|
path = self.get_file(item)
|
|
self.target.write_value(path, value)
|
|
|
|
def get_file(self, item):
|
|
if item not in self.items:
|
|
raise ValueError('item "{}" does not exist for {}'.format(item, self.name))
|
|
filename = '{}{}_{}'.format(self.kind, self.number, item)
|
|
return self.target.path.join(self.path, filename)
|
|
|
|
def __str__(self):
|
|
if self.name != self.label:
|
|
text = 'HS({}, {})'.format(self.name, self.label)
|
|
else:
|
|
text = 'HS({})'.format(self.name)
|
|
return text
|
|
|
|
__repr__ = __str__
|
|
|
|
|
|
class HwmonDevice(object):
|
|
|
|
@property
|
|
def sensors(self):
|
|
all_sensors = []
|
|
for sensors_of_kind in self._sensors.itervalues():
|
|
all_sensors.extend(sensors_of_kind.values())
|
|
return all_sensors
|
|
|
|
def __init__(self, target, path):
|
|
self.target = target
|
|
self.path = path
|
|
self.name = self.target.read_value(self.target.path.join(self.path, 'name'))
|
|
self._sensors = defaultdict(dict)
|
|
path = self.path
|
|
if not path.endswith(self.target.path.sep):
|
|
path += self.target.path.sep
|
|
for entry in self.target.list_directory(path):
|
|
match = HWMON_FILE_REGEX.search(entry)
|
|
if match:
|
|
kind = match.group('kind')
|
|
number = int(match.group('number'))
|
|
item = match.group('item')
|
|
if number not in self._sensors[kind]:
|
|
sensor = HwmonSensor(self, self.path, kind, number)
|
|
self._sensors[kind][number] = sensor
|
|
self._sensors[kind][number].add(item)
|
|
|
|
def get(self, kind, number=None):
|
|
if number is None:
|
|
return [s for _, s in sorted(self._sensors[kind].iteritems(),
|
|
key=lambda x: x[0])]
|
|
else:
|
|
return self._sensors[kind].get(number)
|
|
|
|
def __str__(self):
|
|
return 'HD({})'.format(self.name)
|
|
|
|
__repr__ = __str__
|
|
|
|
|
|
class HwmonModule(Module):
|
|
|
|
name = 'hwmon'
|
|
|
|
@staticmethod
|
|
def probe(target):
|
|
return target.file_exists(HWMON_ROOT)
|
|
|
|
@property
|
|
def sensors(self):
|
|
all_sensors = []
|
|
for device in self.devices:
|
|
all_sensors.extend(device.sensors)
|
|
return all_sensors
|
|
|
|
def __init__(self, target):
|
|
super(HwmonModule, self).__init__(target)
|
|
self.root = HWMON_ROOT
|
|
self.devices = []
|
|
self.scan()
|
|
|
|
def scan(self):
|
|
for entry in self.target.list_directory(self.root):
|
|
if entry.startswith('hwmon'):
|
|
entry_path = self.target.path.join(self.root, entry)
|
|
if self.target.file_exists(self.target.path.join(entry_path, 'name')):
|
|
device = HwmonDevice(self.target, entry_path)
|
|
self.devices.append(device)
|
|
|