import os

import numpy as np


class Config:
    # 文件相关参数
    nRows, nCols, nBands = 256, 1024, 22
    nRgbRows, nRgbCols, nRgbBands = 1024, 4096, 3

    # 需要设置的谱段等参数
    selected_bands = [127, 201, 202, 294]
    bands = [127, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210,
             211, 212, 213, 214, 215, 216, 217, 218, 219, 294]
    is_yellow_min = np.array([0.10167048, 0.1644719, 0.1598884, 0.31534621])
    is_yellow_max = np.array([0.212984, 0.25896924, 0.26509268, 0.51943593])
    is_black_threshold = np.asarray([0.1369, 0.1472, 0.1439, 0.1814])
    black_yellow_bands = [0, 2, 3, 21]
    green_bands = [i for i in range(1, 21)]

    # 光谱模型参数
    blk_size = 4  # 必须是2的倍数，不然会出错
    pixel_model_path = r"./weights/pixel_2022-08-02_15-22.model"
    blk_model_path = r"./weights/rf_4x4_c22_20_sen8_9.model"
    spec_size_threshold = 3

    # rgb模型参数
    rgb_tobacco_model_path = r"weights/tobacco_dt_2022-08-05_10-38.model"
    rgb_background_model_path = r"weights/background_dt_2022-08-22_22-15.model"
    threshold_low, threshold_high = 10, 230
    threshold_s = 190  # 饱和度的最高允许值
    rgb_size_threshold = 4  # rgb的尺寸限制
    ai_path = 'weights/best0823.pt'
    ai_conf_threshold = 0.5

    # mask parameter
    target_size = (256, 256)  # (Width, Height) of mask
    valve_merge_size = 2  # 每两个喷阀当中有任意一个出现杂质则认为都是杂质
    valve_horizontal_padding = 3  # 喷阀横向膨胀的尺寸，应该是奇数,3时表示左右各膨胀1
    max_open_valve_limit = 25  # 最大同时开启喷阀限制,按照电流计算，当前的喷阀可以开启的喷阀 600W的电源 / 12V电源 = 50A, 一个阀门1A
    max_time_spent = 200
    # save part
    offset_vertical = 0

    # logging
    root_dir = os.path.split(os.path.realpath(__file__))[0]