TreeCtrl - A WTL Tree Control with Windows Vista Style Item Selection

<br />
  COLORREF SetBkColor(COLORREF bk)<br />
  {<br />
    m_rgbBackground = bk;<br />
    return CWindowImpl< CTreeCtrl, CTreeViewCtrl >::SetBkColor(bk);<br />
  }<br />
<br />
  void OnLButtonDown( UINT nFlags, CPoint point )<br />
  {<br />
    if ( m_bMultipleSelect )<br />
    {<br />
      // reset selected item if click on a single item<br />
      UINT flags = 0;<br />
      HTREEITEM hNewItem = HitTest( point, &flags );<br />
      if (flags & TVHT_ONITEMBUTTON)<br />
      {<br />
        SetMsgHandled( FALSE );<br />
        return;<br />
      }<br />
<br />
      m_bCtrlKey = ( ( GetKeyState( VK_CONTROL ) & 0x8000 ) != 0 );<br />
      m_bShiftKey = ( ( GetKeyState( VK_SHIFT ) & 0x8000 ) != 0 );<br />
      <br />
      if ( m_bCtrlKey || m_bShiftKey )<br />
        return;<br />
      <br />
      if ( hNewItem != NULL )<br />
      {<br />
        UINT nNewSelectState = ( GetItemState( hNewItem, TVIS_SELECTED ) & TVIS_SELECTED );<br />
        ResetSelected( GetRootItem() );<br />
        SetItemState( hNewItem, nNewSelectState, TVIS_SELECTED );<br />
        m_hFirstSelected = hNewItem;<br />
      }<br />
    }<br />
    <br />
    SetMsgHandled( FALSE );<br />
  }<br />
<br />
  void OnLButtonUp( UINT nFlags, CPoint point )<br />
  {<br />
    SetMsgHandled( FALSE );<br />
    <br />
    if ( !m_bMultipleSelect )<br />
      return;<br />
<br />
    UINT flags = 0;<br />
    HTREEITEM hNewItem = HitTest( point, &flags );<br />
    if (flags & TVHT_ONITEMBUTTON)<br />
      return;<br />
<br />
    HTREEITEM hOldItem = GetSelectedItem();<br />
    <br />
    if ( m_bCtrlKey )<br />
    {<br />
      if ( hNewItem != NULL )<br />
      {<br />
        UINT nNewSelectState = GetItemState( hNewItem, TVIS_SELECTED ) & TVIS_SELECTED ? 0 : TVIS_SELECTED;<br />
        UINT nOldSelectState = hOldItem == NULL ? 0 : GetItemState( hOldItem, TVIS_SELECTED );<br />
        <br />
        // select new item (to get focus)<br />
        SelectItem( hNewItem );<br />
        <br />
        // set highlight state for new item<br />
        if ( GetItemState( hNewItem, TVIS_SELECTED ) != nNewSelectState )<br />
          SetItemState( hNewItem, nNewSelectState, TVIS_SELECTED );<br />
        <br />
        // restore state of old selected item<br />
        if ( hOldItem != NULL && hOldItem != hNewItem && GetItemState( hOldItem, TVIS_SELECTED ) != nOldSelectState )<br />
          SetItemState( hOldItem, nOldSelectState, TVIS_SELECTED );<br />
        <br />
        m_hFirstSelected = NULL;<br />
      }<br />
    }<br />
    else if ( m_bShiftKey )<br />
    {<br />
      // initialise the reference item if this is the first shift selection<br />
      if ( m_hFirstSelected == NULL )<br />
        m_hFirstSelected = GetSelectedItem();<br />
        <br />
      // select new item (to get focus)<br />
      if ( hOldItem == hNewItem )<br />
        SelectItem( NULL ); // to prevent edit<br />
<br />
      if ( m_hFirstSelected != NULL )<br />
        SelectItems( m_hFirstSelected, hNewItem );<br />
    }<br />
    <br />
    m_bCtrlKey = FALSE;<br />
    m_bShiftKey = FALSE;<br />
  }<br />

