/*
 * PROJECT:    PAINT for ReactOS
 * LICENSE:    LGPL-2.0-or-later (https://spdx.org/licenses/LGPL-2.0-or-later)
 * PURPOSE:    Things which should not be in the mouse event handler itself
 * COPYRIGHT:  Copyright 2015 Benedikt Freisen <b.freisen@gmx.net>
 *             Copyright 2021 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
 */

/* INCLUDES *********************************************************/

#include "precomp.h"

INT ToolBase::s_pointSP = 0;
POINT ToolBase::s_pointStack[256] = { { 0 } };

/* FUNCTIONS ********************************************************/

void
regularize(LONG x0, LONG y0, LONG& x1, LONG& y1)
{
    if (abs(x1 - x0) >= abs(y1 - y0))
        y1 = y0 + (y1 > y0 ? abs(x1 - x0) : -abs(x1 - x0));
    else
        x1 = x0 + (x1 > x0 ? abs(y1 - y0) : -abs(y1 - y0));
}

void
roundTo8Directions(LONG x0, LONG y0, LONG& x1, LONG& y1)
{
    if (abs(x1 - x0) >= abs(y1 - y0))
    {
        if (abs(y1 - y0) * 5 < abs(x1 - x0) * 2)
            y1 = y0;
        else
            y1 = y0 + (y1 > y0 ? abs(x1 - x0) : -abs(x1 - x0));
    }
    else
    {
        if (abs(x1 - x0) * 5 < abs(y1 - y0) * 2)
            x1 = x0;
        else
            x1 = x0 + (x1 > x0 ? abs(y1 - y0) : -abs(y1 - y0));
    }
}

BOOL nearlyEqualPoints(INT x0, INT y0, INT x1, INT y1)
{
    INT cxThreshold = toolsModel.GetLineWidth() + UnZoomed(GetSystemMetrics(SM_CXDRAG));
    INT cyThreshold = toolsModel.GetLineWidth() + UnZoomed(GetSystemMetrics(SM_CYDRAG));
    return (abs(x1 - x0) <= cxThreshold) && (abs(y1 - y0) <= cyThreshold);
}

void updateStartAndLast(LONG x, LONG y)
{
    g_ptStart.x = g_ptEnd.x = x;
    g_ptStart.y = g_ptEnd.y = y;
}

void updateLast(LONG x, LONG y)
{
    g_ptEnd.x = x;
    g_ptEnd.y = y;
}

void ToolBase::reset()
{
    s_pointSP = 0;
    g_ptStart.x = g_ptStart.y = g_ptEnd.x = g_ptEnd.y = -1;
    selectionModel.ResetPtStack();
    if (selectionModel.m_bShow)
    {
        selectionModel.Landing();
        selectionModel.HideSelection();
    }
}

void ToolBase::OnCancelDraw()
{
    reset();
    imageModel.NotifyImageChanged();
}

void ToolBase::OnFinishDraw()
{
    reset();
    imageModel.NotifyImageChanged();
}

void ToolBase::beginEvent()
{
    m_hdc = imageModel.GetDC();
    m_fg = paletteModel.GetFgColor();
    m_bg = paletteModel.GetBgColor();
}

void ToolBase::endEvent()
{
    m_hdc = NULL;
}

void ToolBase::OnDrawSelectionOnCanvas(HDC hdc)
{
    if (!selectionModel.m_bShow)
        return;

    RECT rcSelection = selectionModel.m_rc;
    canvasWindow.ImageToCanvas(rcSelection);

    ::InflateRect(&rcSelection, GRIP_SIZE, GRIP_SIZE);
    drawSizeBoxes(hdc, &rcSelection, TRUE);
}

/* TOOLS ********************************************************/

// TOOL_FREESEL
struct FreeSelTool : ToolBase
{
    BOOL m_bLeftButton = FALSE;

    FreeSelTool() : ToolBase(TOOL_FREESEL)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (!selectionModel.IsLanded())
        {
            selectionModel.DrawBackgroundPoly(hdc, selectionModel.m_rgbBack);
            selectionModel.DrawSelection(hdc, paletteModel.GetBgColor(), toolsModel.IsBackgroundTransparent());
        }

