GUITreeCtrl.h

#pragma once
 
#include <set>
#include "DefinitionsGUI.h"
#include "GUITreeCtrlBase.h"
 
namespace FLImaging
{
    namespace GUI
    {
        class CGUIMenuItemBase;
        class CGUIDialogAddSNAPNodeDisplay;
        class CGUIPaneExampleSNAP;
        class CGUIPaneMenu;
        class CGUIPaneBase;
 
        typedef struct sMatchResult
        {
        public:
            sMatchResult(int32_t _i32Start, int32_t _i32End, int32_t _i32AliasIndex)
                : i32Start(_i32Start), i32End(_i32End), i32AliasIndex(_i32AliasIndex){}
 
            int32_t i32Start;
            int32_t i32End;
            int32_t i32AliasIndex;
        }SMatchResult;
 
        class FL_EXPORT CGUITreeCtrl : public CGUITreeCtrlBase
        {
        public:
            CGUITreeCtrl() noexcept;
            CGUITreeCtrl(const CGUITreeCtrl& rh);
            virtual ~CGUITreeCtrl();
 
        public:

            virtual bool RecursiveExpand(HTREEITEM hItem, UINT nCode, bool bSaveToConfig = true) override;

            virtual void GetVerticalRange(HTREEITEM hItemRangeStart, HTREEITEM hItemRangeEnd, std::set<HTREEITEM>& setOut);

            virtual void ClearMultiselectItems();

            virtual HTREEITEM FindHandle(void* pData, HTREEITEM hParent = 0);

            virtual HTREEITEM FindHandle(const Base::CFLString<wchar_t>& strPath, HTREEITEM hParent = 0);

            virtual HTREEITEM FindHandleByFullPath(const Base::CFLString<wchar_t>& strFullPath);

            virtual const CResult SortTreeItem(HTREEITEM hItem = NULL, ESortOrder eSortOrder = ESortOrder_Ascending);

            virtual const CResult SetRootItemName(const Base::CFLString<wchar_t>& strRootItemName);

            virtual const Base::CFLString<wchar_t>& GetRootItemName();
            
            virtual const CResult EnableDragAndDropItem(bool bEnable);

            virtual bool IsDragAndDropItemEnabled();
            
            virtual const CResult SetDefaultImageList();
 
            virtual bool SearchNext();
            virtual bool SearchPrev();
            virtual bool SearchFirst();
            virtual bool SearchLast();
 
            virtual void SetParentPane(CGUIPaneBase* pParentPane);
            virtual CGUIPaneBase* GetParentPane() const;
 
            virtual void SetParentPaneType(EPaneType eParentPaneType);
            virtual EPaneType GetParentPaneType() const;
            
            virtual void EnableAutoSaveExpandStatus(bool bEnable);
            virtual bool IsAutoSaveExpandStatusEnabled() const;
 
            virtual void SetFoundIndex(int32_t i32Idx);
            virtual int32_t GetFoundIndex();
            virtual int32_t GetFoundIndex(HTREEITEM hItem);
            virtual int32_t GetFoundCount();
            virtual HTREEITEM GetFoundItem(int32_t i32Index);
 
            virtual void EnableItem(CGUIMenuItemBase* pMenuItem, bool bEnable);
 
            virtual HTREEITEM MoveTreeItem(HTREEITEM hItem, HTREEITEM hItemTo, HTREEITEM hItemPos = TVI_LAST);
 
        public:
            DECLARE_MESSAGE_MAP()
            virtual BOOL OnNotify(WPARAM wParam, LPARAM lParam, LRESULT* pResult);
            afx_msg void OnNMCustomdraw(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnNMClick(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnLButtonDblClk(UINT nFlags, CPoint point);
            afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
            afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
            afx_msg void OnMouseMove(UINT nFlags, CPoint point);
            afx_msg HBRUSH CtlColor(CDC* /*pDC*/, UINT /*nCtlColor*/);
            afx_msg BOOL OnEraseBkgnd(CDC* pDC);
            afx_msg void OnTvnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnSelchanging(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnSelchanged(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnItemChanged(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnItemChanging(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnBegindrag(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnBeginrdrag(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg void OnTvnDeleteitem(NMHDR* pNMHDR, LRESULT* pResult);
            afx_msg BOOL OnMouseWheel(UINT nFlags, short zDelta, CPoint pt);
 
            afx_msg LRESULT OnCancelAddNode(WPARAM wParam, LPARAM lParam);
            afx_msg LRESULT OnUnselectItem(WPARAM wParam, LPARAM lParam);
 
            virtual BOOL PreTranslateMessage(MSG* pMsg);
 
        protected:
            virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam);
 
        protected:
            CGUIMenuItemBase* GetSNAPNodePropertyMenu(HTREEITEM hItem);
            virtual void InternalSortTreeItem(HTREEITEM hItem, ESortOrder eSortOrder, bool bAllowRoot = false);
            static int CALLBACK CompareDesendingProc(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort);
            static int CALLBACK CompareAscendingProc(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort);
 
        protected:
            Base::CFLString<wchar_t> m_strRootItemName;
            int32_t m_i32FoundIdx;
            std::vector<std::pair<HTREEITEM, std::vector<SMatchResult>>>& m_vctFound;
            std::set<CGUIMenuItemBase*>& m_setMenuItemOnFavorite;
            std::set<CGUIMenuItemBase*>& m_setMenuItemHided;
            std::unordered_map<void*, HTREEITEM>& m_mapDataAndHTreeItem;
            std::unordered_map<uint32_t, HTREEITEM>& m_mapFullPathHashAndHTreeItem;
            HTREEITEM m_hItemVertStart;
            std::set<HTREEITEM>& m_setMultiSelect;
            CImageList m_imgListDefault;
 
            bool m_bDragState;
            bool m_bUnselectItem;
            bool m_bEnableDragAndDropItem;
            bool m_bEnableAutoSaveTreeItemExpandStatus;
 
            HTREEITEM m_hItemUnselect;
            HTREEITEM m_hItemLButtonDown;
            EPaneType m_eParentPaneType;
            TVITEMW m_itemDrag;
 
            CGUIDialogAddSNAPNodeDisplay* m_pDlgAddNode;
            CGUIPaneBase* m_pParentPane;
 
            friend class CGUIManager;
            friend class CGUIPaneMenu;
            friend class CGUIPaneExampleSNAP;
            friend class CGUITreeCtrlSearchUtils;
        };
    }
}