]> git.saurik.com Git - wxWidgets.git/blob - demos/dbbrowse/dbtree.cpp
Applied patch [ 1171467 ] Fix for DocManager not checking OnNewDocument's return...
[wxWidgets.git] / demos / dbbrowse / dbtree.cpp
1 //----------------------------------------------------------------------------------------
2 // Name: DBTree.cpp/.h
3 // Purpose: Tree with Table and Views, branches show Field information
4 // Author: Mark Johnson
5 // Modified by:
6 // Created: 19991129
7 // RCS-ID: $Id$
8 // Copyright: (c) Mark Johnson, Berlin Germany
9 // Licence: wxWindows license
10 //----------------------------------------------------------------------------------------
11 //-- all #ifdefs that the whole Project needs. -------------------------------------------
12 //----------------------------------------------------------------------------------------
13 #ifdef __GNUG__
14 #pragma implementation
15 #pragma interface
16 #endif
17 //----------------------------------------------------------------------------------------
18 // For compilers that support precompilation, includes "wx/wx.h".
19 #include "wx/wxprec.h"
20 //----------------------------------------------------------------------------------------
21 #ifdef __BORLANDC__
22 #pragma hdrstop
23 #endif
24 //----------------------------------------------------------------------------------------
25 #ifndef WX_PRECOMP
26 #include "wx/wx.h"
27 #endif
28 //----------------------------------------------------------------------------------------
29 #ifndef __WXMSW__
30 #endif
31 //----------------------------------------------------------------------------------------
32 //-- all #includes that every .cpp needs --- 19990807.mj10777 ----------------
33 //----------------------------------------------------------------------------------------
34 #include "std.h" // sorgsam Pflegen !
35
36 //----------------------------------------------------------------------------------------
37 //-- Global functions --------------------------------------------------------------------
38 //----------------------------------------------------------------------------------------
39 static inline const wxChar *bool2String(bool b)
40 {
41 return b ? wxEmptyString : _T("not ");
42 }
43
44 //----------------------------------------------------------------------------------------
45 BEGIN_EVENT_TABLE(DBTree, wxTreeCtrl)
46 EVT_MOTION (DBTree::OnMouseMove)
47 EVT_TREE_SEL_CHANGED(TREE_CTRL_DB, DBTree::OnSelChanged)
48 EVT_TREE_ITEM_RIGHT_CLICK(TREE_CTRL_DB,DBTree::OnRightSelect)
49 EVT_MENU(DATA_SHOW,DBTree::OnDBGrid)
50 EVT_MENU(DATA_DB,DBTree::OnDBClass)
51 EVT_MENU(DATA_TABLE,DBTree::OnTableClass)
52 EVT_MENU(DATA_TABLE_ALL,DBTree::OnTableClassAll)
53 END_EVENT_TABLE()
54 //----------------------------------------------------------------------------------------
55 // DBTree implementation
56 //----------------------------------------------------------------------------------------
57 IMPLEMENT_DYNAMIC_CLASS(DBTree, wxTreeCtrl)
58
59 //----------------------------------------------------------------------------------------
60 DBTree::DBTree(wxWindow *parent) : wxTreeCtrl(parent)
61 {
62 }
63
64 //----------------------------------------------------------------------------------------
65 DBTree::DBTree(wxWindow *parent, const wxWindowID id,const wxPoint& pos, const wxSize& size, long style)
66 : wxTreeCtrl(parent, id, pos, size, style)
67 {
68 const int image_size = 16;
69 // Make an image list containing small icons
70 p_imageListNormal = new wxImageList(image_size, image_size, true);
71 // should correspond to TreeIc_xxx enum
72 #if !defined(__WXMSW__)
73 #include "bitmaps/logo.xpm"
74 #include "bitmaps/dsnclose.xpm"
75 #include "bitmaps/dsnopen.xpm"
76 #include "bitmaps/tab.xpm"
77 #include "bitmaps/view.xpm"
78 #include "bitmaps/col.xpm"
79 #include "bitmaps/key.xpm"
80 #include "bitmaps/keyf.xpm"
81 #include "bitmaps/d_open.xpm"
82 #include "bitmaps/d_closed.xpm"
83 #endif
84 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(aLogo)).ConvertToImage().Rescale(image_size, image_size)));
85 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(DsnClosed)).ConvertToImage().Rescale(image_size, image_size)));
86 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(DsnOpen)).ConvertToImage().Rescale(image_size, image_size)));
87 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(TAB)).ConvertToImage().Rescale(image_size, image_size)));
88 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(VIEW)).ConvertToImage().Rescale(image_size, image_size)));
89 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(COL)).ConvertToImage().Rescale(image_size, image_size)));
90 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(KEY)).ConvertToImage().Rescale(image_size, image_size)));
91 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(KEYF)).ConvertToImage().Rescale(image_size, image_size)));
92 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(DocOpen)).ConvertToImage().Rescale(image_size, image_size)));
93 p_imageListNormal->Add(wxBitmap(wxBitmap(wxICON(DocOpen)).ConvertToImage().Rescale(image_size, image_size)));
94 SetImageList(p_imageListNormal);
95 ct_BrowserDB = NULL;
96 popupMenu1 = NULL;
97 popupMenu2 = NULL;
98 }
99
100 //----------------------------------------------------------------------------------------
101 DBTree::~DBTree()
102 {
103 delete ct_BrowserDB;
104
105 // delete (pDoc->db_Br+i_Which);
106 // wxLogMessage("DBTree::~DBTree() - Vor OnCloseDB()");
107 (pDoc->db_Br+i_Which)->OnCloseDB(false);
108 // wxLogMessage("DBTree::~DBTree() - Nach OnCloseDB()");
109 (pDoc->db_Br+i_Which)->db_BrowserDB = NULL;
110 (pDoc->db_Br+i_Which)->ct_BrowserDB = NULL;
111 (pDoc->db_Br+i_Which)->cl_BrowserDB = NULL;
112
113 delete p_imageListNormal;
114 if (popupMenu1) // If the DSN has no Tables, then no delete should be done !
115 delete popupMenu1;
116 if (popupMenu1) // If the DSN has no Tables, then no delete should be done !
117 delete popupMenu2;
118 }
119
120 //----------------------------------------------------------------------------------------
121 #undef TREE_EVENT_HANDLER
122
123 //----------------------------------------------------------------------------------------
124 int DBTree::OnPopulate()
125 {
126 wxTreeItemId Root, Folder, Docu, Funkt;
127 int i,x,y,z=0, TableType;
128 wxString SQL_TYPE, DB_TYPE;
129 SetFont(* pDoc->ft_Doc);
130 //---------------------------------------------------------------------------------------
131 if ((pDoc->db_Br+i_Which)->Initialize(false))
132 {
133 wxStopWatch sw;
134 wxBeginBusyCursor();
135 ct_BrowserDB = (pDoc->db_Br+i_Which)->OnGetCatalog(false);
136 if (ct_BrowserDB)
137 { // Use the wxDatabase Information
138 Temp0.Printf(_T("%s - (%s) (%s)"), s_DSN.c_str(),ct_BrowserDB->catalog, ct_BrowserDB->schema);
139 Root = AddRoot(Temp0,TreeIc_DsnOpen,TreeIc_DsnOpen,new DBTreeData(_T("Root")));
140 for (x=0;x<ct_BrowserDB->numTables;x++)
141 {
142 wxYield();
143 TableType = 0; // TABLE = 1 ; VIEW = 2 ; 0 We are not interested in
144 if (!wxStrcmp((ct_BrowserDB->pTableInf+x)->tableType,_T("TABLE"))) // only TABLES
145 TableType = 1;
146 if (!wxStrcmp((ct_BrowserDB->pTableInf+x)->tableType,_T("VIEW"))) // and VIEWS
147 TableType = 2;
148 if (TableType) // only TABLES or Views
149 {
150 Temp1.Printf(_T("TN(%s"),(ct_BrowserDB->pTableInf+x)->tableName);
151 //----
152 (ct_BrowserDB->pTableInf+x)->pColInf = (pDoc->db_Br+i_Which)->OnGetColumns((ct_BrowserDB->pTableInf+x)->tableName,(ct_BrowserDB->pTableInf+x)->numCols,false);
153 //----
154 if ((ct_BrowserDB->pTableInf+x)->pColInf)
155 {
156 if (TableType == 1) // Table
157 {
158 Temp0.Printf(_("Table-Name(%s) with (%d)Columns ; Remarks(%s)"), (ct_BrowserDB->pTableInf+x)->tableName,
159 (ct_BrowserDB->pTableInf+x)->numCols,(ct_BrowserDB->pTableInf+x)->tableRemarks);
160 Folder = AppendItem(Root,Temp0,TreeIc_TAB,TreeIc_TAB, new DBTreeData(Temp1));
161 }
162 if (TableType == 2) // View
163 {
164 Temp0.Printf(_("View-Name(%s) with (%d)Columns ; Remarks(%s)"), (ct_BrowserDB->pTableInf+x)->tableName,
165 (ct_BrowserDB->pTableInf+x)->numCols,(ct_BrowserDB->pTableInf+x)->tableRemarks);
166 Folder = AppendItem(Root,Temp0,TreeIc_VIEW,TreeIc_VIEW, new DBTreeData(Temp1));
167 }
168 for (y=0;y<(ct_BrowserDB->pTableInf+x)->numCols;y++)
169 {
170 Temp1.Printf(_T("FN(%s"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->colName);
171 // Here is where we find out if the Column is a Primary / Foreign Key
172 if (((ct_BrowserDB->pTableInf+x)->pColInf+y)->PkCol != 0) // Primary Key
173 {
174 Temp2.Printf(_T("(%d) - %s"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->PkCol,((ct_BrowserDB->pTableInf+x)->pColInf+y)->colName);
175 Docu = AppendItem(Folder,Temp2,TreeIc_KEY,TreeIc_KEY,new DBTreeData(Temp1));
176 Temp2 = ((ct_BrowserDB->pTableInf+x)->pColInf+y)->PkTableName;
177 if (Temp2.empty())
178 Temp2 = _("None");
179 Temp2.Printf(_("This Primary Key is used in the following Tables : %s"),Temp2.c_str());
180 Funkt = AppendItem(Docu,Temp2,TreeIc_DocClosed,TreeIc_DocOpen,new DBTreeData(_T("KEY")));
181 }
182 else
183 {
184 if (((ct_BrowserDB->pTableInf+x)->pColInf+y)->FkCol != 0) // Foreign Key
185 {
186 Temp2.Printf(_T("(%d) - %s"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->FkCol,((ct_BrowserDB->pTableInf+x)->pColInf+y)->colName);
187 Docu = AppendItem(Folder,Temp2,TreeIc_KEYF,TreeIc_KEYF,new DBTreeData(Temp1));
188 Temp2.Printf(_("This Foreign Key comes from the following Table : %s"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->FkTableName);
189 Funkt = AppendItem(Docu,Temp2,TreeIc_DocClosed,TreeIc_DocOpen,new DBTreeData(_T("KEYF")));
190 }
191 else
192 Docu = AppendItem(Folder,((ct_BrowserDB->pTableInf+x)->pColInf+y)->colName,TreeIc_COL,TreeIc_COL,new DBTreeData(Temp1));
193 }
194 SQL_TYPE.Printf(_T("SQL_C_???? (%d)"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->sqlDataType);
195 DB_TYPE.Printf(_T("DB_DATA_TYPE_???? (%d)"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->dbDataType);
196 for (i=1;i<=(pDoc->db_Br+i_Which)->i_SqlTyp[0];i++)
197 {
198 if (((ct_BrowserDB->pTableInf+x)->pColInf+y)->sqlDataType == (pDoc->db_Br+i_Which)->i_SqlTyp[i])
199 {
200 SQL_TYPE.Printf(_T("%s(%d) ; "),(pDoc->db_Br+i_Which)->s_SqlTyp[i].c_str(),(pDoc->db_Br+i_Which)->i_SqlTyp[i]);
201 }
202 } // for (i=1;i<=i_SqlTyp[0];i++)
203 wxYield();
204 for (i=1;i<=(pDoc->db_Br+i_Which)->i_dbTyp[0];i++)
205 {
206 if (((ct_BrowserDB->pTableInf+x)->pColInf+y)->dbDataType == (pDoc->db_Br+i_Which)->i_dbTyp[i])
207 {
208 DB_TYPE.Printf(_T("%s(%d)"),(pDoc->db_Br+i_Which)->s_dbTyp[i].c_str(),(pDoc->db_Br+i_Which)->i_dbTyp[i]);
209 }
210 } // for (i=1;i<=i_dbTyp[0];i++)
211 wxYield();
212 SQL_TYPE += DB_TYPE;
213 Funkt = AppendItem(Docu,SQL_TYPE,TreeIc_DocClosed,TreeIc_DocOpen,new DBTreeData(SQL_TYPE));
214 SQL_TYPE.Printf(_T("%10s %d,%d"),((ct_BrowserDB->pTableInf+x)->pColInf+y)->typeName,
215 ((ct_BrowserDB->pTableInf+x)->pColInf+y)->columnLength,((ct_BrowserDB->pTableInf+x)->pColInf+y)->decimalDigits);
216 Funkt = AppendItem(Docu,SQL_TYPE,TreeIc_DocClosed,TreeIc_DocOpen,new DBTreeData(SQL_TYPE));
217 } // for (y=0;y<(ct_BrowserDB->pTableInf+x)->numCols;y++)
218 } // if ((ct_BrowserDB->pTableInf+x)->pColInf)
219 else
220 Folder = AppendItem(Root,Temp0,TreeIc_FolderClosed,TreeIc_FolderOpen, new DBTreeData(Temp1));
221 z++;
222 // if (z % 10 == 0)
223 {
224 #if wxUSE_STATUSBAR
225 Temp0.Printf(_("-I-> DBTree::OnPopulate(%s) - Table %6d (from %d) has been read."),(ct_BrowserDB->pTableInf+x)->tableName,z,ct_BrowserDB->numTables);
226 pDoc->p_MainFrame->SetStatusText(Temp0, 0);
227 #endif // wxUSE_STATUSBAR
228 }
229 wxYield();
230 } // if ((ct_BrowserDB->pTableInf+x)->tableType == "TABLE" or VIEW)
231 // else
232 // wxLogMessage(_("\n-I-> if ! TABLE or VIEW >%s<"),(ct_BrowserDB->pTableInf+x)->tableType);
233 } // for (x=0;x<ct_BrowserDB->numTables;x++)
234 } // if (ct_BrowserDB)
235 else
236 wxLogMessage(_("\n-E-> DBTree::OnPopulate() : Invalid Catalog Pointer : Failed"));
237 wxEndBusyCursor();
238 Temp0.Printf(_("-I-> DBTree::OnPopulate() - %6d Tables have been read. - Time needed : %ld ms"),z,sw.Time());
239 wxLogMessage(Temp0);
240 #if wxUSE_STATUSBAR
241 pDoc->p_MainFrame->SetStatusText(Temp0, 0);
242 #endif // wxUSE_STATUSBAR
243 } // if((pDoc->db_Br+i_Which)->Initialize(false))
244 else
245 {
246 wxLogMessage(_("\n-E-> DBTree::OnPopulate() : A valid Pointer could not be created : Failed"));
247 return 0;
248 }
249 //---------------------------------------------------------------------------------------
250 Expand(Root);
251 //---------------------------------------------------------------------------------------
252 popupMenu1 = NULL;
253 popupMenu1 = new wxMenu;
254 popupMenu1->Append(DATA_DB, _("Make wxDB.cpp/h "));
255 popupMenu1->AppendSeparator();
256 popupMenu1->Append(DATA_TABLE_ALL, _("Make all wxTable.cpp/h classes"));
257 popupMenu2 = NULL;
258 popupMenu2 = new wxMenu;
259 popupMenu2->Append(DATA_SHOW, _("Show Data"));
260 popupMenu2->AppendSeparator();
261 popupMenu2->Append(DATA_TABLE, _("Make wxTable.cpp/h "));
262 //---------------------------------------------------------------------------------------
263 return 0;
264 } // DBTree::OnPopulate()
265
266 //----------------------------------------------------------------------------------------
267 void DBTree::OnSelChanged(wxTreeEvent& WXUNUSED(event))
268 {
269 int i;
270 Temp0.Empty();
271 #if wxUSE_STATUSBAR
272 pDoc->p_MainFrame->SetStatusText(Temp0,0);
273 #endif // wxUSE_STATUSBAR
274 // Get the Information that we need
275 wxTreeItemId itemId = GetSelection();
276 DBTreeData *item = (DBTreeData *)GetItemData(itemId);
277 if ( item != NULL )
278 {
279 int Treffer = 0;
280 Temp1.Printf(_T("%s"),item->m_desc.c_str());
281 //-------------------------------------------------------------------------------------
282 if (Temp1.Contains(_T("ODBC-")))
283 {
284 Temp1 = Temp1.Mid(5,wxSTRING_MAXLEN);
285 for (i=0;i<pDoc->i_DSN;i++)
286 {
287 if (Temp1 == (pDoc->p_DSN+i)->Dsn)
288 {
289 // pDoc->OnChosenDSN(i);
290 }
291 }
292 Treffer++;
293 }
294 //--------------------------------------------------------------------------------------
295 if (Treffer == 0)
296 {
297 //-------------------------------------------------------------------------------------
298 /*
299 Temp0.Printf(_("Item '%s': %sselected, %sexpanded, %sbold,"
300 "%u children (%u immediately under this item)."),
301 item->m_desc.c_str(),
302 bool2String(IsSelected(itemId)),
303 bool2String(IsExpanded(itemId)),
304 bool2String(IsBold(itemId)),
305 GetChildrenCount(itemId),
306 GetChildrenCount(itemId));
307 LogBuf.Printf("-I-> DBTree::OnSelChanged - %s",Temp0.c_str());
308 wxLogMessage( "%s", LogBuf.c_str() );
309 */
310 //-------------------------------------------------------------------------------------
311 }
312 }
313 }
314
315 //----------------------------------------------------------------------------------------
316 void DBTree::OnRightSelect(wxTreeEvent& WXUNUSED(event))
317 {
318 int i;
319 Temp0.Empty();
320 // Get the Information that we need
321 wxTreeItemId itemId = GetSelection();
322 DBTreeData *item = (DBTreeData *)GetItemData(itemId);
323 if ( item != NULL )
324 {
325 int Treffer = 0;
326 Temp1.Printf(_T("%s"),item->m_desc.c_str());
327 //--------------------------------------------------------------------------------------
328 if (!wxStrcmp(_T("Root"),Temp1))
329 {
330 PopupMenu(popupMenu1,TreePos.x,TreePos.y);
331 Treffer++;
332 }
333 for (i=0;i<ct_BrowserDB->numTables;i++)
334 {
335 Temp2.Printf(_T("TN(%s"),(ct_BrowserDB->pTableInf+i)->tableName);
336 if (!wxStrcmp(Temp2,Temp1))
337 {
338 PopupMenu(popupMenu2,TreePos.x,TreePos.y);
339 Treffer++;
340 }
341 }
342 //--------------------------------------------------------------------------------------
343 if (Treffer == 0)
344 {
345 //-------------------------------------------------------------------------------------
346 /*
347 Temp0.Printf(_("Item '%s': %sselected, %sexpanded, %sbold,"
348 "%u children (%u immediately under this item)."),
349 item->m_desc.c_str(),
350 bool2String(IsSelected(itemId)),
351 bool2String(IsExpanded(itemId)),
352 bool2String(IsBold(itemId)),
353 GetChildrenCount(itemId),
354 GetChildrenCount(itemId));
355 LogBuf.Printf("-I-> DBTree::OnSelChanged - %s",Temp0.c_str());
356 wxLogMessage( "%s", LogBuf.c_str() );
357 */
358 //-------------------------------------------------------------------------------------
359 }
360 }
361 }
362
363 //----------------------------------------------------------------------------------------
364 void DBTree::OnDBGrid(wxCommandEvent& WXUNUSED(event))
365 {
366 int i;
367 // Get the Information that we need
368 wxTreeItemId itemId = GetSelection();
369 DBTreeData *item = (DBTreeData *)GetItemData(itemId);
370 if ( item != NULL )
371 {
372 Temp1.Printf(_T("%s"),item->m_desc.c_str());
373 for (i=0;i<ct_BrowserDB->numTables;i++)
374 {
375 Temp2.Printf(_T("TN(%s"),(ct_BrowserDB->pTableInf+i)->tableName);
376 if (!wxStrcmp(Temp2,Temp1))
377 {
378 // Temp0.Printf("(%d) Here is where a GridCtrl for >%s< will be called! ",i,(ct_BrowserDB->pTableInf+i)->tableName);
379 pDoc->OnChosenTbl(1,(ct_BrowserDB->pTableInf+i)->tableName);
380 // wxMessageBox(Temp0);
381 }
382 }
383 }
384 }
385
386 //----------------------------------------------------------------------------------------
387 void DBTree::OnDBClass(wxCommandEvent& WXUNUSED(event))
388 {
389 // int i;
390 // Get the Information that we need
391 wxTreeItemId itemId = GetSelection();
392 DBTreeData *item = (DBTreeData *)GetItemData(itemId);
393 if ( item != NULL )
394 {
395 Temp0.Printf(_("Here is where a wxDB Class for >%s< will be made! "),s_DSN.c_str());
396 wxMessageBox(Temp0);
397 }
398 }
399
400 //----------------------------------------------------------------------------------------
401 void DBTree::OnTableClass(wxCommandEvent& WXUNUSED(event))
402 {
403 int i;
404 // Get the Information that we need
405 wxTreeItemId itemId = GetSelection();
406 DBTreeData *item = (DBTreeData *)GetItemData(itemId);
407 if ( item != NULL )
408 {
409 Temp1.Printf(_T("%s"),item->m_desc.c_str());
410 for (i=0;i<ct_BrowserDB->numTables;i++)
411 {
412 Temp2.Printf(_T("TN(%s"),(ct_BrowserDB->pTableInf+i)->tableName);
413 if (!wxStrcmp(Temp2,Temp1))
414 {
415 Temp0.Printf(_("(%d) Here is where a wxTable Class for >%s< will be made! "),i,(ct_BrowserDB->pTableInf+i)->tableName);
416 wxMessageBox(Temp0);
417 }
418 }
419 }
420 }
421
422 //----------------------------------------------------------------------------------------
423 void DBTree::OnTableClassAll(wxCommandEvent& WXUNUSED(event))
424 {
425 // int i;
426 // Get the Information that we need
427 wxTreeItemId itemId = GetSelection();
428 DBTreeData *item = (DBTreeData *)GetItemData(itemId);
429 if ( item != NULL )
430 {
431 Temp0.Printf(_("Here is where all wxTable Classes in >%s< will be made! "),s_DSN.c_str());
432 wxMessageBox(Temp0);
433 }
434 }
435
436 //----------------------------------------------------------------------------------------
437 void DBTree::OnMouseMove(wxMouseEvent &event)
438 {
439 TreePos = event.GetPosition();
440 }
441 //----------------------------------------------------------------------------------------