make2/libs/libs.py

import os, time, json, requests, shutil, oss2, psutil
from tqdm import tqdm
from PIL import Image, ImageEnhance
import config,libs_db,common
import threading
from concurrent.futures import ThreadPoolExecutor

def find_blender_bin_path():
    base_path = 'C:\\Program Files\\Blender Foundation\\'
    if os.path.exists(base_path):
        for dir in os.listdir(base_path):
            if dir.startswith('Blender'):
                blender_bin_path = base_path + dir + '\\blender.exe'
                return f'"{blender_bin_path}"'
    else:
        print('未找到blender安装目录')
        exit(1)

def resize_photos(photo_path, ratio=0.5):
    for filename in os.listdir(photo_path):
        if filename.endswith('.jpg'):
            img = Image.open(os.path.join(photo_path, filename))
            img = rotate_image(img)
            w, h = img.size
            img = img.resize((int(w * ratio), int(h * ratio)))
            img.save(os.path.join(photo_path, filename))
            
def rotate_image(image):
    # 检查图像的EXIF数据是否包含方向信息
    try:
        exif = image._getexif()
        orientation = exif.get(0x0112)
    except:
        orientation = None

    # 根据方向信息旋转图像
    if orientation == 3:
        image = image.rotate(180, expand=True)
    elif orientation == 6:
        image = image.rotate(270, expand=True)
    elif orientation == 8:
        image = image.rotate(90, expand=True)
    return image

def get_ps_adjust_photo_para(psid):
    res = requests.get(config.urls['get_ps_adjust_photo_para_url'], params={'id': psid})
    print(res.json())
    paras = res.json()['data']
    brightness_factor, saturation_factor, temperature_factor = float(paras['brightness']), float(paras['saturation']), float(paras['colorTemperature'])
    return brightness_factor, saturation_factor, temperature_factor

def adjust_photos(workdir, pid):
    def adjust_brightness(image, brightness_factor):
        if brightness_factor == 1 or brightness_factor == 0 :
            return image
        enhancer = ImageEnhance.Brightness(image)
        adjusted_image = enhancer.enhance(brightness_factor)
        return adjusted_image

    def adjust_saturation(image, saturation_factor):
        if saturation_factor == 1:
            return image
        enhancer = ImageEnhance.Color(image)
        adjusted_image = enhancer.enhance(saturation_factor)
        return adjusted_image

    def adjust_temperature(image, temperature_factor):
        if temperature_factor == 1:
            return image
        r, g, b = image.split()
        r = r.point(lambda i: i * temperature_factor)
        adjusted_image = Image.merge("RGB", (r, g, b))
        return adjusted_image
    if not os.path.exists(os.path.join(workdir, pid, 'photo2')):
        print(f"Directory {os.path.join(workdir, pid, 'photo2')} does not exist")
        return False
    
    psid = getPSid(pid)
    brightness_factor, saturation_factor, temperature_factor = get_ps_adjust_photo_para(psid)

    if (brightness_factor == 1 and saturation_factor == 1 and temperature_factor == 1):
        print("No need to adjust")
        return False
    if os.path.exists(os.path.join(workdir, pid, 'photo3')):
        print(f'{os.path.join(workdir, pid, "photo3")}目录已存在，跳过')
        return
    os.makedirs(os.path.join(workdir, pid, 'photo3'), exist_ok=True)

    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 开始调整图片曝光...')
    start_time = time.time()
    for filename in os.listdir(os.path.join(workdir, pid, 'photo2')):
        if filename.endswith(".jpg"):
            # print(f"Adjusting {filename}:brightness={brightness_factor}, saturation={saturation_factor}, temperature={temperature_factor}")
            image = Image.open(os.path.join(workdir, pid, 'photo2', filename))
            image = rotate_image(image)

            brightened_image = adjust_brightness(image, brightness_factor)
            saturated_image = adjust_saturation(brightened_image, saturation_factor)
            adjusted_image = adjust_temperature(saturated_image, temperature_factor)

            adjusted_image.save(os.path.join(workdir, pid, 'photo3', filename))
    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 图片曝光调整完成，共费时{diff_time(start_time)}')
    return True

