#include "ctlspriv.h" #include "treeview.h" #include "image.h" extern void TruncateString(char *sz, int cch); void TV_GetBackgroundBrush(PTREE pTree, HDC hdc) { if (pTree->clrBk == (COLORREF)-1) { if (pTree->ci.style & WS_DISABLED) pTree->hbrBk = FORWARD_WM_CTLCOLORSTATIC(pTree->ci.hwndParent, hdc, pTree->ci.hwnd, SendMessage); else pTree->hbrBk = FORWARD_WM_CTLCOLOREDIT(pTree->ci.hwndParent, hdc, pTree->ci.hwnd, SendMessage); } } // ---------------------------------------------------------------------------- // // Draws a horizontal or vertical dotted line from the given (x,y) location // for the given length (c). // // ---------------------------------------------------------------------------- void TV_DrawDottedLine(HDC hdc, int x, int y, int c, BOOL fVert) { while (c > 0) { PatBlt(hdc, x, y, 1, 1, PATCOPY); if (fVert) y += 2; else x += 2; c -= 2; } } // ---------------------------------------------------------------------------- // // Draws a plus or minus sign centered around the given (x,y) location and // extending out from that location the given distance (c). // // ---------------------------------------------------------------------------- // TV_DrawPlusMinus is theme aware void TV_DrawPlusMinus(PTREE pTree, HDC hdc, int x, int y, int c, HBRUSH hbrSign, HBRUSH hbrBox, HBRUSH hbrBk, BOOL fPlus) { HRESULT hr = E_FAIL; int n; int p = (c * 7) / 10; n = p * 2 + 1; if (pTree->hTheme) { RECT rc = { x - c, y - c, x + c + 1, y + c + 1 }; hr = DrawThemeBackground(pTree->hTheme, hdc, TVP_GLYPH, fPlus ? GLPS_CLOSED : GLPS_OPENED, &rc, 0); } if (FAILED(hr)) { SelectObject(hdc, hbrBk); PatBlt(hdc, x - c, y - c, 2*c, 2*c, PATCOPY); SelectObject(hdc, hbrSign); if (p >= 5) { PatBlt(hdc, x - p, y - 1, n, 3, PATCOPY); if (fPlus) PatBlt(hdc, x - 1, y - p, 3, n, PATCOPY); p--; n -= 2; } PatBlt(hdc, x - p, y, n, 1, PATCOPY); if (fPlus) PatBlt(hdc, x, y - p, 1, n, PATCOPY); n = c * 2 + 1; SelectObject(hdc, hbrBox); PatBlt(hdc, x - c, y - c, n, 1, PATCOPY); PatBlt(hdc, x - c, y - c, 1, n, PATCOPY); PatBlt(hdc, x - c, y + c, n, 1, PATCOPY); PatBlt(hdc, x + c, y - c, 1, n, PATCOPY); } } // ---------------------------------------------------------------------------- // // Create the bitmaps for the indent area of the tree as follows // if fHasLines && fHasButtons --> 7 bitmaps // if fHasLines && !fHasButtons --> 3 bitmaps // if !fHasLines && fHasButtons --> 2 bitmaps // // sets hStartBmp, hBmp, hdcBits // // If "has lines" then there are three basic bitmaps. // // | | | // | +--- +--- // | | // // (The plan vertical line does not get buttons.) // // Otherwise, there are no lines, so the basic bitmaps are blank. // // If "has buttons", then the basic bitmaps are augmented with buttons. // // [+] [-] // // And if you have "lines at root", you get // // __ // // // And if you have "lines at root" with "has buttons", then you also get // // --[+] --[-] // // So, there are twelve image types. Here they are, with the code names // written underneath. // // | | | | | | | // | +--- +--- [+]-- [+]-- [-]-- [-]-- // | | | | // // "|" "|-" "L" "|-+" "L+" "|--" "L-" // // --- [+]-- [-]-- [+] [-] // // ".-" ".-+" ".--" "+" "-" // // And the master table of which styles get which images. // // // LINES BTNS ROOT | |- L |-+ L+ |-- L- .- .-+ .-- + - // // x 0 1 // x 0 1 2 3 // x 0 1 2 3 // x x 0 1 2 3 4 5 6 // x x 0 1 2 3 // x x x 0 1 2 3 4 5 6 7 8 9 // // ---------------------------------------------------------------------------- void TV_DrawV(PTREE pTree, HDC hdc, int x, int y) // "|" { int xMid; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; TV_DrawDottedLine(hdc, x + xMid, y, pTree->cyItem, TRUE); SelectObject(hdc, hbrOld); } void TV_DrawT(PTREE pTree, HDC hdc, int x, int y) // "|-" { int xMid, yMid; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; TV_DrawDottedLine(hdc, x + xMid, y, pTree->cyItem, TRUE); TV_DrawDottedLine(hdc, x + xMid, y + yMid, pTree->cxIndent - xMid, FALSE); SelectObject(hdc, hbrOld); } void TV_DrawL(PTREE pTree, HDC hdc, int x, int y, BOOL bRoot) // "L" { int xMid, yMid; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; if (!bRoot) { TV_DrawDottedLine(hdc, x + xMid, y, yMid, TRUE); } else { TV_DrawDottedLine(hdc, x + xMid, y + yMid, yMid, TRUE); } TV_DrawDottedLine(hdc, x + xMid, y + yMid, pTree->cxIndent - xMid, FALSE); SelectObject(hdc, hbrOld); } void TV_DrawPML(PTREE pTree, HDC hdc, int x, int y, BOOL fPlus, BOOL fL, BOOL bRoot) { int xMid, yMid, c; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; c = (min(xMid, yMid)) / 2; if (fL) { if (!bRoot) { TV_DrawDottedLine(hdc, x + xMid, y, yMid - c, TRUE); } else { TV_DrawDottedLine(hdc, x + xMid, y + yMid + c, yMid - c, TRUE); } TV_DrawDottedLine(hdc, x + xMid + c, y + yMid, pTree->cxIndent - xMid - c, FALSE); } else { TV_DrawDottedLine(hdc, x + xMid, y, yMid - c, TRUE); TV_DrawDottedLine(hdc, x + xMid + c, y + yMid, pTree->cxIndent - xMid - c, FALSE); TV_DrawDottedLine(hdc, x + xMid, y + yMid + c, yMid - c, TRUE); } TV_DrawPlusMinus(pTree, hdc, x + xMid, y + yMid, c, pTree->hbrText, pTree->hbrLine, pTree->hbrBk, fPlus); SelectObject(hdc, hbrOld); } void TV_DrawTP(PTREE pTree, HDC hdc, int x, int y) // "|-+" { TV_DrawPML(pTree, hdc, x, y, TRUE, FALSE, FALSE); } void TV_DrawLP(PTREE pTree, HDC hdc, int x, int y, BOOL bRoot) // "L+" { TV_DrawPML(pTree, hdc, x, y, TRUE, TRUE, bRoot); } void TV_DrawTM(PTREE pTree, HDC hdc, int x, int y) // "|--" { TV_DrawPML(pTree, hdc, x, y, FALSE, FALSE, FALSE); } void TV_DrawLM(PTREE pTree, HDC hdc, int x, int y, BOOL bRoot) // "L-" { TV_DrawPML(pTree, hdc, x, y, FALSE, TRUE, bRoot); } void TV_DrawH(PTREE pTree, HDC hdc, int x, int y) // ".-" { int xMid, yMid; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; TV_DrawDottedLine(hdc, x + xMid, y + yMid, pTree->cxIndent - xMid, FALSE); SelectObject(hdc, hbrOld); } void TV_DrawHP(PTREE pTree, HDC hdc, int x, int y) // ".-+" { int xMid, yMid, c; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; c = (min(xMid, yMid)) / 2; TV_DrawDottedLine(hdc, x + c, y + yMid, pTree->cxIndent - xMid - c, FALSE); TV_DrawPlusMinus(pTree, hdc, x + xMid, y + yMid, c, pTree->hbrText, pTree->hbrLine, pTree->hbrBk, TRUE); SelectObject(hdc, hbrOld); } void TV_DrawHM(PTREE pTree, HDC hdc, int x, int y) // ".--" { int xMid, yMid, c; HBRUSH hbrOld = SelectObject(hdc, pTree->hbrLine); if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; c = (min(xMid, yMid)) / 2; TV_DrawDottedLine(hdc, x + c, y + yMid, pTree->cxIndent - xMid - c, FALSE); TV_DrawPlusMinus(pTree, hdc, x + xMid, y + yMid, c, pTree->hbrText, pTree->hbrLine, pTree->hbrBk, FALSE); SelectObject(hdc, hbrOld); } void TV_DrawP(PTREE pTree, HDC hdc, int x, int y) // "+" { int xMid, yMid, c; if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; c = (min(xMid, yMid)) / 2; TV_DrawPlusMinus(pTree, hdc, x + xMid, y + yMid, c, pTree->hbrText, pTree->hbrLine, pTree->hbrBk, TRUE); } void TV_DrawM(PTREE pTree, HDC hdc, int x, int y) // "-" { int xMid, yMid, c; if (pTree->hImageList) xMid = (pTree->cxImage - MAGIC_INDENT) / 2; else xMid = pTree->cxIndent / 2; yMid = ((pTree->cyItem / 2) + 1) & ~1; c = (min(xMid, yMid)) / 2; TV_DrawPlusMinus(pTree, hdc, x + xMid, y + yMid, c, pTree->hbrText, pTree->hbrLine, pTree->hbrBk, FALSE); } void TV_DrawState2(PTREE pTree, int iState, HDC hdc, int x, int y, BOOL bRoot) { switch (iState) { case 0: if (pTree->ci.style & TVS_HASLINES) { TV_DrawV(pTree, hdc, x, y); } else { TV_DrawP(pTree, hdc, x, y); } break; case 1: if (pTree->ci.style & TVS_HASLINES) { TV_DrawT(pTree, hdc, x, y); } else { TV_DrawM(pTree, hdc, x, y); } break; case 2: TV_DrawL(pTree, hdc, x, y, bRoot); break; case 3: if (!(pTree->ci.style & TVS_HASBUTTONS)) { TV_DrawH(pTree, hdc, x, y); } else { TV_DrawTP(pTree, hdc, x, y); } break; case 4: TV_DrawLP(pTree, hdc, x, y, bRoot); break; case 5: TV_DrawTM(pTree, hdc, x, y); break; case 6: TV_DrawLM(pTree, hdc, x, y, bRoot); break; case 7: TV_DrawH(pTree, hdc, x, y); break; case 8: TV_DrawHP(pTree, hdc, x, y); break; case 9: TV_DrawHM(pTree, hdc, x, y); break; } } void TV_DrawState(PTREE pTree, int iState, HDC hdc, int x, int y) { TV_DrawState2(pTree, iState, hdc, x, y, FALSE); } // TV_CreateIndentBmps is theme aware void TV_CreateIndentBmps(PTREE pTree) { if (pTree->fRedraw) InvalidateRect(pTree->ci.hwnd, NULL, TRUE); return; } // ---------------------------------------------------------------------------- // // fills in a TVITEM structure based by coying data from the item or // by calling the callback to get it. // // in: // hItem item to get TVITEM struct for // mask which bits of the TVITEM struct you want (TVIF_ flags) // out: // lpItem TVITEM filled in // // ---------------------------------------------------------------------------- void TV_GetItem(PTREE pTree, HTREEITEM hItem, UINT mask, LPTVITEMEX lpItem) { TV_DISPINFO nm; if (!hItem || !lpItem) return; DBG_ValidateTreeItem(hItem, FALSE); nm.item.mask = 0; // We need to check the mask to see if lpItem->pszText is valid // And even then, it might not be, so be paranoid if ((mask & TVIF_TEXT) && lpItem->pszText && lpItem->cchTextMax) { if (hItem->lpstr == LPSTR_TEXTCALLBACK) { nm.item.mask |= TVIF_TEXT; // caller had to fill in pszText and cchTextMax with valid data nm.item.pszText = lpItem->pszText; nm.item.cchTextMax = lpItem->cchTextMax; nm.item.pszText[0] = 0; #ifdef DEBUG pTree->fInTextCallback = TRUE; #endif } else { ASSERT(hItem->lpstr); // we could do this but this is dangerous (when responding // to TVM_GETITEM we would be giving the app a pointer to our data) // lpItem->pszText = hItem->lpstr; StringCchCopy(lpItem->pszText, lpItem->cchTextMax, hItem->lpstr); } } if (mask & TVIF_IMAGE) { if (hItem->iImage == (WORD)I_IMAGECALLBACK) nm.item.mask |= TVIF_IMAGE; else lpItem->iImage = hItem->iImage; } if (mask & TVIF_SELECTEDIMAGE) { if (hItem->iSelectedImage == (WORD)I_IMAGECALLBACK) nm.item.mask |= TVIF_SELECTEDIMAGE; else lpItem->iSelectedImage = hItem->iSelectedImage; } if (mask & TVIF_INTEGRAL) { lpItem->iIntegral = hItem->iIntegral; } if (mask & TVIF_CHILDREN) { switch (hItem->fKids) { case KIDS_COMPUTE: lpItem->cChildren = hItem->hKids ? 1 : 0;// the actual count doesn't matter break; case KIDS_FORCE_YES: lpItem->cChildren = 1;// the actual count doesn't matter break; case KIDS_FORCE_NO: lpItem->cChildren = 0; break; case KIDS_CALLBACK: nm.item.mask |= TVIF_CHILDREN; break; } } // copy out constant parameters (and prepare for callback) // IE4 and IE5.0 did this unconditionally lpItem->state = nm.item.state = hItem->state; // // NOTICE! We do not set TVIF_STATE nm.item.mask and we do not // check for TVIF_STATE in the "any items need to be filled in // by callback?" test a few lines below. This is necessary for // backwards compat. IE5 and earlier did not call the app back // if the only thing you asked for was TVIF_STATE. You can't // change this behavior unless you guard it with a version check, or // apps will break. (They'll get callbacks when they didn't used to.) // Besides, nobody knows that they can customize the state, so it's // not like we're missing out on anything. // lpItem->lParam = nm.item.lParam = hItem->lParam; // any items need to be filled in by callback? if (nm.item.mask & (TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_CHILDREN)) { nm.item.hItem = hItem; CCSendNotify(&pTree->ci, TVN_GETDISPINFO, &nm.hdr); #ifdef DEBUG pTree->fInTextCallback = FALSE; #endif // copy out things that may have been filled in on the callback if (nm.item.mask & TVIF_CHILDREN) lpItem->cChildren = nm.item.cChildren; if (nm.item.mask & TVIF_IMAGE) lpItem->iImage = nm.item.iImage; if (nm.item.mask & TVIF_SELECTEDIMAGE) lpItem->iSelectedImage = nm.item.iSelectedImage; // callback may have redirected pszText to point into its own buffer if (nm.item.mask & TVIF_TEXT) lpItem->pszText = CCReturnDispInfoText(nm.item.pszText, lpItem->pszText, lpItem->cchTextMax); if (nm.item.mask & TVIF_STATE) { lpItem->state = (nm.item.state & nm.item.stateMask) | (lpItem->state & ~nm.item.stateMask); if ((lpItem->state & TVIS_BOLD) && !pTree->hFontBold) TV_CreateBoldFont(pTree); } if (nm.item.mask & TVIF_DI_SETITEM) { if (nm.item.mask & TVIF_TEXT) if (nm.item.pszText) { Str_Set(&hItem->lpstr, nm.item.pszText); } if (nm.item.mask & TVIF_STATE) { // if the bold bit changed, then the width changed if ((hItem->state ^ lpItem->state) & TVIS_BOLD) hItem->iWidth = 0; hItem->state = (WORD) lpItem->state; } if (nm.item.mask & TVIF_IMAGE) hItem->iImage = (WORD) lpItem->iImage; if (nm.item.mask & TVIF_SELECTEDIMAGE) hItem->iSelectedImage = (WORD) lpItem->iSelectedImage; if (nm.item.mask & TVIF_CHILDREN) { switch(nm.item.cChildren) { case I_CHILDRENCALLBACK: hItem->fKids = KIDS_CALLBACK; break; case I_CHILDRENAUTO: hItem->fKids = KIDS_COMPUTE; break; case 0: hItem->fKids = KIDS_FORCE_NO; break; default: hItem->fKids = KIDS_FORCE_YES; break; } } } } } // ---------------------------------------------------------------------------- // // Draws the given item starting at the given (x,y) and extending down and to // the right. // // ---------------------------------------------------------------------------- BOOL TV_ShouldItemDrawBlue(PTREE pTree, TVITEMEX *ti, UINT flags) { return ( (ti->state & TVIS_DROPHILITED) || (!pTree->hDropTarget && !(flags & TVDI_GRAYCTL) && (ti->state & TVIS_SELECTED) && pTree->fFocus)); } #define TV_ShouldItemDrawDisabled(pTree, pti, flags) (flags & TVDI_GRAYCTL) // // Caution: Depending on the user's color scheme, a Gray item may // end up looking Blue if Gray would otherwise be invisible. So make // sure that there are other cues that the user can use to tell whether // the item is "Really Blue" or "Gray masquerading as Blue". // // For example, you might get both is if the treeview is // participating in drag/drop while it is not the active window, // because the selected item gets "Gray masquerading as Blue" and // the drop target gets "Really Blue". But we special-case that // and turn off the selection while we are worrying about drag/drop, // so there is no confusion after all. // BOOL TV_ShouldItemDrawGray(PTREE pTree, TVITEMEX *pti, UINT flags) { return ((flags & TVDI_GRAYCTL) || (!pTree->hDropTarget && ((pti->state & TVIS_SELECTED) && (!pTree->fFocus && (pTree->ci.style & TVS_SHOWSELALWAYS)) ))); } // // Draw a descender line for the item. It is the caller's job to // draw the appropriate glyph at level 0. // void TV_DrawDescender(PTREE pTree, HDC hdc, int x, int y, HTREEITEM hItem) { int i; for (i = 1; i < hItem->iIntegral; i++) { BitBlt(hdc, x, y + i * pTree->cyItem, pTree->cxIndent, pTree->cyItem, pTree->hdcBits, 0, 0, SRCCOPY); } } // // Erase any previous descender line for the item. // void TV_EraseDescender(PTREE pTree, HDC hdc, int x, int y, HTREEITEM hItem) { RECT rc; rc.left = x; rc.right = x + pTree->cxIndent; rc.top = y + pTree->cyItem; rc.bottom = y + hItem->iIntegral * pTree->cyItem; FillRect(hdc, &rc, pTree->hbrBk); } // // Draw (or erase) descenders for siblings and children. // void TV_DrawKinDescender(PTREE pTree, HDC hdc, int x, int y, HTREEITEM hItem, UINT state) { if (hItem->hNext) // Connect to next sibling TV_DrawDescender(pTree, hdc, x, y, hItem); else TV_EraseDescender(pTree, hdc, x, y, hItem); // If any bonus images, then need to connect the image to the kids. if (pTree->himlState || pTree->hImageList) { if (state & (TVIS_EXPANDED | TVIS_EXPANDPARTIAL)) // Connect to expanded kids TV_DrawDescender(pTree, hdc, x + pTree->cxIndent, y, hItem); else TV_EraseDescender(pTree, hdc, x + pTree->cxIndent, y, hItem); } } // TV_DrawItem is theme aware void TV_DrawItem(PTREE pTree, HTREEITEM hItem, HDC hdc, int x, int y, UINT flags) { UINT cxIndent = pTree->cxIndent; COLORREF rgbOldBack = 0, rgbOldText; COLORREF clrBk = CLR_DEFAULT; RECT rc; int iBack, iText; HTREEITEM hItemSave = hItem; LPTSTR lpstr; int cch; UINT etoFlags = ETO_OPAQUE | ETO_CLIPPED; TVITEMEX ti; TCHAR szTemp[MAX_PATH]; int iState = 0; HFONT hFont; //$BOLD DWORD dwRet; NMTVCUSTOMDRAW nmcd; BOOL fItemFocused = ((pTree->fFocus) && (hItem == pTree->hCaret)); DWORD clrTextTemp, clrTextBkTemp; BOOL fSelectedIcon = FALSE; int iOldBkMode = GetBkMode(hdc); rc.top = y; rc.bottom = rc.top + (pTree->cyItem * hItem->iIntegral); rc.left = 0; rc.right = pTree->cxWnd; if (flags & TVDI_ERASE) { // Opaque the whole item FillRect(hdc, &rc, pTree->hbrBk); } // make sure the callbacks don't invalidate this item pTree->hItemPainting = hItem; ti.pszText = szTemp; ti.cchTextMax = ARRAYSIZE(szTemp); ti.stateMask = TVIS_OVERLAYMASK | TVIS_CUT | TVIS_BOLD; //$BOLD TV_GetItem(pTree, hItem, TVIF_IMAGE | TVIF_STATE | TVIF_SELECTEDIMAGE | TVIF_TEXT | TVIF_CHILDREN | TVIF_PARAM, &ti); pTree->hItemPainting = NULL; //////////////// // set up the HDC if (TV_ShouldItemDrawBlue(pTree,&ti,flags)) { // selected iBack = COLOR_HIGHLIGHT; iText = COLOR_HIGHLIGHTTEXT; } else if (TV_ShouldItemDrawDisabled(pTree, &pti, flags)) { iBack = COLOR_3DFACE; iText = COLOR_GRAYTEXT; SetBkMode(hdc, TRANSPARENT); etoFlags &= ~ ETO_OPAQUE; } else if (TV_ShouldItemDrawGray(pTree, &ti, flags)) { // On some color schemes, the BTNFACE color equals the WINDOW color, // and our gray comes out invisible. In such case, change from gray // to blue so you can see it at all. if (GetSysColor(COLOR_WINDOW) != GetSysColor(COLOR_BTNFACE)) { iBack = COLOR_BTNFACE; iText = COLOR_BTNTEXT; } else { iBack = COLOR_HIGHLIGHT; iText = COLOR_HIGHLIGHTTEXT; } } else { // not selected iBack = COLOR_WINDOW; iText = COLOR_WINDOWTEXT; if (hItem == pTree->hHot) { iText = COLOR_HOTLIGHT; } } if (iBack == COLOR_WINDOW && (pTree->clrBk != (COLORREF)-1)) nmcd.clrTextBk = clrTextBkTemp = pTree->clrBk; else nmcd.clrTextBk = clrTextBkTemp = GetSysColor(iBack); if (iText == COLOR_WINDOWTEXT && (pTree->clrText != (COLORREF)-1)) nmcd.clrText = clrTextTemp = pTree->clrText; else nmcd.clrText = clrTextTemp = GetSysColor(iText); // if forcing black and transparent, do so. dc's BkMode should // already be set to TRANSPARENT by caller if (flags & TVDI_TRANSTEXT) { nmcd.clrText = clrTextTemp = 0x000000; etoFlags = 0; // don't opaque nothin' } rgbOldBack = SetBkColor(hdc, nmcd.clrTextBk); rgbOldText = SetTextColor(hdc, nmcd.clrText); if (pTree->ci.style & TVS_RTLREADING) etoFlags |= ETO_RTLREADING; // Figure out which font to use. if (ti.state & TVIS_BOLD) { hFont = pTree->hFontBold; if (hItem == pTree->hHot) { hFont = CCGetHotFont(pTree->hFontBold, &pTree->hFontBoldHot); } } else { hFont = pTree->hFont; if (hItem == pTree->hHot) { hFont = CCGetHotFont(pTree->hFont, &pTree->hFontHot); } } hFont = SelectObject(hdc, hFont); // End HDC setup //////////////// // notify on custom draw then do it! nmcd.nmcd.hdc = hdc; nmcd.nmcd.dwItemSpec = (DWORD_PTR)hItem; nmcd.nmcd.uItemState = 0; nmcd.nmcd.rc = rc; if (flags & TVDI_NOTREE) nmcd.iLevel = 0; else nmcd.iLevel = hItem->iLevel; if (ti.state & TVIS_SELECTED) { fSelectedIcon = TRUE; if (pTree->fFocus || (pTree->ci.style & TVS_SHOWSELALWAYS)) nmcd.nmcd.uItemState |= CDIS_SELECTED; } if (fItemFocused) nmcd.nmcd.uItemState |= CDIS_FOCUS; if (hItem == pTree->hHot) nmcd.nmcd.uItemState |= CDIS_HOT; nmcd.nmcd.lItemlParam = ti.lParam; dwRet = CICustomDrawNotify(&pTree->ci, CDDS_ITEMPREPAINT, &nmcd.nmcd); if (dwRet & CDRF_SKIPDEFAULT) return; fItemFocused = (nmcd.nmcd.uItemState & CDIS_FOCUS); if (nmcd.nmcd.uItemState & CDIS_SELECTED) ti.state |= TVIS_SELECTED; else { ti.state &= ~TVIS_SELECTED; } if (nmcd.clrTextBk != clrTextBkTemp) SetBkColor(hdc, nmcd.clrTextBk); if (nmcd.clrText != clrTextTemp) SetTextColor(hdc, nmcd.clrText); if (pTree->ci.style & TVS_FULLROWSELECT && !(flags & TVDI_TRANSTEXT)) { FillRectClr(hdc, &nmcd.nmcd.rc, GetBkColor(hdc)); etoFlags |= ETO_OPAQUE; clrBk = CLR_NONE; } if (!(flags & TVDI_NOTREE)) { if ((pTree->ci.style & (TVS_HASLINES | TVS_HASBUTTONS)) && (pTree->ci.style & TVS_LINESATROOT)) // Make room for the "plus" at the front of the tree x += cxIndent; } // deal with margin, etc. x += (pTree->cxBorder + (nmcd.iLevel * cxIndent)); y += pTree->cyBorder; // draw image if ((!(flags & TVDI_NOTREE) && !(dwRet & TVCDRF_NOIMAGES)) || (flags & TVDI_FORCEIMAGE)) { int dx, dy; // to clip the images within the borders. COLORREF clrImage = CLR_HILIGHT; COLORREF clrBkImage = clrBk; if (flags & TVDI_NOBK) { clrBkImage = CLR_NONE; } if (pTree->himlState) { iState = TV_StateIndex(&ti); // go figure. in the treeview, 0 for the state image index // means draw nothing... the 0th item is unused. // the listview is 0 based and uses the 0th item. if (iState) { dx = min(pTree->cxState, pTree->cxMax - pTree->cxBorder - x); dy = min(pTree->cyState, pTree->cyItem - (2 * pTree->cyBorder)); ImageList_DrawEx(pTree->himlState, iState, hdc, x, y + max(pTree->cyItem - pTree->cyState, 0), dx, dy, clrBk, CLR_DEFAULT, ILD_NORMAL | (CCDPIScale(pTree->ci)?ILD_SCALE:0)); x += pTree->cxState; } } if (pTree->hImageList) { UINT fStyle = 0; int i = (fSelectedIcon) ? ti.iSelectedImage : ti.iImage; if (ti.state & TVIS_CUT) { fStyle |= ILD_BLEND50; clrImage = ImageList_GetBkColor(pTree->hImageList); } dx = min(pTree->cxImage - MAGIC_INDENT, pTree->cxMax - pTree->cxBorder - x); dy = min(pTree->cyImage, pTree->cyItem - (2 * pTree->cyBorder)); // Draw image stretched (rather than centered) when ImageList allows for scaling ImageList_DrawEx(pTree->hImageList, i, hdc, x + (dx - pTree->cxNativeImage) / 2, y + (max(pTree->cyItem - pTree->cyImage, 0) / 2) + (dy - pTree->cyNativeImage) / 2, pTree->cxNativeImage, pTree->cyNativeImage, clrBkImage, clrImage, fStyle | (ti.state & TVIS_OVERLAYMASK) | (CCDPIScale(pTree->ci)?ILD_SCALE:0)); } } if (pTree->hImageList) { // even if not drawing image, draw text in right place x += pTree->cxImage; } // draw text lpstr = ti.pszText; cch = lstrlen(lpstr); if (!hItem->iWidth || (hItem->lpstr == LPSTR_TEXTCALLBACK)) { TV_ComputeItemWidth(pTree, hItem, hdc); //$BOLD } rc.left = x; rc.top = y + pTree->cyBorder; rc.right = min((x + hItem->iWidth), (pTree->cxMax - pTree->cxBorder)); rc.bottom-= pTree->cyBorder; // Draw the text, unless it's the one we are editing if (pTree->htiEdit != hItem || !IsWindow(pTree->hwndEdit) || !IsWindowVisible(pTree->hwndEdit)) { ExtTextOut(hdc, x + g_cxLabelMargin, y + ((pTree->cyItem - pTree->cyText) / 2) + g_cyBorder, etoFlags, &rc, lpstr, cch, NULL); // Draw the focus rect, if appropriate. if (pTree->fFocus && (fItemFocused) && !(pTree->ci.style & TVS_FULLROWSELECT) && !(flags & (TVDI_TRANSTEXT | TVDI_GRAYCTL))&& !(CCGetUIState(&(pTree->ci)) & UISF_HIDEFOCUS)) { DrawFocusRect(hdc, &rc); } } SetBkColor(hdc, rgbOldBack); SetTextColor(hdc, rgbOldText); // Restore the original font. //$BOLD SelectObject(hdc, hFont); //$BOLD // Notice that we should have opaque'd the rest of the line above if no tree if (!(flags & TVDI_NOTREE)) { int dx, dy; if (pTree->hImageList) x -= pTree->cxImage; if (iState) x -= pTree->cxState; if (pTree->ci.style & TVS_HASLINES) { int i; x -= cxIndent; if (nmcd.iLevel-- || (pTree->ci.style & TVS_LINESATROOT)) { // Special case the first root if (nmcd.iLevel == -1 && hItem == hItem->hParent->hKids) { if (hItem->hNext) { i = 2; // "L" if (ti.cChildren && (pTree->ci.style & TVS_HASBUTTONS)) { i += 2; // "L+" if ((ti.state & (TVIS_EXPANDED | TVIS_EXPANDPARTIAL)) == TVIS_EXPANDED) i += 2; // "L-" } dx = min((int)cxIndent, pTree->cxMax - pTree->cxBorder - x); dy = pTree->cyItem - (2 * pTree->cyBorder); // Pass TRUE to draw the root ("L" upside down) TV_DrawState2(pTree, i, hdc, x, y, TRUE); i = -1; } else { // first root no siblings // if there's no other item, draw just the button if button mode, if (pTree->ci.style & TVS_HASBUTTONS) { if (ti.cChildren) { // hasbuttons, has lines, lines at root i = ((ti.state & (TVIS_EXPANDED | TVIS_EXPANDPARTIAL)) == TVIS_EXPANDED) ? 9 : 8; // ".--" : ".-+" } else { i = 7; // ".-" } } else { i = 3; // ".-" } } } else { i = (hItem->hNext) ? 1 : 2; // "|-" (rep) : "L" if (ti.cChildren && (pTree->ci.style & TVS_HASBUTTONS)) { i += 2; // "|-+" (rep) : "L+" if ((ti.state & (TVIS_EXPANDED | TVIS_EXPANDPARTIAL)) == TVIS_EXPANDED) i += 2; // "|--" (rep) : "L-" } } if (hItem->iIntegral > 1) TV_DrawKinDescender(pTree, hdc, x, y, hItem, ti.state); if (i != -1) { dx = min((int)cxIndent, pTree->cxMax - pTree->cxBorder - x); dy = pTree->cyItem - (2 * pTree->cyBorder); if ((dx > 0) && (dy > 0)) TV_DrawState(pTree, i, hdc, x, y); } while ((--nmcd.iLevel >= 0) || ((pTree->ci.style & TVS_LINESATROOT) && nmcd.iLevel >= -1)) { hItem = hItem->hParent; x -= cxIndent; if (hItem->hNext) { dx = min((int)cxIndent, (pTree->cxMax - pTree->cxBorder - x)); dy = min(pTree->cyItem, pTree->cyWnd - pTree->cyBorder - y); if ((dx > 0) && (dy > 0)) TV_DrawState(pTree, 0, hdc, x, y); TV_DrawDescender(pTree, hdc, x, y, hItemSave); } } } } else { // no lines if ((pTree->ci.style & TVS_HASBUTTONS) && (nmcd.iLevel || pTree->ci.style & TVS_LINESATROOT) && ti.cChildren) { int i = ((ti.state & (TVIS_EXPANDED | TVIS_EXPANDPARTIAL)) == TVIS_EXPANDED) ? 1 : 0; x -= cxIndent; dx = min((int)cxIndent, pTree->cxMax - pTree->cxBorder - x); dy = min(pTree->cyItem, pTree->cyWnd - pTree->cyBorder - y); if ((dx > 0) && (dy > 0)) TV_DrawState(pTree, i, hdc, x, y); } } } if (dwRet & CDRF_NOTIFYPOSTPAINT) { nmcd.nmcd.dwItemSpec = (DWORD_PTR)hItemSave; CICustomDrawNotify(&pTree->ci, CDDS_ITEMPOSTPAINT, &nmcd.nmcd); } SetBkMode(hdc, iOldBkMode); } #define INSERTMARKSIZE 6 BOOL TV_GetInsertMarkRect(PTREE pTree, LPRECT prc) { ASSERT(pTree); if(pTree->htiInsert && TV_GetItemRect(pTree, pTree->htiInsert, prc, TRUE)) { if (pTree->fInsertAfter) prc->top = prc->bottom; else prc->bottom = prc->top; prc->top -= INSERTMARKSIZE/2; prc->bottom += INSERTMARKSIZE/2 + 1; prc->right = pTree->cxWnd - INSERTMARKSIZE; // should always go all the way to right with pad. prc->left -= pTree->cxImage; return TRUE; } return FALSE; } __inline COLORREF TV_GetInsertMarkColor(PTREE pTree) { if (pTree->clrim == CLR_DEFAULT) return g_clrWindowText; else return pTree->clrim; } void TV_DrawTree(PTREE pTree, HDC hdc, BOOL fErase, LPRECT lprc) { int x; int iStart, iCnt; UINT uFlags; RECT rc; NMCUSTOMDRAW nmcd; if (!pTree->fRedraw) return; if (pTree->ci.style & TVS_CHECKBOXES) if (!pTree->himlState) TV_InitCheckBoxes(pTree); x = -pTree->xPos; TV_GetBackgroundBrush(pTree, hdc); rc = *lprc; iStart = lprc->top / pTree->cyItem; if (pTree->cItems && pTree->hTop) { ASSERT(ITEM_VISIBLE(pTree->hTop)); iCnt = pTree->cShowing - pTree->hTop->iShownIndex; } else { iCnt = 0; // Nothing to draw } nmcd.hdc = hdc; /// not implemented yet //if (ptb->ci.hwnd == GetFocus()) //nmcd.uItemState = CDIS_FOCUS; //else nmcd.uItemState = 0; nmcd.lItemlParam = 0; nmcd.rc = rc; pTree->ci.dwCustom = CICustomDrawNotify(&pTree->ci, CDDS_PREPAINT, &nmcd); if (!(pTree->ci.dwCustom & CDRF_SKIPDEFAULT)) { if (iStart <= iCnt) { HTREEITEM hItem; HFONT hOldFont; RECT rcT; int y = 0; for (hItem = pTree->hTop; hItem; ) { if (iStart > hItem->iIntegral) { iStart -= hItem->iIntegral; y += hItem->iIntegral * pTree->cyItem; hItem = TV_GetNextVisItem(hItem); } else break; } hOldFont = pTree->hFont ? SelectObject(hdc, pTree->hFont) : NULL; // TVDI_* for all items uFlags = (pTree->ci.style & WS_DISABLED) ? TVDI_GRAYCTL : 0; if (fErase) uFlags |= TVDI_ERASE; // loop from the first visible item until either all visible items are // drawn or there are no more items to draw for ( ; hItem && y < lprc->bottom; hItem = TV_GetNextVisItem(hItem)) { TV_DrawItem(pTree, hItem, hdc, x, y, uFlags); y += pTree->cyItem * hItem->iIntegral; } // // handle drawing the InsertMark next to this item. // if(TV_GetInsertMarkRect(pTree, &rcT)) CCDrawInsertMark(hdc, &rcT, FALSE, TV_GetInsertMarkColor(pTree)); if (hOldFont) SelectObject(hdc, hOldFont); rc.top = y; } if (fErase) // Opaque out everything we have not drawn explicitly FillRect(hdc, &rc, pTree->hbrBk); // notify parent afterwards if they want us to if (pTree->ci.dwCustom & CDRF_NOTIFYPOSTPAINT) { CICustomDrawNotify(&pTree->ci, CDDS_POSTPAINT, &nmcd); } } } // ---------------------------------------------------------------------------- // // Set up for paint, call DrawTree, and clean up after paint. // // ---------------------------------------------------------------------------- void TV_Paint(PTREE pTree, HDC hdc) { PAINTSTRUCT ps; if (hdc) { // hdc != 0 indicates a subclassed paint -- use the hdc passed in SetRect(&ps.rcPaint, 0, 0, pTree->cxWnd, pTree->cyWnd); TV_DrawTree(pTree, hdc, TRUE, &ps.rcPaint); } else { BeginPaint(pTree->ci.hwnd, &ps); TV_DrawTree(pTree, ps.hdc, ps.fErase, &ps.rcPaint); EndPaint(pTree->ci.hwnd, &ps); } } // ---------------------------------------------------------------------------- // Create an imagelist to be used for dragging. // // 1) create mask and image bitmap matching the select bounds size // 2) draw the text to both bitmaps (in black for now) // 3) create an imagelist with these bitmaps // 4) make a dithered copy of the image onto the new imagelist // ---------------------------------------------------------------------------- HIMAGELIST TV_CreateDragImage(PTREE pTree, HTREEITEM hItem) { HDC hdcMem = NULL; HBITMAP hbmImage = NULL; HBITMAP hbmMask = NULL; HBITMAP hbmOld; HIMAGELIST himl = NULL; BOOL bMirroredWnd = (pTree->ci.dwExStyle&RTL_MIRRORED_WINDOW); int dx, dy; int iSrc; TVITEMEX ti; if (!pTree->hImageList) return NULL; if (hItem == NULL) hItem = pTree->htiDrag; if (hItem == NULL) return NULL; dx = hItem->iWidth + pTree->cxImage; dy = pTree->cyItem; if (!(hdcMem = CreateCompatibleDC(NULL))) goto CDI_Exit; if (!(hbmImage = CreateColorBitmap(dx, dy))) goto CDI_Exit; if (!(hbmMask = CreateMonoBitmap(dx, dy))) goto CDI_Exit; // // Mirror the memory DC so that the transition from // mirrored(memDC)->non-mirrored(imagelist DCs)->mirrored(screenDC) // is consistent. [samera] // if (bMirroredWnd) { SET_DC_RTL_MIRRORED(hdcMem); } // prepare for drawing the item if (pTree->hFont) SelectObject(hdcMem, pTree->hFont); SetBkMode(hdcMem, TRANSPARENT); /* ** draw the text to both bitmaps */ hbmOld = SelectObject(hdcMem, hbmImage); // fill image with black for transparency PatBlt(hdcMem, 0, 0, dx, dy, BLACKNESS); TV_DrawItem(pTree, hItem, hdcMem, 0, 0, TVDI_NOIMAGE | TVDI_NOTREE | TVDI_TRANSTEXT); // // If the header is RTL mirrored, then // mirror the Memory DC, so that when copying back // we don't get any image-flipping. [samera] // if (bMirroredWnd) MirrorBitmapInDC(hdcMem, hbmImage); SelectObject(hdcMem, hbmMask); // fill mask with white for transparency PatBlt(hdcMem, 0, 0, dx, dy, WHITENESS); TV_DrawItem(pTree, hItem, hdcMem, 0, 0, TVDI_NOIMAGE | TVDI_NOTREE | TVDI_TRANSTEXT); // // If the header is RTL mirrored, then // mirror the Memory DC, so that when copying back // we don't get any image-flipping. [samera] // if (bMirroredWnd) MirrorBitmapInDC(hdcMem, hbmMask); // unselect objects that we used SelectObject(hdcMem, hbmOld); SelectObject(hdcMem, g_hfontSystem); /* ** make an image list that for now only has the text */ if (!(himl = ImageList_Create(dx, dy, ILC_MASK, 1, 0))) goto CDI_Exit; ImageList_SetBkColor(himl, CLR_NONE); ImageList_Add(himl, hbmImage, hbmMask); /* ** make a dithered copy of the image part onto our bitmaps ** (need both bitmap and mask to be dithered) */ TV_GetItem(pTree, hItem, TVIF_IMAGE, &ti); iSrc = ti.iImage; ImageList_CopyDitherImage(himl, 0, 0, (pTree->cyItem - pTree->cyImage) / 2, pTree->hImageList, iSrc, ((pTree->ci.dwExStyle & dwExStyleRTLMirrorWnd) ? ILD_MIRROR : 0L) | (hItem->state & TVIS_OVERLAYMASK)); CDI_Exit: if (hdcMem) DeleteObject(hdcMem); if (hbmImage) DeleteObject(hbmImage); if (hbmMask) DeleteObject(hbmMask); return himl; } #define COLORKEY RGB(0xF4, 0x0, 0x0) LRESULT TV_GenerateDragImage(PTREE pTree, SHDRAGIMAGE* pshdi) { LRESULT lRet = 0; HBITMAP hbmpOld = NULL; HTREEITEM hItem = pTree->htiDrag; RECT rc; HDC hdcDragImage; if (hItem == NULL) return FALSE; hdcDragImage = CreateCompatibleDC(NULL); if (!hdcDragImage) return 0; // After this rc contains the bounds of all the items in Client Coordinates. // // Mirror the the DC, if the listview is mirrored. // if (pTree->ci.dwExStyle & RTL_MIRRORED_WINDOW) { SET_DC_RTL_MIRRORED(hdcDragImage); } TV_GetItemRect(pTree, hItem, &rc, TRUE); // Subtract off the image... rc.left -= pTree->cxImage; pshdi->sizeDragImage.cx = RECTWIDTH(rc); pshdi->sizeDragImage.cy = RECTHEIGHT(rc); pshdi->hbmpDragImage = CreateBitmap( pshdi->sizeDragImage.cx, pshdi->sizeDragImage.cy, GetDeviceCaps(hdcDragImage, PLANES), GetDeviceCaps(hdcDragImage, BITSPIXEL), NULL); if (pshdi->hbmpDragImage) { COLORREF clrBkSave; RECT rcImage = {0, 0, pshdi->sizeDragImage.cx, pshdi->sizeDragImage.cy}; hbmpOld = SelectObject(hdcDragImage, pshdi->hbmpDragImage); pshdi->crColorKey = COLORKEY; FillRectClr(hdcDragImage, &rcImage, pshdi->crColorKey); // Calculate the offset... The cursor should be in the bitmap rect. if (pTree->ci.dwExStyle & RTL_MIRRORED_WINDOW) pshdi->ptOffset.x = rc.right - pTree->ptCapture.x; else pshdi->ptOffset.x = pTree->ptCapture.x - rc.left; pshdi->ptOffset.y = pTree->ptCapture.y - rc.top; clrBkSave = pTree->clrBk; pTree->clrBk = COLORKEY; TV_DrawItem(pTree, hItem, hdcDragImage, 0, 0, TVDI_NOTREE | TVDI_TRANSTEXT | TVDI_FORCEIMAGE | TVDI_NOBK); pTree->clrBk = clrBkSave; SelectObject(hdcDragImage, hbmpOld); DeleteDC(hdcDragImage); // We're passing back the created HBMP. return 1; } return lRet; }