# -*- coding:utf-8 -*-
# title           :Body工具类
# description     :Body工具类
# author          :Python超人
# date            :2023-03-26
# link            :https://gitcode.net/pythoncr/
# python_version  :3.8
# ==============================================================================


def gen_bodies_from_image(pixel_image, params, body_template=None):
    """
    根据像素图片以及参数，自动生成星球，注意图片像素不能太多，否则会导致电脑运行太慢
    @param pixel_image:
    @param params:
    @return:
    """
    from PIL import Image

    img = Image.open(pixel_image).convert('RGBA')
    width, height = img.size
    interval_factor = 20  # 星球间距因子
    if body_template is None:
        body_template = 'Body(name="%s", mass=mass, color=(%d, %d, %d), size_scale=%.4f, ' \
                        'init_position=get_position([0, %g * D, %g * D], %.4f), ' \
                        'init_velocity=[0, 0, 0], ignore_mass=True)'
    bodies_str = "["

    # 以图片像素为坐标，对角线的距离
    distance_hw = pow(pow(width, 2) + pow(height, 2), 1 / 2)

    for h in range(0, height):
        # row = []
        for w in range(0, width):
            # 以图片像素为坐标，每个像素点到中心的距离
            distance_to_center = pow(pow(w - width / 2, 2) + pow(h - height / 2, 2), 1 / 2)
            # 让 body 从中心开始，离摄像机越远， body 的缩放值越大（scale 就越大，）
            scale = (distance_to_center / (distance_hw * 10) + 1)  # 中心最近 1.0 ~ 1.05
            # TODO: 队列反向排列（中心最远 1.05 ~ 1.0）
            # scale = 1.05 - scale + 1.0
            # print(scale)
            # 获取像素的颜色
            pixel = img.getpixel((w, h))
            # 对于纯白色的颜色，就忽略，不生成星球（这样图片中，纯白色越多，对电脑的压力就越小）
            if pixel[0] >= 255 and pixel[1] >= 255 and pixel[1] >= 255:
                continue
            body_str = body_template % (f"星球{h}:{w}", pixel[0], pixel[1], pixel[2], scale,
                                        (width - w) * interval_factor, (height - h) * interval_factor, scale)
            bodies_str += body_str + ",\n"

    bodies_str += "]"
    return eval(bodies_str, params)


if __name__ == '__main__':
    import random
    from bodies import Body

    D = 600
    mass = 0.9e25
    # camera_pos = 左-右+、上+下-、前+后-
    camera_pos = (-100 * D, 0, -5000 * D)


    def get_position(pos, scale):
        # [ 远+近-  , 左+右-  , 上+下-]
        return pos[0] + (scale - 1.0) * 300 * (random.randint(90, 110)) * D, pos[1], pos[2]
        # return pos[0], pos[1], pos[2]


    bodies: list = gen_bodies_from_image(pixel_image="../images/eye.png",
                                         params={"D": D,
                                                 "Body": Body,
                                                 "mass": mass,
                                                 "get_position": get_position,
                                                 "camera_pos": camera_pos})

    print(bodies)