def getPSid(pid):
    res = requests.get(config.urls['get_psid_url'], params={'pid': pid})
    print('get_psid_url:', res.url)
    print('res:', res.text)
    res = json.loads(res.text)
    return str(res['data'])

def getHeadCount(pid):
    res = requests.get(config.urls['get_printinfo_url'], params={'id': pid})
    print('get_printinfo_url:', res.url)
    print('res:', res.text)
    if res.status_code != 200:
        print('获取人数失败，程序退出')
        exit(1)
    res = json.loads(res.text)
    return res['data']['headcount']

def get_ps_type(pid):
    # return 1:圆形影棚 2:方形影棚
    res = requests.get(config.urls['get_ps_type_url'], params={'pid': pid})
    return res.json()['data']['type']

def find_valid_camera_on_oss(pid):
    if get_ps_type(pid) == 1:
        print('当前拍照影棚为：圆形影棚')
        cameras = (103, 93, 113)
    else:
        print('当前拍照影棚为：方形影棚')
        cameras = (74, 64, 84)
    find_camera = 0
    for camera in cameras:
        objectkey = f'photos/{pid}/photo2/{camera}_8.jpg'
        find = config.oss_bucket.object_exists(objectkey)
        if find:
            find_camera = camera
            break

    print('找到有效正脸相机:', find_camera)
    if find_camera == 0:
        print('{cameras}没有找到照片，程序退出')
        exit(1)
    return find_camera

def aliyun_face(pid):
    high = False
    style = 'imm/detectface'
    camera = find_valid_camera_on_oss(pid)
    objectkey = f'photos/{pid}/photo2/{camera}_8.jpg'
    try:
        res = config.oss_bucket.get_object(objectkey, process=style)
    except oss2.exceptions.NoSuchKey:
        print('没有找到文件:', objectkey)
        return high
    res = json.loads(res.read())
    if res['success']:
        if res['Faces'] is None:
            print('no face')
            return None
        else:
            print('faces num:', len(res['Faces']))
        for face in res['Faces']:
            print('-' * 20)
            print('face_id:', face['FaceId'])
            print('gender:', face['Gender'])
            print('age:', face['Age'])

            if face['Gender'] == 'FEMALE' and face['Age'] < 22: high = True
            if face['Gender'] == 'MALE' and face['Age'] < 15: high = True
    else:
        print('face detect failed...')
    return high

def down_obj_from_oss(workdir, pid, action):
    if os.path.exists(os.path.join(workdir, action, pid)):
        print(f'目录{os.path.join(workdir, action, pid)}已存在，跳过')
        return
    else:
        os.makedirs(os.path.join(workdir, action, pid))

    # 根据前缀获取文件列表
    prefix = f'objs/{action}/{pid}/'
    filelist = oss2.ObjectIteratorV2(config.oss_bucket, prefix=prefix)
    print('正在下载:', prefix)
    obj_filename = ""
    for file in filelist:
        filename = file.key.split('/')[-1]
        if filename.endswith('.obj'): 
            obj_filename = filename
        # print('正在下载:', file.key)
        localfile = os.path.join(workdir, action, pid, filename)
        config.oss_bucket.get_object_to_file(file.key, localfile)

    return obj_filename

def set_photos_join_type(workdir, pid, photoN, mesh = '0', texture='0'):
    photoN_path = os.path.join(workdir, pid, photoN)
    for xmp in os.listdir(photoN_path):
        if xmp.endswith('.xmp'):
            xmp_path = os.path.join(photoN_path, xmp)
            with open(xmp_path, 'r') as f:
                lines = f.readlines()
                lines = [line.replace('xcr:InMeshing="0"', f'xcr:InMeshing="{mesh}"') for line in lines]
                lines = [line.replace('xcr:InMeshing="1"', f'xcr:InMeshing="{mesh}"') for line in lines]
                lines = [line.replace('xcr:InTexturing="0"', f'xcr:InTexturing="{texture}"') for line in lines]
                lines = [line.replace('xcr:InTexturing="1"', f'xcr:InTexturing="{texture}"') for line in lines]
                with open(xmp_path, 'w') as f:
                    f.writelines(lines)

