mbEditorPro2.0 release

mbEditorPro2.0/arcane/objects/common/SparseData.py

@@ -0,0 +1,108 @@
+#  • ▌ ▄ ·. ▄▄▄▄·     ▄▄▄ .·▄▄▄▄  ▪  ▄▄▄▄▄      ▄▄▄       ▄▄▄·▄▄▄       
+#  ·██ ▐███▪▐█ ▀█▪    ▀▄.▀·██▪ ██ ██ •██  ▪     ▀▄ █·    ▐█ ▄█▀▄ █·▪    
+#  ▐█ ▌▐▌▐█·▐█▀▀█▄    ▐▀▀▪▄▐█· ▐█▌▐█· ▐█.▪ ▄█▀▄ ▐▀▀▄      ██▀·▐▀▀▄  ▄█▀▄
+#  ██ ██▌▐█▌██▄▪▐█    ▐█▄▄▌██. ██ ▐█▌ ▐█▌·▐█▌.▐▌▐█•█▌    ▐█▪·•▐█•█▌▐█▌.▐▌
+#  ▀▀  █▪▀▀▀·▀▀▀▀      ▀▀▀ ▀▀▀▀▀• ▀▀▀ ▀▀▀  ▀█▄▀▪.▀  ▀    .▀   .▀  ▀ ▀█▄▀▪
+#                Magicbane Emulator Project © 2013 - 2022
+#                           www.magicbane.com
+
+from collections import OrderedDict
+
+from arcane.enums.arc_sparse import *
+from arcane.util import ResStream
+
+
+class SparseData:
+    def __init__(self) -> None:
+        self.sp_data = OrderedDict()
+
+    def load_binary(self, stream: ResStream):
+        sparse_tag = stream.read_dword()
+        while sparse_tag:
+            sparse_type = SPARSE_TAG_TO_SPARSE_TYPE[sparse_tag]
+            value = None
+
+            if sparse_type == SPARSE_VAL_LONG:
+                value = stream.read_dword()
+
+            elif sparse_type == SPARSE_VAL_FLOAT:
+                value = stream.read_float()
+
+            elif sparse_type == SPARSE_VAL_BOOL:
+                value = stream.read_bool()
+
+            elif sparse_type == SPARSE_UID:
+                value = stream.read_dword()
+
+            elif sparse_type == SPARSE_REF_VECTOR3:
+                value = stream.read_tuple()
+
+            elif sparse_type == SPARSE_REF_ARC_STRING:
+                value = stream.read_string()
+
+            elif sparse_type == SPARSE_REF_MERCHANT_DATA:
+                value = [
+                    stream.read_qword(),
+                    stream.read_qword(),
+                    stream.read_dword(),
+                ]
+
+            elif sparse_type == SPARSE_REF_ARC_CACHE_ID:
+                value = stream.read_qword()
+
+            elif sparse_type == SPARSE_PTR_ACTION_RESPONSE:
+                value = stream.read_dword()
+
+            elif sparse_type == SPARSE_ENUM_ITEM_TYPE:
+                value = stream.read_dword()
+
+            self.sp_data[sparse_tag] = value
+            sparse_tag = stream.read_dword()
+
+    def save_binary(self, stream: ResStream):
+        for tag, value in self.sp_data.items():
+            stream.write_dword(tag)
+
+            sparse_type = SPARSE_TAG_TO_SPARSE_TYPE[tag]
+            if sparse_type == SPARSE_VAL_LONG:
+                stream.write_dword(value)
+
+            elif sparse_type == SPARSE_VAL_FLOAT:
+                stream.write_float(value)
+
+            elif sparse_type == SPARSE_VAL_BOOL:
+                stream.write_bool(value)
+
+            elif sparse_type == SPARSE_UID:
+                stream.write_dword(value)
+
+            elif sparse_type == SPARSE_REF_VECTOR3:
+                stream.write_tuple(value)
+
+            elif sparse_type == SPARSE_REF_ARC_STRING:
+                stream.write_string(value)
+
+            elif sparse_type == SPARSE_REF_MERCHANT_DATA:
+                stream.write_qword(value[0])
+                stream.write_qword(value[1])
+                stream.write_qword(value[2])
+
+            elif sparse_type == SPARSE_REF_ARC_CACHE_ID:
+                stream.write_qword(value)
+
+            elif sparse_type == SPARSE_PTR_ACTION_RESPONSE:
+                stream.write_dword(value)
+
+            elif sparse_type == SPARSE_ENUM_ITEM_TYPE:
+                stream.write_dword(value)
+        stream.write_dword(0)
+
+    def load_json(self, data):
+        for key, value in data.items():
+            self.sp_data[STRING_TO_SPARSE_TAG[key]] = value
+
+    def save_json(self):
+        data = OrderedDict()
+        for tag, value in self.sp_data.items():
+            data[SPARSE_TAG_TO_STRING[tag]] = value
+        return data