#include "stdafx.h"
#include "ApredTreeCtrl2.h"


#define ROFFSET 7
#define WIDE	10
#define WIDE2	5
#define EXPANDED_WIDE	8

#define BADGE_HIDE                  0
#define BADGE_DOT                   1
#define BADGE_NUMBER                2
#define BADGE_DOT_WIDTH             12
#define BADGE_NUMBER_WIDTH          20


IMPLEMENT_DYNAMIC(CApredTreeCtrl2, CTreeCtrl)

CApredTreeCtrl2::CApredTreeCtrl2()
{
	m_hBrushItem[0] = NULL;
	m_hBrushItem[1] = NULL;
	m_hBrushItem[2] = NULL;
	m_crItemBk[0] = GetSysColor(COLOR_WINDOW);
	m_crItemBk[1] = GetSysColor(COLOR_HIGHLIGHT);
	m_crItemBk[2] = RGB(204, 206, 219);
	m_crItemBtn[0] = RGB(30, 30, 30);
	m_crItemBtn[1] = RGB(255, 255, 255);
	m_crItemBtn[2] = RGB(255, 255, 255);
}

CApredTreeCtrl2::~CApredTreeCtrl2()
{
	if (m_hBrushItem[0] != NULL) {
		::DeleteObject(m_hBrushItem[0]);
	}
	if (m_hBrushItem[1] != NULL) {
		::DeleteObject(m_hBrushItem[1]);
	}
	if (m_hBrushItem[2] != NULL) {
		::DeleteObject(m_hBrushItem[2]);
	}
}

BEGIN_MESSAGE_MAP(CApredTreeCtrl2, CTreeCtrl)
	ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, &CApredTreeCtrl2::OnNMCustomdraw)
END_MESSAGE_MAP()


void CApredTreeCtrl2::OnNMCustomdraw(NMHDR* pNMHDR, LRESULT* pResult)
{
	LPNMCUSTOMDRAW pNMCD = reinterpret_cast<LPNMCUSTOMDRAW>(pNMHDR);
    LPNMTVCUSTOMDRAW lpnmcd = (LPNMTVCUSTOMDRAW)pNMCD;
    HTREEITEM hItem;
    UINT unt;
    *pResult = 0;


    switch (lpnmcd->nmcd.dwDrawStage)
    {
    case CDDS_PREPAINT:
        *pResult = CDRF_NOTIFYITEMDRAW;
        m_items.clear();
        break;

    case CDDS_ITEMPREPAINT:
        m_items[(HTREEITEM)lpnmcd->nmcd.dwItemSpec] = lpnmcd->iLevel;
        hItem = (HTREEITEM)lpnmcd->nmcd.dwItemSpec;
        if (hItem != NULL) {
            if ( (GetItemState(hItem, TVIS_SELECTED) & TVIS_SELECTED)== TVIS_SELECTED) {
                if (GetFocus() != this) {
                    lpnmcd->clrTextBk = RGB(204, 206, 219);
                }
            }
        }
        *pResult = CDRF_NEWFONT | CDRF_NOTIFYPOSTPAINT;   // ¶©ÔÄ»æÖƽáÊø֪ͨ

        break;


    case CDDS_ITEMPOSTPAINT: // Ïî»æÖƽáÊø֪ͨ
		unt = GetIndent();
		hItem = (HTREEITEM)lpnmcd->nmcd.dwItemSpec;
		auto iter = m_items.find(hItem);
		if (iter != m_items.end()) {
			CRect button_rect = pNMCD->rc;
			button_rect.right = button_rect.left + unt + 5;
			int off = unt * iter->second;
			button_rect.OffsetRect(off, 0);
			DrawItemButton(hItem, pNMCD->hdc, &button_rect);

			// ÊÇ·ñÓÐСԲµã
			CRect rcItem = pNMCD->rc;
			Gdiplus::Graphics graphics(pNMCD->hdc);
			graphics.SetSmoothingMode(Gdiplus::SmoothingModeAntiAlias);
			if (m_badges.find(hItem) != m_badges.end()) {
				BADGE& badge = m_badges[hItem];
				if (badge.type == BADGE_DOT) {
					Gdiplus::SolidBrush brush(Gdiplus::Color(GetRValue(badge.badgeBackground), GetGValue(badge.badgeBackground), GetBValue(badge.badgeBackground)));
					int x = rcItem.right - 18 - BADGE_DOT_WIDTH;
					int y = rcItem.top + (rcItem.Height() - BADGE_DOT_WIDTH) / 2;
					graphics.FillEllipse(&brush, x, y, BADGE_DOT_WIDTH, BADGE_DOT_WIDTH);
				}
				else if (badge.type == BADGE_NUMBER) {
					Gdiplus::SolidBrush brush(Gdiplus::Color(GetRValue(badge.badgeBackground), GetGValue(badge.badgeBackground), GetBValue(badge.badgeBackground)));
					int x = rcItem.right - 18 - BADGE_NUMBER_WIDTH;
					int y = rcItem.top + (rcItem.Height() - BADGE_NUMBER_WIDTH) / 2;
					graphics.FillEllipse(&brush, x, y, BADGE_NUMBER_WIDTH, BADGE_NUMBER_WIDTH);
					RECT rcBadge;
					rcBadge.left = x;
					rcBadge.right = rcBadge.left + BADGE_NUMBER_WIDTH;
					rcBadge.top = y;
					rcBadge.bottom = rcBadge.top + BADGE_NUMBER_WIDTH;
					::SetTextColor(pNMCD->hdc, badge.badgeForeground);
					char szBuffer[32];
					sprintf_s(szBuffer, 32, "%d%s", min(badge.number, 9), badge.number > 9 ? "+" : "");
					DrawText(pNMCD->hdc, szBuffer, (int)strlen(szBuffer), &rcBadge, DT_CENTER | DT_VCENTER | DT_SINGLELINE);
				}
			}

			*pResult = CDRF_SKIPDEFAULT;
		}

        break;
    }
}

