path: root/alf/alfnotebook.cpp

#include "alfpriv.h"
#include "alfcompat.h"

#define ALF_NB_ADDTAB       (ALF_WM__BASE + 200)
#define ALF_NB_TABCOUNT     (ALF_WM__BASE + 201)
#define ALF_NB_GETPANEL     (ALF_WM__BASE + 202)
#define ALF_NB_GETSELINDEX  (ALF_WM__BASE + 203)
#define ALF_NB_GETPNLINDEX  (ALF_WM__BASE + 204)

#define TABP_BODY 10

TCHAR *_alf_notebookClass = NULL;

typedef struct {
    HWND   hwndTabCtrl;
    HTHEME hTheme;
} ALFNotebookPriv;

static ALFNotebookPriv *
ALF_Notebook_CreatePriv(void)
{
    ALFNotebookPriv *priv = ALF_New(ALFNotebookPriv, 1);

    return priv;
}

void
ALF_Notebook_FreePriv(ALFNotebookPriv *priv)
{
    ALF_Compat_CloseThemeData(priv->hTheme);
    priv->hTheme = NULL;
    ALF_Free(priv);
}

static int
ALF_Notebook_InternalTabCount(HWND notebook, HWND tabControl)
{
    (void)notebook;
    return TabCtrl_GetItemCount(tabControl);
}

static HWND
ALF_Notebook_InternalTabPanel(HWND notebook, HWND tabControl, int index)
{
    (void)notebook;

    TCITEM tci;
    ZeroMemory(&tci, sizeof(tci));

    tci.mask = TCIF_PARAM;

    if (TabCtrl_GetItem(tabControl, index, &tci)) {
        return (HWND)tci.lParam;
    } else {
        return NULL;
    }
}

static int
ALF_Notebook_InternalSelectedIndex(HWND notebook, HWND tabControl)
{
    (void)notebook;

    return TabCtrl_GetCurSel(tabControl);
}


static void
ALF_Notebook_InternalHandleTabChange(HWND hwndNotebook, HWND hwndTabCtrl)
{
    int selectedIndex = ALF_Notebook_InternalSelectedIndex(hwndNotebook, hwndTabCtrl);
    int n = ALF_Notebook_InternalTabCount(hwndNotebook, hwndTabCtrl);

    for (int i = 0; i < n; ++i) {
        if (i == selectedIndex) {
            SetWindowPos(ALF_Notebook_InternalTabPanel(hwndNotebook, hwndTabCtrl, i), HWND_TOP, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOMOVE | SWP_SHOWWINDOW);
        } else {
            ShowWindow(ALF_Notebook_InternalTabPanel(hwndNotebook, hwndTabCtrl, i), SW_HIDE);
        }
    }
}

static void
ALF_Notebook_InternalHandleThemeChange(HWND hwndNotebook, ALFNotebookPriv *priv)
{
    ALF_Compat_CloseThemeData(priv->hTheme);
    priv->hTheme = NULL;
    if (ALF_Compat_IsAppThemed())
        priv->hTheme = ALF_Compat_OpenThemeData(hwndNotebook, L"TAB");

    InvalidateRect(hwndNotebook, NULL, TRUE);
}


static HWND
ALF_Notebook_InternalAddTab(HWND notebook, HWND tabControl, const TCHAR *title)
{
    RECT r;
    GetClientRect(tabControl, &r);
    TabCtrl_AdjustRect(tabControl, FALSE, &r);
    MapWindowRect(tabControl, notebook, &r);

    HWND hwndPanel = ALF_CreatePanelWindow(notebook, (WORD)-1);
    SetWindowPos(hwndPanel, NULL, r.left, r.top, r.right - r.left, r.bottom - r.top, SWP_HIDEWINDOW|SWP_NOACTIVATE|SWP_NOZORDER|SWP_NOOWNERZORDER);

    TCITEM tie;
    ZeroMemory(&tie, sizeof(tie));

    tie.mask = TCIF_TEXT | TCIF_PARAM;
    tie.pszText = (TCHAR*)title;
    tie.lParam = (LPARAM)hwndPanel;

    TabCtrl_InsertItem(tabControl, ALF_Notebook_InternalTabCount(notebook, tabControl), &tie);
    ALF_Notebook_InternalHandleTabChange(notebook, tabControl);

    return hwndPanel;
}


