from enum import Enum

# 根据标签修改
class tlabel(Enum):
    BBLOCK   = 1    # 蓝色方块
    RBLOCK   = 2    # 红色方块
    HOSPITAL = 3    # 医院
    BBALL    = 4    # 蓝球
    YBALL    = 5    # 黄球
    TOWER    = 6    # 通信塔
    RBALL    = 7    # 红球
    BASKET   = 8    # 球筐
    MARKL    = 9    # 指向标
    MARKR    = 10   # 指向标
    SPILLAR  = 11   # 小柱体 (红色)
    MPILLAR  = 12   # 中柱体 (蓝色)
    LPILLAR  = 13   # 大柱体 (红色)
    SIGN     = 14   # 文字标牌
    TARGET   = 15   # 目标靶
    SHELTER  = 16   # 停车区
    BASE     = 17   # 基地

class label_filter:
    def __init__(self, list_src):
        self.num = 0
        self.pos = []
        self.list = list_src # 获取目标检测输出的接口 (含标签，位置，置信度)
        # TODO 添加置信度阈值
        pass
    def get(self, tlabel):
        # TODO 循环查找匹配的标签值
        # TODO 返回对应标签的个数，以及坐标列表
        return self.num, self.pos
    def find(self, tlabel):
        # TODO 遍历返回的列表，有对应标签则返回 True
        return False
    def aim_left(self, tlabel):
        # TODO 如果标签存在，则返回列表中位置最靠左的目标框和中心的偏移值
        error = 0
        return error
    def aim_right(self, tlabel):
        # TODO 如果标签存在，则返回列表中位置最靠右的目标框和中心的偏移值
        error = 0
        return error
    def aim_near(self, tlabel):
        # TODO 如果标签存在，则返回列表中位置最近的目标框和中心的偏移值
        error = 0
        return error