import os
import cv2

# ----------------------------
# 配置
# ----------------------------
SOURCE_ROOT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/cls-61/class4"
TARGET_ROOT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/cls-61/class4c"
TARGET_SIZE = 640  # resize 尺寸

# ----------------------------
# 全局 ROI (x, y, w, h)
# ----------------------------
GLOBAL_ROI = [445, 540, 931, 319]
#GLOBAL_ROI = [604, 182, 594, 252]


# ----------------------------
# 主处理函数
# ----------------------------
def process_images():
    x, y, w, h = GLOBAL_ROI

    # 遍历主目录下的所有子文件夹
    for sub_dir_name in sorted(os.listdir(SOURCE_ROOT_DIR)):
        src_sub_dir = os.path.join(SOURCE_ROOT_DIR, sub_dir_name)

        if not os.path.isdir(src_sub_dir):
            continue  # 跳过非文件夹

        tgt_sub_dir = os.path.join(TARGET_ROOT_DIR, sub_dir_name)
        os.makedirs(tgt_sub_dir, exist_ok=True)

        print(f"\n📁 处理文件夹: {sub_dir_name}")

        for file in os.listdir(src_sub_dir):
            if not file.lower().endswith((".jpg", ".png", ".jpeg", ".bmp", ".tif", ".tiff")):
                continue

            img_path = os.path.join(src_sub_dir, file)
            img = cv2.imread(img_path)

            if img is None:
                print(f"❌ 无法读取图片: {img_path}")
                continue

            h_img, w_img = img.shape[:2]

            # ROI 边界保护
            x1 = max(0, x)
            y1 = max(0, y)
            x2 = min(w_img, x + w)
            y2 = min(h_img, y + h)

            cropped = img[y1:y2, x1:x2]
            if cropped.size == 0:
                print(f"❌ 裁剪结果为空: {img_path}")
                continue

            resized = cv2.resize(cropped, (TARGET_SIZE, TARGET_SIZE), interpolation=cv2.INTER_AREA)

            tgt_path = os.path.join(tgt_sub_dir, file)
            cv2.imwrite(tgt_path, resized)

            print(f"✅ {sub_dir_name}/{file}")

    print("\n🎉 全部图片处理完成")


# ----------------------------
# 入口
# ----------------------------
if __name__ == "__main__":
    process_images()