        if (canvasWindow.m_drawing)
        {
            selectionModel.DrawFramePoly(hdc);
        }
    }

    void OnDrawOverlayOnCanvas(HDC hdc) override
    {
        OnDrawSelectionOnCanvas(hdc);
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        selectionModel.Landing();
        if (bLeftButton)
        {
            selectionModel.HideSelection();
            selectionModel.ResetPtStack();
            POINT pt = { x, y };
            selectionModel.PushToPtStack(pt);
        }
        m_bLeftButton = bLeftButton;
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        if (bLeftButton)
        {
            POINT pt = { x, y };
            imageModel.Clamp(pt);
            selectionModel.PushToPtStack(pt);
            imageModel.NotifyImageChanged();
        }
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        if (bLeftButton)
        {
            if (selectionModel.PtStackSize() > 2)
            {
                selectionModel.BuildMaskFromPtStack();
                selectionModel.m_bShow = TRUE;
            }
            else
            {
                selectionModel.ResetPtStack();
                selectionModel.m_bShow = FALSE;
            }
            imageModel.NotifyImageChanged();
        }
        else
        {
            POINT pt = { x, y };
            canvasWindow.ClientToScreen(&pt);
            mainWindow.TrackPopupMenu(pt, 0);
        }
    }

    void OnFinishDraw() override
    {
        selectionModel.Landing();
        ToolBase::OnFinishDraw();
    }

    void OnCancelDraw() override
    {
        selectionModel.HideSelection();
        ToolBase::OnCancelDraw();
    }
};

// TOOL_RECTSEL
struct RectSelTool : ToolBase
{
    BOOL m_bLeftButton = FALSE;

    RectSelTool() : ToolBase(TOOL_RECTSEL)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (!selectionModel.IsLanded())
        {
            selectionModel.DrawBackgroundRect(hdc, selectionModel.m_rgbBack);
            selectionModel.DrawSelection(hdc, paletteModel.GetBgColor(), toolsModel.IsBackgroundTransparent());
        }

        if (canvasWindow.m_drawing)
        {
            RECT rc = selectionModel.m_rc;
            if (!::IsRectEmpty(&rc))
                RectSel(hdc, rc.left, rc.top, rc.right, rc.bottom);
        }
    }

    void OnDrawOverlayOnCanvas(HDC hdc) override
    {
        OnDrawSelectionOnCanvas(hdc);
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        selectionModel.Landing();
        if (bLeftButton)
        {
            selectionModel.HideSelection();
        }
        m_bLeftButton = bLeftButton;
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        if (bLeftButton)
        {
            POINT pt = { x, y };
            imageModel.Clamp(pt);
            selectionModel.SetRectFromPoints(g_ptStart, pt);
            imageModel.NotifyImageChanged();
        }
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        POINT pt = { x, y };
        if (bLeftButton)
        {
            imageModel.Clamp(pt);
            selectionModel.SetRectFromPoints(g_ptStart, pt);
            selectionModel.m_bShow = !selectionModel.m_rc.IsRectEmpty();
            imageModel.NotifyImageChanged();
        }
        else
        {
            canvasWindow.ClientToScreen(&pt);
            mainWindow.TrackPopupMenu(pt, 0);
        }
    }

    void OnFinishDraw() override
    {
        selectionModel.Landing();
        ToolBase::OnFinishDraw();
    }

    void OnCancelDraw() override
    {
        selectionModel.HideSelection();
        ToolBase::OnCancelDraw();
    }
};

struct TwoPointDrawTool : ToolBase
{
    BOOL m_bLeftButton = FALSE;
    BOOL m_bDrawing = FALSE;

    TwoPointDrawTool(TOOLTYPE type) : ToolBase(type)
    {
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        m_bLeftButton = bLeftButton;
        m_bDrawing = TRUE;
        g_ptStart.x = g_ptEnd.x = x;
        g_ptStart.y = g_ptEnd.y = y;
        imageModel.NotifyImageChanged();
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        g_ptEnd.x = x;
        g_ptEnd.y = y;
        imageModel.NotifyImageChanged();
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        g_ptEnd.x = x;
        g_ptEnd.y = y;
        imageModel.PushImageForUndo();
        OnDrawOverlayOnImage(m_hdc);
        m_bDrawing = FALSE;
        imageModel.NotifyImageChanged();
    }

