samples/treectrl/treetest.h

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        treectrl.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
   9 // Licence:     wxWindows license
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 #define USE_GENERIC_TREECTRL 0
  13
  14 #if USE_GENERIC_TREECTRL
  15 #include "wx/generic/treectlg.h"
  16 #ifndef wxTreeCtrl
  17 #define wxTreeCtrl wxGenericTreeCtrl
  18 #define sm_classwxTreeCtrl sm_classwxGenericTreeCtrl
  19 #endif
  20 #endif
  21
  22 // Define a new application type
  23 class MyApp : public wxApp
  24 {
  25 public:
  26     MyApp() { m_showImages = true; m_showStates = true; m_showButtons = false; }
  27
  28     bool OnInit();
  29
  30     void SetShowImages(bool show) { m_showImages = show; }
  31     bool ShowImages() const { return m_showImages; }
  32
  33     void SetShowStates(bool show) { m_showStates = show; }
  34     bool ShowStates() const { return m_showStates; }
  35
  36     void SetShowButtons(bool show) { m_showButtons = show; }
  37     bool ShowButtons() const { return m_showButtons; }
  38
  39 private:
  40     bool m_showImages, m_showStates, m_showButtons;
  41 };
  42
  43 class MyTreeItemData : public wxTreeItemData
  44 {
  45 public:
  46     MyTreeItemData(const wxString& desc) : m_desc(desc) { }
  47
  48     void ShowInfo(wxTreeCtrl *tree);
  49     const wxChar *GetDesc() const { return m_desc.c_str(); }
  50
  51 private:
  52     wxString m_desc;
  53 };
  54
  55 class MyTreeCtrl : public wxTreeCtrl
  56 {
  57 public:
  58     enum
  59     {
  60         TreeCtrlIcon_File,
  61         TreeCtrlIcon_FileSelected,
  62         TreeCtrlIcon_Folder,
  63         TreeCtrlIcon_FolderSelected,
  64         TreeCtrlIcon_FolderOpened
  65     };
  66
  67     MyTreeCtrl() { m_alternateImages = false; m_alternateStates = false; }
  68     MyTreeCtrl(wxWindow *parent, const wxWindowID id,
  69                const wxPoint& pos, const wxSize& size,
  70                long style);
  71     virtual ~MyTreeCtrl(){};
  72
  73     void OnBeginDrag(wxTreeEvent& event);
  74     void OnBeginRDrag(wxTreeEvent& event);
  75     void OnEndDrag(wxTreeEvent& event);
  76     void OnBeginLabelEdit(wxTreeEvent& event);
  77     void OnEndLabelEdit(wxTreeEvent& event);
  78     void OnDeleteItem(wxTreeEvent& event);
  79     void OnContextMenu(wxContextMenuEvent& event);
  80     void OnItemMenu(wxTreeEvent& event);
  81     void OnGetInfo(wxTreeEvent& event);
  82     void OnSetInfo(wxTreeEvent& event);
  83     void OnItemExpanded(wxTreeEvent& event);
  84     void OnItemExpanding(wxTreeEvent& event);
  85     void OnItemCollapsed(wxTreeEvent& event);
  86     void OnItemCollapsing(wxTreeEvent& event);
  87     void OnSelChanged(wxTreeEvent& event);
  88     void OnSelChanging(wxTreeEvent& event);
  89     void OnTreeKeyDown(wxTreeEvent& event);
  90     void OnItemActivated(wxTreeEvent& event);
  91     void OnItemStateClick(wxTreeEvent& event);
  92     void OnItemRClick(wxTreeEvent& event);
  93
  94     void OnRMouseDown(wxMouseEvent& event);
  95     void OnRMouseUp(wxMouseEvent& event);
  96     void OnRMouseDClick(wxMouseEvent& event);
  97
  98     void GetItemsRecursively(const wxTreeItemId& idParent,
  99                              wxTreeItemIdValue cookie = 0);
 100
 101     void CreateImageList(int size = 16);
 102     void CreateButtonsImageList(int size = 11);
 103     void CreateStateImageList(bool del = false);
 104
 105     void AddTestItemsToTree(size_t numChildren, size_t depth);
 106
 107     void DoSortChildren(const wxTreeItemId& item, bool reverse = false)
 108         { m_reverseSort = reverse; wxTreeCtrl::SortChildren(item); }
 109     void DoEnsureVisible() { if (m_lastItem.IsOk()) EnsureVisible(m_lastItem); }
 110
 111     void DoToggleIcon(const wxTreeItemId& item);
 112     void DoToggleState(const wxTreeItemId& item);
 113
 114     void ShowMenu(wxTreeItemId id, const wxPoint& pt);
 115
 116     int ImageSize(void) const { return m_imageSize; }
 117
 118     void SetLastItem(wxTreeItemId id) { m_lastItem = id; }
 119
 120     void SetAlternateImages(bool show) { m_alternateImages = show; }
 121     bool AlternateImages() const { return m_alternateImages; }
 122
 123     void SetAlternateStates(bool show) { m_alternateStates = show; }
 124     bool AlternateStates() const { return m_alternateStates; }
 125
 126     void ResetBrokenStateImages()
 127     {
 128         const size_t count = GetStateImageList()->GetImageCount();
 129         int state = count > 0 ? count - 1 : wxTREE_ITEMSTATE_NONE;
 130         DoResetBrokenStateImages(GetRootItem(), 0, state);
 131     }
 132
 133 protected:
 134     virtual int OnCompareItems(const wxTreeItemId& i1, const wxTreeItemId& i2);
 135
 136     // is this the test item which we use in several event handlers?
 137     bool IsTestItem(const wxTreeItemId& item)
 138     {
 139         // the test item is the first child folder
 140         return GetItemParent(item) == GetRootItem() && !GetPrevSibling(item);
 141     }
 142
 143 private:
 144     void AddItemsRecursively(const wxTreeItemId& idParent,
 145                              size_t nChildren,
 146                              size_t depth,
 147                              size_t folder);
 148
 149     void DoResetBrokenStateImages(const wxTreeItemId& idParent,
 150                                   wxTreeItemIdValue cookie, int state);
 151
 152     void LogEvent(const wxChar *name, const wxTreeEvent& event);
 153
 154     int          m_imageSize;               // current size of images
 155     bool         m_reverseSort;             // flag for OnCompareItems
 156     wxTreeItemId m_lastItem,                // for OnEnsureVisible()
 157                  m_draggedItem;             // item being dragged right now
 158     bool         m_alternateImages;
 159     bool         m_alternateStates;
 160
 161     // NB: due to an ugly wxMSW hack you _must_ use DECLARE_DYNAMIC_CLASS()
 162     //     if you want your overloaded OnCompareItems() to be called.
 163     //     OTOH, if you don't want it you may omit the next line - this will
 164     //     make default (alphabetical) sorting much faster under wxMSW.
 165     DECLARE_DYNAMIC_CLASS(MyTreeCtrl)
 166     DECLARE_EVENT_TABLE()
 167 };
 168
 169 // Define a new frame type
 170 class MyFrame: public wxFrame
 171 {
 172 public:
 173     // ctor and dtor
 174     MyFrame(const wxString& title, int x, int y, int w, int h);
 175     virtual ~MyFrame();
 176
 177     // menu callbacks
 178     void OnQuit(wxCommandEvent& event);
 179     void OnAbout(wxCommandEvent& event);
 180     void OnClearLog(wxCommandEvent& event);
 181
 182     void OnTogButtons(wxCommandEvent& event)
 183         { TogStyle(event.GetId(), wxTR_HAS_BUTTONS); }
 184     void OnTogTwist(wxCommandEvent& event)
 185         { TogStyle(event.GetId(), wxTR_TWIST_BUTTONS); }
 186     void OnTogLines(wxCommandEvent& event)
 187         { TogStyle(event.GetId(), wxTR_NO_LINES); }
 188     void OnTogEdit(wxCommandEvent& event)
 189         { TogStyle(event.GetId(), wxTR_EDIT_LABELS); }
 190     void OnTogHideRoot(wxCommandEvent& event)
 191         { TogStyle(event.GetId(), wxTR_HIDE_ROOT); }
 192     void OnTogRootLines(wxCommandEvent& event)
 193         { TogStyle(event.GetId(), wxTR_LINES_AT_ROOT); }
 194     void OnTogBorder(wxCommandEvent& event)
 195         { TogStyle(event.GetId(), wxTR_ROW_LINES); }
 196     void OnTogFullHighlight(wxCommandEvent& event)
 197         { TogStyle(event.GetId(), wxTR_FULL_ROW_HIGHLIGHT); }
 198
 199     void OnResetStyle(wxCommandEvent& WXUNUSED(event))
 200         { CreateTreeWithDefStyle(); }
 201
 202     void OnSetFgColour(wxCommandEvent& event);
 203     void OnSetBgColour(wxCommandEvent& event);
 204
 205     void OnHighlight(wxCommandEvent& event);
 206     void OnDump(wxCommandEvent& event);
 207 #ifndef NO_MULTIPLE_SELECTION
 208     void OnDumpSelected(wxCommandEvent& event);
 209     void OnSelect(wxCommandEvent& event);
 210     void OnUnselect(wxCommandEvent& event);
 211     void OnToggleSel(wxCommandEvent& event);
 212 #endif // NO_MULTIPLE_SELECTION
 213     void OnSelectRoot(wxCommandEvent& event);
 214     void OnDelete(wxCommandEvent& event);
 215     void OnDeleteChildren(wxCommandEvent& event);
 216     void OnDeleteAll(wxCommandEvent& event);
 217
 218     void OnRecreate(wxCommandEvent& event);
 219     void OnToggleButtons(wxCommandEvent& event);
 220     void OnToggleImages(wxCommandEvent& event);
 221     void OnToggleStates(wxCommandEvent& event);
 222     void OnToggleAlternateImages(wxCommandEvent& event);
 223     void OnToggleAlternateStates(wxCommandEvent& event);
 224     void OnSetImageSize(wxCommandEvent& event);
 225     void OnCollapseAndReset(wxCommandEvent& event);
 226
 227     void OnSetBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(true); }
 228     void OnClearBold(wxCommandEvent& WXUNUSED(event)) { DoSetBold(false); }
 229
 230     void OnEnsureVisible(wxCommandEvent& event);
 231     void OnSetFocus(wxCommandEvent& event);
 232
 233     void OnCount(wxCommandEvent& event);
 234     void OnCountRec(wxCommandEvent& event);
 235
 236     void OnRename(wxCommandEvent& event);
 237     void OnSort(wxCommandEvent& WXUNUSED(event)) { DoSort(); }
 238     void OnSortRev(wxCommandEvent& WXUNUSED(event)) { DoSort(true); }
 239
 240     void OnAddItem(wxCommandEvent& event);
 241     void OnInsertItem(wxCommandEvent& event);
 242
 243     void OnIncIndent(wxCommandEvent& event);
 244     void OnDecIndent(wxCommandEvent& event);
 245
 246     void OnIncSpacing(wxCommandEvent& event);
 247     void OnDecSpacing(wxCommandEvent& event);
 248
 249     void OnToggleIcon(wxCommandEvent& event);
 250     void OnToggleState(wxCommandEvent& event);
 251
 252     void OnShowFirstVisible(wxCommandEvent& WXUNUSED(event))
 253         { DoShowFirstOrLast(&wxTreeCtrl::GetFirstVisibleItem, "first visible"); }
 254 #ifdef wxHAS_LAST_VISIBLE // we don't have it currently but may add later
 255     void OnShowLastVisible(wxCommandEvent& WXUNUSED(event))
 256         { DoShowFirstOrLast(&wxTreeCtrl::GetLastVisibleItem, "last visible"); }
 257 #endif // wxHAS_LAST_VISIBLE
 258
 259     void OnShowNextVisible(wxCommandEvent& WXUNUSED(event))
 260         { DoShowRelativeItem(&wxTreeCtrl::GetNextVisible, "next visible"); }
 261     void OnShowPrevVisible(wxCommandEvent& WXUNUSED(event))
 262         { DoShowRelativeItem(&wxTreeCtrl::GetPrevVisible, "previous visible"); }
 263
 264     void OnShowParent(wxCommandEvent& WXUNUSED(event))
 265         { DoShowRelativeItem(&wxTreeCtrl::GetItemParent, "parent"); }
 266     void OnShowPrevSibling(wxCommandEvent& WXUNUSED(event))
 267         { DoShowRelativeItem(&wxTreeCtrl::GetPrevSibling, "previous sibling"); }
 268     void OnShowNextSibling(wxCommandEvent& WXUNUSED(event))
 269         { DoShowRelativeItem(&wxTreeCtrl::GetNextSibling, "next sibling"); }
 270
 271     void OnIdle(wxIdleEvent& event);
 272     void OnSize(wxSizeEvent& event);
 273
 274 private:
 275     void TogStyle(int id, long flag);
 276
 277     void DoSort(bool reverse = false);
 278
 279     void Resize();
 280
 281     void CreateTreeWithDefStyle();
 282     void CreateTree(long style);
 283
 284     // common parts of OnShowFirst/LastVisible() and OnShowNext/PrevVisible()
 285     typedef wxTreeItemId (wxTreeCtrl::*TreeFunc0_t)() const;
 286     void DoShowFirstOrLast(TreeFunc0_t pfn, const wxString& label);
 287
 288     typedef wxTreeItemId (wxTreeCtrl::*TreeFunc1_t)(const wxTreeItemId&) const;
 289     void DoShowRelativeItem(TreeFunc1_t pfn, const wxString& label);
 290
 291
 292     wxPanel *m_panel;
 293     MyTreeCtrl *m_treeCtrl;
 294 #if wxUSE_LOG
 295     wxTextCtrl *m_textCtrl;
 296 #endif // wxUSE_LOG
 297
 298     void DoSetBold(bool bold = true);
 299
 300     DECLARE_EVENT_TABLE()
 301 };
 302
 303 // menu and control ids
 304 enum
 305 {
 306     TreeTest_Quit = wxID_EXIT,
 307     TreeTest_About = wxID_ABOUT,
 308     TreeTest_ClearLog = wxID_CLEAR,
 309     TreeTest_TogButtons = wxID_HIGHEST,
 310     TreeTest_TogTwist,
 311     TreeTest_TogLines,
 312     TreeTest_TogEdit,
 313     TreeTest_TogHideRoot,
 314     TreeTest_TogRootLines,
 315     TreeTest_TogBorder,
 316     TreeTest_TogFullHighlight,
 317     TreeTest_SetFgColour,
 318     TreeTest_SetBgColour,
 319     TreeTest_ResetStyle,
 320     TreeTest_Highlight,
 321     TreeTest_Dump,
 322     TreeTest_DumpSelected,
 323     TreeTest_Count,
 324     TreeTest_CountRec,
 325     TreeTest_Sort,
 326     TreeTest_SortRev,
 327     TreeTest_SetBold,
 328     TreeTest_ClearBold,
 329     TreeTest_Rename,
 330     TreeTest_Delete,
 331     TreeTest_DeleteChildren,
 332     TreeTest_DeleteAll,
 333     TreeTest_Recreate,
 334     TreeTest_ToggleImages,
 335     TreeTest_ToggleStates,
 336     TreeTest_ToggleAlternateImages,
 337     TreeTest_ToggleAlternateStates,
 338     TreeTest_ToggleButtons,
 339     TreeTest_SetImageSize,
 340     TreeTest_ToggleSel,
 341     TreeTest_CollapseAndReset,
 342     TreeTest_EnsureVisible,
 343     TreeTest_SetFocus,
 344     TreeTest_AddItem,
 345     TreeTest_InsertItem,
 346     TreeTest_IncIndent,
 347     TreeTest_DecIndent,
 348     TreeTest_IncSpacing,
 349     TreeTest_DecSpacing,
 350     TreeTest_ToggleIcon,
 351     TreeTest_ToggleState,
 352     TreeTest_Select,
 353     TreeTest_Unselect,
 354     TreeTest_SelectRoot,
 355     TreeTest_ShowFirstVisible,
 356     TreeTest_ShowLastVisible,
 357     TreeTest_ShowNextVisible,
 358     TreeTest_ShowPrevVisible,
 359     TreeTest_ShowParent,
 360     TreeTest_ShowPrevSibling,
 361     TreeTest_ShowNextSibling,
 362     TreeTest_Ctrl = 1000
 363 };