tocctrlmodel.h

#ifndef FEATURE_TOC_TOCCTRLMODEL_H
#define FEATURE_TOC_TOCCTRLMODEL_H
 
#include "wx/wxprec.h"
#ifndef WX_PRECOMP
#include "wx/wx.h"
#endif
 
#include <wx/dataview.h>
 
#include "tmlayerproperties.h"
 
class TocCtrlModelNode;
 
WX_DEFINE_ARRAY_PTR(TocCtrlModelNode*, TocCtrlModelNodePtrArray);
 
class TocCtrlModelNode {
  public:
    TocCtrlModelNode(TocCtrlModelNode* parent, tmLayerProperties* layerprop);
 
    ~TocCtrlModelNode();
 
    void GetAllChildRecursive(TocCtrlModelNodePtrArray& node_array, TocCtrlModelNode* startnode = nullptr);
    bool IsMyChildren(TocCtrlModelNode* node);
 
    bool IsContainer() const;
 
    TocCtrlModelNode* GetParent();
 
    TocCtrlModelNodePtrArray& GetChildren();
 
    void Insert(TocCtrlModelNode* child, unsigned int n);
 
    void Append(TocCtrlModelNode* child);
 
    unsigned int GetChildCount() const;
 
    // public to avoid getters/setters
    bool m_Container;
    tmLayerProperties* m_LayerProp;
 
  private:
    TocCtrlModelNode* m_Parent;
    TocCtrlModelNodePtrArray m_Children;
};
 
class TocCtrlModel : public wxDataViewModel {
  public:
    TocCtrlModel();
 
    ~TocCtrlModel() {
        delete m_Root;
    }
 
    bool IsChecked(const wxDataViewItem& item) const;
    void SetChecked(const wxDataViewItem& item, bool check = true);
 
    wxString NodeGetTitle(TocCtrlModelNode* node);
    bool NodeSetTitle(TocCtrlModelNode* node, const wxString& title);
 
    TocCtrlModelNode* NodeAdd(TocCtrlModelNode* parent, tmLayerProperties* layerprop);
    TocCtrlModelNode* NodeInsert(TocCtrlModelNode* parent, tmLayerProperties* layerprop, int index = 0);
    bool NodeMove(wxDataViewItemArray& selectedItems, TocCtrlModelNode* destination, int proposedIndex = wxNOT_FOUND);
    void NodeRecursiveAdd(TocCtrlModelNode* parent, TocCtrlModelNode* start);
    void NodeRecursiveRemove(TocCtrlModelNode* start);
 
    // model function
    void Delete(const wxDataViewItem& item);
 
    void Clear();
 
    virtual void GetValue(wxVariant& variant, const wxDataViewItem& item, unsigned int col) const wxOVERRIDE;
 
    virtual bool SetValue(const wxVariant& variant, const wxDataViewItem& item, unsigned int col) wxOVERRIDE;
 
    virtual bool IsEnabled(const wxDataViewItem& item, unsigned int col) const wxOVERRIDE;
 
    virtual wxDataViewItem GetParent(const wxDataViewItem& item) const wxOVERRIDE;
 
    wxDataViewItem GetRoot() const;
 
    virtual bool IsContainer(const wxDataViewItem& item) const wxOVERRIDE;
 
    virtual unsigned int GetChildren(const wxDataViewItem& parent, wxDataViewItemArray& array) const wxOVERRIDE;
 
    virtual unsigned int GetColumnCount() const wxOVERRIDE;
    virtual wxString GetColumnType(unsigned int) const wxOVERRIDE;
 
    static wxDataViewItem ConvertFromNode(const TocCtrlModelNode* node);
    static TocCtrlModelNode* ConvertFromDataViewItem(const wxDataViewItem& item);
 
  private:
    TocCtrlModelNode* m_Root;
    wxImageList m_ImageList;
};
 
#endif  // FEATURE_TOC_TOCCTRLMODEL_H