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     const char *GetDesc() const { return m_desc.c_str(); }
  26
  27 private:
  28     wxString m_desc;
  29 };
  30
  31 class MyTreeCtrl : public wxTreeCtrl
  32 {
  33 public:
  34     enum
  35     {
  36         TreeCtrlIcon_File,
  37         TreeCtrlIcon_FileSelected,
  38         TreeCtrlIcon_Folder,
  39         TreeCtrlIcon_FolderSelected,
  40         TreeCtrlIcon_FolderOpened
  41     };
  42
  43     MyTreeCtrl() { }
  44     MyTreeCtrl(wxWindow *parent, const wxWindowID id,
  45                const wxPoint& pos, const wxSize& size,
  46                long style);
  47     virtual ~MyTreeCtrl();
  48
  49     void OnBeginDrag(wxTreeEvent& event);
  50     void OnBeginRDrag(wxTreeEvent& event);
  51     void OnEndDrag(wxTreeEvent& event);
  52     void OnBeginLabelEdit(wxTreeEvent& event);
  53     void OnEndLabelEdit(wxTreeEvent& event);
  54     void OnDeleteItem(wxTreeEvent& event);
  55     void OnGetInfo(wxTreeEvent& event);
  56     void OnSetInfo(wxTreeEvent& event);
  57     void OnItemExpanded(wxTreeEvent& event);
  58     void OnItemExpanding(wxTreeEvent& event);
  59     void OnItemCollapsed(wxTreeEvent& event);
  60     void OnItemCollapsing(wxTreeEvent& event);
  61     void OnSelChanged(wxTreeEvent& event);
  62     void OnSelChanging(wxTreeEvent& event);
  63     void OnTreeKeyDown(wxTreeEvent& event);
  64     void OnItemActivated(wxTreeEvent& event);
  65     void OnRMouseDClick(wxMouseEvent& event);
  66
  67     void GetItemsRecursively(const wxTreeItemId& idParent, long cookie);
  68
  69     void AddTestItemsToTree(size_t numChildren, size_t depth);
  70
  71     void DoSortChildren(const wxTreeItemId& item, bool reverse = FALSE)
  72         { m_reverseSort = reverse; wxTreeCtrl::SortChildren(item); }
  73     void DoEnsureVisible() { EnsureVisible(m_lastItem); }
  74
  75     void DoToggleIcon(const wxTreeItemId& item);
  76
  77 protected:
  78     virtual int OnCompareItems(const wxTreeItemId& i1, const wxTreeItemId& i2);
  79
  80     // is this the test item which we use in several event handlers?
  81     bool IsTestItem(const wxTreeItemId& item)
  82     {
  83         // the test item is the first child folder
  84         return GetParent(item) == GetRootItem() && !GetPrevSibling(item);
  85     }
  86
  87 private:
  88     void AddItemsRecursively(const wxTreeItemId& idParent,
  89                              size_t nChildren,
  90                              size_t depth,
  91                              size_t folder);
  92
  93     wxImageList *m_imageListNormal;
  94     bool         m_reverseSort;             // flag for OnCompareItems
  95     wxTreeItemId m_lastItem,                // for OnEnsureVisible()
  96                  m_draggedItem;             // item being dragged right now
  97
  98     // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
  99     //     if you want your overloaded OnCompareItems() to be called.
 100     //     OTOH, if you don't want it you may omit the next line - this will
 101     //     make default (alphabetical) sorting much faster under wxMSW.
 102     DECLARE_DYNAMIC_CLASS(MyTreeCtrl)
 103     DECLARE_EVENT_TABLE()
 104 };
 105
 106 // Define a new frame type
 107 class MyFrame: public wxFrame
 108 {
 109 public:
 110     // ctor and dtor
 111     MyFrame(const wxString& title, int x, int y, int w, int h);
 112     virtual ~MyFrame();
 113
 114     // menu callbacks
 115     void OnQuit(wxCommandEvent& event);
 116     void OnAbout(wxCommandEvent& event);
 117
 118     void OnDump(wxCommandEvent& event);
 119 #ifndef NO_MULTIPLE_SELECTION
 120     void OnDumpSelected(wxCommandEvent& event);
 121     void OnSelect(wxCommandEvent& event);
 122     void OnUnselect(wxCommandEvent& event);
 123 #endif // NO_MULTIPLE_SELECTION
 124     void OnDelete(wxCommandEvent& event);
 125     void OnDeleteChildren(wxCommandEvent& event);
 126     void OnDeleteAll(wxCommandEvent& event);
 127     void OnRecreate(wxCommandEvent& event);
 128     void OnCollapseAndReset(wxCommandEvent& event);
 129
 130     void OnSetBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(TRUE); }
 131     void OnClearBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(FALSE); }
 132
 133     void OnEnsureVisible(wxCommandEvent& event);
 134
 135     void OnCount(wxCommandEvent& event);
 136     void OnCountRec(wxCommandEvent& event);
 137
 138     void OnRename(wxCommandEvent& event);
 139     void OnSort(wxCommandEvent& event) { DoSort(); }
 140     void OnSortRev(wxCommandEvent& event) { DoSort(TRUE); }
 141
 142     void OnAddItem(wxCommandEvent& event);
 143     void OnInsertItem(wxCommandEvent& event);
 144
 145     void OnIncIndent(wxCommandEvent& event);
 146     void OnDecIndent(wxCommandEvent& event);
 147
 148     void OnIncSpacing(wxCommandEvent& event);
 149     void OnDecSpacing(wxCommandEvent& event);
 150
 151     void OnToggleIcon(wxCommandEvent& event);
 152
 153 private:
 154     void DoSort(bool reverse = FALSE);
 155
 156     MyTreeCtrl *m_treeCtrl;
 157
 158     void DoSetBold(bool bold = TRUE);
 159
 160     DECLARE_EVENT_TABLE()
 161 };
 162
 163 // menu and control ids
 164 enum
 165 {
 166     TreeTest_Quit,
 167     TreeTest_About,
 168     TreeTest_Dump,
 169     TreeTest_DumpSelected,
 170     TreeTest_Count,
 171     TreeTest_CountRec,
 172     TreeTest_Sort,
 173     TreeTest_SortRev,
 174     TreeTest_Bold,
 175     TreeTest_UnBold,
 176     TreeTest_Rename,
 177     TreeTest_Delete,
 178     TreeTest_DeleteChildren,
 179     TreeTest_DeleteAll,
 180     TreeTest_Recreate,
 181     TreeTest_CollapseAndReset,
 182     TreeTest_EnsureVisible,
 183     TreeTest_AddItem,
 184     TreeTest_InsertItem,
 185     TreeTest_IncIndent,
 186     TreeTest_DecIndent,
 187     TreeTest_IncSpacing,
 188     TreeTest_DecSpacing,
 189     TreeTest_ToggleIcon,
 190     TreeTest_Select,
 191     TreeTest_Unselect,
 192     TreeTest_Ctrl = 1000
 193 };