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