void CApredTreeCtrl2::DrawItemButton(HTREEITEM hItem, HDC hDC, CRect* pRect)
{
	if (!ItemHasChildren(hItem)) {
		return;
	}


	// °´Å¥ÒªË¢Ò»ÏÂ
	::FillRect(hDC, pRect, GetItemBkBrush(hItem));
	if ((GetItemState(hItem, TVIS_EXPANDED) & TVIS_EXPANDED) == TVIS_EXPANDED) {
		int nBottomOffset = (pRect->Height() - EXPANDED_WIDE) / 2;
		POINT pt[3];
		pt[0].x = pRect->right - ROFFSET - EXPANDED_WIDE;
		pt[0].y = pRect->bottom - nBottomOffset;
		pt[1].x = pRect->right - ROFFSET;
		pt[1].y = pRect->bottom - nBottomOffset;
		pt[2].x = pRect->right - ROFFSET;
		pt[2].y = pRect->bottom - nBottomOffset - EXPANDED_WIDE;

		HBRUSH hBrush = GetItemBtnBrush(hItem);
		HPEN hPen = GetItemBtnPen(hItem);
		HBRUSH hOldBrush = (HBRUSH)::SelectObject(hDC, hBrush);
		HPEN hOldPen = (HPEN)::SelectObject(hDC, hPen);
		::Polygon(hDC, pt, 3);
		::SelectObject(hDC, hOldBrush);
		::SelectObject(hDC, hOldPen);
		::DeleteObject(hBrush);
		::DeleteObject(hPen);
	}
	else {
		int nBottomOffset = (pRect->Height() - WIDE) / 2;
		POINT pt[3];
		pt[0].x = pRect->right - ROFFSET - WIDE2;
		pt[0].y = pRect->bottom - nBottomOffset - WIDE;
		pt[1].x = pRect->right - ROFFSET - WIDE2;
		pt[1].y = pRect->bottom - nBottomOffset;
		pt[2].x = pRect->right - ROFFSET;
		pt[2].y = pRect->bottom - nBottomOffset - WIDE2;

		HBRUSH hBrush = GetItemBtnBrush(hItem);
		HPEN hPen = GetItemBtnPen(hItem);
		HBRUSH hOldBrush = (HBRUSH)::SelectObject(hDC, hBrush);
		HPEN hOldPen = (HPEN)::SelectObject(hDC, hPen);
		::Polygon(hDC, pt, 3);
		::SelectObject(hDC, hOldBrush);
		::SelectObject(hDC, hOldPen);
		::DeleteObject(hBrush);
		::DeleteObject(hPen);
	}
}

