Trie.h

#pragma once
#include <unordered_map>
#include <string>
#include <vector>
 
namespace FLImaging
{
    namespace Foundation
    {
        class FL_EXPORT CTrieNode
        {
        public:
            CTrieNode();
 
            CTrieNode(const CTrieNode& rhs);
 
            CTrieNode(const CTrieNode* pRhs);
 
            virtual ~CTrieNode();
 
            virtual const CResult Assign(const CTrieNode& rhs);
 
            virtual const CResult Assign(const CTrieNode* pRhs);
 
            virtual const CResult Clear();
 
            virtual CTrieNode* DuplicateObject() const;
            virtual CTrieNode* CreateObject() const;
 
        public:
            std::unordered_map<wchar_t, CTrieNode*>& m_mapChildren;
 
            bool m_bEndOfWord;
 
            std::vector<std::wstring>& m_vctWstrWords;
        };
 
        class FL_EXPORT CTrie
        {
        public:
            CTrie();
 
            CTrie(const CTrie& rhs);
 
            CTrie(const CTrie* pRhs);
 
            virtual ~CTrie();
 
            virtual const CResult Assign(const CTrie& rhs);
 
            virtual const CResult Assign(const CTrie* pRhs);
 
            virtual const CResult SetNodeTemplate(const CTrieNode* pTrieNodeTemplate);
 
            virtual const CTrieNode* GetNodeTemplate() const;
 
            virtual CTrieNode* Insert(const std::wstring& wstrWord);
            
            virtual const CResult Remove(const std::wstring& wstrWord);
 
            virtual CTrieNode* Rename(const std::wstring& wstrOldWord, const std::wstring& wstrNewWord);
 
            virtual bool FindExact(const std::wstring& wstrWord) const;
 
            virtual bool FindSubstring(const std::wstring& wstrSubstring, std::vector<std::wstring>& vctResults) const;
 
            virtual const CResult Clear();
 
            virtual const CTrieNode* FindExactNode(const std::wstring& wstrWord) const;
 
        protected:
            virtual const CTrieNode* FindNode(const std::wstring& wstrKey) const;
 
            virtual bool RemoveHelper(CTrieNode* pNode, const std::wstring& wstrWord, int32_t i32Depth);
 
            virtual void CollectWords(const CTrieNode* pNode, std::vector<std::wstring>& vctResults) const;
 
        protected:
            virtual wchar_t NormalizeChar(wchar_t ch) const;
 
        protected:
            CTrieNode* m_pRoot;
 
            CTrieNode* m_pNodeTemplate;
        };
    }
}