    void OnFinishDraw() override
    {
        m_bDrawing = FALSE;
        ToolBase::OnFinishDraw();
    }

    void OnCancelDraw() override
    {
        m_bDrawing = FALSE;
        ToolBase::OnCancelDraw();
    }
};

struct SmoothDrawTool : ToolBase
{
    SmoothDrawTool(TOOLTYPE type) : ToolBase(type)
    {
    }

    virtual void draw(BOOL bLeftButton, LONG x, LONG y) = 0;

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        imageModel.PushImageForUndo();
        g_ptStart.x = g_ptEnd.x = x;
        g_ptStart.y = g_ptEnd.y = y;
        imageModel.NotifyImageChanged();
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        draw(bLeftButton, x, y);
        imageModel.NotifyImageChanged();
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        draw(bLeftButton, x, y);
        OnFinishDraw();
    }

    void OnFinishDraw() override
    {
        ToolBase::OnFinishDraw();
    }

    void OnCancelDraw() override
    {
        OnButtonUp(FALSE, 0, 0);
        imageModel.Undo(TRUE);
        ToolBase::OnCancelDraw();
    }
};

// TOOL_RUBBER
struct RubberTool : SmoothDrawTool
{
    RubberTool() : SmoothDrawTool(TOOL_RUBBER)
    {
    }

    void draw(BOOL bLeftButton, LONG x, LONG y) override
    {
        if (bLeftButton)
            Erase(m_hdc, g_ptEnd.x, g_ptEnd.y, x, y, m_bg, toolsModel.GetRubberRadius());
        else
            Replace(m_hdc, g_ptEnd.x, g_ptEnd.y, x, y, m_fg, m_bg, toolsModel.GetRubberRadius());
        g_ptEnd.x = x;
        g_ptEnd.y = y;
    }
};

// TOOL_FILL
struct FillTool : ToolBase
{
    FillTool() : ToolBase(TOOL_FILL)
    {
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        imageModel.PushImageForUndo();
        Fill(m_hdc, x, y, bLeftButton ? m_fg : m_bg);
    }
};

// TOOL_COLOR
struct ColorTool : ToolBase
{
    ColorTool() : ToolBase(TOOL_COLOR)
    {
    }

    void fetchColor(BOOL bLeftButton, LONG x, LONG y)
    {
        COLORREF rgbColor;

        if (0 <= x && x < imageModel.GetWidth() && 0 <= y && y < imageModel.GetHeight())
            rgbColor = GetPixel(m_hdc, x, y);
        else
            rgbColor = RGB(255, 255, 255); // Outside is white

        if (bLeftButton)
            paletteModel.SetFgColor(rgbColor);
        else
            paletteModel.SetBgColor(rgbColor);
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        fetchColor(bLeftButton, x, y);
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        fetchColor(bLeftButton, x, y);
        toolsModel.SetActiveTool(toolsModel.GetOldActiveTool());
    }
};

// TOOL_ZOOM
struct ZoomTool : ToolBase
{
    ZoomTool() : ToolBase(TOOL_ZOOM)
    {
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        imageModel.PushImageForUndo();
        if (bLeftButton)
        {
            if (toolsModel.GetZoom() < MAX_ZOOM)
                zoomTo(toolsModel.GetZoom() * 2, x, y);
        }
        else
        {
            if (toolsModel.GetZoom() > MIN_ZOOM)
                zoomTo(toolsModel.GetZoom() / 2, x, y);
        }
    }
};

// TOOL_PEN
struct PenTool : SmoothDrawTool
{
    PenTool() : SmoothDrawTool(TOOL_PEN)
    {
    }

    void draw(BOOL bLeftButton, LONG x, LONG y) override
    {
        COLORREF rgb = bLeftButton ? m_fg : m_bg;
        Line(m_hdc, g_ptEnd.x, g_ptEnd.y, x, y, rgb, 1);
        ::SetPixelV(m_hdc, x, y, rgb);
        g_ptEnd.x = x;
        g_ptEnd.y = y;
    }
};

// TOOL_BRUSH
struct BrushTool : SmoothDrawTool
{
    BrushTool() : SmoothDrawTool(TOOL_BRUSH)
    {
    }

