Corrige y agrega pruebas con clave variable.

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

diff --git a/src/btree.cpp b/src/btree.cpp
index 3027c9a30d972b9efb7c0662bc9a3e7a5d53d733..87471779f5f5ea15c12b7243e6b2842c40a51a54 100644 (file)
--- a/src/btree.cpp
+++ b/src/btree.cpp
@@ -516,17 +516,20 @@ void BTree::DelKeyFromLeaf (Clave *k, uint node_num, uint padre)
                } else {
                        std::cout << "NADIE ME PUEDE PRESTAR, FUNDIR NODOS\n";
                        uint join1, join2;
+                       int tipoh;
                        if (hi != 0) {
                                std::cout << "Join con Hermano Izquierdo\n";
                                join1 = hi;
                                join2 = node_num;
+                               tipoh = 0;
                        } else {
                                std::cout << "Join con Hermano Derecho\n";
                                join1 = node_num;
                                join2 = hd;
+                               tipoh = 1;
                        }
 
-                       JoinNodes (join1, join2, padre);
+                       JoinNodes (join1, join2, padre, tipoh);
                        return;
                }
 
@@ -540,12 +543,29 @@ void BTree::DelKeyFromLeaf (Clave *k, uint node_num, uint padre)
        std::cout << "Borrado de una hoja listo\n";
 }
 
