PolynomialGUI.cpp

// PolynomialGUI.cpp
#include "PolynomialGUI.h"
#include "resource.h"
#include <sstream>
#include <fstream>

#define MAX_LOADSTRING 100

extern HINSTANCE hInst;
extern WCHAR szWindowClass[MAX_LOADSTRING];
extern WCHAR szTitle[MAX_LOADSTRING];

PolynomialGUI::PolynomialGUI(HINSTANCE hInst)
    : hInstance(hInst), hWnd(nullptr), currentPoly(nullptr) {
    hEditPoly1 = hEditPoly2 = nullptr;
    hBtnSplit = hBtnAdd = hBtnMultiply = hBtnSave = hBtnDivide = nullptr;
}

PolynomialGUI::~PolynomialGUI() {
    delete currentPoly;
}

void PolynomialGUI::createControls() {
    // 创建输入框1
    hEditPoly1 = CreateWindowW(
        L"EDIT",
        L"3x^2+5x+7",
        WS_CHILD | WS_VISIBLE | WS_BORDER,
        20, 20, 300, 30,
        hWnd,
        (HMENU)1001,
        hInstance,
        nullptr);

    // "拆分多项式"按钮
    hBtnSplit = CreateWindowW(
        L"BUTTON",
        L"拆分多项式",
        WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON,
        20, 60, 120, 30,
        hWnd,
        (HMENU)1002,
        hInstance,
        nullptr);

    // 创建输入框2
    hEditPoly2 = CreateWindowW(
        L"EDIT",
        L"2x^2+3x+1",
        WS_CHILD | WS_VISIBLE | WS_BORDER,
        20, 100, 300, 30,
        hWnd,
        (HMENU)1003,
        hInstance,
        nullptr);

    // "多项式相加"按钮
    hBtnAdd = CreateWindowW(
        L"BUTTON",
        L"多项式相加",
        WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON,
        20, 140, 120, 30,
        hWnd,
        (HMENU)1004,
        hInstance,
        nullptr);

    // "多项式相乘"按钮
    hBtnMultiply = CreateWindowW(
        L"BUTTON",
        L"多项式相乘",
        WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON,
        20, 180, 120, 30,
        hWnd,
        (HMENU)1005,
        hInstance,
        nullptr);

    // "保存结果"按钮
    hBtnSave = CreateWindowW(
        L"BUTTON",
        L"保存结果",
        WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON,
        20, 220, 120, 30,
        hWnd,
        (HMENU)1006,
        hInstance,
        nullptr);

    // "多项式相除"按钮
    hBtnDivide = CreateWindowW(
        L"BUTTON",
        L"多项式相除",
        WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON,
        20, 260, 120, 30,
        hWnd,
        (HMENU)1007,
        hInstance,
        nullptr);
}

void PolynomialGUI::drawBarChart(HDC hdc) {
    if (!currentPoly || !currentPoly->getHead()) return;

    int x = LEFT_X;
    for (Term* t = currentPoly->getHead(); t; t = t->next) {
        int h = abs(t->coef) * SCALE;
        COLORREF col = (t->exp & 1) ? RGB(255, 100, 100) : RGB(100, 100, 255);
        HBRUSH hbr = CreateSolidBrush(col);
        RECT rc = { x, TOP_Y, x + BAR_W, TOP_Y + h };
        FillRect(hdc, &rc, hbr);
        DeleteObject(hbr);

        // 在柱下方标指数
        wchar_t label[16];
        wsprintfW(label, L"x^%d", t->exp);
        TextOutW(hdc, x, TOP_Y + h + 5, label, (int)wcslen(label));

        x += BAR_W + GAP;
    }
}

void PolynomialGUI::updateDisplay(const std::wstring& text, Polynomial* poly) {
    displayText = text;
    delete currentPoly;
    if (poly) {
        currentPoly = new Polynomial(*poly);
    }
    else {
        currentPoly = nullptr;
    }
    InvalidateRect(hWnd, nullptr, TRUE);
}