COLORREF CApredTreeCtrl2::SetBkColor(COLORREF crColor)
{
	if (m_hBrushItem[0] != NULL) {
		::DeleteObject(m_hBrushItem[0]);
	}
	m_hBrushItem[0] = CreateSolidBrush(crColor);

	return CTreeCtrl::SetBkColor(crColor);
}

int CApredTreeCtrl2::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CTreeCtrl::OnCreate(lpCreateStruct) == -1)
		return -1;

	m_crItemBk[0] = GetBkColor();
	m_crItemBk[1] = GetSysColor(COLOR_HIGHLIGHT);
	m_hBrushItem[0] = CreateSolidBrush(m_crItemBk[0]);
	m_hBrushItem[1] = CreateSolidBrush(m_crItemBk[1]);
	m_hBrushItem[2] = CreateSolidBrush(m_crItemBk[2]);

	return 0;
}

HBRUSH CApredTreeCtrl2::GetItemBkBrush(HTREEITEM hItem)
{
	BOOL bSelected = (GetItemState(hItem, TVIS_SELECTED) & TVIS_SELECTED) == TVIS_SELECTED;
	BOOL bHilited = (GetItemState(hItem, TVIS_DROPHILITED) & TVIS_DROPHILITED) == TVIS_DROPHILITED;;
	HTREEITEM hClickedItem = GetDropHilightItem();


	if (bHilited) {
		return m_hBrushItem[1];
	}

	if (!bSelected) {
		return m_hBrushItem[0];
	}

	if (GetFocus() == this) {
		if (hClickedItem == NULL) {
			return m_hBrushItem[1];
		}
		else {
			return m_hBrushItem[0];
		}
	}


	return m_hBrushItem[2];
}

HBRUSH CApredTreeCtrl2::GetItemBtnBrush(HTREEITEM hItem)
{
	BOOL bExpanded = (GetItemState(hItem, TVIS_EXPANDED) & TVIS_EXPANDED) == TVIS_EXPANDED;
	BOOL bSelected = (GetItemState(hItem, TVIS_SELECTED) & TVIS_SELECTED) == TVIS_SELECTED;
	BOOL bHilited = (GetItemState(hItem, TVIS_DROPHILITED) & TVIS_DROPHILITED) == TVIS_DROPHILITED;;
	HTREEITEM hClickedItem = GetDropHilightItem();


	// Õ¹¿ªÊ±ÊµÐÄÈý½Ç
	if (bExpanded) {
		if (bHilited) {
			return CreateSolidBrush(m_crItemBtn[1]);
		}

		if (!bSelected) {
			return CreateSolidBrush(m_crItemBtn[0]);
		}

		if (GetFocus() == this && hClickedItem == NULL) {
			return CreateSolidBrush(m_crItemBtn[1]);
		}

		return CreateSolidBrush(m_crItemBtn[0]);
	}
	else {
		// ÊÕÆðʱ¿ÕÐÄÈý½Ç
		if (bHilited) {
			return CreateSolidBrush(m_crItemBk[1]);
		}

		if (!bSelected) {
			return CreateSolidBrush(m_crItemBk[0]);
		}

		if (GetFocus() == this && hClickedItem == NULL) {
			return CreateSolidBrush(m_crItemBk[1]);
		}

		return CreateSolidBrush(m_crItemBk[2]);
	}
}

