xmr: bp - memory view optimized

ph4r05 · Aug 17, 2018 · 83ba7a6 · 83ba7a6
1 parent b517906
commit 83ba7a6
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/apps/monero/xmr/bulletproof.py b/src/apps/monero/xmr/bulletproof.py
@@ -274,6 +274,7 @@ class KeyV(object):
     def __init__(self, elems=64, src=None, buffer=None):
         self.current_idx = 0
         self.d = None
+        self.mv = None
         self.size = elems
         if src:
             self.d = bytearray(src.d)
@@ -283,14 +284,18 @@ def __init__(self, elems=64, src=None, buffer=None):
             self.size = len(buffer) // 32
         else:
             self.d = bytearray(32 * elems)
+        self._set_mv()
+
+    def _set_mv(self):
+        self.mv = memoryview(self.d)
 
     def __getitem__(self, item):
         """
         Returns corresponding 32 byte array
         :param item:
         :return:
         """
-        return memoryview(self.d)[item * 32 : (item + 1) * 32]
+        return self.mv[item * 32 : (item + 1) * 32]
 
     def __setitem__(self, key, value):
         """
@@ -326,10 +331,14 @@ def slice_r(self, start, stop):
         res = KeyV(stop - start)
         return self.slice(res, start, stop)
 
+    def copy_from(self, src):
+        self.size = self.size
+        self.d = bytearray(self.d)
+        self._set_mv()
+
     def copy(self, dst=None):
         if dst:
-            dst.size = self.size
-            dst.d = bytearray(self.d)
+            dst.copy_from(self)
         else:
             dst = KeyV(src=self)
         return dst
@@ -342,6 +351,7 @@ def resize(self, nsize, chop=False):
         else:
             self.d = bytearray(nsize * 32)
         self.size = nsize
+        self._set_mv()
 
 
 class KeyVEval(KeyV):