Check number of bytes remaining before reading texture strings
This commit is contained in:
parent
613cb20678
commit
f451be4d18
|
@ -138,6 +138,9 @@ class Reader:
|
|||
def how_much_left(self, pos):
|
||||
return self.end_pos - pos
|
||||
|
||||
def bytes_remaining(self):
|
||||
return self.end_pos - self.file.tell()
|
||||
|
||||
def skip_until(self, header):
|
||||
while (self.could_have_child() and header not in self.peak_next_header()):
|
||||
self.skip_bytes(1)
|
||||
|
|
|
@ -165,18 +165,22 @@ def _read_matd(matd: Reader) -> Material:
|
|||
|
||||
elif next_header == "TX0D":
|
||||
with matd.read_child() as tx0d:
|
||||
if tx0d.bytes_remaining() > 0:
|
||||
mat.texture0 = tx0d.read_string()
|
||||
|
||||
elif next_header == "TX1D":
|
||||
with matd.read_child() as tx1d:
|
||||
if tx1d.bytes_remaining() > 0:
|
||||
mat.texture1 = tx1d.read_string()
|
||||
|
||||
elif next_header == "TX2D":
|
||||
with matd.read_child() as tx2d:
|
||||
if tx2d.bytes_remaining() > 0:
|
||||
mat.texture2 = tx2d.read_string()
|
||||
|
||||
elif next_header == "TX3D":
|
||||
with matd.read_child() as tx3d:
|
||||
if tx3d.bytes_remaining() > 0:
|
||||
mat.texture3 = tx3d.read_string()
|
||||
|
||||
else:
|
||||
|
|
Loading…
Reference in New Issue