From f2e28edc054dd2beaf63dc1d6a7d0d824de5ec7d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= <sybren@blender.org>
Date: Mon, 15 Jul 2024 14:27:11 +0200
Subject: [PATCH] Implement reader & writer for int8_t type

---
 blender_asset_tracer/blendfile/dna.py    |  1 +
 blender_asset_tracer/blendfile/dna_io.py | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/blender_asset_tracer/blendfile/dna.py b/blender_asset_tracer/blendfile/dna.py
index 12dff21..e4a536e 100644
--- a/blender_asset_tracer/blendfile/dna.py
+++ b/blender_asset_tracer/blendfile/dna.py
@@ -260,6 +260,7 @@ class Struct:
             b"short": endian.read_short,
             b"uint64_t": endian.read_ulong,
             b"float": endian.read_float,
+            b"int8_t": endian.read_int8,
         }
         try:
             simple_reader = simple_readers[dna_type.dna_type_id]
diff --git a/blender_asset_tracer/blendfile/dna_io.py b/blender_asset_tracer/blendfile/dna_io.py
index ac46794..918f2e4 100644
--- a/blender_asset_tracer/blendfile/dna_io.py
+++ b/blender_asset_tracer/blendfile/dna_io.py
@@ -28,6 +28,7 @@ import typing
 class EndianIO:
     # TODO(Sybren): note as UCHAR: struct.Struct = None and move actual structs to LittleEndianTypes
     UCHAR = struct.Struct(b"<B")
+    SINT8 = struct.Struct(b"<b")
     USHORT = struct.Struct(b"<H")
     USHORT2 = struct.Struct(b"<HH")  # two shorts in a row
     SSHORT = struct.Struct(b"<h")
@@ -62,6 +63,14 @@ class EndianIO:
     def write_char(cls, fileobj: typing.IO[bytes], value: int):
         return cls._write(fileobj, cls.UCHAR, value)
 
+    @classmethod
+    def read_int8(cls, fileobj: typing.IO[bytes]):
+        return cls._read(fileobj, cls.SINT8)
+
+    @classmethod
+    def write_int8(cls, fileobj: typing.IO[bytes], value: int):
+        return cls._write(fileobj, cls.SINT8, value)
+
     @classmethod
     def read_ushort(cls, fileobj: typing.IO[bytes]):
         return cls._read(fileobj, cls.USHORT)
@@ -207,6 +216,7 @@ class EndianIO:
         """
         return {
             b"char": cls.write_char,
+            b"int8": cls.write_int8,
             b"ushort": cls.write_ushort,
             b"short": cls.write_short,
             b"uint": cls.write_uint,
@@ -222,6 +232,7 @@ class LittleEndianTypes(EndianIO):
 
 class BigEndianTypes(LittleEndianTypes):
     UCHAR = struct.Struct(b">B")
+    SINT8 = struct.Struct(b">b")
     USHORT = struct.Struct(b">H")
     USHORT2 = struct.Struct(b">HH")  # two shorts in a row
     SSHORT = struct.Struct(b">h")