X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/16f32ee7652c15bdbe46a85cdc0ecf036bd4003e..bdcb213757b966746ffc73bf01ec2e838d867cd3:/src/common/treebase.cpp diff --git a/src/common/treebase.cpp b/src/common/treebase.cpp index 08220fa503..f10e2af927 100644 --- a/src/common/treebase.cpp +++ b/src/common/treebase.cpp @@ -183,6 +183,9 @@ wxSize wxTreeCtrlBase::DoGetBestSize() const void wxTreeCtrlBase::ExpandAll() { + if ( IsEmpty() ) + return; + ExpandAllChildren(GetRootItem()); } @@ -202,5 +205,33 @@ void wxTreeCtrlBase::ExpandAllChildren(const wxTreeItemId& item) } } +void wxTreeCtrlBase::CollapseAll() +{ + if ( IsEmpty() ) + return; + + CollapseAllChildren(GetRootItem()); +} + +void wxTreeCtrlBase::CollapseAllChildren(const wxTreeItemId& item) +{ + // first (recursively) collapse all the children + wxTreeItemIdValue cookie; + for ( wxTreeItemId idCurr = GetFirstChild(item, cookie); + idCurr.IsOk(); + idCurr = GetNextChild(item, cookie) ) + { + CollapseAllChildren(idCurr); + } + + // then collapse this element too + Collapse(item); +} + +bool wxTreeCtrlBase::IsEmpty() const +{ + return !GetRootItem().IsOk(); +} + #endif // wxUSE_TREECTRL