samples/treectrl/treetest.h

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        treetest.h
   3 // Purpose:     wxTreeCtrl sample
   4 // Author:      Julian Smart
   5 // Modified by:
   6 // Created:     04/01/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Julian Smart and Markus Holzem
   9 // Licence:     wxWindows license
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // Define a new application type
  13 class MyApp : public wxApp
  14 {
  15 public:
  16   bool OnInit();
  17 };
  18
  19 class MyTreeItemData : public wxTreeItemData
  20 {
  21 public:
  22   MyTreeItemData(const wxString& desc) : m_desc(desc) { }
  23
  24   void ShowInfo(wxTreeCtrl *tree);
  25
  26 private:
  27   wxString m_desc;
  28 };
  29
  30 class MyTreeCtrl : public wxTreeCtrl
  31 {
  32 public:
  33   enum
  34   {
  35     TreeCtrlIcon_File,
  36     TreeCtrlIcon_Folder
  37   };
  38
  39   MyTreeCtrl() { }
  40   MyTreeCtrl(wxWindow *parent, const wxWindowID id,
  41              const wxPoint& pos, const wxSize& size,
  42              long style);
  43   virtual ~MyTreeCtrl();
  44
  45   void OnBeginDrag(wxTreeEvent& event);
  46   void OnBeginRDrag(wxTreeEvent& event);
  47   void OnBeginLabelEdit(wxTreeEvent& event);
  48   void OnEndLabelEdit(wxTreeEvent& event);
  49   void OnDeleteItem(wxTreeEvent& event);
  50   void OnGetInfo(wxTreeEvent& event);
  51   void OnSetInfo(wxTreeEvent& event);
  52   void OnItemExpanded(wxTreeEvent& event);
  53   void OnItemExpanding(wxTreeEvent& event);
  54   void OnItemCollapsed(wxTreeEvent& event);
  55   void OnItemCollapsing(wxTreeEvent& event);
  56   void OnSelChanged(wxTreeEvent& event);
  57   void OnSelChanging(wxTreeEvent& event);
  58   void OnTreeKeyDown(wxTreeEvent& event);
  59   void OnItemActivated(wxTreeEvent& event);
  60
  61   void GetItemsRecursively(const wxTreeItemId& idParent, long cookie);
  62
  63   void AddTestItemsToTree(size_t numChildren,
  64                           size_t depth);
  65
  66   void DoSortChildren(const wxTreeItemId& item, bool reverse = FALSE)
  67     { m_reverseSort = reverse; wxTreeCtrl::SortChildren(item); }
  68   void DoEnsureVisible() { EnsureVisible(m_lastItem); }
  69
  70 protected:
  71   virtual int OnCompareItems(const wxTreeItemId& item1,
  72                              const wxTreeItemId& item2);
  73
  74 private:
  75   void AddItemsRecursively(const wxTreeItemId& idParent,
  76                            size_t nChildren,
  77                            size_t depth,
  78                            size_t folder);
  79
  80   wxImageList *m_imageListNormal;
  81   bool         m_reverseSort;           // flag for OnCompareItems
  82   wxTreeItemId m_lastItem;              // for OnEnsureVisible()
  83
  84   // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
  85   //     if you want your overloaded OnCompareItems() to be called.
  86   //     OTOH, if you don't want it you may omit the next line - this will
  87   //     make default (alphabetical) sorting much faster under wxMSW.
  88   DECLARE_DYNAMIC_CLASS(MyTreeCtrl)
  89   DECLARE_EVENT_TABLE()
  90 };
  91
  92 // Define a new frame type
  93 class MyFrame: public wxFrame
  94 {
  95 public:
  96   // ctor and dtor
  97   MyFrame(const wxString& title, int x, int y, int w, int h);
  98   virtual ~MyFrame();
  99
 100   // menu callbacks
 101   void OnQuit(wxCommandEvent& event);
 102   void OnAbout(wxCommandEvent& event);
 103   void OnDump(wxCommandEvent& event);
 104   void OnDelete(wxCommandEvent& event);
 105   void OnDeleteChildren(wxCommandEvent& event);
 106   void OnDeleteAll(wxCommandEvent& event);
 107   void OnRecreate(wxCommandEvent& event);
 108
 109   void OnSetBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(TRUE); }
 110   void OnClearBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(FALSE); }
 111
 112   void OnEnsureVisible(wxCommandEvent& event);
 113
 114   void OnRename(wxCommandEvent& event);
 115   void OnSort(wxCommandEvent& event) { DoSort(); }
 116   void OnSortRev(wxCommandEvent& event) { DoSort(TRUE); }
 117
 118 private:
 119   void DoSort(bool reverse = FALSE);
 120
 121   MyTreeCtrl *m_treeCtrl;
 122
 123   void DoSetBold(bool bold = TRUE);
 124
 125   DECLARE_EVENT_TABLE()
 126 };
 127
 128 // menu and control ids
 129 enum
 130 {
 131   TreeTest_Quit,
 132   TreeTest_About,
 133   TreeTest_Dump,
 134   TreeTest_Sort,
 135   TreeTest_SortRev,
 136   TreeTest_Bold,
 137   TreeTest_UnBold,
 138   TreeTest_Rename,
 139   TreeTest_Delete,
 140   TreeTest_DeleteChildren,
 141   TreeTest_DeleteAll,
 142   TreeTest_Recreate,
 143   TreeTest_EnsureVisible,
 144   TreeTest_Ctrl = 100
 145 };