/* dbx_tree: tree database driver for Miranda IM Copyright 2007-2010 Michael "Protogenes" Kunz, This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ #pragma once #include "BTree.h" #include "BlockManager.h" #include "Logger.h" template <typename TKey, uint16_t SizeParam = 4> class CFileBTree : public CBTree<TKey, SizeParam> { private: protected: CBlockManager & m_BlockManager; uint16_t cSignature; virtual void PrepareInsertOperation(); virtual TNode * CreateNewNode(TNodeRef & NodeRef); virtual void DeleteNode(TNodeRef Node); virtual TNode * Read(TNodeRef Node); virtual void Write(TNodeRef Node); public: CFileBTree(CBlockManager & BlockManager, typename CBTree<TKey, SizeParam>::TNodeRef RootNode, uint16_t Signature); virtual ~CFileBTree(); }; template <typename TKey, uint16_t SizeParam> CFileBTree<TKey, SizeParam>::CFileBTree(CBlockManager & BlockManager, typename CBTree<TKey, SizeParam>::TNodeRef RootNode, uint16_t Signature) : CBTree<TKey, SizeParam>::CBTree(RootNode), m_BlockManager(BlockManager) { cSignature = Signature; CBTree<TKey, SizeParam>::m_DestroyTree = false; } template <typename TKey, uint16_t SizeParam> CFileBTree<TKey, SizeParam>::~CFileBTree() { CBTree<TKey, SizeParam>::m_DestroyTree = false; } template <typename TKey, uint16_t SizeParam> void CFileBTree<TKey, SizeParam>::PrepareInsertOperation() { } template <typename TKey, uint16_t SizeParam> typename CBTree<TKey, SizeParam>::TNode * CFileBTree<TKey, SizeParam>::CreateNewNode(typename CBTree<TKey, SizeParam>::TNodeRef & NodeRef) { return reinterpret_cast<TNode*>( m_BlockManager.CreateBlock<uint32_t>(NodeRef, cSignature << 16, sizeof(typename CBTree<TKey, SizeParam>::TNode) - 4) - 1); } template <typename TKey, uint16_t SizeParam> void CFileBTree<TKey, SizeParam>::DeleteNode(typename CBTree<TKey, SizeParam>::TNodeRef Node) { m_BlockManager.DeleteBlock(Node); } template <typename TKey, uint16_t SizeParam> typename CBTree<TKey, SizeParam>::TNode * CFileBTree<TKey, SizeParam>::Read(typename CBTree<TKey, SizeParam>::TNodeRef Node) { uint32_t sig = 0; uint32_t size = 0; TNode * res = reinterpret_cast<TNode*>( m_BlockManager.ReadBlock<uint32_t>(Node, size, sig) - 1); /// HACK using knowledge about the blockmanager here CHECK(res->Signature == cSignature, logCRITICAL, _T("Signature check failed")); return res; } template <typename TKey, uint16_t SizeParam> void CFileBTree<TKey, SizeParam>::Write(typename CBTree<TKey, SizeParam>::TNodeRef Node) { m_BlockManager.UpdateBlock(Node, 0); }