static LRESULT CALLBACK
ALF_Notebook_TabCtrlSubclassProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam, UINT_PTR uIdSubclass, DWORD_PTR dwRefData)
{
    (void)uIdSubclass;
    (void)dwRefData;

    if (uMsg == WM_ERASEBKGND) {
        // see: http://www.virtualdub.org/blog/pivot/entry.php?id=291
        HDC hdc = (HDC)wParam;

        RECT r;
        GetClientRect(hwnd, &r);
        TabCtrl_AdjustRect(hwnd, FALSE, &r);

        ExcludeClipRect(hdc, r.left, r.top, r.right, r.bottom);
    }
    if (uMsg == 0x2000 + WM_NOTIFY) {
        NMHDR *pnmh = (NMHDR*)lParam;
        if (pnmh->code == TCN_SELCHANGE) {
            ALF_Notebook_InternalHandleTabChange(GetParent(hwnd), hwnd);

            return TRUE;
        }
    }
    if (uMsg == WM_DESTROY) {
        ALF_Compat_RemoveWindowSubclass(hwnd, ALF_Notebook_TabCtrlSubclassProc, 0);
    }

    return ALF_Compat_DefSubclassProc(hwnd, uMsg, wParam, lParam);
}

static void
ALF_Notebook_InternalPaint(HWND hwnd, ALFNotebookPriv *priv, HDC dc, RECT *f)
{
    if (priv->hTheme != NULL) {
        RECT r;
        GetClientRect(priv->hwndTabCtrl, &r);
        TabCtrl_AdjustRect(priv->hwndTabCtrl, FALSE, &r);
        MapWindowRect(priv->hwndTabCtrl, hwnd, &r);

        ALF_Compat_DrawThemeBackground(priv->hTheme, dc, TABP_BODY, 0, &r, NULL);

        ExcludeClipRect(dc, r.left, r.top, r.right, r.bottom);
    }

    FillRect(dc, f, GetSysColorBrush(COLOR_BTNFACE));
}

