From 32032e8333c00dbd437c78700af3d39b54c73678 Mon Sep 17 00:00:00 2001
From: Stewie <fortiersteven1@gmail.com>
Date: Tue, 30 Aug 2022 19:36:46 -0400
Subject: [PATCH] Function to grab TSS from file for Event

---
 ToolsTOPX.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/ToolsTOPX.py b/ToolsTOPX.py
index 30becc5..d990b8e 100644
--- a/ToolsTOPX.py
+++ b/ToolsTOPX.py
@@ -197,6 +197,23 @@ class ToolsTOPX(ToolsTales):
         self.extract_tss_XML(tss, event_file, '../Data/{}/Events'.format(self.repo_name))
         return tss
         
+    def get_tss_from_event(self, event_file):
+        
+        with open(event_file, "rb") as event_f:
+            data = event_f.read()
+            file_offset = struct.unpack("<I", data[0x6C:0x70])[0]
+            tss_data = data[file_offset:len(data)]
+            dirname = os.path.dirname(event_file)
+            self.mkdir( os.path.join(dirname, "TSS"))
+            file_tss = os.path.join(dirname, "TSS", os.path.basename(event_file).replace(".dat",".tss"))
+            with open(file_tss, "wb") as f:
+                f.write(tss_data)
+            
+            return io.BytesIO(tss_data), file_tss
+            
+            
+            
+            
         
         root = etree.Element('SceneText')