]> git.saurik.com Git - wxWidgets.git/blame_incremental - src/generic/treectrl.cpp
projects / wxWidgets.git / blame_incremental
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (non-incremental) | history | HEAD
Made various Motif fixes, wxListBox/wxChoice derive from wxControlWithItems,
[wxWidgets.git] / src / generic / treectrl.cpp
... / ...
CommitLineData
1/////////////////////////////////////////////////////////////////////////////
2// Name: treectrl.cpp
3// Purpose: generic tree control implementation
4// Author: Robert Roebling
5// Created: 01/02/97
6// Modified: 22/10/98 - almost total rewrite, simpler interface (VZ)
7// Id: $Id$
8// Copyright: (c) 1998 Robert Roebling, Julian Smart and Markus Holzem
9// Licence: wxWindows licence
10/////////////////////////////////////////////////////////////////////////////
11
12// =============================================================================
13// declarations
14// =============================================================================
15
16// -----------------------------------------------------------------------------
17// headers
18// -----------------------------------------------------------------------------
19
20#ifdef __GNUG__
21 #pragma implementation "treectrl.h"
22#endif
23
24// For compilers that support precompilation, includes "wx.h".
25#include "wx/wxprec.h"
26
27#ifdef __BORLANDC__
28#pragma hdrstop
29#endif
30
31#include "wx/treectrl.h"
32#include "wx/generic/imaglist.h"
33#include "wx/settings.h"
34#include "wx/log.h"
35#include "wx/intl.h"
36#include "wx/dynarray.h"
37#include "wx/arrimpl.cpp"
38#include "wx/dcclient.h"
39#include "wx/msgdlg.h"
40
41// -----------------------------------------------------------------------------
42// array types
43// -----------------------------------------------------------------------------
44
45class WXDLLEXPORT wxGenericTreeItem;
46
47WX_DEFINE_ARRAY(wxGenericTreeItem *, wxArrayGenericTreeItems);
48WX_DEFINE_OBJARRAY(wxArrayTreeItemIds);
49
50// ----------------------------------------------------------------------------
51// constants
52// ----------------------------------------------------------------------------
53
54static const int NO_IMAGE = -1;
55
56// -----------------------------------------------------------------------------
57// private classes
58// -----------------------------------------------------------------------------
59
60// a tree item
61class WXDLLEXPORT wxGenericTreeItem
62{
63public:
64 // ctors & dtor
65 wxGenericTreeItem() { m_data = NULL; }
66 wxGenericTreeItem( wxGenericTreeItem *parent,
67 const wxString& text,
68 wxDC& dc,
69 int image, int selImage,
70 wxTreeItemData *data );
71
72 ~wxGenericTreeItem();
73
74 // trivial accessors
75 wxArrayGenericTreeItems& GetChildren() { return m_children; }
76
77 const wxString& GetText() const { return m_text; }
78 int GetImage(wxTreeItemIcon which = wxTreeItemIcon_Normal) const
79 { return m_images[which]; }
80 wxTreeItemData *GetData() const { return m_data; }
81
82 // returns the current image for the item (depending on its
83 // selected/expanded/whatever state)
84 int GetCurrentImage() const;
85
86 void SetText( const wxString &text );
87 void SetImage(int image, wxTreeItemIcon which) { m_images[which] = image; }
88 void SetData(wxTreeItemData *data) { m_data = data; }
89
90 void SetHasPlus(bool has = TRUE) { m_hasPlus = has; }
91
92 void SetBold(bool bold) { m_isBold = bold; }
93
94 int GetX() const { return m_x; }
95 int GetY() const { return m_y; }
96
97 void SetX(int x) { m_x = x; }
98 void SetY(int y) { m_y = y; }
99
100 int GetHeight() const { return m_height; }
101 int GetWidth() const { return m_width; }
102
103 void SetHeight(int h) { m_height = h; }
104 void SetWidth(int w) { m_width = w; }
105
106
107 wxGenericTreeItem *GetParent() const { return m_parent; }
108
109 // operations
110 // deletes all children notifying the treectrl about it if !NULL pointer
111 // given
112 void DeleteChildren(wxTreeCtrl *tree = NULL);
113 // FIXME don't know what is it for
114 void Reset();
115
116 // get count of all children (and grand children if 'recursively')
117 size_t GetChildrenCount(bool recursively = TRUE) const;
118
119 void Insert(wxGenericTreeItem *child, size_t index)
120 { m_children.Insert(child, index); }
121
122 void SetCross( int x, int y );
123 void GetSize( int &x, int &y, const wxTreeCtrl* );
124
125 // return the item at given position (or NULL if no item), onButton is TRUE
126 // if the point belongs to the item's button, otherwise it lies on the
127 // button's label
128 wxGenericTreeItem *HitTest( const wxPoint& point, const wxTreeCtrl *, int &flags);
129
130 void Expand() { m_isCollapsed = FALSE; }
131 void Collapse() { m_isCollapsed = TRUE; }
132
133 void SetHilight( bool set = TRUE ) { m_hasHilight = set; }
134
135 // status inquiries
136 bool HasChildren() const { return !m_children.IsEmpty(); }
137 bool IsSelected() const { return m_hasHilight; }
138 bool IsExpanded() const { return !m_isCollapsed; }
139 bool HasPlus() const { return m_hasPlus || HasChildren(); }
140 bool IsBold() const { return m_isBold; }
141
142private:
143 wxString m_text;
144
145 // tree ctrl images for the normal, selected, expanded and expanded+selected
146 // states
147 int m_images[wxTreeItemIcon_Max];
148
149 wxTreeItemData *m_data;
150
151 // use bitfields to save size
152 int m_isCollapsed :1;
153 int m_hasHilight :1; // same as focused
154 int m_hasPlus :1; // used for item which doesn't have
155 // children but still has a [+] button
156 int m_isBold :1; // render the label in bold font
157
158 int m_x, m_y;
159 long m_height, m_width;
160 int m_xCross, m_yCross;
161 int m_level;
162 wxArrayGenericTreeItems m_children;
163 wxGenericTreeItem *m_parent;
164};
165
166// =============================================================================
167// implementation
168// =============================================================================
169
170
171// -----------------------------------------------------------------------------
172// wxTreeRenameTimer (internal)
173// -----------------------------------------------------------------------------
174
175wxTreeRenameTimer::wxTreeRenameTimer( wxTreeCtrl *owner )
176{
177 m_owner = owner;
178}
179
180void wxTreeRenameTimer::Notify()
181{
182 m_owner->OnRenameTimer();
183}
184
185//-----------------------------------------------------------------------------
186// wxTreeTextCtrl (internal)
187//-----------------------------------------------------------------------------
188
189IMPLEMENT_DYNAMIC_CLASS(wxTreeTextCtrl,wxTextCtrl);
190
191BEGIN_EVENT_TABLE(wxTreeTextCtrl,wxTextCtrl)
192 EVT_CHAR (wxTreeTextCtrl::OnChar)
193 EVT_KILL_FOCUS (wxTreeTextCtrl::OnKillFocus)
194END_EVENT_TABLE()
195
196wxTreeTextCtrl::wxTreeTextCtrl( wxWindow *parent, const wxWindowID id,
197 bool *accept, wxString *res, wxTreeCtrl *owner,
198 const wxString &value, const wxPoint &pos, const wxSize &size,
199#if wxUSE_VALIDATORS
200# if defined(__VISAGECPP__)
201 int style, const wxValidator* validator, const wxString &name ) :
202# else
203 int style, const wxValidator& validator, const wxString &name ) :
204# endif
205#endif
206 wxTextCtrl( parent, id, value, pos, size, style, validator, name )
207{
208 m_res = res;
209 m_accept = accept;
210 m_owner = owner;
211 (*m_accept) = FALSE;
212 (*m_res) = "";
213 m_startValue = value;
214}
215
216void wxTreeTextCtrl::OnChar( wxKeyEvent &event )
217{
218 if (event.m_keyCode == WXK_RETURN)
219 {
220 (*m_accept) = TRUE;
221 (*m_res) = GetValue();
222 m_owner->SetFocus();
223 return;
224 }
225 if (event.m_keyCode == WXK_ESCAPE)
226 {
227 (*m_accept) = FALSE;
228 (*m_res) = "";
229 m_owner->SetFocus();
230 return;
231 }
232 event.Skip();
233}
234
235void wxTreeTextCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) )
236{
237 if (wxPendingDelete.Member(this)) return;
238
239 wxPendingDelete.Append(this);
240
241 if ((*m_accept) && ((*m_res) != m_startValue))
242 m_owner->OnRenameAccept();
243}
244
245#define PIXELS_PER_UNIT 10
246// -----------------------------------------------------------------------------
247// wxTreeEvent
248// -----------------------------------------------------------------------------
249
250IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent, wxNotifyEvent)
251
252wxTreeEvent::wxTreeEvent( wxEventType commandType, int id )
253 : wxNotifyEvent( commandType, id )
254{
255 m_code = 0;
256 m_itemOld = (wxGenericTreeItem *)NULL;
257}
258
259// -----------------------------------------------------------------------------
260// wxGenericTreeItem
261// -----------------------------------------------------------------------------
262
263wxGenericTreeItem::wxGenericTreeItem(wxGenericTreeItem *parent,
264 const wxString& text,
265 wxDC& dc,
266 int image, int selImage,
267 wxTreeItemData *data)
268 : m_text(text)
269{
270 m_images[wxTreeItemIcon_Normal] = image;
271 m_images[wxTreeItemIcon_Selected] = selImage;
272 m_images[wxTreeItemIcon_Expanded] = NO_IMAGE;
273 m_images[wxTreeItemIcon_SelectedExpanded] = NO_IMAGE;
274
275 m_data = data;
276 m_x = m_y = 0;
277 m_xCross = m_yCross = 0;
278
279 m_level = 0;
280
281 m_isCollapsed = TRUE;
282 m_hasHilight = FALSE;
283 m_hasPlus = FALSE;
284 m_isBold = FALSE;
285
286 m_parent = parent;
287
288 dc.GetTextExtent( m_text, &m_width, &m_height );
289 // TODO : Add the width of the image
290 // PB : We don't know which image is shown (image, selImage)
291 // We don't even know imageList from the treectrl this item belongs to !!!
292 // At this point m_width doesn't mean much, this can be remove !
293}
294
295wxGenericTreeItem::~wxGenericTreeItem()
296{
297 delete m_data;
298
299 wxASSERT_MSG( m_children.IsEmpty(),
300 wxT("please call DeleteChildren() before deleting the item") );
301}
302
303void wxGenericTreeItem::DeleteChildren(wxTreeCtrl *tree)
304{
305 size_t count = m_children.Count();
306 for ( size_t n = 0; n < count; n++ )
307 {
308 wxGenericTreeItem *child = m_children[n];
309 if ( tree )
310 {
311 tree->SendDeleteEvent(child);
312 }
313
314 child->DeleteChildren(tree);
315 delete child;
316 }
317
318 m_children.Empty();
319}
320
321void wxGenericTreeItem::SetText( const wxString &text )
322{
323 m_text = text;
324}
325
326void wxGenericTreeItem::Reset()
327{
328 m_text.Empty();
329 for ( int i = 0; i < wxTreeItemIcon_Max; i++ )
330 {
331 m_images[i] = NO_IMAGE;
332 }
333
334 m_data = NULL;
335 m_x = m_y =
336 m_height = m_width = 0;
337 m_xCross =
338 m_yCross = 0;
339
340 m_level = 0;
341
342 DeleteChildren();
343 m_isCollapsed = TRUE;
344
345 m_parent = (wxGenericTreeItem *)NULL;
346}
347
348size_t wxGenericTreeItem::GetChildrenCount(bool recursively) const
349{
350 size_t count = m_children.Count();
351 if ( !recursively )
352 return count;
353
354 size_t total = count;
355 for ( size_t n = 0; n < count; ++n )
356 {
357 total += m_children[n]->GetChildrenCount();
358 }
359
360 return total;
361}
362
363void wxGenericTreeItem::SetCross( int x, int y )
364{
365 m_xCross = x;
366 m_yCross = y;
367}
368
369void wxGenericTreeItem::GetSize( int &x, int &y, const wxTreeCtrl *theTree )
370{
371 int bottomY=m_y+theTree->GetLineHeight(this);
372 if ( y < bottomY ) y = bottomY;
373 int width = m_x + m_width;
374 if ( x < width ) x = width;
375
376 if (IsExpanded())
377 {
378 size_t count = m_children.Count();
379 for ( size_t n = 0; n < count; ++n )
380 {
381 m_children[n]->GetSize( x, y, theTree );
382 }
383 }
384}
385
386wxGenericTreeItem *wxGenericTreeItem::HitTest( const wxPoint& point,
387 const wxTreeCtrl *theTree,
388 int &flags)
389{
390 if ((point.y > m_y) && (point.y < m_y + theTree->GetLineHeight(this)))
391 {
392 if (point.y<m_y+theTree->GetLineHeight(this)/2)
393 flags |= wxTREE_HITTEST_ONITEMUPPERPART;
394 else
395 flags |= wxTREE_HITTEST_ONITEMLOWERPART;
396
397 // 5 is the size of the plus sign
398 if ((point.x > m_xCross-5) && (point.x < m_xCross+5) &&
399 (point.y > m_yCross-5) && (point.y < m_yCross+5) &&
400 (IsExpanded() || HasPlus()))
401 {
402 flags|=wxTREE_HITTEST_ONITEMBUTTON;
403 return this;
404 }
405
406 if ((point.x >= m_x) && (point.x <= m_x+m_width))
407 {
408 int image_w = -1;
409 int image_h;
410
411 // assuming every image (normal and selected ) has the same size !
412 if ( (GetImage() != NO_IMAGE) && theTree->m_imageListNormal )
413 theTree->m_imageListNormal->GetSize(GetImage(), image_w, image_h);
414
415 if ((image_w != -1) && (point.x <= m_x + image_w + 1))
416 flags |= wxTREE_HITTEST_ONITEMICON;
417 else
418 flags |= wxTREE_HITTEST_ONITEMLABEL;
419
420 return this;
421 }
422
423 if (point.x < m_x)
424 flags |= wxTREE_HITTEST_ONITEMIDENT;
425 if (point.x > m_x+m_width)
426 flags |= wxTREE_HITTEST_ONITEMRIGHT;
427
428 return this;
429 }
430 else
431 {
432 if (!m_isCollapsed)
433 {
434 size_t count = m_children.Count();
435 for ( size_t n = 0; n < count; n++ )
436 {
437 wxGenericTreeItem *res = m_children[n]->HitTest( point, theTree, flags );
438 if ( res != NULL )
439 return res;
440 }
441 }
442 }
443
444 flags|=wxTREE_HITTEST_NOWHERE;
445 return NULL;
446}
447
448int wxGenericTreeItem::GetCurrentImage() const
449{
450 int image = NO_IMAGE;
451 if ( IsExpanded() )
452 {
453 if ( IsSelected() )
454 {
455 image = GetImage(wxTreeItemIcon_SelectedExpanded);
456 }
457
458 if ( image == NO_IMAGE )
459 {
460 // we usually fall back to the normal item, but try just the
461 // expanded one (and not selected) first in this case
462 image = GetImage(wxTreeItemIcon_Expanded);
463 }
464 }
465 else // not expanded
466 {
467 if ( IsSelected() )
468 image = GetImage(wxTreeItemIcon_Selected);
469 }
470
471 // may be it doesn't have the specific image we want, try the default one
472 // instead
473 if ( image == NO_IMAGE )
474 {
475 image = GetImage();
476 }
477
478 return image;
479}
480
481// -----------------------------------------------------------------------------
482// wxTreeCtrl implementation
483// -----------------------------------------------------------------------------
484
485IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxScrolledWindow)
486
487BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow)
488 EVT_PAINT (wxTreeCtrl::OnPaint)
489 EVT_MOUSE_EVENTS (wxTreeCtrl::OnMouse)
490 EVT_CHAR (wxTreeCtrl::OnChar)
491 EVT_SET_FOCUS (wxTreeCtrl::OnSetFocus)
492 EVT_KILL_FOCUS (wxTreeCtrl::OnKillFocus)
493 EVT_IDLE (wxTreeCtrl::OnIdle)
494END_EVENT_TABLE()
495
496// -----------------------------------------------------------------------------
497// construction/destruction
498// -----------------------------------------------------------------------------
499
500void wxTreeCtrl::Init()
501{
502 m_current =
503 m_key_current =
504 m_anchor = (wxGenericTreeItem *) NULL;
505 m_hasFocus = FALSE;
506 m_dirty = FALSE;
507
508 m_xScroll = 0;
509 m_yScroll = 0;
510 m_lineHeight = 10;
511 m_indent = 15;
512 m_spacing = 18;
513
514 m_hilightBrush = new wxBrush
515 (
516 wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT),
517 wxSOLID
518 );
519
520 m_imageListNormal =
521 m_imageListState = (wxImageList *) NULL;
522
523 m_dragCount = 0;
524
525 m_renameTimer = new wxTreeRenameTimer( this );
526}
527
528bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id,
529 const wxPoint& pos, const wxSize& size,
530 long style,
531#if wxUSE_VALIDATORS
532# if defined(__VISAGECPP__)
533 const wxValidator *validator,
534# else
535 const wxValidator &validator,
536# endif
537#endif
538 const wxString& name )
539{
540 Init();
541
542 wxScrolledWindow::Create( parent, id, pos, size, style|wxHSCROLL|wxVSCROLL, name );
543
544#if wxUSE_VALIDATORS
545 SetValidator( validator );
546#endif
547
548 SetBackgroundColour( *wxWHITE );
549// m_dottedPen = wxPen( "grey", 0, wxDOT );
550 m_dottedPen = wxPen( "grey", 0, 0 );
551
552 return TRUE;
553}
554
555wxTreeCtrl::~wxTreeCtrl()
556{
557 wxDELETE( m_hilightBrush );
558
559 DeleteAllItems();
560
561 delete m_renameTimer;
562}
563
564// -----------------------------------------------------------------------------
565// accessors
566// -----------------------------------------------------------------------------
567
568size_t wxTreeCtrl::GetCount() const
569{
570 return m_anchor == NULL ? 0u : m_anchor->GetChildrenCount();
571}
572
573void wxTreeCtrl::SetIndent(unsigned int indent)
574{
575 m_indent = indent;
576 m_dirty = TRUE;
577 Refresh();
578}
579
580void wxTreeCtrl::SetSpacing(unsigned int spacing)
581{
582 m_spacing = spacing;
583 m_dirty = TRUE;
584 Refresh();
585}
586
587size_t wxTreeCtrl::GetChildrenCount(const wxTreeItemId& item, bool recursively)
588{
589 wxCHECK_MSG( item.IsOk(), 0u, wxT("invalid tree item") );
590
591 return item.m_pItem->GetChildrenCount(recursively);
592}
593
594// -----------------------------------------------------------------------------
595// functions to work with tree items
596// -----------------------------------------------------------------------------
597
598wxString wxTreeCtrl::GetItemText(const wxTreeItemId& item) const
599{
600 wxCHECK_MSG( item.IsOk(), wxT(""), wxT("invalid tree item") );
601
602 return item.m_pItem->GetText();
603}
604
605int wxTreeCtrl::GetItemImage(const wxTreeItemId& item,
606 wxTreeItemIcon which) const
607{
608 wxCHECK_MSG( item.IsOk(), -1, wxT("invalid tree item") );
609
610 return item.m_pItem->GetImage(which);
611}
612
613wxTreeItemData *wxTreeCtrl::GetItemData(const wxTreeItemId& item) const
614{
615 wxCHECK_MSG( item.IsOk(), NULL, wxT("invalid tree item") );
616
617 return item.m_pItem->GetData();
618}
619
620void wxTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text)
621{
622 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
623
624 wxClientDC dc(this);
625 wxGenericTreeItem *pItem = item.m_pItem;
626 pItem->SetText(text);
627 CalculateSize(pItem, dc);
628 RefreshLine(pItem);
629}
630
631void wxTreeCtrl::SetItemImage(const wxTreeItemId& item,
632 int image,
633 wxTreeItemIcon which)
634{
635 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
636
637 wxGenericTreeItem *pItem = item.m_pItem;
638 pItem->SetImage(image, which);
639
640 wxClientDC dc(this);
641 CalculateSize(pItem, dc);
642 RefreshLine(pItem);
643}
644
645void wxTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data)
646{
647 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
648
649 item.m_pItem->SetData(data);
650}
651
652void wxTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has)
653{
654 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
655
656 wxGenericTreeItem *pItem = item.m_pItem;
657 pItem->SetHasPlus(has);
658 RefreshLine(pItem);
659}
660
661void wxTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold)
662{
663 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
664
665 // avoid redrawing the tree if no real change
666 wxGenericTreeItem *pItem = item.m_pItem;
667 if ( pItem->IsBold() != bold )
668 {
669 pItem->SetBold(bold);
670 RefreshLine(pItem);
671 }
672}
673
674// -----------------------------------------------------------------------------
675// item status inquiries
676// -----------------------------------------------------------------------------
677
678bool wxTreeCtrl::IsVisible(const wxTreeItemId& WXUNUSED(item)) const
679{
680 wxFAIL_MSG(wxT("not implemented"));
681
682 return TRUE;
683}
684
685bool wxTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const
686{
687 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
688
689 return !item.m_pItem->GetChildren().IsEmpty();
690}
691
692bool wxTreeCtrl::IsExpanded(const wxTreeItemId& item) const
693{
694 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
695
696 return item.m_pItem->IsExpanded();
697}
698
699bool wxTreeCtrl::IsSelected(const wxTreeItemId& item) const
700{
701 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
702
703 return item.m_pItem->IsSelected();
704}
705
706bool wxTreeCtrl::IsBold(const wxTreeItemId& item) const
707{
708 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
709
710 return item.m_pItem->IsBold();
711}
712
713// -----------------------------------------------------------------------------
714// navigation
715// -----------------------------------------------------------------------------
716
717wxTreeItemId wxTreeCtrl::GetParent(const wxTreeItemId& item) const
718{
719 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
720
721 return item.m_pItem->GetParent();
722}
723
724wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item, long& cookie) const
725{
726 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
727
728 cookie = 0;
729 return GetNextChild(item, cookie);
730}
731
732wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& item, long& cookie) const
733{
734 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
735
736 wxArrayGenericTreeItems& children = item.m_pItem->GetChildren();
737 if ( (size_t)cookie < children.Count() )
738 {
739 return children.Item(cookie++);
740 }
741 else
742 {
743 // there are no more of them
744 return wxTreeItemId();
745 }
746}
747
748wxTreeItemId wxTreeCtrl::GetLastChild(const wxTreeItemId& item) const
749{
750 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
751
752 wxArrayGenericTreeItems& children = item.m_pItem->GetChildren();
753 return (children.IsEmpty() ? wxTreeItemId() : wxTreeItemId(children.Last()));
754}
755
756wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
757{
758 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
759
760 wxGenericTreeItem *i = item.m_pItem;
761 wxGenericTreeItem *parent = i->GetParent();
762 if ( parent == NULL )
763 {
764 // root item doesn't have any siblings
765 return wxTreeItemId();
766 }
767
768 wxArrayGenericTreeItems& siblings = parent->GetChildren();
769 int index = siblings.Index(i);
770 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
771
772 size_t n = (size_t)(index + 1);
773 return n == siblings.Count() ? wxTreeItemId() : wxTreeItemId(siblings[n]);
774}
775
776wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
777{
778 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
779
780 wxGenericTreeItem *i = item.m_pItem;
781 wxGenericTreeItem *parent = i->GetParent();
782 if ( parent == NULL )
783 {
784 // root item doesn't have any siblings
785 return wxTreeItemId();
786 }
787
788 wxArrayGenericTreeItems& siblings = parent->GetChildren();
789 int index = siblings.Index(i);
790 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
791
792 return index == 0 ? wxTreeItemId()
793 : wxTreeItemId(siblings[(size_t)(index - 1)]);
794}
795
796wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const
797{
798 wxFAIL_MSG(wxT("not implemented"));
799
800 return wxTreeItemId();
801}
802
803wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
804{
805 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
806
807 wxFAIL_MSG(wxT("not implemented"));
808
809 return wxTreeItemId();
810}
811
812wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
813{
814 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
815
816 wxFAIL_MSG(wxT("not implemented"));
817
818 return wxTreeItemId();
819}
820
821// -----------------------------------------------------------------------------
822// operations
823// -----------------------------------------------------------------------------
824
825wxTreeItemId wxTreeCtrl::DoInsertItem(const wxTreeItemId& parentId,
826 size_t previous,
827 const wxString& text,
828 int image, int selImage,
829 wxTreeItemData *data)
830{
831 wxGenericTreeItem *parent = parentId.m_pItem;
832 if ( !parent )
833 {
834 // should we give a warning here?
835 return AddRoot(text, image, selImage, data);
836 }
837
838 wxClientDC dc(this);
839 wxGenericTreeItem *item = new wxGenericTreeItem(parent,
840 text, dc,
841 image, selImage,
842 data);
843
844 if ( data != NULL )
845 {
846 data->m_pItem = item;
847 }
848
849 parent->Insert( item, previous );
850
851 m_dirty = TRUE;
852
853 return item;
854}
855
856wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text,
857 int image, int selImage,
858 wxTreeItemData *data)
859{
860 wxCHECK_MSG( !m_anchor, wxTreeItemId(), wxT("tree can have only one root") );
861
862 wxClientDC dc(this);
863 m_anchor = new wxGenericTreeItem((wxGenericTreeItem *)NULL, text, dc,
864 image, selImage, data);
865 if ( data != NULL )
866 {
867 data->m_pItem = m_anchor;
868 }
869
870 Refresh();
871 AdjustMyScrollbars();
872
873 return m_anchor;
874}
875
876wxTreeItemId wxTreeCtrl::PrependItem(const wxTreeItemId& parent,
877 const wxString& text,
878 int image, int selImage,
879 wxTreeItemData *data)
880{
881 return DoInsertItem(parent, 0u, text, image, selImage, data);
882}
883
884wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parentId,
885 const wxTreeItemId& idPrevious,
886 const wxString& text,
887 int image, int selImage,
888 wxTreeItemData *data)
889{
890 wxGenericTreeItem *parent = parentId.m_pItem;
891 if ( !parent )
892 {
893 // should we give a warning here?
894 return AddRoot(text, image, selImage, data);
895 }
896
897 int index = parent->GetChildren().Index(idPrevious.m_pItem);
898 wxASSERT_MSG( index != wxNOT_FOUND,
899 wxT("previous item in wxTreeCtrl::InsertItem() is not a sibling") );
900 return DoInsertItem(parentId, (size_t)++index, text, image, selImage, data);
901}
902
903wxTreeItemId wxTreeCtrl::AppendItem(const wxTreeItemId& parentId,
904 const wxString& text,
905 int image, int selImage,
906 wxTreeItemData *data)
907{
908 wxGenericTreeItem *parent = parentId.m_pItem;
909 if ( !parent )
910 {
911 // should we give a warning here?
912 return AddRoot(text, image, selImage, data);
913 }
914
915 return DoInsertItem(parent, parent->GetChildren().Count(), text,
916 image, selImage, data);
917}
918
919void wxTreeCtrl::SendDeleteEvent(wxGenericTreeItem *item)
920{
921 wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, GetId() );
922 event.m_item = item;
923 event.SetEventObject( this );
924 ProcessEvent( event );
925}
926
927void wxTreeCtrl::DeleteChildren(const wxTreeItemId& itemId)
928{
929 wxGenericTreeItem *item = itemId.m_pItem;
930 item->DeleteChildren(this);
931
932 m_dirty = TRUE;
933}
934
935void wxTreeCtrl::Delete(const wxTreeItemId& itemId)
936{
937 wxGenericTreeItem *item = itemId.m_pItem;
938 wxGenericTreeItem *parent = item->GetParent();
939
940 if ( parent )
941 {
942 parent->GetChildren().Remove(item);
943 }
944
945 item->DeleteChildren(this);
946 SendDeleteEvent(item);
947 delete item;
948
949 m_dirty = TRUE;
950}
951
952void wxTreeCtrl::DeleteAllItems()
953{
954 if ( m_anchor )
955 {
956 m_anchor->DeleteChildren(this);
957 delete m_anchor;
958
959 m_anchor = NULL;
960
961 m_dirty = TRUE;
962 }
963}
964
965void wxTreeCtrl::Expand(const wxTreeItemId& itemId)
966{
967 wxGenericTreeItem *item = itemId.m_pItem;
968
969 if ( !item->HasPlus() )
970 return;
971
972 if ( item->IsExpanded() )
973 return;
974
975 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, GetId() );
976 event.m_item = item;
977 event.SetEventObject( this );
978
979// if ( ProcessEvent( event ) && event.m_code ) TODO: Was this a typo ?
980 if ( ProcessEvent( event ) && !event.IsAllowed() )
981 {
982 // cancelled by program
983 return;
984 }
985
986 item->Expand();
987 CalculatePositions();
988
989 RefreshSubtree(item);
990
991 event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED);
992 ProcessEvent( event );
993}
994
995void wxTreeCtrl::Collapse(const wxTreeItemId& itemId)
996{
997 wxGenericTreeItem *item = itemId.m_pItem;
998
999 if ( !item->IsExpanded() )
1000 return;
1001
1002 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, GetId() );
1003 event.m_item = item;
1004 event.SetEventObject( this );
1005 if ( ProcessEvent( event ) && !event.IsAllowed() )
1006 {
1007 // cancelled by program
1008 return;
1009 }
1010
1011 item->Collapse();
1012
1013 wxArrayGenericTreeItems& children = item->GetChildren();
1014 size_t count = children.Count();
1015 for ( size_t n = 0; n < count; n++ )
1016 {
1017 Collapse(children[n]);
1018 }
1019
1020 CalculatePositions();
1021
1022 RefreshSubtree(item);
1023
1024 event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED);
1025 ProcessEvent( event );
1026}
1027
1028void wxTreeCtrl::CollapseAndReset(const wxTreeItemId& item)
1029{
1030 Collapse(item);
1031 DeleteChildren(item);
1032}
1033
1034void wxTreeCtrl::Toggle(const wxTreeItemId& itemId)
1035{
1036 wxGenericTreeItem *item = itemId.m_pItem;
1037
1038 if ( item->IsExpanded() )
1039 Collapse(itemId);
1040 else
1041 Expand(itemId);
1042}
1043
1044void wxTreeCtrl::Unselect()
1045{
1046 if ( m_current )
1047 {
1048 m_current->SetHilight( FALSE );
1049 RefreshLine( m_current );
1050 }
1051}
1052
1053void wxTreeCtrl::UnselectAllChildren(wxGenericTreeItem *item)
1054{
1055 item->SetHilight(FALSE);
1056 RefreshLine(item);
1057
1058 if (item->HasChildren())
1059 {
1060 wxArrayGenericTreeItems& children = item->GetChildren();
1061 size_t count = children.Count();
1062 for ( size_t n = 0; n < count; ++n )
1063 UnselectAllChildren(children[n]);
1064 }
1065}
1066
1067void wxTreeCtrl::UnselectAll()
1068{
1069 UnselectAllChildren(GetRootItem().m_pItem);
1070}
1071
1072// Recursive function !
1073// To stop we must have crt_item<last_item
1074// Algorithm :
1075// Tag all next children, when no more children,
1076// Move to parent (not to tag)
1077// Keep going... if we found last_item, we stop.
1078bool wxTreeCtrl::TagNextChildren(wxGenericTreeItem *crt_item, wxGenericTreeItem *last_item, bool select)
1079{
1080 wxGenericTreeItem *parent = crt_item->GetParent();
1081
1082 if ( parent == NULL ) // This is root item
1083 return TagAllChildrenUntilLast(crt_item, last_item, select);
1084
1085 wxArrayGenericTreeItems& children = parent->GetChildren();
1086 int index = children.Index(crt_item);
1087 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
1088
1089 size_t count = children.Count();
1090 for (size_t n=(size_t)(index+1); n<count; ++n)
1091 if (TagAllChildrenUntilLast(children[n], last_item, select)) return TRUE;
1092
1093 return TagNextChildren(parent, last_item, select);
1094}
1095
1096bool wxTreeCtrl::TagAllChildrenUntilLast(wxGenericTreeItem *crt_item, wxGenericTreeItem *last_item, bool select)
1097{
1098 crt_item->SetHilight(select);
1099 RefreshLine(crt_item);
1100
1101 if (crt_item==last_item) return TRUE;
1102
1103 if (crt_item->HasChildren())
1104 {
1105 wxArrayGenericTreeItems& children = crt_item->GetChildren();
1106 size_t count = children.Count();
1107 for ( size_t n = 0; n < count; ++n )
1108 if (TagAllChildrenUntilLast(children[n], last_item, select)) return TRUE;
1109 }
1110
1111 return FALSE;
1112}
1113
1114void wxTreeCtrl::SelectItemRange(wxGenericTreeItem *item1, wxGenericTreeItem *item2)
1115{
1116 // item2 is not necessary after item1
1117 wxGenericTreeItem *first=NULL, *last=NULL;
1118
1119 // choice first' and 'last' between item1 and item2
1120 if (item1->GetY()<item2->GetY())
1121 {
1122 first=item1;
1123 last=item2;
1124 }
1125 else
1126 {
1127 first=item2;
1128 last=item1;
1129 }
1130
1131 bool select = m_current->IsSelected();
1132
1133 if ( TagAllChildrenUntilLast(first,last,select) )
1134 return;
1135
1136 TagNextChildren(first,last,select);
1137}
1138
1139void wxTreeCtrl::SelectItem(const wxTreeItemId& itemId,
1140 bool unselect_others,
1141 bool extended_select)
1142{
1143 wxCHECK_RET( itemId.IsOk(), wxT("invalid tree item") );
1144
1145 bool is_single=!(GetWindowStyleFlag() & wxTR_MULTIPLE);
1146 wxGenericTreeItem *item = itemId.m_pItem;
1147
1148 //wxCHECK_RET( ( (!unselect_others) && is_single),
1149 // wxT("this is a single selection tree") );
1150
1151 // to keep going anyhow !!!
1152 if (is_single)
1153 {
1154 if (item->IsSelected())
1155 return; // nothing to do
1156 unselect_others = TRUE;
1157 extended_select = FALSE;
1158 }
1159 else if ( unselect_others && item->IsSelected() )
1160 {
1161 // selection change if there is more than one item currently selected
1162 wxArrayTreeItemIds selected_items;
1163 if ( GetSelections(selected_items) == 1 )
1164 return;
1165 }
1166
1167 wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGING, GetId() );
1168 event.m_item = item;
1169 event.m_itemOld = m_current;
1170 event.SetEventObject( this );
1171 // TODO : Here we don't send any selection mode yet !
1172
1173 if ( GetEventHandler()->ProcessEvent( event ) && !event.IsAllowed() )
1174 return;
1175
1176 // ctrl press
1177 if (unselect_others)
1178 {
1179 if (is_single) Unselect(); // to speed up thing
1180 else UnselectAll();
1181 }
1182
1183 // shift press
1184 if (extended_select)
1185 {
1186 if (m_current == NULL) m_current=m_key_current=GetRootItem().m_pItem;
1187 // don't change the mark (m_current)
1188 SelectItemRange(m_current, item);
1189 }
1190 else
1191 {
1192 bool select=TRUE; // the default
1193
1194 // Check if we need to toggle hilight (ctrl mode)
1195 if (!unselect_others)
1196 select=!item->IsSelected();
1197
1198 m_current = m_key_current = item;
1199 m_current->SetHilight(select);
1200 RefreshLine( m_current );
1201 }
1202
1203 event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGED);
1204 GetEventHandler()->ProcessEvent( event );
1205}
1206
1207void wxTreeCtrl::FillArray(wxGenericTreeItem *item,
1208 wxArrayTreeItemIds &array) const
1209{
1210 if ( item->IsSelected() )
1211 array.Add(wxTreeItemId(item));
1212
1213 if ( item->HasChildren() )
1214 {
1215 wxArrayGenericTreeItems& children = item->GetChildren();
1216 size_t count = children.GetCount();
1217 for ( size_t n = 0; n < count; ++n )
1218 FillArray(children[n],array);
1219 }
1220}
1221
1222size_t wxTreeCtrl::GetSelections(wxArrayTreeItemIds &array) const
1223{
1224 array.Empty();
1225 FillArray(GetRootItem().m_pItem, array);
1226
1227 return array.Count();
1228}
1229
1230void wxTreeCtrl::EnsureVisible(const wxTreeItemId& item)
1231{
1232 if (!item.IsOk()) return;
1233
1234 wxGenericTreeItem *gitem = item.m_pItem;
1235
1236 // first expand all parent branches
1237 wxGenericTreeItem *parent = gitem->GetParent();
1238 while ( parent )
1239 {
1240 Expand(parent);
1241 parent = parent->GetParent();
1242 }
1243
1244 //if (parent) CalculatePositions();
1245
1246 ScrollTo(item);
1247}
1248
1249void wxTreeCtrl::ScrollTo(const wxTreeItemId &item)
1250{
1251 if (!item.IsOk()) return;
1252
1253 // We have to call this here because the label in
1254 // question might just have been added and no screen
1255 // update taken place.
1256 if (m_dirty) wxYield();
1257
1258 wxGenericTreeItem *gitem = item.m_pItem;
1259
1260 // now scroll to the item
1261 int item_y = gitem->GetY();
1262
1263 int start_x = 0;
1264 int start_y = 0;
1265 ViewStart( &start_x, &start_y );
1266 start_y *= PIXELS_PER_UNIT;
1267
1268 int client_h = 0;
1269 int client_w = 0;
1270 GetClientSize( &client_w, &client_h );
1271
1272 if (item_y < start_y+3)
1273 {
1274 // going down
1275 int x = 0;
1276 int y = 0;
1277 m_anchor->GetSize( x, y, this );
1278 y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1279 int x_pos = GetScrollPos( wxHORIZONTAL );
1280 // Item should appear at top
1281 SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, item_y/PIXELS_PER_UNIT );
1282 }
1283 else if (item_y+GetLineHeight(gitem) > start_y+client_h)
1284 {
1285 // going up
1286 int x = 0;
1287 int y = 0;
1288 m_anchor->GetSize( x, y, this );
1289 y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1290 item_y += PIXELS_PER_UNIT+2;
1291 int x_pos = GetScrollPos( wxHORIZONTAL );
1292 // Item should appear at bottom
1293 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 );
1294 }
1295}
1296
1297// FIXME: tree sorting functions are not reentrant and not MT-safe!
1298static wxTreeCtrl *s_treeBeingSorted = NULL;
1299
1300static int LINKAGEMODE tree_ctrl_compare_func(wxGenericTreeItem **item1,
1301 wxGenericTreeItem **item2)
1302{
1303 wxCHECK_MSG( s_treeBeingSorted, 0, wxT("bug in wxTreeCtrl::SortChildren()") );
1304
1305 return s_treeBeingSorted->OnCompareItems(*item1, *item2);
1306}
1307
1308int wxTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
1309 const wxTreeItemId& item2)
1310{
1311 return wxStrcmp(GetItemText(item1), GetItemText(item2));
1312}
1313
1314void wxTreeCtrl::SortChildren(const wxTreeItemId& itemId)
1315{
1316 wxCHECK_RET( itemId.IsOk(), wxT("invalid tree item") );
1317
1318 wxGenericTreeItem *item = itemId.m_pItem;
1319
1320 wxCHECK_RET( !s_treeBeingSorted,
1321 wxT("wxTreeCtrl::SortChildren is not reentrant") );
1322
1323 wxArrayGenericTreeItems& children = item->GetChildren();
1324 if ( children.Count() > 1 )
1325 {
1326 s_treeBeingSorted = this;
1327 children.Sort(tree_ctrl_compare_func);
1328 s_treeBeingSorted = NULL;
1329
1330 m_dirty = TRUE;
1331 }
1332 //else: don't make the tree dirty as nothing changed
1333}
1334
1335wxImageList *wxTreeCtrl::GetImageList() const
1336{
1337 return m_imageListNormal;
1338}
1339
1340wxImageList *wxTreeCtrl::GetStateImageList() const
1341{
1342 return m_imageListState;
1343}
1344
1345void wxTreeCtrl::SetImageList(wxImageList *imageList)
1346{
1347 m_imageListNormal = imageList;
1348
1349 // Calculate a m_lineHeight value from the image sizes.
1350 // May be toggle off. Then wxTreeCtrl will spread when
1351 // necessary (which might look ugly).
1352#if 1
1353 wxClientDC dc(this);
1354 m_lineHeight = (int)(dc.GetCharHeight() + 4);
1355 int
1356 width = 0,
1357 height = 0,
1358 n = m_imageListNormal->GetImageCount();
1359 for(int i = 0; i < n ; i++)
1360 {
1361 m_imageListNormal->GetSize(i, width, height);
1362 if(height > m_lineHeight) m_lineHeight = height;
1363 }
1364
1365 if (m_lineHeight<40) m_lineHeight+=4; // at least 4 pixels (odd such that a line can be drawn in between)
1366 else m_lineHeight+=m_lineHeight/10; // otherwise 10% extra spacing
1367
1368#endif
1369}
1370
1371void wxTreeCtrl::SetStateImageList(wxImageList *imageList)
1372{
1373 m_imageListState = imageList;
1374}
1375
1376// -----------------------------------------------------------------------------
1377// helpers
1378// -----------------------------------------------------------------------------
1379
1380void wxTreeCtrl::AdjustMyScrollbars()
1381{
1382 if (m_anchor)
1383 {
1384 int x = 0;
1385 int y = 0;
1386 m_anchor->GetSize( x, y, this );
1387 //y += GetLineHeight(m_anchor);
1388 y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1389 int x_pos = GetScrollPos( wxHORIZONTAL );
1390 int y_pos = GetScrollPos( wxVERTICAL );
1391 SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, y_pos );
1392 }
1393 else
1394 {
1395 SetScrollbars( 0, 0, 0, 0 );
1396 }
1397}
1398
1399int wxTreeCtrl::GetLineHeight(wxGenericTreeItem *item) const
1400{
1401 if (GetWindowStyleFlag() & wxTR_HAS_VARIABLE_ROW_HEIGHT)
1402 return item->GetHeight();
1403 else
1404 return m_lineHeight;
1405}
1406
1407void wxTreeCtrl::PaintItem(wxGenericTreeItem *item, wxDC& dc)
1408{
1409 // render bold items in bold
1410 wxFont fontOld;
1411 wxFont fontNew;
1412
1413 if (item->IsBold())
1414 {
1415 fontOld = dc.GetFont();
1416 if (fontOld.Ok())
1417 {
1418 // VZ: is there any better way to make a bold variant of old font?
1419 fontNew = wxFont( fontOld.GetPointSize(),
1420 fontOld.GetFamily(),
1421 fontOld.GetStyle(),
1422 wxBOLD,
1423 fontOld.GetUnderlined());
1424 dc.SetFont(fontNew);
1425 }
1426 else
1427 {
1428 wxFAIL_MSG(wxT("wxDC::GetFont() failed!"));
1429 }
1430 }
1431
1432 long text_w = 0;
1433 long text_h = 0;
1434 dc.GetTextExtent( item->GetText(), &text_w, &text_h );
1435
1436 int image_h = 0;
1437 int image_w = 0;
1438 int image = item->GetCurrentImage();
1439 if ( image != NO_IMAGE )
1440 {
1441 m_imageListNormal->GetSize( image, image_w, image_h );
1442 image_w += 4;
1443 }
1444
1445 int total_h = GetLineHeight(item);
1446
1447 dc.DrawRectangle( item->GetX()-2, item->GetY(), item->GetWidth()+2, total_h );
1448
1449 if ( image != NO_IMAGE )
1450 {
1451 dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
1452 m_imageListNormal->Draw( image, dc,
1453 item->GetX(),
1454 item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
1455 wxIMAGELIST_DRAW_TRANSPARENT );
1456 dc.DestroyClippingRegion();
1457 }
1458
1459 dc.SetBackgroundMode(wxTRANSPARENT);
1460 dc.DrawText( item->GetText(), image_w + item->GetX(), item->GetY()
1461 + ((total_h > text_h) ? (total_h - text_h)/2 : 0));
1462
1463 // restore normal font for bold items
1464 if (fontOld.Ok())
1465 {
1466 dc.SetFont( fontOld);
1467 }
1468}
1469
1470// Now y stands for the top of the item, whereas it used to stand for middle !
1471void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
1472{
1473 int horizX = level*m_indent;
1474
1475 item->SetX( horizX+m_indent+m_spacing );
1476 item->SetY( y );
1477
1478 int oldY = y;
1479 y+=GetLineHeight(item)/2;
1480
1481 item->SetCross( horizX+m_indent, y );
1482
1483 int exposed_x = dc.LogicalToDeviceX( 0 );
1484 int exposed_y = dc.LogicalToDeviceY( item->GetY() );
1485
1486 if (IsExposed( exposed_x, exposed_y, 10000, GetLineHeight(item) )) // 10000 = very much
1487 {
1488 int startX = horizX;
1489 int endX = horizX + (m_indent-5);
1490
1491// if (!item->HasChildren()) endX += (m_indent+5);
1492 if (!item->HasChildren()) endX += 20;
1493
1494 dc.DrawLine( startX, y, endX, y );
1495
1496 if (item->HasPlus())
1497 {
1498 dc.DrawLine( horizX+(m_indent+5), y, horizX+(m_indent+15), y );
1499 dc.SetPen( *wxGREY_PEN );
1500 dc.SetBrush( *wxWHITE_BRUSH );
1501 dc.DrawRectangle( horizX+(m_indent-5), y-4, 11, 9 );
1502
1503 dc.SetPen( *wxBLACK_PEN );
1504 dc.DrawLine( horizX+(m_indent-2), y, horizX+(m_indent+3), y );
1505 if (!item->IsExpanded())
1506 dc.DrawLine( horizX+m_indent, y-2, horizX+m_indent, y+3 );
1507
1508 dc.SetPen( m_dottedPen );
1509 }
1510
1511 if (item->IsSelected())
1512 {
1513 dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) );
1514
1515 dc.SetBrush( *m_hilightBrush );
1516
1517 if (m_hasFocus)
1518 dc.SetPen( *wxBLACK_PEN );
1519 else
1520 dc.SetPen( *wxTRANSPARENT_PEN );
1521
1522 PaintItem(item, dc);
1523
1524 dc.SetPen( m_dottedPen );
1525 dc.SetTextForeground( *wxBLACK );
1526 dc.SetBrush( *wxWHITE_BRUSH );
1527 }
1528 else
1529 {
1530 dc.SetBrush( *wxWHITE_BRUSH );
1531 dc.SetPen( *wxTRANSPARENT_PEN );
1532
1533 PaintItem(item, dc);
1534
1535 dc.SetPen( m_dottedPen );
1536 }
1537 }
1538
1539 y = oldY+GetLineHeight(item);
1540
1541 if (item->IsExpanded())
1542 {
1543 oldY+=GetLineHeight(item)/2;
1544 int semiOldY=y; // (=y) for stupid compilator
1545
1546 wxArrayGenericTreeItems& children = item->GetChildren();
1547 size_t n, count = children.Count();
1548 for ( n = 0; n < count; ++n )
1549 {
1550 semiOldY=y;
1551 PaintLevel( children[n], dc, level+1, y );
1552 }
1553
1554 // it may happen that the item is expanded but has no items (when you
1555 // delete all its children for example) - don't draw the vertical line
1556 // in this case
1557 if (count > 0)
1558 {
1559 semiOldY+=GetLineHeight(children[--n])/2;
1560 dc.DrawLine( horizX+m_indent, oldY+5, horizX+m_indent, semiOldY );
1561 }
1562 }
1563}
1564
1565void wxTreeCtrl::DrawBorder(wxTreeItemId &item)
1566{
1567 if (!item) return;
1568
1569 wxGenericTreeItem *i=item.m_pItem;
1570
1571 wxClientDC dc(this);
1572 PrepareDC( dc );
1573 dc.SetLogicalFunction(wxINVERT);
1574
1575 int w,h,x;
1576 ViewStart(&x,&h); // we only need x
1577 GetClientSize(&w,&h); // we only need w
1578
1579 h=GetLineHeight(i)+1;
1580 // 2 white column at border
1581 dc.DrawRectangle( PIXELS_PER_UNIT*x+2, i->GetY()-1, w-6, h);
1582}
1583
1584void wxTreeCtrl::DrawLine(wxTreeItemId &item, bool below)
1585{
1586 if (!item) return;
1587
1588 wxGenericTreeItem *i=item.m_pItem;
1589
1590 wxClientDC dc(this);
1591 PrepareDC( dc );
1592 dc.SetLogicalFunction(wxINVERT);
1593
1594 int w,h,y;
1595 GetSize(&w,&h);
1596
1597 if (below) y=i->GetY()+GetLineHeight(i)-1;
1598 else y=i->GetY();
1599
1600 dc.DrawLine( 0, y, w, y);
1601}
1602
1603// -----------------------------------------------------------------------------
1604// wxWindows callbacks
1605// -----------------------------------------------------------------------------
1606
1607void wxTreeCtrl::OnPaint( wxPaintEvent &WXUNUSED(event) )
1608{
1609 if ( !m_anchor)
1610 return;
1611
1612 wxPaintDC dc(this);
1613 PrepareDC( dc );
1614
1615 dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
1616
1617 dc.SetPen( m_dottedPen );
1618 //if(GetImageList() == NULL)
1619 // m_lineHeight = (int)(dc.GetCharHeight() + 4);
1620
1621 int y = 2;
1622 PaintLevel( m_anchor, dc, 0, y );
1623}
1624
1625void wxTreeCtrl::OnSetFocus( wxFocusEvent &WXUNUSED(event) )
1626{
1627 m_hasFocus = TRUE;
1628
1629 if (m_current) RefreshLine( m_current );
1630}
1631
1632void wxTreeCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) )
1633{
1634 m_hasFocus = FALSE;
1635
1636 if (m_current) RefreshLine( m_current );
1637}
1638
1639void wxTreeCtrl::OnChar( wxKeyEvent &event )
1640{
1641 wxTreeEvent te( wxEVT_COMMAND_TREE_KEY_DOWN, GetId() );
1642 te.m_code = event.KeyCode();
1643 te.SetEventObject( this );
1644 GetEventHandler()->ProcessEvent( te );
1645
1646 if ( (m_current == 0) || (m_key_current == 0) )
1647 {
1648 event.Skip();
1649 return;
1650 }
1651
1652 bool is_multiple=(GetWindowStyleFlag() & wxTR_MULTIPLE);
1653 bool extended_select=(event.ShiftDown() && is_multiple);
1654 bool unselect_others=!(extended_select || (event.ControlDown() && is_multiple));
1655
1656 switch (event.KeyCode())
1657 {
1658 case '+':
1659 case WXK_ADD:
1660 if (m_current->HasPlus() && !IsExpanded(m_current))
1661 {
1662 Expand(m_current);
1663 }
1664 break;
1665
1666 case '-':
1667 case WXK_SUBTRACT:
1668 if (IsExpanded(m_current))
1669 {
1670 Collapse(m_current);
1671 }
1672 break;
1673
1674 case '*':
1675 case WXK_MULTIPLY:
1676 Toggle(m_current);
1677 break;
1678
1679 case ' ':
1680 case WXK_RETURN:
1681 {
1682 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
1683 event.m_item = m_current;
1684 event.m_code = 0;
1685 event.SetEventObject( this );
1686 GetEventHandler()->ProcessEvent( event );
1687 }
1688 break;
1689
1690 // up goes to the previous sibling or to the last of its children if
1691 // it's expanded
1692 case WXK_UP:
1693 {
1694 wxTreeItemId prev = GetPrevSibling( m_key_current );
1695 if (!prev)
1696 {
1697 prev = GetParent( m_key_current );
1698 if (prev)
1699 {
1700 long cockie = 0;
1701 wxTreeItemId current = m_key_current;
1702 if (current == GetFirstChild( prev, cockie ))
1703 {
1704 // otherwise we return to where we came from
1705 SelectItem( prev, unselect_others, extended_select );
1706 m_key_current=prev.m_pItem;
1707 EnsureVisible( prev );
1708 break;
1709 }
1710 }
1711 }
1712 if (prev)
1713 {
1714 while ( IsExpanded(prev) && HasChildren(prev) )
1715 {
1716 wxTreeItemId child = GetLastChild(prev);
1717 if ( child )
1718 {
1719 prev = child;
1720 }
1721 }
1722
1723 SelectItem( prev, unselect_others, extended_select );
1724 m_key_current=prev.m_pItem;
1725 EnsureVisible( prev );
1726 }
1727 }
1728 break;
1729
1730 // left arrow goes to the parent
1731 case WXK_LEFT:
1732 {
1733 wxTreeItemId prev = GetParent( m_current );
1734 if (prev)
1735 {
1736 EnsureVisible( prev );
1737 SelectItem( prev, unselect_others, extended_select );
1738 }
1739 }
1740 break;
1741
1742 case WXK_RIGHT:
1743 // this works the same as the down arrow except that we also expand the
1744 // item if it wasn't expanded yet
1745 Expand(m_current);
1746 // fall through
1747
1748 case WXK_DOWN:
1749 {
1750 if (IsExpanded(m_key_current) && HasChildren(m_key_current))
1751 {
1752 long cookie = 0;
1753 wxTreeItemId child = GetFirstChild( m_key_current, cookie );
1754 SelectItem( child, unselect_others, extended_select );
1755 m_key_current=child.m_pItem;
1756 EnsureVisible( child );
1757 }
1758 else
1759 {
1760 wxTreeItemId next = GetNextSibling( m_key_current );
1761// if (next == 0)
1762 if (!next)
1763 {
1764 wxTreeItemId current = m_key_current;
1765 while (current && !next)
1766 {
1767 current = GetParent( current );
1768 if (current) next = GetNextSibling( current );
1769 }
1770 }
1771// if (next != 0)
1772 if (next)
1773 {
1774 SelectItem( next, unselect_others, extended_select );
1775 m_key_current=next.m_pItem;
1776 EnsureVisible( next );
1777 }
1778 }
1779 }
1780 break;
1781
1782 // <End> selects the last visible tree item
1783 case WXK_END:
1784 {
1785 wxTreeItemId last = GetRootItem();
1786
1787 while ( last.IsOk() && IsExpanded(last) )
1788 {
1789 wxTreeItemId lastChild = GetLastChild(last);
1790
1791 // it may happen if the item was expanded but then all of
1792 // its children have been deleted - so IsExpanded() returned
1793 // TRUE, but GetLastChild() returned invalid item
1794 if ( !lastChild )
1795 break;
1796
1797 last = lastChild;
1798 }
1799
1800 if ( last.IsOk() )
1801 {
1802 EnsureVisible( last );
1803 SelectItem( last, unselect_others, extended_select );
1804 }
1805 }
1806 break;
1807
1808 // <Home> selects the root item
1809 case WXK_HOME:
1810 {
1811 wxTreeItemId prev = GetRootItem();
1812 if (prev)
1813 {
1814 EnsureVisible( prev );
1815 SelectItem( prev, unselect_others, extended_select );
1816 }
1817 }
1818 break;
1819
1820 default:
1821 event.Skip();
1822 }
1823}
1824
1825wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& flags)
1826{
1827 // We have to call this here because the label in
1828 // question might just have been added and no screen
1829 // update taken place.
1830 if (m_dirty) wxYield();
1831
1832 wxClientDC dc(this);
1833 PrepareDC(dc);
1834 long x = dc.DeviceToLogicalX( (long)point.x );
1835 long y = dc.DeviceToLogicalY( (long)point.y );
1836 int w, h;
1837 GetSize(&w, &h);
1838
1839 flags=0;
1840 if (point.x<0) flags|=wxTREE_HITTEST_TOLEFT;
1841 if (point.x>w) flags|=wxTREE_HITTEST_TORIGHT;
1842 if (point.y<0) flags|=wxTREE_HITTEST_ABOVE;
1843 if (point.y>h) flags|=wxTREE_HITTEST_BELOW;
1844
1845 return m_anchor->HitTest( wxPoint(x, y), this, flags);
1846}
1847
1848/* **** */
1849
1850void wxTreeCtrl::Edit( const wxTreeItemId& item )
1851{
1852 if (!item.IsOk()) return;
1853
1854 m_currentEdit = item.m_pItem;
1855
1856 wxTreeEvent te( wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT, GetId() );
1857 te.m_item = m_currentEdit;
1858 te.SetEventObject( this );
1859 GetEventHandler()->ProcessEvent( te );
1860
1861 if (!te.IsAllowed()) return;
1862
1863 // We have to call this here because the label in
1864 // question might just have been added and no screen
1865 // update taken place.
1866 if (m_dirty) wxYield();
1867
1868 wxString s = m_currentEdit->GetText();
1869 int x = m_currentEdit->GetX();
1870 int y = m_currentEdit->GetY();
1871 int w = m_currentEdit->GetWidth();
1872 int h = m_currentEdit->GetHeight();
1873
1874 int image_h = 0;
1875 int image_w = 0;
1876
1877 int image = m_currentEdit->GetCurrentImage();
1878 if ( image != NO_IMAGE )
1879 {
1880 m_imageListNormal->GetSize( image, image_w, image_h );
1881 image_w += 4;
1882 }
1883 x += image_w;
1884 w -= image_w + 4; // I don't know why +4 is needed
1885
1886 wxClientDC dc(this);
1887 PrepareDC( dc );
1888 x = dc.LogicalToDeviceX( x );
1889 y = dc.LogicalToDeviceY( y );
1890
1891 wxTreeTextCtrl *text = new wxTreeTextCtrl(
1892 this, -1, &m_renameAccept, &m_renameRes, this, s, wxPoint(x-4,y-4), wxSize(w+11,h+8) );
1893 text->SetFocus();
1894}
1895
1896void wxTreeCtrl::OnRenameTimer()
1897{
1898 Edit( m_current );
1899}
1900
1901void wxTreeCtrl::OnRenameAccept()
1902{
1903 wxTreeEvent le( wxEVT_COMMAND_TREE_END_LABEL_EDIT, GetId() );
1904 le.m_item = m_currentEdit;
1905 le.SetEventObject( this );
1906 le.m_label = m_renameRes;
1907 GetEventHandler()->ProcessEvent( le );
1908
1909 if (!le.IsAllowed()) return;
1910
1911 SetItemText( m_currentEdit, m_renameRes );
1912}
1913
1914void wxTreeCtrl::OnMouse( wxMouseEvent &event )
1915{
1916 if ( !(event.LeftUp() || event.RightDown() || event.LeftDClick() || event.Dragging()) ) return;
1917
1918 if ( !m_anchor ) return;
1919
1920 wxClientDC dc(this);
1921 PrepareDC(dc);
1922 long x = dc.DeviceToLogicalX( (long)event.GetX() );
1923 long y = dc.DeviceToLogicalY( (long)event.GetY() );
1924
1925 int flags=0;
1926 wxGenericTreeItem *item = m_anchor->HitTest( wxPoint(x,y), this, flags);
1927 bool onButton = flags & wxTREE_HITTEST_ONITEMBUTTON;
1928
1929 if (event.Dragging())
1930 {
1931 if (m_dragCount == 0)
1932 m_dragStart = wxPoint(x,y);
1933
1934 m_dragCount++;
1935
1936 if (m_dragCount != 3) return;
1937
1938 int command = wxEVT_COMMAND_TREE_BEGIN_DRAG;
1939 if (event.RightIsDown()) command = wxEVT_COMMAND_TREE_BEGIN_RDRAG;
1940
1941 wxTreeEvent nevent( command, GetId() );
1942 nevent.m_item = m_current;
1943 nevent.SetEventObject(this);
1944 GetEventHandler()->ProcessEvent(nevent);
1945 return;
1946 }
1947 else
1948 {
1949 m_dragCount = 0;
1950 }
1951
1952 if (item == NULL) return; /* we hit the blank area */
1953
1954 if (event.RightDown()) {
1955 wxTreeEvent nevent(wxEVT_COMMAND_TREE_ITEM_RIGHT_CLICK,GetId());
1956 nevent.m_item=item;
1957 nevent.m_code=0;
1958 nevent.SetEventObject(this);
1959 GetEventHandler()->ProcessEvent(nevent);
1960 return;
1961 }
1962
1963 if (event.LeftUp() && (item == m_current) &&
1964 (flags & wxTREE_HITTEST_ONITEMLABEL) &&
1965 HasFlag(wxTR_EDIT_LABELS) )
1966 {
1967 m_renameTimer->Start( 100, TRUE );
1968 return;
1969 }
1970
1971 bool is_multiple=(GetWindowStyleFlag() & wxTR_MULTIPLE);
1972 bool extended_select=(event.ShiftDown() && is_multiple);
1973 bool unselect_others=!(extended_select || (event.ControlDown() && is_multiple));
1974
1975 if (onButton)
1976 {
1977 Toggle( item );
1978 if (is_multiple)
1979 return;
1980 }
1981
1982 SelectItem(item, unselect_others, extended_select);
1983
1984 if (event.LeftDClick())
1985 {
1986 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
1987 event.m_item = item;
1988 event.m_code = 0;
1989 event.SetEventObject( this );
1990 GetEventHandler()->ProcessEvent( event );
1991 }
1992}
1993
1994void wxTreeCtrl::OnIdle( wxIdleEvent &WXUNUSED(event) )
1995{
1996 /* after all changes have been done to the tree control,
1997 * we actually redraw the tree when everything is over */
1998
1999 if (!m_dirty)
2000 return;
2001
2002 m_dirty = FALSE;
2003
2004 CalculatePositions();
2005 Refresh();
2006 AdjustMyScrollbars();
2007}
2008
2009void wxTreeCtrl::CalculateSize( wxGenericTreeItem *item, wxDC &dc )
2010{
2011 long text_w = 0;
2012 long text_h = 0;
2013
2014 wxFont fontOld;
2015 wxFont fontNew;
2016 if (item->IsBold())
2017 {
2018 fontOld = dc.GetFont();
2019 if (fontOld.Ok())
2020 {
2021 // VZ: is there any better way to make a bold variant of old font?
2022 fontNew = wxFont( fontOld.GetPointSize(),
2023 fontOld.GetFamily(),
2024 fontOld.GetStyle(),
2025 wxBOLD,
2026 fontOld.GetUnderlined());
2027 dc.SetFont(fontNew);
2028 }
2029 else
2030 {
2031 wxFAIL_MSG(wxT("wxDC::GetFont() failed!"));
2032 }
2033 }
2034
2035 dc.GetTextExtent( item->GetText(), &text_w, &text_h );
2036 text_h+=2;
2037
2038 // restore normal font for bold items
2039 if (fontOld.Ok())
2040 dc.SetFont( fontOld);
2041
2042 int image_h = 0;
2043 int image_w = 0;
2044 int image = item->GetCurrentImage();
2045 if ( image != NO_IMAGE )
2046 {
2047 m_imageListNormal->GetSize( image, image_w, image_h );
2048 image_w += 4;
2049 }
2050
2051 int total_h = (image_h > text_h) ? image_h : text_h;
2052
2053 if (total_h<40) total_h+=4; // at least 4 pixels
2054 else total_h+=total_h/10; // otherwise 10% extra spacing
2055
2056 item->SetHeight(total_h);
2057 if (total_h>m_lineHeight) m_lineHeight=total_h;
2058
2059 item->SetWidth(image_w+text_w+2);
2060}
2061
2062// -----------------------------------------------------------------------------
2063// for developper : y is now the top of the level
2064// not the middle of it !
2065void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
2066{
2067 int horizX = level*m_indent;
2068
2069 CalculateSize( item, dc );
2070
2071 // set its position
2072 item->SetX( horizX+m_indent+m_spacing );
2073 item->SetY( y );
2074 y+=GetLineHeight(item);
2075
2076 if ( !item->IsExpanded() )
2077 {
2078 // we dont need to calculate collapsed branches
2079 return;
2080 }
2081
2082 wxArrayGenericTreeItems& children = item->GetChildren();
2083 size_t n, count = children.Count();
2084 for (n = 0; n < count; ++n )
2085 CalculateLevel( children[n], dc, level+1, y ); // recurse
2086}
2087
2088void wxTreeCtrl::CalculatePositions()
2089{
2090 if ( !m_anchor ) return;
2091
2092 wxClientDC dc(this);
2093 PrepareDC( dc );
2094
2095 dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
2096
2097 dc.SetPen( m_dottedPen );
2098 //if(GetImageList() == NULL)
2099 // m_lineHeight = (int)(dc.GetCharHeight() + 4);
2100
2101 int y = 2;
2102 CalculateLevel( m_anchor, dc, 0, y ); // start recursion
2103}
2104
2105void wxTreeCtrl::RefreshSubtree(wxGenericTreeItem *item)
2106{
2107 wxClientDC dc(this);
2108 PrepareDC(dc);
2109
2110 int cw = 0;
2111 int ch = 0;
2112 GetClientSize( &cw, &ch );
2113
2114 wxRect rect;
2115 rect.x = dc.LogicalToDeviceX( 0 );
2116 rect.width = cw;
2117 rect.y = dc.LogicalToDeviceY( item->GetY() );
2118 rect.height = ch;
2119
2120 Refresh( TRUE, &rect );
2121
2122 AdjustMyScrollbars();
2123}
2124
2125void wxTreeCtrl::RefreshLine( wxGenericTreeItem *item )
2126{
2127 wxClientDC dc(this);
2128 PrepareDC( dc );
2129
2130 int cw = 0;
2131 int ch = 0;
2132 GetClientSize( &cw, &ch );
2133
2134 wxRect rect;
2135 rect.x = dc.LogicalToDeviceX( 0 );
2136 rect.y = dc.LogicalToDeviceY( item->GetY() );
2137 rect.width = cw;
2138 rect.height = GetLineHeight(item); //dc.GetCharHeight() + 6;
2139
2140 Refresh( TRUE, &rect );
2141}
2142
wxWidgets (Impactor)