def set_photo_join_type(workdir, pid, photoN, camera_id, mesh = '0', texture='0'):
    if photoN == 'photo1':
        filename = os.path.join(workdir, pid, photoN, f'{camera_id}_1.xmp')
    else:
        filename = os.path.join(workdir, pid, photoN, f'{camera_id}_8.xmp')
    with open(filename, 'r') as f:
        lines = f.readlines()
        lines = [line.replace('xcr:InMeshing="0"', f'xcr:InMeshing="{mesh}"') for line in lines]
        lines = [line.replace('xcr:InMeshing="1"', f'xcr:InMeshing="{mesh}"') for line in lines]
        lines = [line.replace('xcr:InTexturing="0"', f'xcr:InTexturing="{texture}"') for line in lines]
        lines = [line.replace('xcr:InTexturing="1"', f'xcr:InTexturing="{texture}"') for line in lines]
        with open(filename, 'w') as f:
            f.writelines(lines)

def down_from_oss(oss_client, workdir, pid, per=100,photoPath=""):
    start_time = time.time()
    path = os.path.join(workdir, pid)
    if os.path.exists(path): 
        print(f"Directory {path} already exists, skip")
        return
    os.makedirs(os.path.join(path, 'photo1'))
    os.makedirs(os.path.join(path, 'photo2'))
    
    psid = getPSid(pid)
    # 根据前缀获取文件列表
    prefix = f'photos/{pid}/'
    filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix)
    for file in tqdm(filelist):
        filename = file.key.split('/')[-1]
        localfile = ""
        # print('正在下载:', file.key)
        if photoPath == "":
            if filename.endswith('_1.jpg'):
                localfile = os.path.join(path, 'photo1', filename)
            else:
                localfile = os.path.join(path, 'photo2', filename)
        else:
            if photoPath=="1":
                if filename.endswith('_1.jpg'):
                    localfile = os.path.join(path, 'photo1', filename)
            else:
                if filename.endswith('_8.jpg'):
                    localfile = os.path.join(path, 'photo2', filename)
        if localfile == "":
            continue
        style = f'image/resize,p_{per}'
        if per == 100:
            oss_client.get_object_to_file(file.key, localfile)
        else:
            oss_client.get_object_to_file(file.key, localfile, process=style)
        
        #判断localfile 是否有包含 photo2
    
    
    #遍历处理photo2的数数据
    # if str(psid) == "96" or str(pid) == "118994":
    #     path = os.path.join(workdir, pid, 'photo2')
    #     files = []
    #     for fileName in os.listdir(path):
    #         if ".jpg" in fileName:
    #             files.append(path+"\\"+fileName)
    #     beginTime =  time.time()
    #     with ThreadPoolExecutor(max_workers=6) as executor:
    #         executor.map(process_image, files)
              
    #     print(f'{localfile}灰度处理费时{diff_time(beginTime)}')

     
    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片下载完成, 共费时{diff_time(start_time)}')

#灰度处理图片
def process_image(localfile):
    if ".jpg" in localfile:
        common.remove_gray_and_sharpening(localfile)

def get_defineDistances(psid):
    res = ''
    distances = libs_db.get_floor_sticker_distances(psid).split(';')
    print("distances",distances)
    for d in distances:
        p1, p2, distance = d.split(' ')
        res = res + f' -defineDistance {p1} {p2} {distance}'
    return res.strip()

def is_running(psname):
    for p in psutil.process_iter(['name']):
        if psname.strip() in p.info['name']:
            return True
    return False

def diff_time(start_time):
    # 按照分:秒的方式返回时间差
    end_time = time.time()
    diff = end_time - start_time
    m, s = divmod(diff, 60)
    return f'{int(m)}分{int(s)}秒'

def down_obj_from_oss_for_fix(workdir, pid, action):
    if os.path.exists(os.path.join(workdir, action, pid)):
        print(f'目录{os.path.join(workdir, action, pid)}已存在，跳过')
        return
    else:
        os.makedirs(os.path.join(workdir, action, pid))

    # 根据前缀获取文件列表
    prefix = f'objs/{action}/{pid}/'
    filelist = oss2.ObjectIteratorV2(config.oss_bucket, prefix=prefix)
    print('正在下载:', prefix)
    obj_filename = ""
    for file in filelist:
        filename = file.key.split('/')[-1]
        if filename.endswith(f'{pid}.obj') == False:
            continue
        
        if filename.endswith('.obj'): 
            obj_filename = filename

        # print('正在下载:', file.key)
        localfile = os.path.join(workdir, action, pid, filename)
        config.oss_bucket.get_object_to_file(file.key, localfile)
    return obj_filename