void PolynomialGUI::onSplit() {
    HWND hEdit = GetDlgItem(hWnd, 1001);
    wchar_t buf[256] = {};
    GetWindowTextW(hEdit, buf, 256);

    Polynomial poly(buf);

    std::wstringstream ss;
    for (Term* t = poly.getHead(); t; t = t->next) {
        wchar_t tmp[64];
        wsprintfW(tmp, L"%dx^%d  ", t->coef, t->exp);
        ss << tmp;
    }

    updateDisplay(ss.str(), &poly);
}

void PolynomialGUI::onAdd() {
    // 读第一个多项式
    HWND h1 = GetDlgItem(hWnd, 1001);
    wchar_t buf1[256];
    GetWindowTextW(h1, buf1, 256);
    Polynomial p1(buf1);

    // 读第二个多项式
    HWND h2 = GetDlgItem(hWnd, 1003);
    wchar_t buf2[256];
    GetWindowTextW(h2, buf2, 256);
    Polynomial p2(buf2);

    // 相加
    Polynomial sum = p1.add(p2);

    // 显示结果
    std::wstring poly1 = p1.toString();
    std::wstring poly2 = p2.toString();
    std::wstring result = sum.toString();

    std::wstring display = L"(" + poly1 + L") + (" + poly2 + L") = " + result;
    updateDisplay(display, &sum);
}

void PolynomialGUI::onMultiply() {
    // 读第一个多项式
    HWND h1 = GetDlgItem(hWnd, 1001);
    wchar_t buf1[256];
    GetWindowTextW(h1, buf1, 256);
    Polynomial p1(buf1);

    // 读第二个多项式
    HWND h2 = GetDlgItem(hWnd, 1003);
    wchar_t buf2[256];
    GetWindowTextW(h2, buf2, 256);
    Polynomial p2(buf2);

    // 相乘
    Polynomial prod = p1.multiply(p2);

    // 显示结果
    std::wstring poly1 = p1.toString();
    std::wstring poly2 = p2.toString();
    std::wstring result = prod.toString();

    std::wstring display = L"(" + poly1 + L") * (" + poly2 + L") = " + result;
    updateDisplay(display, &prod);
}

void PolynomialGUI::onSave() {
    std::wofstream fs("result.txt");
    if (fs) {
        fs << displayText;
        fs.close();
        MessageBoxW(hWnd, L"已保存到程序目录下的 result.txt", L"提示", MB_OK);
    }
    else {
        MessageBoxW(hWnd, L"保存文件失败", L"错误", MB_OK | MB_ICONERROR);
    }
}

