# lib/__init__.py
# 完整导入 lib 目录下所有模块的公共接口，确保无未定义引用

# ======================================
# 1. 导入 pvtv2.py 中的类和模型
# ======================================
from .pvtv2 import (
    # 基础组件类
    Mlp,
    Attention,
    Block,
    OverlapPatchEmbed,
    PyramidVisionTransformerImpr,
    DWConv,
    # PVTv2 模型实例
    pvt_v2_b0,
    pvt_v2_b1,
    pvt_v2_b2,
    pvt_v2_b3,
    pvt_v2_b4,
    pvt_v2_b5
)

# ======================================
# 2. 导入 resnet.py 中的类和模型
# ======================================
from .resnet import (
    # ResNet 基础类
    ResNet,
    # ResNet 模型实例
    resnet18,
    resnet34,
    resnet50,
    resnet101,
    resnet152
)

# ======================================
# 3. 导入 networks.py 中的模型类（关键：按实际类名调整）
# ======================================
# 注意：如果 networks.py 中的主模型类是 EMCADNet，保留以下导入；
# 如果是其他名称（如 EMCAD），请替换为实际类名（与 networks.py 完全一致）
from .networks import EMCADNet

# ======================================
# 4. 导入 decoders.py 中的类和函数（基于你提供的完整代码）
# ======================================
from .decoders import (
    # 工具函数
    gcd,
    _init_weights,
    act_layer,
    channel_shuffle,
    MSCBLayer,
    # 解码模块类
    MSDC,
    MSCB,
    EUCB,
    LGAG,
    CAB,
    SAB,
    EMCAD  # decoders.py 中定义的 EMCAD 解码模块
)

# ======================================
# 全局公共接口：定义可通过 from lib import * 导入的内容
# ======================================
__all__ = [
    # ------------------------------
    # pvtv2.py
    # ------------------------------
    'Mlp',
    'Attention',
    'Block',
    'OverlapPatchEmbed',
    'PyramidVisionTransformerImpr',
    'DWConv',
    'pvt_v2_b0',
    'pvt_v2_b1',
    'pvt_v2_b2',
    'pvt_v2_b3',
    'pvt_v2_b4',
    'pvt_v2_b5',
    # ------------------------------
    # resnet.py
    # ------------------------------
    'ResNet',
    'resnet18',
    'resnet34',
    'resnet50',
    'resnet101',
    'resnet152',
    # ------------------------------
    # networks.py
    # ------------------------------
    'EMCADNet',  # 与 networks.py 中的类名保持一致
    # ------------------------------
    # decoders.py
    # ------------------------------
    'gcd',
    '_init_weights',
    'act_layer',
    'channel_shuffle',
    'MSCBLayer',
    'MSDC',
    'MSCB',
    'EUCB',
    'LGAG',
    'CAB',
    'SAB',
    'EMCAD'
]