python: Add a Log class to control libsigrok logging.

[libsigrok.git] / bindings / python / sigrok / core / classes.py

diff --git a/bindings/python/sigrok/core/classes.py b/bindings/python/sigrok/core/classes.py
index cf955e9430801a4f8b18135121d1a98b7abbb033..89049cc4a15b94f993209fd6e196ff84260ad632 100644 (file)
--- a/bindings/python/sigrok/core/classes.py
+++ b/bindings/python/sigrok/core/classes.py
@@ -19,11 +19,11 @@
 
 from functools import partial
 from fractions import Fraction
-from lowlevel import *
-import lowlevel
+from .lowlevel import *
+from . import lowlevel
 import itertools
 
-__all__ = ['Error', 'Context', 'Driver', 'Device', 'Session', 'Packet']
+__all__ = ['Error', 'Context', 'Driver', 'Device', 'Session', 'Packet', 'Log']
 
 class Error(Exception):
 
@@ -54,7 +54,7 @@ def gvariant_to_python(value):
         return Fraction(
             g_variant_get_uint64(g_variant_get_child_value(value, 0)),
             g_variant_get_uint64(g_variant_get_child_value(value, 1)))
-    raise NotImplementedError, (
+    raise NotImplementedError(
         "Can't convert GVariant type '%s' to a Python type." % type_string)
 
 def python_to_gvariant(value):
@@ -73,7 +73,7 @@ def python_to_gvariant(value):
         result = g_variant_new_tuple(array, 2)
         delete_gvariant_ptr_array(array)
         return result
-    raise NotImplementedError, (
+    raise NotImplementedError(
         "Can't convert Python '%s' to a GVariant." % type(value))
 
 def callback_wrapper(session, callback, device_ptr, packet_ptr):
@@ -152,8 +152,8 @@ class Device(object):
         try:
             check(sr_config_get(self.driver.struct, key, data, self.struct))
         except Error as error:
-            if error.errno == SR_ERR_ARG:
-                raise NotImplementedError, (
+            if error.errno == SR_ERR_NA:
+                raise NotImplementedError(
                     "Device does not implement %s" % name)
             else:
                 raise AttributeError
@@ -225,8 +225,11 @@ class Packet(object):
             if self.struct.type == SR_DF_LOGIC:
                 self._payload = Logic(self,
                     void_ptr_to_sr_datafeed_logic_ptr(pointer))
+            elif self.struct.type == SR_DF_ANALOG:
+                self._payload = Analog(self,
+                    void_ptr_to_sr_datafeed_analog_ptr(pointer))
             else:
-                raise NotImplementedError, (
+                raise NotImplementedError(
                     "No Python mapping for packet type %ѕ" % self.struct.type)
         return self._payload
 
@@ -243,6 +246,60 @@ class Logic(object):
             self._data = cdata(self.struct.data, self.struct.length)
         return self._data
 
+class Analog(object):
+
+    def __init__(self, packet, struct):
+        self.packet = packet
+        self.struct = struct
+        self._data = None
+
+    @property
+    def num_samples(self):
+        return self.struct.num_samples
+
+    @property
+    def mq(self):
+        return self.struct.mq
+
+    @property
+    def unit(self):
+        return self.struct.unit
+
+    @property
+    def mqflags(self):
+        return self.struct.mqflags
+
+    @property
+    def data(self):
+        if self._data is None:
+            self._data = float_array.frompointer(self.struct.data)
+        return self._data
+
+class Log(object):
+
+    NONE = SR_LOG_NONE
+    ERR = SR_LOG_ERR
+    WARN = SR_LOG_WARN
+    INFO = SR_LOG_INFO
+    DBG = SR_LOG_DBG
+    SPEW = SR_LOG_SPEW
+
+    @property
+    def level(self):
+        return sr_log_loglevel_get()
+
+    @level.setter
+    def level(self, l):
+        check(sr_log_loglevel_set(l))
+
+    @property
+    def domain(self):
+        return sr_log_logdomain_get()
+
+    @domain.setter
+    def domain(self, d):
+        check(sr_log_logdomain_set(d))
+
 for symbol_name in dir(lowlevel):
     prefix = 'SR_DF_'
     if symbol_name.startswith(prefix):