| #!/usr/bin/env python3.4 |
| # |
| # Copyright (C) 2017 The Android Open Source Project |
| # |
| # 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. |
| # |
| |
| # A helper class to generate MIXED property values that can be |
| # set as the value for a diagnostic frame |
| # Spritually, the same as DiagnosticEventBuilder.java |
| |
| from diagnostic_sensors import DIAGNOSTIC_SENSOR_INTEGER_LAST_SYSTEM_INDEX |
| from diagnostic_sensors import DIAGNOSTIC_SENSOR_FLOAT_LAST_SYSTEM_INDEX |
| |
| class DiagnosticEventBuilder(object): |
| class ByteArray(object): |
| def __init__(self, numElements): |
| self.count = numElements |
| if 0 == (numElements % 8): |
| self.data = bytearray(numElements/8) |
| else: |
| # if not a multiple of 8, add one extra byte |
| self.data = bytearray(1+numElements/8) |
| |
| def _getIndices(self, bit): |
| if (bit < 0) or (bit >= self.count): |
| raise IndexError("index %d not in range [0,%d)" % (bit, self.count)) |
| byteIdx = bit / 8 |
| bitIdx = (bit % 8) |
| return byteIdx, bitIdx |
| |
| def setBit(self, bit): |
| byteIdx, bitIdx = self._getIndices(bit) |
| bitValue = pow(2,bitIdx) |
| self.data[byteIdx] = self.data[byteIdx] | bitValue |
| |
| def getBit(self, bit): |
| byteIdx, bitIdx = self._getIndices(bit) |
| bitValue = pow(2,bitIdx) |
| return 0 != self.data[byteIdx] & bitValue |
| |
| def __str__(self): |
| return str(self.data) |
| |
| def __init__(self, propConfig): |
| self.string_value = "" |
| self.bytes = "" |
| self.numIntSensors = propConfig.config[0].config_array[0] + \ |
| DIAGNOSTIC_SENSOR_INTEGER_LAST_SYSTEM_INDEX + 1 |
| self.numFloatSensors = propConfig.config[0].config_array[1] + \ |
| DIAGNOSTIC_SENSOR_FLOAT_LAST_SYSTEM_INDEX + 1 |
| self.bitmask = DiagnosticEventBuilder.ByteArray( |
| self.numIntSensors+self.numFloatSensors) |
| self.int32_values = [0] * self.numIntSensors |
| self.float_values = [0.0] * self.numFloatSensors |
| |
| def addIntSensor(self, idx, value): |
| self.int32_values[idx] = value |
| self.bitmask.setBit(idx) |
| return self |
| |
| def addFloatSensor(self, idx, value): |
| self.float_values[idx] = value |
| self.bitmask.setBit(len(self.int32_values)+idx) |
| return self |
| |
| def setStringValue(self, string): |
| self.string_value = string |
| return self |
| |
| def build(self): |
| self.bytes_value = str(self.bitmask) |
| return self |
| |
| def __str__(self): |
| s = "diagnostic event {\n" |
| for x in ['string_value', 'int32_values', 'float_values']: |
| s = s + "\t%s: %s\n" % (x, self.__dict__[x]) |
| return s + "}" |