]> git.saurik.com Git - wxWidgets.git/blob - samples/treelay/treelay.cpp
This is how wxPlotWindow would look like with the
[wxWidgets.git] / samples / treelay / treelay.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: treelay.cpp
3 // Purpose: wxTreeLayout sample
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 // For compilers that support precompilation, includes "wx.h".
13 #include "wx/wxprec.h"
14
15 #ifdef __BORLANDC__
16 #pragma hdrstop
17 #endif
18
19 #ifndef WX_PRECOMP
20 #include "wx/wx.h"
21 #endif
22
23 #include "wx/treelay.h"
24
25 #include "treelay.h"
26
27 wxTreeLayoutStored *myTree = NULL;
28
29 // A macro needed for some compilers (AIX) that need 'main' to be defined
30 // in the application itself.
31 IMPLEMENT_APP(MyApp)
32
33 // The `main program' equivalent, creating the windows and returning the
34 // main frame
35 bool MyApp::OnInit()
36 {
37 // Create the main frame window
38 MyFrame* frame = new MyFrame(NULL, "Tree Test", wxPoint(-1, -1), wxSize(400, 550));
39
40 // Give it a status line
41 frame->CreateStatusBar(2);
42
43 // Give it an icon
44 #ifdef __WINDOWS__
45 wxIcon icon("tree_icn");
46 frame->SetIcon(icon);
47 #endif
48
49 // Make a menubar
50 wxMenu *file_menu = new wxMenu;
51 file_menu->Append(TEST_LEFT_RIGHT, "&Left to right", "Redraw left to right");
52 file_menu->Append(TEST_TOP_BOTTOM, "&Top to bottom", "Redraw top to bottom");
53 file_menu->AppendSeparator();
54 file_menu->Append(TEST_QUIT, "E&xit", "Quit program");
55
56 wxMenu *help_menu = new wxMenu;
57 help_menu->Append(TEST_ABOUT, "&About", "About Tree Test");
58
59 wxMenuBar* menu_bar = new wxMenuBar;
60
61 menu_bar->Append(file_menu, "&File");
62 menu_bar->Append(help_menu, "&Help");
63
64 // Associate the menu bar with the frame
65 frame->SetMenuBar(menu_bar);
66
67 MyCanvas *canvas = new MyCanvas(frame);
68
69 // Give it scrollbars: the virtual canvas is 20 * 50 = 1000 pixels in each direction
70 canvas->SetScrollbars(20, 20, 50, 50);
71 frame->canvas = canvas;
72
73 myTree = new wxTreeLayoutStored();
74
75 wxClientDC dc(canvas);
76 wxFont font(10, wxROMAN, wxNORMAL, wxBOLD);
77 dc.SetFont(font);
78 TreeTest(*myTree, dc);
79
80 frame->Show(TRUE);
81
82 frame->SetStatusText("Hello, tree!");
83
84 // Return the main frame window
85 return TRUE;
86 }
87
88 void MyApp::TreeTest(wxTreeLayoutStored& tree, wxDC& dc)
89 {
90 tree.Initialize(200);
91
92 tree.AddChild("animal");
93 tree.AddChild("mammal", "animal");
94 tree.AddChild("insect", "animal");
95 tree.AddChild("bird", "animal");
96
97 tree.AddChild("man", "mammal");
98 tree.AddChild("cat", "mammal");
99 tree.AddChild("dog", "mammal");
100 tree.AddChild("giraffe", "mammal");
101 tree.AddChild("elephant", "mammal");
102 tree.AddChild("donkey", "mammal");
103 tree.AddChild("horse", "mammal");
104
105 tree.AddChild("fido", "dog");
106 tree.AddChild("domestic cat", "cat");
107 tree.AddChild("lion", "cat");
108 tree.AddChild("tiger", "cat");
109 tree.AddChild("felix", "domestic cat");
110 tree.AddChild("socks", "domestic cat");
111
112 tree.AddChild("beetle", "insect");
113 tree.AddChild("earwig", "insect");
114 tree.AddChild("eagle", "bird");
115 tree.AddChild("bluetit", "bird");
116 tree.AddChild("sparrow", "bird");
117 tree.AddChild("blackbird", "bird");
118 tree.AddChild("emu", "bird");
119 tree.AddChild("crow", "bird");
120
121 tree.DoLayout(dc);
122 }
123
124 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
125 EVT_MENU(TEST_QUIT, MyFrame::OnQuit)
126 EVT_MENU(TEST_ABOUT, MyFrame::OnAbout)
127 EVT_MENU(TEST_LEFT_RIGHT, MyFrame::OnLeftRight)
128 EVT_MENU(TEST_TOP_BOTTOM, MyFrame::OnTopBottom)
129 EVT_CLOSE(MyFrame::OnCloseWindow)
130 END_EVENT_TABLE()
131
132 // Define my frame constructor
133 MyFrame::MyFrame(wxWindow *parent, const wxString& title, const wxPoint& pos, const wxSize& size):
134 wxFrame(parent, -1, title, pos, size)
135 {
136 }
137
138 void MyFrame::OnQuit(wxCommandEvent& event)
139 {
140 Close(TRUE);
141 }
142
143 void MyFrame::OnLeftRight(wxCommandEvent& event)
144 {
145 if (myTree)
146 {
147 myTree->SetOrientation(FALSE);
148 wxClientDC dc(canvas);
149 wxFont font(10, wxROMAN, wxNORMAL, wxBOLD);
150 dc.SetFont(font);
151 wxGetApp().TreeTest(*myTree, dc);
152 canvas->Refresh();
153 }
154 }
155
156 void MyFrame::OnTopBottom(wxCommandEvent& event)
157 {
158 if (myTree)
159 {
160 myTree->SetOrientation(TRUE);
161 wxClientDC dc(canvas);
162 wxFont font(10, wxROMAN, wxNORMAL, wxBOLD);
163 dc.SetFont(font);
164 wxGetApp().TreeTest(*myTree, dc);
165 canvas->Refresh();
166 }
167 }
168
169 void MyFrame::OnAbout(wxCommandEvent& event)
170 {
171 (void)wxMessageBox("wxWindows tree library demo Vsn 2.0\nAuthor: Julian Smart (c) 1998", "About tree test");
172 }
173
174 void MyFrame::OnCloseWindow(wxCloseEvent& event)
175 {
176 Destroy();
177 }
178
179 BEGIN_EVENT_TABLE(MyCanvas, wxScrolledWindow)
180 EVT_PAINT(MyCanvas::OnPaint)
181 END_EVENT_TABLE()
182
183 // Define a constructor for my canvas
184 MyCanvas::MyCanvas(wxWindow *parent):
185 wxScrolledWindow(parent, -1)
186 {
187 SetBackgroundColour(*wxWHITE);
188 }
189
190 // Define the repainting behaviour
191 void MyCanvas::OnPaint(wxPaintEvent& event)
192 {
193 wxPaintDC dc(this);
194 PrepareDC(dc);
195 if (myTree)
196 {
197 wxFont font(10, wxROMAN, wxNORMAL, wxBOLD);
198 dc.SetFont(font);
199 myTree->Draw(dc);
200 }
201 }
202