# config.py
import os

# 路径配置
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(BASE_DIR, "data")
RAW_VIDEO_DIR = os.path.join(DATA_DIR, "raw_videos")
FRAMES_DIR = os.path.join(DATA_DIR, "frames")
ANNOTATIONS_DIR = os.path.join(DATA_DIR, "annotations")
PROCESSED_DIR = os.path.join(DATA_DIR, "processed")
MODEL_DIR = os.path.join(BASE_DIR, "models")

# 创建目录
os.makedirs(RAW_VIDEO_DIR, exist_ok=True)
os.makedirs(FRAMES_DIR, exist_ok=True)
os.makedirs(ANNOTATIONS_DIR, exist_ok=True)
os.makedirs(PROCESSED_DIR, exist_ok=True)
os.makedirs(MODEL_DIR, exist_ok=True)

# 图像处理参数
IMG_HEIGHT = 84
IMG_WIDTH = 84
IMG_CHANNELS = 3
CROP_TOP_RATIO = 0.08   # 裁剪顶部8%（去掉状态栏）
CROP_BOTTOM_RATIO = 0.05 # 裁剪底部5%（去掉按钮栏）

# 视频抽帧参数
EXTRACT_FPS = 5  # 每秒抽5帧

# 动作空间（16个动作）
ACTIONS = [
    'move_up', 'move_down', 'move_left', 'move_right',
    'move_upleft', 'move_upright', 'move_downleft', 'move_downright',
    'attack', 'skill_1', 'skill_2', 'skill_3',
    'recall', 'heal', 'summoner', 'upgrade'
]

NUM_ACTIONS = len(ACTIONS)

# 训练参数
BATCH_SIZE = 64
LEARNING_RATE = 0.0001
NUM_EPOCHS = 50
TRAIN_SPLIT = 0.8

# 设备
DEVICE = None  # 运行时自动检测