]> git.saurik.com Git - wxWidgets.git/blob - src/generic/treelay.cpp
One more function was out of alphanumeric sequence
[wxWidgets.git] / src / generic / treelay.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: treelay.h
3 // Purpose: wxTreeLayout class
4 // Author: Julian Smart
5 // Modified by:
6 // Created: 7/4/98
7 // RCS-ID: $Id$
8 // Copyright: (c) 1998 Julian Smart
9 // Licence: wxWindows licence
10 ///////////////////////////////////////////////////////////////////////////////
11
12 #ifdef __GNUG__
13 #pragma implementation "wxtree.h"
14 #endif
15
16 // For compilers that support precompilation, includes "wx.h".
17 #include <wx/wxprec.h>
18
19 #ifdef __BORLANDC__
20 #pragma hdrstop
21 #endif
22
23 #ifndef WX_PRECOMP
24 #include "wx/wx.h"
25 #endif
26
27 #if wxUSE_TREELAYOUT
28
29 #include <wx/treelay.h>
30
31 /*
32 * Abstract tree
33 *
34 */
35
36 IMPLEMENT_ABSTRACT_CLASS(wxTreeLayout, wxObject)
37
38 wxTreeLayout::wxTreeLayout()
39 {
40 m_xSpacing = 16;
41 m_ySpacing = 20;
42 m_topMargin = 5;
43 m_leftMargin = 5;
44 m_orientation = FALSE;
45 m_parentNode = 0;
46 }
47
48 void wxTreeLayout::DoLayout(wxDC& dc, long topId)
49 {
50 if (topId != -1)
51 SetTopNode(topId);
52
53 long actualTopId = GetTopNode();
54 long id = actualTopId;
55 while (id != -1)
56 {
57 SetNodeX(id, 0);
58 SetNodeY(id, 0);
59 ActivateNode(id, FALSE);
60 id = GetNextNode(id);
61 }
62 m_lastY = m_topMargin;
63 m_lastX = m_leftMargin;
64 CalcLayout(actualTopId, 0, dc);
65 }
66
67 void wxTreeLayout::Draw(wxDC& dc)
68 {
69 dc.Clear();
70 DrawBranches(dc);
71 DrawNodes(dc);
72 }
73
74 void wxTreeLayout::DrawNodes(wxDC& dc)
75 {
76 long id = GetTopNode();
77 while (id != -1)
78 {
79 if (NodeActive(id))
80 DrawNode(id, dc);
81 id = GetNextNode(id);
82 }
83 }
84
85 void wxTreeLayout::DrawBranches(wxDC& dc)
86 {
87 long id = GetTopNode();
88 while (id != -1)
89 {
90 if (GetNodeParent(id) > -1)
91 {
92 long parent = GetNodeParent(id);
93 if (NodeActive(parent))
94 DrawBranch(parent, id, dc);
95 }
96 id = GetNextNode(id);
97 }
98 }
99
100 void wxTreeLayout::DrawNode(long id, wxDC& dc)
101 {
102 wxChar buf[80];
103 wxString name(GetNodeName(id));
104 if (name != wxT(""))
105 wxSprintf(buf, wxT("%s"), (const wxChar*) name);
106 else
107 wxSprintf(buf, wxT("<unnamed>"));
108
109 long x = 80;
110 long y = 20;
111 dc.GetTextExtent(buf, &x, &y);
112 dc.DrawText(buf, GetNodeX(id), (long)(GetNodeY(id) - (y/2.0)));
113 }
114
115 void wxTreeLayout::DrawBranch(long from, long to, wxDC& dc)
116 {
117 long w, h;
118 GetNodeSize(from, &w, &h, dc);
119 dc.DrawLine(GetNodeX(from)+w, GetNodeY(from),
120 GetNodeX(to), GetNodeY(to));
121 }
122
123 void wxTreeLayout::Initialize(void)
124 {
125 }
126
127 void wxTreeLayout::GetNodeSize(long id, long *x, long *y, wxDC& dc)
128 {
129 wxString name(GetNodeName(id));
130 if (name != wxT(""))
131 dc.GetTextExtent(name, x, y);
132 else
133 {
134 *x = 70; *y = 20;
135 }
136 }
137
138 void wxTreeLayout::CalcLayout(long nodeId, int level, wxDC& dc)
139 {
140 wxList children;
141 GetChildren(nodeId, children);
142 int n = children.Number();
143
144 if (m_orientation == FALSE)
145 {
146 // Left to right
147 // X Calculations
148 if (level == 0)
149 SetNodeX(nodeId, m_leftMargin);
150 else
151 {
152 long x = 0;
153 long y = 0;
154 long parentId = GetNodeParent(nodeId);
155 if (parentId != -1)
156 GetNodeSize(parentId, &x, &y, dc);
157 SetNodeX(nodeId, (long)(GetNodeX(parentId) + m_xSpacing + x));
158 }
159
160 wxNode *node = children.First();
161 while (node)
162 {
163 CalcLayout((long)node->Data(), level+1, dc);
164 node = node->Next();
165 }
166
167 // Y Calculations
168 long averageY;
169 ActivateNode(nodeId, TRUE);
170
171 if (n > 0)
172 {
173 averageY = 0;
174 node = children.First();
175 while (node)
176 {
177 averageY += GetNodeY((long)node->Data());
178 node = node->Next();
179 }
180 averageY = averageY / n;
181 SetNodeY(nodeId, averageY);
182 }
183 else
184 {
185 SetNodeY(nodeId, m_lastY);
186 long x, y;
187 GetNodeSize(nodeId, &x, &y, dc);
188
189 m_lastY = m_lastY + y + m_ySpacing;
190 }
191 }
192 else
193 {
194 // Top to bottom
195
196 // Y Calculations
197 if (level == 0)
198 SetNodeY(nodeId, m_topMargin);
199 else
200 {
201 long x = 0;
202 long y = 0;
203 long parentId = GetNodeParent(nodeId);
204 if (parentId != -1)
205 GetNodeSize(parentId, &x, &y, dc);
206 SetNodeY(nodeId, (long)(GetNodeY(parentId) + m_ySpacing + y));
207 }
208
209 wxNode *node = children.First();
210 while (node)
211 {
212 CalcLayout((long)node->Data(), level+1, dc);
213 node = node->Next();
214 }
215
216 // X Calculations
217 long averageX;
218 ActivateNode(nodeId, TRUE);
219
220 if (n > 0)
221 {
222 averageX = 0;
223 node = children.First();
224 while (node)
225 {
226 averageX += GetNodeX((long)node->Data());
227 node = node->Next();
228 }
229 averageX = averageX / n;
230 SetNodeX(nodeId, averageX);
231 }
232 else
233 {
234 SetNodeX(nodeId, m_lastX);
235 long x, y;
236 GetNodeSize(nodeId, &x, &y, dc);
237
238 m_lastX = m_lastX + x + m_xSpacing;
239 }
240 }
241 }
242
243 /*
244 * Tree with storage
245 *
246 */
247
248 IMPLEMENT_DYNAMIC_CLASS(wxTreeLayoutStored, wxTreeLayout)
249
250 wxTreeLayoutStored::wxTreeLayoutStored(int n):wxTreeLayout()
251 {
252 m_nodes = NULL;
253 m_maxNodes = 0;
254 Initialize(n);
255 }
256
257 wxTreeLayoutStored::~wxTreeLayoutStored(void)
258 {
259 if (m_nodes)
260 delete[] m_nodes;
261 }
262
263 void wxTreeLayoutStored::Initialize(int n)
264 {
265 m_maxNodes = n;
266 wxTreeLayout::Initialize();
267 if (m_nodes) delete[] m_nodes;
268 m_nodes = new wxStoredNode[m_maxNodes];
269 int i;
270 for (i = 0; i < n; i++)
271 {
272 m_nodes[i].m_name = "";
273 m_nodes[i].m_active = FALSE;
274 m_nodes[i].m_parentId = -1;
275 m_nodes[i].m_x = 0;
276 m_nodes[i].m_y = 0;
277 }
278 m_num = 0;
279 }
280
281 long wxTreeLayoutStored::AddChild(const wxString& name, const wxString& parent)
282 {
283 if (m_num < (m_maxNodes -1 ))
284 {
285 long i = -1;
286 if (parent != wxT(""))
287 i = NameToId(parent);
288 else m_parentNode = m_num;
289
290 m_nodes[m_num].m_parentId = i;
291 m_nodes[m_num].m_name = name;
292 m_nodes[m_num].m_x = m_nodes[m_num].m_y = 0;
293 m_nodes[m_num].m_clientData = 0;
294 m_num ++;
295
296 return (m_num - 1);
297 }
298 else
299 return -1;
300 }
301
302 long wxTreeLayoutStored::NameToId(const wxString& name)
303 {
304 long i;
305 for (i = 0; i < m_num; i++)
306 if (name == m_nodes[i].m_name)
307 return i;
308 return -1;
309 }
310
311 void wxTreeLayoutStored::GetChildren(long id, wxList& list)
312 {
313 long currentId = GetTopNode();
314 while (currentId != -1)
315 {
316 if (id == GetNodeParent(currentId))
317 list.Append((wxObject *)currentId);
318 currentId = GetNextNode(currentId);
319 }
320 }
321
322 wxStoredNode* wxTreeLayoutStored::GetNode(long idx) const
323 {
324 wxASSERT(idx < m_num);
325
326 return &m_nodes[idx];
327 };
328
329 long wxTreeLayoutStored::GetNodeX(long id)
330 {
331 wxASSERT(id < m_num);
332
333 return (long)m_nodes[id].m_x;
334 }
335
336 long wxTreeLayoutStored::GetNodeY(long id)
337 {
338 wxASSERT(id < m_num);
339
340 return (long)m_nodes[id].m_y;
341 }
342
343 void wxTreeLayoutStored::SetNodeX(long id, long x)
344 {
345 wxASSERT(id < m_num);
346
347 m_nodes[id].m_x = (int)x;
348 }
349
350 void wxTreeLayoutStored::SetNodeY(long id, long y)
351 {
352 wxASSERT(id < m_num);
353
354 m_nodes[id].m_y = (int)y;
355 }
356
357 void wxTreeLayoutStored::SetNodeName(long id, const wxString& name)
358 {
359 wxASSERT(id < m_num);
360
361 m_nodes[id].m_name = name;
362 }
363
364 wxString wxTreeLayoutStored::GetNodeName(long id)
365 {
366 wxASSERT(id < m_num);
367
368 return m_nodes[id].m_name;
369 }
370
371 long wxTreeLayoutStored::GetNodeParent(long id)
372 {
373 if (id != -1)
374 {
375 wxASSERT(id < m_num);
376
377 return m_nodes[id].m_parentId;
378 }
379 else
380 return -1;
381 }
382
383 long wxTreeLayoutStored::GetNextNode(long id)
384 {
385 wxASSERT(id < m_num);
386
387 if ((id != -1) && (id < (m_num - 1)))
388 return id + 1;
389 else
390 return -1;
391 }
392
393 void wxTreeLayoutStored::SetClientData(long id, long clientData)
394 {
395 wxASSERT(id < m_num);
396
397 m_nodes[id].m_clientData = clientData;
398 }
399
400 long wxTreeLayoutStored::GetClientData(long id) const
401 {
402 wxASSERT(id < m_num);
403
404 return m_nodes[id].m_clientData;
405 }
406
407 void wxTreeLayoutStored::ActivateNode(long id, bool active)
408 {
409 wxASSERT(id < m_num);
410
411 m_nodes[id].m_active = active;
412 }
413
414 bool wxTreeLayoutStored::NodeActive(long id)
415 {
416 wxASSERT(id < m_num);
417
418 return m_nodes[id].m_active;
419 }
420
421 wxString wxTreeLayoutStored::HitTest(wxMouseEvent& event, wxDC& dc)
422 {
423 wxPoint pt = event.GetPosition();
424 wxCoord x = pt.x;
425 wxCoord y = pt.y;
426
427 int i;
428 for (i = 0; i < m_maxNodes; i++)
429 {
430 long width, height;
431 dc.GetTextExtent(m_nodes[i].m_name, &width, &height);
432
433 if ( (x >= (m_nodes[i].m_x-10)) && (x < (m_nodes[i].m_x + width+10)) &&
434 (y >= m_nodes[i].m_y-10) && (y < (m_nodes[i].m_y + height+10)) )
435 {
436 return m_nodes[i].m_name;
437 }
438 }
439
440 return wxString("");
441 }
442
443 #endif
444 // wxUSE_TREELAYOUT