X-Git-Url: https://git.llucax.com/z.facultad/75.52/treemulator.git/blobdiff_plain/1b6152a8b901df0e8b57c35b9846a6e609b8b7df..44c4c40e48cc4a72e81e0af80842738c45d6116f:/src/btree.h

diff --git a/src/btree.h b/src/btree.h
index 744079b..999d798 100644
--- a/src/btree.h
+++ b/src/btree.h
@@ -44,18 +44,23 @@
  *    tomar una idea simimar)
  */
 
+#include <iostream>
 #include <string>
+#include <list>
 #include "common.h"
+#include "clave.h"
+#include "clave_fija.h"
+#include "btree_data.h"
 
 /* alias para codear menos :) */
 
 /** Encabezado del archivo BTree */
-typedef struct _btree_file_ {
+struct BTreeFileHeader {
 	uint block_size;
-} BTreeFileHeader;
+};
 
 /** Encabezado de un bloque */
-typedef struct _btree_header_ {
+struct BTreeNodeHeader {
 	/** Indica a que nivel corresponde un bloque
 	 *
 	 *  nivel == 0 : una hoja
@@ -71,7 +76,7 @@ typedef struct _btree_header_ {
 
 	/** Cantidad de elementos en el nodo */
 	unsigned int item_count;
-} BTreeNodeHeader;
+};
 
 /** Crea un nuevo arbol B
  *
@@ -81,18 +86,27 @@ typedef struct _btree_header_ {
  */
 class BTree {
 	public:
-		BTree (const char *filename, unsigned int block_size, bool create_new_file = false);
+		BTree (const std::string &filename, unsigned int block_size, bool create_new_file = false);
 		~BTree ();
 
-		void AddKey ();
-		void DelKey ();
+		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);
+
+		void WriteFileHeader ();
+
+		void WriteBlock (uchar *block, uint num);
+		uchar *ReadBlock (uint num);
+		uchar *NewBlock (uint &num);
 
-	private:
-		void write_tree_header ();
-		void write_block (uchar *block, 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;
 
@@ -101,6 +115,10 @@ class BTree {
 		 *  \TODO Ver si vale la pena
 		 */
 		FILE *fp;
+
+
+		/* DEBUG */
+		void PrintNode (uint num);
 };
 
 #endif // _B_TREE_H