import os
import argparse
import re
from PIL import Image

def create_gif(image_folder, output_path, fps=10, loop=0, sort_order='asc'):
    """
    從指定目錄的圖片創(chuàng)建GIF動圖
    參數(shù):
    image_folder (str): 圖片目錄路徑
    output_path (str): GIF輸出路徑
    fps (int/float): 每秒幀數(shù),默認(rèn)0.5
    loop (int): 循環(huán)次數(shù)(0=無限),默認(rèn)0
    sort_order (str): 排序方式(asc=升序/desc=降序),默認(rèn)asc
    """
    # 支持的圖片格式
    IMAGE_EXTS = ('.png', '.jpg', '.jpeg', '.bmp', '.webp')
    
    # 篩選目錄內(nèi)的圖片文件
    image_files = [
        os.path.join(image_folder, f) 
        for f in os.listdir(image_folder) 
        if f.lower().endswith(IMAGE_EXTS)
    ]
    
    if not image_files:
        raise ValueError(f"目錄 '{image_folder}' 未找到圖片文件")
    
    # 提取文件名中的數(shù)字作為排序依據(jù)
    def sort_key(filename):
        basename = os.path.basename(filename)
        match = re.search(r'\d+', basename)
        return int(match.group()) if match else basename
    
    # 根據(jù)參數(shù)切換升/降序排序
    reverse_flag = True if sort_order == 'desc' else False
    image_files.sort(key=sort_key, reverse=reverse_flag)
    
    # 打印排序后的圖片列表
    print(f"找到 {len(image_files)} 張圖片，排序方式: {sort_order}")
    for i, f in enumerate(image_files):
        print(f"  {i+1}. {os.path.basename(f)}")
    
    # 打開并處理圖片（轉(zhuǎn)RGB兼容透明通道）
    images = []
    for img_path in image_files:
        try:
            img = Image.open(img_path)
            img = img.convert('RGB')  # 統(tǒng)一轉(zhuǎn)為RGB模式
            images.append(img)
        except Exception as e:
            print(f"警告: 無法打開圖片 {img_path} - {e}")
    
    if not images:
        raise ValueError("無可用圖片文件")
    
    # 計算每幀持續(xù)時間（毫秒）
    duration = 1000 / fps
    
    # 保存GIF
    images[0].save(
        output_path,
        save_all=True,
        append_images=images[1:],
        duration=duration,
        loop=loop
    )
    
    # 打印生成結(jié)果
    print(f"\nGIF已保存至: {output_path}")
    print(f"幀率: {fps} FPS | 圖片數(shù)量: {len(images)} | 循環(huán)次數(shù): {'無限' if loop == 0 else loop}")

def main():
    # 命令行參數(shù)解析
    parser = argparse.ArgumentParser(description='批量圖片轉(zhuǎn)GIF動圖工具')
    parser.add_argument('--input_dir', default='./coco_test/', help='輸入圖片目錄（默認(rèn): ./coco_test/）')
    parser.add_argument('--output_file', default='Output_GIF/coco_test-202601.gif', help='輸出GIF文件路徑')
    parser.add_argument('-f', '--fps', type=float, default=0.5, help='每秒幀數(shù)（默認(rèn): 0.5）')
    parser.add_argument('-l', '--loop', type=int, default=0, help='循環(huán)次數(shù)（0=無限，默認(rèn): 0）')
    parser.add_argument('-s', '--sort_order', choices=['asc', 'desc'], default='asc', help='排序方式（asc=升序/desc=降序，默認(rèn): asc）')
    
    args = parser.parse_args()
    
    # 校驗輸入目錄是否存在
    if not os.path.isdir(args.input_dir):
        print(f"錯誤: 輸入目錄 '{args.input_dir}' 不存在")
        return
    
    # 自動補全GIF擴展名
    if not args.output_file.lower().endswith('.gif'):
        args.output_file += '.gif'
    
    # 自動創(chuàng)建輸出目錄（如果不存在）
    output_dir = os.path.dirname(args.output_file)
    if output_dir and not os.path.exists(output_dir):
        os.makedirs(output_dir, exist_ok=True)
        print(f"提示: 自動創(chuàng)建輸出目錄 '{output_dir}'")
    
    # 生成GIF
    try:
        create_gif(args.input_dir, args.output_file, args.fps, args.loop, args.sort_order)
    except Exception as e:
        print(f"錯誤: {e}")

