pvkoala 1.0.1__py3-none-any.whl → 2.0.0__py3-none-any.whl

pvkoala/_koala.py

@@ -16,7 +16,27 @@ from typing import Sequence
 
 
 class KoalaError(Exception):
-    pass
+    def __init__(self, message: str = '', message_stack: Sequence[str] = None):
+        super().__init__(message)
+
+        self._message = message
+        self._message_stack = list() if message_stack is None else message_stack
+
+    def __str__(self):
+        message = self._message
+        if len(self._message_stack) > 0:
+            message += ':'
+            for i in range(len(self._message_stack)):
+                message += '\n  [%d] %s' % (i, self._message_stack[i])
+        return message
+
+    @property
+    def message(self) -> str:
+        return self._message
+
+    @property
+    def message_stack(self) -> Sequence[str]:
+        return self._message_stack
 
 
 class KoalaMemoryError(KoalaError):
@@ -123,15 +143,34 @@ class Koala(object):
 
         library = cdll.LoadLibrary(library_path)
 
+        set_sdk_func = library.pv_set_sdk
+        set_sdk_func.argtypes = [c_char_p]
+        set_sdk_func.restype = None
+
+        set_sdk_func('python'.encode('utf-8'))
+
+        self._get_error_stack_func = library.pv_get_error_stack
+        self._get_error_stack_func.argtypes = [POINTER(POINTER(c_char_p)), POINTER(c_int)]
+        self._get_error_stack_func.restype = self.PicovoiceStatuses
+
+        self._free_error_stack_func = library.pv_free_error_stack
+        self._free_error_stack_func.argtypes = [POINTER(c_char_p)]
+        self._free_error_stack_func.restype = None
+
         init_func = library.pv_koala_init
         init_func.argtypes = [c_char_p, c_char_p, POINTER(POINTER(self.CKoala))]
         init_func.restype = self.PicovoiceStatuses
 
         self._handle = POINTER(self.CKoala)()
 
-        status = init_func(access_key.encode(), model_path.encode(), byref(self._handle))
+        status = init_func(
+            access_key.encode(),
+            model_path.encode(),
+            byref(self._handle))
         if status is not self.PicovoiceStatuses.SUCCESS:
-            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status]()
+            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status](
+                message='Initialization failed',
+                message_stack=self._get_error_stack())
 
         self._delete_func = library.pv_koala_delete
         self._delete_func.argtypes = [POINTER(self.CKoala)]
@@ -144,7 +183,10 @@ class Koala(object):
         status = delay_sample_func(self._handle, delay_sample)
         if status is not self.PicovoiceStatuses.SUCCESS:
             self.delete()
-            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status]()
+            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status](
+                message='Failed to get delay samples',
+                message_stack=self._get_error_stack())
+
         self._delay_sample = delay_sample.value
 
         self._process_func = library.pv_koala_process
@@ -193,7 +235,9 @@ class Koala(object):
 
         status = self._process_func(self._handle, pcm, enhanced_pcm)
         if status is not self.PicovoiceStatuses.SUCCESS:
-            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status]()
+            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status](
+                message='Processing failed',
+                message_stack=self._get_error_stack())
 
         # noinspection PyTypeChecker
         return list(enhanced_pcm)
@@ -206,7 +250,9 @@ class Koala(object):
 
         status = self._reset_func(self._handle)
         if status is not self.PicovoiceStatuses.SUCCESS:
-            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status]()
+            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status](
+                message='Reset failed',
+                message_stack=self._get_error_stack())
 
     def delete(self) -> None:
         """Releases resources acquired by Koala."""
@@ -239,6 +285,21 @@ class Koala(object):
 
         return self._version
 
