// comctl32.dll versions
#define wxUSE_COMCTL32_SAFELY 0
-// Mingw32 is a bit mental even though this is done in winundef
-#ifdef GetFirstChild
- #undef GetFirstChild
-#endif
-
-#ifdef GetNextSibling
- #undef GetNextSibling
-#endif
-
-#if defined(__WIN95__)
-
#include "wx/app.h"
#include "wx/log.h"
#include "wx/dynarray.h"
#define TVM_SETTEXTCOLOR (TV_FIRST + 30)
#endif
-// a macro to hide the ugliness of nested casts
-#define HITEM(item) (HTREEITEM)(WXHTREEITEM)(item)
+// macros to hide the cast ugliness
+// --------------------------------
+
+// ptr is the real item id, i.e. wxTreeItemId::m_pItem
+#define HITEM_PTR(ptr) (HTREEITEM)(ptr)
+
+// item here is a wxTreeItemId
+#define HITEM(item) HITEM_PTR((item).m_pItem)
// the native control doesn't support multiple selections under MSW and we
// have 2 ways to emulate them: either using TVS_CHECKBOXES style and let
private:
wxArrayTreeItemIds& m_selections;
+
+ DECLARE_NO_COPY_CLASS(TraverseSelections)
};
// internal class for counting tree items
private:
size_t m_count;
+
+ DECLARE_NO_COPY_CLASS(TraverseCounter)
};
// ----------------------------------------------------------------------------
bool wxTreeTraversal::Traverse(const wxTreeItemId& root, bool recursively)
{
- long cookie;
+ wxTreeItemIdValue cookie;
wxTreeItemId child = m_tree->GetFirstChild(root, cookie);
while ( child.IsOk() )
{
m_textCtrl = NULL;
m_hasAnyAttr = false;
m_dragImage = NULL;
- m_htSelStart = 0;
m_pVirtualRoot = NULL;
// initialize the global array of events now as it can't be done statically
// delete any attributes
if ( m_hasAnyAttr )
{
- for ( wxNode *node = m_attrs.Next(); node; node = m_attrs.Next() )
- {
- delete (wxTreeItemAttr *)node->GetData();
- }
+ WX_CLEAR_HASH_MAP(wxMapTreeAttr, m_attrs);
// prevent TVN_DELETEITEM handler from deleting the attributes again!
m_hasAnyAttr = false;
wxImageList *wxTreeCtrl::GetStateImageList() const
{
- return m_imageListNormal;
+ return m_imageListState;
}
void wxTreeCtrl::SetAnyImageList(wxImageList *imageList, int which)
case wxTreeItemIcon_Normal:
{
const int imageNormalOld = GetItemImage(item);
- const int imageSelOld = GetItemSelectedImage(item);
+ const int imageSelOld =
+ GetItemImage(item, wxTreeItemIcon_Selected);
// always set the normal image
imageNormal = image;
// the wxTreeItemIndirectData GetItemXXXImage() will use it to
// get the images
imageNormal = GetItemImage(item);
- imageSel = GetItemSelectedImage(item);
+ imageSel = GetItemImage(item, wxTreeItemIcon_Selected);
// if it doesn't have it yet, add it
wxTreeItemIndirectData *data = new
wxColour wxTreeCtrl::GetItemTextColour(const wxTreeItemId& item) const
{
- long id = (long)(WXHTREEITEM)item;
- wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
- if ( !attr )
- {
- return wxNullColour;
- }
+ wxMapTreeAttr::const_iterator it = m_attrs.find(item.m_pItem);
- return attr->GetTextColour();
+ return it == m_attrs.end() ? wxNullColour : it->second->GetTextColour();
}
wxColour wxTreeCtrl::GetItemBackgroundColour(const wxTreeItemId& item) const
{
- long id = (long)(WXHTREEITEM)item;
- wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
- if ( !attr )
- {
- return wxNullColour;
- }
+ wxMapTreeAttr::const_iterator it = m_attrs.find(item.m_pItem);
- return attr->GetBackgroundColour();
+ return it == m_attrs.end() ? wxNullColour : it->second->GetBackgroundColour();
}
wxFont wxTreeCtrl::GetItemFont(const wxTreeItemId& item) const
{
- long id = (long)(WXHTREEITEM)item;
- wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
- if ( !attr )
- {
- return wxNullFont;
- }
+ wxMapTreeAttr::const_iterator it = m_attrs.find(item.m_pItem);
- return attr->GetFont();
+ return it == m_attrs.end() ? wxNullFont : it->second->GetFont();
}
void wxTreeCtrl::SetItemTextColour(const wxTreeItemId& item,
const wxColour& col)
{
- m_hasAnyAttr = true;
-
- long id = (long)(WXHTREEITEM)item;
- wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
- if ( !attr )
+ wxTreeItemAttr *attr;
+ wxMapTreeAttr::iterator it = m_attrs.find(item.m_pItem);
+ if ( it == m_attrs.end() )
{
+ m_hasAnyAttr = true;
+
+ m_attrs[item.m_pItem] =
attr = new wxTreeItemAttr;
- m_attrs.Put(id, (wxObject *)attr);
+ }
+ else
+ {
+ attr = it->second;
}
attr->SetTextColour(col);
void wxTreeCtrl::SetItemBackgroundColour(const wxTreeItemId& item,
const wxColour& col)
{
- m_hasAnyAttr = true;
-
- long id = (long)(WXHTREEITEM)item;
- wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
- if ( !attr )
+ wxTreeItemAttr *attr;
+ wxMapTreeAttr::iterator it = m_attrs.find(item.m_pItem);
+ if ( it == m_attrs.end() )
{
+ m_hasAnyAttr = true;
+
+ m_attrs[item.m_pItem] =
attr = new wxTreeItemAttr;
- m_attrs.Put(id, (wxObject *)attr);
+ }
+ else // already in the hash
+ {
+ attr = it->second;
}
attr->SetBackgroundColour(col);
void wxTreeCtrl::SetItemFont(const wxTreeItemId& item, const wxFont& font)
{
- m_hasAnyAttr = true;
-
- long id = (long)(WXHTREEITEM)item;
- wxTreeItemAttr *attr = (wxTreeItemAttr *)m_attrs.Get(id);
- if ( !attr )
+ wxTreeItemAttr *attr;
+ wxMapTreeAttr::iterator it = m_attrs.find(item.m_pItem);
+ if ( it == m_attrs.end() )
{
+ m_hasAnyAttr = true;
+
+ m_attrs[item.m_pItem] =
attr = new wxTreeItemAttr;
- m_attrs.Put(id, (wxObject *)attr);
+ }
+ else // already in the hash
+ {
+ attr = it->second;
}
attr->SetFont(font);
// this ugliness comes directly from MSDN - it *is* the correct way to pass
// the HTREEITEM with TVM_GETITEMRECT
- *(WXHTREEITEM *)&rect = (WXHTREEITEM)item;
+ *(HTREEITEM *)&rect = HITEM(item);
// false means get item rect for the whole item, not only text
return SendMessage(GetHwnd(), TVM_GETITEMRECT, false, (LPARAM)&rect) != 0;
if ( GET_VIRTUAL_ROOT() )
return TVI_ROOT;
- return wxTreeItemId((WXHTREEITEM) TreeView_GetRoot(GetHwnd()));
+ return wxTreeItemId(TreeView_GetRoot(GetHwnd()));
}
wxTreeItemId wxTreeCtrl::GetSelection() const
{
- wxCHECK_MSG( !(m_windowStyle & wxTR_MULTIPLE), (long)(WXHTREEITEM)0,
+ wxCHECK_MSG( !(m_windowStyle & wxTR_MULTIPLE), wxTreeItemId(),
wxT("this only works with single selection controls") );
- return wxTreeItemId((WXHTREEITEM) TreeView_GetSelection(GetHwnd()));
+ return wxTreeItemId(TreeView_GetSelection(GetHwnd()));
}
wxTreeItemId wxTreeCtrl::GetItemParent(const wxTreeItemId& item) const
}
}
- return wxTreeItemId((WXHTREEITEM)hItem);
+ return wxTreeItemId(hItem);
}
wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item,
- long& _cookie) const
+ wxTreeItemIdValue& cookie) const
{
// remember the last child returned in 'cookie'
- _cookie = (long)TreeView_GetChild(GetHwnd(), HITEM(item));
+ cookie = TreeView_GetChild(GetHwnd(), HITEM(item));
+
+ return wxTreeItemId(cookie);
+}
+
+wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& WXUNUSED(item),
+ wxTreeItemIdValue& cookie) const
+{
+ wxTreeItemId item(TreeView_GetNextSibling(GetHwnd(),
+ HITEM(wxTreeItemId(cookie))));
+ cookie = item.m_pItem;
- return wxTreeItemId((WXHTREEITEM)_cookie);
+ return item;
+}
+
+#if WXWIN_COMPATIBILITY_2_4
+
+wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item,
+ long& cookie) const
+{
+ cookie = (long)TreeView_GetChild(GetHwnd(), HITEM(item));
+
+ return wxTreeItemId((void *)cookie);
}
wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& WXUNUSED(item),
- long& _cookie) const
+ long& cookie) const
{
- wxTreeItemId l = wxTreeItemId((WXHTREEITEM)TreeView_GetNextSibling(GetHwnd(),
- HITEM(_cookie)));
- _cookie = (long)l;
+ wxTreeItemId item(TreeView_GetNextSibling
+ (
+ GetHwnd(),
+ HITEM(wxTreeItemId((void *)cookie)
+ )));
+ cookie = (long)item.m_pItem;
- return l;
+ return item;
}
+#endif // WXWIN_COMPATIBILITY_2_4
+
wxTreeItemId wxTreeCtrl::GetLastChild(const wxTreeItemId& item) const
{
// can this be done more efficiently?
- long cookie;
+ wxTreeItemIdValue cookie;
wxTreeItemId childLast,
child = GetFirstChild(item, cookie);
wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
{
- return wxTreeItemId((WXHTREEITEM) TreeView_GetNextSibling(GetHwnd(), HITEM(item)));
+ return wxTreeItemId(TreeView_GetNextSibling(GetHwnd(), HITEM(item)));
}
wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
{
- return wxTreeItemId((WXHTREEITEM) TreeView_GetPrevSibling(GetHwnd(), HITEM(item)));
+ return wxTreeItemId(TreeView_GetPrevSibling(GetHwnd(), HITEM(item)));
}
wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const
{
- return wxTreeItemId((WXHTREEITEM) TreeView_GetFirstVisible(GetHwnd()));
+ return wxTreeItemId(TreeView_GetFirstVisible(GetHwnd()));
}
wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
{
wxASSERT_MSG( IsVisible(item), wxT("The item you call GetNextVisible() for must be visible itself!"));
- return wxTreeItemId((WXHTREEITEM) TreeView_GetNextVisible(GetHwnd(), HITEM(item)));
+ return wxTreeItemId(TreeView_GetNextVisible(GetHwnd(), HITEM(item)));
}
wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
{
wxASSERT_MSG( IsVisible(item), wxT("The item you call GetPrevVisible() for must be visible itself!"));
- return wxTreeItemId((WXHTREEITEM) TreeView_GetPrevVisible(GetHwnd(), HITEM(item)));
+ return wxTreeItemId(TreeView_GetPrevVisible(GetHwnd(), HITEM(item)));
}
// ----------------------------------------------------------------------------
if ( data != NULL )
{
// associate the application tree item with Win32 tree item handle
- data->SetId((WXHTREEITEM)id);
+ data->SetId(id);
}
- return wxTreeItemId((WXHTREEITEM)id);
+ return wxTreeItemId(id);
}
// for compatibility only
+#if WXWIN_COMPATIBILITY_2_4
+
wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parent,
const wxString& text,
int image, int selImage,
long insertAfter)
{
- return DoInsertItem(parent, (WXHTREEITEM)insertAfter, text,
+ return DoInsertItem(parent, wxTreeItemId((void *)insertAfter), text,
image, selImage, NULL);
}
+#endif // WXWIN_COMPATIBILITY_2_4
+
wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text,
int image, int selectedImage,
wxTreeItemData *data)
return TVI_ROOT;
}
- return DoInsertItem(wxTreeItemId((long)(WXHTREEITEM) 0), (long)(WXHTREEITEM) 0,
+ return DoInsertItem(wxTreeItemId(), wxTreeItemId(),
text, image, selectedImage, data);
}
int image, int selectedImage,
wxTreeItemData *data)
{
- return DoInsertItem(parent, (WXHTREEITEM) TVI_FIRST,
+ return DoInsertItem(parent, TVI_FIRST,
text, image, selectedImage, data);
}
wxTreeItemData *data)
{
// find the item from index
- long cookie;
+ wxTreeItemIdValue cookie;
wxTreeItemId idPrev, idCur = GetFirstChild(parent, cookie);
while ( index != 0 && idCur.IsOk() )
{
int image, int selectedImage,
wxTreeItemData *data)
{
- return DoInsertItem(parent, (WXHTREEITEM) TVI_LAST,
+ return DoInsertItem(parent, TVI_LAST,
text, image, selectedImage, data);
}
// delete all children (but don't delete the item itself)
void wxTreeCtrl::DeleteChildren(const wxTreeItemId& item)
{
- long cookie;
+ wxTreeItemIdValue cookie;
- wxArrayLong children;
+ wxArrayTreeItemIds children;
wxTreeItemId child = GetFirstChild(item, cookie);
while ( child.IsOk() )
{
- children.Add((long)(WXHTREEITEM)child);
+ children.Add(child);
child = GetNextChild(item, cookie);
}
size_t nCount = children.Count();
for ( size_t n = 0; n < nCount; n++ )
{
- if ( !TreeView_DeleteItem(GetHwnd(), (HTREEITEM)children[n]) )
+ if ( !TreeView_DeleteItem(GetHwnd(), HITEM_PTR(children[n])) )
{
wxLogLastError(wxT("TreeView_DeleteItem"));
}
DoExpand(item, TVE_TOGGLE);
}
+#if WXWIN_COMPATIBILITY_2_4
void wxTreeCtrl::ExpandItem(const wxTreeItemId& item, int action)
{
DoExpand(item, action);
}
+#endif
void wxTreeCtrl::Unselect()
{
wxT("doesn't make sense, may be you want UnselectAll()?") );
// just remove the selection
- SelectItem(wxTreeItemId((long) (WXHTREEITEM) 0));
+ SelectItem(wxTreeItemId());
}
void wxTreeCtrl::UnselectAll()
for ( size_t n = 0; n < count; n++ )
{
#if wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
- SetItemCheck(selections[n], false);
+ SetItemCheck(HITEM_PTR(selections[n]), false);
#else // !wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
- ::UnselectItem(GetHwnd(), HITEM(selections[n]));
+ ::UnselectItem(GetHwnd(), HITEM_PTR(selections[n]));
#endif // wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE/!wxUSE_CHECKBOXES_IN_MULTI_SEL_TREE
}
}
#undef TRANSLATE_FLAG
- return wxTreeItemId((WXHTREEITEM) hitTestInfo.hItem);
+ return wxTreeItemId(hitTestInfo.hItem);
}
bool wxTreeCtrl::GetBoundingRect(const wxTreeItemId& item,
::SetFocus(GetHwnd(), htItem);
// reset on any click without Shift
- m_htSelStart = 0;
+ m_htSelStart.Unset();
processed = true;
}
if ( !m_htSelStart )
{
// take the focused item
- m_htSelStart = (WXHTREEITEM)
- TreeView_GetSelection(GetHwnd());
+ m_htSelStart = TreeView_GetSelection(GetHwnd());
}
SelectRange(GetHwnd(), HITEM(m_htSelStart), htItem,
{
// avoid doing anything if we click on the only
// currently selected item
+
wxArrayTreeItemIds selections;
size_t count = GetSelections(selections);
if ( count == 0 ||
- count > 1 ||
- HITEM(selections[0]) != htItem )
+ count > 1 ||
+ HITEM_PTR(selections[0]) != htItem )
{
- // clear the previously selected items
- UnselectAll();
-
- // prevent the click from starting in-place editing
- // which should only happen if we click on the
- // already selected item (and nothing else is
- // selected)
- TreeView_SelectItem(GetHwnd(), 0);
+ // clear the previously selected items, if the
+ // user clicked outside of the present selection.
+ // otherwise, perform the deselection on mouse-up.
+ // this allows multiple drag and drop to work.
+
+ if (IsItemSelected(GetHwnd(), htItem))
+ {
+ ::SetFocus(GetHwnd(), htItem);
+ }
+ else
+ {
+ UnselectAll();
+
+ // prevent the click from starting in-place editing
+ // which should only happen if we click on the
+ // already selected item (and nothing else is
+ // selected)
+
+ TreeView_SelectItem(GetHwnd(), 0);
+ ::SelectItem(GetHwnd(), htItem);
+ }
}
// reset on any click without Shift
- m_htSelStart = 0;
+ m_htSelStart.Unset();
}
}
break;
break;
case WM_LBUTTONUP:
+
+ // facilitates multiple drag-and-drop
+ if (htItem && isMultiple)
+ {
+ wxArrayTreeItemIds selections;
+ size_t count = GetSelections(selections);
+
+ if (count > 1 &&
+ !(wParam & MK_CONTROL) &&
+ !(wParam & MK_SHIFT))
+ {
+ UnselectAll();
+ TreeView_SelectItem(GetHwnd(), htItem);
+ }
+ }
+
+ // fall through
+
case WM_RBUTTONUP:
if ( m_dragImage )
{
// generate the drag end event
wxTreeEvent event(wxEVT_COMMAND_TREE_END_DRAG, m_windowId);
- event.m_item = (WXHTREEITEM)htItem;
+ event.m_item = htItem;
event.m_pointDrag = wxPoint(x, y);
event.SetEventObject(this);
{
// TreeView_GetItemRect() will return false if item is not visible,
// which may happen perfectly well
- if ( TreeView_GetItemRect(GetHwnd(), HITEM(selections[n]),
+ if ( TreeView_GetItemRect(GetHwnd(), HITEM_PTR(selections[n]),
&rect, true) )
{
::InvalidateRect(GetHwnd(), &rect, false);
HTREEITEM htSel = (HTREEITEM)TreeView_GetSelection(GetHwnd());
if ( !m_htSelStart )
{
- m_htSelStart = (WXHTREEITEM)htSel;
+ m_htSelStart = htSel;
}
if ( wParam == VK_SPACE )
NM_TREEVIEW *tv = (NM_TREEVIEW *)lParam;
- event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
+ event.m_item = tv->itemNew.hItem;
event.m_pointDrag = wxPoint(tv->ptDrag.x, tv->ptDrag.y);
// don't allow dragging by default: the user code must
eventType = wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT;
TV_DISPINFO *info = (TV_DISPINFO *)lParam;
- event.m_item = (WXHTREEITEM) info->item.hItem;
+ event.m_item = info->item.hItem;
event.m_label = info->item.pszText;
event.m_editCancelled = false;
}
eventType = wxEVT_COMMAND_TREE_DELETE_ITEM;
NM_TREEVIEW *tv = (NM_TREEVIEW *)lParam;
- event.m_item = (WXHTREEITEM)tv->itemOld.hItem;
+ event.m_item = tv->itemOld.hItem;
if ( m_hasAnyAttr )
{
- delete (wxTreeItemAttr *)m_attrs.
- Delete((long)tv->itemOld.hItem);
+ wxMapTreeAttr::iterator it = m_attrs.find(tv->itemOld.hItem);
+ if ( it != m_attrs.end() )
+ {
+ delete it->second;
+ m_attrs.erase(it);
+ }
}
}
break;
eventType = wxEVT_COMMAND_TREE_END_LABEL_EDIT;
TV_DISPINFO *info = (TV_DISPINFO *)lParam;
- event.m_item = (WXHTREEITEM)info->item.hItem;
+ event.m_item = info->item.hItem;
event.m_label = info->item.pszText;
if (info->item.pszText == NULL)
{
TV_DISPINFO *info = (TV_DISPINFO *)lParam;
- event.m_item = (WXHTREEITEM) info->item.hItem;
+ event.m_item = info->item.hItem;
break;
}
eventType = gs_expandEvents[what][how];
- event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
+ event.m_item = tv->itemNew.hItem;
}
break;
// similar to the ones from a "real" WM_KEYDOWN so that
// CreateKeyEvent() works correctly
WXLPARAM lParam =
-// (::GetKeyState(VK_MENU) & 0x100 ? KF_ALTDOWN : 0) << 16;
- // Returns different negative values on WinME and WinNT,
- // so simply test for negative value.
(::GetKeyState(VK_MENU) < 0 ? KF_ALTDOWN : 0) << 16;
WXWPARAM wParam = info->wVKey;
hdr->code == TVN_SELCHANGEDW)
{
NM_TREEVIEWW* tv = (NM_TREEVIEWW *)lParam;
- event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
- event.m_itemOld = (WXHTREEITEM) tv->itemOld.hItem;
+ event.m_item = tv->itemNew.hItem;
+ event.m_itemOld = tv->itemOld.hItem;
}
else
{
NM_TREEVIEWA* tv = (NM_TREEVIEWA *)lParam;
- event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
- event.m_itemOld = (WXHTREEITEM) tv->itemOld.hItem;
+ event.m_item = tv->itemNew.hItem;
+ event.m_itemOld = tv->itemOld.hItem;
}
}
break;
case CDDS_ITEMPREPAINT:
{
- wxTreeItemAttr *attr =
- (wxTreeItemAttr *)m_attrs.Get(nmcd.dwItemSpec);
+ wxMapTreeAttr::iterator
+ it = m_attrs.find((void *)nmcd.dwItemSpec);
- if ( !attr )
+ if ( it == m_attrs.end() )
{
// nothing to do for this item
*result = CDRF_DODEFAULT;
break;
}
+ wxTreeItemAttr * const attr = it->second;
+
HFONT hFont;
if ( attr->HasFont() )
{
return true;
#endif // _WIN32_IE >= 0x300
+ case NM_CLICK:
+ {
+ DWORD pos = GetMessagePos();
+ POINT point;
+ point.x = LOWORD(pos);
+ point.y = HIWORD(pos);
+ ::MapWindowPoints(HWND_DESKTOP, GetHwnd(), &point, 1);
+ int flags = 0;
+ wxTreeItemId item = HitTest(wxPoint(point.x, point.y), flags);
+ if (flags & wxTREE_HITTEST_ONITEMSTATEICON)
+ {
+ event.m_item = item;
+ eventType = wxEVT_COMMAND_TREE_STATE_IMAGE_CLICK;
+ }
+ break;
+ }
+
case NM_DBLCLK:
case NM_RCLICK:
{
{
if ( tvhti.flags & TVHT_ONITEM )
{
- event.m_item = (WXHTREEITEM) tvhti.hItem;
+ event.m_item = tvhti.hItem;
eventType = (int)hdr->code == NM_DBLCLK
? wxEVT_COMMAND_TREE_ITEM_ACTIVATED
: wxEVT_COMMAND_TREE_ITEM_RIGHT_CLICK;
// comctl32.dll or our code?
{
NM_TREEVIEW* tv = (NM_TREEVIEW *)lParam;
- wxTreeItemId id = (WXHTREEITEM)tv->itemNew.hItem;
+ wxTreeItemId id(tv->itemNew.hItem);
int image = GetItemImage(id, wxTreeItemIcon_Expanded);
if ( image != -1 )
return processed;
}
-#endif // __WIN95__
+// ----------------------------------------------------------------------------
+// State control.
+// ----------------------------------------------------------------------------
+
+// why do they define INDEXTOSTATEIMAGEMASK but not the inverse?
+#define STATEIMAGEMASKTOINDEX(state) (((state) & TVIS_STATEIMAGEMASK) >> 12)
+
+void wxTreeCtrl::SetState(const wxTreeItemId& node, int state)
+{
+ TV_ITEM tvi;
+ tvi.hItem = (HTREEITEM)node.m_pItem;
+ tvi.mask = TVIF_STATE;
+ tvi.stateMask = TVIS_STATEIMAGEMASK;
+
+ // Select the specified state, or -1 == cycle to the next one.
+ if ( state == -1 )
+ {
+ TreeView_GetItem(GetHwnd(), &tvi);
+
+ state = STATEIMAGEMASKTOINDEX(tvi.state) + 1;
+ if ( state == m_imageListState->GetImageCount() )
+ state = 1;
+ }
+
+ wxCHECK_RET( state < m_imageListState->GetImageCount(),
+ _T("wxTreeCtrl::SetState(): item index out of bounds") );
+
+ tvi.state = INDEXTOSTATEIMAGEMASK(state);
+
+ TreeView_SetItem(GetHwnd(), &tvi);
+}
+
+int wxTreeCtrl::GetState(const wxTreeItemId& node)
+{
+ TV_ITEM tvi;
+ tvi.hItem = (HTREEITEM)node.m_pItem;
+ tvi.mask = TVIF_STATE;
+ tvi.stateMask = TVIS_STATEIMAGEMASK;
+ TreeView_GetItem(GetHwnd(), &tvi);
+
+ return STATEIMAGEMASKTOINDEX(tvi.state);
+}
#endif // wxUSE_TREECTRL
+