X-Git-Url: https://git.llucax.com/z.facultad/75.52/treemulator.git/blobdiff_plain/e994ad95d6fcb7372527146482ce55cea669b6c9..6a4238b2894a1cf20a5b8ccb0575f018ae25a5b3:/src/btree.cpp

diff --git a/src/btree.cpp b/src/btree.cpp
index 66f4026..adb0ae7 100644
--- a/src/btree.cpp
+++ b/src/btree.cpp
@@ -71,6 +71,7 @@ void BTree::AddKey (const Clave &k)
 		WriteKeys (node, node_header, node_keys);
 		WriteNodoHeader (node, &node_header);
 		WriteBlock (node, left);
+		DeleteKeys (node_keys);
 		delete [] node;
 
 		/* Leo y actualizo la Raiz */
@@ -87,7 +88,11 @@ void BTree::AddKey (const Clave &k)
 		WriteKeys (node, node_header, node_keys);
 		WriteNodoHeader (node, &node_header);
 		WriteBlock (node, 0);
+		delete [] node;
+		DeleteKeys (node_keys);
 		PrintNode (0);
+
+		delete kout;
 	}
 }
 
@@ -132,6 +137,8 @@ Clave* BTree::AddKeyLeafR (const Clave *k, uint node_num, uint &left_child, uint
 		WriteKeys (node, node_header, node_keys);
 		WriteNodoHeader (node, &node_header);
 		WriteBlock (node, node_num);
+		DeleteKeys (node_keys);
+		delete [] node;
 
 		PrintNode (node_num);
 	} else {
@@ -208,6 +215,8 @@ Clave* BTree::AddKeyLeafR (const Clave *k, uint node_num, uint &left_child, uint
 		WriteKeys (new_node, new_node_header, new_node_keys);
 		WriteNodoHeader (new_node, &new_node_header);
 		WriteBlock (new_node, new_node_num);
+		DeleteKeys (old_node_keys);
+		DeleteKeys (new_node_keys);
 
 		PrintNode (node_num);
 		PrintNode (new_node_num);
@@ -256,9 +265,14 @@ Clave* BTree::AddKeyOtherR (const Clave *k, uint node_num, uint &left_child, uin
 	} else {
 		k = AddKeyR (k, (*ultima)->getChild (), left_child, right_child);
 	}
+	DeleteKeys (node_keys);
 
 	/* Nada que hacer */
-	if (!k) return NULL;
+	if (data) delete data;
+	if (!k) {
+		delete [] node;
+		return NULL;
+	}
 
 	data = new BTreeData (k->Clone (), right_child);
 
@@ -278,6 +292,8 @@ Clave* BTree::AddKeyOtherR (const Clave *k, uint node_num, uint &left_child, uin
 		WriteKeys (node, node_header, node_keys);
 		WriteNodoHeader (node, &node_header);
 		WriteBlock (node, node_num);
+		DeleteKeys (node_keys);
+		delete [] node;
 
 		PrintNode (node_num);
 	} else {
@@ -358,6 +374,8 @@ Clave* BTree::AddKeyOtherR (const Clave *k, uint node_num, uint &left_child, uin
 		WriteKeys (new_node, new_node_header, new_node_keys);
 		WriteNodoHeader (new_node, &new_node_header);
 		WriteBlock (new_node, new_node_num);
+		DeleteKeys (old_node_keys);
+		DeleteKeys (new_node_keys);
 
 		PrintNode (node_num);
 		PrintNode (new_node_num);
@@ -440,7 +458,9 @@ void BTree::WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<BTre
 
 	while (it != keys.end ()) {
 		BTreeData *d = (*it);
-		memcpy (node, d->ToArray(), d->Size ());
+		uchar *n = d->ToArray ();
+		memcpy (node, n, d->Size ());
+		delete [] n;
 		node += d->Size ();
 		node_header.free_space -= d->Size ();
 		node_header.item_count++;
@@ -471,6 +491,7 @@ void BTree::PrintNode (uint num)
 	std::cout << std::endl;
 
 	delete [] node;
+	DeleteKeys (node_keys);
 }
 
 uchar *BTree::NewBlock (uint &num)
@@ -495,3 +516,67 @@ uchar *BTree::NewBlock (uint &num)
 	return node;
 }
 
+bool BTree::FindKey (const Clave &k)
+{
+	return FindKeyR (&k, 0);
+}
+
+bool BTree::FindKeyR (const Clave *k, uint node_num)
+{
+	std::list<BTreeData *> node_keys;
+	BTreeNodeHeader node_header;
+
+	/* Leo el nodo raiz para empezar a agregar */
+	uchar *node = ReadBlock (node_num);
+	ReadNodoHeader (node, &node_header);
+	node_keys = ReadKeys (node, node_header);
+
+	std::list<BTreeData *>::iterator it = node_keys.begin ();
+	std::list<BTreeData *>::iterator posterior;
+	std::list<BTreeData *>::iterator ultima;
+	
+	/* Se supone que la primera es un hijo :) */
+	BTreeData *lchild;
+	if (node_header.level != 0) {
+		lchild = (*it++);
+	}
+	posterior = it;
+
+	BTreeData *data;
+	if (node_header.level == 0)
+		data = new BTreeLeafData ((Clave *)k);
+	else
+		data = new BTreeData ((Clave *)k, 0);
+
+	while (it != node_keys.end ()) {
+		if ((*data) == (*(*it))) {
+			/* La encontre!, retorno */
+			delete [] node;
+			DeleteKeys (node_keys);
+			return true;
+		}
+
+		if ((*data) < (*(*it)))
+			break;
+		ultima = it;
+		it++;
+	}
+
+	/* TODO: Aca faltaria liberar memoria */
+	if (it == posterior)
+		return FindKeyR (k, lchild->getChild ());
+		
+	return FindKeyR (k, (*ultima)->getChild ());
+}
+
+void BTree::DeleteKeys (std::list<BTreeData *> &keys)
+{
+	std::list<BTreeData *>::iterator it = keys.begin ();
+
+	while (it != keys.end ()) {
+		BTreeData *d = (*it);
+		delete d;
+		it++;
+	}
+}
+