void PolynomialGUI::onDivide() {
    // 读被除式、除式
    HWND h1 = GetDlgItem(hWnd, 1001);
    HWND h2 = GetDlgItem(hWnd, 1003);
    wchar_t buf1[256], buf2[256];
    GetWindowTextW(h1, buf1, 256);
    GetWindowTextW(h2, buf2, 256);

    Polynomial p1(buf1);
    Polynomial p2(buf2);

    // 检查除式是否为空
    if (p2.isZero()) {
        MessageBoxW(hWnd, L"除式不能为 0！", L"错误", MB_OK);
        return;
    }

    try {
        // 执行除法 - 不使用结构化绑定
        auto divisionResult = p1.divide(p2);
        Polynomial quotient = divisionResult.first;
        Polynomial remainder = divisionResult.second;

        // 使用 toString 显示美观结果
        std::wstring poly1_str = p1.toString();
        std::wstring poly2_str = p2.toString();
        std::wstring quotient_str = quotient.toString();
        std::wstring remainder_str = remainder.toString();

        // 构建清晰的显示字符串
        std::wstringstream ss;
        ss << L"多项式除法：\r\n";
        ss << L"=========================\r\n";
        ss << L"被除式: " + poly1_str + L"\r\n";
        ss << L"除式:   " + poly2_str + L"\r\n";
        ss << L"-------------------------\r\n";

        if (remainder_str == L"0") {
            // 能整除的情况
            ss << L"商:     " + quotient_str + L"\r\n";
            ss << L"余式:   " + remainder_str + L"\r\n";
            ss << L"-------------------------\r\n";
            ss << L"结果:   (" + poly1_str + L") ÷ (" + poly2_str + L") = " + quotient_str;
        }
        else if (quotient_str == L"0") {
            // 商为0的情况（不能整除）
            ss << L"商:     " + quotient_str + L"\r\n";
            ss << L"余式:   " + remainder_str + L"\r\n";
            ss << L"-------------------------\r\n";
            ss << L"结果:   (" + poly1_str + L") ÷ (" + poly2_str + L") = ";
            ss << L"(" + remainder_str + L")/(" + poly2_str + L")";
        }
        else {
            // 一般情况（有商有余式）
            ss << L"商:     " + quotient_str + L"\r\n";
            ss << L"余式:   " + remainder_str + L"\r\n";
            ss << L"-------------------------\r\n";
            ss << L"结果:   (" + poly1_str + L") ÷ (" + poly2_str + L") = ";
            ss << quotient_str + L" + (" + remainder_str + L")/(" + poly2_str + L")";
        }

        updateDisplay(ss.str(), &quotient);

        // 保存到文件
        std::wofstream fs("division.txt");
        if (fs) {
            fs << L"多项式除法结果" << std::endl;
            fs << L"==================" << std::endl;
            fs << L"被除式: " << poly1_str << std::endl;
            fs << L"除式: " << poly2_str << std::endl;
            fs << L"商: " << quotient_str << std::endl;
            fs << L"余式: " << remainder_str << std::endl;
            fs << std::endl << L"结果显示:" << std::endl;
            fs << displayText << std::endl;
            fs.close();
            MessageBoxW(hWnd, L"除法结果已保存到 division.txt", L"提示", MB_OK);
        }
    }
    catch (const std::exception& e) {
        MessageBoxW(hWnd, L"多项式除法出错", L"错误", MB_OK | MB_ICONERROR);
    }
}

bool PolynomialGUI::createWindow(int nCmdShow) {
    hWnd = CreateWindowW(
        szWindowClass,
        szTitle,
        WS_OVERLAPPEDWINDOW,
        CW_USEDEFAULT, 0,
        800, 900,
        nullptr,
        nullptr,
        hInstance,
        nullptr);

    if (!hWnd) {
        return false;
    }

    createControls();

    ShowWindow(hWnd, nCmdShow);
    UpdateWindow(hWnd);

    return true;
}

LRESULT PolynomialGUI::handleMessage(UINT msg, WPARAM wParam, LPARAM lParam) {
    switch (msg) {
    case WM_COMMAND: {
        int wmId = LOWORD(wParam);
        switch (wmId) {
        case IDM_ABOUT:
            DialogBox(hInstance, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd,
                [](HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) -> INT_PTR {
                    switch (message) {
                    case WM_INITDIALOG:
                        return (INT_PTR)TRUE;
                    case WM_COMMAND:
                        if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL) {
                            EndDialog(hDlg, LOWORD(wParam));
                            return (INT_PTR)TRUE;
                        }
                        break;
                    }
                    return (INT_PTR)FALSE;
                });
            break;
        case IDM_EXIT:
            DestroyWindow(hWnd);
            break;
        case 1002:
            onSplit();
            break;
        case 1004:
            onAdd();
            break;
        case 1005:
            onMultiply();
            break;
        case 1006:
            onSave();
            break;
        case 1007:
            onDivide();
            break;
        default:
            return DefWindowProc(hWnd, msg, wParam, lParam);
        }
        break;
    }
    case WM_PAINT: {
        PAINTSTRUCT ps;
        HDC hdc = BeginPaint(hWnd, &ps);

        // 1. 左侧文字区
        RECT textRc = { 20, 320, 400, 580 };
        SetBkMode(hdc, TRANSPARENT);
        DrawTextW(hdc, displayText.c_str(), (int)displayText.length(),
            &textRc, DT_LEFT | DT_WORDBREAK | DT_EXPANDTABS);

        // 2. 右侧柱形图
        drawBarChart(hdc);

        EndPaint(hWnd, &ps);
        break;
    }
    case WM_DESTROY:
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hWnd, msg, wParam, lParam);
    }
    return 0;
}