TreeView.h

#include "Control.h"
#include "ImageList.h"
#include <commctrl.h>
 
namespace sw
{
    enum class TreeViewImageList {
        Normal = TVSIL_NORMAL, 
        State  = TVSIL_STATE,  
    };
 
    class TreeViewNode : public IToString<TreeViewNode>,
                         public IEqualityComparable<TreeViewNode>
    {
    private:
        HWND _hwnd;
 
        HTREEITEM _hitem;
 
    public:
        TreeViewNode() = default;
 
        TreeViewNode(HWND hwnd, HTREEITEM hitem);
 
        HWND GetOwnerHandle() const;
 
        HTREEITEM GetHandle() const;
 
        std::wstring ToString() const;
 
        bool Equals(const TreeViewNode &other) const;
 
        bool IsNull() const;
 
        explicit operator bool() const;
 
        std::wstring GetText() const;
 
        bool SetText(const std::wstring &text);
 
        TreeViewNode GetParent() const;
 
        TreeViewNode GetNextNode() const;
 
        TreeViewNode GetPreviousNode() const;
 
        TreeViewNode GetFirstChildNode() const;
 
        TreeViewNode InsertAfter(const std::wstring &text);
 
        TreeViewNode AddChild(const std::wstring &text);
 
        bool IsSelected() const;
 
        bool Select();
 
        bool Delete();
 
        bool IsExpanded() const;
 
        bool SetExpand(bool expand);
 
        bool Expand();
 
        bool Collapse();
 
        void *GetUserData() const;
 
        bool SetUserData(void *data);
 
        bool IsChecked() const;
 
        void SetCheck(bool check);
 
        bool SetImages(int imageIndex, int selectedImageIndex);
 
        int GetChildCount() const;
 
        int DeleteAllChildren();
    };
 
    class TreeViewItemExpandingEventArgs : public TypedRoutedEventArgs<TreeView_ItemExpanding, CancelableEventArgs>
    {
    public:
        bool action;
 
        TreeViewNode node;
 
        TreeViewItemExpandingEventArgs(bool action, const TreeViewNode &node)
            : action(action), node(node)
        {
        }
    };
 
    class TreeViewItemExpandedEventArgs : public TypedRoutedEventArgs<TreeView_ItemExpanding>
    {
    public:
        bool action;
 
        TreeViewNode node;
 
        TreeViewItemExpandedEventArgs(bool action, const TreeViewNode &node)
            : action(action), node(node)
        {
        }
    };
 
    class TreeViewCheckStateChangedEventArgs : public TypedRoutedEventArgs<TreeView_CheckStateChanged>
    {
    public:
        int checkState;
 
        TreeViewNode node;
 
        TreeViewCheckStateChangedEventArgs(int checkState, const TreeViewNode &node)
            : checkState(checkState), node(node)
        {
        }
    };
 
    class TreeView : public Control
    {
    private:
        using TBase = Control;
 
    public:
        const ReadOnlyProperty<TreeViewNode> Root;
 
        const ReadOnlyProperty<TreeViewNode> SelectedItem;
 
        const ReadOnlyProperty<int> AllItemsCount;
 
        const Property<bool> CheckBoxes;
 
        const Property<Color> LineColor;
 
        const Property<double> IndentWidth;
 
    public:
        TreeView();
 
    protected:
        virtual void OnSetBackColor(Color color, bool redraw) override;
 
        virtual void OnSetTextColor(Color color, bool redraw) override;
 
        virtual bool OnNotified(NMHDR *pNMHDR, LRESULT &result) override;
 
        virtual void OnSelectionChanged();
 
        virtual bool OnClicked(NMHDR *pNMHDR, LRESULT &result);
 
        virtual bool OnDoubleClicked(NMHDR *pNMHDR, LRESULT &result);
 
        virtual void OnGetDispInfo(NMTVDISPINFOW *pNMInfo);
 
        virtual bool OnItemExpanding(NMTREEVIEWW *pNMTV);
 
        virtual void OnItemExpanded(NMTREEVIEWW *pNMTV);
 
        virtual void OnItemChanged(NMTVITEMCHANGE *pNMInfo);
 
    public:
        void Clear();
 
        TreeViewNode AddItem(const std::wstring &text);
 
        ImageList GetImageList(TreeViewImageList imageList);
 
        HIMAGELIST SetImageList(TreeViewImageList imageList, HIMAGELIST value);
 
    private:
        TreeViewNode _GetRoot();
 
        TreeViewNode _GetSelectedItem();
 
        TreeViewNode _InsertItem(HTREEITEM hParent, HTREEITEM hInsertAfter, const std::wstring &text);
    };
}