    void draw(BOOL bLeftButton, LONG x, LONG y) override
    {
        COLORREF rgb = bLeftButton ? m_fg : m_bg;
        Brush(m_hdc, g_ptEnd.x, g_ptEnd.y, x, y, rgb, toolsModel.GetBrushStyle());
        g_ptEnd.x = x;
        g_ptEnd.y = y;
    }
};

// TOOL_AIRBRUSH
struct AirBrushTool : SmoothDrawTool
{
    AirBrushTool() : SmoothDrawTool(TOOL_AIRBRUSH)
    {
    }

    void draw(BOOL bLeftButton, LONG x, LONG y) override
    {
        COLORREF rgb = bLeftButton ? m_fg : m_bg;
        Airbrush(m_hdc, x, y, rgb, toolsModel.GetAirBrushWidth());
    }
};

// TOOL_TEXT
struct TextTool : ToolBase
{
    TextTool() : ToolBase(TOOL_TEXT)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (canvasWindow.m_drawing)
        {
            RECT rc = selectionModel.m_rc;
            if (!::IsRectEmpty(&rc))
                RectSel(hdc, rc.left, rc.top, rc.right, rc.bottom);
        }
    }

    void UpdatePoint(LONG x, LONG y)
    {
        POINT pt = { x, y };
        imageModel.Clamp(pt);
        selectionModel.SetRectFromPoints(g_ptStart, pt);
        imageModel.NotifyImageChanged();
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        if (!textEditWindow.IsWindow())
            textEditWindow.Create(canvasWindow);

        UpdatePoint(x, y);
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        UpdatePoint(x, y);
    }

    void draw(HDC hdc)
    {
        CString szText;
        textEditWindow.GetWindowText(szText);

        RECT rc;
        textEditWindow.InvalidateEditRect();
        textEditWindow.GetEditRect(&rc);
        ::InflateRect(&rc, -GRIP_SIZE / 2, -GRIP_SIZE / 2);

        // Draw the text
        INT style = (toolsModel.IsBackgroundTransparent() ? 0 : 1);
        Text(hdc, rc.left, rc.top, rc.right, rc.bottom, m_fg, m_bg, szText,
             textEditWindow.GetFont(), style);
    }

    void quit()
    {
        if (textEditWindow.IsWindow())
            textEditWindow.ShowWindow(SW_HIDE);
        selectionModel.HideSelection();
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        POINT pt = { x, y };
        imageModel.Clamp(pt);
        selectionModel.SetRectFromPoints(g_ptStart, pt);

        BOOL bTextBoxShown = ::IsWindowVisible(textEditWindow);
        if (bTextBoxShown)
        {
            if (textEditWindow.GetWindowTextLength() > 0)
            {
                imageModel.PushImageForUndo();
                draw(m_hdc);
            }
            if (::IsRectEmpty(&selectionModel.m_rc))
            {
                quit();
                return;
            }
        }

        if (registrySettings.ShowTextTool)
        {
            if (!fontsDialog.IsWindow())
                fontsDialog.Create(mainWindow);

            fontsDialog.ShowWindow(SW_SHOWNOACTIVATE);
        }

        RECT rc = selectionModel.m_rc;

        // Enlarge if tool small
        INT cxMin = CX_MINTEXTEDIT, cyMin = CY_MINTEXTEDIT;
        if (selectionModel.m_rc.IsRectEmpty())
        {
            SetRect(&rc, x, y, x + cxMin, y + cyMin);
        }
        else
        {
            if (rc.right - rc.left < cxMin)
                rc.right = rc.left + cxMin;
            if (rc.bottom - rc.top < cyMin)
                rc.bottom = rc.top + cyMin;
        }

        if (!textEditWindow.IsWindow())
            textEditWindow.Create(canvasWindow);

        textEditWindow.SetWindowText(NULL);
        textEditWindow.ValidateEditRect(&rc);
        textEditWindow.ShowWindow(SW_SHOWNOACTIVATE);
        textEditWindow.SetFocus();
    }

    void OnFinishDraw() override
    {
        if (textEditWindow.GetWindowTextLength() > 0)
        {
            imageModel.PushImageForUndo();
            draw(m_hdc);
        }
        quit();
        ToolBase::OnFinishDraw();
    }

    void OnCancelDraw() override
    {
        quit();
        ToolBase::OnCancelDraw();
    }
};

