DataParse.py

import io
import struct
import numpy as np

def VertexParse(dataBytes, datatype, offset, stride):
    dataOut = []
    i = offset
    while i < len(dataBytes):
        if datatype == "_vertex_format_attribute_short4":
            dataOut.append(np.array(struct.unpack('<hhhh', dataBytes[i:i+8])))
        elif datatype == "_vertex_format_attribute_float4":
            dataOut.append(np.array(struct.unpack('<ffff', dataBytes[i:i+16])))
        else:
            print("Unknown type, please file an issue for this item...")
            return
        i += stride
            
    return dataOut
    
class DataParse:
    def __init__(self, byteData):
        self.data = io.BytesIO(byteData)
        return
    
    def seek(self, address):
        self.data.seek(address, 0)
        return
    
    def read(self, length):
        return self.data.read(length)
        
    def readString(self, length):
        return self.data.read(length).decode('utf-8').rstrip('\0')
        
    def readUTF(self):
        length = int.from_bytes(self.data.read(2), byteorder='little')
        return self.data.read(length).decode('utf-8')
    
    def readInt8(self):
        return struct.unpack('<b', self.data.read(1))[0]
    
    def readInt16(self):
        return struct.unpack('<h', self.data.read(2))[0]
    
    def readInt32(self):
        return struct.unpack('<l', self.data.read(4))[0]
    
    def readInt64(self):
        return struct.unpack('<q', self.data.read(8))[0]
    
    def readFloat(self):
        return struct.unpack('f', self.data.read(4))[0]
    
    def readVector2D(self):
        return np.array([self.readFloat(), self.readFloat()], dtype='float')
    
    def readVector3D(self):
        return np.array([self.readFloat(), self.readFloat(), self.readFloat()], dtype='float')
    
    def readVector4D(self):
        return np.array([self.readFloat(), self.readFloat(), self.readFloat(), self.readFloat()], dtype='float')