GamesUnpack/atlas_unpack.py at master · Kin-L/GamesUnpack · 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
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
# -*- coding: utf-8 -*-
import os
from PIL import Image
from loguru import logger


def premultiply_alpha(image):
    '''
    若输出的贴图文件应用了预乘alpha
    :param image:
    :return:
    '''
    if image.mode != "RGBA":
        image = image.convert("RGBA")
    for i in range(image.width):
        for j in range(image.height):
            red, green, blue, alpha = image.getpixel((i, j))
            if alpha > 0:
                image.putpixel((i, j),
                               tuple([min(max(int(p / alpha * 255), 0), 255) for p in (red, green, blue)] + [alpha]))
    return image


def split_atlas(fileName, output_path: str = None, atlas_path: str = None): # type: ignore
    '''
    :param fileName: 文件名
    :param output_path: 输出目录
    :param atlas_path: 图集.atlas文件
    :return:
    '''
    logger.info(f"开始拆分图集: {atlas_path}")
    if output_path is None:
        output_path = os.path.join(os.getcwd(), fileName)
    if atlas_path is None:
        atlas_path = fileName + '.atlas'

    if not os.path.exists(atlas_path):
        logger.error(f"图集文件不存在: {atlas_path}")
        return

    with open(atlas_path, 'r', encoding='utf-8-sig') as atlas:
        os.makedirs(output_path, exist_ok=True)
        num = 0
        logger.debug(f"读取图集文件: {atlas_path}")
        while True:
            _line = atlas.readline()
            if not _line:
                num += 1
                if num > 5:
                    break
            elif ".png" in _line:
                png_name = _line.strip("\n")
                png_path = os.path.join(os.path.split(atlas_path)[0], png_name)
                if not os.path.exists(png_path):
                    logger.warning(f"未找到 PNG 文件: {png_path}")
                    png_path = os.path.join(os.path.split(atlas_path)[0], "Textures", png_name)
                    if not os.path.exists(png_path):
                        logger.error(f"仍然未找到 PNG 文件: {png_path}")
                        raise ValueError(f"找不到图像文件: {png_name}")

                ori_image = Image.open(png_path)

                big_image_size = atlas.readline()
                big_image_width, big_image_height = list(map(int, big_image_size.split(":")[1].split(",")))
                big_image = ori_image.resize((big_image_width, big_image_height))

                atlas.readline()  # Skip lines
                atlas.readline()
                atlas.readline()

                png_name = None
                rotate_angle = 0
                offset_x, offset_y = 0, 0
                while True:
                    line1 = atlas.readline()  # name
                    if len(line1) == 0:
                        break
                    elif len(line1.strip("\n")) == 0:
                        continue
                    elif line1.strip("\n").endswith('.png'):
                        png_name = line1.strip("\n")
                        png_path = os.path.join(os.path.split(atlas_path)[0], png_name)
                        ori_image = Image.open(png_path)
                        for i in range(4):
                            if i == 0:
                                big_image_size = atlas.readline()
                                big_image_width, big_image_height = list(map(int, big_image_size.split(":")[1].split(",")))
                                big_image = ori_image.resize((big_image_width, big_image_height))
                            else:
                                _line = atlas.readline()
                    else:
                        line1 = line1.strip("\n")
                        rotate = atlas.readline().split(":")[1].strip("\n")
                        xy = atlas.readline()  # xy
                        size = atlas.readline()  # size
                        orig = atlas.readline()  # orig
                        offset = atlas.readline().strip("\n")  # offset
                        index = atlas.readline()  # index
                        rotate = rotate.strip(" ")
                        if rotate == "true":
                            rotate_angle = 90
                        elif rotate == "false":
                            rotate_angle = 0
                        else:
                            rotate_angle = int(rotate)
                        rotate = rotate_angle > 0
                        name = line1.replace("\n", "") + ".png"
                        if '/' in name:
                            os.makedirs(os.path.join(output_path, '/'.join(name.split('/')[:-1])), exist_ok=True)
                        width, height = list(map(int, size.split(":")[1].split(",")))
                        ltx, lty = list(map(int, xy.split(":")[1].split(",")))
                        origx, origy = list(map(int, orig.split(":")[1].split(",")))
                        offset_x, offset_y = list(map(int, offset.split(":")[1].split(",")))
                        if rotate_angle == 0 or rotate_angle == 180:
                            rbx = ltx + width
                            rby = lty + height
                        elif rotate_angle == 90:
                            rbx = ltx + height
                            rby = lty + width
                        elif rotate_angle == 270:
                            rbx = ltx + height
                            rby = lty + width

                        result_image = Image.new("RGBA", (origx, origy))
                        rect_on_big = big_image.crop((ltx, lty, rbx, rby))
                        if rotate:
                            rect_on_big = rect_on_big.rotate(-rotate_angle, expand=True)

                        result_image.paste(rect_on_big, (offset_x, origy - height - offset_y))
                        result_image.save(output_path + '/' + name)
                        logger.info(f"保存图像: {output_path + '/' + name}")


# atlas图理文件解包
if __name__ == '__main__':
    # input_path = filedialog.askdirectory(title="选择需要批量解包的目录")
    # logger.info(f"选择的目录: {input_path}")

    # for root, dirs, files in os.walk(input_path):
    #     for filename in files:
    #         if filename.endswith(".atlas"):
    #             atlas_path = os.path.join(root, filename)
    #             unitName = os.path.split(filename)[1]
    #             outputPath = os.path.join(root, 'images')  # sub

    #             if not os.path.exists(outputPath):
    #                 os.makedirs(outputPath)
    #                 logger.info(f"创建输出目录: {outputPath}")

    #             try:
    #                 split_atlas(unitName, output_path=outputPath, atlas_path=atlas_path)
    #             except Exception as e:
    #                 logger.error(f"解包失败: {e}，跳过: {atlas_path}")
    logger.debug("程序已退出")