if __name__ == "__main__":
    main()

import os
import argparse
import re
from PIL import Image
import numpy as np

def create_gif(image_folder, output_path, fps=10, loop=0, 
               reduce_colors=True, max_colors=256, 
               optimize_frames=True, resize_factor=1.0, 
               compress_level=6, quality=85, sort_order='asc'):
    """
    從指定目錄圖片創(chuàng)建GIF動圖（支持壓縮/縮放/調(diào)色）
    參數(shù):
    image_folder (str): 圖片目錄路徑
    output_path (str): GIF輸出路徑
    fps (int/float): 每秒幀數(shù)，默認(rèn)10
    loop (int): 循環(huán)次數(shù)（0=無限），默認(rèn)0
    reduce_colors (bool): 是否減少顏色數(shù)，默認(rèn)True
    max_colors (int): 最大顏色數(shù)(1-256)，默認(rèn)256
    optimize_frames (bool): 是否優(yōu)化幀（差異編碼），默認(rèn)True
    resize_factor (float): 縮放因子(0.1-1.0)，默認(rèn)1.0（不縮放）
    compress_level (int): 壓縮級別(0-9)，默認(rèn)6
    quality (int): 調(diào)色板質(zhì)量(1-100)，默認(rèn)85
    sort_order (str): 排序方式（asc=升序/desc=降序），默認(rèn)asc
    """
    # 支持的圖片格式
    IMAGE_EXTS = ('.png', '.jpg', '.jpeg', '.bmp', '.webp')
    
    # 篩選目錄內(nèi)的圖片文件
    image_files = [
        os.path.join(image_folder, f) 
        for f in os.listdir(image_folder) 
        if f.lower().endswith(IMAGE_EXTS)
    ]
    
    if not image_files:
        raise ValueError(f"目錄 '{image_folder}' 未找到圖片文件")
    
    # 提取文件名中的數(shù)字作為排序依據(jù)
    def sort_key(filename):
        basename = os.path.basename(filename)
        match = re.search(r'\d+', basename)
        return int(match.group()) if match else basename
    
    # 根據(jù)參數(shù)切換升/降序排序
    reverse = True if sort_order == 'desc' else False
    image_files.sort(key=sort_key, reverse=reverse)
    
    # 打印排序結(jié)果
    print(f"找到 {len(image_files)} 張圖片，排序方式: {sort_order}")
    for i, f in enumerate(image_files):
        print(f"  {i+1}. {os.path.basename(f)}")
    
    # 加載并統(tǒng)一圖片格式/尺寸
    images = []
    first_size = None
    
    for img_path in image_files:
        try:
            img = Image.open(img_path)
            
            # 校驗圖片尺寸一致性
            if first_size is None:
                first_size = img.size
                print(f"\n基準(zhǔn)尺寸: {first_size[0]}x{first_size[1]}")
            elif img.size != first_size:
                print(f"警告: 圖片 {os.path.basename(img_path)} 尺寸 {img.size} 與基準(zhǔn)不一致")
            
            # 處理透明通道（轉(zhuǎn)為白色背景）
            if img.mode == 'RGBA':
                background = Image.new('RGB', img.size, (255, 255, 255))
                background.paste(img, mask=img.split()[-1])
                img = background
            else:
                img = img.convert('RGB')
            
            images.append(img)
            
        except Exception as e:
            print(f"警告: 無法打開圖片 {img_path} - {e}")
    
    if not images:
        raise ValueError("無可用圖片文件")
    
    print(f"\n成功加載 {len(images)} 張圖片")
    
    # 圖片縮放（按需）
    if resize_factor != 1.0 and 0.1 <= resize_factor <= 1.0:
        new_size = (int(first_size[0] * resize_factor), 
                   int(first_size[1] * resize_factor))
        print(f"縮放圖片: {first_size} → {new_size} (縮放因子: {resize_factor})")
        
        for i in range(len(images)):
            images[i] = images[i].resize(new_size, Image.Resampling.LANCZOS)
    
    # 減少顏色數(shù)（按需）
    if reduce_colors and max_colors < 256:
        print(f"減少顏色數(shù)至 {max_colors} 色")
        
        for i in range(len(images)):
            images[i] = images[i].quantize(colors=max_colors, 
                                           method=Image.Quantize.MEDIANCUT)
            images[i] = images[i].convert('RGB')
    
    # 計算每幀時長（毫秒）
    duration = 1000 / fps
    
    # 自動創(chuàng)建輸出目錄（不存在時）
    output_dir = os.path.dirname(os.path.abspath(output_path))
    if output_dir:
        os.makedirs(output_dir, exist_ok=True)
        print(f"\n輸出目錄已確保存在: {output_dir}")
    
    # 生成并保存GIF（帶優(yōu)化參數(shù)）
    print("正在生成GIF文件...")
    images[0].save(
        output_path,
        save_all=True,
        append_images=images[1:],
        duration=duration,
        loop=loop,
        optimize=optimize_frames,
        compress_level=compress_level,
        quality=quality,
        disposal=2,
        dither=Image.Dither.FLOYDSTEINBERG
    )
    
    # 輸出生成結(jié)果
    file_size = os.path.getsize(output_path) / (1024 * 1024)
    print(f"\n? GIF已保存至: {output_path}")
    print(f"?? 文件大小: {file_size:.2f} MB | ??? 幀率: {fps} FPS (每幀 {duration:.1f}ms)")
    print(f"??? 圖片數(shù)量: {len(images)} | ?? 循環(huán)次數(shù): {'無限' if loop == 0 else loop}")
    if resize_factor != 1.0:
        print(f"?? 縮放因子: {resize_factor}")
    if reduce_colors:
        print(f"?? 顏色數(shù)限制: {max_colors}")
    print(f"??? 壓縮級別: {compress_level} | ?? 質(zhì)量設(shè)置: {quality}")