def down_obj_from_oss_for_repeat_texture(workdir, pid, action):
    print(os.path.join(workdir, pid))
    if not os.path.exists(os.path.join(workdir, pid)):
        print("目录不存在,无法处理")
        return "error"

    #判断是否有专门新建的目录用来重新贴图
    newPath = os.path.join(workdir, pid,'repeat_texture','completion_cloud')
    if not os.path.exists(newPath):
        os.makedirs(newPath)


    # 根据前缀获取文件列表
    prefix = f'objs/{action}/{pid}/ai/repeat_texture/completion_cloud'
    filelist = oss2.ObjectIteratorV2(config.oss_bucket, prefix=prefix)
    print('正在下载:', prefix)
    for file in filelist:
        filename = file.key.split('/')[-1]
        if filename == "":
            continue
        localfile = os.path.join(newPath, filename)
        config.oss_bucket.get_object_to_file(file.key, localfile)
    return newPath




def down_obj_from_oss_auto(pid,action,pathStr):
    #pathStr = os.path.join(workdir, str(pid)+"_ai")
    if not os.path.exists(pathStr):
        os.makedirs(pathStr)

    # 根据前缀获取文件列表
    prefix = f'objs/{action}/{pid}'
    filelist = oss2.ObjectIteratorV2(config.oss_bucket, prefix=prefix)
    print('正在下载:', prefix)
    for file in filelist:
        filename = file.key.split('/')[-1]
        if filename == "":
            continue
        
        if filename != f"{pid}.jpg" and filename != f"{pid}.obj" and filename != f"{pid}.mtl":
            continue

        print(filename)
        localfile = os.path.join(pathStr, filename)
        config.oss_bucket.get_object_to_file(file.key, localfile)



def down_from_oss_photo2(oss_client, workdir, pid, per=100,photoPath=""):
    start_time = time.time()
    path = os.path.join(workdir, pid)
    if os.path.exists(path): 
        print(f"Directory {path} already exists, skip")
        return
    # os.makedirs(os.path.join(path, 'photo1'))
    os.makedirs(os.path.join(path, 'photo2'))
    
    psid = getPSid(pid)
    # 根据前缀获取文件列表
    prefix = f'photos/{pid}/photo2'
    filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix)
    for file in tqdm(filelist):
        filename = file.key.split('/')[-1]
        localfile = ""
        # print('正在下载:', file.key)
        if photoPath == "":
            if filename.endswith('_1.jpg'):
                localfile = os.path.join(path, 'photo1', filename)
            else:
                localfile = os.path.join(path, 'photo2', filename)
        else:
            if photoPath=="1":
                if filename.endswith('_1.jpg'):
                    localfile = os.path.join(path, 'photo1', filename)
            else:
                if filename.endswith('_8.jpg'):
                    localfile = os.path.join(path, 'photo2', filename)
        if localfile == "":
            continue
        style = f'image/resize,p_{per}'
        if per == 100:
            oss_client.get_object_to_file(file.key, localfile)
        else:
            oss_client.get_object_to_file(file.key, localfile, process=style)
        
        #判断localfile 是否有包含 photo2
    
    
    #遍历处理photo2的数数据
    # if str(psid) == "96" or str(pid) == "118994":
    #     path = os.path.join(workdir, pid, 'photo2')
    #     files = []
    #     for fileName in os.listdir(path):
    #         if ".jpg" in fileName:
    #             files.append(path+"\\"+fileName)
    #     beginTime =  time.time()
    #     with ThreadPoolExecutor(max_workers=6) as executor:
    #         executor.map(process_image, files)
              
    #     print(f'{localfile}灰度处理费时{diff_time(beginTime)}')

     
    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 图片下载完成, 共费时{diff_time(start_time)}')