HPEN CApredTreeCtrl2::GetItemBtnPen(HTREEITEM hItem)
{
	BOOL bExpanded = (GetItemState(hItem, TVIS_EXPANDED) & TVIS_EXPANDED) == TVIS_EXPANDED;
	BOOL bSelected = (GetItemState(hItem, TVIS_SELECTED) & TVIS_SELECTED) == TVIS_SELECTED;
	BOOL bHilited = (GetItemState(hItem, TVIS_DROPHILITED) & TVIS_DROPHILITED) == TVIS_DROPHILITED;;
	HTREEITEM hClickedItem = GetDropHilightItem();


	// Õ¹¿ªÊ±ÊµÐÄÈý½Ç
	if (bExpanded) {
		if (bHilited) {
			return CreatePen(PS_SOLID, 1, m_crItemBtn[1]);
		}

		if (!bSelected) {
			return CreatePen(PS_SOLID, 1, m_crItemBtn[0]);
		}

		if (GetFocus() == this && hClickedItem == NULL) {
			return CreatePen(PS_SOLID, 1, m_crItemBtn[1]);
		}

		return CreatePen(PS_SOLID, 1, m_crItemBtn[0]);
	}
	else {
		// ÊÕÆðʱ¿ÕÐÄÈý½Ç
		if (bHilited) {
			return CreatePen(PS_SOLID, 1, m_crItemBtn[1]);
		}

		if (!bSelected) {
			return CreatePen(PS_SOLID, 1, m_crItemBtn[0]);
		}

		if (GetFocus() == this && hClickedItem == NULL) {
			return CreatePen(PS_SOLID, 1, m_crItemBtn[1]);
		}

		return CreatePen(PS_SOLID, 1, m_crItemBtn[0]);
	}
}

void CApredTreeCtrl2::PreSubclassWindow()
{
	m_crItemBk[0] = GetBkColor();
	m_crItemBk[1] = GetSysColor(COLOR_HIGHLIGHT);
	m_hBrushItem[0] = CreateSolidBrush(m_crItemBk[0]);
	m_hBrushItem[1] = CreateSolidBrush(m_crItemBk[1]);
	m_hBrushItem[2] = CreateSolidBrush(m_crItemBk[2]);

	CTreeCtrl::PreSubclassWindow();
}

void CApredTreeCtrl2::SetItemBadge(HTREEITEM hItem, COLORREF badgeBackground, COLORREF badgeForeground)
{
	if (m_badges.find(hItem) == m_badges.end()) {
		BADGE badge;
		badge.badgeBackground = badgeBackground;
		badge.badgeForeground = badgeForeground;
		badge.type = BADGE_HIDE;
		m_badges[hItem] = badge;
	}
	else {
		BADGE& badge = m_badges[hItem];
		badge.badgeBackground = badgeBackground;
		badge.badgeForeground = badgeForeground;
		badge.type = BADGE_HIDE;
	}
}

void CApredTreeCtrl2::ShowItemBadgeNumber(HTREEITEM hItem, int number)
{
	if (m_badges.find(hItem) == m_badges.end()) {
		return;
	}

	BADGE& badge = m_badges[hItem];
	badge.type = BADGE_NUMBER;
	badge.number = number;
	InvalidateRect(NULL, TRUE);
}

void CApredTreeCtrl2::ShowItemBadgeDotMode(HTREEITEM hItem)
{
	if (m_badges.find(hItem) == m_badges.end()) {
		return;
	}

	BADGE& badge = m_badges[hItem];
	if (badge.type != BADGE_DOT) {
		badge.type = BADGE_DOT;
		InvalidateRect(NULL, TRUE);
	}
}

void CApredTreeCtrl2::HideItemBadge(HTREEITEM hItem)
{
	if (m_badges.find(hItem) == m_badges.end()) {
		return;
	}

	BADGE& badge = m_badges[hItem];
	if (badge.type != BADGE_HIDE) {
		badge.type = BADGE_HIDE;
		InvalidateRect(NULL, TRUE);
	}
}