Agrego clave variable.

[z.facultad/75.52/treemulator.git] / src / btree.h

diff --git a/src/btree.h b/src/btree.h
index 081c43687c884a4b69fd00fdc3b5878409cc335b..a67fa4c4b9d1e086458f80e2cd367869411f448d 100644 (file)
--- a/src/btree.h
+++ b/src/btree.h
@@ -46,8 +46,12 @@
 
 #include <iostream>
 #include <string>
+#include <list>
 #include "common.h"
 #include "clave.h"
+#include "clave_fija.h"
+#include "clave_variable.h"
+#include "btree_data.h"
 
 /* alias para codear menos :) */
 
@@ -83,29 +87,48 @@ struct BTreeNodeHeader {
  */
 class BTree {
        public:
-               BTree (const std::string &filename, unsigned int block_size, bool create_new_file = false);
+               BTree (const std::string &filename, unsigned int block_size, int k_t = KEY_FIXED, bool create_new_file = false);
                ~BTree ();
 
-               void AddKey (Clave &k);
-               void DelKey (Clave &k);
+               /** Tipos de clave a usar */
+               enum {
+                       KEY_FIXED,
+                       KEY_VARIABLE
+               };
 
-       private:
-               void write_tree_header ();
-               void write_block (uchar *block, uint num);
+               void AddKey (const Clave &k);
+               void DelKey (const Clave &k);
 
+       protected:
+               Clave* AddKeyR (const Clave *k, uint node_num, uint &left_child, uint &right_child);
+               Clave* AddKeyOtherR (const Clave *k, uint node_num, uint &left_child, uint &right_child);
+               Clave* AddKeyLeafR (const Clave *k, uint node_num, uint &left_child, uint &right_child);
+
+               void WriteFileHeader ();
+
+               void WriteBlock (uchar *block, uint num);
                uchar *ReadBlock (uint num);
+               uchar *NewBlock (uint &num);
 
                void ReadNodoHeader (uchar *node, BTreeNodeHeader *header);
                void WriteNodoHeader (uchar *node, BTreeNodeHeader *header);
 
+               std::list<BTreeData *> ReadKeys (uchar *node, BTreeNodeHeader &node_header);
+               void WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<BTreeData *> &keys);
+
                std::string filename;
                BTreeFileHeader header;
+               int key_type;
 
                /** Apunta al archivo de datos, asi se abre solo 1 vez
                 *
                 *  \TODO Ver si vale la pena
                 */
                FILE *fp;
+
+
+               /* DEBUG */
+               void PrintNode (uint num);
 };
 
 #endif // _B_TREE_H