import platform,sys,redis,time,requests,json,atexit
import os, sys, time, argparse, requests, json, re, oss2
import bpy
import open3d as o3d
import shutil
import psutil
import trimesh
from PIL import Image
from pyvirtualdisplay import Display
# os.environ['DISPLAY'] = ':1'
AccessKeyId = 'LTAI5tSReWm8hz7dSYxxth8f'
AccessKeySecret = '8ywTDF9upPAtvgXtLKALY2iMYHIxdS'
Endpoint = 'oss-cn-shanghai.aliyuncs.com'
Bucket = 'suwa3d-website'
oss_client = oss2.Bucket(oss2.Auth(AccessKeyId, AccessKeySecret), Endpoint, Bucket)

# 创建虚拟显示
display = Display(visible=0, size=(500, 500))
display.start()

# pid = sys.argv[1]
# target_folder = '/data/datasets/complate/preview'
# source_folder = f'/data/datasets/complate/objs/{pid}'

# 定义函数来递归遍历文件夹下的所有文件
def list_files(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            yield os.path.join(root, file)


def trim_whitespace(image_path, output_path):
    # 打开图片
    image = Image.open(image_path)

    # 获取图片的像素数据
    data = image.getdata()

    # 检测图片边界
    left = image.width
    right = 0
    top = image.height
    bottom = 0
    for x in range(image.width):
        for y in range(image.height):
            # 如果像素不是白色，则更新边界坐标
            if data[x + y * image.width] != (255, 255, 255, 255):
                left = min(left, x)
                right = max(right, x)
                top = min(top, y)
                bottom = max(bottom, y)

    # 裁剪图片
    image = image.crop((left, top, right + 1, bottom + 1))

    # 保存裁剪后的图片
    image.save(output_path)

def remove_background(image_path, output_path):
    # 打开图片
    image = Image.open(image_path)

    # 将图片转换为带有透明通道的 RGBA 模式
    image = image.convert("RGBA")

    # 获取图片的像素数据
    data = image.getdata()

    # 新的像素数据列表
    new_data = []

    # 将白色背景的像素转换为透明
    for item in data:
        # 如果像素为白色，则设置为完全透明
        if item[:3] == (255, 255, 255):
            new_data.append((255, 255, 255, 0))
        else:
            new_data.append(item)

    # 更新图片的像素数据
    image.putdata(new_data)

    # 保存图片
    image.save(output_path, "PNG")

#生成预览图
def createImage(pid):


    target_folder = '/data/datasets/complate/preview'
    source_folder = f'/data/datasets/complate/objs/{pid}'

    for file_path in list_files(source_folder):
        if ".obj" not in file_path:
            continue
        #print("文件路径：",file_path)
        #判断文件是否存在
        if not os.path.exists(file_path):
            continue



        # 加载OBJ文件
        mesh = trimesh.load_mesh(file_path)

        # 生成封面图
        image = mesh.scene().save_image(resolution=[400, 300])
        # 转换为Pillow的Image对象
        # pil_image = Image.frombytes('RGB', (image.width, image.height), image.data)
        # # 保存封面图
        # pil_image.save(f'{target_folder}/{pid}_preview.png')
        #保存封面图
        with open(f'{target_folder}/{pid}_preview.png', 'wb') as f:
            f.write(image)
        
        time.sleep(2)
        print("处理封面图片.....")
        remove_background(f'{target_folder}/{pid}_preview.png',f'{target_folder}/{pid}_preview.png')
        print("封面图片处理完成.....")
        time.sleep(1)

        oss_client.put_object_from_file(f'print_ticket_view/{pid}_preview.png', f'{target_folder}/{pid}_preview.png')

        break
    # 关闭虚拟显示
    #display.stop()
# if __name__ == "__main__":
#     AccessKeyId = 'LTAI5tSReWm8hz7dSYxxth8f'
#     AccessKeySecret = '8ywTDF9upPAtvgXtLKALY2iMYHIxdS'
#     Endpoint = 'oss-cn-shanghai.aliyuncs.com'
#     Bucket = 'suwa3d-website'
#     oss_client = oss2.Bucket(oss2.Auth(AccessKeyId, AccessKeySecret), Endpoint, Bucket)
#     pid = sys.argv[1]
#     target_folder = '/data/datasets/complate/preview'
#     source_folder = f'/data/datasets/complate/objs/{pid}'
#     createImage()