#include "MainFrame.h"

constexpr auto g_Border = 2;
CMainFrame::CMainFrame(const wxString& title) : wxFrame(NULL, wxID_ANY, title)
{
    SetSize(wxSize(600, 300));
    init();
}

CMainFrame::~CMainFrame()
{
}

void CMainFrame::init()
{
    auto* panel = new wxPanel(this);
    auto* sizerTop = new wxBoxSizer(wxVERTICAL);

    auto* sizeA = new wxBoxSizer(wxHORIZONTAL);
    auto* pic_path = new wxTextCtrl(panel, wxID_ANY);
    auto* select_pic = new wxButton(panel, wxID_ANY, wxT("选择"));
    sizeA->Add(pic_path, 1, wxALL | wxEXPAND, g_Border);
    sizeA->Add(select_pic, 0, wxALL, g_Border);
    sizerTop->Add(sizeA, 0, wxALL | wxEXPAND, g_Border);

    win_img_ = new CImagePanel(panel);
    sizerTop->Add(win_img_, 1, wxALL | wxEXPAND, g_Border);
    img_ = cv::imread("D:/2.png");
    win_img_->set_img(img_);

    auto* sizeB = new wxBoxSizer(wxHORIZONTAL);
    auto* btn2Blue = new wxButton(panel, wxID_ANY, wxT("蓝底"));
    auto* btn2White = new wxButton(panel, wxID_ANY, wxT("白底"));
    sizeB->Add(btn2Blue, 1, wxALL | wxEXPAND, g_Border);
    sizeB->Add(btn2White, 1, wxALL | wxEXPAND, g_Border);

    btn2Blue->Bind(wxEVT_BUTTON, &CMainFrame::turn2blue, this);
    btn2White->Bind(wxEVT_BUTTON, &CMainFrame::turn2white, this);
    sizerTop->Add(sizeB, 0, wxALL | wxEXPAND, g_Border);

    panel->SetSizer(sizerTop);
    panel->Layout();
}

void CMainFrame::turn2blue(wxCommandEvent& event)
{
    // 定义白色的阈值范围
    cv::Scalar lowerWhite(200, 200, 200); // 白色下限 (B, G, R)
    cv::Scalar upperWhite(255, 255, 255); // 白色上限 (B, G, R)

    // 创建蓝色背景
    cv::Mat mask;
    cv::inRange(img_, lowerWhite, upperWhite, mask); // 检测白色区域

    // 将白色区域替换为蓝色
    cv::Mat blueBackground(img_.size(), img_.type(), cv::Scalar(255, 0, 0)); // 蓝色 (B, G, R)
    blueBackground.copyTo(img_, mask); // 通过掩码复制蓝色区域到原图
    win_img_->set_img(img_);

    // cv::Mat mask = cv::Mat::zeros(img_.size(), CV_8UC1);
    // cv::Rect rect(10, 10, img_.cols - 10, img_.rows - 10);
    // cv::grabCut(img_, mask, rect, cv::Mat(), cv::Mat(), 5,
    //             cv::GC_INIT_WITH_RECT);

    // // 创建一个蓝色背景，BGR格式的蓝色是(255, 0, 0)
    // cv::Mat blueBackground(img_.size(), img_.type(), cv::Scalar(255, 0, 0));

    // // 将前景区域（mask == cv::GC_PR_FGD）复制到蓝色背景中
    // cv::Mat result;
    // img_.copyTo(result, mask);   // 前景区域从原图像复制

    // // 将背景区域填充为蓝色（mask != cv::GC_PR_FGD）
    // cv::Mat backgroundMask = (mask != cv::GC_PR_FGD);   // 背景区域
    // blueBackground.copyTo(result, backgroundMask);      // 将背景区域填充为蓝色

    // // 将处理后的结果显示到窗口中
    // win_img_->set_img(result);
}

void CMainFrame::turn2white(wxCommandEvent& event)
{
}