X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/e55ad60e195f1007db921b2a73a3cac98ed9df65..8a0681f9338345d0ef32b2ad809d49fd20861119:/src/generic/treectrl.cpp diff --git a/src/generic/treectrl.cpp b/src/generic/treectrl.cpp index 109827378a..36f8be5bfa 100644 --- a/src/generic/treectrl.cpp +++ b/src/generic/treectrl.cpp @@ -1,193 +1,378 @@ ///////////////////////////////////////////////////////////////////////////// // Name: treectrl.cpp -// Purpose: +// Purpose: generic tree control implementation // Author: Robert Roebling // Created: 01/02/97 +// Modified: 22/10/98 - almost total rewrite, simpler interface (VZ) // Id: $Id$ // Copyright: (c) 1998 Robert Roebling, Julian Smart and Markus Holzem // Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// +// ============================================================================= +// declarations +// ============================================================================= + +// ----------------------------------------------------------------------------- +// headers +// ----------------------------------------------------------------------------- + #ifdef __GNUG__ -#pragma implementation "treectrl.h" + #pragma implementation "treectrl.h" +#endif + +// For compilers that support precompilation, includes "wx.h". +#include "wx/wxprec.h" + +#ifdef __BORLANDC__ +#pragma hdrstop #endif #include "wx/treectrl.h" +#include "wx/generic/imaglist.h" #include "wx/settings.h" #include "wx/log.h" -#include +#include "wx/intl.h" +#include "wx/dynarray.h" +#include "wx/arrimpl.cpp" +#include "wx/dcclient.h" +#include "wx/msgdlg.h" -//----------------------------------------------------------------------------- -// wxTreeItem -//----------------------------------------------------------------------------- +// ----------------------------------------------------------------------------- +// array types +// ----------------------------------------------------------------------------- -IMPLEMENT_DYNAMIC_CLASS(wxTreeItem, wxObject) +class WXDLLEXPORT wxGenericTreeItem; -wxTreeItem::wxTreeItem() +WX_DEFINE_ARRAY(wxGenericTreeItem *, wxArrayGenericTreeItems); +WX_DEFINE_OBJARRAY(wxArrayTreeItemIds); + +// ---------------------------------------------------------------------------- +// constants +// ---------------------------------------------------------------------------- + +static const int NO_IMAGE = -1; + +// ----------------------------------------------------------------------------- +// private classes +// ----------------------------------------------------------------------------- + +// a tree item +class WXDLLEXPORT wxGenericTreeItem { - m_mask = 0; - m_itemId = 0; - m_state = 0; - m_stateMask = 0; - m_image = -1; - m_selectedImage = -1; - m_children = 0; - m_data = 0; -} +public: + // ctors & dtor + wxGenericTreeItem() { m_data = NULL; } + wxGenericTreeItem( wxGenericTreeItem *parent, + const wxString& text, + wxDC& dc, + int image, int selImage, + wxTreeItemData *data ); -//----------------------------------------------------------------------------- -// wxTreeEvent -//----------------------------------------------------------------------------- + ~wxGenericTreeItem(); + + // trivial accessors + wxArrayGenericTreeItems& GetChildren() { return m_children; } + + const wxString& GetText() const { return m_text; } + int GetImage(wxTreeItemIcon which = wxTreeItemIcon_Normal) const + { return m_images[which]; } + wxTreeItemData *GetData() const { return m_data; } + + // returns the current image for the item (depending on its + // selected/expanded/whatever state) + int GetCurrentImage() const; + + void SetText( const wxString &text ); + void SetImage(int image, wxTreeItemIcon which) { m_images[which] = image; } + void SetData(wxTreeItemData *data) { m_data = data; } + + void SetHasPlus(bool has = TRUE) { m_hasPlus = has; } + + void SetBold(bool bold) { m_isBold = bold; } + + int GetX() const { return m_x; } + int GetY() const { return m_y; } + + void SetX(int x) { m_x = x; } + void SetY(int y) { m_y = y; } + + int GetHeight() const { return m_height; } + int GetWidth() const { return m_width; } + + void SetHeight(int h) { m_height = h; } + void SetWidth(int w) { m_width = w; } + + + wxGenericTreeItem *GetParent() const { return m_parent; } + + // operations + // deletes all children notifying the treectrl about it if !NULL + // pointer given + void DeleteChildren(wxTreeCtrl *tree = NULL); + // FIXME don't know what is it for + void Reset(); -IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent,wxCommandEvent) + // get count of all children (and grand children if 'recursively') + size_t GetChildrenCount(bool recursively = TRUE) const; -wxTreeEvent::wxTreeEvent( wxEventType commandType, int id ) : - wxCommandEvent( commandType, id ) + void Insert(wxGenericTreeItem *child, size_t index) + { m_children.Insert(child, index); } + + void SetCross( int x, int y ); + void GetSize( int &x, int &y, const wxTreeCtrl* ); + + // return the item at given position (or NULL if no item), onButton is + // TRUE if the point belongs to the item's button, otherwise it lies + // on the button's label + wxGenericTreeItem *HitTest( const wxPoint& point, const wxTreeCtrl *, int &flags); + + void Expand() { m_isCollapsed = FALSE; } + void Collapse() { m_isCollapsed = TRUE; } + + void SetHilight( bool set = TRUE ) { m_hasHilight = set; } + + // status inquiries + bool HasChildren() const { return !m_children.IsEmpty(); } + bool IsSelected() const { return m_hasHilight; } + bool IsExpanded() const { return !m_isCollapsed; } + bool HasPlus() const { return m_hasPlus || HasChildren(); } + bool IsBold() const { return m_isBold; } + + // attributes + // get them - may be NULL + wxTreeItemAttr *GetAttributes() const { return m_attr; } + // get them ensuring that the pointer is not NULL + wxTreeItemAttr& Attr() + { + if ( !m_attr ) + m_attr = new wxTreeItemAttr; + + return *m_attr; + } + +private: + wxString m_text; + + // tree ctrl images for the normal, selected, expanded and + // expanded+selected states + int m_images[wxTreeItemIcon_Max]; + + wxTreeItemData *m_data; + + // use bitfields to save size + int m_isCollapsed :1; + int m_hasHilight :1; // same as focused + int m_hasPlus :1; // used for item which doesn't have + // children but has a [+] button + int m_isBold :1; // render the label in bold font + + int m_x, m_y; + long m_height, m_width; + int m_xCross, m_yCross; + int m_level; + + wxArrayGenericTreeItems m_children; + wxGenericTreeItem *m_parent; + + wxTreeItemAttr *m_attr; +}; + +// ============================================================================= +// implementation +// ============================================================================= + + +// ----------------------------------------------------------------------------- +// wxTreeRenameTimer (internal) +// ----------------------------------------------------------------------------- + +wxTreeRenameTimer::wxTreeRenameTimer( wxTreeCtrl *owner ) { - m_code = 0; - m_oldItem = 0; + m_owner = owner; +} + +void wxTreeRenameTimer::Notify() +{ + m_owner->OnRenameTimer(); } //----------------------------------------------------------------------------- -// wxGenericTreeItem +// wxTreeTextCtrl (internal) //----------------------------------------------------------------------------- -IMPLEMENT_DYNAMIC_CLASS(wxGenericTreeItem,wxObject) +IMPLEMENT_DYNAMIC_CLASS(wxTreeTextCtrl,wxTextCtrl); + +BEGIN_EVENT_TABLE(wxTreeTextCtrl,wxTextCtrl) + EVT_CHAR (wxTreeTextCtrl::OnChar) + EVT_KILL_FOCUS (wxTreeTextCtrl::OnKillFocus) +END_EVENT_TABLE() -wxGenericTreeItem::wxGenericTreeItem( wxGenericTreeItem *parent ) +wxTreeTextCtrl::wxTreeTextCtrl( wxWindow *parent, const wxWindowID id, + bool *accept, wxString *res, wxTreeCtrl *owner, + const wxString &value, const wxPoint &pos, const wxSize &size, +#if wxUSE_VALIDATORS + int style, const wxValidator& validator, const wxString &name ) : +#endif + wxTextCtrl( parent, id, value, pos, size, style, validator, name ) { - Reset(); - m_parent = parent; - m_hasHilight = FALSE; + m_res = res; + m_accept = accept; + m_owner = owner; + (*m_accept) = FALSE; + (*m_res) = ""; + m_startValue = value; } -wxGenericTreeItem::wxGenericTreeItem( wxGenericTreeItem *parent, const wxTreeItem& item, wxDC *dc ) +void wxTreeTextCtrl::OnChar( wxKeyEvent &event ) { - Reset(); - SetItem( item, dc ); - m_parent = parent; - m_hasHilight = FALSE; + if (event.m_keyCode == WXK_RETURN) + { + (*m_accept) = TRUE; + (*m_res) = GetValue(); + m_owner->SetFocus(); + return; + } + if (event.m_keyCode == WXK_ESCAPE) + { + (*m_accept) = FALSE; + (*m_res) = ""; + m_owner->SetFocus(); + return; + } + event.Skip(); } -void wxGenericTreeItem::SetItem( const wxTreeItem &item, wxDC *dc ) +void wxTreeTextCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) ) { - if ((item.m_mask & wxTREE_MASK_HANDLE) == wxTREE_MASK_HANDLE) - m_itemId = item.m_itemId; - if ((item.m_mask & wxTREE_MASK_STATE) == wxTREE_MASK_STATE) - m_state = item.m_state; - if ((item.m_mask & wxTREE_MASK_TEXT) == wxTREE_MASK_TEXT) - m_text = item.m_text; - if ((item.m_mask & wxTREE_MASK_IMAGE) == wxTREE_MASK_IMAGE) - m_image = item.m_image; - if ((item.m_mask & wxTREE_MASK_SELECTED_IMAGE) == wxTREE_MASK_SELECTED_IMAGE) - m_selectedImage = item.m_selectedImage; - if ((item.m_mask & wxTREE_MASK_CHILDREN) == wxTREE_MASK_CHILDREN) - m_hasChildren = (item.m_children > 0); - if ((item.m_mask & wxTREE_MASK_DATA) == wxTREE_MASK_DATA) - m_data = item.m_data; - long lw = 0; - long lh = 0; - dc->GetTextExtent( m_text, &lw, &lh ); - m_width = lw; - m_height = lh; + if (wxPendingDelete.Member(this)) return; + + wxPendingDelete.Append(this); + + if ((*m_accept) && ((*m_res) != m_startValue)) + m_owner->OnRenameAccept(); } -void wxGenericTreeItem::SetText( const wxString &text, wxDC *dc ) +#define PIXELS_PER_UNIT 10 +// ----------------------------------------------------------------------------- +// wxTreeEvent +// ----------------------------------------------------------------------------- + +IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent, wxNotifyEvent) + +wxTreeEvent::wxTreeEvent( wxEventType commandType, int id ) + : wxNotifyEvent( commandType, id ) { - m_text = text; - long lw = 0; - long lh = 0; - dc->GetTextExtent( m_text, &lw, &lh ); - m_width = lw; - m_height = lh; + m_code = 0; + m_itemOld = (wxGenericTreeItem *)NULL; } -void wxGenericTreeItem::Reset() +// ----------------------------------------------------------------------------- +// wxGenericTreeItem +// ----------------------------------------------------------------------------- + +wxGenericTreeItem::wxGenericTreeItem(wxGenericTreeItem *parent, + const wxString& text, + wxDC& dc, + int image, int selImage, + wxTreeItemData *data) + : m_text(text) { - m_itemId = -1; - m_state = 0; - m_text = ""; - m_image = -1; - m_selectedImage = -1; -// m_children = 0; - m_hasChildren = FALSE; - m_data = 0; - m_x = 0; - m_y = 0; - m_height = 0; - m_width = 0; - m_xCross = 0; - m_yCross = 0; + m_images[wxTreeItemIcon_Normal] = image; + m_images[wxTreeItemIcon_Selected] = selImage; + m_images[wxTreeItemIcon_Expanded] = NO_IMAGE; + m_images[wxTreeItemIcon_SelectedExpanded] = NO_IMAGE; + + m_data = data; + m_x = m_y = 0; + m_xCross = m_yCross = 0; + m_level = 0; - m_children.DeleteContents( TRUE ); + m_isCollapsed = TRUE; - m_parent = (wxGenericTreeItem *) NULL; -} + m_hasHilight = FALSE; + m_hasPlus = FALSE; + m_isBold = FALSE; -void wxGenericTreeItem::GetItem( wxTreeItem &item ) const -{ - if ((item.m_mask & wxTREE_MASK_STATE) == wxTREE_MASK_STATE) - item.m_state = m_state; - if ((item.m_mask & wxTREE_MASK_TEXT) == wxTREE_MASK_TEXT) - item.m_text = m_text; - if ((item.m_mask & wxTREE_MASK_IMAGE) == wxTREE_MASK_IMAGE) - item.m_image = m_image; - if ((item.m_mask & wxTREE_MASK_SELECTED_IMAGE) == wxTREE_MASK_SELECTED_IMAGE) - item.m_selectedImage = m_selectedImage; - if ((item.m_mask & wxTREE_MASK_CHILDREN) == wxTREE_MASK_CHILDREN) - item.m_children = (int)m_hasChildren; - if ((item.m_mask & wxTREE_MASK_DATA) == wxTREE_MASK_DATA) - item.m_data = m_data; -} + m_parent = parent; -bool wxGenericTreeItem::HasChildren() -{ - return m_hasChildren; + m_attr = (wxTreeItemAttr *)NULL; + + dc.GetTextExtent( m_text, &m_width, &m_height ); + // TODO : Add the width of the image + // PB : We don't know which image is shown (image, selImage) + // We don't even know imageList from the treectrl this item belongs to !!! + // At this point m_width doesn't mean much, this can be remove ! } -bool wxGenericTreeItem::HasPlus() +wxGenericTreeItem::~wxGenericTreeItem() { - if ( !HasChildren() ) - return FALSE; + delete m_data; + + delete m_attr; - return !IsExpanded(); + wxASSERT_MSG( m_children.IsEmpty(), + wxT("please call DeleteChildren() before deleting the item") ); } -int wxGenericTreeItem::NumberOfVisibleDescendents() +void wxGenericTreeItem::DeleteChildren(wxTreeCtrl *tree) { - int ret = m_children.Number(); - wxNode *node = m_children.First(); - while (node) + size_t count = m_children.Count(); + for ( size_t n = 0; n < count; n++ ) { - wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data(); - ret += item->NumberOfVisibleDescendents(); - node = node->Next(); + wxGenericTreeItem *child = m_children[n]; + if ( tree ) + { + tree->SendDeleteEvent(child); + } + + child->DeleteChildren(tree); + delete child; } - return ret; + + m_children.Empty(); } -int wxGenericTreeItem::NumberOfVisibleChildren() +void wxGenericTreeItem::SetText( const wxString &text ) { - return m_isCollapsed ? 0 : m_children.Number(); + m_text = text; } -wxGenericTreeItem *wxGenericTreeItem::FindItem( long itemId ) const +void wxGenericTreeItem::Reset() { - if (m_itemId == itemId) return (wxGenericTreeItem*)(this); - wxNode *node = m_children.First(); - while (node) + m_text.Empty(); + for ( int i = 0; i < wxTreeItemIcon_Max; i++ ) { - wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data(); - wxGenericTreeItem *res = item->FindItem( itemId ); - if (res) return (wxGenericTreeItem*)(res); - node = node->Next(); + m_images[i] = NO_IMAGE; } - return (wxGenericTreeItem *) NULL; + + m_data = NULL; + m_x = m_y = + m_height = m_width = 0; + m_xCross = + m_yCross = 0; + + m_level = 0; + + DeleteChildren(); + m_isCollapsed = TRUE; + + m_parent = (wxGenericTreeItem *)NULL; } -void wxGenericTreeItem::AddChild( wxGenericTreeItem *child ) +size_t wxGenericTreeItem::GetChildrenCount(bool recursively) const { - m_children.Append( child ); + size_t count = m_children.Count(); + if ( !recursively ) + return count; + + size_t total = count; + for ( size_t n = 0; n < count; ++n ) + { + total += m_children[n]->GetChildrenCount(); + } + + return total; } void wxGenericTreeItem::SetCross( int x, int y ) @@ -196,955 +381,1789 @@ void wxGenericTreeItem::SetCross( int x, int y ) m_yCross = y; } -void wxGenericTreeItem::GetSize( int &x, int &y ) +void wxGenericTreeItem::GetSize( int &x, int &y, const wxTreeCtrl *theTree ) { - if (y < m_y + 10) y = m_y +10; + int bottomY=m_y+theTree->GetLineHeight(this); + if ( y < bottomY ) y = bottomY; int width = m_x + m_width; - if (width > x) x = width; - wxNode *node = m_children.First(); - while (node) + if ( x < width ) x = width; + + if (IsExpanded()) { - wxGenericTreeItem *item = (wxGenericTreeItem*)node->Data(); - item->GetSize( x, y ); - node = node->Next(); + size_t count = m_children.Count(); + for ( size_t n = 0; n < count; ++n ) + { + m_children[n]->GetSize( x, y, theTree ); + } } } -long wxGenericTreeItem::HitTest( const wxPoint& point, int &flags ) +wxGenericTreeItem *wxGenericTreeItem::HitTest( const wxPoint& point, + const wxTreeCtrl *theTree, + int &flags) { - if ((point.y > m_y) && (point.y < m_y+m_height)) + if ((point.y > m_y) && (point.y < m_y + theTree->GetLineHeight(this))) { - if ((point.x > m_xCross-5) && - (point.x < m_xCross+5) && - (point.y > m_yCross-5) && - (point.y < m_yCross+5) && - (m_hasChildren)) - { - flags = wxTREE_HITTEST_ONITEMBUTTON; - return m_itemId; - } - if ((point.x > m_x) && (point.x < m_x+m_width)) + if (point.yGetLineHeight(this)/2) + flags |= wxTREE_HITTEST_ONITEMUPPERPART; + else + flags |= wxTREE_HITTEST_ONITEMLOWERPART; + + // 5 is the size of the plus sign + if ((point.x > m_xCross-5) && (point.x < m_xCross+5) && + (point.y > m_yCross-5) && (point.y < m_yCross+5) && + (IsExpanded() || HasPlus())) { - flags = wxTREE_HITTEST_ONITEMLABEL; - return m_itemId; + flags|=wxTREE_HITTEST_ONITEMBUTTON; + return this; } - if (point.x > m_x) + + if ((point.x >= m_x) && (point.x <= m_x+m_width)) { - flags = wxTREE_HITTEST_ONITEMRIGHT; - return m_itemId; + int image_w = -1; + int image_h; + + // assuming every image (normal and selected ) has the same size ! + if ( (GetImage() != NO_IMAGE) && theTree->m_imageListNormal ) + theTree->m_imageListNormal->GetSize(GetImage(), image_w, image_h); + + if ((image_w != -1) && (point.x <= m_x + image_w + 1)) + flags |= wxTREE_HITTEST_ONITEMICON; + else + flags |= wxTREE_HITTEST_ONITEMLABEL; + + return this; } - flags = wxTREE_HITTEST_ONITEMINDENT; - return m_itemId; + + if (point.x < m_x) + flags |= wxTREE_HITTEST_ONITEMINDENT; + if (point.x > m_x+m_width) + flags |= wxTREE_HITTEST_ONITEMRIGHT; + + return this; } else { if (!m_isCollapsed) { - wxNode *node = m_children.First(); - while (node) + size_t count = m_children.Count(); + for ( size_t n = 0; n < count; n++ ) { - wxGenericTreeItem *child = (wxGenericTreeItem*)node->Data(); - long res = child->HitTest( point, flags ); - if (res != -1) return res; - node = node->Next(); + wxGenericTreeItem *res = m_children[n]->HitTest( point, theTree, flags ); + if ( res != NULL ) + return res; } } } - return -1; + + flags|=wxTREE_HITTEST_NOWHERE; + return NULL; } -void wxGenericTreeItem::PrepareEvent( wxTreeEvent &event ) +int wxGenericTreeItem::GetCurrentImage() const { - event.m_item.m_itemId = m_itemId; - event.m_item.m_state = m_state; - event.m_item.m_text = m_text; - event.m_item.m_image = m_image; - event.m_item.m_selectedImage = m_selectedImage; - event.m_item.m_children = (int)m_hasChildren; - event.m_item.m_data = m_data; - event.m_oldItem = 0; - event.m_code = 0; - event.m_pointDrag.x = 0; - event.m_pointDrag.y = 0; + int image = NO_IMAGE; + if ( IsExpanded() ) + { + if ( IsSelected() ) + { + image = GetImage(wxTreeItemIcon_SelectedExpanded); + } + + if ( image == NO_IMAGE ) + { + // we usually fall back to the normal item, but try just the + // expanded one (and not selected) first in this case + image = GetImage(wxTreeItemIcon_Expanded); + } + } + else // not expanded + { + if ( IsSelected() ) + image = GetImage(wxTreeItemIcon_Selected); + } + + // may be it doesn't have the specific image we want, try the default one + // instead + if ( image == NO_IMAGE ) + { + image = GetImage(); + } + + return image; } -void wxGenericTreeItem::SendKeyDown( wxWindow *target ) +// ----------------------------------------------------------------------------- +// wxTreeCtrl implementation +// ----------------------------------------------------------------------------- + +IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxScrolledWindow) + +BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow) + EVT_PAINT (wxTreeCtrl::OnPaint) + EVT_MOUSE_EVENTS (wxTreeCtrl::OnMouse) + EVT_CHAR (wxTreeCtrl::OnChar) + EVT_SET_FOCUS (wxTreeCtrl::OnSetFocus) + EVT_KILL_FOCUS (wxTreeCtrl::OnKillFocus) + EVT_IDLE (wxTreeCtrl::OnIdle) +END_EVENT_TABLE() + +// ----------------------------------------------------------------------------- +// construction/destruction +// ----------------------------------------------------------------------------- + +void wxTreeCtrl::Init() { - wxTreeEvent event( wxEVT_COMMAND_TREE_KEY_DOWN, target->GetId() ); - PrepareEvent( event ); - event.SetEventObject( target ); - target->ProcessEvent( event ); + m_current = + m_key_current = + m_anchor = (wxGenericTreeItem *) NULL; + m_hasFocus = FALSE; + m_dirty = FALSE; + + m_xScroll = 0; + m_yScroll = 0; + m_lineHeight = 10; + m_indent = 15; + m_spacing = 18; + + m_hilightBrush = new wxBrush + ( + wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), + wxSOLID + ); + + m_imageListNormal = + m_imageListState = (wxImageList *) NULL; + + m_dragCount = 0; + + m_renameTimer = new wxTreeRenameTimer( this ); + + m_normalFont = wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ); + m_boldFont = wxFont( m_normalFont.GetPointSize(), + m_normalFont.GetFamily(), + m_normalFont.GetStyle(), + wxBOLD, + m_normalFont.GetUnderlined()); } -void wxGenericTreeItem::SendSelected( wxWindow *target ) +bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id, + const wxPoint& pos, const wxSize& size, + long style, +#if wxUSE_VALIDATORS + const wxValidator &validator, +#endif + const wxString& name ) { - wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGED, target->GetId() ); - PrepareEvent( event ); - event.SetEventObject( target ); - target->ProcessEvent( event ); + Init(); + + wxScrolledWindow::Create( parent, id, pos, size, style|wxHSCROLL|wxVSCROLL, name ); + +#if wxUSE_VALIDATORS + SetValidator( validator ); +#endif + + SetBackgroundColour( *wxWHITE ); +// m_dottedPen = wxPen( "grey", 0, wxDOT ); + m_dottedPen = wxPen( "grey", 0, 0 ); + + return TRUE; } -void wxGenericTreeItem::SendDelete( wxWindow *target ) +wxTreeCtrl::~wxTreeCtrl() { - wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, target->GetId() ); - PrepareEvent( event ); - event.SetEventObject( target ); - target->ProcessEvent( event ); + wxDELETE( m_hilightBrush ); + + DeleteAllItems(); + + delete m_renameTimer; } -void wxGenericTreeItem::SendExpand( wxWindow *target ) +// ----------------------------------------------------------------------------- +// accessors +// ----------------------------------------------------------------------------- + +size_t wxTreeCtrl::GetCount() const { - m_isCollapsed = FALSE; + return m_anchor == NULL ? 0u : m_anchor->GetChildrenCount(); +} - wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, target->GetId() ); - event.SetEventObject( target ); - PrepareEvent( event ); - target->ProcessEvent( event ); +void wxTreeCtrl::SetIndent(unsigned int indent) +{ + m_indent = indent; + m_dirty = TRUE; + Refresh(); +} - event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED); - PrepareEvent( event ); - target->ProcessEvent( event ); +void wxTreeCtrl::SetSpacing(unsigned int spacing) +{ + m_spacing = spacing; + m_dirty = TRUE; + Refresh(); } -void wxGenericTreeItem::SendCollapse( wxWindow *target ) +size_t wxTreeCtrl::GetChildrenCount(const wxTreeItemId& item, bool recursively) { - m_isCollapsed = TRUE; + wxCHECK_MSG( item.IsOk(), 0u, wxT("invalid tree item") ); - wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, target->GetId() ); - event.SetEventObject( target ); - PrepareEvent( event ); - target->ProcessEvent( event ); + return item.m_pItem->GetChildrenCount(recursively); +} - event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED); - PrepareEvent( event ); - target->ProcessEvent( event ); +// ----------------------------------------------------------------------------- +// functions to work with tree items +// ----------------------------------------------------------------------------- + +wxString wxTreeCtrl::GetItemText(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxT(""), wxT("invalid tree item") ); + + return item.m_pItem->GetText(); } -void wxGenericTreeItem::SetHilight( bool set ) +int wxTreeCtrl::GetItemImage(const wxTreeItemId& item, + wxTreeItemIcon which) const { - m_hasHilight = set; + wxCHECK_MSG( item.IsOk(), -1, wxT("invalid tree item") ); + + return item.m_pItem->GetImage(which); } -bool wxGenericTreeItem::HasHilight() +wxTreeItemData *wxTreeCtrl::GetItemData(const wxTreeItemId& item) const { - return m_hasHilight; + wxCHECK_MSG( item.IsOk(), NULL, wxT("invalid tree item") ); + + return item.m_pItem->GetData(); } -//----------------------------------------------------------------------------- -// wxTreeCtrl -//----------------------------------------------------------------------------- +void wxTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text) +{ + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); -IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl,wxScrolledWindow) + wxClientDC dc(this); + wxGenericTreeItem *pItem = item.m_pItem; + pItem->SetText(text); + CalculateSize(pItem, dc); + RefreshLine(pItem); +} -BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow) - EVT_PAINT (wxTreeCtrl::OnPaint) - EVT_MOUSE_EVENTS (wxTreeCtrl::OnMouse) - EVT_CHAR (wxTreeCtrl::OnChar) - EVT_SET_FOCUS (wxTreeCtrl::OnSetFocus) - EVT_KILL_FOCUS (wxTreeCtrl::OnKillFocus) -END_EVENT_TABLE() +void wxTreeCtrl::SetItemImage(const wxTreeItemId& item, + int image, + wxTreeItemIcon which) +{ + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + wxGenericTreeItem *pItem = item.m_pItem; + pItem->SetImage(image, which); + + wxClientDC dc(this); + CalculateSize(pItem, dc); + RefreshLine(pItem); +} -wxTreeCtrl::wxTreeCtrl() +void wxTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data) { - m_current = (wxGenericTreeItem *) NULL; - m_anchor = (wxGenericTreeItem *) NULL; - m_hasFocus = FALSE; - m_xScroll = 0; - m_yScroll = 0; - m_lastId = 0; - m_lineHeight = 10; - m_indent = 15; - m_isCreated = FALSE; - m_hilightBrush = new wxBrush( wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), wxSOLID ); - m_imageList = (wxImageList *) NULL; - m_smallImageList = (wxImageList *) NULL; + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + item.m_pItem->SetData(data); } -wxTreeCtrl::wxTreeCtrl(wxWindow *parent, wxWindowID id, - const wxPoint& pos, const wxSize& size, - long style, const wxString& name ) +void wxTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has) { - m_current = (wxGenericTreeItem *) NULL; - m_anchor = (wxGenericTreeItem *) NULL; - m_hasFocus = FALSE; - m_xScroll = 0; - m_yScroll = 0; - m_lastId = 0; - m_lineHeight = 10; - m_indent = 15; - m_isCreated = FALSE; - m_hilightBrush = new wxBrush( wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT), wxSOLID ); - m_imageList = (wxImageList *) NULL; - m_smallImageList = (wxImageList *) NULL; - Create( parent, id, pos, size, style, name ); + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + wxGenericTreeItem *pItem = item.m_pItem; + pItem->SetHasPlus(has); + RefreshLine(pItem); } -wxTreeCtrl::~wxTreeCtrl() +void wxTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold) { - if (m_hilightBrush) delete m_hilightBrush; - if (m_anchor) delete m_anchor; + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + // avoid redrawing the tree if no real change + wxGenericTreeItem *pItem = item.m_pItem; + if ( pItem->IsBold() != bold ) + { + pItem->SetBold(bold); + RefreshLine(pItem); + } } -bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id, - const wxPoint& pos, const wxSize& size, - long style, const wxString& name ) +void wxTreeCtrl::SetItemTextColour(const wxTreeItemId& item, + const wxColour& col) { - wxScrolledWindow::Create( parent, id, pos, size, style, name ); - SetBackgroundColour( *wxWHITE ); - m_dottedPen = wxPen( *wxBLACK, 0, 0 ); + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + wxGenericTreeItem *pItem = item.m_pItem; + pItem->Attr().SetTextColour(col); + RefreshLine(pItem); +} + +void wxTreeCtrl::SetItemBackgroundColour(const wxTreeItemId& item, + const wxColour& col) +{ + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + wxGenericTreeItem *pItem = item.m_pItem; + pItem->Attr().SetBackgroundColour(col); + RefreshLine(pItem); +} + +void wxTreeCtrl::SetItemFont(const wxTreeItemId& item, const wxFont& font) +{ + wxCHECK_RET( item.IsOk(), wxT("invalid tree item") ); + + wxGenericTreeItem *pItem = item.m_pItem; + pItem->Attr().SetFont(font); + RefreshLine(pItem); +} + +// ----------------------------------------------------------------------------- +// item status inquiries +// ----------------------------------------------------------------------------- + +bool wxTreeCtrl::IsVisible(const wxTreeItemId& WXUNUSED(item)) const +{ + wxFAIL_MSG(wxT("not implemented")); + return TRUE; } -int wxTreeCtrl::GetCount() const +bool wxTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") ); + + return !item.m_pItem->GetChildren().IsEmpty(); +} + +bool wxTreeCtrl::IsExpanded(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") ); + + return item.m_pItem->IsExpanded(); +} + +bool wxTreeCtrl::IsSelected(const wxTreeItemId& item) const { - if (!m_anchor) return 0; - return m_anchor->NumberOfVisibleDescendents(); + wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") ); + + return item.m_pItem->IsSelected(); } -long wxTreeCtrl::InsertItem( long parent, const wxString& label, int image, - int selImage, long WXUNUSED(insertAfter) ) +bool wxTreeCtrl::IsBold(const wxTreeItemId& item) const { - wxGenericTreeItem *p = (wxGenericTreeItem *) NULL; - if (parent == 0) + wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") ); + + return item.m_pItem->IsBold(); +} + +// ----------------------------------------------------------------------------- +// navigation +// ----------------------------------------------------------------------------- + +wxTreeItemId wxTreeCtrl::GetParent(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + return item.m_pItem->GetParent(); +} + +wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item, long& cookie) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + cookie = 0; + return GetNextChild(item, cookie); +} + +wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& item, long& cookie) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + wxArrayGenericTreeItems& children = item.m_pItem->GetChildren(); + if ( (size_t)cookie < children.Count() ) { - if (m_anchor) return -1; + return children.Item(cookie++); } else { - p = FindItem( parent ); - if (!p) return -1; + // there are no more of them + return wxTreeItemId(); } - wxTreeItem item; - m_lastId++; - item.m_mask = wxTREE_MASK_HANDLE; - item.m_itemId = m_lastId; - if (!label.IsNull() || (label == "")) - { - item.m_text = label; - item.m_mask |= wxTREE_MASK_TEXT; - } - if (image >= 0) +} + +wxTreeItemId wxTreeCtrl::GetLastChild(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + wxArrayGenericTreeItems& children = item.m_pItem->GetChildren(); + return (children.IsEmpty() ? wxTreeItemId() : wxTreeItemId(children.Last())); +} + +wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + wxGenericTreeItem *i = item.m_pItem; + wxGenericTreeItem *parent = i->GetParent(); + if ( parent == NULL ) { - item.m_image = image; - item.m_mask |= wxTREE_MASK_IMAGE; + // root item doesn't have any siblings + return wxTreeItemId(); } - if (selImage >= 0) + + wxArrayGenericTreeItems& siblings = parent->GetChildren(); + int index = siblings.Index(i); + wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent? + + size_t n = (size_t)(index + 1); + return n == siblings.Count() ? wxTreeItemId() : wxTreeItemId(siblings[n]); +} + +wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + wxGenericTreeItem *i = item.m_pItem; + wxGenericTreeItem *parent = i->GetParent(); + if ( parent == NULL ) { - item.m_selectedImage = selImage; - item.m_mask |= wxTREE_MASK_SELECTED_IMAGE; + // root item doesn't have any siblings + return wxTreeItemId(); } - wxClientDC dc(this); - wxGenericTreeItem *new_child = new wxGenericTreeItem( p, item, &dc ); - if (p) - p->AddChild( new_child ); - else - m_anchor = new_child; + wxArrayGenericTreeItems& siblings = parent->GetChildren(); + int index = siblings.Index(i); + wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent? - if (p) - { - CalculatePositions(); + return index == 0 ? wxTreeItemId() + : wxTreeItemId(siblings[(size_t)(index - 1)]); +} - if (!p->HasChildren()) p->m_hasChildren = TRUE; +wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const +{ + wxFAIL_MSG(wxT("not implemented")); - int ch = 0; - GetClientSize( (int *) NULL, &ch ); + return wxTreeItemId(); +} - PrepareDC( dc ); - - wxRectangle rect; - rect.x = dc.LogicalToDeviceX( 0 ); - rect.y = 0; - rect.width = 10000; - rect.height = ch; +wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + wxFAIL_MSG(wxT("not implemented")); - if (p->m_children.Number() == 1) + return wxTreeItemId(); +} + +wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const +{ + wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") ); + + wxFAIL_MSG(wxT("not implemented")); + + return wxTreeItemId(); +} + +// ----------------------------------------------------------------------------- +// operations +// ----------------------------------------------------------------------------- + +wxTreeItemId wxTreeCtrl::DoInsertItem(const wxTreeItemId& parentId, + size_t previous, + const wxString& text, + int image, int selImage, + wxTreeItemData *data) +{ + wxGenericTreeItem *parent = parentId.m_pItem; + if ( !parent ) { - rect.y = dc.LogicalToDeviceY( p->m_y ); + // should we give a warning here? + return AddRoot(text, image, selImage, data); } - else + + wxClientDC dc(this); + wxGenericTreeItem *item = + new wxGenericTreeItem( parent, text, dc, image, selImage, data ); + + if ( data != NULL ) + { + data->m_pItem = item; + } + + parent->Insert( item, previous ); + + m_dirty = TRUE; + + return item; +} + +wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text, + int image, int selImage, + wxTreeItemData *data) +{ + wxCHECK_MSG( !m_anchor, wxTreeItemId(), wxT("tree can have only one root") ); + + wxClientDC dc(this); + m_anchor = new wxGenericTreeItem((wxGenericTreeItem *)NULL, text, dc, + image, selImage, data); + if ( data != NULL ) + { + data->m_pItem = m_anchor; + } + + if (!HasFlag(wxTR_MULTIPLE)) { - wxNode *node = p->m_children.Member( new_child )->Previous(); - wxGenericTreeItem* last_child = (wxGenericTreeItem*)node->Data(); - rect.y = dc.LogicalToDeviceY( last_child->m_y ); + m_current = m_key_current = m_anchor; + m_current->SetHilight( TRUE ); } + Refresh(); AdjustMyScrollbars(); - if (rect.height > 0) Refresh( FALSE, &rect); + return m_anchor; +} + +wxTreeItemId wxTreeCtrl::PrependItem(const wxTreeItemId& parent, + const wxString& text, + int image, int selImage, + wxTreeItemData *data) +{ + return DoInsertItem(parent, 0u, text, image, selImage, data); +} + +wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parentId, + const wxTreeItemId& idPrevious, + const wxString& text, + int image, int selImage, + wxTreeItemData *data) +{ + wxGenericTreeItem *parent = parentId.m_pItem; + if ( !parent ) + { + // should we give a warning here? + return AddRoot(text, image, selImage, data); } - else + + int index = parent->GetChildren().Index(idPrevious.m_pItem); + wxASSERT_MSG( index != wxNOT_FOUND, + wxT("previous item in wxTreeCtrl::InsertItem() is not a sibling") ); + return DoInsertItem(parentId, (size_t)++index, text, image, selImage, data); +} + +wxTreeItemId wxTreeCtrl::AppendItem(const wxTreeItemId& parentId, + const wxString& text, + int image, int selImage, + wxTreeItemData *data) +{ + wxGenericTreeItem *parent = parentId.m_pItem; + if ( !parent ) { - AdjustMyScrollbars(); + // should we give a warning here? + return AddRoot(text, image, selImage, data); + } - Refresh(); + return DoInsertItem(parent, parent->GetChildren().Count(), text, + image, selImage, data); +} + +void wxTreeCtrl::SendDeleteEvent(wxGenericTreeItem *item) +{ + wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, GetId() ); + event.m_item = item; + event.SetEventObject( this ); + ProcessEvent( event ); +} + +void wxTreeCtrl::DeleteChildren(const wxTreeItemId& itemId) +{ + wxGenericTreeItem *item = itemId.m_pItem; + item->DeleteChildren(this); + + m_dirty = TRUE; +} + +void wxTreeCtrl::Delete(const wxTreeItemId& itemId) +{ + wxGenericTreeItem *item = itemId.m_pItem; + wxGenericTreeItem *parent = item->GetParent(); + + if ( parent ) + { + parent->GetChildren().Remove( item ); // remove by value + } + + item->DeleteChildren(this); + SendDeleteEvent(item); + delete item; + + m_dirty = TRUE; +} + +void wxTreeCtrl::DeleteAllItems() +{ + if ( m_anchor ) + { + m_anchor->DeleteChildren(this); + delete m_anchor; + + m_anchor = NULL; + + m_dirty = TRUE; + } +} + +void wxTreeCtrl::Expand(const wxTreeItemId& itemId) +{ + wxGenericTreeItem *item = itemId.m_pItem; + + if ( !item->HasPlus() ) + return; + + if ( item->IsExpanded() ) + return; + + wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, GetId() ); + event.m_item = item; + event.SetEventObject( this ); + +// if ( ProcessEvent( event ) && event.m_code ) TODO: Was this a typo ? + if ( ProcessEvent( event ) && !event.IsAllowed() ) + { + // cancelled by program + return; } - return m_lastId; + item->Expand(); + CalculatePositions(); + + RefreshSubtree(item); + + event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED); + ProcessEvent( event ); } -long wxTreeCtrl::InsertItem( long parent, wxTreeItem &info, long WXUNUSED(insertAfter) ) +void wxTreeCtrl::Collapse(const wxTreeItemId& itemId) { - int oldMask = info.m_mask; - wxGenericTreeItem *p = (wxGenericTreeItem *) NULL; - if (parent == 0) + wxGenericTreeItem *item = itemId.m_pItem; + + if ( !item->IsExpanded() ) + return; + + wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, GetId() ); + event.m_item = item; + event.SetEventObject( this ); + if ( ProcessEvent( event ) && !event.IsAllowed() ) { - if (m_anchor) return -1; + // cancelled by program + return; } + + item->Collapse(); + + wxArrayGenericTreeItems& children = item->GetChildren(); + size_t count = children.Count(); + for ( size_t n = 0; n < count; n++ ) + { + Collapse(children[n]); + } + + CalculatePositions(); + + RefreshSubtree(item); + + event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED); + ProcessEvent( event ); +} + +void wxTreeCtrl::CollapseAndReset(const wxTreeItemId& item) +{ + Collapse(item); + DeleteChildren(item); +} + +void wxTreeCtrl::Toggle(const wxTreeItemId& itemId) +{ + wxGenericTreeItem *item = itemId.m_pItem; + + if ( item->IsExpanded() ) + Collapse(itemId); else + Expand(itemId); +} + +void wxTreeCtrl::Unselect() +{ + if ( m_current ) { - p = FindItem( parent ); - if (!p) + m_current->SetHilight( FALSE ); + RefreshLine( m_current ); + } +} + +void wxTreeCtrl::UnselectAllChildren(wxGenericTreeItem *item) +{ + item->SetHilight(FALSE); + RefreshLine(item); + + if (item->HasChildren()) { - printf( "TreeItem not found.\n" ); - return -1; + wxArrayGenericTreeItems& children = item->GetChildren(); + size_t count = children.Count(); + for ( size_t n = 0; n < count; ++n ) + UnselectAllChildren(children[n]); } - } - long ret = 0; - if ((info.m_mask & wxTREE_MASK_HANDLE) == 0) +} + +void wxTreeCtrl::UnselectAll() +{ + UnselectAllChildren(GetRootItem().m_pItem); +} + +// Recursive function ! +// To stop we must have crt_itemGetParent(); + + if ( parent == NULL ) // This is root item + return TagAllChildrenUntilLast(crt_item, last_item, select); + + wxArrayGenericTreeItems& children = parent->GetChildren(); + int index = children.Index(crt_item); + wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent? + + size_t count = children.Count(); + for (size_t n=(size_t)(index+1); nSetHilight(select); + RefreshLine(crt_item); + + if (crt_item==last_item) return TRUE; + + if (crt_item->HasChildren()) + { + wxArrayGenericTreeItems& children = crt_item->GetChildren(); + size_t count = children.Count(); + for ( size_t n = 0; n < count; ++n ) + if (TagAllChildrenUntilLast(children[n], last_item, select)) return TRUE; + } + + return FALSE; +} + +void wxTreeCtrl::SelectItemRange(wxGenericTreeItem *item1, wxGenericTreeItem *item2) +{ + // item2 is not necessary after item1 + wxGenericTreeItem *first=NULL, *last=NULL; + + // choice first' and 'last' between item1 and item2 + if (item1->GetY()GetY()) { - m_lastId++; - info.m_itemId = m_lastId; - info.m_mask |= wxTREE_MASK_HANDLE; - ret = m_lastId; + first=item1; + last=item2; } else { - ret = info.m_itemId; + first=item2; + last=item1; } - wxClientDC dc(this); - wxGenericTreeItem *new_child = new wxGenericTreeItem( p, info, &dc ); - if (p) - p->AddChild( new_child ); - else - m_anchor = new_child; + bool select = m_current->IsSelected(); - if (p) - { - CalculatePositions(); + if ( TagAllChildrenUntilLast(first,last,select) ) + return; + + TagNextChildren(first,last,select); +} - if (!p->HasChildren()) p->m_hasChildren = TRUE; +void wxTreeCtrl::SelectItem(const wxTreeItemId& itemId, + bool unselect_others, + bool extended_select) +{ + wxCHECK_RET( itemId.IsOk(), wxT("invalid tree item") ); - int ch = 0; - GetClientSize( (int *) NULL, &ch ); + bool is_single=!(GetWindowStyleFlag() & wxTR_MULTIPLE); + wxGenericTreeItem *item = itemId.m_pItem; - PrepareDC( dc ); - - wxRectangle rect; - rect.x = dc.LogicalToDeviceX( 0 ); - rect.y = 0; - rect.width = 10000; - rect.height = ch; + //wxCHECK_RET( ( (!unselect_others) && is_single), + // wxT("this is a single selection tree") ); - if (p->m_children.Number() == 1) + // to keep going anyhow !!! + if (is_single) { - rect.y = dc.LogicalToDeviceY( p->m_y ); + if (item->IsSelected()) + return; // nothing to do + unselect_others = TRUE; + extended_select = FALSE; } - else + else if ( unselect_others && item->IsSelected() ) { - wxNode *node = p->m_children.Member( new_child )->Previous(); - wxGenericTreeItem* last_child = (wxGenericTreeItem*)node->Data(); - rect.y = dc.LogicalToDeviceY( last_child->m_y ); + // selection change if there is more than one item currently selected + wxArrayTreeItemIds selected_items; + if ( GetSelections(selected_items) == 1 ) + return; } - AdjustMyScrollbars(); + wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGING, GetId() ); + event.m_item = item; + event.m_itemOld = m_current; + event.SetEventObject( this ); + // TODO : Here we don't send any selection mode yet ! - if (rect.height > 0) Refresh( FALSE, &rect); - } - else - { - AdjustMyScrollbars(); + if ( GetEventHandler()->ProcessEvent( event ) && !event.IsAllowed() ) + return; + + // ctrl press + if (unselect_others) + { + if (is_single) Unselect(); // to speed up thing + else UnselectAll(); + } + + // shift press + if (extended_select) + { + if (m_current == NULL) m_current=m_key_current=GetRootItem().m_pItem; + // don't change the mark (m_current) + SelectItemRange(m_current, item); + } + else + { + bool select=TRUE; // the default + + // Check if we need to toggle hilight (ctrl mode) + if (!unselect_others) + select=!item->IsSelected(); - Refresh(); - } + m_current = m_key_current = item; + m_current->SetHilight(select); + RefreshLine( m_current ); + } - info.m_mask = oldMask; - return ret; + event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGED); + GetEventHandler()->ProcessEvent( event ); } -bool wxTreeCtrl::ExpandItem( long item, int action ) +void wxTreeCtrl::FillArray(wxGenericTreeItem *item, + wxArrayTreeItemIds &array) const { - wxGenericTreeItem *i = FindItem( item ); - if (!i) - return FALSE; + if ( item->IsSelected() ) + array.Add(wxTreeItemId(item)); - switch (action) - { - case wxTREE_EXPAND_EXPAND: + if ( item->HasChildren() ) { - i->SendExpand( this ); - break; + wxArrayGenericTreeItems& children = item->GetChildren(); + size_t count = children.GetCount(); + for ( size_t n = 0; n < count; ++n ) + FillArray(children[n],array); } +} - case wxTREE_EXPAND_COLLAPSE_RESET: - case wxTREE_EXPAND_COLLAPSE: - { - wxNode *node = i->m_children.First(); - while (node) - { - wxGenericTreeItem *child = (wxGenericTreeItem*)node->Data(); - if ( child->IsExpanded() ) - ExpandItem( child->m_itemId, wxTREE_EXPAND_COLLAPSE ); - node = node->Next(); - } +size_t wxTreeCtrl::GetSelections(wxArrayTreeItemIds &array) const +{ + array.Empty(); + FillArray(GetRootItem().m_pItem, array); - CalculatePositions(); + return array.Count(); +} - i->SendCollapse( this ); - break; - } +void wxTreeCtrl::EnsureVisible(const wxTreeItemId& item) +{ + if (!item.IsOk()) return; - case wxTREE_EXPAND_TOGGLE: + wxGenericTreeItem *gitem = item.m_pItem; + + // first expand all parent branches + wxGenericTreeItem *parent = gitem->GetParent(); + while ( parent ) { - if ( i->IsExpanded() ) - ExpandItem( item, wxTREE_EXPAND_COLLAPSE ); - else - ExpandItem( item, wxTREE_EXPAND_EXPAND ); - return TRUE; + Expand(parent); + parent = parent->GetParent(); } - } - wxClientDC dc(this); - PrepareDC(dc); - - int cw = 0; - int ch = 0; - GetClientSize( &cw, &ch ); - - wxRect rect; - rect.x = dc.LogicalToDeviceX( 0 ); - rect.width = cw; - rect.y = dc.LogicalToDeviceY( i->m_y ); - rect.height = ch; - Refresh( TRUE, &rect ); - - AdjustMyScrollbars(); + //if (parent) CalculatePositions(); - return TRUE; + ScrollTo(item); } -void wxTreeCtrl::DeleteItem( long item ) +void wxTreeCtrl::ScrollTo(const wxTreeItemId &item) { - wxGenericTreeItem *pItem = FindItem( item ); - wxCHECK_RET( pItem != NULL, _("wxTreeCtrl::DeleteItem: no such pItem.") ); + if (!item.IsOk()) return; - pItem->m_parent->m_children.DeleteObject(pItem); + // We have to call this here because the label in + // question might just have been added and no screen + // update taken place. + if (m_dirty) wxYield(); - Refresh(); -} + wxGenericTreeItem *gitem = item.m_pItem; -void wxTreeCtrl::DeleteChildren( long item ) -{ - wxGenericTreeItem *pItem = FindItem( item ); - wxCHECK_RET( pItem != NULL, _("wxTreeCtrl::DeleteChildren: no such pItem.") ); + // now scroll to the item + int item_y = gitem->GetY(); - pItem->m_children.Clear(); + int start_x = 0; + int start_y = 0; + ViewStart( &start_x, &start_y ); + start_y *= PIXELS_PER_UNIT; - Refresh(); -} + int client_h = 0; + int client_w = 0; + GetClientSize( &client_w, &client_h ); -bool wxTreeCtrl::DeleteAllItems() -{ - delete m_anchor; - m_anchor = (wxGenericTreeItem *) NULL; - Refresh(); - return TRUE; + if (item_y < start_y+3) + { + // going down + int x = 0; + int y = 0; + m_anchor->GetSize( x, y, this ); + y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels + int x_pos = GetScrollPos( wxHORIZONTAL ); + // Item should appear at top + SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, item_y/PIXELS_PER_UNIT ); + } + else if (item_y+GetLineHeight(gitem) > start_y+client_h) + { + // going up + int x = 0; + int y = 0; + m_anchor->GetSize( x, y, this ); + y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels + item_y += PIXELS_PER_UNIT+2; + int x_pos = GetScrollPos( wxHORIZONTAL ); + // Item should appear at bottom + SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, (item_y+GetLineHeight(gitem)-client_h)/PIXELS_PER_UNIT ); + } } -bool wxTreeCtrl::GetItem( wxTreeItem &info ) const +// FIXME: tree sorting functions are not reentrant and not MT-safe! +static wxTreeCtrl *s_treeBeingSorted = NULL; + +static int LINKAGEMODE tree_ctrl_compare_func(wxGenericTreeItem **item1, + wxGenericTreeItem **item2) { - wxGenericTreeItem *i = FindItem( info.m_itemId ); - if (!i) return FALSE; - i->GetItem( info ); - return TRUE; + wxCHECK_MSG( s_treeBeingSorted, 0, wxT("bug in wxTreeCtrl::SortChildren()") ); + + return s_treeBeingSorted->OnCompareItems(*item1, *item2); } -long wxTreeCtrl::GetItemData( long item ) const +int wxTreeCtrl::OnCompareItems(const wxTreeItemId& item1, + const wxTreeItemId& item2) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return 0; - return i->m_data; + return wxStrcmp(GetItemText(item1), GetItemText(item2)); } -wxString wxTreeCtrl::GetItemText( long item ) const +void wxTreeCtrl::SortChildren(const wxTreeItemId& itemId) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return ""; - return i->m_text; + wxCHECK_RET( itemId.IsOk(), wxT("invalid tree item") ); + + wxGenericTreeItem *item = itemId.m_pItem; + + wxCHECK_RET( !s_treeBeingSorted, + wxT("wxTreeCtrl::SortChildren is not reentrant") ); + + wxArrayGenericTreeItems& children = item->GetChildren(); + if ( children.Count() > 1 ) + { + s_treeBeingSorted = this; + children.Sort(tree_ctrl_compare_func); + s_treeBeingSorted = NULL; + + m_dirty = TRUE; + } + //else: don't make the tree dirty as nothing changed } -int wxTreeCtrl::GetItemImage(long item) const +wxImageList *wxTreeCtrl::GetImageList() const { - wxGenericTreeItem *i = FindItem( item ); - return i == 0 ? -1 : i->GetImage(); + return m_imageListNormal; } -long wxTreeCtrl::GetParent( long item ) const +wxImageList *wxTreeCtrl::GetStateImageList() const { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return -1; - i = i->m_parent; - if (!i) return -1; - return i->m_parent->m_itemId; + return m_imageListState; } -long wxTreeCtrl::GetRootItem() const +void wxTreeCtrl::SetImageList(wxImageList *imageList) { - if (m_anchor) return m_anchor->m_itemId; - return -1; + m_imageListNormal = imageList; + + // Calculate a m_lineHeight value from the image sizes. + // May be toggle off. Then wxTreeCtrl will spread when + // necessary (which might look ugly). +#if 1 + wxClientDC dc(this); + m_lineHeight = (int)(dc.GetCharHeight() + 4); + int + width = 0, + height = 0, + n = m_imageListNormal->GetImageCount(); + for(int i = 0; i < n ; i++) + { + m_imageListNormal->GetSize(i, width, height); + if(height > m_lineHeight) m_lineHeight = height; + } + + if (m_lineHeight<40) m_lineHeight+=4; // at least 4 pixels (odd such that a line can be drawn in between) + else m_lineHeight+=m_lineHeight/10; // otherwise 10% extra spacing + +#endif } -long wxTreeCtrl::GetChild( long item ) const +void wxTreeCtrl::SetStateImageList(wxImageList *imageList) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return -1; - if (!i->HasChildren()) return -1; - wxNode *node = i->m_children.First(); - i = (wxGenericTreeItem *)node->Data(); - return i->m_itemId; + m_imageListState = imageList; } -long wxTreeCtrl::GetNextItem( long item, int code ) const +// ----------------------------------------------------------------------------- +// helpers +// ----------------------------------------------------------------------------- + +void wxTreeCtrl::AdjustMyScrollbars() { - switch (code) - { - case wxTREE_NEXT_CARET: return GetSelection(); - case wxTREE_NEXT_CHILD: return GetChild( item ); - case wxTREE_NEXT_ROOT: return GetRootItem(); - case wxTREE_NEXT_PARENT: return GetParent( item ); - case wxTREE_NEXT_NEXT: + if (m_anchor) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return -1; - if (i->m_parent) - { - wxNode *node = i->m_parent->m_children.Member( i ); - if (!node) return -1; - node = node->Next(); - if (!node) return -1; - i = (wxGenericTreeItem *)node->Data(); - return i->m_itemId; - } - break; + int x = 0; + int y = 0; + m_anchor->GetSize( x, y, this ); + //y += GetLineHeight(m_anchor); + y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels + int x_pos = GetScrollPos( wxHORIZONTAL ); + int y_pos = GetScrollPos( wxVERTICAL ); + SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, y_pos ); } - case wxTREE_NEXT_PREVIOUS: + else { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return -1; - if (i->m_parent) - { - wxNode *node = i->m_parent->m_children.Member( i ); - if (!node) return -1; - node = node->Previous(); - if (!node) return -1; - i = (wxGenericTreeItem *)node->Data(); - return i->m_itemId; - } - break; + SetScrollbars( 0, 0, 0, 0 ); } - } - return -1; } -long wxTreeCtrl::GetSelection() const +int wxTreeCtrl::GetLineHeight(wxGenericTreeItem *item) const { - return m_current ? m_current->GetItemId() : -1; + if (GetWindowStyleFlag() & wxTR_HAS_VARIABLE_ROW_HEIGHT) + return item->GetHeight(); + else + return m_lineHeight; } -bool wxTreeCtrl::SelectItem( long itemId ) +void wxTreeCtrl::PaintItem(wxGenericTreeItem *item, wxDC& dc) { - wxGenericTreeItem *pItem = FindItem(itemId); - if ( !pItem ) { - wxLogDebug(_("Can't select an item %d which doesn't exist."), itemId); + wxTreeItemAttr *attr = item->GetAttributes(); + if ( attr && attr->HasFont() ) + dc.SetFont(attr->GetFont()); + else if (item->IsBold()) + dc.SetFont(m_boldFont); - return FALSE; - } + long text_w = 0; + long text_h = 0; + dc.GetTextExtent( item->GetText(), &text_w, &text_h ); + + int image_h = 0; + int image_w = 0; + int image = item->GetCurrentImage(); + if ( image != NO_IMAGE ) + { + m_imageListNormal->GetSize( image, image_w, image_h ); + image_w += 4; + } - SelectItem(pItem); + int total_h = GetLineHeight(item); - return TRUE; -} + dc.DrawRectangle( item->GetX()-2, item->GetY(), item->GetWidth()+2, total_h ); -void wxTreeCtrl::SelectItem(wxGenericTreeItem *item) -{ - if (m_current != item) - { - if (m_current) + if ( image != NO_IMAGE ) { - m_current->SetHilight( FALSE ); - RefreshLine( m_current ); + dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h ); + m_imageListNormal->Draw( image, dc, + item->GetX(), + item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0), + wxIMAGELIST_DRAW_TRANSPARENT ); + dc.DestroyClippingRegion(); } - m_current = item; - m_current->SetHilight( TRUE ); - RefreshLine( m_current ); - m_current->SendSelected( this ); - } + bool hasBgCol = attr && attr->HasBackgroundColour(); + dc.SetBackgroundMode(hasBgCol ? wxSOLID : wxTRANSPARENT); + if ( hasBgCol ) + dc.SetTextBackground(attr->GetBackgroundColour()); + dc.DrawText( item->GetText(), image_w + item->GetX(), item->GetY() + + ((total_h > text_h) ? (total_h - text_h)/2 : 0)); + + // restore normal font + dc.SetFont( m_normalFont ); } -bool wxTreeCtrl::ItemHasChildren( long item ) const +// Now y stands for the top of the item, whereas it used to stand for middle ! +void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y ) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return FALSE; - return i->m_hasChildren; + int horizX = level*m_indent; + + item->SetX( horizX+m_indent+m_spacing ); + item->SetY( y ); + + int oldY = y; + y+=GetLineHeight(item)/2; + + item->SetCross( horizX+m_indent, y ); + + int exposed_x = dc.LogicalToDeviceX( 0 ); + int exposed_y = dc.LogicalToDeviceY( item->GetY() ); + + if (IsExposed( exposed_x, exposed_y, 10000, GetLineHeight(item) )) // 10000 = very much + { + int startX = horizX; + int endX = horizX + (m_indent-5); + +// if (!item->HasChildren()) endX += (m_indent+5); + if (!item->HasChildren()) endX += 20; + + dc.DrawLine( startX, y, endX, y ); + + if (item->HasPlus()) + { + dc.DrawLine( horizX+(m_indent+5), y, horizX+(m_indent+15), y ); + dc.SetPen( *wxGREY_PEN ); + dc.SetBrush( *wxWHITE_BRUSH ); + dc.DrawRectangle( horizX+(m_indent-5), y-4, 11, 9 ); + + dc.SetPen( *wxBLACK_PEN ); + dc.DrawLine( horizX+(m_indent-2), y, horizX+(m_indent+3), y ); + if (!item->IsExpanded()) + dc.DrawLine( horizX+m_indent, y-2, horizX+m_indent, y+3 ); + + dc.SetPen( m_dottedPen ); + } + + wxPen *pen = wxTRANSPARENT_PEN; + wxBrush *brush; // FIXME is this really needed? + wxColour colText; + + if ( item->IsSelected() ) + { + colText = wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ); + + brush = m_hilightBrush; + + if ( m_hasFocus ) + pen = wxBLACK_PEN; + + } + else + { + wxTreeItemAttr *attr = item->GetAttributes(); + if ( attr && attr->HasTextColour() ) + colText = attr->GetTextColour(); + else + colText = *wxBLACK; + + brush = wxWHITE_BRUSH; + } + + // prepare to draw + dc.SetTextForeground(colText); + dc.SetPen(*pen); + dc.SetBrush(*brush); + + // draw + PaintItem(item, dc); + + // restore DC objects + dc.SetBrush( *wxWHITE_BRUSH ); + dc.SetPen( m_dottedPen ); + dc.SetTextForeground( *wxBLACK ); + } + + y = oldY+GetLineHeight(item); + + if (item->IsExpanded()) + { + oldY+=GetLineHeight(item)/2; + int semiOldY=0; + + wxArrayGenericTreeItems& children = item->GetChildren(); + size_t n, count = children.Count(); + for ( n = 0; n < count; ++n ) + { + semiOldY=y; + PaintLevel( children[n], dc, level+1, y ); + } + + // it may happen that the item is expanded but has no items (when you + // delete all its children for example) - don't draw the vertical line + // in this case + if (count > 0) + { + semiOldY+=GetLineHeight(children[--n])/2; + dc.DrawLine( horizX+m_indent, oldY+5, horizX+m_indent, semiOldY ); + } + } } -void wxTreeCtrl::SetIndent( int indent ) +void wxTreeCtrl::DrawBorder(wxTreeItemId &item) { - m_indent = indent; - Refresh(); + if (!item) return; + + wxGenericTreeItem *i=item.m_pItem; + + wxClientDC dc(this); + PrepareDC( dc ); + dc.SetLogicalFunction(wxINVERT); + + int w,h,x; + ViewStart(&x,&h); // we only need x + GetClientSize(&w,&h); // we only need w + + h=GetLineHeight(i)+1; + // 2 white column at border + dc.DrawRectangle( PIXELS_PER_UNIT*x+2, i->GetY()-1, w-6, h); } -int wxTreeCtrl::GetIndent() const +void wxTreeCtrl::DrawLine(wxTreeItemId &item, bool below) { - return m_indent; + if (!item) return; + + wxGenericTreeItem *i=item.m_pItem; + + wxClientDC dc(this); + PrepareDC( dc ); + dc.SetLogicalFunction(wxINVERT); + + int w,h,y; + GetSize(&w,&h); + + if (below) y=i->GetY()+GetLineHeight(i)-1; + else y=i->GetY(); + + dc.DrawLine( 0, y, w, y); } -bool wxTreeCtrl::SetItem( wxTreeItem &info ) +// ----------------------------------------------------------------------------- +// wxWindows callbacks +// ----------------------------------------------------------------------------- + +void wxTreeCtrl::OnPaint( wxPaintEvent &WXUNUSED(event) ) { - wxGenericTreeItem *i = FindItem( info.m_itemId ); - if (!i) return FALSE; - wxClientDC dc(this); - i->SetItem( info, &dc ); - Refresh(); - return TRUE; + if ( !m_anchor) + return; + + wxPaintDC dc(this); + PrepareDC( dc ); + + dc.SetFont( m_normalFont ); + dc.SetPen( m_dottedPen ); + + // this is now done dynamically + //if(GetImageList() == NULL) + // m_lineHeight = (int)(dc.GetCharHeight() + 4); + + int y = 2; + PaintLevel( m_anchor, dc, 0, y ); } -bool wxTreeCtrl::SetItemData( long item, long data ) +void wxTreeCtrl::OnSetFocus( wxFocusEvent &WXUNUSED(event) ) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return FALSE; - i->m_data = data; - return TRUE; + m_hasFocus = TRUE; + + if (m_current) RefreshLine( m_current ); } -bool wxTreeCtrl::SetItemText( long item, const wxString &text ) +void wxTreeCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) ) { - wxGenericTreeItem *i = FindItem( item ); - if (!i) return FALSE; - wxClientDC dc(this); - i->SetText( text, &dc ); - return TRUE; + m_hasFocus = FALSE; + + if (m_current) RefreshLine( m_current ); } -void wxTreeCtrl::SetItemImage(long item, int image, int imageSel) const +void wxTreeCtrl::OnChar( wxKeyEvent &event ) { - wxGenericTreeItem *i = FindItem( item ); - if ( i != 0 ) - { - i->SetImage(image); - i->SetSelectedImage(imageSel); - } -} + wxTreeEvent te( wxEVT_COMMAND_TREE_KEY_DOWN, GetId() ); + te.m_code = event.KeyCode(); + te.SetEventObject( this ); + GetEventHandler()->ProcessEvent( te ); -int wxTreeCtrl::GetItemState( long item, long stateMask ) const -{ - wxGenericTreeItem *i = FindItem( item ); - if (!i) return 0; - int ret = 0; - - // Currently, an item is always selected _and_ focussed in wxGTK - - if (stateMask & wxTREE_STATE_FOCUSED) - if (m_current == i) - ret += wxTREE_STATE_FOCUSED; - - if (stateMask & wxTREE_STATE_SELECTED) - if (m_current == i) - ret += wxTREE_STATE_SELECTED; - - if (stateMask & wxTREE_STATE_EXPANDED) - if (i->IsExpanded()) - ret += wxTREE_STATE_EXPANDED; - - return ret; -} - -bool wxTreeCtrl::SetItemState( long item, long state, long stateMask ) -{ - wxGenericTreeItem *i = FindItem( item ); - if (!i) return FALSE; - - if (stateMask & wxTREE_STATE_FOCUSED) - { - if (state & wxTREE_STATE_FOCUSED) - SelectItem( i ); - } - - if (stateMask & wxTREE_STATE_SELECTED) - { - if (state & wxTREE_STATE_SELECTED) - SelectItem( i ); - } - - if (stateMask & wxTREE_STATE_EXPANDED) - { - if (state & wxTREE_STATE_EXPANDED) - ExpandItem( i->m_itemId, wxTREE_EXPAND_EXPAND ); - else - ExpandItem( i->m_itemId, wxTREE_EXPAND_COLLAPSE ); - } - - return TRUE; + if ( (m_current == 0) || (m_key_current == 0) ) + { + event.Skip(); + return; + } + + bool is_multiple=(GetWindowStyleFlag() & wxTR_MULTIPLE); + bool extended_select=(event.ShiftDown() && is_multiple); + bool unselect_others=!(extended_select || (event.ControlDown() && is_multiple)); + + switch (event.KeyCode()) + { + case '+': + case WXK_ADD: + if (m_current->HasPlus() && !IsExpanded(m_current)) + { + Expand(m_current); + } + break; + + case '-': + case WXK_SUBTRACT: + if (IsExpanded(m_current)) + { + Collapse(m_current); + } + break; + + case '*': + case WXK_MULTIPLY: + Toggle(m_current); + break; + + case ' ': + case WXK_RETURN: + { + wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() ); + event.m_item = m_current; + event.m_code = 0; + event.SetEventObject( this ); + GetEventHandler()->ProcessEvent( event ); + } + break; + + // up goes to the previous sibling or to the last of its children if + // it's expanded + case WXK_UP: + { + wxTreeItemId prev = GetPrevSibling( m_key_current ); + if (!prev) + { + prev = GetParent( m_key_current ); + if (prev) + { + long cockie = 0; + wxTreeItemId current = m_key_current; + if (current == GetFirstChild( prev, cockie )) + { + // otherwise we return to where we came from + SelectItem( prev, unselect_others, extended_select ); + m_key_current=prev.m_pItem; + EnsureVisible( prev ); + break; + } + } + } + if (prev) + { + while ( IsExpanded(prev) && HasChildren(prev) ) + { + wxTreeItemId child = GetLastChild(prev); + if ( child ) + { + prev = child; + } + } + + SelectItem( prev, unselect_others, extended_select ); + m_key_current=prev.m_pItem; + EnsureVisible( prev ); + } + } + break; + + // left arrow goes to the parent + case WXK_LEFT: + { + wxTreeItemId prev = GetParent( m_current ); + if (prev) + { + EnsureVisible( prev ); + SelectItem( prev, unselect_others, extended_select ); + } + } + break; + + case WXK_RIGHT: + // this works the same as the down arrow except that we also expand the + // item if it wasn't expanded yet + Expand(m_current); + // fall through + + case WXK_DOWN: + { + if (IsExpanded(m_key_current) && HasChildren(m_key_current)) + { + long cookie = 0; + wxTreeItemId child = GetFirstChild( m_key_current, cookie ); + SelectItem( child, unselect_others, extended_select ); + m_key_current=child.m_pItem; + EnsureVisible( child ); + } + else + { + wxTreeItemId next = GetNextSibling( m_key_current ); +// if (next == 0) + if (!next) + { + wxTreeItemId current = m_key_current; + while (current && !next) + { + current = GetParent( current ); + if (current) next = GetNextSibling( current ); + } + } +// if (next != 0) + if (next) + { + SelectItem( next, unselect_others, extended_select ); + m_key_current=next.m_pItem; + EnsureVisible( next ); + } + } + } + break; + + // selects the last visible tree item + case WXK_END: + { + wxTreeItemId last = GetRootItem(); + + while ( last.IsOk() && IsExpanded(last) ) + { + wxTreeItemId lastChild = GetLastChild(last); + + // it may happen if the item was expanded but then all of + // its children have been deleted - so IsExpanded() returned + // TRUE, but GetLastChild() returned invalid item + if ( !lastChild ) + break; + + last = lastChild; + } + + if ( last.IsOk() ) + { + EnsureVisible( last ); + SelectItem( last, unselect_others, extended_select ); + } + } + break; + + // selects the root item + case WXK_HOME: + { + wxTreeItemId prev = GetRootItem(); + if (prev) + { + EnsureVisible( prev ); + SelectItem( prev, unselect_others, extended_select ); + } + } + break; + + default: + event.Skip(); + } } -long wxTreeCtrl::HitTest( const wxPoint& point, int &flags ) +wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& flags) { - flags = 0; - if (!m_anchor) return -1; - return m_anchor->HitTest( point, flags ); + // We have to call this here because the label in + // question might just have been added and no screen + // update taken place. + if (m_dirty) wxYield(); + + wxClientDC dc(this); + PrepareDC(dc); + long x = dc.DeviceToLogicalX( (long)point.x ); + long y = dc.DeviceToLogicalY( (long)point.y ); + int w, h; + GetSize(&w, &h); + + flags=0; + if (point.x<0) flags|=wxTREE_HITTEST_TOLEFT; + if (point.x>w) flags|=wxTREE_HITTEST_TORIGHT; + if (point.y<0) flags|=wxTREE_HITTEST_ABOVE; + if (point.y>h) flags|=wxTREE_HITTEST_BELOW; + + return m_anchor->HitTest( wxPoint(x, y), this, flags); } -wxImageList *wxTreeCtrl::GetImageList( int which ) const +/* **** */ + +void wxTreeCtrl::Edit( const wxTreeItemId& item ) { - if (which == wxIMAGE_LIST_NORMAL) return m_imageList; - return m_smallImageList; + if (!item.IsOk()) return; + + m_currentEdit = item.m_pItem; + + wxTreeEvent te( wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT, GetId() ); + te.m_item = m_currentEdit; + te.SetEventObject( this ); + GetEventHandler()->ProcessEvent( te ); + + if (!te.IsAllowed()) return; + + // We have to call this here because the label in + // question might just have been added and no screen + // update taken place. + if (m_dirty) wxYield(); + + wxString s = m_currentEdit->GetText(); + int x = m_currentEdit->GetX(); + int y = m_currentEdit->GetY(); + int w = m_currentEdit->GetWidth(); + int h = m_currentEdit->GetHeight(); + + int image_h = 0; + int image_w = 0; + + int image = m_currentEdit->GetCurrentImage(); + if ( image != NO_IMAGE ) + { + m_imageListNormal->GetSize( image, image_w, image_h ); + image_w += 4; + } + x += image_w; + w -= image_w + 4; // I don't know why +4 is needed + + wxClientDC dc(this); + PrepareDC( dc ); + x = dc.LogicalToDeviceX( x ); + y = dc.LogicalToDeviceY( y ); + + wxTreeTextCtrl *text = new wxTreeTextCtrl( + this, -1, &m_renameAccept, &m_renameRes, this, s, wxPoint(x-4,y-4), wxSize(w+11,h+8) ); + text->SetFocus(); } -void wxTreeCtrl::SetImageList( wxImageList *imageList, int which ) +void wxTreeCtrl::OnRenameTimer() { - if (which == wxIMAGE_LIST_NORMAL) - { - if (m_imageList) delete m_imageList; - m_imageList = imageList; - } - else - { - if (m_smallImageList) delete m_smallImageList; - m_smallImageList = imageList; - } + Edit( m_current ); } -void wxTreeCtrl::AdjustMyScrollbars() +void wxTreeCtrl::OnRenameAccept() { - if (m_anchor) - { - int x = 0; - int y = 0; - m_anchor->GetSize( x, y ); - y += 2*m_lineHeight; - int x_pos = GetScrollPos( wxHORIZONTAL ); - int y_pos = GetScrollPos( wxVERTICAL ); - SetScrollbars( 10, 10, x/10, y/10, x_pos, y_pos ); - } - else - { - SetScrollbars( 0, 0, 0, 0 ); - } + wxTreeEvent le( wxEVT_COMMAND_TREE_END_LABEL_EDIT, GetId() ); + le.m_item = m_currentEdit; + le.SetEventObject( this ); + le.m_label = m_renameRes; + GetEventHandler()->ProcessEvent( le ); + + if (!le.IsAllowed()) return; + + SetItemText( m_currentEdit, m_renameRes ); } -void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxPaintDC &dc, int level, int &y ) +void wxTreeCtrl::OnMouse( wxMouseEvent &event ) { - int horizX = level*m_indent; + if ( !(event.LeftUp() || event.RightDown() || event.LeftDClick() || event.Dragging()) ) return; - item->m_x = horizX+33; - item->m_y = y-m_lineHeight/3; - item->m_height = m_lineHeight; + if ( !m_anchor ) return; - item->SetCross( horizX+15, y ); + wxClientDC dc(this); + PrepareDC(dc); + long x = dc.DeviceToLogicalX( (long)event.GetX() ); + long y = dc.DeviceToLogicalY( (long)event.GetY() ); - int oldY = y; + int flags=0; + wxGenericTreeItem *item = m_anchor->HitTest( wxPoint(x,y), this, flags); + bool onButton = flags & wxTREE_HITTEST_ONITEMBUTTON; - int exposed_x = dc.LogicalToDeviceX( 0 ); - int exposed_y = dc.LogicalToDeviceY( item->m_y-2 ); - - if (IsExposed( exposed_x, exposed_y, 1000, m_lineHeight+4 )) - { - int startX = horizX; - int endX = horizX + 10; + if (event.Dragging()) + { + if (m_dragCount == 0) + m_dragStart = wxPoint(x,y); - if (!item->HasChildren()) endX += 20; - - dc.DrawLine( startX, y, endX, y ); + m_dragCount++; - if (item->HasChildren()) - { - dc.DrawLine( horizX+20, y, horizX+30, y ); - dc.SetPen( *wxGREY_PEN ); - dc.DrawRectangle( horizX+10, y-4, 11, 9 ); - dc.SetPen( *wxBLACK_PEN ); - dc.DrawLine( horizX+13, y, horizX+18, y ); - if (item->HasPlus()) - dc.DrawLine( horizX+15, y-2, horizX+15, y+3 ); - } + if (m_dragCount != 3) return; - if (item->HasHilight()) - { - dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) ); - dc.SetBrush( *m_hilightBrush ); - long tw, th; - dc.GetTextExtent( item->m_text, &tw, &th ); - if (m_hasFocus) - { - dc.SetPen( *wxBLACK_PEN ); - dc.DrawRectangle( item->m_x-2, item->m_y-2, tw+4, th+4 ); - } - else - { - dc.SetPen( *wxTRANSPARENT_PEN ); - dc.DrawRectangle( item->m_x-2, item->m_y-2, tw+4, th+4 ); - } - dc.DrawText( item->m_text, item->m_x, item->m_y ); + int command = wxEVT_COMMAND_TREE_BEGIN_DRAG; + if (event.RightIsDown()) command = wxEVT_COMMAND_TREE_BEGIN_RDRAG; - dc.SetPen( *wxBLACK_PEN ); - dc.SetTextForeground( *wxBLACK ); - dc.SetBrush( *wxWHITE_BRUSH ); - } - else - { - dc.SetBrush( *wxWHITE_BRUSH ); - dc.SetPen( *wxTRANSPARENT_PEN ); - long tw, th; - dc.GetTextExtent( item->m_text, &tw, &th ); - dc.DrawRectangle( item->m_x-2, item->m_y-2, tw+4, th+4 ); - dc.DrawText( item->m_text, item->m_x, item->m_y ); - dc.SetPen( *wxBLACK_PEN ); - } - } + wxTreeEvent nevent( command, GetId() ); + nevent.m_item = m_current; + nevent.SetEventObject(this); + GetEventHandler()->ProcessEvent(nevent); + return; + } + else + { + m_dragCount = 0; + } - if (item->NumberOfVisibleChildren() == 0) return; + if (item == NULL) return; /* we hit the blank area */ - int semiOldY = y; + if (event.RightDown()) { + wxTreeEvent nevent(wxEVT_COMMAND_TREE_ITEM_RIGHT_CLICK,GetId()); + nevent.m_item=item; + nevent.m_code=0; + nevent.SetEventObject(this); + GetEventHandler()->ProcessEvent(nevent); + return; + } - wxNode *node = item->m_children.First(); - while (node) - { - wxGenericTreeItem *child = (wxGenericTreeItem *)node->Data(); + if (event.LeftUp() && (item == m_current) && + (flags & wxTREE_HITTEST_ONITEMLABEL) && + HasFlag(wxTR_EDIT_LABELS) ) + { + m_renameTimer->Start( 100, TRUE ); + return; + } - y += m_lineHeight; - semiOldY = y; + bool is_multiple=(GetWindowStyleFlag() & wxTR_MULTIPLE); + bool extended_select=(event.ShiftDown() && is_multiple); + bool unselect_others=!(extended_select || (event.ControlDown() && is_multiple)); - PaintLevel( child, dc, level+1, y ); + if (onButton) + { + Toggle( item ); + if (is_multiple) + return; + } - node = node->Next(); - } + SelectItem(item, unselect_others, extended_select); - dc.DrawLine( horizX+15, oldY+5, horizX+15, semiOldY ); + if (event.LeftDClick()) + { + wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() ); + event.m_item = item; + event.m_code = 0; + event.SetEventObject( this ); + GetEventHandler()->ProcessEvent( event ); + } } -void wxTreeCtrl::OnPaint( const wxPaintEvent &WXUNUSED(event) ) +void wxTreeCtrl::OnIdle( wxIdleEvent &WXUNUSED(event) ) { - if (!m_anchor) return; - - wxPaintDC dc(this); - PrepareDC( dc ); + /* after all changes have been done to the tree control, + * we actually redraw the tree when everything is over */ - dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_SYSTEM_FONT ) ); + if (!m_dirty) + return; - dc.SetPen( m_dottedPen ); - m_lineHeight = (int)(dc.GetCharHeight() + 4); - - int y = m_lineHeight / 2 + 2; - PaintLevel( m_anchor, dc, 0, y ); -} + m_dirty = FALSE; -void wxTreeCtrl::OnSetFocus( const wxFocusEvent &WXUNUSED(event) ) -{ - m_hasFocus = TRUE; - if (m_current) RefreshLine( m_current ); + CalculatePositions(); + Refresh(); + AdjustMyScrollbars(); } -void wxTreeCtrl::OnKillFocus( const wxFocusEvent &WXUNUSED(event) ) +void wxTreeCtrl::CalculateSize( wxGenericTreeItem *item, wxDC &dc ) { - m_hasFocus = FALSE; - if (m_current) RefreshLine( m_current ); -} + long text_w = 0; + long text_h = 0; -void wxTreeCtrl::OnChar( wxKeyEvent &event ) -{ - event.Skip(); -} + if (item->IsBold()) + dc.SetFont(m_boldFont); -void wxTreeCtrl::OnMouse( const wxMouseEvent &event ) -{ - if (!event.LeftDown() && - !event.LeftDClick()) return; + dc.GetTextExtent( item->GetText(), &text_w, &text_h ); + text_h+=2; - wxClientDC dc(this); - PrepareDC(dc); - long x = dc.DeviceToLogicalX( (long)event.GetX() ); - long y = dc.DeviceToLogicalY( (long)event.GetY() ); + // restore normal font + dc.SetFont( m_normalFont ); - int flag = 0; - long id = HitTest( wxPoint(x,y), flag ); - if (id == -1) - return; - wxGenericTreeItem *item = FindItem( id ); + int image_h = 0; + int image_w = 0; + int image = item->GetCurrentImage(); + if ( image != NO_IMAGE ) + { + m_imageListNormal->GetSize( image, image_w, image_h ); + image_w += 4; + } - if (!item) return; - if ((flag != wxTREE_HITTEST_ONITEMBUTTON) && - (flag != wxTREE_HITTEST_ONITEMLABEL)) return; + int total_h = (image_h > text_h) ? image_h : text_h; - SelectItem(item); + if (total_h<40) total_h+=4; // at least 4 pixels + else total_h+=total_h/10; // otherwise 10% extra spacing - if (event.LeftDClick()) - m_current->SendKeyDown( this ); + item->SetHeight(total_h); + if (total_h>m_lineHeight) m_lineHeight=total_h; - if (flag == wxTREE_HITTEST_ONITEMBUTTON) - { - ExpandItem( item->m_itemId, wxTREE_EXPAND_TOGGLE ); - return; - } + item->SetWidth(image_w+text_w+2); } -void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxPaintDC &dc, int level, int &y ) +// ----------------------------------------------------------------------------- +// for developper : y is now the top of the level +// not the middle of it ! +void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y ) { - int horizX = level*m_indent; + int horizX = level*m_indent; - item->m_x = horizX+33; - item->m_y = y-m_lineHeight/3-2; - item->m_height = m_lineHeight; + CalculateSize( item, dc ); - if (item->NumberOfVisibleChildren() == 0) return; - - wxNode *node = item->m_children.First(); - while (node) - { - wxGenericTreeItem *child = (wxGenericTreeItem *)node->Data(); + // set its position + item->SetX( horizX+m_indent+m_spacing ); + item->SetY( y ); + y+=GetLineHeight(item); - y += m_lineHeight; - CalculateLevel( child, dc, level+1, y ); + if ( !item->IsExpanded() ) + { + // we dont need to calculate collapsed branches + return; + } - node = node->Next(); - } + wxArrayGenericTreeItems& children = item->GetChildren(); + size_t n, count = children.Count(); + for (n = 0; n < count; ++n ) + CalculateLevel( children[n], dc, level+1, y ); // recurse } void wxTreeCtrl::CalculatePositions() { - if (!m_anchor) - return; + if ( !m_anchor ) return; - wxClientDC dc(this); - PrepareDC( dc ); + wxClientDC dc(this); + PrepareDC( dc ); - dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_SYSTEM_FONT ) ); + dc.SetFont( m_normalFont ); - dc.SetPen( m_dottedPen ); - m_lineHeight = (int)(dc.GetCharHeight() + 4); + dc.SetPen( m_dottedPen ); + //if(GetImageList() == NULL) + // m_lineHeight = (int)(dc.GetCharHeight() + 4); - int y = m_lineHeight / 2 + 2; - CalculateLevel( m_anchor, dc, 0, y ); + int y = 2; + CalculateLevel( m_anchor, dc, 0, y ); // start recursion } -wxGenericTreeItem *wxTreeCtrl::FindItem( long itemId ) const +void wxTreeCtrl::RefreshSubtree(wxGenericTreeItem *item) { - if (!m_anchor) return (wxGenericTreeItem *) NULL; - return m_anchor->FindItem( itemId ); + wxClientDC dc(this); + PrepareDC(dc); + + int cw = 0; + int ch = 0; + GetClientSize( &cw, &ch ); + + wxRect rect; + rect.x = dc.LogicalToDeviceX( 0 ); + rect.width = cw; + rect.y = dc.LogicalToDeviceY( item->GetY() ); + rect.height = ch; + + Refresh( TRUE, &rect ); + + AdjustMyScrollbars(); } void wxTreeCtrl::RefreshLine( wxGenericTreeItem *item ) { - if (!item) return; - wxClientDC dc(this); - PrepareDC( dc ); - wxRect rect; - rect.x = dc.LogicalToDeviceX( item->m_x-2 ); - rect.y = dc.LogicalToDeviceY( item->m_y-2 ); - rect.width = 1000; - rect.height = dc.GetCharHeight()+6; - Refresh( TRUE, &rect ); -} + wxClientDC dc(this); + PrepareDC( dc ); + + int cw = 0; + int ch = 0; + GetClientSize( &cw, &ch ); + wxRect rect; + rect.x = dc.LogicalToDeviceX( 0 ); + rect.y = dc.LogicalToDeviceY( item->GetY() ); + rect.width = cw; + rect.height = GetLineHeight(item); //dc.GetCharHeight() + 6; + Refresh( TRUE, &rect ); +}