TreeView.h

#include "ImageList.h"
#include "ItemsControl.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;
 
        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();
    };
 
    // clang-format off
 
    struct TreeViewItemExpandingEventArgs : TypedRoutedEventArgs<TreeView_ItemExpanding, CancelableEventArgs> {
        bool action;       // true表示展开，false表示折叠
        TreeViewNode node; // 正在展开或折叠的节点
        TreeViewItemExpandingEventArgs(bool action, const TreeViewNode &node): action(action), node(node) {}
    };
 
    struct TreeViewItemExpandedEventArgs : TypedRoutedEventArgs<TreeView_ItemExpanding> {
        bool action;       // true表示展开，false表示折叠
        TreeViewNode node; // 已展开或折叠的节点
        TreeViewItemExpandedEventArgs(bool action, const TreeViewNode &node): action(action), node(node) {}
    };
 
    struct TreeViewCheckStateChangedEventArgs : TypedRoutedEventArgs<TreeView_CheckStateChanged> {
        int checkState;    // 复选框的新状态，0表示未选中，1表示选中，-1表示无复选框
        TreeViewNode node; // 复选框状态改变的节点
        TreeViewCheckStateChangedEventArgs(int checkState, const TreeViewNode &node): checkState(checkState), node(node) {}
    };
 
    // clang-format on
 
    class TreeView : public ItemsControl<TreeViewNode>
    {
    private:
        using TBase = ItemsControl<TreeViewNode>;
 
    public:
        const ReadOnlyProperty<TreeViewNode> Root;
 
        const ReadOnlyProperty<int> AllItemsCount;
 
        const Property<bool> CheckBoxes;
 
        const Property<Color> LineColor;
 
        const Property<double> IndentWidth;
 
    public:
        TreeView();
 
    protected:
        virtual int GetItemsCount() override;
 
        virtual int GetSelectedIndex() override;
 
        virtual void SetSelectedIndex(int index) override;
 
        virtual TreeViewNode GetSelectedItem() override;
 
        virtual void SetBackColor(Color color, bool redraw) override;
 
        virtual void SetTextColor(Color color, bool redraw) override;
 
        virtual bool OnNotified(NMHDR *pNMHDR, LRESULT &result) override;
 
        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:
        virtual void Clear() override;
 
        virtual TreeViewNode GetItemAt(int index) override;
 
        virtual bool AddItem(const TreeViewNode &item) override;
 
        virtual bool InsertItem(int index, const TreeViewNode &item) override;
 
        virtual bool UpdateItem(int index, const TreeViewNode &newValue) override;
 
        virtual bool RemoveItemAt(int index) override;
 
        TreeViewNode AddItem(const std::wstring &text);
 
        TreeViewNode InsertItem(int index, const std::wstring &text);
 
        ImageList GetImageList(TreeViewImageList imageList);
 
        HIMAGELIST SetImageList(TreeViewImageList imageList, HIMAGELIST value);
 
    private:
        TreeViewNode _GetRoot();
 
        TreeViewNode _InsertItem(HTREEITEM hParent, HTREEITEM hInsertAfter, const std::wstring &text);
    };
}