static LRESULT CALLBACK
ALF__NotebookWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    if (ALF_ShouldMessageBubble(hwnd, uMsg, wParam, lParam))
        return SendMessage(GetParent(hwnd), uMsg, wParam, lParam);

    ALFNotebookPriv *priv = (ALFNotebookPriv *)GetWindowLongPtr(hwnd, 0);

    if (uMsg == WM_CREATE && priv == NULL) {
        priv = ALF_Notebook_CreatePriv();
        priv->hwndTabCtrl = CreateWindowEx(0,
                                           WC_TABCONTROL,
                                           TEXT(""),
                                           WS_CHILD | WS_VISIBLE | WS_TABSTOP | WS_CLIPSIBLINGS,
                                           0, 0, 100, 100,
                                           hwnd,
                                           NULL,
                                           ALF_HINSTANCE,
                                           NULL);

        ALF_Compat_SetWindowSubclass(priv->hwndTabCtrl, ALF_Notebook_TabCtrlSubclassProc, 0, 0);

        SetWindowLongPtr(hwnd, 0, (LONG_PTR)priv);
        ALF_Notebook_InternalHandleThemeChange(hwnd, priv);
    }

    if (!priv)
        return DefWindowProc(hwnd, uMsg, wParam, lParam);

    if (uMsg == WM_DESTROY) {
        ALF_Notebook_FreePriv(priv);
        SetWindowLongPtr(hwnd, 0, 0);
        priv = NULL;
    } else if (uMsg == ALF_WM_APPLYFONTS) {
        int n = ALF_Notebook_InternalTabCount(hwnd, priv->hwndTabCtrl);
        for (int i = 0; i < n; ++i) {
            HWND p = ALF_Notebook_InternalTabPanel(hwnd, priv->hwndTabCtrl, i);
            SendMessage(p, ALF_WM_APPLYFONTS, wParam, lParam);
        }

        return TRUE;
    } else if (uMsg == WM_SETFONT) {
        SendMessage(priv->hwndTabCtrl, WM_SETFONT, wParam, lParam);

        int n = ALF_Notebook_InternalTabCount(hwnd, priv->hwndTabCtrl);
        for (int i = 0; i < n; ++i) {
            HWND p = ALF_Notebook_InternalTabPanel(hwnd, priv->hwndTabCtrl, i);
            SendMessage(p, WM_SETFONT, wParam, lParam);
        }
    } else if (uMsg == ALF_WM_QUERYSIZE) {
        int n = ALF_Notebook_InternalTabCount(hwnd, priv->hwndTabCtrl);
        RECT r = { 0, 0, 0, 0 };
        for (int i = 0; i < n; ++i) {
            HWND p = ALF_Notebook_InternalTabPanel(hwnd, priv->hwndTabCtrl, i);
            SIZE s = { 0, 0 };
            SendMessage(p, ALF_WM_QUERYSIZE, 0, (LPARAM)&s);
            if (s.cx > r.right)
                r.right = s.cx;
            if (s.cy > r.bottom)
                r.bottom = s.cy;
        }

        if (!r.right)
            r.right = 100; // FIXME!
        if (!r.bottom)
            r.bottom = 100; // FIXME!

        TabCtrl_AdjustRect(priv->hwndTabCtrl, TRUE, &r);

        SIZE *ps = (SIZE*)lParam;
        if (!ps->cx) {
            ps->cx = r.right - r.left;
        }

        if (!ps->cy) {
            ps->cy = r.bottom - r.top;
        }

        return 0;
    } else if (uMsg == WM_WINDOWPOSCHANGED) {
        WINDOWPOS *pos = (WINDOWPOS *)lParam;
        if (!(pos->flags & SWP_NOSIZE)) {
            RECT r;
            GetClientRect(hwnd, &r);
            MoveWindow(priv->hwndTabCtrl, 0, 0, r.right - r.left, r.bottom - r.top, FALSE);
            InvalidateRect(priv->hwndTabCtrl, NULL, TRUE);

            GetClientRect(priv->hwndTabCtrl, &r);
            TabCtrl_AdjustRect(priv->hwndTabCtrl, FALSE, &r);
            MapWindowRect(priv->hwndTabCtrl, hwnd, &r);

            int n = ALF_Notebook_InternalTabCount(hwnd, priv->hwndTabCtrl);
            for (int i = 0; i < n; ++i) {
                HWND p = ALF_Notebook_InternalTabPanel(hwnd, priv->hwndTabCtrl, i);

                MoveWindow(p, r.left, r.top, r.right - r.left, r.bottom - r.top, FALSE);
                InvalidateRect(p, NULL, TRUE);
            }
        }
    } else if (uMsg == ALF_NB_ADDTAB) {
        return (LRESULT)ALF_Notebook_InternalAddTab(hwnd, priv->hwndTabCtrl, (TCHAR *)lParam);
    } else if (uMsg == ALF_NB_TABCOUNT) {
        return (LRESULT)ALF_Notebook_InternalTabCount(hwnd, priv->hwndTabCtrl);
    } else if (uMsg == ALF_NB_GETPANEL) {
        return (LRESULT)ALF_Notebook_InternalTabPanel(hwnd, priv->hwndTabCtrl, (int)lParam);
    } else if (uMsg == ALF_NB_GETSELINDEX) {
        return (LRESULT)ALF_Notebook_InternalSelectedIndex(hwnd, priv->hwndTabCtrl);
    } else if (uMsg == WM_THEMECHANGED) {
        ALF_Notebook_InternalHandleThemeChange(hwnd, priv);
    } else if (uMsg == WM_ERASEBKGND) {
        return TRUE;
    } else if (uMsg == WM_PRINTCLIENT) {
        RECT r = {0, 0, 0, 0};
        GetClientRect(hwnd, &r);

        ALF_Notebook_InternalPaint(hwnd, priv, (HDC)wParam, &r);
        return TRUE;
    } else if (uMsg == WM_PAINT) {
        PAINTSTRUCT ps;
        HDC dc = BeginPaint(hwnd, &ps);

        ALF_Notebook_InternalPaint(hwnd, priv, dc, &ps.rcPaint);

        EndPaint(hwnd, &ps);
        return 0;
    }

    return DefWindowProc(hwnd, uMsg, wParam, lParam);
}