-void BTree::JoinNodes (uint node1, uint node2, uint padre)
+void BTree::JoinNodes (uint node1, uint node2, uint padre, int tipohermano)
 {
        uchar *n1, *n2, *npadre;
        BTreeNodeHeader nh1, nh2, nhp;
        std::list<BTreeData *> nk1, nk2, nkpadre;
 
+       if (node1 == node2) {
+               std::cout << "PANIC : No puedo juntar el mismo nodo con si mismo!!\n";
+               exit (1);
+       }
+       if (node1 == padre) {
+               std::cout << "PANIC : No puedo juntar el mismo nodo con si mismo!!\n";
+               exit (1);
+       }
+       if (node2 == padre) {
+               std::cout << "PANIC : No puedo juntar el mismo nodo con si mismo!!\n";
+               exit (1);
+       }
+
+       PrintNode (padre);
+       PrintNode (node1);
+       PrintNode (node2);
+
        /* Leo los nodos */
        n1 = ReadBlock (node1);
        n2 = ReadBlock (node2);
@@ -567,26 +587,31 @@ void BTree::JoinNodes (uint node1, uint node2, uint padre)
        nk2 = ReadKeys (n2, nh2);
        nkpadre = ReadKeys (npadre, nhp);
 
-       /* Busco la clave a juntar con los nodos */
+       /* Busco la clave del padre a juntar con los nodos */
        std::list<BTreeData *>::iterator it = nkpadre.begin ();
        std::list<BTreeData *>::iterator borrar_padre;
        std::list<BTreeData *>::iterator sig;
+       std::list<BTreeData *>::iterator anterior = it;
        
        Clave *cpadre;
        BTreeData *lchild = (*it++);
 
        if (lchild->getChild () == node1) {
                cpadre = (*it)->getClave ();
-               //nkpadre.erase (it);
                borrar_padre = it;
        } else {
                while (it != nkpadre.end ()) {
-                       if ((*it)->getChild () == node1)
-                               break;
+                       if (tipohermano == 0) {
+                               if ((*it)->getChild () == node2)
+                                       break;
+                       } else {
+                               if ((*it)->getChild () == node1)
+                                       break;
+                       }
+                       anterior = it;
                        it++;
                }
                cpadre = (*it)->getClave ();
-               //nkpadre.erase (it);
                borrar_padre = it;
        }
        if (it == nkpadre.end ()) {
@@ -604,7 +629,7 @@ void BTree::JoinNodes (uint node1, uint node2, uint padre)
                newkeys.push_back ( new BTreeLeafData ((*i)->getClave ()->Clone ()));
                i++;
        }
-       if (cpadre)
+       //if (tipohermano == 0)
                newkeys.push_back ( new BTreeLeafData (cpadre->Clone ()));
        i = nk2.begin ();
        while (i != nk2.end ()) {
@@ -618,27 +643,39 @@ void BTree::JoinNodes (uint node1, uint node2, uint padre)
                exit (1);
        }
 
-/*     if (padre == 0) {
+       /* Para el padre, tener 2 items significa tener solo 1 clave, ya que
+        * el otro item es el LeftChild!
+        */
+       if ((padre == 0) && (nhp.item_count == 2)) {
+               /* Si junte 2 nodos, cuyo padre era la raiz, y esta tenia
+                * solo una clave, quiere decir que lo que me queda
+                * es de nuevo solo una raiz con todas las claves
+                */
                nhp.level = 0;
                WriteKeys (npadre, nhp, newkeys);
                WriteNodoHeader (npadre, &nhp);
-               WriteBlock (npadre, padre);*/
+               WriteBlock (npadre, padre);
 
         /* TODO: Recuperar nodo1 y nodo2 */
-//     } else {
+       } else {
                WriteKeys (n1, nh1, newkeys);
                WriteNodoHeader (n1, &nh1);
                WriteBlock (n1, node1);
 
                /* TODO : Recuperar node2 */
                /* Actualizo punero al padre */
-               (*sig)->setChild (node1);
+               (*anterior)->setChild (node1);
        
                nkpadre.erase (borrar_padre);
                WriteKeys (npadre, nhp, nkpadre);
                WriteNodoHeader (npadre, &nhp);
                WriteBlock (npadre, padre);
-       //}
+       }
+
+       std::cout << " ----- Luego de Fundir -----\n";
+       PrintNode (node1);
+       PrintNode (padre);
+       std::cout << " ---------------------------\n";
 
        DeleteKeys (nk1);
        DeleteKeys (nk2);
@@ -804,7 +841,7 @@ void BTree::DelKeyFromNode (Clave *k, uint node_num, uint padre, uint left, uint
        BTreeNodeHeader node_header;
        std::list<BTreeData *> node_keys;
 
-       node = ReadBlock (padre);
+       node = ReadBlock (node_num);
        ReadNodoHeader (node, &node_header);
        node_keys = ReadKeys (node, node_header);
 
@@ -844,6 +881,10 @@ void BTree::DelKeyFromNode (Clave *k, uint node_num, uint padre, uint left, uint
 
                std::list<BTreeData *>::iterator it = node_keys.begin ();
                while (it != node_keys.end ()) {
+                       std::string ss1, ss2;
+                       ss1 = *data;
+                       ss2 = *(*it);
+                       std::cout << ss1 << " == " << ss2 << std::endl;
                        if ((*data) == (*(*it))) {
                                break;
                        }
@@ -864,7 +905,7 @@ void BTree::DelKeyFromNode (Clave *k, uint node_num, uint padre, uint left, uint
                /* Grabo los nodos */
                WriteKeys (node, node_header, node_keys);
                WriteNodoHeader (node, &node_header);
-               WriteBlock (node, padre);
+               WriteBlock (node, node_num);
                DeleteKeys (node_keys);
                delete [] node;
 
@@ -875,7 +916,8 @@ void BTree::DelKeyFromNode (Clave *k, uint node_num, uint padre, uint left, uint
                delete [] node_r;
 
                std::cout << "Grabe todo en disco ...\n";
-
+               PrintNode (node_num);
+               PrintNode (right);
                /* Ahora debo eliminar la clave que puse en el nodo hoja */
                std::cout << "Borro la clave desde la hoja!\n";
 
@@ -948,11 +990,44 @@ std::list<BTreeData *> BTree::ReadKeys (uchar *node, BTreeNodeHeader &node_heade
                keys.push_back (data);
        }
 
+       DeAbrevKey (keys);
        return keys;
 }
 
+void BTree::AbrevKey (std::list<BTreeData *> &lst)
+{
+       /* Claves Fijas No se abrevian */
+       if (key_type == KEY_FIXED) return;
+
+       BTreeData *primera = NULL;
+       std::list<BTreeData *>::iterator it = lst.begin ();
+
+       while (it != lst.end ()) {
+               if ((*it)->Abrev (primera) == false)
+                       primera = (*it);
+               it++;
+       }
+}
+
+void BTree::DeAbrevKey (std::list<BTreeData *> &lst)
+{
+       /* Claves Fijas No se abrevian */
+       if (key_type == KEY_FIXED) return;
+
+       BTreeData *primera = NULL;
+       std::list<BTreeData *>::iterator it = lst.begin ();
+
+       while (it != lst.end ()) {
+               if ((*it)->DesAbrev (primera) == false)
+                       primera = (*it);
+               it++;
+       }
+}
+
 void BTree::WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<BTreeData *> &keys)
 {
+       AbrevKey (keys);
+
        std::list<BTreeData *>::iterator it = keys.begin ();
 
        node += sizeof (BTreeNodeHeader);
@@ -974,7 +1049,7 @@ void BTree::WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<BTre
                it++;
        }
 
-       /* TODO : incrementar node_header.item_count aca o fuera de este metodo? */
+       DeAbrevKey (keys);
 }
                
 void BTree::PrintNode (uint num)