191 lines
3.9 KiB
C++
191 lines
3.9 KiB
C++
#ifndef UI_CONTROL_TREEVIEW_H_
|
||
#define UI_CONTROL_TREEVIEW_H_
|
||
|
||
#pragma once
|
||
|
||
namespace ui
|
||
{
|
||
|
||
#define ROOT_NODE_DEPTH -1
|
||
|
||
class TreeView;
|
||
class UILIB_API TreeNode : public ListContainerElement
|
||
{
|
||
public:
|
||
TreeNode();
|
||
TreeNode(const TreeNode& r) = delete;
|
||
TreeNode& operator=(const TreeNode& r) = delete;
|
||
|
||
/// 重写父类方法，提供个性化功能，请参考父类声明
|
||
virtual bool IsVisible() const override;
|
||
virtual void SetInternVisible(bool bVisible) override;
|
||
virtual void SetWindow(Window* pManager, Box* pParent, bool bInit = true) override;
|
||
|
||
/**
|
||
* @brief 子项被点击时触发
|
||
* @param[in] param 消息体
|
||
* @return 始终返回 true
|
||
*/
|
||
virtual bool OnClickItem(EventArgs* param);
|
||
|
||
/**
|
||
* @brief 设置子项所属的属性列表容器
|
||
* @param[in] pTreeView 容器指针
|
||
* @return 无
|
||
*/
|
||
void SetTreeView(TreeView* pTreeView);
|
||
|
||
/**
|
||
* @brief 获取父节点指针
|
||
* @return 返回父节点指针
|
||
*/
|
||
TreeNode* GetParentNode();
|
||
|
||
/**
|
||
* @brief 设置父节点
|
||
* @param[in] pParentTreeNode 父节点指针
|
||
* @return 无
|
||
*/
|
||
void SetParentNode(TreeNode* pParentTreeNode);
|
||
|
||
/**
|
||
* @brief 添加子节点
|
||
* @param[in] pTreeNode 子节点指针
|
||
* @return 成功返回 true，失败返回 false
|
||
*/
|
||
bool AddChildNode(TreeNode* pTreeNode);
|
||
|
||
/**
|
||
* @brief 在指定位置添加子节点
|
||
* @param[in] pTreeNode 子节点指针
|
||
* @param[in] iIndex 要插入哪个位置之后
|
||
* @return 成功返回 true，失败返回 false
|
||
*/
|
||
bool AddChildNodeAt(TreeNode* pTreeNode, std::size_t iIndex);
|
||
|
||
/**
|
||
* @brief 从指定位置移除一个子节点
|
||
* @param[in] index 要移除的子节点索引
|
||
* @return 成功返回 true，失败返回 false
|
||
*/
|
||
bool RemoveChildNodeAt(std::size_t index);
|
||
|
||
/**
|
||
* @brief 根据子节点指针移除一个子节点
|
||
* @param[in] pTreeNode 子节点指针
|
||
* @return 成功返回 true，失败返回 false
|
||
*/
|
||
bool RemoveChildNode(TreeNode* pTreeNode);
|
||
|
||
/**
|
||
* @brief 移除所有子节点
|
||
* @return 无
|
||
*/
|
||
void RemoveAllChildNode();
|
||
|
||
/**
|
||
* @brief 递归获取所有子节点个数
|
||
* @return 返回所有子节点个数
|
||
*/
|
||
int GetDescendantNodeCount();
|
||
|
||
/**
|
||
* @brief 获取下一级子节点个数
|
||
* @return 返回子节点个数
|
||
*/
|
||
std::size_t GetChildNodeCount();
|
||
|
||
/**
|
||
* @brief 获取一个子节点指针
|
||
* @param[in] iIndex 要获取的子节点索引
|
||
* @return 返回子节点指针
|
||
*/
|
||
TreeNode* GetChildNode(std::size_t iIndex);
|
||
|
||
/**
|
||
* @brief 根据子节点指针获取子节点位置索引
|
||
* @param[in] pTreeNode 子节点指针
|
||
* @return 返回位置索引
|
||
*/
|
||
int GetChildNodeIndex(TreeNode* pTreeNode);
|
||
|
||
/**
|
||
* @brief 判断是否展开状态
|
||
* @return 返回 true 为展开状态，否则为 false
|
||
*/
|
||
bool IsExpand() const;
|
||
|
||
/**
|
||
* @brief 设置是否展开显示子节点
|
||
* @param[in] bExpand 为 true 时展开，为 false 是不展开
|
||
* @return 无
|
||
*/
|
||
void SetExpand(bool bExpand);
|
||
|
||
/**
|
||
* @brief 获取子项层级
|
||
* @return 返回当前层级
|
||
*/
|
||
int GetDepth();
|
||
|
||
private:
|
||
/**
|
||
* @brief 删除自身
|
||
* @return 成功返回 true，失败返回 false
|
||
*/
|
||
bool RemoveSelf();
|
||
|
||
private:
|
||
int m_iDepth;
|
||
bool m_bExpand;
|
||
TreeView *m_pTreeView;
|
||
TreeNode *m_pParentTreeNode;
|
||
std::vector<TreeNode*> m_aTreeNodes;
|
||
};
|
||
|
||
class UILIB_API TreeView : public ListBox
|
||
{
|
||
public:
|
||
TreeView(void);
|
||
|
||
/// 重写父类方法，提供个性化功能，请参考父类声明
|
||
virtual void SetAttribute(const std::wstring& strName, const std::wstring& strValue) override;
|
||
|
||
/**
|
||
* @brief 获取根节点
|
||
* @return 返回根节点指针
|
||
*/
|
||
TreeNode* GetRootNode() { return m_rootNode.get(); }
|
||
|
||
/**
|
||
* @brief 获取子节点缩进值
|
||
* @return 返回子节点缩进值
|
||
*/
|
||
int GetIndent() { return m_iIndent; }
|
||
|
||
/**
|
||
* @brief 设置子节点缩进值
|
||
* @param[in] indent 要设置的缩进值
|
||
* @return 无
|
||
*/
|
||
void SetIndent(int indent) { m_iIndent = indent; }
|
||
|
||
private:
|
||
//以下函数故意私有化，表明禁止使用；应该使用TreeNode中的相关函数
|
||
bool Add(Control* pControl) override;
|
||
bool AddAt(Control* pControl, std::size_t iIndex) override;
|
||
bool Remove(Control* pControl) override;
|
||
bool RemoveAt(std::size_t iIndex) override;
|
||
void RemoveAll() override;
|
||
|
||
virtual void SetWindow(Window* pManager, Box* pParent, bool bInit = true) override;
|
||
|
||
private:
|
||
int m_iIndent;
|
||
std::unique_ptr<TreeNode> m_rootNode;
|
||
};
|
||
|
||
}
|
||
|
||
#endif // UI_CONTROL_TREEVIEW_H_
|