135,144c135,136
< 		HTREEITEM hItem = GetNextVisibleItem( GetRootItem() );
< 		
< 		SetRedraw( FALSE );
< 		
< 		// clear selection upto the first item
< 		for ( ; hItem != NULL && hItem != hItemFrom && hItem != hItemTo; hItem = GetNextVisibleItem( hItem ) )
< 			SetItemState( hItem, 0, TVIS_SELECTED );
< 
< 		// is item visible?
< 		if ( hItem != NULL )
---
> 		HTREEITEM hItem = GetRootItem();
> 		for ( ; hItem != NULL; hItem = GetNextVisibleItem( hItem ) )
146,148c138,142
< 			SelectItem( hItemTo );
< 
< 			if ( hItem == hItemTo )
---
> 			if ( hItem == hItemFrom )
> 			{
> 				break;
> 			}
> 			else if ( hItem == hItemTo )
152a146
> 				break;
153,155c148,149
< 
< 			// go through remaining visible items
< 			for ( BOOL bSelectItem = TRUE; hItem != NULL; hItem = GetNextVisibleItem( hItem ) )
---
> 		}
> 		if ( hItem == NULL )
157,161c151
< 				SetItemState( hItem, bSelectItem ? TVIS_SELECTED : 0, TVIS_SELECTED );
< 
< 				// do we need to start removing items from selection?
< 				if ( hItem == hItemTo ) 
< 					bSelectItem = FALSE;
---
> 			return FALSE;
163a153,162
> 		
> 		SetRedraw( FALSE );
> 		
> 		// clear whole selection
> 		ResetSelected( GetRootItem() );
> 
> 		// is item visible?
> 		for ( hItem = hItemFrom; hItem != hItemTo; hItem = GetNextVisibleItem( hItem ) )
> 		{
> 			SetItemState( hItem, TVIS_SELECTED, TVIS_SELECTED );
164a164
> 		SetItemState( hItemTo, TVIS_SELECTED, TVIS_SELECTED );
242c243,245
< 		if ( ( nChar == VK_UP || nChar == VK_DOWN ) && GetKeyState( VK_SHIFT ) & 0x8000 )
---
> 		if ( m_bMultipleSelect && ( nChar == VK_UP || nChar == VK_DOWN ) )
> 		{
> 			if ( GetKeyState( VK_SHIFT ) & 0x8000 )
265c268,290
< 		else if ( nChar >= VK_SPACE )
---
> 			else	// no shift key
> 			{
> 				HTREEITEM hSel = GetSelectedItem();
> 				m_hFirstSelected = NULL;
> 				ResetSelected( GetRootItem() );
> 				SelectItem( hSel );
> 			}
> 		}
> 		else if ( nChar == VK_SPACE && (GetStyle() & TVS_CHECKBOXES) == TVS_CHECKBOXES )
> 		{
> 			HTREEITEM hSel = GetSelectedItem();
> 			BOOL bChecked = GetCheckState(hSel);
> 			CSimpleArray<HTREEITEM> aSel;
> 			GetSelectedItems(aSel);
> 			for (int ctr = 0; ctr < aSel.GetSize(); ctr++)
> 			{
> 				if (aSel[ctr] != hSel)
> 				{
> 					SetCheckState(aSel[ctr], !bChecked);
> 				}
> 			}
> 		}
> 		else if ( isalnum(nChar) )
300c325,326
< 			if ( HitTest( point, NULL ) != NULL )
---
> 			HTREEITEM hNewItem = HitTest( point, NULL );
> 			if ( hNewItem != NULL )
302a328
> 				UINT nItemSelState = GetItemState(hNewItem, TVIS_SELECTED) & TVIS_SELECTED;
303c330,331
< 				m_hFirstSelected = NULL;
---
> 				SetItemState(hNewItem, nItemSelState, TVIS_SELECTED);
> 				m_hFirstSelected = hNewItem;

"Could not include file atlapp.h. File not found."