// TOOL_LINE
struct LineTool : TwoPointDrawTool
{
    LineTool() : TwoPointDrawTool(TOOL_LINE)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (!m_bDrawing)
            return;
        if (GetAsyncKeyState(VK_SHIFT) < 0)
            roundTo8Directions(g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y);
        COLORREF rgb = m_bLeftButton ? m_fg : m_bg;
        Line(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, rgb, toolsModel.GetLineWidth());
    }
};

// TOOL_BEZIER
struct BezierTool : ToolBase
{
    BOOL m_bLeftButton = FALSE;
    BOOL m_bDrawing = FALSE;

    BezierTool() : ToolBase(TOOL_BEZIER)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc)
    {
        if (!m_bDrawing)
            return;

        COLORREF rgb = (m_bLeftButton ? m_fg : m_bg);
        switch (s_pointSP)
        {
            case 1:
                Line(hdc, s_pointStack[0].x, s_pointStack[0].y, s_pointStack[1].x, s_pointStack[1].y, rgb,
                     toolsModel.GetLineWidth());
                break;
            case 2:
                Bezier(hdc, s_pointStack[0], s_pointStack[2], s_pointStack[2], s_pointStack[1], rgb, toolsModel.GetLineWidth());
                break;
            case 3:
                Bezier(hdc, s_pointStack[0], s_pointStack[2], s_pointStack[3], s_pointStack[1], rgb, toolsModel.GetLineWidth());
                break;
        }
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        m_bLeftButton = bLeftButton;

        if (!m_bDrawing)
        {
            m_bDrawing = TRUE;
            s_pointStack[s_pointSP].x = s_pointStack[s_pointSP + 1].x = x;
            s_pointStack[s_pointSP].y = s_pointStack[s_pointSP + 1].y = y;
            ++s_pointSP;
        }
        else
        {
            ++s_pointSP;
            s_pointStack[s_pointSP].x = x;
            s_pointStack[s_pointSP].y = y;
        }

        imageModel.NotifyImageChanged();
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        s_pointStack[s_pointSP].x = x;
        s_pointStack[s_pointSP].y = y;
        imageModel.NotifyImageChanged();
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        s_pointStack[s_pointSP].x = x;
        s_pointStack[s_pointSP].y = y;
        if (s_pointSP >= 3)
        {
            OnFinishDraw();
            return;
        }
        imageModel.NotifyImageChanged();
    }

    void OnCancelDraw() override
    {
        m_bDrawing = FALSE;
        ToolBase::OnCancelDraw();
    }

    void OnFinishDraw() override
    {
        imageModel.PushImageForUndo();
        OnDrawOverlayOnImage(m_hdc);
        m_bDrawing = FALSE;
        ToolBase::OnFinishDraw();
    }
};

// TOOL_RECT
struct RectTool : TwoPointDrawTool
{
    RectTool() : TwoPointDrawTool(TOOL_RECT)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (!m_bDrawing)
            return;
        if (GetAsyncKeyState(VK_SHIFT) < 0)
            regularize(g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y);
        if (m_bLeftButton)
            Rect(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, m_fg, m_bg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle());
        else
            Rect(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, m_bg, m_fg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle());
    }
};

// TOOL_SHAPE
struct ShapeTool : ToolBase
{
    BOOL m_bLeftButton = FALSE;
    BOOL m_bClosed = FALSE;

