fix: Add wrap ctor to MutableRawBuffer

mrousavy · Apr 21, 2024 · 8227762 · 8227762
1 parent 40c02ef
commit 8227762
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/package/android/src/main/cpp/MutableRawBuffer.h b/package/android/src/main/cpp/MutableRawBuffer.h
@@ -18,12 +18,17 @@ using namespace facebook;
 class MutableRawBuffer : public jsi::MutableBuffer {
 
 public:
-  explicit MutableRawBuffer(size_t size) {
+  explicit MutableRawBuffer(uint8_t* data, size_t size, bool freeOnDealloc) {
     _size = size;
-    _data = new uint8_t[size];
+    _data = data;
+    _freeOnDealloc = freeOnDealloc;
   }
+  explicit MutableRawBuffer(size_t size): MutableRawBuffer(new uint8_t[size], size, true) { }
+
   ~MutableRawBuffer() {
-    delete[] _data;
+    if (_freeOnDealloc) {
+      delete[] _data;
+    }
   }
 
 public:
@@ -37,6 +42,7 @@ class MutableRawBuffer : public jsi::MutableBuffer {
 private:
   uint8_t* _data;
   size_t _size;
+  bool _freeOnDealloc;
 };
 
 } // namespace vision
diff --git a/package/ios/FrameProcessors/MutableRawBuffer.h b/package/ios/FrameProcessors/MutableRawBuffer.h
@@ -18,12 +18,17 @@ using namespace facebook;
 class MutableRawBuffer : public jsi::MutableBuffer {
 
 public:
-  explicit MutableRawBuffer(size_t size) {
+  explicit MutableRawBuffer(uint8_t* data, size_t size, bool freeOnDealloc) {
     _size = size;
-    _data = new uint8_t[size];
+    _data = data;
+    _freeOnDealloc = freeOnDealloc;
   }
+  explicit MutableRawBuffer(size_t size): MutableRawBuffer(new uint8_t[size], size, true) { }
+
   ~MutableRawBuffer() {
-    delete[] _data;
+    if (_freeOnDealloc) {
+      delete[] _data;
+    }
   }
 
 public:
@@ -37,6 +42,7 @@ class MutableRawBuffer : public jsi::MutableBuffer {
 private:
   uint8_t* _data;
   size_t _size;
+  bool _freeOnDealloc;
 };
 
 } // namespace vision