def main():
    # 命令行參數(shù)解析
    parser = argparse.ArgumentParser(description='GIF生成工具（支持壓縮/縮放/調(diào)色/排序）')
    parser.add_argument('--input_dir', default='./THUD_Robot_data/', help='輸入圖片目錄（默認(rèn): ./Gif_bev_Global_Graph_Range/）')
    parser.add_argument('--output_file', default='Output_GIF/THUD_Robot_data-test.gif', help='輸出GIF文件路徑')
    parser.add_argument('-f', '--fps', type=float, default=0.5, help='每秒幀數(shù)（默認(rèn): 0.5）')
    parser.add_argument('-l', '--loop', type=int, default=0, help='循環(huán)次數(shù)（0=無限，默認(rèn): 0）')
    parser.add_argument('-s', '--sort_order', choices=['asc', 'desc'], default='asc',help='排序方式（asc=升序/desc=降序，默認(rèn): asc）')
    
    # 壓縮/優(yōu)化相關(guān)參數(shù)
    parser.add_argument('--no-reduce-colors', action='store_false', default=True, dest='reduce_colors',help='不減少顏色數(shù)')
    parser.add_argument('--max-colors', type=int, default=256,help='最大顏色數(shù)(1-256)，默認(rèn)256')
    parser.add_argument('--resize', type=float, default=0.6,help='縮放因子(0.1-1.0)，默認(rèn)0.6')
    parser.add_argument('--compress', type=int, default=3, choices=range(0, 10),help='壓縮級別(0-9)，默認(rèn)3')
    parser.add_argument('--quality', type=int, default=95,help='調(diào)色板質(zhì)量(1-100)，默認(rèn)95')
    
    args = parser.parse_args()
    
    # 校驗輸入目錄
    if not os.path.isdir(args.input_dir):
        print(f"? 錯誤: 輸入目錄 '{args.input_dir}' 不存在")
        return
    
    # 自動補全GIF擴展名
    if not args.output_file.lower().endswith('.gif'):
        args.output_file += '.gif'
    
    # 執(zhí)行GIF生成
    try:
        create_gif(
            args.input_dir, 
            args.output_file, 
            args.fps, 
            args.loop,
            reduce_colors=args.reduce_colors,
            max_colors=args.max_colors,
            resize_factor=args.resize,
            compress_level=args.compress,
            quality=args.quality,
            sort_order=args.sort_order
        )
    except Exception as e:
        print(f"\n? 錯誤: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    main()