    ShapeTool() : ToolBase(TOOL_SHAPE)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc)
    {
        if (s_pointSP <= 0)
            return;

        if (m_bLeftButton)
            Poly(hdc, s_pointStack, s_pointSP + 1, m_fg, m_bg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle(), m_bClosed, FALSE);
        else
            Poly(hdc, s_pointStack, s_pointSP + 1, m_bg, m_fg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle(), m_bClosed, FALSE);
    }

    void OnButtonDown(BOOL bLeftButton, LONG x, LONG y, BOOL bDoubleClick) override
    {
        m_bLeftButton = bLeftButton;
        m_bClosed = FALSE;

        if ((s_pointSP > 0) && (GetAsyncKeyState(VK_SHIFT) < 0))
            roundTo8Directions(s_pointStack[s_pointSP - 1].x, s_pointStack[s_pointSP - 1].y, x, y);

        s_pointStack[s_pointSP].x = x;
        s_pointStack[s_pointSP].y = y;

        if (s_pointSP && bDoubleClick)
        {
            OnFinishDraw();
            return;
        }

        if (s_pointSP == 0)
        {
            s_pointSP++;
            s_pointStack[s_pointSP].x = x;
            s_pointStack[s_pointSP].y = y;
        }

        imageModel.NotifyImageChanged();
    }

    void OnMouseMove(BOOL bLeftButton, LONG x, LONG y) override
    {
        if ((s_pointSP > 0) && (GetAsyncKeyState(VK_SHIFT) < 0))
            roundTo8Directions(s_pointStack[s_pointSP - 1].x, s_pointStack[s_pointSP - 1].y, x, y);

        s_pointStack[s_pointSP].x = x;
        s_pointStack[s_pointSP].y = y;

        imageModel.NotifyImageChanged();
    }

    void OnButtonUp(BOOL bLeftButton, LONG x, LONG y) override
    {
        if ((s_pointSP > 0) && (GetAsyncKeyState(VK_SHIFT) < 0))
            roundTo8Directions(s_pointStack[s_pointSP - 1].x, s_pointStack[s_pointSP - 1].y, x, y);

        m_bClosed = FALSE;
        if (nearlyEqualPoints(x, y, s_pointStack[0].x, s_pointStack[0].y))
        {
            OnFinishDraw();
            return;
        }
        else
        {
            s_pointSP++;
            s_pointStack[s_pointSP].x = x;
            s_pointStack[s_pointSP].y = y;
        }

        if (s_pointSP == _countof(s_pointStack))
            s_pointSP--;

        imageModel.NotifyImageChanged();
    }

    void OnCancelDraw() override
    {
        ToolBase::OnCancelDraw();
    }

    void OnFinishDraw() override
    {
        if (s_pointSP)
        {
            --s_pointSP;
            m_bClosed = TRUE;

            imageModel.PushImageForUndo();
            OnDrawOverlayOnImage(m_hdc);
        }

        m_bClosed = FALSE;
        s_pointSP = 0;

        ToolBase::OnFinishDraw();
    }
};

// TOOL_ELLIPSE
struct EllipseTool : TwoPointDrawTool
{
    EllipseTool() : TwoPointDrawTool(TOOL_ELLIPSE)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (!m_bDrawing)
            return;
        if (GetAsyncKeyState(VK_SHIFT) < 0)
            regularize(g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y);
        if (m_bLeftButton)
            Ellp(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, m_fg, m_bg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle());
        else
            Ellp(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, m_bg, m_fg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle());
    }
};

// TOOL_RRECT
struct RRectTool : TwoPointDrawTool
{
    RRectTool() : TwoPointDrawTool(TOOL_RRECT)
    {
    }

    void OnDrawOverlayOnImage(HDC hdc) override
    {
        if (!m_bDrawing)
            return;
        if (GetAsyncKeyState(VK_SHIFT) < 0)
            regularize(g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y);
        if (m_bLeftButton)
            RRect(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, m_fg, m_bg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle());
        else
            RRect(hdc, g_ptStart.x, g_ptStart.y, g_ptEnd.x, g_ptEnd.y, m_bg, m_fg, toolsModel.GetLineWidth(), toolsModel.GetShapeStyle());
    }
};

/*static*/ ToolBase*
ToolBase::createToolObject(TOOLTYPE type)
{
    switch (type)
    {
        case TOOL_FREESEL:  return new FreeSelTool();
        case TOOL_RECTSEL:  return new RectSelTool();
        case TOOL_RUBBER:   return new RubberTool();
        case TOOL_FILL:     return new FillTool();
        case TOOL_COLOR:    return new ColorTool();
        case TOOL_ZOOM:     return new ZoomTool();
        case TOOL_PEN:      return new PenTool();
        case TOOL_BRUSH:    return new BrushTool();
        case TOOL_AIRBRUSH: return new AirBrushTool();
        case TOOL_TEXT:     return new TextTool();
        case TOOL_LINE:     return new LineTool();
        case TOOL_BEZIER:   return new BezierTool();
        case TOOL_RECT:     return new RectTool();
        case TOOL_SHAPE:    return new ShapeTool();
        case TOOL_ELLIPSE:  return new EllipseTool();
        case TOOL_RRECT:    return new RRectTool();
    }
    UNREACHABLE;
    return NULL;
}