@@ -137,7 +137,10 @@ def _get_texture_size(pixel_format: VtfPF, width, height):
137
137
return width * height // 2
138
138
elif pixel_format in (VtfPF .DXT3 , VtfPF .DXT5 ):
139
139
return width * height
140
- elif pixel_format in (VtfPF .A8 , VtfPF .I8 ,):
140
+ elif pixel_format in (
141
+ VtfPF .A8 ,
142
+ VtfPF .I8 ,
143
+ ):
141
144
return width * height
142
145
elif pixel_format in (VtfPF .UV88 , VtfPF .IA88 ):
143
146
return width * height * 2
@@ -205,7 +208,7 @@ def _write_image(fp: BufferedIOBase, im: Image.Image, pixel_format: VtfPF):
205
208
206
209
def _closest_power (x ):
207
210
possible_results = round (log (x , 2 )), ceil (log (x , 2 ))
208
- return 2 ** min (possible_results , key = lambda z : abs (x - 2 ** z ))
211
+ return 2 ** min (possible_results , key = lambda z : abs (x - 2 ** z ))
209
212
210
213
211
214
class VtfImageFile (ImageFile .ImageFile ):
@@ -246,8 +249,15 @@ def _open(self):
246
249
# flags = CompiledVtfFlags(header.flags)
247
250
pixel_format = VtfPF (header .pixel_format )
248
251
low_format = VtfPF (header .low_pixel_format )
249
- if pixel_format in (VtfPF .DXT1_ONEBITALPHA , VtfPF .DXT1 , VtfPF .DXT3 , VtfPF .DXT5 ,
250
- VtfPF .RGBA8888 , VtfPF .BGRA8888 ,VtfPF .A8 ):
252
+ if pixel_format in (
253
+ VtfPF .DXT1_ONEBITALPHA ,
254
+ VtfPF .DXT1 ,
255
+ VtfPF .DXT3 ,
256
+ VtfPF .DXT5 ,
257
+ VtfPF .RGBA8888 ,
258
+ VtfPF .BGRA8888 ,
259
+ VtfPF .A8 ,
260
+ ):
251
261
self .mode = "RGBA"
252
262
elif pixel_format in (VtfPF .RGB888 , VtfPF .BGR888 , VtfPF .UV88 ):
253
263
self .mode = "RGB"
@@ -309,9 +319,14 @@ def _save(im, fp, filename):
309
319
310
320
if pixel_format == VtfPF .DXT1_ONEBITALPHA :
311
321
flags |= CompiledVtfFlags .ONEBITALPHA
312
- elif pixel_format in (VtfPF .DXT3 , VtfPF .DXT5 ,
313
- VtfPF .RGBA8888 , VtfPF .BGRA8888 ,
314
- VtfPF .A8 , VtfPF .IA88 ):
322
+ elif pixel_format in (
323
+ VtfPF .DXT3 ,
324
+ VtfPF .DXT5 ,
325
+ VtfPF .RGBA8888 ,
326
+ VtfPF .BGRA8888 ,
327
+ VtfPF .A8 ,
328
+ VtfPF .IA88 ,
329
+ ):
315
330
flags |= CompiledVtfFlags .EIGHTBITALPHA
316
331
else :
317
332
pass
0 commit comments