void
ALF_RegisterNotebookClass(void)
{
    WNDCLASS cls;
    ZeroMemory(&cls, sizeof(cls));

    TCHAR  classNameBuf[256];
    ALF_BuildUniqueName(classNameBuf, TEXT("ALFNotebook."), (ULONG_PTR)&_alf_notebookClass);

    cls.hInstance = ALF_HINSTANCE;
    cls.hCursor = LoadCursor(NULL, (LPTSTR)IDC_ARROW);
    cls.lpszClassName = classNameBuf;
    cls.cbWndExtra = sizeof(ALFNotebookPriv*);
    cls.lpfnWndProc = ALF__NotebookWindowProc;

    ATOM classatom = RegisterClass(&cls);
    if (!classatom)
        MessageBox(NULL, TEXT("FATAL: Could not register notebook class"), NULL, MB_OK);

    _alf_notebookClass = MAKEINTATOM(classatom);
}

// tab control
HWND
ALF_AddNotebook(HWND parent, WORD id, UINT x, UINT y)
{
    HWND hwndNtbk  = CreateWindowEx(WS_EX_CONTROLPARENT,
                                    _alf_notebookClass,
                                    TEXT(""),
                                    WS_CHILD | WS_VISIBLE | WS_CLIPCHILDREN,
                                    0, 0, 100, 100,
                                    parent,
                                    (HMENU)(int)id,
                                    ALF_HINSTANCE,
                                    NULL);

    ALFWidgetLayoutParams p;
    ZeroMemory(&p, sizeof(p));
    p.hwnd = hwndNtbk;
    p.x = x;
    p.y = y;
    p.width = 0;
    p.height = 0;
    p.flags = ALF_QUERYSIZE | ALF_MESSAGEFONT | ALF_SENDAPPLYFONTS;

    ALF_AddWidgetEx(parent, &p);

    return hwndNtbk;
}

HWND
ALF_NotebookAddTab(HWND notebook, const TCHAR *title)
{
    return (HWND)SendMessage(notebook, ALF_NB_ADDTAB, 0, (LPARAM)title);
}

int
ALF_NotebookTabCount(HWND notebook)
{
    return (int)SendMessage(notebook, ALF_NB_TABCOUNT, 0, 0);
}

int
ALF_NotebookTabIndex(HWND notebook, HWND tabPanel);

HWND
ALF_NotebookTabPanel(HWND notebook, int index)
{
    return (HWND)SendMessage(notebook, ALF_NB_GETPANEL, 0, (WPARAM)index);
}

int
ALF_NotebookSelectedIndex(HWND notebook)
{
    return (int)SendMessage(notebook, ALF_NB_GETSELINDEX, 0, 0);
}

HWND
ALF_NotebookSelectedPanel(HWND notebook)
{
    int i = ALF_NotebookSelectedIndex(notebook);
    if (i >= 0) {
        return ALF_NotebookTabPanel(notebook, i);
    } else {
        return NULL;
    }
}