+    def _get_error_stack(self) -> Sequence[str]:
+        message_stack_ref = POINTER(c_char_p)()
+        message_stack_depth = c_int()
+        status = self._get_error_stack_func(byref(message_stack_ref), byref(message_stack_depth))
+        if status is not self.PicovoiceStatuses.SUCCESS:
+            raise self._PICOVOICE_STATUS_TO_EXCEPTION[status](message='Unable to get Koala error state')
+
+        message_stack = list()
+        for i in range(message_stack_depth.value):
+            message_stack.append(message_stack_ref[i].decode('utf-8'))
+
+        self._free_error_stack_func(message_stack_ref)
+
+        return message_stack
+
 
 __all__ = [
     'Koala',

{pvkoala-1.0.1.dist-info → pvkoala-2.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pvkoala
-Version: 1.0.1
+Version: 2.0.0
 Summary: Koala Noise Suppression Engine.
 Home-page: https://github.com/Picovoice/koala
 Author: Picovoice

pvkoala-2.0.0.dist-info/RECORD

@@ -0,0 +1,19 @@
+pvkoala/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+pvkoala/__init__.py,sha256=46DQVzd7KMWWHnksLGV0vT6VPuZQe9Ivvdf2VQjLRhY,570
+pvkoala/_factory.py,sha256=iZqM5lgr130Z_BjDgSi3maA-4dGO4J9abwUlNsX84Vo,1458
+pvkoala/_koala.py,sha256=2NwR_XwjcllPA8FHTE4Q-ofp3X78YaCBGuEZNiafPdo,10689
+pvkoala/_util.py,sha256=wU3RwME0llwlBkMCmVRmfKHGsrQJrGZwxbtxkfvjSCo,3135
+pvkoala/lib/common/koala_params.pv,sha256=A_92Jxj36MvGe3oboq70wFubecCELhAkNZvTEJk6vfQ,4007743
+pvkoala/lib/jetson/cortex-a57-aarch64/libpv_koala.so,sha256=R6ylMuZDy3GNJTpI_i4S2qesQk1_3GulWnl3ZhhKuKg,142656
+pvkoala/lib/linux/x86_64/libpv_koala.so,sha256=9sE-zswQZEt6i2S1kFYgc9ypkWTFFTpWN6xkjJ7XMAk,163504
+pvkoala/lib/mac/arm64/libpv_koala.dylib,sha256=5bswF2_yIr4dA6z9Ff8IIyQnVBFwX8eBD3Ioon2_VvU,203502
+pvkoala/lib/mac/x86_64/libpv_koala.dylib,sha256=rg65W2omZecWDu6uFOe-iOyrCsMiGjOwNINBNAAFGAE,218256
+pvkoala/lib/raspberry-pi/cortex-a53/libpv_koala.so,sha256=5IwELivbaznf4mn-uGO9FUJUkx3Kwn_s3avPLa46Vz8,129188
+pvkoala/lib/raspberry-pi/cortex-a53-aarch64/libpv_koala.so,sha256=7PyFuqYL1XcvumWb0VDYzjLSUtrb2RlM0fcXXdrxrKw,142656
+pvkoala/lib/raspberry-pi/cortex-a72/libpv_koala.so,sha256=0HpQr4t6IXiu_o-GS3mV5QczmgvrdYEVZ8zUM0bp-kA,133284
+pvkoala/lib/raspberry-pi/cortex-a72-aarch64/libpv_koala.so,sha256=LjFS24G2TIaJtcpZoELC5-tvaw7QAAdOyM2UZJMHZrY,142656
+pvkoala/lib/windows/amd64/libpv_koala.dll,sha256=h586bu40qiAw2FG06e9Eg1EX4cBZvS2jHodwvaEsNkI,282112
+pvkoala-2.0.0.dist-info/METADATA,sha256=cHRGEixxK3jMCC41V3TIxueiHx729hB8-6Mu9vDC3pU,2785
+pvkoala-2.0.0.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
+pvkoala-2.0.0.dist-info/top_level.txt,sha256=Gb-hF6eAZIh7Bm1LfLuwsCQ_0y7FWSIuYSQX0p3T6No,8
+pvkoala-2.0.0.dist-info/RECORD,,

{pvkoala-1.0.1.dist-info → pvkoala-2.0.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: bdist_wheel (0.34.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

pvkoala-1.0.1.dist-info/RECORD

@@ -1,19 +0,0 @@
-pvkoala/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-pvkoala/__init__.py,sha256=46DQVzd7KMWWHnksLGV0vT6VPuZQe9Ivvdf2VQjLRhY,570
-pvkoala/_factory.py,sha256=iZqM5lgr130Z_BjDgSi3maA-4dGO4J9abwUlNsX84Vo,1458
-pvkoala/_koala.py,sha256=KWZlVNRFV0nmrNy2KaNOZU6uMZlXVHFxheEVgpVHsBo,8370
-pvkoala/_util.py,sha256=wU3RwME0llwlBkMCmVRmfKHGsrQJrGZwxbtxkfvjSCo,3135
-pvkoala/lib/common/koala_params.pv,sha256=3lDAuNhemokD-BmhHS1jJzvb3x3F_7u0AA_msQGc61o,4007735
-pvkoala/lib/jetson/cortex-a57-aarch64/libpv_koala.so,sha256=O84bKNt2ccsaWcfKmwekTYw7uUG0eUdVu5SNku5l6Kg,133600
-pvkoala/lib/linux/x86_64/libpv_koala.so,sha256=ZK--RIMsHSv_AUEC6KiuuLJW4EpVT4klk9d5Yr-YAqg,150352
-pvkoala/lib/mac/arm64/libpv_koala.dylib,sha256=xor3S9k-rgk85h03je-mhAo_ZEn4lPnDCl-uK70CFxU,203166
-pvkoala/lib/mac/x86_64/libpv_koala.dylib,sha256=tbfpxLQnxT4vLOLu2tkQNa2FoKjmLmlSa4uzwl29mrs,217912
-pvkoala/lib/raspberry-pi/cortex-a53/libpv_koala.so,sha256=QTVgoqL49iOSDXycT_nAOc7On-txbXcztlFX4tFwi8Y,120620
-pvkoala/lib/raspberry-pi/cortex-a53-aarch64/libpv_koala.so,sha256=WTRuKpSNUsm6so1FaQUwP_boTE2Xs_n-VGQGrSj_W4U,133688
-pvkoala/lib/raspberry-pi/cortex-a72/libpv_koala.so,sha256=fQe4_Td1tWfC5GVvqY-5sk0s7sFmYH1G41WaupHyoGI,124716
-pvkoala/lib/raspberry-pi/cortex-a72-aarch64/libpv_koala.so,sha256=7Udm5LN-kqD9ICs20AveKw81ZFE6RuD3mmH6UzfZIDA,133688
-pvkoala/lib/windows/amd64/libpv_koala.dll,sha256=uWdtDcuYFpUe-0w2OHy8RRfQwrd0IKHpMGXZk2XFGs0,303104
-pvkoala-1.0.1.dist-info/METADATA,sha256=WXc6hhx1_yxkiL2AGcaUzd2zLCGrVHWy0gkxF6cGh6E,2785
-pvkoala-1.0.1.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-pvkoala-1.0.1.dist-info/top_level.txt,sha256=Gb-hF6eAZIh7Bm1LfLuwsCQ_0y7FWSIuYSQX0p3T6No,8
-pvkoala-1.0.1.dist-info/RECORD,,