]> git.saurik.com Git - wxWidgets.git/blame - src/generic/treectlg.cpp
Header change for smooth Aqua buttons in wxTreeCtrl.
[wxWidgets.git] / src / generic / treectlg.cpp
CommitLineData
c801d85f 1/////////////////////////////////////////////////////////////////////////////
941830cb 2// Name: treectlg.cpp
f135ff73 3// Purpose: generic tree control implementation
c801d85f
KB
4// Author: Robert Roebling
5// Created: 01/02/97
f135ff73 6// Modified: 22/10/98 - almost total rewrite, simpler interface (VZ)
389cdc7a 7// Id: $Id$
c801d85f 8// Copyright: (c) 1998 Robert Roebling, Julian Smart and Markus Holzem
29d87bba 9// Licence: wxWindows licence
c801d85f
KB
10/////////////////////////////////////////////////////////////////////////////
11
f135ff73
VZ
12// =============================================================================
13// declarations
14// =============================================================================
15
16// -----------------------------------------------------------------------------
17// headers
18// -----------------------------------------------------------------------------
19
c801d85f 20#ifdef __GNUG__
941830cb 21 #pragma implementation "treectlg.h"
c801d85f
KB
22#endif
23
1e6d9499
JS
24// For compilers that support precompilation, includes "wx.h".
25#include "wx/wxprec.h"
26
27#ifdef __BORLANDC__
1e6feb95 28 #pragma hdrstop
1e6d9499
JS
29#endif
30
1e6feb95
VZ
31#if wxUSE_TREECTRL
32
618a5e38 33#include "wx/treebase.h"
941830cb 34#include "wx/generic/treectlg.h"
618a5e38
RR
35#include "wx/timer.h"
36#include "wx/textctrl.h"
941830cb 37#include "wx/imaglist.h"
c801d85f 38#include "wx/settings.h"
f135ff73 39#include "wx/dcclient.h"
c801d85f 40
f135ff73
VZ
41// -----------------------------------------------------------------------------
42// array types
43// -----------------------------------------------------------------------------
c801d85f 44
1e6d9499
JS
45class WXDLLEXPORT wxGenericTreeItem;
46
8a985b84 47WX_DEFINE_EXPORTED_ARRAY(wxGenericTreeItem *, wxArrayGenericTreeItems);
941830cb 48//WX_DEFINE_OBJARRAY(wxArrayTreeItemIds);
c801d85f 49
8dc99046
VZ
50// ----------------------------------------------------------------------------
51// constants
52// ----------------------------------------------------------------------------
53
54static const int NO_IMAGE = -1;
55
3dbeaa52
VZ
56#define PIXELS_PER_UNIT 10
57
f135ff73
VZ
58// -----------------------------------------------------------------------------
59// private classes
60// -----------------------------------------------------------------------------
61
3dbeaa52
VZ
62// timer used for enabling in-place edit
63class WXDLLEXPORT wxTreeRenameTimer: public wxTimer
64{
65public:
941830cb 66 wxTreeRenameTimer( wxGenericTreeCtrl *owner );
3dbeaa52
VZ
67
68 void Notify();
69
70private:
941830cb 71 wxGenericTreeCtrl *m_owner;
3dbeaa52
VZ
72};
73
74// control used for in-place edit
75class WXDLLEXPORT wxTreeTextCtrl: public wxTextCtrl
76{
77public:
3dbeaa52
VZ
78 wxTreeTextCtrl( wxWindow *parent,
79 const wxWindowID id,
80 bool *accept,
81 wxString *res,
941830cb 82 wxGenericTreeCtrl *owner,
3dbeaa52
VZ
83 const wxString &value = wxEmptyString,
84 const wxPoint &pos = wxDefaultPosition,
85 const wxSize &size = wxDefaultSize,
27316302 86 int style = wxSIMPLE_BORDER,
3dbeaa52
VZ
87 const wxValidator& validator = wxDefaultValidator,
88 const wxString &name = wxTextCtrlNameStr );
89
90 void OnChar( wxKeyEvent &event );
c13cace1 91 void OnKeyUp( wxKeyEvent &event );
3dbeaa52
VZ
92 void OnKillFocus( wxFocusEvent &event );
93
94private:
95 bool *m_accept;
96 wxString *m_res;
618a5e38 97 wxGenericTreeCtrl *m_owner;
3dbeaa52 98 wxString m_startValue;
dd5a32cc 99 bool m_finished;
3dbeaa52
VZ
100
101 DECLARE_EVENT_TABLE()
3dbeaa52
VZ
102};
103
f135ff73
VZ
104// a tree item
105class WXDLLEXPORT wxGenericTreeItem
c801d85f 106{
f135ff73 107public:
9ec64fa7
VZ
108 // ctors & dtor
109 wxGenericTreeItem() { m_data = NULL; }
110 wxGenericTreeItem( wxGenericTreeItem *parent,
618a5e38 111 const wxString& text,
618a5e38
RR
112 int image,
113 int selImage,
114 wxTreeItemData *data );
f135ff73 115
9ec64fa7 116 ~wxGenericTreeItem();
f135ff73 117
9ec64fa7
VZ
118 // trivial accessors
119 wxArrayGenericTreeItems& GetChildren() { return m_children; }
f135ff73 120
9ec64fa7
VZ
121 const wxString& GetText() const { return m_text; }
122 int GetImage(wxTreeItemIcon which = wxTreeItemIcon_Normal) const
3dbeaa52 123 { return m_images[which]; }
9ec64fa7 124 wxTreeItemData *GetData() const { return m_data; }
f135ff73 125
9ec64fa7
VZ
126 // returns the current image for the item (depending on its
127 // selected/expanded/whatever state)
128 int GetCurrentImage() const;
8dc99046 129
9ec64fa7
VZ
130 void SetText( const wxString &text );
131 void SetImage(int image, wxTreeItemIcon which) { m_images[which] = image; }
132 void SetData(wxTreeItemData *data) { m_data = data; }
f135ff73 133
9ec64fa7 134 void SetHasPlus(bool has = TRUE) { m_hasPlus = has; }
f135ff73 135
9ec64fa7 136 void SetBold(bool bold) { m_isBold = bold; }
ef44a621 137
9ec64fa7
VZ
138 int GetX() const { return m_x; }
139 int GetY() const { return m_y; }
f135ff73 140
9ec64fa7
VZ
141 void SetX(int x) { m_x = x; }
142 void SetY(int y) { m_y = y; }
f135ff73 143
9ec64fa7
VZ
144 int GetHeight() const { return m_height; }
145 int GetWidth() const { return m_width; }
91b8de8d 146
9ec64fa7
VZ
147 void SetHeight(int h) { m_height = h; }
148 void SetWidth(int w) { m_width = w; }
91b8de8d 149
9ec64fa7 150 wxGenericTreeItem *GetParent() const { return m_parent; }
c801d85f 151
9ec64fa7
VZ
152 // operations
153 // deletes all children notifying the treectrl about it if !NULL
154 // pointer given
941830cb 155 void DeleteChildren(wxGenericTreeCtrl *tree = NULL);
f135ff73 156
9ec64fa7
VZ
157 // get count of all children (and grand children if 'recursively')
158 size_t GetChildrenCount(bool recursively = TRUE) const;
f135ff73 159
9ec64fa7 160 void Insert(wxGenericTreeItem *child, size_t index)
f135ff73
VZ
161 { m_children.Insert(child, index); }
162
941830cb 163 void GetSize( int &x, int &y, const wxGenericTreeCtrl* );
f135ff73 164
9ec64fa7
VZ
165 // return the item at given position (or NULL if no item), onButton is
166 // TRUE if the point belongs to the item's button, otherwise it lies
167 // on the button's label
618a5e38
RR
168 wxGenericTreeItem *HitTest( const wxPoint& point,
169 const wxGenericTreeCtrl *,
170 int &flags,
171 int level );
f135ff73 172
9ec64fa7
VZ
173 void Expand() { m_isCollapsed = FALSE; }
174 void Collapse() { m_isCollapsed = TRUE; }
f135ff73 175
9ec64fa7 176 void SetHilight( bool set = TRUE ) { m_hasHilight = set; }
f135ff73 177
9ec64fa7
VZ
178 // status inquiries
179 bool HasChildren() const { return !m_children.IsEmpty(); }
ef8698d6 180 bool IsSelected() const { return m_hasHilight != 0; }
9ec64fa7
VZ
181 bool IsExpanded() const { return !m_isCollapsed; }
182 bool HasPlus() const { return m_hasPlus || HasChildren(); }
ef8698d6 183 bool IsBold() const { return m_isBold != 0; }
9ec64fa7
VZ
184
185 // attributes
186 // get them - may be NULL
187 wxTreeItemAttr *GetAttributes() const { return m_attr; }
188 // get them ensuring that the pointer is not NULL
189 wxTreeItemAttr& Attr()
190 {
191 if ( !m_attr )
618a5e38 192 {
9ec64fa7 193 m_attr = new wxTreeItemAttr;
618a5e38
RR
194 m_ownsAttr = TRUE;
195 }
9ec64fa7
VZ
196 return *m_attr;
197 }
618a5e38
RR
198 // set them
199 void SetAttributes(wxTreeItemAttr *attr)
200 {
201 if ( m_ownsAttr ) delete m_attr;
202 m_attr = attr;
203 m_ownsAttr = FALSE;
204 }
205 // set them and delete when done
206 void AssignAttributes(wxTreeItemAttr *attr)
207 {
208 SetAttributes(attr);
209 m_ownsAttr = TRUE;
210 }
f135ff73
VZ
211
212private:
618a5e38
RR
213 // since there can be very many of these, we save size by chosing
214 // the smallest representation for the elements and by ordering
215 // the members to avoid padding.
216 wxString m_text; // label to be rendered for item
217
218 wxTreeItemData *m_data; // user-provided data
219
220 wxArrayGenericTreeItems m_children; // list of children
221 wxGenericTreeItem *m_parent; // parent of this item
222
223 wxTreeItemAttr *m_attr; // attributes???
9ec64fa7
VZ
224
225 // tree ctrl images for the normal, selected, expanded and
226 // expanded+selected states
618a5e38 227 short m_images[wxTreeItemIcon_Max];
9ec64fa7 228
618a5e38
RR
229 wxCoord m_x; // (virtual) offset from top
230 short m_y; // (virtual) offset from left
231 short m_width; // width of this item
232 unsigned char m_height; // height of this item
9ec64fa7
VZ
233
234 // use bitfields to save size
235 int m_isCollapsed :1;
236 int m_hasHilight :1; // same as focused
237 int m_hasPlus :1; // used for item which doesn't have
238 // children but has a [+] button
239 int m_isBold :1; // render the label in bold font
618a5e38 240 int m_ownsAttr :1; // delete attribute when done
f135ff73
VZ
241};
242
243// =============================================================================
244// implementation
245// =============================================================================
246
06b466c7
VZ
247// ----------------------------------------------------------------------------
248// private functions
249// ----------------------------------------------------------------------------
250
251// translate the key or mouse event flags to the type of selection we're
252// dealing with
253static void EventFlagsToSelType(long style,
254 bool shiftDown,
255 bool ctrlDown,
dfc6cd93
SB
256 bool &is_multiple,
257 bool &extended_select,
258 bool &unselect_others)
06b466c7 259{
dfc6cd93
SB
260 is_multiple = (style & wxTR_MULTIPLE) != 0;
261 extended_select = shiftDown && is_multiple;
262 unselect_others = !(extended_select || (ctrlDown && is_multiple));
06b466c7 263}
e179bd65
RR
264
265// -----------------------------------------------------------------------------
266// wxTreeRenameTimer (internal)
267// -----------------------------------------------------------------------------
268
941830cb 269wxTreeRenameTimer::wxTreeRenameTimer( wxGenericTreeCtrl *owner )
e179bd65
RR
270{
271 m_owner = owner;
272}
273
274void wxTreeRenameTimer::Notify()
275{
276 m_owner->OnRenameTimer();
277}
278
279//-----------------------------------------------------------------------------
280// wxTreeTextCtrl (internal)
281//-----------------------------------------------------------------------------
282
e179bd65
RR
283BEGIN_EVENT_TABLE(wxTreeTextCtrl,wxTextCtrl)
284 EVT_CHAR (wxTreeTextCtrl::OnChar)
c13cace1 285 EVT_KEY_UP (wxTreeTextCtrl::OnKeyUp)
e179bd65
RR
286 EVT_KILL_FOCUS (wxTreeTextCtrl::OnKillFocus)
287END_EVENT_TABLE()
288
674ac8b9
VZ
289wxTreeTextCtrl::wxTreeTextCtrl( wxWindow *parent,
290 const wxWindowID id,
291 bool *accept,
292 wxString *res,
941830cb 293 wxGenericTreeCtrl *owner,
674ac8b9
VZ
294 const wxString &value,
295 const wxPoint &pos,
296 const wxSize &size,
297 int style,
298 const wxValidator& validator,
299 const wxString &name )
618a5e38 300 : wxTextCtrl( parent, id, value, pos, size, style, validator, name )
e179bd65
RR
301{
302 m_res = res;
303 m_accept = accept;
304 m_owner = owner;
5f1ea0ee 305 (*m_accept) = FALSE;
3dbeaa52 306 (*m_res) = wxEmptyString;
5f1ea0ee 307 m_startValue = value;
dd5a32cc 308 m_finished = FALSE;
e179bd65
RR
309}
310
311void wxTreeTextCtrl::OnChar( wxKeyEvent &event )
312{
313 if (event.m_keyCode == WXK_RETURN)
314 {
315 (*m_accept) = TRUE;
316 (*m_res) = GetValue();
33ac7e6f 317
dd5a32cc 318 if ((*m_res) != m_startValue)
bce1406b 319 m_owner->OnRenameAccept();
33ac7e6f 320
618a5e38
RR
321 if (!wxPendingDelete.Member(this))
322 wxPendingDelete.Append(this);
dd5a32cc
RR
323
324 m_finished = TRUE;
325 m_owner->SetFocus(); // This doesn't work. TODO.
618a5e38 326
e179bd65
RR
327 return;
328 }
329 if (event.m_keyCode == WXK_ESCAPE)
330 {
331 (*m_accept) = FALSE;
332 (*m_res) = "";
33ac7e6f 333
bce1406b
RR
334 if (!wxPendingDelete.Member(this))
335 wxPendingDelete.Append(this);
33ac7e6f 336
dd5a32cc
RR
337 m_finished = TRUE;
338 m_owner->SetFocus(); // This doesn't work. TODO.
339
e179bd65
RR
340 return;
341 }
342 event.Skip();
c13cace1
VS
343}
344
345void wxTreeTextCtrl::OnKeyUp( wxKeyEvent &event )
346{
dd5a32cc
RR
347 if (m_finished)
348 {
349 event.Skip();
350 return;
351 }
352
c13cace1
VS
353 // auto-grow the textctrl:
354 wxSize parentSize = m_owner->GetSize();
355 wxPoint myPos = GetPosition();
356 wxSize mySize = GetSize();
357 int sx, sy;
dd5a32cc 358 GetTextExtent(GetValue() + _T("M"), &sx, &sy);
c13cace1
VS
359 if (myPos.x + sx > parentSize.x) sx = parentSize.x - myPos.x;
360 if (mySize.x > sx) sx = mySize.x;
361 SetSize(sx, -1);
33ac7e6f 362
c13cace1 363 event.Skip();
e179bd65
RR
364}
365
dd5a32cc 366void wxTreeTextCtrl::OnKillFocus( wxFocusEvent &event )
e179bd65 367{
dd5a32cc
RR
368 if (m_finished)
369 {
370 event.Skip();
371 return;
372 }
373
bce1406b
RR
374 if (!wxPendingDelete.Member(this))
375 wxPendingDelete.Append(this);
9dfbf520 376
dd5a32cc
RR
377 (*m_accept) = TRUE;
378 (*m_res) = GetValue();
379
380 if ((*m_res) != m_startValue)
5f1ea0ee 381 m_owner->OnRenameAccept();
e179bd65
RR
382}
383
f135ff73 384// -----------------------------------------------------------------------------
c801d85f 385// wxGenericTreeItem
f135ff73 386// -----------------------------------------------------------------------------
c801d85f 387
f135ff73
VZ
388wxGenericTreeItem::wxGenericTreeItem(wxGenericTreeItem *parent,
389 const wxString& text,
f135ff73
VZ
390 int image, int selImage,
391 wxTreeItemData *data)
392 : m_text(text)
c801d85f 393{
00e12320
RR
394 m_images[wxTreeItemIcon_Normal] = image;
395 m_images[wxTreeItemIcon_Selected] = selImage;
396 m_images[wxTreeItemIcon_Expanded] = NO_IMAGE;
397 m_images[wxTreeItemIcon_SelectedExpanded] = NO_IMAGE;
8dc99046 398
00e12320
RR
399 m_data = data;
400 m_x = m_y = 0;
f135ff73 401
00e12320
RR
402 m_isCollapsed = TRUE;
403 m_hasHilight = FALSE;
404 m_hasPlus = FALSE;
405 m_isBold = FALSE;
c801d85f 406
00e12320 407 m_parent = parent;
f135ff73 408
00e12320 409 m_attr = (wxTreeItemAttr *)NULL;
618a5e38 410 m_ownsAttr = FALSE;
06b466c7 411
00e12320
RR
412 // We don't know the height here yet.
413 m_width = 0;
414 m_height = 0;
edaa81ae 415}
c801d85f 416
f135ff73 417wxGenericTreeItem::~wxGenericTreeItem()
c801d85f 418{
00e12320 419 delete m_data;
4832f7c0 420
618a5e38 421 if (m_ownsAttr) delete m_attr;
9ec64fa7 422
00e12320
RR
423 wxASSERT_MSG( m_children.IsEmpty(),
424 wxT("please call DeleteChildren() before deleting the item") );
372edb9d
VZ
425}
426
941830cb 427void wxGenericTreeItem::DeleteChildren(wxGenericTreeCtrl *tree)
372edb9d 428{
00e12320
RR
429 size_t count = m_children.Count();
430 for ( size_t n = 0; n < count; n++ )
a43a4f9d 431 {
00e12320
RR
432 wxGenericTreeItem *child = m_children[n];
433 if (tree)
434 tree->SendDeleteEvent(child);
a43a4f9d 435
00e12320
RR
436 child->DeleteChildren(tree);
437 delete child;
438 }
372edb9d 439
00e12320 440 m_children.Empty();
edaa81ae 441}
c801d85f 442
91b8de8d 443void wxGenericTreeItem::SetText( const wxString &text )
c801d85f 444{
00e12320 445 m_text = text;
edaa81ae 446}
c801d85f 447
4832f7c0 448size_t wxGenericTreeItem::GetChildrenCount(bool recursively) const
c801d85f 449{
00e12320
RR
450 size_t count = m_children.Count();
451 if ( !recursively )
452 return count;
4832f7c0 453
00e12320 454 size_t total = count;
f2593d0d 455 for (size_t n = 0; n < count; ++n)
00e12320
RR
456 {
457 total += m_children[n]->GetChildrenCount();
458 }
c801d85f 459
00e12320 460 return total;
edaa81ae 461}
c801d85f 462
618a5e38
RR
463void wxGenericTreeItem::GetSize( int &x, int &y,
464 const wxGenericTreeCtrl *theButton )
c801d85f 465{
618a5e38 466 int bottomY=m_y+theButton->GetLineHeight(this);
00e12320
RR
467 if ( y < bottomY ) y = bottomY;
468 int width = m_x + m_width;
469 if ( x < width ) x = width;
f135ff73 470
00e12320 471 if (IsExpanded())
4832f7c0 472 {
00e12320
RR
473 size_t count = m_children.Count();
474 for ( size_t n = 0; n < count; ++n )
475 {
618a5e38 476 m_children[n]->GetSize( x, y, theButton );
00e12320 477 }
df875e59 478 }
edaa81ae 479}
c801d85f 480
618a5e38
RR
481wxGenericTreeItem *wxGenericTreeItem::HitTest(const wxPoint& point,
482 const wxGenericTreeCtrl *theCtrl,
483 int &flags,
484 int level)
c801d85f 485{
618a5e38
RR
486 // for a hidden root node, don't evaluate it, but do evaluate children
487 if ( !(level == 0 && theCtrl->HasFlag(wxTR_HIDE_ROOT)) )
c801d85f 488 {
618a5e38
RR
489 // evaluate the item
490 int h = theCtrl->GetLineHeight(this);
491 if ((point.y > m_y) && (point.y < m_y + h))
f2593d0d 492 {
618a5e38
RR
493 int y_mid = m_y + h/2;
494 if (point.y < y_mid )
495 flags |= wxTREE_HITTEST_ONITEMUPPERPART;
496 else
497 flags |= wxTREE_HITTEST_ONITEMLOWERPART;
d3a9f4af 498
618a5e38
RR
499 // 5 is the size of the plus sign
500 int xCross = m_x - theCtrl->GetSpacing();
501 if ((point.x > xCross-5) && (point.x < xCross+5) &&
502 (point.y > y_mid-5) && (point.y < y_mid+5) &&
503 HasPlus() && theCtrl->HasButtons() )
504 {
505 flags |= wxTREE_HITTEST_ONITEMBUTTON;
506 return this;
507 }
0ae7f2a2 508
618a5e38
RR
509 if ((point.x >= m_x) && (point.x <= m_x+m_width))
510 {
511 int image_w = -1;
512 int image_h;
91b8de8d 513
618a5e38
RR
514 // assuming every image (normal and selected) has the same size!
515 if ( (GetImage() != NO_IMAGE) && theCtrl->m_imageListNormal )
516 theCtrl->m_imageListNormal->GetSize(GetImage(),
517 image_w, image_h);
518
519 if ((image_w != -1) && (point.x <= m_x + image_w + 1))
520 flags |= wxTREE_HITTEST_ONITEMICON;
521 else
522 flags |= wxTREE_HITTEST_ONITEMLABEL;
523
524 return this;
525 }
526
527 if (point.x < m_x)
528 flags |= wxTREE_HITTEST_ONITEMINDENT;
529 if (point.x > m_x+m_width)
530 flags |= wxTREE_HITTEST_ONITEMRIGHT;
91b8de8d 531
f2593d0d
RR
532 return this;
533 }
91b8de8d 534
618a5e38
RR
535 // if children are expanded, fall through to evaluate them
536 if (m_isCollapsed) return (wxGenericTreeItem*) NULL;
f2593d0d 537 }
618a5e38
RR
538
539 // evaluate children
540 size_t count = m_children.Count();
541 for ( size_t n = 0; n < count; n++ )
c801d85f 542 {
618a5e38
RR
543 wxGenericTreeItem *res = m_children[n]->HitTest( point,
544 theCtrl,
545 flags,
546 level + 1 );
547 if ( res != NULL )
548 return res;
edaa81ae 549 }
f135ff73 550
f2593d0d 551 return (wxGenericTreeItem*) NULL;
edaa81ae 552}
c801d85f 553
8dc99046
VZ
554int wxGenericTreeItem::GetCurrentImage() const
555{
556 int image = NO_IMAGE;
557 if ( IsExpanded() )
558 {
559 if ( IsSelected() )
560 {
561 image = GetImage(wxTreeItemIcon_SelectedExpanded);
562 }
563
564 if ( image == NO_IMAGE )
565 {
566 // we usually fall back to the normal item, but try just the
567 // expanded one (and not selected) first in this case
568 image = GetImage(wxTreeItemIcon_Expanded);
569 }
570 }
571 else // not expanded
572 {
573 if ( IsSelected() )
574 image = GetImage(wxTreeItemIcon_Selected);
575 }
576
618a5e38
RR
577 // maybe it doesn't have the specific image we want,
578 // try the default one instead
579 if ( image == NO_IMAGE ) image = GetImage();
8dc99046
VZ
580
581 return image;
582}
583
f135ff73 584// -----------------------------------------------------------------------------
941830cb 585// wxGenericTreeCtrl implementation
f135ff73
VZ
586// -----------------------------------------------------------------------------
587
941830cb 588IMPLEMENT_DYNAMIC_CLASS(wxGenericTreeCtrl, wxScrolledWindow)
f135ff73 589
941830cb
JS
590BEGIN_EVENT_TABLE(wxGenericTreeCtrl,wxScrolledWindow)
591 EVT_PAINT (wxGenericTreeCtrl::OnPaint)
592 EVT_MOUSE_EVENTS (wxGenericTreeCtrl::OnMouse)
593 EVT_CHAR (wxGenericTreeCtrl::OnChar)
594 EVT_SET_FOCUS (wxGenericTreeCtrl::OnSetFocus)
595 EVT_KILL_FOCUS (wxGenericTreeCtrl::OnKillFocus)
596 EVT_IDLE (wxGenericTreeCtrl::OnIdle)
f135ff73
VZ
597END_EVENT_TABLE()
598
618a5e38 599#if !defined(__WXMSW__) || defined(__WIN16__) || defined(__WXUNIVERSAL__)
233058c7
JS
600/*
601 * wxTreeCtrl has to be a real class or we have problems with
602 * the run-time information.
603 */
604
605IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxGenericTreeCtrl)
606#endif
607
f135ff73
VZ
608// -----------------------------------------------------------------------------
609// construction/destruction
610// -----------------------------------------------------------------------------
91b8de8d 611
941830cb 612void wxGenericTreeCtrl::Init()
c801d85f 613{
618a5e38 614 m_current = m_key_current = m_anchor = (wxGenericTreeItem *) NULL;
00e12320
RR
615 m_hasFocus = FALSE;
616 m_dirty = FALSE;
f135ff73 617
00e12320
RR
618 m_lineHeight = 10;
619 m_indent = 15;
620 m_spacing = 18;
f135ff73 621
b771aa29
VZ
622 m_hilightBrush = new wxBrush
623 (
624 wxSystemSettings::GetSystemColour
625 (
626 wxSYS_COLOUR_HIGHLIGHT
627 ),
628 wxSOLID
629 );
630
631 m_hilightUnfocusedBrush = new wxBrush
632 (
633 wxSystemSettings::GetSystemColour
634 (
635 wxSYS_COLOUR_BTNSHADOW
636 ),
637 wxSOLID
638 );
f135ff73 639
618a5e38 640 m_imageListNormal = m_imageListButtons =
00e12320 641 m_imageListState = (wxImageList *) NULL;
618a5e38 642 m_ownsImageListNormal = m_ownsImageListButtons =
46cd520d 643 m_ownsImageListState = FALSE;
978f38c2 644
00e12320 645 m_dragCount = 0;
3dbeaa52 646 m_isDragging = FALSE;
618a5e38 647 m_dropTarget = m_oldSelection = (wxGenericTreeItem *)NULL;
9dfbf520 648
00e12320 649 m_renameTimer = new wxTreeRenameTimer( this );
f6bcfd97 650 m_lastOnSame = FALSE;
f38374d0 651
00e12320
RR
652 m_normalFont = wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT );
653 m_boldFont = wxFont( m_normalFont.GetPointSize(),
618a5e38
RR
654 m_normalFont.GetFamily(),
655 m_normalFont.GetStyle(),
656 wxBOLD,
657 m_normalFont.GetUnderlined());
edaa81ae 658}
c801d85f 659
618a5e38
RR
660bool wxGenericTreeCtrl::Create(wxWindow *parent,
661 wxWindowID id,
662 const wxPoint& pos,
663 const wxSize& size,
664 long style,
665 const wxValidator &validator,
666 const wxString& name )
c801d85f 667{
2593f033
RR
668#ifdef __WXMAC__
669 int major,minor;
670 wxGetOsVersion( &major, &minor );
671
672 if (style & wxTR_HAS_BUTTONS) style |= wxTR_MAC_BUTTONS;
f6eaeda1 673 if (style & wxTR_HAS_BUTTONS) style &= ~wxTR_HAS_BUTTONS;
2593f033
RR
674 style &= ~wxTR_LINES_AT_ROOT;
675 style |= wxTR_NO_LINES;
676 if (major < 10)
677 style |= wxTR_ROW_LINES;
17d5bdf9
RR
678 if (major >= 10)
679 style |= wxTR_AQUA_BUTTONS;
2593f033
RR
680#endif
681
17d5bdf9 682
618a5e38
RR
683 wxScrolledWindow::Create( parent, id, pos, size,
684 style|wxHSCROLL|wxVSCROLL, name );
685
686 // If the tree display has no buttons, but does have
687 // connecting lines, we can use a narrower layout.
688 // It may not be a good idea to force this...
689 if (!HasButtons() && !HasFlag(wxTR_NO_LINES))
690 {
691 m_indent= 10;
692 m_spacing = 10;
693 }
978f38c2 694
ce4169a4 695#if wxUSE_VALIDATORS
00e12320 696 SetValidator( validator );
ce4169a4 697#endif
f135ff73 698
91fc2bdc 699 SetBackgroundColour( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_LISTBOX ) );
c22886bd 700
00e12320 701// m_dottedPen = wxPen( "grey", 0, wxDOT ); too slow under XFree86
9e0a12c9 702 m_dottedPen = wxPen( wxT("grey"), 0, 0 );
f135ff73 703
00e12320 704 return TRUE;
edaa81ae 705}
c801d85f 706
941830cb 707wxGenericTreeCtrl::~wxGenericTreeCtrl()
c801d85f 708{
b771aa29
VZ
709 delete m_hilightBrush;
710 delete m_hilightUnfocusedBrush;
a43a4f9d 711
00e12320 712 DeleteAllItems();
9dfbf520 713
00e12320 714 delete m_renameTimer;
46cd520d
VS
715 if (m_ownsImageListNormal) delete m_imageListNormal;
716 if (m_ownsImageListState) delete m_imageListState;
618a5e38 717 if (m_ownsImageListButtons) delete m_imageListButtons;
edaa81ae 718}
c801d85f 719
f135ff73
VZ
720// -----------------------------------------------------------------------------
721// accessors
722// -----------------------------------------------------------------------------
723
941830cb 724size_t wxGenericTreeCtrl::GetCount() const
c801d85f 725{
f2593d0d 726 return m_anchor == NULL ? 0u : m_anchor->GetChildrenCount();
edaa81ae 727}
c801d85f 728
941830cb 729void wxGenericTreeCtrl::SetIndent(unsigned int indent)
c801d85f 730{
f2593d0d
RR
731 m_indent = indent;
732 m_dirty = TRUE;
cf724bce
RR
733}
734
941830cb 735void wxGenericTreeCtrl::SetSpacing(unsigned int spacing)
cf724bce 736{
f2593d0d
RR
737 m_spacing = spacing;
738 m_dirty = TRUE;
f135ff73 739}
74bedbeb 740
941830cb 741size_t wxGenericTreeCtrl::GetChildrenCount(const wxTreeItemId& item, bool recursively)
4832f7c0 742{
f2593d0d 743 wxCHECK_MSG( item.IsOk(), 0u, wxT("invalid tree item") );
4832f7c0 744
941830cb 745 return ((wxGenericTreeItem*) item.m_pItem)->GetChildrenCount(recursively);
4832f7c0
VZ
746}
747
618a5e38
RR
748void wxGenericTreeCtrl::SetWindowStyle(const long styles)
749{
750 // right now, just sets the styles. Eventually, we may
751 // want to update the inherited styles, but right now
752 // none of the parents has updatable styles
753 m_windowStyle = styles;
754 m_dirty = TRUE;
755}
756
f135ff73
VZ
757// -----------------------------------------------------------------------------
758// functions to work with tree items
759// -----------------------------------------------------------------------------
74bedbeb 760
941830cb 761wxString wxGenericTreeCtrl::GetItemText(const wxTreeItemId& item) const
f135ff73 762{
f2593d0d 763 wxCHECK_MSG( item.IsOk(), wxT(""), wxT("invalid tree item") );
4832f7c0 764
941830cb 765 return ((wxGenericTreeItem*) item.m_pItem)->GetText();
edaa81ae 766}
74bedbeb 767
941830cb 768int wxGenericTreeCtrl::GetItemImage(const wxTreeItemId& item,
8dc99046 769 wxTreeItemIcon which) const
74bedbeb 770{
f2593d0d 771 wxCHECK_MSG( item.IsOk(), -1, wxT("invalid tree item") );
4832f7c0 772
941830cb 773 return ((wxGenericTreeItem*) item.m_pItem)->GetImage(which);
edaa81ae 774}
c801d85f 775
941830cb 776wxTreeItemData *wxGenericTreeCtrl::GetItemData(const wxTreeItemId& item) const
c801d85f 777{
f2593d0d 778 wxCHECK_MSG( item.IsOk(), NULL, wxT("invalid tree item") );
4832f7c0 779
941830cb 780 return ((wxGenericTreeItem*) item.m_pItem)->GetData();
edaa81ae 781}
c801d85f 782
941830cb 783void wxGenericTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text)
f135ff73 784{
f2593d0d 785 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
4832f7c0 786
f2593d0d 787 wxClientDC dc(this);
941830cb 788 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
f2593d0d
RR
789 pItem->SetText(text);
790 CalculateSize(pItem, dc);
791 RefreshLine(pItem);
f135ff73 792}
c801d85f 793
941830cb 794void wxGenericTreeCtrl::SetItemImage(const wxTreeItemId& item,
8dc99046
VZ
795 int image,
796 wxTreeItemIcon which)
f135ff73 797{
223d09f6 798 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
4832f7c0 799
941830cb 800 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
8dc99046 801 pItem->SetImage(image, which);
4832f7c0 802
8dc99046
VZ
803 wxClientDC dc(this);
804 CalculateSize(pItem, dc);
805 RefreshLine(pItem);
edaa81ae 806}
c801d85f 807
941830cb 808void wxGenericTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data)
c801d85f 809{
f2593d0d 810 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
4832f7c0 811
941830cb 812 ((wxGenericTreeItem*) item.m_pItem)->SetData(data);
edaa81ae 813}
c801d85f 814
941830cb 815void wxGenericTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has)
c801d85f 816{
f2593d0d 817 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
4832f7c0 818
941830cb 819 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
f2593d0d
RR
820 pItem->SetHasPlus(has);
821 RefreshLine(pItem);
edaa81ae 822}
c801d85f 823
941830cb 824void wxGenericTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold)
ef44a621 825{
9ec64fa7 826 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
ef44a621 827
9ec64fa7 828 // avoid redrawing the tree if no real change
941830cb 829 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
9ec64fa7
VZ
830 if ( pItem->IsBold() != bold )
831 {
832 pItem->SetBold(bold);
833 RefreshLine(pItem);
834 }
835}
836
941830cb 837void wxGenericTreeCtrl::SetItemTextColour(const wxTreeItemId& item,
9ec64fa7
VZ
838 const wxColour& col)
839{
840 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
841
941830cb 842 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
9ec64fa7
VZ
843 pItem->Attr().SetTextColour(col);
844 RefreshLine(pItem);
845}
846
941830cb 847void wxGenericTreeCtrl::SetItemBackgroundColour(const wxTreeItemId& item,
9ec64fa7
VZ
848 const wxColour& col)
849{
850 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
851
941830cb 852 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
9ec64fa7
VZ
853 pItem->Attr().SetBackgroundColour(col);
854 RefreshLine(pItem);
855}
856
941830cb 857void wxGenericTreeCtrl::SetItemFont(const wxTreeItemId& item, const wxFont& font)
9ec64fa7
VZ
858{
859 wxCHECK_RET( item.IsOk(), wxT("invalid tree item") );
860
941830cb 861 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
9ec64fa7 862 pItem->Attr().SetFont(font);
ef44a621 863 RefreshLine(pItem);
ef44a621
VZ
864}
865
3da2715f
JS
866bool wxGenericTreeCtrl::SetFont( const wxFont &font )
867{
868 wxScrolledWindow::SetFont(font);
869
870 m_normalFont = font ;
871 m_boldFont = wxFont( m_normalFont.GetPointSize(),
872 m_normalFont.GetFamily(),
873 m_normalFont.GetStyle(),
874 wxBOLD,
875 m_normalFont.GetUnderlined());
876
877 return TRUE;
878}
879
880
f135ff73
VZ
881// -----------------------------------------------------------------------------
882// item status inquiries
883// -----------------------------------------------------------------------------
884
1ed01484 885bool wxGenericTreeCtrl::IsVisible(const wxTreeItemId& item) const
c801d85f 886{
1ed01484
JS
887 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
888
c22886bd 889 // An item is only visible if it's not a descendant of a collapsed item
1ed01484 890 wxGenericTreeItem *pItem = (wxGenericTreeItem*) item.m_pItem;
c22886bd
RR
891 wxGenericTreeItem* parent = pItem->GetParent();
892 while (parent)
893 {
894 if (!parent->IsExpanded())
895 return FALSE;
896 parent = parent->GetParent();
897 }
1ed01484
JS
898
899 int startX, startY;
900 GetViewStart(& startX, & startY);
901
902 wxSize clientSize = GetClientSize();
903
904 wxRect rect;
905 if (!GetBoundingRect(item, rect))
906 return FALSE;
c22886bd
RR
907 if (rect.GetWidth() == 0 || rect.GetHeight() == 0)
908 return FALSE;
1ed01484
JS
909 if (rect.GetBottom() < 0 || rect.GetTop() > clientSize.y)
910 return FALSE;
911 if (rect.GetRight() < 0 || rect.GetLeft() > clientSize.x)
912 return FALSE;
f135ff73 913
f2593d0d 914 return TRUE;
edaa81ae 915}
c801d85f 916
941830cb 917bool wxGenericTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const
c801d85f 918{
f2593d0d 919 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
4832f7c0 920
59a2e635
VZ
921 // consider that the item does have children if it has the "+" button: it
922 // might not have them (if it had never been expanded yet) but then it
923 // could have them as well and it's better to err on this side rather than
924 // disabling some operations which are restricted to the items with
925 // children for an item which does have them
607d9cfc 926 return ((wxGenericTreeItem*) item.m_pItem)->HasPlus();
edaa81ae 927}
c801d85f 928
941830cb 929bool wxGenericTreeCtrl::IsExpanded(const wxTreeItemId& item) const
c801d85f 930{
f2593d0d 931 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
4832f7c0 932
941830cb 933 return ((wxGenericTreeItem*) item.m_pItem)->IsExpanded();
f135ff73 934}
29d87bba 935
941830cb 936bool wxGenericTreeCtrl::IsSelected(const wxTreeItemId& item) const
f135ff73 937{
f2593d0d 938 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
4832f7c0 939
941830cb 940 return ((wxGenericTreeItem*) item.m_pItem)->IsSelected();
f135ff73 941}
29d87bba 942
941830cb 943bool wxGenericTreeCtrl::IsBold(const wxTreeItemId& item) const
ef44a621 944{
f2593d0d 945 wxCHECK_MSG( item.IsOk(), FALSE, wxT("invalid tree item") );
ef44a621 946
941830cb 947 return ((wxGenericTreeItem*) item.m_pItem)->IsBold();
ef44a621
VZ
948}
949
f135ff73
VZ
950// -----------------------------------------------------------------------------
951// navigation
952// -----------------------------------------------------------------------------
29d87bba 953
941830cb 954wxTreeItemId wxGenericTreeCtrl::GetParent(const wxTreeItemId& item) const
f135ff73 955{
618a5e38 956 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
389cdc7a 957
618a5e38 958 return ((wxGenericTreeItem*) item.m_pItem)->GetParent();
f135ff73 959}
29d87bba 960
941830cb 961wxTreeItemId wxGenericTreeCtrl::GetFirstChild(const wxTreeItemId& item, long& cookie) const
f135ff73 962{
618a5e38 963 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
29d87bba 964
618a5e38
RR
965 cookie = 0;
966 return GetNextChild(item, cookie);
f135ff73 967}
29d87bba 968
941830cb 969wxTreeItemId wxGenericTreeCtrl::GetNextChild(const wxTreeItemId& item, long& cookie) const
f135ff73 970{
618a5e38 971 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
29d87bba 972
618a5e38
RR
973 wxArrayGenericTreeItems& children = ((wxGenericTreeItem*) item.m_pItem)->GetChildren();
974 if ( (size_t)cookie < children.Count() )
975 {
976 return children.Item((size_t)cookie++);
977 }
978 else
979 {
980 // there are no more of them
981 return wxTreeItemId();
982 }
f135ff73 983}
29d87bba 984
941830cb 985wxTreeItemId wxGenericTreeCtrl::GetLastChild(const wxTreeItemId& item) const
978f38c2 986{
618a5e38 987 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
978f38c2 988
618a5e38
RR
989 wxArrayGenericTreeItems& children = ((wxGenericTreeItem*) item.m_pItem)->GetChildren();
990 return (children.IsEmpty() ? wxTreeItemId() : wxTreeItemId(children.Last()));
978f38c2
VZ
991}
992
941830cb 993wxTreeItemId wxGenericTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
f135ff73 994{
618a5e38 995 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
f135ff73 996
618a5e38
RR
997 wxGenericTreeItem *i = (wxGenericTreeItem*) item.m_pItem;
998 wxGenericTreeItem *parent = i->GetParent();
999 if ( parent == NULL )
1000 {
1001 // root item doesn't have any siblings
1002 return wxTreeItemId();
1003 }
4832f7c0 1004
618a5e38
RR
1005 wxArrayGenericTreeItems& siblings = parent->GetChildren();
1006 int index = siblings.Index(i);
1007 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
29d87bba 1008
618a5e38
RR
1009 size_t n = (size_t)(index + 1);
1010 return n == siblings.Count() ? wxTreeItemId() : wxTreeItemId(siblings[n]);
edaa81ae 1011}
c801d85f 1012
941830cb 1013wxTreeItemId wxGenericTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
c801d85f 1014{
618a5e38 1015 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
f135ff73 1016
618a5e38
RR
1017 wxGenericTreeItem *i = (wxGenericTreeItem*) item.m_pItem;
1018 wxGenericTreeItem *parent = i->GetParent();
1019 if ( parent == NULL )
1020 {
1021 // root item doesn't have any siblings
1022 return wxTreeItemId();
1023 }
4832f7c0 1024
618a5e38
RR
1025 wxArrayGenericTreeItems& siblings = parent->GetChildren();
1026 int index = siblings.Index(i);
1027 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
29d87bba 1028
618a5e38
RR
1029 return index == 0 ? wxTreeItemId()
1030 : wxTreeItemId(siblings[(size_t)(index - 1)]);
f135ff73 1031}
389cdc7a 1032
1ed01484
JS
1033// Only for internal use right now, but should probably be public
1034wxTreeItemId wxGenericTreeCtrl::GetNext(const wxTreeItemId& item) const
f135ff73 1035{
618a5e38
RR
1036 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
1037
1038 wxGenericTreeItem *i = (wxGenericTreeItem*) item.m_pItem;
1039
1040 // First see if there are any children.
1041 wxArrayGenericTreeItems& children = i->GetChildren();
1042 if (children.GetCount() > 0)
1043 {
1044 return children.Item(0);
1045 }
1046 else
1047 {
1048 // Try a sibling of this or ancestor instead
1049 wxTreeItemId p = item;
1050 wxTreeItemId toFind;
1051 do
1052 {
1053 toFind = GetNextSibling(p);
1054 p = GetParent(p);
1055 } while (p.IsOk() && !toFind.IsOk());
1056 return toFind;
1057 }
1ed01484
JS
1058}
1059
1ed01484
JS
1060wxTreeItemId wxGenericTreeCtrl::GetFirstVisibleItem() const
1061{
618a5e38
RR
1062 wxTreeItemId id = GetRootItem();
1063 if (!id.IsOk())
1ed01484 1064 return id;
1ed01484 1065
618a5e38
RR
1066 do
1067 {
1068 if (IsVisible(id))
1069 return id;
1070 id = GetNext(id);
1071 } while (id.IsOk());
1072
1073 return wxTreeItemId();
1ed01484
JS
1074}
1075
941830cb 1076wxTreeItemId wxGenericTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
f135ff73 1077{
618a5e38 1078 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
29d87bba 1079
618a5e38
RR
1080 wxTreeItemId id = item;
1081 if (id.IsOk())
1082 {
1083 while (id = GetNext(id), id.IsOk())
1084 {
1085 if (IsVisible(id))
1086 return id;
1087 }
1088 }
1089 return wxTreeItemId();
f135ff73 1090}
29d87bba 1091
941830cb 1092wxTreeItemId wxGenericTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
f135ff73 1093{
618a5e38 1094 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), wxT("invalid tree item") );
29d87bba 1095
618a5e38 1096 wxFAIL_MSG(wxT("not implemented"));
29d87bba 1097
618a5e38 1098 return wxTreeItemId();
edaa81ae 1099}
c801d85f 1100
f135ff73
VZ
1101// -----------------------------------------------------------------------------
1102// operations
1103// -----------------------------------------------------------------------------
1104
941830cb 1105wxTreeItemId wxGenericTreeCtrl::DoInsertItem(const wxTreeItemId& parentId,
f135ff73
VZ
1106 size_t previous,
1107 const wxString& text,
1108 int image, int selImage,
1109 wxTreeItemData *data)
c801d85f 1110{
941830cb 1111 wxGenericTreeItem *parent = (wxGenericTreeItem*) parentId.m_pItem;
f49f2b0c
RR
1112 if ( !parent )
1113 {
1114 // should we give a warning here?
1115 return AddRoot(text, image, selImage, data);
1116 }
4832f7c0 1117
618a5e38
RR
1118 m_dirty = TRUE; // do this first so stuff below doesn't cause flicker
1119
f38374d0 1120 wxGenericTreeItem *item =
2b84e565 1121 new wxGenericTreeItem( parent, text, image, selImage, data );
74bedbeb 1122
f49f2b0c
RR
1123 if ( data != NULL )
1124 {
40fab78b 1125 data->m_pItem = (long) item;
f49f2b0c 1126 }
74bedbeb 1127
f49f2b0c 1128 parent->Insert( item, previous );
ef44a621 1129
f49f2b0c 1130 return item;
4c681997
RR
1131}
1132
941830cb 1133wxTreeItemId wxGenericTreeCtrl::AddRoot(const wxString& text,
f135ff73
VZ
1134 int image, int selImage,
1135 wxTreeItemData *data)
4c681997 1136{
f49f2b0c 1137 wxCHECK_MSG( !m_anchor, wxTreeItemId(), wxT("tree can have only one root") );
389cdc7a 1138
618a5e38
RR
1139 m_dirty = TRUE; // do this first so stuff below doesn't cause flicker
1140
2b84e565 1141 m_anchor = new wxGenericTreeItem((wxGenericTreeItem *)NULL, text,
f135ff73 1142 image, selImage, data);
618a5e38
RR
1143 if (HasFlag(wxTR_HIDE_ROOT))
1144 {
1145 // if root is hidden, make sure we can navigate
1146 // into children
1147 m_anchor->SetHasPlus();
1148 Expand(m_anchor);
1149 }
f49f2b0c
RR
1150 if ( data != NULL )
1151 {
40fab78b 1152 data->m_pItem = (long) m_anchor;
f49f2b0c 1153 }
f38374d0 1154
f49f2b0c
RR
1155 if (!HasFlag(wxTR_MULTIPLE))
1156 {
1157 m_current = m_key_current = m_anchor;
06b466c7 1158 m_current->SetHilight( TRUE );
f49f2b0c 1159 }
389cdc7a 1160
f49f2b0c 1161 return m_anchor;
edaa81ae 1162}
c801d85f 1163
941830cb 1164wxTreeItemId wxGenericTreeCtrl::PrependItem(const wxTreeItemId& parent,
f135ff73
VZ
1165 const wxString& text,
1166 int image, int selImage,
1167 wxTreeItemData *data)
c801d85f 1168{
f2593d0d 1169 return DoInsertItem(parent, 0u, text, image, selImage, data);
edaa81ae 1170}
c801d85f 1171
941830cb 1172wxTreeItemId wxGenericTreeCtrl::InsertItem(const wxTreeItemId& parentId,
f135ff73
VZ
1173 const wxTreeItemId& idPrevious,
1174 const wxString& text,
1175 int image, int selImage,
1176 wxTreeItemData *data)
c801d85f 1177{
941830cb 1178 wxGenericTreeItem *parent = (wxGenericTreeItem*) parentId.m_pItem;
f2593d0d
RR
1179 if ( !parent )
1180 {
1181 // should we give a warning here?
1182 return AddRoot(text, image, selImage, data);
1183 }
c801d85f 1184
941830cb 1185 int index = parent->GetChildren().Index((wxGenericTreeItem*) idPrevious.m_pItem);
f2593d0d 1186 wxASSERT_MSG( index != wxNOT_FOUND,
941830cb 1187 wxT("previous item in wxGenericTreeCtrl::InsertItem() is not a sibling") );
06b466c7 1188
f2593d0d
RR
1189 return DoInsertItem(parentId, (size_t)++index, text, image, selImage, data);
1190}
1191
941830cb 1192wxTreeItemId wxGenericTreeCtrl::InsertItem(const wxTreeItemId& parentId,
f2593d0d
RR
1193 size_t before,
1194 const wxString& text,
1195 int image, int selImage,
1196 wxTreeItemData *data)
1197{
941830cb 1198 wxGenericTreeItem *parent = (wxGenericTreeItem*) parentId.m_pItem;
f2593d0d
RR
1199 if ( !parent )
1200 {
1201 // should we give a warning here?
1202 return AddRoot(text, image, selImage, data);
1203 }
1204
1205 return DoInsertItem(parentId, before, text, image, selImage, data);
edaa81ae 1206}
c801d85f 1207
941830cb 1208wxTreeItemId wxGenericTreeCtrl::AppendItem(const wxTreeItemId& parentId,
f135ff73
VZ
1209 const wxString& text,
1210 int image, int selImage,
1211 wxTreeItemData *data)
74bedbeb 1212{
941830cb 1213 wxGenericTreeItem *parent = (wxGenericTreeItem*) parentId.m_pItem;
f2593d0d
RR
1214 if ( !parent )
1215 {
1216 // should we give a warning here?
1217 return AddRoot(text, image, selImage, data);
1218 }
f135ff73 1219
f2593d0d
RR
1220 return DoInsertItem( parent, parent->GetChildren().Count(), text,
1221 image, selImage, data);
74bedbeb
VZ
1222}
1223
941830cb 1224void wxGenericTreeCtrl::SendDeleteEvent(wxGenericTreeItem *item)
a43a4f9d 1225{
f2593d0d 1226 wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, GetId() );
40fab78b 1227 event.m_item = (long) item;
f2593d0d
RR
1228 event.SetEventObject( this );
1229 ProcessEvent( event );
a43a4f9d
VZ
1230}
1231
941830cb 1232void wxGenericTreeCtrl::DeleteChildren(const wxTreeItemId& itemId)
372edb9d 1233{
618a5e38
RR
1234 m_dirty = TRUE; // do this first so stuff below doesn't cause flicker
1235
941830cb 1236 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
a43a4f9d 1237 item->DeleteChildren(this);
372edb9d
VZ
1238}
1239
941830cb 1240void wxGenericTreeCtrl::Delete(const wxTreeItemId& itemId)
c801d85f 1241{
618a5e38
RR
1242 m_dirty = TRUE; // do this first so stuff below doesn't cause flicker
1243
941830cb 1244 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
ff5bf259 1245
618a5e38
RR
1246 // don't stay with invalid m_key_current or we will crash in
1247 // the next call to OnChar()
aaa2f297
VZ
1248 bool changeKeyCurrent = FALSE;
1249 wxGenericTreeItem *itemKey = m_key_current;
618a5e38 1250 while ( itemKey )
aaa2f297
VZ
1251 {
1252 if ( itemKey == item )
1253 {
1254 // m_key_current is a descendant of the item being deleted
1255 changeKeyCurrent = TRUE;
618a5e38 1256 break;
aaa2f297 1257 }
618a5e38 1258 itemKey = itemKey->GetParent();
aaa2f297
VZ
1259 }
1260
1261 wxGenericTreeItem *parent = item->GetParent();
97d7bfb8
RR
1262 if ( parent )
1263 {
1264 parent->GetChildren().Remove( item ); // remove by value
1265 }
f135ff73 1266
aaa2f297
VZ
1267 if ( changeKeyCurrent )
1268 {
1269 // may be NULL or not
1270 m_key_current = parent;
1271 }
1272
97d7bfb8
RR
1273 item->DeleteChildren(this);
1274 SendDeleteEvent(item);
1275 delete item;
edaa81ae 1276}
c801d85f 1277
941830cb 1278void wxGenericTreeCtrl::DeleteAllItems()
c801d85f 1279{
97d7bfb8
RR
1280 if ( m_anchor )
1281 {
618a5e38
RR
1282 m_dirty = TRUE;
1283
97d7bfb8
RR
1284 m_anchor->DeleteChildren(this);
1285 delete m_anchor;
a43a4f9d 1286
97d7bfb8 1287 m_anchor = NULL;
97d7bfb8 1288 }
edaa81ae
RR
1289}
1290
941830cb 1291void wxGenericTreeCtrl::Expand(const wxTreeItemId& itemId)
edaa81ae 1292{
941830cb 1293 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
f135ff73 1294
941830cb 1295 wxCHECK_RET( item, _T("invalid item in wxGenericTreeCtrl::Expand") );
f6bcfd97 1296
f2593d0d
RR
1297 if ( !item->HasPlus() )
1298 return;
978f38c2 1299
f2593d0d
RR
1300 if ( item->IsExpanded() )
1301 return;
f135ff73 1302
f2593d0d 1303 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, GetId() );
40fab78b 1304 event.m_item = (long) item;
f2593d0d 1305 event.SetEventObject( this );
004fd0c8 1306
f2593d0d
RR
1307 if ( ProcessEvent( event ) && !event.IsAllowed() )
1308 {
1309 // cancelled by program
1310 return;
1311 }
4832f7c0 1312
f2593d0d
RR
1313 item->Expand();
1314 CalculatePositions();
f135ff73 1315
f2593d0d 1316 RefreshSubtree(item);
f135ff73 1317
f2593d0d
RR
1318 event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED);
1319 ProcessEvent( event );
edaa81ae
RR
1320}
1321
941830cb 1322void wxGenericTreeCtrl::ExpandAll(const wxTreeItemId& item)
f6bcfd97
BP
1323{
1324 Expand(item);
1325 if ( IsExpanded(item) )
1326 {
1327 long cookie;
1328 wxTreeItemId child = GetFirstChild(item, cookie);
1329 while ( child.IsOk() )
1330 {
1331 ExpandAll(child);
1332
1333 child = GetNextChild(item, cookie);
1334 }
1335 }
1336}
1337
941830cb 1338void wxGenericTreeCtrl::Collapse(const wxTreeItemId& itemId)
edaa81ae 1339{
941830cb 1340 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
f135ff73 1341
f2593d0d
RR
1342 if ( !item->IsExpanded() )
1343 return;
f135ff73 1344
f2593d0d 1345 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, GetId() );
40fab78b 1346 event.m_item = (long) item;
f2593d0d
RR
1347 event.SetEventObject( this );
1348 if ( ProcessEvent( event ) && !event.IsAllowed() )
1349 {
1350 // cancelled by program
1351 return;
1352 }
4832f7c0 1353
f2593d0d 1354 item->Collapse();
f135ff73 1355
618a5e38 1356#if 0 // TODO why should items be collapsed recursively?
f2593d0d
RR
1357 wxArrayGenericTreeItems& children = item->GetChildren();
1358 size_t count = children.Count();
1359 for ( size_t n = 0; n < count; n++ )
1360 {
1361 Collapse(children[n]);
1362 }
618a5e38 1363#endif
f135ff73 1364
f2593d0d 1365 CalculatePositions();
f135ff73 1366
f2593d0d 1367 RefreshSubtree(item);
f135ff73 1368
f2593d0d
RR
1369 event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED);
1370 ProcessEvent( event );
edaa81ae 1371}
c801d85f 1372
941830cb 1373void wxGenericTreeCtrl::CollapseAndReset(const wxTreeItemId& item)
c801d85f 1374{
00e12320
RR
1375 Collapse(item);
1376 DeleteChildren(item);
edaa81ae 1377}
c801d85f 1378
941830cb 1379void wxGenericTreeCtrl::Toggle(const wxTreeItemId& itemId)
c801d85f 1380{
941830cb 1381 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
389cdc7a 1382
00e12320
RR
1383 if (item->IsExpanded())
1384 Collapse(itemId);
1385 else
1386 Expand(itemId);
f135ff73 1387}
389cdc7a 1388
941830cb 1389void wxGenericTreeCtrl::Unselect()
f135ff73 1390{
00e12320
RR
1391 if (m_current)
1392 {
1393 m_current->SetHilight( FALSE );
1394 RefreshLine( m_current );
1395 }
edaa81ae 1396}
c801d85f 1397
941830cb 1398void wxGenericTreeCtrl::UnselectAllChildren(wxGenericTreeItem *item)
389cdc7a 1399{
00e12320
RR
1400 if (item->IsSelected())
1401 {
1402 item->SetHilight(FALSE);
1403 RefreshLine(item);
1404 }
d3a9f4af 1405
00e12320 1406 if (item->HasChildren())
88ac883a 1407 {
00e12320
RR
1408 wxArrayGenericTreeItems& children = item->GetChildren();
1409 size_t count = children.Count();
1410 for ( size_t n = 0; n < count; ++n )
1411 {
1412 UnselectAllChildren(children[n]);
1413 }
88ac883a
VZ
1414 }
1415}
f135ff73 1416
941830cb 1417void wxGenericTreeCtrl::UnselectAll()
88ac883a 1418{
941830cb 1419 UnselectAllChildren((wxGenericTreeItem*) GetRootItem().m_pItem);
88ac883a
VZ
1420}
1421
1422// Recursive function !
1423// To stop we must have crt_item<last_item
91b8de8d 1424// Algorithm :
88ac883a 1425// Tag all next children, when no more children,
d3a9f4af 1426// Move to parent (not to tag)
91b8de8d 1427// Keep going... if we found last_item, we stop.
941830cb 1428bool wxGenericTreeCtrl::TagNextChildren(wxGenericTreeItem *crt_item, wxGenericTreeItem *last_item, bool select)
88ac883a
VZ
1429{
1430 wxGenericTreeItem *parent = crt_item->GetParent();
1431
00e12320
RR
1432 if (parent == NULL) // This is root item
1433 return TagAllChildrenUntilLast(crt_item, last_item, select);
88ac883a 1434
91b8de8d 1435 wxArrayGenericTreeItems& children = parent->GetChildren();
88ac883a
VZ
1436 int index = children.Index(crt_item);
1437 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
1438
1439 size_t count = children.Count();
1440 for (size_t n=(size_t)(index+1); n<count; ++n)
00e12320
RR
1441 {
1442 if (TagAllChildrenUntilLast(children[n], last_item, select)) return TRUE;
1443 }
88ac883a
VZ
1444
1445 return TagNextChildren(parent, last_item, select);
1446}
1447
941830cb 1448bool wxGenericTreeCtrl::TagAllChildrenUntilLast(wxGenericTreeItem *crt_item, wxGenericTreeItem *last_item, bool select)
88ac883a 1449{
00e12320
RR
1450 crt_item->SetHilight(select);
1451 RefreshLine(crt_item);
d3a9f4af 1452
06b466c7 1453 if (crt_item==last_item)
00e12320 1454 return TRUE;
88ac883a 1455
00e12320 1456 if (crt_item->HasChildren())
88ac883a 1457 {
00e12320
RR
1458 wxArrayGenericTreeItems& children = crt_item->GetChildren();
1459 size_t count = children.Count();
1460 for ( size_t n = 0; n < count; ++n )
1461 {
06b466c7 1462 if (TagAllChildrenUntilLast(children[n], last_item, select))
00e12320 1463 return TRUE;
06b466c7 1464 }
88ac883a 1465 }
d3a9f4af 1466
c0de7af4 1467 return FALSE;
88ac883a
VZ
1468}
1469
941830cb 1470void wxGenericTreeCtrl::SelectItemRange(wxGenericTreeItem *item1, wxGenericTreeItem *item2)
88ac883a 1471{
f2593d0d
RR
1472 // item2 is not necessary after item1
1473 wxGenericTreeItem *first=NULL, *last=NULL;
88ac883a 1474
f2593d0d
RR
1475 // choice first' and 'last' between item1 and item2
1476 if (item1->GetY()<item2->GetY())
06b466c7 1477 {
f2593d0d
RR
1478 first=item1;
1479 last=item2;
1480 }
1481 else
1482 {
1483 first=item2;
1484 last=item1;
1485 }
88ac883a 1486
f2593d0d 1487 bool select = m_current->IsSelected();
d3a9f4af 1488
f2593d0d
RR
1489 if ( TagAllChildrenUntilLast(first,last,select) )
1490 return;
88ac883a 1491
f2593d0d 1492 TagNextChildren(first,last,select);
88ac883a
VZ
1493}
1494
941830cb 1495void wxGenericTreeCtrl::SelectItem(const wxTreeItemId& itemId,
c193b707
VZ
1496 bool unselect_others,
1497 bool extended_select)
d3a9f4af 1498{
223d09f6 1499 wxCHECK_RET( itemId.IsOk(), wxT("invalid tree item") );
8b04a037 1500
88ac883a 1501 bool is_single=!(GetWindowStyleFlag() & wxTR_MULTIPLE);
941830cb 1502 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
88ac883a
VZ
1503
1504 //wxCHECK_RET( ( (!unselect_others) && is_single),
223d09f6 1505 // wxT("this is a single selection tree") );
88ac883a
VZ
1506
1507 // to keep going anyhow !!!
d3a9f4af 1508 if (is_single)
8dc99046
VZ
1509 {
1510 if (item->IsSelected())
1511 return; // nothing to do
1512 unselect_others = TRUE;
1513 extended_select = FALSE;
1514 }
1515 else if ( unselect_others && item->IsSelected() )
1516 {
1517 // selection change if there is more than one item currently selected
1518 wxArrayTreeItemIds selected_items;
1519 if ( GetSelections(selected_items) == 1 )
1520 return;
1521 }
d3a9f4af 1522
f135ff73 1523 wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGING, GetId() );
40fab78b
JS
1524 event.m_item = (long) item;
1525 event.m_itemOld = (long) m_current;
f135ff73 1526 event.SetEventObject( this );
91b8de8d 1527 // TODO : Here we don't send any selection mode yet !
d3a9f4af 1528
f98e2558 1529 if ( GetEventHandler()->ProcessEvent( event ) && !event.IsAllowed() )
618a5e38 1530 return;
f135ff73 1531
2cc78389
RR
1532 wxTreeItemId parent = GetParent( itemId );
1533 while (parent.IsOk())
1534 {
1535 if (!IsExpanded(parent))
1536 Expand( parent );
cdb3cffe 1537
2cc78389
RR
1538 parent = GetParent( parent );
1539 }
cdb3cffe 1540
2cc78389 1541 EnsureVisible( itemId );
cdb3cffe 1542
88ac883a
VZ
1543 // ctrl press
1544 if (unselect_others)
389cdc7a 1545 {
88ac883a 1546 if (is_single) Unselect(); // to speed up thing
c193b707 1547 else UnselectAll();
edaa81ae 1548 }
f135ff73 1549
88ac883a 1550 // shift press
d3a9f4af 1551 if (extended_select)
88ac883a 1552 {
aaa2f297
VZ
1553 if ( !m_current )
1554 {
618a5e38 1555 m_current = m_key_current = (wxGenericTreeItem*) GetRootItem().m_pItem;
aaa2f297
VZ
1556 }
1557
88ac883a
VZ
1558 // don't change the mark (m_current)
1559 SelectItemRange(m_current, item);
1560 }
1561 else
1562 {
c0de7af4 1563 bool select=TRUE; // the default
88ac883a 1564
c193b707
VZ
1565 // Check if we need to toggle hilight (ctrl mode)
1566 if (!unselect_others)
618a5e38 1567 select=!item->IsSelected();
88ac883a 1568
91b8de8d 1569 m_current = m_key_current = item;
c193b707
VZ
1570 m_current->SetHilight(select);
1571 RefreshLine( m_current );
88ac883a 1572 }
389cdc7a 1573
f135ff73 1574 event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGED);
6daa0637 1575 GetEventHandler()->ProcessEvent( event );
389cdc7a
VZ
1576}
1577
941830cb 1578void wxGenericTreeCtrl::FillArray(wxGenericTreeItem *item,
9dfbf520 1579 wxArrayTreeItemIds &array) const
c801d85f 1580{
8dc99046 1581 if ( item->IsSelected() )
9dfbf520 1582 array.Add(wxTreeItemId(item));
91b8de8d 1583
9dfbf520 1584 if ( item->HasChildren() )
91b8de8d 1585 {
9dfbf520
VZ
1586 wxArrayGenericTreeItems& children = item->GetChildren();
1587 size_t count = children.GetCount();
1588 for ( size_t n = 0; n < count; ++n )
f6bcfd97 1589 FillArray(children[n], array);
91b8de8d
RR
1590 }
1591}
1592
941830cb 1593size_t wxGenericTreeCtrl::GetSelections(wxArrayTreeItemIds &array) const
91b8de8d 1594{
618a5e38
RR
1595 array.Empty();
1596 wxTreeItemId idRoot = GetRootItem();
1597 if ( idRoot.IsOk() )
1598 {
1599 FillArray((wxGenericTreeItem*) idRoot.m_pItem, array);
1600 }
1601 //else: the tree is empty, so no selections
91b8de8d 1602
618a5e38 1603 return array.Count();
91b8de8d
RR
1604}
1605
941830cb 1606void wxGenericTreeCtrl::EnsureVisible(const wxTreeItemId& item)
d3a9f4af 1607{
91b8de8d
RR
1608 if (!item.IsOk()) return;
1609
941830cb 1610 wxGenericTreeItem *gitem = (wxGenericTreeItem*) item.m_pItem;
ef44a621 1611
f65635b5
VZ
1612 // first expand all parent branches
1613 wxGenericTreeItem *parent = gitem->GetParent();
5391f772 1614 while ( parent )
f65635b5 1615 {
8dc99046 1616 Expand(parent);
f65635b5
VZ
1617 parent = parent->GetParent();
1618 }
1619
5391f772 1620 //if (parent) CalculatePositions();
91b8de8d
RR
1621
1622 ScrollTo(item);
1623}
1624
941830cb 1625void wxGenericTreeCtrl::ScrollTo(const wxTreeItemId &item)
91b8de8d
RR
1626{
1627 if (!item.IsOk()) return;
1628
dc6c62a9
RR
1629 // We have to call this here because the label in
1630 // question might just have been added and no screen
1631 // update taken place.
cd66f45b 1632 if (m_dirty) wxYieldIfNeeded();
dc6c62a9 1633
941830cb 1634 wxGenericTreeItem *gitem = (wxGenericTreeItem*) item.m_pItem;
91b8de8d 1635
f65635b5 1636 // now scroll to the item
0659e7ee 1637 int item_y = gitem->GetY();
8dc99046 1638
0659e7ee
RR
1639 int start_x = 0;
1640 int start_y = 0;
1e6feb95 1641 GetViewStart( &start_x, &start_y );
91b8de8d 1642 start_y *= PIXELS_PER_UNIT;
978f38c2 1643
a93109d5
RR
1644 int client_h = 0;
1645 int client_w = 0;
1646 GetClientSize( &client_w, &client_h );
ef44a621 1647
0659e7ee
RR
1648 if (item_y < start_y+3)
1649 {
91b8de8d 1650 // going down
0659e7ee
RR
1651 int x = 0;
1652 int y = 0;
91b8de8d
RR
1653 m_anchor->GetSize( x, y, this );
1654 y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
c7a9fa36 1655 x += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
0659e7ee 1656 int x_pos = GetScrollPos( wxHORIZONTAL );
c193b707 1657 // Item should appear at top
91b8de8d 1658 SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, item_y/PIXELS_PER_UNIT );
0659e7ee 1659 }
91b8de8d 1660 else if (item_y+GetLineHeight(gitem) > start_y+client_h)
0659e7ee 1661 {
c7a9fa36
RR
1662 // going up
1663 int x = 0;
1664 int y = 0;
1665 m_anchor->GetSize( x, y, this );
1666 y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1667 x += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
1668 item_y += PIXELS_PER_UNIT+2;
1669 int x_pos = GetScrollPos( wxHORIZONTAL );
c193b707 1670 // Item should appear at bottom
c7a9fa36 1671 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 );
0659e7ee 1672 }
edaa81ae 1673}
c801d85f 1674
e1ee62bd 1675// FIXME: tree sorting functions are not reentrant and not MT-safe!
941830cb 1676static wxGenericTreeCtrl *s_treeBeingSorted = NULL;
0659e7ee 1677
004fd0c8 1678static int LINKAGEMODE tree_ctrl_compare_func(wxGenericTreeItem **item1,
e1ee62bd 1679 wxGenericTreeItem **item2)
edaa81ae 1680{
941830cb 1681 wxCHECK_MSG( s_treeBeingSorted, 0, wxT("bug in wxGenericTreeCtrl::SortChildren()") );
e1ee62bd
VZ
1682
1683 return s_treeBeingSorted->OnCompareItems(*item1, *item2);
0659e7ee
RR
1684}
1685
941830cb 1686int wxGenericTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
e1ee62bd 1687 const wxTreeItemId& item2)
0659e7ee 1688{
87138c52 1689 return wxStrcmp(GetItemText(item1), GetItemText(item2));
e1ee62bd
VZ
1690}
1691
941830cb 1692void wxGenericTreeCtrl::SortChildren(const wxTreeItemId& itemId)
e1ee62bd 1693{
223d09f6 1694 wxCHECK_RET( itemId.IsOk(), wxT("invalid tree item") );
e1ee62bd 1695
941830cb 1696 wxGenericTreeItem *item = (wxGenericTreeItem*) itemId.m_pItem;
978f38c2 1697
e1ee62bd 1698 wxCHECK_RET( !s_treeBeingSorted,
941830cb 1699 wxT("wxGenericTreeCtrl::SortChildren is not reentrant") );
e1ee62bd 1700
91b8de8d 1701 wxArrayGenericTreeItems& children = item->GetChildren();
e1ee62bd
VZ
1702 if ( children.Count() > 1 )
1703 {
618a5e38
RR
1704 m_dirty = TRUE;
1705
e1ee62bd
VZ
1706 s_treeBeingSorted = this;
1707 children.Sort(tree_ctrl_compare_func);
1708 s_treeBeingSorted = NULL;
e1ee62bd
VZ
1709 }
1710 //else: don't make the tree dirty as nothing changed
edaa81ae
RR
1711}
1712
941830cb 1713wxImageList *wxGenericTreeCtrl::GetImageList() const
edaa81ae 1714{
f135ff73 1715 return m_imageListNormal;
edaa81ae
RR
1716}
1717
618a5e38
RR
1718wxImageList *wxGenericTreeCtrl::GetButtonsImageList() const
1719{
1720 return m_imageListButtons;
1721}
1722
941830cb 1723wxImageList *wxGenericTreeCtrl::GetStateImageList() const
c801d85f 1724{
f135ff73 1725 return m_imageListState;
edaa81ae 1726}
c801d85f 1727
618a5e38 1728void wxGenericTreeCtrl::CalculateLineHeight()
e2414cbe 1729{
f2593d0d
RR
1730 wxClientDC dc(this);
1731 m_lineHeight = (int)(dc.GetCharHeight() + 4);
06b466c7 1732
618a5e38
RR
1733 if ( m_imageListNormal )
1734 {
1735 // Calculate a m_lineHeight value from the normal Image sizes.
1736 // May be toggle off. Then wxGenericTreeCtrl will spread when
1737 // necessary (which might look ugly).
1738 int n = m_imageListNormal->GetImageCount();
1739 for (int i = 0; i < n ; i++)
1740 {
1741 int width = 0, height = 0;
1742 m_imageListNormal->GetSize(i, width, height);
1743 if (height > m_lineHeight) m_lineHeight = height;
1744 }
1745 }
1746
1747 if (m_imageListButtons)
f2593d0d 1748 {
618a5e38
RR
1749 // Calculate a m_lineHeight value from the Button image sizes.
1750 // May be toggle off. Then wxGenericTreeCtrl will spread when
1751 // necessary (which might look ugly).
1752 int n = m_imageListButtons->GetImageCount();
1753 for (int i = 0; i < n ; i++)
1754 {
1755 int width = 0, height = 0;
1756 m_imageListButtons->GetSize(i, width, height);
1757 if (height > m_lineHeight) m_lineHeight = height;
1758 }
f2593d0d 1759 }
91b8de8d 1760
618a5e38 1761 if (m_lineHeight < 30)
f2593d0d 1762 m_lineHeight += 2; // at least 2 pixels
06b466c7 1763 else
f2593d0d 1764 m_lineHeight += m_lineHeight/10; // otherwise 10% extra spacing
edaa81ae 1765}
e2414cbe 1766
618a5e38
RR
1767void wxGenericTreeCtrl::SetImageList(wxImageList *imageList)
1768{
1769 if (m_ownsImageListNormal) delete m_imageListNormal;
1770 m_imageListNormal = imageList;
1771 m_ownsImageListNormal = FALSE;
1772 m_dirty = TRUE;
1773 CalculateLineHeight();
1774}
1775
941830cb 1776void wxGenericTreeCtrl::SetStateImageList(wxImageList *imageList)
e2414cbe 1777{
46cd520d 1778 if (m_ownsImageListState) delete m_imageListState;
f135ff73 1779 m_imageListState = imageList;
46cd520d
VS
1780 m_ownsImageListState = FALSE;
1781}
1782
618a5e38
RR
1783void wxGenericTreeCtrl::SetButtonsImageList(wxImageList *imageList)
1784{
1785 if (m_ownsImageListButtons) delete m_imageListButtons;
1786 m_imageListButtons = imageList;
1787 m_ownsImageListButtons = FALSE;
1788 m_dirty = TRUE;
1789 CalculateLineHeight();
1790}
1791
46cd520d
VS
1792void wxGenericTreeCtrl::AssignImageList(wxImageList *imageList)
1793{
1794 SetImageList(imageList);
1795 m_ownsImageListNormal = TRUE;
1796}
1797
1798void wxGenericTreeCtrl::AssignStateImageList(wxImageList *imageList)
1799{
1800 SetStateImageList(imageList);
1801 m_ownsImageListState = TRUE;
edaa81ae 1802}
e2414cbe 1803
618a5e38
RR
1804void wxGenericTreeCtrl::AssignButtonsImageList(wxImageList *imageList)
1805{
1806 SetButtonsImageList(imageList);
1807 m_ownsImageListButtons = TRUE;
1808}
1809
f135ff73
VZ
1810// -----------------------------------------------------------------------------
1811// helpers
1812// -----------------------------------------------------------------------------
0659e7ee 1813
941830cb 1814void wxGenericTreeCtrl::AdjustMyScrollbars()
c801d85f 1815{
0659e7ee
RR
1816 if (m_anchor)
1817 {
618a5e38 1818 int x = 0, y = 0;
91b8de8d 1819 m_anchor->GetSize( x, y, this );
c193b707 1820 y += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
c7a9fa36 1821 x += PIXELS_PER_UNIT+2; // one more scrollbar unit + 2 pixels
0659e7ee
RR
1822 int x_pos = GetScrollPos( wxHORIZONTAL );
1823 int y_pos = GetScrollPos( wxVERTICAL );
91b8de8d 1824 SetScrollbars( PIXELS_PER_UNIT, PIXELS_PER_UNIT, x/PIXELS_PER_UNIT, y/PIXELS_PER_UNIT, x_pos, y_pos );
0659e7ee
RR
1825 }
1826 else
1827 {
1828 SetScrollbars( 0, 0, 0, 0 );
1829 }
edaa81ae 1830}
c801d85f 1831
941830cb 1832int wxGenericTreeCtrl::GetLineHeight(wxGenericTreeItem *item) const
91b8de8d 1833{
dc6c62a9
RR
1834 if (GetWindowStyleFlag() & wxTR_HAS_VARIABLE_ROW_HEIGHT)
1835 return item->GetHeight();
1836 else
1837 return m_lineHeight;
91b8de8d
RR
1838}
1839
941830cb 1840void wxGenericTreeCtrl::PaintItem(wxGenericTreeItem *item, wxDC& dc)
ef44a621 1841{
618a5e38
RR
1842 // TODO implement "state" icon on items
1843
9ec64fa7
VZ
1844 wxTreeItemAttr *attr = item->GetAttributes();
1845 if ( attr && attr->HasFont() )
1846 dc.SetFont(attr->GetFont());
1847 else if (item->IsBold())
eff869aa 1848 dc.SetFont(m_boldFont);
ef44a621 1849
618a5e38 1850 long text_w = 0, text_h = 0;
bbe0af5b 1851 dc.GetTextExtent( item->GetText(), &text_w, &text_h );
ef44a621 1852
618a5e38 1853 int image_h = 0, image_w = 0;
8dc99046
VZ
1854 int image = item->GetCurrentImage();
1855 if ( image != NO_IMAGE )
bbe0af5b 1856 {
2c8e4738
VZ
1857 if ( m_imageListNormal )
1858 {
1859 m_imageListNormal->GetSize( image, image_w, image_h );
1860 image_w += 4;
1861 }
1862 else
1863 {
1864 image = NO_IMAGE;
1865 }
bbe0af5b 1866 }
ef44a621 1867
91b8de8d
RR
1868 int total_h = GetLineHeight(item);
1869
b771aa29 1870 if ( item->IsSelected() )
cdb3cffe 1871 {
b771aa29 1872 dc.SetBrush(*(m_hasFocus ? m_hilightBrush : m_hilightUnfocusedBrush));
cdb3cffe 1873 }
afa6a1a1 1874 else
c0fba4d1
VS
1875 {
1876 wxColour colBg;
1877 if ( attr && attr->HasBackgroundColour() )
1878 colBg = attr->GetBackgroundColour();
1879 else
1880 colBg = m_backgroundColour;
1881 dc.SetBrush(wxBrush(colBg, wxSOLID));
1882 }
afa6a1a1 1883
cdb3cffe 1884 int offset = HasFlag(wxTR_ROW_LINES) ? 1 : 0;
b77d9650 1885
c6f4913a 1886 if ( HasFlag(wxTR_FULL_ROW_HIGHLIGHT) )
a69cb1cc 1887 {
c6f4913a
VS
1888 int x, y, w, h;
1889
1890 DoGetPosition(&x, &y);
1891 DoGetSize(&w, &h);
1892 dc.DrawRectangle(x, item->GetY()+offset, w, total_h-offset);
a69cb1cc
JS
1893 }
1894 else
b77d9650 1895 {
c6f4913a
VS
1896 if ( item->IsSelected() && image != NO_IMAGE )
1897 {
1898 // If it's selected, and there's an image, then we should
1899 // take care to leave the area under the image painted in the
1900 // background colour.
1901 dc.DrawRectangle( item->GetX() + image_w - 2, item->GetY()+offset,
1902 item->GetWidth() - image_w + 2, total_h-offset );
1903 }
1904 else
1905 {
1906 dc.DrawRectangle( item->GetX()-2, item->GetY()+offset,
1907 item->GetWidth()+2, total_h-offset );
1908 }
b77d9650 1909 }
ef44a621 1910
8dc99046 1911 if ( image != NO_IMAGE )
bbe0af5b 1912 {
d30b4d20 1913 dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
8dc99046 1914 m_imageListNormal->Draw( image, dc,
49cd56ef 1915 item->GetX(),
d701d432 1916 item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
bbe0af5b
RR
1917 wxIMAGELIST_DRAW_TRANSPARENT );
1918 dc.DestroyClippingRegion();
1919 }
ef44a621 1920
afa6a1a1 1921 dc.SetBackgroundMode(wxTRANSPARENT);
f6bcfd97
BP
1922 int extraH = (total_h > text_h) ? (total_h - text_h)/2 : 0;
1923 dc.DrawText( item->GetText(),
1924 (wxCoord)(image_w + item->GetX()),
1925 (wxCoord)(item->GetY() + extraH));
ef44a621 1926
eff869aa
RR
1927 // restore normal font
1928 dc.SetFont( m_normalFont );
ef44a621
VZ
1929}
1930
91b8de8d 1931// Now y stands for the top of the item, whereas it used to stand for middle !
941830cb 1932void wxGenericTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
c801d85f 1933{
618a5e38
RR
1934 int x = level*m_indent;
1935 if (!HasFlag(wxTR_HIDE_ROOT))
1936 {
1937 x += m_indent;
1938 }
1939 else if (level == 0)
1940 {
2b84e565
VS
1941 // always expand hidden root
1942 int origY = y;
618a5e38 1943 wxArrayGenericTreeItems& children = item->GetChildren();
2b84e565
VS
1944 int count = children.Count();
1945 if (count > 0)
1946 {
1947 int n = 0, oldY;
1948 do {
1949 oldY = y;
1950 PaintLevel(children[n], dc, 1, y);
1951 } while (++n < count);
1952
e76520fd 1953 if (!HasFlag(wxTR_NO_LINES) && HasFlag(wxTR_LINES_AT_ROOT) && count > 0)
2b84e565
VS
1954 {
1955 // draw line down to last child
1956 origY += GetLineHeight(children[0])>>1;
1957 oldY += GetLineHeight(children[n-1])>>1;
1958 dc.DrawLine(3, origY, 3, oldY);
1959 }
1960 }
618a5e38
RR
1961 return;
1962 }
91b8de8d 1963
618a5e38
RR
1964 item->SetX(x+m_spacing);
1965 item->SetY(y);
389cdc7a 1966
618a5e38
RR
1967 int h = GetLineHeight(item);
1968 int y_top = y;
1969 int y_mid = y_top + (h>>1);
1970 y += h;
4832f7c0 1971
618a5e38
RR
1972 int exposed_x = dc.LogicalToDeviceX(0);
1973 int exposed_y = dc.LogicalToDeviceY(y_top);
233058c7 1974
618a5e38 1975 if (IsExposed(exposed_x, exposed_y, 10000, h)) // 10000 = very much
bbe0af5b 1976 {
c6f4913a
VS
1977 wxPen *pen =
1978#ifndef __WXMAC__
1979 // don't draw rect outline if we already have the
1980 // background color under Mac
1981 (item->IsSelected() && m_hasFocus) ? wxBLACK_PEN :
1982#endif // !__WXMAC__
1983 wxTRANSPARENT_PEN;
1984
1985 wxColour colText;
1986 if ( item->IsSelected() )
1987 {
1988 colText = wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHTTEXT);
1989 }
1990 else
1991 {
1992 wxTreeItemAttr *attr = item->GetAttributes();
1993 if (attr && attr->HasTextColour())
1994 colText = attr->GetTextColour();
1995 else
1996 colText = wxSystemSettings::GetSystemColour(wxSYS_COLOUR_WINDOWTEXT);
1997 }
1998
1999 // prepare to draw
2000 dc.SetTextForeground(colText);
2001 dc.SetPen(*pen);
2002
2003 // draw
2004 PaintItem(item, dc);
2005
2006 if (HasFlag(wxTR_ROW_LINES))
2007 {
2008 // if the background colour is white, choose a
2009 // contrasting color for the lines
2010 dc.SetPen(*((GetBackgroundColour() == *wxWHITE)
2011 ? wxMEDIUM_GREY_PEN : wxWHITE_PEN));
2012 dc.DrawLine(0, y_top, 10000, y_top);
2013 dc.DrawLine(0, y, 10000, y);
2014 }
2015
2016 // restore DC objects
2017 dc.SetBrush(*wxWHITE_BRUSH);
2018 dc.SetPen(m_dottedPen);
2019 dc.SetTextForeground(*wxBLACK);
2020
618a5e38 2021 if (item->HasPlus() && HasButtons()) // should the item show a button?
cdb3cffe 2022 {
618a5e38
RR
2023 if (!HasFlag(wxTR_NO_LINES))
2024 {
2b84e565 2025 if (x > (signed)m_indent)
618a5e38 2026 dc.DrawLine(x - m_indent, y_mid, x - 5, y_mid);
2b84e565
VS
2027 else if (HasFlag(wxTR_LINES_AT_ROOT))
2028 dc.DrawLine(3, y_mid, x - 5, y_mid);
618a5e38
RR
2029 dc.DrawLine(x + 5, y_mid, x + m_spacing, y_mid);
2030 }
2031
2032 if (m_imageListButtons != NULL)
c22886bd 2033 {
618a5e38 2034 // draw the image button here
2b84e565
VS
2035 int image_h = 0, image_w = 0, image = wxTreeItemIcon_Normal;
2036 if (item->IsExpanded()) image = wxTreeItemIcon_Expanded;
618a5e38 2037 if (item->IsSelected())
2b84e565 2038 image += wxTreeItemIcon_Selected - wxTreeItemIcon_Normal;
618a5e38
RR
2039 m_imageListButtons->GetSize(image, image_w, image_h);
2040 int xx = x - (image_w>>1);
2041 int yy = y_mid - (image_h>>1);
2042 dc.SetClippingRegion(xx, yy, image_w, image_h);
2043 m_imageListButtons->Draw(image, dc, xx, yy,
2044 wxIMAGELIST_DRAW_TRANSPARENT);
2045 dc.DestroyClippingRegion();
2046 }
2047 else if (HasFlag(wxTR_TWIST_BUTTONS))
2048 {
2049 // draw the twisty button here
17d5bdf9 2050
c22886bd 2051 wxPoint button[3];
17d5bdf9
RR
2052 dc.SetBrush(*m_hilightBrush);
2053
2054 if (HasFlag(wxTR_AQUA_BUTTONS))
c22886bd 2055 {
17d5bdf9
RR
2056 dc.SetPen(*wxTRANSPARENT_PEN);
2057
2058 if (item->IsExpanded())
2059 {
2060 button[0].x = x-6;
2061 button[0].y = y_mid-2;
2062 button[1].x = x+6;
2063 button[1].y = y_mid-2;
2064 button[2].x = x;
2065 button[2].y = y_mid+7;
2066 }
2067 else
2068 {
2069 button[0].y = y_mid-6;
2070 button[0].x = x-2;
2071 button[1].y = y_mid+6;
2072 button[1].x = x-2;
2073 button[2].y = y_mid;
2074 button[2].x = x+7;
2075 }
c22886bd
RR
2076 }
2077 else
2078 {
17d5bdf9
RR
2079 dc.SetPen(*wxBLACK_PEN);
2080
2081 if (item->IsExpanded())
2082 {
2083 button[0].x = x-5;
2084 button[0].y = y_mid-2;
2085 button[1].x = x+5;
2086 button[1].y = y_mid-2;
2087 button[2].x = x;
2088 button[2].y = y_mid+3;
2089 }
2090 else
2091 {
2092 button[0].y = y_mid-5;
2093 button[0].x = x-2;
2094 button[1].y = y_mid+5;
2095 button[1].x = x-2;
2096 button[2].y = y_mid;
2097 button[2].x = x+3;
2098 }
c22886bd 2099 }
618a5e38 2100 dc.DrawPolygon(3, button);
cdb3cffe 2101
618a5e38 2102 dc.SetPen(m_dottedPen);
c22886bd 2103 }
618a5e38 2104 else // if (HasFlag(wxTR_HAS_BUTTONS))
c22886bd 2105 {
618a5e38
RR
2106 // draw the plus sign here
2107 dc.SetPen(*wxGREY_PEN);
2108 dc.SetBrush(*wxWHITE_BRUSH);
2109 dc.DrawRectangle(x-5, y_mid-4, 11, 9);
2110 dc.SetPen(*wxBLACK_PEN);
2111 dc.DrawLine(x-2, y_mid, x+3, y_mid);
c22886bd 2112 if (!item->IsExpanded())
618a5e38
RR
2113 dc.DrawLine(x, y_mid-2, x, y_mid+3);
2114 dc.SetPen(m_dottedPen);
c22886bd 2115 }
bbe0af5b 2116 }
618a5e38
RR
2117 else if (!HasFlag(wxTR_NO_LINES)) // no button; maybe a line?
2118 {
2119 // draw the horizontal line here
2120 int x_start = x;
2b84e565 2121 if (x > (signed)m_indent)
618a5e38 2122 x_start -= m_indent;
2b84e565
VS
2123 else if (HasFlag(wxTR_LINES_AT_ROOT))
2124 x_start = 3;
618a5e38
RR
2125 dc.DrawLine(x_start, y_mid, x + m_spacing, y_mid);
2126 }
f135ff73 2127 }
d3a9f4af 2128
bbe0af5b
RR
2129 if (item->IsExpanded())
2130 {
91b8de8d 2131 wxArrayGenericTreeItems& children = item->GetChildren();
618a5e38 2132 int count = children.Count();
f65635b5 2133 if (count > 0)
9ec64fa7 2134 {
618a5e38
RR
2135 int n = 0, oldY;
2136 ++level;
2137 do {
2138 oldY = y;
2139 PaintLevel(children[n], dc, level, y);
2140 } while (++n < count);
2141
e76520fd 2142 if (!HasFlag(wxTR_NO_LINES) && count > 0)
618a5e38
RR
2143 {
2144 // draw line down to last child
2b84e565 2145 oldY += GetLineHeight(children[n-1])>>1;
618a5e38
RR
2146 if (HasButtons()) y_mid += 5;
2147 dc.DrawLine(x, y_mid, x, oldY);
2148 }
9ec64fa7 2149 }
bbe0af5b 2150 }
4c681997 2151}
c801d85f 2152
941830cb 2153void wxGenericTreeCtrl::DrawDropEffect(wxGenericTreeItem *item)
3dbeaa52
VZ
2154{
2155 if ( item )
2156 {
2157 if ( item->HasPlus() )
2158 {
2159 // it's a folder, indicate it by a border
2160 DrawBorder(item);
2161 }
2162 else
2163 {
2164 // draw a line under the drop target because the item will be
2165 // dropped there
2166 DrawLine(item, TRUE /* below */);
2167 }
2168
2169 SetCursor(wxCURSOR_BULLSEYE);
2170 }
2171 else
2172 {
2173 // can't drop here
2174 SetCursor(wxCURSOR_NO_ENTRY);
2175 }
2176}
2177
941830cb 2178void wxGenericTreeCtrl::DrawBorder(const wxTreeItemId &item)
91b8de8d 2179{
941830cb 2180 wxCHECK_RET( item.IsOk(), _T("invalid item in wxGenericTreeCtrl::DrawLine") );
91b8de8d 2181
941830cb 2182 wxGenericTreeItem *i = (wxGenericTreeItem*) item.m_pItem;
91b8de8d 2183
1044a386 2184 wxClientDC dc(this);
91b8de8d
RR
2185 PrepareDC( dc );
2186 dc.SetLogicalFunction(wxINVERT);
3dbeaa52 2187 dc.SetBrush(*wxTRANSPARENT_BRUSH);
91b8de8d 2188
3dbeaa52
VZ
2189 int w = i->GetWidth() + 2;
2190 int h = GetLineHeight(i) + 2;
91b8de8d 2191
3dbeaa52 2192 dc.DrawRectangle( i->GetX() - 1, i->GetY() - 1, w, h);
91b8de8d
RR
2193}
2194
941830cb 2195void wxGenericTreeCtrl::DrawLine(const wxTreeItemId &item, bool below)
91b8de8d 2196{
941830cb 2197 wxCHECK_RET( item.IsOk(), _T("invalid item in wxGenericTreeCtrl::DrawLine") );
91b8de8d 2198
941830cb 2199 wxGenericTreeItem *i = (wxGenericTreeItem*) item.m_pItem;
91b8de8d 2200
1044a386 2201 wxClientDC dc(this);
91b8de8d
RR
2202 PrepareDC( dc );
2203 dc.SetLogicalFunction(wxINVERT);
d3a9f4af 2204
3dbeaa52
VZ
2205 int x = i->GetX(),
2206 y = i->GetY();
2207 if ( below )
2208 {
2209 y += GetLineHeight(i) - 1;
2210 }
91b8de8d 2211
3dbeaa52 2212 dc.DrawLine( x, y, x + i->GetWidth(), y);
91b8de8d
RR
2213}
2214
f135ff73
VZ
2215// -----------------------------------------------------------------------------
2216// wxWindows callbacks
2217// -----------------------------------------------------------------------------
2218
941830cb 2219void wxGenericTreeCtrl::OnPaint( wxPaintEvent &WXUNUSED(event) )
c801d85f 2220{
0659e7ee
RR
2221 wxPaintDC dc(this);
2222 PrepareDC( dc );
29d87bba 2223
941830cb
JS
2224 if ( !m_anchor)
2225 return;
2226
eff869aa 2227 dc.SetFont( m_normalFont );
0659e7ee 2228 dc.SetPen( m_dottedPen );
f38374d0 2229
eff869aa 2230 // this is now done dynamically
91b8de8d
RR
2231 //if(GetImageList() == NULL)
2232 // m_lineHeight = (int)(dc.GetCharHeight() + 4);
29d87bba 2233
91b8de8d 2234 int y = 2;
0659e7ee 2235 PaintLevel( m_anchor, dc, 0, y );
edaa81ae 2236}
c801d85f 2237
b771aa29 2238void wxGenericTreeCtrl::OnSetFocus( wxFocusEvent &event )
c801d85f 2239{
0659e7ee 2240 m_hasFocus = TRUE;
978f38c2 2241
b771aa29
VZ
2242 RefreshSelected();
2243
2244 event.Skip();
edaa81ae 2245}
c801d85f 2246
b771aa29 2247void wxGenericTreeCtrl::OnKillFocus( wxFocusEvent &event )
c801d85f 2248{
0659e7ee 2249 m_hasFocus = FALSE;
978f38c2 2250
b771aa29
VZ
2251 RefreshSelected();
2252
2253 event.Skip();
edaa81ae 2254}
c801d85f 2255
941830cb 2256void wxGenericTreeCtrl::OnChar( wxKeyEvent &event )
c801d85f 2257{
978f38c2 2258 wxTreeEvent te( wxEVT_COMMAND_TREE_KEY_DOWN, GetId() );
b09bda68 2259 te.m_evtKey = event;
978f38c2 2260 te.SetEventObject( this );
68eb5f63
VZ
2261 if ( GetEventHandler()->ProcessEvent( te ) )
2262 {
2263 // intercepted by the user code
2264 return;
2265 }
435fe83e 2266
91b8de8d 2267 if ( (m_current == 0) || (m_key_current == 0) )
978f38c2
VZ
2268 {
2269 event.Skip();
2270 return;
2271 }
ef44a621 2272
06b466c7
VZ
2273 // how should the selection work for this event?
2274 bool is_multiple, extended_select, unselect_others;
2275 EventFlagsToSelType(GetWindowStyleFlag(),
2276 event.ShiftDown(),
2277 event.ControlDown(),
dfc6cd93
SB
2278 is_multiple, extended_select, unselect_others);
2279
2280 // + : Expand
2281 // - : Collaspe
f6bcfd97 2282 // * : Expand all/Collapse all
dfc6cd93
SB
2283 // ' ' | return : activate
2284 // up : go up (not last children!)
2285 // down : go down
2286 // left : go to parent
2287 // right : open if parent and go next
2288 // home : go to root
2289 // end : go to last item without opening parents
978f38c2
VZ
2290 switch (event.KeyCode())
2291 {
2292 case '+':
2293 case WXK_ADD:
2294 if (m_current->HasPlus() && !IsExpanded(m_current))
2295 {
2296 Expand(m_current);
2297 }
2298 break;
ef44a621 2299
f6bcfd97
BP
2300 case '*':
2301 case WXK_MULTIPLY:
2302 if ( !IsExpanded(m_current) )
2303 {
2304 // expand all
2305 ExpandAll(m_current);
2306 break;
2307 }
2308 //else: fall through to Collapse() it
2309
978f38c2
VZ
2310 case '-':
2311 case WXK_SUBTRACT:
2312 if (IsExpanded(m_current))
2313 {
2314 Collapse(m_current);
2315 }
2316 break;
ef44a621 2317
978f38c2
VZ
2318 case ' ':
2319 case WXK_RETURN:
2320 {
2321 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
40fab78b 2322 event.m_item = (long) m_current;
978f38c2
VZ
2323 event.SetEventObject( this );
2324 GetEventHandler()->ProcessEvent( event );
2325 }
2326 break;
ef44a621 2327
618a5e38
RR
2328 // up goes to the previous sibling or to the last
2329 // of its children if it's expanded
978f38c2
VZ
2330 case WXK_UP:
2331 {
91b8de8d 2332 wxTreeItemId prev = GetPrevSibling( m_key_current );
978f38c2
VZ
2333 if (!prev)
2334 {
91b8de8d 2335 prev = GetParent( m_key_current );
618a5e38
RR
2336 if ((prev == GetRootItem()) && HasFlag(wxTR_HIDE_ROOT))
2337 {
2338 break; // don't go to root if it is hidden
2339 }
c193b707
VZ
2340 if (prev)
2341 {
618a5e38 2342 long cookie = 0;
91b8de8d 2343 wxTreeItemId current = m_key_current;
618a5e38
RR
2344 // TODO: Huh? If we get here, we'd better be the first child of our parent. How else could it be?
2345 if (current == GetFirstChild( prev, cookie ))
90e58684
RR
2346 {
2347 // otherwise we return to where we came from
88ac883a 2348 SelectItem( prev, unselect_others, extended_select );
941830cb 2349 m_key_current= (wxGenericTreeItem*) prev.m_pItem;
c193b707 2350 EnsureVisible( prev );
90e58684 2351 break;
c193b707 2352 }
978f38c2
VZ
2353 }
2354 }
2355 if (prev)
2356 {
69a282d4 2357 while ( IsExpanded(prev) && HasChildren(prev) )
978f38c2 2358 {
69a282d4
VZ
2359 wxTreeItemId child = GetLastChild(prev);
2360 if ( child )
2361 {
2362 prev = child;
2363 }
978f38c2 2364 }
69a282d4 2365
88ac883a 2366 SelectItem( prev, unselect_others, extended_select );
941830cb 2367 m_key_current=(wxGenericTreeItem*) prev.m_pItem;
978f38c2
VZ
2368 EnsureVisible( prev );
2369 }
2370 }
2371 break;
ef44a621 2372
978f38c2
VZ
2373 // left arrow goes to the parent
2374 case WXK_LEFT:
2375 {
2376 wxTreeItemId prev = GetParent( m_current );
618a5e38
RR
2377 if ((prev == GetRootItem()) && HasFlag(wxTR_HIDE_ROOT))
2378 {
2379 // don't go to root if it is hidden
2380 prev = GetPrevSibling( m_current );
2381 }
978f38c2
VZ
2382 if (prev)
2383 {
2384 EnsureVisible( prev );
88ac883a 2385 SelectItem( prev, unselect_others, extended_select );
978f38c2
VZ
2386 }
2387 }
2388 break;
ef44a621 2389
978f38c2 2390 case WXK_RIGHT:
618a5e38
RR
2391 // this works the same as the down arrow except that we
2392 // also expand the item if it wasn't expanded yet
978f38c2
VZ
2393 Expand(m_current);
2394 // fall through
2395
2396 case WXK_DOWN:
ef44a621 2397 {
91b8de8d 2398 if (IsExpanded(m_key_current) && HasChildren(m_key_current))
978f38c2
VZ
2399 {
2400 long cookie = 0;
91b8de8d 2401 wxTreeItemId child = GetFirstChild( m_key_current, cookie );
88ac883a 2402 SelectItem( child, unselect_others, extended_select );
941830cb 2403 m_key_current=(wxGenericTreeItem*) child.m_pItem;
978f38c2
VZ
2404 EnsureVisible( child );
2405 }
2406 else
2407 {
91b8de8d 2408 wxTreeItemId next = GetNextSibling( m_key_current );
b62c3631 2409 if (!next)
978f38c2 2410 {
91b8de8d 2411 wxTreeItemId current = m_key_current;
978f38c2
VZ
2412 while (current && !next)
2413 {
2414 current = GetParent( current );
2415 if (current) next = GetNextSibling( current );
2416 }
2417 }
b62c3631 2418 if (next)
978f38c2 2419 {
88ac883a 2420 SelectItem( next, unselect_others, extended_select );
941830cb 2421 m_key_current=(wxGenericTreeItem*) next.m_pItem;
978f38c2
VZ
2422 EnsureVisible( next );
2423 }
2424 }
ef44a621 2425 }
978f38c2 2426 break;
ef44a621 2427
978f38c2
VZ
2428 // <End> selects the last visible tree item
2429 case WXK_END:
2430 {
2431 wxTreeItemId last = GetRootItem();
2432
2433 while ( last.IsOk() && IsExpanded(last) )
2434 {
2435 wxTreeItemId lastChild = GetLastChild(last);
2436
2437 // it may happen if the item was expanded but then all of
2438 // its children have been deleted - so IsExpanded() returned
2439 // TRUE, but GetLastChild() returned invalid item
2440 if ( !lastChild )
2441 break;
2442
2443 last = lastChild;
2444 }
2445
2446 if ( last.IsOk() )
2447 {
2448 EnsureVisible( last );
88ac883a 2449 SelectItem( last, unselect_others, extended_select );
978f38c2
VZ
2450 }
2451 }
2452 break;
2453
2454 // <Home> selects the root item
2455 case WXK_HOME:
2456 {
2457 wxTreeItemId prev = GetRootItem();
618a5e38
RR
2458 if (!prev) break;
2459 if (HasFlag(wxTR_HIDE_ROOT))
978f38c2 2460 {
618a5e38
RR
2461 long dummy;
2462 prev = GetFirstChild(prev, dummy);
2463 if (!prev) break;
978f38c2 2464 }
618a5e38
RR
2465 EnsureVisible( prev );
2466 SelectItem( prev, unselect_others, extended_select );
978f38c2
VZ
2467 }
2468 break;
2469
2470 default:
2471 event.Skip();
2472 }
edaa81ae 2473}
c801d85f 2474
941830cb 2475wxTreeItemId wxGenericTreeCtrl::HitTest(const wxPoint& point, int& flags)
4f22cf8d 2476{
618a5e38
RR
2477 // JACS: removed wxYieldIfNeeded() because it can cause the window
2478 // to be deleted from under us if a close window event is pending
dc6c62a9 2479
91b8de8d
RR
2480 int w, h;
2481 GetSize(&w, &h);
91b8de8d 2482 flags=0;
618a5e38
RR
2483 if (point.x<0) flags |= wxTREE_HITTEST_TOLEFT;
2484 if (point.x>w) flags |= wxTREE_HITTEST_TORIGHT;
2485 if (point.y<0) flags |= wxTREE_HITTEST_ABOVE;
2486 if (point.y>h) flags |= wxTREE_HITTEST_BELOW;
2487 if (flags) return wxTreeItemId();
d3a9f4af 2488
618a5e38
RR
2489 if (m_anchor == NULL)
2490 {
2491 flags = wxTREE_HITTEST_NOWHERE;
2492 return wxTreeItemId();
2493 }
5716a1ab 2494
618a5e38
RR
2495 wxClientDC dc(this);
2496 PrepareDC(dc);
2497 wxCoord x = dc.DeviceToLogicalX( point.x );
2498 wxCoord y = dc.DeviceToLogicalY( point.y );
2499 wxGenericTreeItem *hit = m_anchor->HitTest(wxPoint(x, y),
2500 this,
2501 flags,
2502 0 );
2503 if (hit == NULL)
2504 {
2505 flags = wxTREE_HITTEST_NOWHERE;
2506 return wxTreeItemId();
2507 }
2508 return hit;
4f22cf8d
RR
2509}
2510
8fb3bfe2
JS
2511// get the bounding rectangle of the item (or of its label only)
2512bool wxGenericTreeCtrl::GetBoundingRect(const wxTreeItemId& item,
2513 wxRect& rect,
33ac7e6f 2514 bool WXUNUSED(textOnly)) const
8fb3bfe2 2515{
601c163b 2516 wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid item in wxGenericTreeCtrl::GetBoundingRect") );
8fb3bfe2
JS
2517
2518 wxGenericTreeItem *i = (wxGenericTreeItem*) item.m_pItem;
2519
2520 int startX, startY;
2521 GetViewStart(& startX, & startY);
2522
1ed01484 2523 rect.x = i->GetX() - startX*PIXELS_PER_UNIT;
c22886bd 2524 rect.y = i->GetY() - startY*PIXELS_PER_UNIT;
1ed01484 2525 rect.width = i->GetWidth();
c22886bd
RR
2526 //rect.height = i->GetHeight();
2527 rect.height = GetLineHeight(i);
8fb3bfe2
JS
2528
2529 return TRUE;
8fb3bfe2
JS
2530}
2531
e179bd65
RR
2532/* **** */
2533
941830cb 2534void wxGenericTreeCtrl::Edit( const wxTreeItemId& item )
e179bd65
RR
2535{
2536 if (!item.IsOk()) return;
2537
941830cb 2538 m_currentEdit = (wxGenericTreeItem*) item.m_pItem;
9dfbf520 2539
e179bd65 2540 wxTreeEvent te( wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT, GetId() );
40fab78b 2541 te.m_item = (long) m_currentEdit;
e179bd65
RR
2542 te.SetEventObject( this );
2543 GetEventHandler()->ProcessEvent( te );
2544
2545 if (!te.IsAllowed()) return;
8dc99046 2546
dc6c62a9
RR
2547 // We have to call this here because the label in
2548 // question might just have been added and no screen
2549 // update taken place.
cd66f45b 2550 if (m_dirty) wxYieldIfNeeded();
9dfbf520 2551
e179bd65
RR
2552 wxString s = m_currentEdit->GetText();
2553 int x = m_currentEdit->GetX();
2554 int y = m_currentEdit->GetY();
2555 int w = m_currentEdit->GetWidth();
2556 int h = m_currentEdit->GetHeight();
9dfbf520 2557
5f1ea0ee
RR
2558 int image_h = 0;
2559 int image_w = 0;
8dc99046
VZ
2560
2561 int image = m_currentEdit->GetCurrentImage();
2562 if ( image != NO_IMAGE )
5f1ea0ee 2563 {
2c8e4738
VZ
2564 if ( m_imageListNormal )
2565 {
2566 m_imageListNormal->GetSize( image, image_w, image_h );
2567 image_w += 4;
2568 }
6359fa0e 2569 else
2c8e4738
VZ
2570 {
2571 wxFAIL_MSG(_T("you must create an image list to use images!"));
2572 }
5f1ea0ee
RR
2573 }
2574 x += image_w;
2575 w -= image_w + 4; // I don't know why +4 is needed
e179bd65
RR
2576
2577 wxClientDC dc(this);
2578 PrepareDC( dc );
2579 x = dc.LogicalToDeviceX( x );
2580 y = dc.LogicalToDeviceY( y );
2581
6359fa0e
VZ
2582 wxTreeTextCtrl *text = new wxTreeTextCtrl(this, -1,
2583 &m_renameAccept,
2584 &m_renameRes,
2585 this,
2586 s,
2587 wxPoint(x-4,y-4),
2588 wxSize(w+11,h+8));
e179bd65
RR
2589 text->SetFocus();
2590}
2591
941830cb 2592void wxGenericTreeCtrl::OnRenameTimer()
e179bd65
RR
2593{
2594 Edit( m_current );
2595}
2596
941830cb 2597void wxGenericTreeCtrl::OnRenameAccept()
e179bd65 2598{
618a5e38 2599 // TODO if the validator fails this causes a crash
e179bd65 2600 wxTreeEvent le( wxEVT_COMMAND_TREE_END_LABEL_EDIT, GetId() );
40fab78b 2601 le.m_item = (long) m_currentEdit;
e179bd65
RR
2602 le.SetEventObject( this );
2603 le.m_label = m_renameRes;
2604 GetEventHandler()->ProcessEvent( le );
9dfbf520 2605
e179bd65 2606 if (!le.IsAllowed()) return;
9dfbf520 2607
5f1ea0ee 2608 SetItemText( m_currentEdit, m_renameRes );
e179bd65 2609}
9dfbf520 2610
941830cb 2611void wxGenericTreeCtrl::OnMouse( wxMouseEvent &event )
c801d85f 2612{
bbe0af5b 2613 if ( !m_anchor ) return;
978f38c2 2614
3dbeaa52
VZ
2615 // we process left mouse up event (enables in-place edit), right down
2616 // (pass to the user code), left dbl click (activate item) and
2617 // dragging/moving events for items drag-and-drop
f6bcfd97
BP
2618 if ( !(event.LeftDown() ||
2619 event.LeftUp() ||
3dbeaa52
VZ
2620 event.RightDown() ||
2621 event.LeftDClick() ||
2622 event.Dragging() ||
2623 ((event.Moving() || event.RightUp()) && m_isDragging)) )
2624 {
2625 event.Skip();
2626
2627 return;
2628 }
2629
bbe0af5b
RR
2630 wxClientDC dc(this);
2631 PrepareDC(dc);
06b466c7
VZ
2632 wxCoord x = dc.DeviceToLogicalX( event.GetX() );
2633 wxCoord y = dc.DeviceToLogicalY( event.GetY() );
29d87bba 2634
3dbeaa52 2635 int flags = 0;
618a5e38
RR
2636 wxGenericTreeItem *item = m_anchor->HitTest( wxPoint(x,y),
2637 this,
2638 flags,
2639 0 );
3dbeaa52 2640
3dbeaa52 2641 if ( event.Dragging() && !m_isDragging )
bbe0af5b 2642 {
fd9811b1 2643 if (m_dragCount == 0)
8dc99046
VZ
2644 m_dragStart = wxPoint(x,y);
2645
fd9811b1 2646 m_dragCount++;
8dc99046 2647
3dbeaa52
VZ
2648 if (m_dragCount != 3)
2649 {
2650 // wait until user drags a bit further...
2651 return;
2652 }
8dc99046 2653
3dbeaa52
VZ
2654 wxEventType command = event.RightIsDown()
2655 ? wxEVT_COMMAND_TREE_BEGIN_RDRAG
2656 : wxEVT_COMMAND_TREE_BEGIN_DRAG;
8dc99046 2657
fd9811b1 2658 wxTreeEvent nevent( command, GetId() );
40fab78b 2659 nevent.m_item = (long) m_current;
fd9811b1 2660 nevent.SetEventObject(this);
3dbeaa52
VZ
2661
2662 // by default the dragging is not supported, the user code must
2663 // explicitly allow the event for it to take place
2664 nevent.Veto();
2665
2666 if ( GetEventHandler()->ProcessEvent(nevent) && nevent.IsAllowed() )
2667 {
2668 // we're going to drag this item
2669 m_isDragging = TRUE;
2670
b3a7510d
VZ
2671 // remember the old cursor because we will change it while
2672 // dragging
2673 m_oldCursor = m_cursor;
2674
2675 // in a single selection control, hide the selection temporarily
2676 if ( !(GetWindowStyleFlag() & wxTR_MULTIPLE) )
2677 {
941830cb 2678 m_oldSelection = (wxGenericTreeItem*) GetSelection().m_pItem;
b3a7510d
VZ
2679
2680 if ( m_oldSelection )
2681 {
2682 m_oldSelection->SetHilight(FALSE);
2683 RefreshLine(m_oldSelection);
2684 }
2685 }
2686
3dbeaa52
VZ
2687 CaptureMouse();
2688 }
bbe0af5b 2689 }
3dbeaa52 2690 else if ( event.Moving() )
fd9811b1 2691 {
3dbeaa52
VZ
2692 if ( item != m_dropTarget )
2693 {
2694 // unhighlight the previous drop target
2695 DrawDropEffect(m_dropTarget);
fd9811b1 2696
3dbeaa52 2697 m_dropTarget = item;
978f38c2 2698
3dbeaa52
VZ
2699 // highlight the current drop target if any
2700 DrawDropEffect(m_dropTarget);
fb882e1c 2701
cd66f45b 2702 wxYieldIfNeeded();
3dbeaa52 2703 }
e179bd65 2704 }
3dbeaa52
VZ
2705 else if ( (event.LeftUp() || event.RightUp()) && m_isDragging )
2706 {
2707 // erase the highlighting
2708 DrawDropEffect(m_dropTarget);
9dfbf520 2709
4f5fffcc
RR
2710 if ( m_oldSelection )
2711 {
2712 m_oldSelection->SetHilight(TRUE);
2713 RefreshLine(m_oldSelection);
2714 m_oldSelection = (wxGenericTreeItem *)NULL;
2715 }
2716
3dbeaa52
VZ
2717 // generate the drag end event
2718 wxTreeEvent event(wxEVT_COMMAND_TREE_END_DRAG, GetId());
88ac883a 2719
40fab78b 2720 event.m_item = (long) item;
3dbeaa52
VZ
2721 event.m_pointDrag = wxPoint(x, y);
2722 event.SetEventObject(this);
91b8de8d 2723
3dbeaa52 2724 (void)GetEventHandler()->ProcessEvent(event);
29d87bba 2725
3dbeaa52
VZ
2726 m_isDragging = FALSE;
2727 m_dropTarget = (wxGenericTreeItem *)NULL;
2728
2729 ReleaseMouse();
2730
b3a7510d 2731 SetCursor(m_oldCursor);
3dbeaa52 2732
cd66f45b 2733 wxYieldIfNeeded();
3dbeaa52
VZ
2734 }
2735 else
bbe0af5b 2736 {
3dbeaa52
VZ
2737 // here we process only the messages which happen on tree items
2738
2739 m_dragCount = 0;
2740
2741 if (item == NULL) return; /* we hit the blank area */
2742
2743 if ( event.RightDown() )
2744 {
2745 wxTreeEvent nevent(wxEVT_COMMAND_TREE_ITEM_RIGHT_CLICK, GetId());
40fab78b 2746 nevent.m_item = (long) item;
05a7f61d
VZ
2747 CalcScrolledPosition(x, y,
2748 &nevent.m_pointDrag.x,
2749 &nevent.m_pointDrag.y);
3dbeaa52
VZ
2750 nevent.SetEventObject(this);
2751 GetEventHandler()->ProcessEvent(nevent);
2752 }
80d2803f 2753 else if ( event.LeftUp() )
f6bcfd97 2754 {
80d2803f 2755 if ( m_lastOnSame )
f6bcfd97 2756 {
80d2803f
VZ
2757 if ( (item == m_current) &&
2758 (flags & wxTREE_HITTEST_ONITEMLABEL) &&
2759 HasFlag(wxTR_EDIT_LABELS) )
2760 {
bdb310a7
VZ
2761 if ( m_renameTimer->IsRunning() )
2762 m_renameTimer->Stop();
2763
80d2803f
VZ
2764 m_renameTimer->Start( 100, TRUE );
2765 }
f6bcfd97 2766
80d2803f
VZ
2767 m_lastOnSame = FALSE;
2768 }
3dbeaa52 2769 }
0326c494 2770 else // !RightDown() && !LeftUp() ==> LeftDown() || LeftDClick()
3dbeaa52 2771 {
f6bcfd97
BP
2772 if ( event.LeftDown() )
2773 {
2774 m_lastOnSame = item == m_current;
2775 }
2776
0326c494
VZ
2777 if ( flags & wxTREE_HITTEST_ONITEMBUTTON )
2778 {
2779 // only toggle the item for a single click, double click on
2780 // the button doesn't do anything (it toggles the item twice)
2781 if ( event.LeftDown() )
2782 {
2783 Toggle( item );
2784 }
2785
2786 // don't select the item if the button was clicked
2787 return;
2788 }
2789
3dbeaa52
VZ
2790 // how should the selection work for this event?
2791 bool is_multiple, extended_select, unselect_others;
2792 EventFlagsToSelType(GetWindowStyleFlag(),
2793 event.ShiftDown(),
2794 event.ControlDown(),
dfc6cd93 2795 is_multiple, extended_select, unselect_others);
3dbeaa52 2796
3dbeaa52
VZ
2797 SelectItem(item, unselect_others, extended_select);
2798
618a5e38
RR
2799 // For some reason, Windows isn't recognizing a left double-click,
2800 // so we need to simulate it here. Allow 200 milliseconds for now.
3dbeaa52
VZ
2801 if ( event.LeftDClick() )
2802 {
f6bcfd97
BP
2803 // double clicking should not start editing the item label
2804 m_renameTimer->Stop();
2805 m_lastOnSame = FALSE;
2806
ebb987b7
VZ
2807 // send activate event first
2808 wxTreeEvent nevent( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
2809 nevent.m_item = (long) item;
ebb987b7
VZ
2810 CalcScrolledPosition(x, y,
2811 &nevent.m_pointDrag.x,
2812 &nevent.m_pointDrag.y);
2813 nevent.SetEventObject( this );
2814 if ( !GetEventHandler()->ProcessEvent( nevent ) )
618a5e38 2815 {
ebb987b7
VZ
2816 // if the user code didn't process the activate event,
2817 // handle it ourselves by toggling the item when it is
2818 // double clicked
2819 if ( item->HasPlus() )
2820 {
2821 Toggle(item);
2822 }
618a5e38 2823 }
3dbeaa52
VZ
2824 }
2825 }
bbe0af5b 2826 }
edaa81ae 2827}
c801d85f 2828
941830cb 2829void wxGenericTreeCtrl::OnIdle( wxIdleEvent &WXUNUSED(event) )
3db7be80 2830{
bbe0af5b
RR
2831 /* after all changes have been done to the tree control,
2832 * we actually redraw the tree when everything is over */
ef44a621 2833
618a5e38 2834 if (!m_dirty) return;
ef44a621 2835
bbe0af5b 2836 m_dirty = FALSE;
3db7be80 2837
bbe0af5b 2838 CalculatePositions();
91b8de8d 2839 Refresh();
bbe0af5b 2840 AdjustMyScrollbars();
3db7be80
RR
2841}
2842
941830cb 2843void wxGenericTreeCtrl::CalculateSize( wxGenericTreeItem *item, wxDC &dc )
d3a9f4af 2844{
e06b9569
JS
2845 wxCoord text_w = 0;
2846 wxCoord text_h = 0;
9dfbf520 2847
a62867a5 2848 if (item->IsBold())
eff869aa 2849 dc.SetFont(m_boldFont);
9dfbf520 2850
91b8de8d 2851 dc.GetTextExtent( item->GetText(), &text_w, &text_h );
a62867a5 2852 text_h+=2;
91b8de8d 2853
eff869aa
RR
2854 // restore normal font
2855 dc.SetFont( m_normalFont );
9dfbf520 2856
91b8de8d
RR
2857 int image_h = 0;
2858 int image_w = 0;
8dc99046
VZ
2859 int image = item->GetCurrentImage();
2860 if ( image != NO_IMAGE )
91b8de8d 2861 {
2c8e4738
VZ
2862 if ( m_imageListNormal )
2863 {
2864 m_imageListNormal->GetSize( image, image_w, image_h );
2865 image_w += 4;
2866 }
91b8de8d
RR
2867 }
2868
2869 int total_h = (image_h > text_h) ? image_h : text_h;
2870
618a5e38 2871 if (total_h < 30)
f2593d0d 2872 total_h += 2; // at least 2 pixels
06b466c7 2873 else
f2593d0d 2874 total_h += total_h/10; // otherwise 10% extra spacing
91b8de8d
RR
2875
2876 item->SetHeight(total_h);
6cedba09
VZ
2877 if (total_h>m_lineHeight)
2878 m_lineHeight=total_h;
bbe0af5b 2879
91b8de8d
RR
2880 item->SetWidth(image_w+text_w+2);
2881}
2882
2883// -----------------------------------------------------------------------------
2884// for developper : y is now the top of the level
2885// not the middle of it !
941830cb 2886void wxGenericTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
c801d85f 2887{
618a5e38
RR
2888 int x = level*m_indent;
2889 if (!HasFlag(wxTR_HIDE_ROOT))
2890 {
2891 x += m_indent;
2892 }
2893 else if (level == 0)
2894 {
2895 // a hidden root is not evaluated, but its
2896 // children are always calculated
2897 goto Recurse;
2898 }
389cdc7a 2899
91b8de8d 2900 CalculateSize( item, dc );
d3a9f4af 2901
91b8de8d 2902 // set its position
618a5e38 2903 item->SetX( x+m_spacing );
91b8de8d 2904 item->SetY( y );
618a5e38 2905 y += GetLineHeight(item);
4c681997 2906
bbe0af5b
RR
2907 if ( !item->IsExpanded() )
2908 {
618a5e38 2909 // we don't need to calculate collapsed branches
bbe0af5b
RR
2910 return;
2911 }
389cdc7a 2912
618a5e38 2913 Recurse:
91b8de8d
RR
2914 wxArrayGenericTreeItems& children = item->GetChildren();
2915 size_t n, count = children.Count();
618a5e38 2916 ++level;
91b8de8d 2917 for (n = 0; n < count; ++n )
618a5e38 2918 CalculateLevel( children[n], dc, level, y ); // recurse
edaa81ae 2919}
c801d85f 2920
941830cb 2921void wxGenericTreeCtrl::CalculatePositions()
c801d85f 2922{
bbe0af5b 2923 if ( !m_anchor ) return;
29d87bba 2924
bbe0af5b
RR
2925 wxClientDC dc(this);
2926 PrepareDC( dc );
29d87bba 2927
eff869aa 2928 dc.SetFont( m_normalFont );
29d87bba 2929
bbe0af5b 2930 dc.SetPen( m_dottedPen );
91b8de8d
RR
2931 //if(GetImageList() == NULL)
2932 // m_lineHeight = (int)(dc.GetCharHeight() + 4);
29d87bba 2933
8dc99046 2934 int y = 2;
f65635b5 2935 CalculateLevel( m_anchor, dc, 0, y ); // start recursion
edaa81ae 2936}
c801d85f 2937
941830cb 2938void wxGenericTreeCtrl::RefreshSubtree(wxGenericTreeItem *item)
c801d85f 2939{
e6527f9d
RR
2940 if (m_dirty) return;
2941
bbe0af5b
RR
2942 wxClientDC dc(this);
2943 PrepareDC(dc);
4832f7c0 2944
bbe0af5b
RR
2945 int cw = 0;
2946 int ch = 0;
2947 GetClientSize( &cw, &ch );
4832f7c0 2948
bbe0af5b
RR
2949 wxRect rect;
2950 rect.x = dc.LogicalToDeviceX( 0 );
2951 rect.width = cw;
2952 rect.y = dc.LogicalToDeviceY( item->GetY() );
2953 rect.height = ch;
f135ff73 2954
bbe0af5b 2955 Refresh( TRUE, &rect );
f135ff73 2956
bbe0af5b 2957 AdjustMyScrollbars();
edaa81ae 2958}
c801d85f 2959
941830cb 2960void wxGenericTreeCtrl::RefreshLine( wxGenericTreeItem *item )
c801d85f 2961{
e6527f9d
RR
2962 if (m_dirty) return;
2963
bbe0af5b
RR
2964 wxClientDC dc(this);
2965 PrepareDC( dc );
2966
5391f772
SB
2967 int cw = 0;
2968 int ch = 0;
2969 GetClientSize( &cw, &ch );
2970
bbe0af5b 2971 wxRect rect;
5391f772
SB
2972 rect.x = dc.LogicalToDeviceX( 0 );
2973 rect.y = dc.LogicalToDeviceY( item->GetY() );
2974 rect.width = cw;
91b8de8d 2975 rect.height = GetLineHeight(item); //dc.GetCharHeight() + 6;
978f38c2 2976
bbe0af5b 2977 Refresh( TRUE, &rect );
edaa81ae 2978}
c801d85f 2979
b771aa29
VZ
2980void wxGenericTreeCtrl::RefreshSelected()
2981{
2982 // TODO: this is awfully inefficient, we should keep the list of all
2983 // selected items internally, should be much faster
2984 if ( m_anchor )
2985 RefreshSelectedUnder(m_anchor);
2986}
2987
2988void wxGenericTreeCtrl::RefreshSelectedUnder(wxGenericTreeItem *item)
2989{
2990 if ( item->IsSelected() )
2991 RefreshLine(item);
2992
2993 const wxArrayGenericTreeItems& children = item->GetChildren();
2994 size_t count = children.GetCount();
2995 for ( size_t n = 0; n < count; n++ )
2996 {
2997 RefreshSelectedUnder(children[n]);
2998 }
2999}
3000
7009f411
VZ
3001// ----------------------------------------------------------------------------
3002// changing colours: we need to refresh the tree control
3003// ----------------------------------------------------------------------------
3004
3005bool wxGenericTreeCtrl::SetBackgroundColour(const wxColour& colour)
3006{
3007 if ( !wxWindow::SetBackgroundColour(colour) )
3008 return FALSE;
3009
3010 Refresh();
3011
3012 return TRUE;
3013}
3014
0800a4ba 3015bool wxGenericTreeCtrl::SetForegroundColour(const wxColour& colour)
7009f411
VZ
3016{
3017 if ( !wxWindow::SetForegroundColour(colour) )
3018 return FALSE;
3019
3020 Refresh();
3021
3022 return TRUE;
3023}
3024
1e6feb95 3025#endif // wxUSE_TREECTRL