Luigis-Mansion-Blender-Toolkit/gx_texture.py at 3c596b1f64d7d506ed37e131cec6f95ec8df3766 · Astral-C/Luigis-Mansion-Blender-Toolkit · GitHub

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
import bpy
import struct
from io import BytesIO
from wwlib_texture_utils import *

class gx_texture():

    @staticmethod
    def cfrom_rgb565(color):
        r = ((color >> 8) & 0xF8) | ((color >> 11) & 0x7)
        g = ((color >> 3) & 0xFC) | ((color >> 5) & 0x3)
        b = ((color << 3) & 0xF8) | (color & 0x7)
        return (r, g, b, 0xFF)

    @staticmethod
    def cfrom_rgb5A3(color):
        r = 0x00
        g = 0x00
        b = 0x00
        a = 0x00
        if(color & 0x8000 == 0x8000):
            a = 0xFF
            r = (color >> 0x7C00) >> 10
            g = (color & 0x3E0) >> 5
            b = (color & 0x1F)

            r = r << 3 | r >> 2
            g = g << 3 | g >> 2
            b = b << 3 | b >> 2

        else:
            a = (color & 0x7000) >> 12
            r = (color & 0xF00) >> 8
            g = (color & 0xF0) >> 4
            b = (color & 0xF)

            a = a << 5 | a << 2 | a >> 1
            r = r << 4 | r
            g = g << 4 | g
            b = b << 4 | b

        return (r, g, b, a)

    @staticmethod
    def bi_from_rgb5A3(stream, w, h, offset, tex_index):
        stream.seek(offset)
        img = bpy.data.images.new("IMG{}".format(tex_index), width=w, height=h, alpha=True, float_buffer=False)

        pixels = [None] * w * h
        for ty in range(0, h, 4):
            for tx in range(0, w, 4):
                for by in range(4):
                    for bx in range(4):
                        color = gx_texture.cfrom_rgb5A3(stream.readUInt16())
                        pixels[(ty + by) * w + (tx + bx)] = color

        img.pixels = [chan/255 for px in pixels for chan in px]
        return img

    @staticmethod
    def bi_from_rgb565(stream, w, h, offset, tex_index):
        stream.seek(offset)
        img = bpy.data.images.new("IMG{}".format(tex_index), width=w, height=h, alpha=True, float_buffer=False)

        pixels = [None] * w * h
        for ty in range(0, h, 4):
            for tx in range(0, w, 4):
                for by in range(4):
                    for bx in range(4):
                        color = gx_texture.cfrom_rgb565(stream.readUInt16())
                        pixels[(ty + by) * w + (tx + bx)] = color

        img.pixels = [chan/255 for px in pixels for chan in px]
        return img

    @staticmethod
    def bi_from_i4(stream, w, h, offset, tex_index):
        stream.seek(offset)
        img = bpy.data.images.new("IMG{}".format(tex_index), width=w, height=h, alpha=False, float_buffer=False)

        pixels = [None] * w * h

        for ty in range(0, h, 8):
            for tx in range(0, w, 8):

                block_pixels = decode_i4_block(ImageFormat.I4, stream, 0, BLOCK_DATA_SIZES[ImageFormat.I4], [])

                for by in range(0, 8, 4):
                    for bx in range(0, 8, 4):
                        for y in range(4):
                            for x in range(4):
                                if((ty + by + y) < h and (tx + bx + (3-x)) < w):
                                    pxl = block_pixels[(by + y) * BLOCK_WIDTHS[ImageFormat.CMPR] + (bx + (3-x))]
                                    pixels[(ty + by + y) * w + (tx + bx + (3-x))] = (pxl[0],pxl[1],pxl[2],255)

        img.pixels = [chan / 255 for px in pixels for chan in px]

        return img

    @staticmethod
    def bi_from_cmpr(stream, w, h, offset, tex_index):
        stream.seek(offset)
        img = bpy.data.images.new("IMG{}".format(tex_index), width=w, height=h, alpha=True, float_buffer=False)
        pixels = [None] * w * h

        for ty in range(0, h, 8):
            for tx in range(0, w, 8):

                block_pixels = decode_cmpr_block(ImageFormat.CMPR, stream, 0, BLOCK_DATA_SIZES[ImageFormat.CMPR], [], None)

                for by in range(0, 8, 4):
                    for bx in range(0, 8, 4):
                        for y in range(4):
                            for x in range(4):
                                if((ty + by + y) < h and (tx + bx + (3-x)) < w):
                                    pixels[(ty + by + y) * w + (tx + bx + (3-x))] = block_pixels[(by + y) * BLOCK_WIDTHS[ImageFormat.CMPR] + (bx + (3-x))]

        img.pixels = [chan / 255 for px in pixels for chan in px]
        return img