forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathTexture2DConverter.py
More file actions
454 lines (392 loc) · 15.3 KB
/
Copy pathTexture2DConverter.py
File metadata and controls
454 lines (392 loc) · 15.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
from __future__ import annotations
import struct
from copy import copy
from io import BytesIO
from typing import TYPE_CHECKING, Dict, Tuple, Union
import astc_encoder
import texture2ddecoder
from PIL import Image
from ..enums import BuildTarget, TextureFormat
from ..helpers import TextureSwizzler
if TYPE_CHECKING:
from ..classes import Texture2D
TF = TextureFormat
def image_to_texture2d(
img: Image.Image, target_texture_format: Union[TF, int], flip: bool = True
) -> Tuple[bytes, TextureFormat]:
if isinstance(target_texture_format, int):
target_texture_format = TextureFormat(target_texture_format)
import etcpak
if flip:
img = img.transpose(Image.FLIP_TOP_BOTTOM)
# DXT
if target_texture_format in [TF.DXT1, TF.DXT1Crunched]:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_bc1(raw_img, img.width, img.height)
tex_format = TF.DXT1
elif target_texture_format in [TF.DXT5, TF.DXT5Crunched]:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_bc3(raw_img, img.width, img.height)
tex_format = TF.DXT5
elif target_texture_format in [TF.BC4]:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_bc4(raw_img, img.width, img.height)
tex_format = TF.BC4
elif target_texture_format in [TF.BC5]:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_bc5(raw_img, img.width, img.height)
tex_format = TF.BC5
elif target_texture_format in [TF.BC7]:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_bc7(raw_img, img.width, img.height)
tex_format = TF.BC7
# ETC
elif target_texture_format in [TF.ETC_RGB4, TF.ETC_RGB4Crunched, TF.ETC_RGB4_3DS]:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_etc1_rgb(raw_img, img.width, img.height)
tex_format = TF.ETC_RGB4
elif target_texture_format == TF.ETC2_RGB:
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_etc2_rgb(raw_img, img.width, img.height)
tex_format = TF.ETC2_RGB
elif (
target_texture_format in [TF.ETC2_RGBA8, TF.ETC2_RGBA8Crunched, TF.ETC2_RGBA1]
or "_RGB_" in target_texture_format.name
):
raw_img = img.tobytes("raw", "RGBA")
enc_img = etcpak.compress_etc2_rgba(raw_img, img.width, img.height)
tex_format = TF.ETC2_RGBA8
# ASTC
elif target_texture_format.name.startswith("ASTC"):
raw_img = img.tobytes("raw", "RGBA")
block_size = tuple(
map(int, target_texture_format.name.rsplit("_", 1)[1].split("x"))
)
config = astc_encoder.ASTCConfig(
astc_encoder.ASTCProfile.LDR, *block_size, 1, 100
)
context = astc_encoder.ASTCContext(config)
raw_img = astc_encoder.ASTCImage(
astc_encoder.ASTCType.U8, img.width, img.height, 1, raw_img
)
if img.mode == "RGB":
tex_format = getattr(TF, f"ASTC_RGB_{block_size[0]}x{block_size[1]}")
else:
tex_format = getattr(TF, f"ASTC_RGBA_{block_size[0]}x{block_size[1]}")
swizzle = astc_encoder.ASTCSwizzle.from_str("RGBA")
enc_img = context.compress(raw_img, swizzle)
tex_format = target_texture_format
# A
elif target_texture_format == TF.Alpha8:
enc_img = img.tobytes("raw", "A")
tex_format = TF.Alpha8
# R - should probably be moerged into #A, as pure R is used as Alpha
# but need test data for this first
elif target_texture_format in [
TF.R8,
TF.R16,
TF.RHalf,
TF.RFloat,
TF.EAC_R,
TF.EAC_R_SIGNED,
]:
enc_img = img.tobytes("raw", "R")
tex_format = TF.R8
# RGBA
elif target_texture_format in [
TF.RGB565,
TF.RGB24,
TF.RGB9e5Float,
TF.PVRTC_RGB2,
TF.PVRTC_RGB4,
TF.ATC_RGB4,
]:
enc_img = img.tobytes("raw", "RGB")
tex_format = TF.RGB24
# everything else defaulted to RGBA
else:
enc_img = img.tobytes("raw", "RGBA")
tex_format = TF.RGBA32
return enc_img, tex_format
def assert_rgba(img: Image.Image, target_texture_format: TextureFormat) -> Image.Image:
if img.mode == "RGB":
img = img.convert("RGBA")
assert (
img.mode == "RGBA"
), f"{target_texture_format} compression only supports RGB & RGBA images" # noqa: E501
return img
def get_image_from_texture2d(
texture_2d: Texture2D,
flip: bool = True,
) -> Image.Image:
"""converts the given texture into PIL.Image
:param texture_2d: texture to be converterd
:type texture_2d: Texture2D
:param flip: flips the image back to the original (all Unity textures are flipped by default)
:type flip: bool
:return: PIL.Image object
:rtype: Image
"""
return parse_image_data(
texture_2d.get_image_data(),
texture_2d.m_Width,
texture_2d.m_Height,
texture_2d.m_TextureFormat,
texture_2d.object_reader.version,
texture_2d.object_reader.platform,
getattr(texture_2d, "m_PlatformBlob", None),
flip,
)
def parse_image_data(
image_data: bytes,
width: int,
height: int,
texture_format: Union[int, TextureFormat],
version: tuple,
platform: int,
platform_blob: bytes = None,
flip: bool = True,
) -> Image.Image:
image_data = copy(bytes(image_data))
if not image_data:
raise ValueError("Texture2D has no image data")
selection = CONV_TABLE[texture_format]
if len(selection) == 0:
raise NotImplementedError(
f"Not implemented texture format: {texture_format.name}"
)
if platform == BuildTarget.XBOX360 and texture_format in XBOX_SWAP_FORMATS:
image_data = swap_bytes_for_xbox(image_data)
elif platform == BuildTarget.Switch and platform_blob is not None:
gobsPerBlock = TextureSwizzler.get_switch_gobs_per_block(platform_blob)
block_size = TextureSwizzler.TEXTUREFORMAT_BLOCK_SIZE_MAP[texture_format]
padded_size = TextureSwizzler.get_padded_texture_size(
width, height, *block_size, gobsPerBlock
)
image_data = TextureSwizzler.deswizzle(
image_data, *padded_size, *block_size, gobsPerBlock
)
if not isinstance(texture_format, TextureFormat):
texture_format = TextureFormat(texture_format)
if "Crunched" in texture_format.name:
version = version
if (
version[0] > 2017
or (version[0] == 2017 and version[1] >= 3) # 2017.3 and up
or texture_format == TF.ETC_RGB4Crunched
or texture_format == TF.ETC2_RGBA8Crunched
):
image_data = texture2ddecoder.unpack_unity_crunch(image_data)
else:
image_data = texture2ddecoder.unpack_crunch(image_data)
img = selection[0](image_data, width, height, *selection[1:])
if img and flip:
return img.transpose(Image.FLIP_TOP_BOTTOM)
return img
def swap_bytes_for_xbox(image_data: bytes) -> bytes:
"""swaps the texture bytes
This is required for textures deployed on XBOX360.
"""
for i in range(0, len(image_data), 2):
image_data[i : i + 2] = image_data[i : i + 2][::-1]
return image_data
def pillow(
image_data: bytes,
width: int,
height: int,
mode: str,
codec: str,
args,
swap: tuple = None,
) -> Image.Image:
img = (
Image.frombytes(mode, (width, height), image_data, codec, args)
if width
else Image.new(mode, (width, height))
)
if swap:
channels = img.split()
img = Image.merge(mode, [channels[x] for x in swap])
return img
def atc(image_data: bytes, width: int, height: int, alpha: bool) -> Image.Image:
if alpha:
image_data = texture2ddecoder.decode_atc_rgba8(image_data, width, height)
else:
image_data = texture2ddecoder.decode_atc_rgb4(image_data, width, height)
return Image.frombytes("RGBA", (width, height), image_data, "raw", "BGRA")
ASTC_CONTEXTS: Dict[Tuple[int, int], astc_encoder.ASTCContext] = {}
def astc(image_data: bytes, width: int, height: int, block_size: tuple) -> Image.Image:
context = ASTC_CONTEXTS.get(block_size)
if context is None:
config = astc_encoder.ASTCConfig(
astc_encoder.ASTCProfile.LDR,
*block_size,
1,
100,
astc_encoder.ASTCConfigFlags.USE_DECODE_UNORM8,
)
context = ASTC_CONTEXTS[block_size] = astc_encoder.ASTCContext(config)
image = astc_encoder.ASTCImage(astc_encoder.ASTCType.U8, width, height, 1)
context.decompress(image_data, image, astc_encoder.ASTCSwizzle.from_str("RGBA"))
return Image.frombytes("RGBA", (width, height), image.data, "raw", "RGBA")
def pvrtc(image_data: bytes, width: int, height: int, fmt: bool) -> Image.Image:
image_data = texture2ddecoder.decode_pvrtc(image_data, width, height, fmt)
return Image.frombytes("RGBA", (width, height), image_data, "raw", "BGRA")
def etc(image_data: bytes, width: int, height: int, fmt: list) -> Image.Image:
if fmt[0] == 1:
image_data = texture2ddecoder.decode_etc1(image_data, width, height)
elif fmt[0] == 2:
if fmt[1] == "RGB":
image_data = texture2ddecoder.decode_etc2(image_data, width, height)
elif fmt[1] == "A1":
image_data = texture2ddecoder.decode_etc2a1(image_data, width, height)
elif fmt[1] == "A8":
image_data = texture2ddecoder.decode_etc2a8(image_data, width, height)
else:
raise NotImplementedError("unknown etc mode")
return Image.frombytes("RGBA", (width, height), image_data, "raw", "BGRA")
def eac(image_data: bytes, width: int, height: int, fmt: list) -> Image.Image:
if fmt == "EAC_R":
image_data = texture2ddecoder.decode_eacr(image_data, width, height)
elif fmt == "EAC_R_SIGNED":
image_data = texture2ddecoder.decode_eacr_signed(image_data, width, height)
elif fmt == "EAC_RG":
image_data = texture2ddecoder.decode_eacrg(image_data, width, height)
elif fmt == "EAC_RG_SIGNED":
image_data = texture2ddecoder.decode_eacrg_signed(image_data, width, height)
return Image.frombytes("RGBA", (width, height), image_data, "raw", "BGRA")
def half(
image_data: bytes,
width: int,
height: int,
mode: str,
codec: str,
args,
swap: tuple = None,
) -> Image.Image:
# convert half-float to int8
stream = BytesIO(image_data)
image_data = bytes(
int(struct.unpack("e", stream.read(2))[0] * 256)
for _ in range(width * height * len(codec))
)
return pillow(image_data, width, height, mode, codec, args, swap)
RG_PADDING_MAP = {
"RGE": 16,
"RGF": 32,
"RG;16": 16,
"RG;16s": 16,
"RG;8s": 8,
}
def rg(
image_data: bytes, width: int, height: int, mode: str, codec: str, args
) -> Image.Image:
# convert rg to rgb by adding in zeroes
padding_size = RG_PADDING_MAP[codec]
stream = BytesIO(image_data)
padding = bytes(padding_size)
rgb_data = b"".join(
stream.read(padding_size * 2) + padding
for _ in range(image_data / (2 * padding_size))
)
if codec == "RGE":
return half(rgb_data, width, height, mode, "RGB", args)
else:
return pillow(rgb_data, width, height, mode, codec.replace("RG", "RGB"), args)
def rgb9e5float(image_data: bytes, width: int, height: int) -> Image.Image:
rgb = bytearray(width * height * 3)
for i, (n,) in enumerate(struct.iter_unpack("<i", image_data)):
scale = n >> 27 & 0x1F
scalef = 2 ** (scale - 24)
scaleb = scalef * 255.0
b = (n >> 18 & 0x1FF) * scaleb
g = (n >> 9 & 0x1FF) * scaleb
r = (n & 0x1FF) * scaleb
offset = i * 3
rgb[offset : offset + 3] = [r, g, b]
return Image.frombytes("RGB", (width, height), rgb, "raw", "RGB")
CONV_TABLE = {
# FORMAT FUNC #ARGS.....
# ----------------------- -------- -------- ------------ ----------------- ------------ ----------
(TF.Alpha8, pillow, "RGBA", "raw", "A"),
(TF.ARGB4444, pillow, "RGBA", "raw", "RGBA;4B", (2, 1, 0, 3)),
(TF.RGB24, pillow, "RGB", "raw", "RGB"),
(TF.RGBA32, pillow, "RGBA", "raw", "RGBA"),
(TF.ARGB32, pillow, "RGBA", "raw", "ARGB"),
(TF.ARGBFloat, pillow, "RGBA", "raw", "RGBAF", (2, 1, 0, 3)),
(TF.RGB565, pillow, "RGB", "raw", "BGR;16"),
(TF.BGR24, pillow, "RGB", "raw", "BGR"),
(TF.R8, pillow, "RGB", "raw", "R"),
(TF.R16, pillow, "RGB", "raw", "R;16"),
(TF.RG16, rg, "RGB", "raw", "RG"),
(TF.DXT1, pillow, "RGBA", "bcn", 1),
(TF.DXT3, pillow, "RGBA", "bcn", 2),
(TF.DXT5, pillow, "RGBA", "bcn", 3),
(TF.RGBA4444, pillow, "RGBA", "raw", "RGBA;4B", (3, 2, 1, 0)),
(TF.BGRA32, pillow, "RGBA", "raw", "BGRA"),
(TF.RHalf, half, "R", "raw", "R"),
(TF.RGHalf, rg, "RGB", "raw", "RGE"),
(TF.RGBAHalf, half, "RGB", "raw", "RGB"),
(TF.RFloat, pillow, "RGB", "raw", "RF"),
(TF.RGFloat, rg, "RGB", "raw", "RGF"),
(TF.RGBAFloat, pillow, "RGBA", "raw", "RGBAF"),
(TF.YUY2,),
(TF.RGB9e5Float, rgb9e5float),
(TF.BC4, pillow, "L", "bcn", 4),
(TF.BC5, pillow, "RGB", "bcn", 5),
(TF.BC6H, pillow, "RGBA", "bcn", 6),
(TF.BC7, pillow, "RGBA", "bcn", 7),
(TF.DXT1Crunched, pillow, "RGBA", "bcn", 1),
(TF.DXT5Crunched, pillow, "RGBA", "bcn", 3),
(TF.PVRTC_RGB2, pvrtc, True),
(TF.PVRTC_RGBA2, pvrtc, True),
(TF.PVRTC_RGB4, pvrtc, False),
(TF.PVRTC_RGBA4, pvrtc, False),
(TF.ETC_RGB4, etc, (1,)),
(TF.ATC_RGB4, atc, False),
(TF.ATC_RGBA8, atc, True),
(TF.EAC_R, eac, "EAC_R"),
(TF.EAC_R_SIGNED, eac, "EAC_R:SIGNED"),
(TF.EAC_RG, eac, "EAC_RG"),
(TF.EAC_RG_SIGNED, eac, "EAC_RG_SIGNED"),
(TF.ETC2_RGB, etc, (2, "RGB")),
(TF.ETC2_RGBA1, etc, (2, "A1")),
(TF.ETC2_RGBA8, etc, (2, "A8")),
(TF.ASTC_RGB_4x4, astc, (4, 4)),
(TF.ASTC_RGB_5x5, astc, (5, 5)),
(TF.ASTC_RGB_6x6, astc, (6, 6)),
(TF.ASTC_RGB_8x8, astc, (8, 8)),
(TF.ASTC_RGB_10x10, astc, (10, 10)),
(TF.ASTC_RGB_12x12, astc, (12, 12)),
(TF.ASTC_RGBA_4x4, astc, (4, 4)),
(TF.ASTC_RGBA_5x5, astc, (5, 5)),
(TF.ASTC_RGBA_6x6, astc, (6, 6)),
(TF.ASTC_RGBA_8x8, astc, (8, 8)),
(TF.ASTC_RGBA_10x10, astc, (10, 10)),
(TF.ASTC_RGBA_12x12, astc, (12, 12)),
(TF.ETC_RGB4_3DS, etc, (1,)),
(TF.ETC_RGBA8_3DS, etc, (1,)),
(TF.ETC_RGB4Crunched, etc, (1,)),
(TF.ETC2_RGBA8Crunched, etc, (2, "A8")),
(TF.ASTC_HDR_4x4, astc, (4, 4)),
(TF.ASTC_HDR_5x5, astc, (5, 5)),
(TF.ASTC_HDR_6x6, astc, (6, 6)),
(TF.ASTC_HDR_8x8, astc, (8, 8)),
(TF.ASTC_HDR_10x10, astc, (10, 10)),
(TF.ASTC_HDR_12x12, astc, (12, 12)),
(TF.RG32, rg, "RGB", "raw", "RG;16"),
(TF.RGB48, pillow, "RGB", "raw", "RGB;16"),
(TF.RGBA64, pillow, "RGBA", "raw", "RGBA;16"),
(TF.R8_SIGNED, pillow, "R", "raw", "R;8s"),
(TF.RG16_SIGNED, rg, "RGB", "raw", "RG;8s"),
(TF.RGB24_SIGNED, pillow, "RGB", "raw", "RGB;8s"),
(TF.RGBA32_SIGNED, pillow, "RGBA", "raw", "RGBA;8s"),
(TF.R16_SIGNED, pillow, "R", "raw", "R;16s"),
(TF.RG32_SIGNED, rg, "RGB", "raw", "RG;16s"),
(TF.RGB48_SIGNED, pillow, "RGB", "raw", "RGB;16s"),
(TF.RGBA64_SIGNED, pillow, "RGBA", "raw", "RGBA;16s"),
}
# format conv_table to a dict
CONV_TABLE = {line[0]: line[1:] for line in CONV_TABLE}
# XBOX Swap Formats
XBOX_SWAP_FORMATS = [TF.RGB565, TF.DXT1, TF.DXT1Crunched, TF.DXT5, TF.DXT5Crunched]