+ wxCHECK_MSG( s_treeBeingSorted, 0, _T("bug in wxTreeCtrl::SortChildren()") );
+
+ return s_treeBeingSorted->OnCompareItems(*item1, *item2);
+}
+
+int wxTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
+ const wxTreeItemId& item2)
+{
+ return wxStrcmp(GetItemText(item1), GetItemText(item2));
+}
+
+void wxTreeCtrl::SortChildren(const wxTreeItemId& itemId)
+{
+ wxCHECK_RET( itemId.IsOk(), _T("invalid tree item") );
+
+ wxGenericTreeItem *item = itemId.m_pItem;
+
+ wxCHECK_RET( !s_treeBeingSorted,
+ _T("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
+}
+
+wxImageList *wxTreeCtrl::GetImageList() const
+{
+ return m_imageListNormal;
+}
+
+wxImageList *wxTreeCtrl::GetStateImageList() const
+{
+ return m_imageListState;
+}
+
+void wxTreeCtrl::SetImageList(wxImageList *imageList)
+{
+ 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
+ wxPaintDC 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
+}
+
+void wxTreeCtrl::SetStateImageList(wxImageList *imageList)
+{
+ m_imageListState = imageList;
+}
+
+// -----------------------------------------------------------------------------
+// helpers
+// -----------------------------------------------------------------------------
+
+void wxTreeCtrl::AdjustMyScrollbars()
+{
+ if (m_anchor)
+ {
+ 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 );
+ }
+ else
+ {
+ SetScrollbars( 0, 0, 0, 0 );
+ }
+}
+
+int wxTreeCtrl::GetLineHeight(wxGenericTreeItem *item) const
+{
+ if (GetWindowStyleFlag() & wxTR_HAS_VARIABLE_ROW_HEIGHT)
+ return item->GetHeight();
+ else
+ return m_lineHeight;
+}
+
+void wxTreeCtrl::PaintItem(wxGenericTreeItem *item, wxDC& dc)
+{
+ // render bold items in bold
+ wxFont fontOld;
+ wxFont fontNew;
+
+ if (item->IsBold())
+ {
+ fontOld = dc.GetFont();
+ if (fontOld.Ok())
+ {
+ // VZ: is there any better way to make a bold variant of old font?
+ fontNew = wxFont( fontOld.GetPointSize(),
+ fontOld.GetFamily(),
+ fontOld.GetStyle(),
+ wxBOLD,
+ fontOld.GetUnderlined());
+ dc.SetFont(fontNew);
+ }
+ else
+ {
+ wxFAIL_MSG(_T("wxDC::GetFont() failed!"));
+ }
+ }
+
+ long text_w = 0;
+ long text_h = 0;
+ dc.GetTextExtent( item->GetText(), &text_w, &text_h );
+
+ int image_h = 0;
+ int image_w = 0;
+ if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
+ {
+ m_imageListNormal->GetSize( item->GetSelectedImage(), image_w, image_h );
+ image_w += 4;
+ }
+ else if (item->GetImage() != -1)
+ {
+ m_imageListNormal->GetSize( item->GetImage(), image_w, image_h );
+ image_w += 4;
+ }
+
+ int total_h = GetLineHeight(item);
+
+ dc.DrawRectangle( item->GetX()-2, item->GetY(), item->GetWidth()+2, total_h );
+
+ if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
+ {
+ dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
+ m_imageListNormal->Draw( item->GetSelectedImage(), dc,
+ item->GetX(),
+ item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
+ wxIMAGELIST_DRAW_TRANSPARENT );
+ dc.DestroyClippingRegion();
+ }
+ else if (item->GetImage() != -1)
+ {
+ dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
+ m_imageListNormal->Draw( item->GetImage(), dc,
+ item->GetX(),
+ item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
+ wxIMAGELIST_DRAW_TRANSPARENT );
+ dc.DestroyClippingRegion();
+ }
+
+ dc.SetBackgroundMode(wxTRANSPARENT);
+ dc.DrawText( item->GetText(), image_w + item->GetX(), item->GetY()
+ + ((total_h > text_h) ? (total_h - text_h)/2 : 0));
+
+ // restore normal font for bold items
+ if (fontOld.Ok())
+ {
+ dc.SetFont( fontOld);
+ }
+}
+
+// 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 )
+{
+ 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 );
+ }
+
+ if (item->HasHilight())
+ {
+ dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) );
+
+ dc.SetBrush( *m_hilightBrush );
+
+ if (m_hasFocus)
+ dc.SetPen( *wxBLACK_PEN );
+ else
+ dc.SetPen( *wxTRANSPARENT_PEN );
+
+ PaintItem(item, dc);
+
+ dc.SetPen( m_dottedPen );
+ dc.SetTextForeground( *wxBLACK );
+ dc.SetBrush( *wxWHITE_BRUSH );
+ }
+ else
+ {
+ dc.SetBrush( *wxWHITE_BRUSH );
+ dc.SetPen( *wxTRANSPARENT_PEN );
+
+ PaintItem(item, dc);
+
+ dc.SetPen( m_dottedPen );
+ }
+ }
+
+ y = oldY+GetLineHeight(item);
+
+ if (item->IsExpanded())
+ {
+ oldY+=GetLineHeight(item)/2;
+ int semiOldY=y; // (=y) for stupid compilator
+
+ 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::DrawBorder(wxTreeItemId &item)
+{
+ if (!item) return;
+
+ wxGenericTreeItem *i=item.m_pItem;
+
+ wxPaintDC 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);
+}
+
+void wxTreeCtrl::DrawLine(wxTreeItemId &item, bool below)
+{
+ if (!item) return;
+
+ wxGenericTreeItem *i=item.m_pItem;
+
+ wxPaintDC 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);
+}
+
+// -----------------------------------------------------------------------------
+// wxWindows callbacks
+// -----------------------------------------------------------------------------
+
+void wxTreeCtrl::OnPaint( wxPaintEvent &WXUNUSED(event) )
+{
+ if ( !m_anchor)
+ return;
+
+ wxPaintDC dc(this);
+ PrepareDC( dc );
+
+ dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
+
+ dc.SetPen( m_dottedPen );
+ //if(GetImageList() == NULL)
+ // m_lineHeight = (int)(dc.GetCharHeight() + 4);
+
+ int y = 2;
+ PaintLevel( m_anchor, dc, 0, y );
+}
+
+void wxTreeCtrl::OnSetFocus( wxFocusEvent &WXUNUSED(event) )
+{
+ m_hasFocus = TRUE;
+
+ if (m_current) RefreshLine( m_current );
+}
+
+void wxTreeCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) )
+{
+ m_hasFocus = FALSE;
+
+ if (m_current) RefreshLine( m_current );
+}
+
+void wxTreeCtrl::OnChar( wxKeyEvent &event )
+{
+ wxTreeEvent te( wxEVT_COMMAND_TREE_KEY_DOWN, GetId() );
+ te.m_code = event.KeyCode();
+ te.SetEventObject( this );
+ GetEventHandler()->ProcessEvent( te );
+
+ 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;
+
+ // <End> 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;
+
+ // <Home> selects the root item
+ case WXK_HOME:
+ {
+ wxTreeItemId prev = GetRootItem();
+ if (prev)
+ {
+ EnsureVisible( prev );
+ SelectItem( prev, unselect_others, extended_select );
+ }
+ }
+ break;
+
+ default:
+ event.Skip();
+ }
+}
+
+wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& 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);
+}
+
+/* **** */
+
+void wxTreeCtrl::Edit( const wxTreeItemId& item )
+{
+ 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;