src/btree.cpp

   1
   2 #include "btree.h"
   3
   4 BTree::BTree (const std::string &name, unsigned int block_size, int kt, bool create_new_file)
   5 {
   6         key_type = kt;
   7         uchar *node;
   8         BTreeNodeHeader nh;
   9
  10         fp = fopen (name.c_str(), "wb+");
  11         if (!fp) {
  12                 /* TODO : mandar una exception ? */
  13                 return;
  14         }
  15
  16         /* Nombre de archivo */
  17         filename = name;
  18
  19         /* Inicializo el header */
  20         header.block_size = block_size;
  21         WriteFileHeader ();
  22
  23         /* Creo el primer bloque vacio */
  24         node = new uchar[block_size];
  25         ReadNodoHeader (node, &nh);
  26         nh.level = 0;
  27         nh.free_space = block_size - sizeof (BTreeNodeHeader);
  28         nh.item_count = 0;
  29         WriteNodoHeader (node, &nh);
  30         WriteBlock (node, 0);
  31
  32         delete [] node;
  33 }
  34
  35 BTree::~BTree ()
  36 {
  37         fclose (fp);
  38 }
  39
  40 void BTree::WriteFileHeader ()
  41 {
  42         fseek (fp, 0L, SEEK_SET);
  43         fwrite (&header, 1, sizeof (BTreeFileHeader), fp);
  44 }
  45
  46 void BTree::WriteBlock (uchar *block, uint num)
  47 {
  48         num++;
  49         fseek (fp, num*header.block_size, SEEK_SET);
  50         fwrite (block, 1, header.block_size, fp);
  51 }
  52
  53 void BTree::AddKey (const Clave &k)
  54 {
  55         uint left, right;
  56         Clave *kout = AddKeyR (k.Clone (), 0, left, right);
  57
  58         if (kout) {
  59                 unsigned short level;
  60                 /* Debo dejar la raiz en el nodo 0, por lo que paso el nodo
  61                  * que esta usando el hijo izquierdo a un nuevo nodo */
  62                 std::list<BTreeData *> node_keys;
  63                 BTreeNodeHeader node_header;
  64                 uchar *node = ReadBlock (left);
  65                 ReadNodoHeader (node, &node_header);
  66                 node_keys = ReadKeys (node, node_header);
  67                 level = node_header.level + 1;
  68
  69                 uchar *new_node = NewBlock (left);
  70                 delete [] new_node; /* No me interesa, voy a usar lo leio antes */
  71
  72                 WriteKeys (node, node_header, node_keys);
  73                 WriteNodoHeader (node, &node_header);
  74                 WriteBlock (node, left);
  75                 DeleteKeys (node_keys);
  76                 delete [] node;
  77
  78                 /* Leo y actualizo la Raiz */
  79                 node = ReadBlock (0);
  80                 ReadNodoHeader (node, &node_header);
  81                 node_keys = std::list<BTreeData *>();
  82
  83                 node_keys.push_back (new BTreeChildData (left));
  84                 node_keys.push_back (new BTreeData (kout, right));
  85
  86                 node_header.level = level;
  87                 node_header.item_count = 1;
  88
  89                 WriteKeys (node, node_header, node_keys);
  90                 WriteNodoHeader (node, &node_header);
  91                 WriteBlock (node, 0);
  92                 delete [] node;
  93                 DeleteKeys (node_keys);
  94                 PrintNode (0);
  95         }
  96 }
  97
  98 Clave* BTree::AddKeyR (const Clave *k, uint node_num, uint &left_child, uint &right_child)
  99 {
 100         uchar *node = ReadBlock (node_num);
 101         BTreeNodeHeader node_header;
 102         ReadNodoHeader (node, &node_header);
 103         delete [] node;
 104
 105         if (node_header.level == 0)
 106                 return AddKeyLeafR (k, node_num, left_child, right_child);
 107
 108         return AddKeyOtherR (k, node_num, left_child, right_child);
 109 }
 110
 111 Clave* BTree::AddKeyLeafR (const Clave *k, uint node_num, uint &left_child, uint &right_child)
 112 {
 113         Clave *kout = NULL;
 114         std::list<BTreeData *> node_keys;
 115
 116         BTreeData *data = new BTreeLeafData (k->Clone ());
 117
 118         /* Leo el nodo raiz para empezar a agregar */
 119         uchar *node = ReadBlock (node_num);
 120         BTreeNodeHeader node_header;
 121         ReadNodoHeader (node, &node_header);
 122
 123         if (node_header.free_space > data->Size ()) {
 124                 BTreeData *datait;
 125                 node_keys = ReadKeys (node, node_header);
 126                 std::list<BTreeData *>::iterator it = node_keys.begin ();
 127
 128                 while (it != node_keys.end ()) {
 129                         datait = (*it);
 130                         if ((*data) < (*datait))
 131                                 /* Me pase, lo agrego aca! */
 132                                 break;
 133                         it++;
 134                 }
 135                 node_keys.insert (it, data);
 136                 WriteKeys (node, node_header, node_keys);
 137                 WriteNodoHeader (node, &node_header);
 138                 WriteBlock (node, node_num);
 139                 DeleteKeys (node_keys);
 140                 delete [] node;
 141
 142                 PrintNode (node_num);
 143         } else {
 144                 /* Split : Creo e inicializo el nuevo nodo */
 145                 std::list<BTreeData *> new_node_keys;
 146                 std::list<BTreeData *> old_node_keys;
 147                 BTreeNodeHeader new_node_header;
 148                 uint new_node_num;
 149                 uchar *new_node = NewBlock (new_node_num);
 150                 ReadNodoHeader (new_node, &new_node_header);
 151                 new_node_header.level = node_header.level;
 152
 153                 node_keys = ReadKeys (node, node_header);
 154                 new_node_keys = ReadKeys (new_node, new_node_header);
 155
 156                 /* Agrego la clave en la lista que ya tengo de manera ordenada */
 157                 std::list<BTreeData *>::iterator it = node_keys.begin ();
 158                 std::list<BTreeData *>::iterator previt = node_keys.begin ();
 159
 160                 while (it != node_keys.end ()) {
 161                         BTreeData *datait;
 162                         datait = (*it);
 163                         if ((*data) < (*datait))
 164                                 /* Me pase, lo agrego aca! */
 165                                 break;
 166                         previt = it;
 167                         it++;
 168                 }
 169                 if (it != node_keys.end ())
 170                         node_keys.insert (it, data);
 171                 else
 172                         node_keys.push_back (data);
 173
 174                 /* Tengo que guardar claves hasta ocupar nodo size/2 en cada nodo
 175                  * y subir la clave del medio */
 176                 node_header.item_count = 0;
 177                 node_header.free_space = header.block_size - sizeof (BTreeNodeHeader);
 178
 179                 uint total_size = 0;
 180                 it = node_keys.begin ();
 181                 while (it != node_keys.end ()) {
 182                         BTreeData *datait;
 183                         datait = (*it);
 184                         total_size += datait->Size ();
 185                         it++;
 186                         /* Hack : Si me quedo con todas las claves, en el caso de ser
 187                          * del mismo tama#o se desbalancea. Hay que ver que efecto
 188                          * puede tener en el caso de claves de long. variable
 189                          */
 190                         if (it == node_keys.end ())
 191                                 total_size -= datait->Size ();
 192                 }
 193
 194                 it = node_keys.begin ();
 195                 uint used = 0;
 196                 while (used < total_size/2) {
 197                         BTreeData *d = (*it);
 198                         old_node_keys.push_back (d);
 199                         used += d->Size ();
 200                         it++;
 201                 }
 202                 kout = (*it++)->getClave (); // Esta se retorna al "padre" para que se la agregue
 203
 204                 while (it != node_keys.end ()) {
 205                         BTreeData *d = (*it);
 206                         new_node_keys.push_back (d);
 207                         it++;
 208                 }
 209
 210                 /* Guardo */
 211                 WriteKeys (node, node_header, old_node_keys);
 212                 WriteNodoHeader (node, &node_header);
 213                 WriteBlock (node, node_num);
 214                 WriteKeys (new_node, new_node_header, new_node_keys);
 215                 WriteNodoHeader (new_node, &new_node_header);
 216                 WriteBlock (new_node, new_node_num);
 217                 DeleteKeys (old_node_keys);
 218                 DeleteKeys (new_node_keys);
 219
 220                 PrintNode (node_num);
 221                 PrintNode (new_node_num);
 222
 223                 /* Paso los hijos */
 224                 left_child = node_num;
 225                 right_child = new_node_num;
 226                 delete [] new_node;
 227                 delete [] node;
 228         }
 229
 230         return kout;
 231 }
 232
 233 Clave* BTree::AddKeyOtherR (const Clave *k, uint node_num, uint &left_child, uint &right_child)
 234 {
 235         Clave *kout = NULL;
 236         std::list<BTreeData *> node_keys;
 237
 238         BTreeData *data = new BTreeLeafData (k->Clone ());
 239
 240         /* Leo el nodo raiz para empezar a agregar */
 241         uchar *node = ReadBlock (node_num);
 242         BTreeNodeHeader node_header;
 243         ReadNodoHeader (node, &node_header);
 244
 245         node_keys = ReadKeys (node, node_header);
 246
 247         std::list<BTreeData *>::iterator it = node_keys.begin ();
 248         std::list<BTreeData *>::iterator posterior;
 249         std::list<BTreeData *>::iterator ultima;
 250
 251         /* Se supone que la primera es un hijo :) */
 252         BTreeData *lchild = (*it++);
 253         posterior = it;
 254
 255         while (it != node_keys.end ()) {
 256                 if ((*data) < (*(*it)))
 257                         break;
 258                 ultima = it;
 259                 it++;
 260         }
 261
 262         if (it == posterior) {
 263                 k = AddKeyR (k, lchild->getChild (), left_child, right_child);
 264         } else {
 265                 k = AddKeyR (k, (*ultima)->getChild (), left_child, right_child);
 266         }
 267         DeleteKeys (node_keys);
 268
 269         /* Nada que hacer */
 270         if (data) delete data;
 271         if (!k) {
 272                 delete [] node;
 273                 return NULL;
 274         }
 275
 276         data = new BTreeData (k->Clone (), right_child);
 277
 278         if (node_header.free_space > data->Size ()) {
 279                 BTreeData *datait;
 280                 node_keys = ReadKeys (node, node_header);
 281                 std::list<BTreeData *>::iterator it = node_keys.begin ();
 282
 283                 while (it != node_keys.end ()) {
 284                         datait = (*it);
 285                         if ((*data) < (*datait))
 286                                 /* Me pase, lo agrego aca! */
 287                                 break;
 288                         it++;
 289                 }
 290                 node_keys.insert (it, data);
 291                 WriteKeys (node, node_header, node_keys);
 292                 WriteNodoHeader (node, &node_header);
 293                 WriteBlock (node, node_num);
 294                 DeleteKeys (node_keys);
 295                 delete [] node;
 296
 297                 PrintNode (node_num);
 298         } else {
 299                 /* Split : Creo e inicializo el nuevo nodo */
 300                 std::list<BTreeData *> new_node_keys;
 301                 std::list<BTreeData *> old_node_keys;
 302                 BTreeNodeHeader new_node_header;
 303                 uint new_node_num;
 304                 uchar *new_node = NewBlock (new_node_num);
 305                 ReadNodoHeader (new_node, &new_node_header);
 306                 new_node_header.level = node_header.level;
 307
 308                 node_keys = ReadKeys (node, node_header);
 309                 new_node_keys = ReadKeys (new_node, new_node_header);
 310
 311                 /* Agrego la clave en la lista que ya tengo de manera ordenada */
 312                 std::list<BTreeData *>::iterator it = node_keys.begin ();
 313                 std::list<BTreeData *>::iterator previt = node_keys.begin ();
 314
 315                 previt = ++it;
 316
 317                 while (it != node_keys.end ()) {
 318                         BTreeData *datait;
 319                         datait = (*it);
 320                         if ((*data) < (*datait))
 321                                 /* Me pase, lo agrego aca! */
 322                                 break;
 323                         previt = it;
 324                         it++;
 325                 }
 326                 if (it != node_keys.end ())
 327                         node_keys.insert (it, data);
 328                 else
 329                         node_keys.push_back (data);
 330
 331                 /* Tengo que guardar claves hasta ocupar nodo size/2 en cada nodo
 332                  * y subir la clave del medio */
 333                 node_header.item_count = 0;
 334                 node_header.free_space = header.block_size - sizeof (BTreeNodeHeader);
 335
 336                 uint total_size = 0;
 337                 it = node_keys.begin ();
 338                 while (it != node_keys.end ()) {
 339                         BTreeData *datait;
 340                         datait = (*it);
 341                         total_size += datait->Size ();
 342                         it++;
 343                         /* Hack : Si me quedo con todas las claves, en el caso de ser
 344                          * del mismo tama#o se desbalancea. Hay que ver que efecto
 345                          * puede tener en el caso de claves de long. variable
 346                          */
 347                         if (it == node_keys.end ())
 348                                 total_size -= datait->Size ();
 349                 }
 350
 351                 it = node_keys.begin ();
 352                 uint used = 0;
 353                 while (used < total_size/2) {
 354                         BTreeData *d = (*it);
 355                         old_node_keys.push_back (d);
 356                         used += d->Size ();
 357                         it++;
 358                 }
 359                 kout = (*it)->getClave (); // Esta se retorna al "padre" para que se la agregue
 360
 361                 new_node_keys.push_back ( new BTreeChildData ((*it)->getChild ()));
 362                 it++;
 363                 while (it != node_keys.end ()) {
 364                         BTreeData *d = (*it);
 365                         new_node_keys.push_back (d);
 366                         it++;
 367                 }
 368
 369                 /* Guardo */
 370                 WriteKeys (node, node_header, old_node_keys);
 371                 WriteNodoHeader (node, &node_header);
 372                 WriteBlock (node, node_num);
 373                 WriteKeys (new_node, new_node_header, new_node_keys);
 374                 WriteNodoHeader (new_node, &new_node_header);
 375                 WriteBlock (new_node, new_node_num);
 376                 DeleteKeys (old_node_keys);
 377                 DeleteKeys (new_node_keys);
 378
 379                 PrintNode (node_num);
 380                 PrintNode (new_node_num);
 381
 382                 /* Paso los hijos */
 383                 left_child = node_num;
 384                 right_child = new_node_num;
 385                 delete [] new_node;
 386                 delete [] node;
 387         }
 388
 389         return kout;
 390 }
 391
 392 void BTree::DelKey (const Clave &k)
 393 {
 394         DelKeyR (new BTreeLeafData (k.Clone ()), 0, 0);
 395 }
 396
 397 void BTree::DelKeyR (BTreeData *k, uint node_num, uint padre)
 398 {
 399         std::list<BTreeData *> node_keys;
 400         BTreeNodeHeader node_header;
 401         uchar *node;
 402
 403         node = ReadBlock (node_num);
 404         ReadNodoHeader (node, &node_header);
 405         node_keys = ReadKeys (node, node_header);
 406
 407         std::list<BTreeData *>::iterator it = node_keys.begin ();
 408         std::list<BTreeData *>::iterator ultima;
 409         std::list<BTreeData *>::iterator posterior;
 410
 411         BTreeData *lchild;
 412         if (node_header.level != 0) {
 413                 lchild = (*it++);
 414         }
 415         posterior = it;
 416
 417         while (it != node_keys.end ()) {
 418                 if ((*k) == (*(*it))) {
 419                         /* La encontre!, retorno */
 420                         if (node_header.level == 0) {
 421                                 DelKeyFromLeaf (k->getClave (), node_num, padre);
 422                         } else {
 423                                 uint left, right;
 424                                 if (it == posterior) {
 425                                         left = lchild->getChild ();
 426                                         right = (*it)->getChild ();
 427                                 } else {
 428                                         left = (*ultima)->getChild ();
 429                                         right = (*it)->getChild ();
 430                                 }
 431                                 std::cout << "Eliminar de Nodo con hijos : " << left << " y " << right << std::endl;
 432                                 DelKeyFromNode (k->getClave (), node_num, padre, left, right);
 433                         }
 434                         return;
 435                 }
 436
 437                 if ((*k) < (*(*it)))
 438                         break;
 439                 ultima = it;
 440                 it++;
 441         }
 442
 443         /* Si llego aca y estoy en nivel 0 (una hoja) quiere
 444          * decir que no lo encontre
 445          */
 446         if (node_header.level == 0) {
 447                 std::cout << "Clave no encontrada\n";
 448                 return;
 449         }
 450
 451         /* TODO: Aca faltaria liberar memoria */
 452         if (it == posterior) {
 453                 DelKeyR (k, lchild->getChild (), node_num);
 454         } else {
 455                 DelKeyR (k, (*ultima)->getChild (), node_num);
 456         }
 457 }
 458
 459 void BTree::DelKeyFromLeaf (Clave *k, uint node_num, uint padre)
 460 {
 461         BTreeData *data;
 462         uchar *node;
 463         BTreeNodeHeader node_header;
 464         std::list<BTreeData *> node_keys;
 465
 466         node = ReadBlock (node_num);
 467         ReadNodoHeader (node, &node_header);
 468         node_keys = ReadKeys (node, node_header);
 469
 470         data = new BTreeLeafData (k);
 471
 472         std::list<BTreeData *>::iterator it;
 473         it = node_keys.begin ();
 474         while (it != node_keys.end ()) {
 475                 if ((*data) == (*(*it))) {
 476                         node_keys.erase (it);
 477                         break;
 478                 }
 479                 it++;
 480         }
 481
 482         delete data;
 483
 484         /* TODO : que pasa cuando el nodo se queda vacio o con
 485          * muy pocas claveS??
 486          */
 487         WriteKeys (node, node_header, node_keys);
 488         WriteNodoHeader (node, &node_header);
 489         WriteBlock (node, node_num);
 490
 491         /* Veo si se cumple la condición de minimalidad */
 492         if ((node_header.free_space <= ((header.block_size-sizeof(BTreeNodeHeader))/2)) && (node_num != 0)) {
 493                 /* Oops! Debo pedir prestada clave */
 494                 uint hi, hd;
 495                 Clave *pedida;
 496
 497                 FindBrothers (node_num, padre, hi, hd);
 498
 499                 if ((pedida = GetKey (hi, 1)) != NULL) {
 500                         std::string s = *pedida;
 501                         std::cout << "Clave Pedida : " << s << std::endl;
 502
 503                         pedida = ReplaceKeyInFather (node_num, padre, pedida);
 504
 505                         node_keys.insert (node_keys.begin (), new BTreeLeafData (pedida));
 506                 } else if ((pedida = GetKey (hd, 0)) != NULL) {
 507                         std::string s = *pedida;
 508                         std::cout << "Clave Pedida : " << s << std::endl;
 509
 510                         pedida = ReplaceKeyInFather (node_num, padre, pedida);
 511
 512                         node_keys.push_back (new BTreeLeafData (pedida));
 513                 } else {
 514                         std::cout << "NADIE ME PUEDE PRESTAR, FUNDIR NODOS\n";
 515                         uint join1, join2;
 516                         if (hi != 0) {
 517                                 join1 = hi;
 518                                 join2 = node_num;
 519                         } else {
 520                                 join1 = node_num;
 521                                 join2 = hd;
 522                         }
 523
 524                         JoinNodes (join1, join2, padre);
 525                         return;
 526                 }
 527
 528                 WriteKeys (node, node_header, node_keys);
 529                 WriteNodoHeader (node, &node_header);
 530                 WriteBlock (node, node_num);
 531         }
 532
 533         DeleteKeys (node_keys);
 534         delete [] node;
 535         std::cout << "Borrado de una hoja listo\n";
 536 }
 537
 538 void BTree::JoinNodes (uint node1, uint node2, uint padre)
 539 {
 540         uchar *n1, *n2, *npadre;
 541         BTreeNodeHeader nh1, nh2, nhp;
 542         std::list<BTreeData *> nk1, nk2, nkpadre;
 543
 544         /* Leo los nodos */
 545         n1 = ReadBlock (node1);
 546         n2 = ReadBlock (node2);
 547         npadre = ReadBlock (padre);
 548
 549         ReadNodoHeader (n1, &nh1);
 550         ReadNodoHeader (n2, &nh2);
 551         ReadNodoHeader (npadre, &nhp);
 552
 553         nk1 = ReadKeys (n1, nh1);
 554         nk2 = ReadKeys (n2, nh2);
 555         nkpadre = ReadKeys (npadre, nhp);
 556
 557         /* Busco la clave a juntar con los nodos */
 558         std::list<BTreeData *>::iterator it = nkpadre.begin ();
 559         std::list<BTreeData *>::iterator sig;
 560
 561         Clave *cpadre;
 562         BTreeData *lchild = (*it++);
 563
 564         if (lchild->getChild () == node1) {
 565                 cpadre = (*it)->getClave ();
 566                 nkpadre.erase (it);
 567         } else {
 568                 while (it != nkpadre.end ()) {
 569                         if ((*it)->getChild () == node1)
 570                                 break;
 571                         it++;
 572                 }
 573                 cpadre = (*it)->getClave ();
 574                 nkpadre.erase (it);
 575         }
 576         sig = it++;
 577
 578         std::list<BTreeData *> newkeys;
 579         std::list<BTreeData *>::iterator i;
 580
 581         i = nk1.begin ();
 582         while (i != nk1.end ()) {
 583                 newkeys.push_back ( new BTreeLeafData ((*i)->getClave ()->Clone ()));
 584                 i++;
 585         }
 586         if (cpadre)
 587                 newkeys.push_back ( new BTreeLeafData (cpadre->Clone ()));
 588         i = nk2.begin ();
 589         while (i != nk2.end ()) {
 590                 newkeys.push_back ( new BTreeLeafData ((*i)->getClave ()->Clone ()));
 591                 i++;
 592         }
 593
 594         if (padre == 0) {
 595                 nhp.level = 0;
 596                 WriteKeys (npadre, nhp, newkeys);
 597                 WriteNodoHeader (npadre, &nhp);
 598                 WriteBlock (npadre, padre);
 599
 600                 /* TODO: Recuperar nodo1 y nodo2 */
 601         } else {
 602                 WriteKeys (n1, nh1, newkeys);
 603                 WriteNodoHeader (n1, &nh1);
 604                 WriteBlock (n1, node1);
 605
 606                 /* TODO : Recuperar node2 */
 607                 /* Actualizo punero al padre */
 608                 (*sig)->setChild (node1);
 609                 WriteKeys (npadre, nhp, nkpadre);
 610                 WriteNodoHeader (npadre, &nhp);
 611                 WriteBlock (npadre, padre);
 612         }
 613
 614         DeleteKeys (nk1);
 615         DeleteKeys (nk2);
 616         DeleteKeys (nkpadre);
 617         DeleteKeys (newkeys);
 618
 619         delete [] n1;
 620         delete [] n2;
 621         delete [] npadre;
 622 }
 623
 624 Clave *BTree::GetKey (uint node_num, char maxmin)
 625 {
 626         if (node_num == 0) return NULL;
 627
 628         uchar *node;
 629         BTreeNodeHeader node_header;
 630         std::list<BTreeData *> node_keys;
 631
 632         node = ReadBlock (node_num);
 633         ReadNodoHeader (node, &node_header);
 634         node_keys = ReadKeys (node, node_header);
 635
 636         std::list<BTreeData *>::iterator it = node_keys.begin ();
 637
 638         if (node_header.level != 0) it++;
 639
 640         Clave *k;
 641         uint free = node_header.free_space + (*it)->Size ();
 642         if (free > ((header.block_size - sizeof (BTreeNodeHeader))/2)) {
 643                 WriteKeys (node, node_header, node_keys);
 644                 WriteNodoHeader (node, &node_header);
 645                 WriteBlock (node, node_num);
 646                 DeleteKeys (node_keys);
 647
 648                 delete [] node;
 649
 650                 return NULL;
 651         }
 652
 653         if (maxmin == 0) {
 654                 k = (*it)->getClave ()->Clone ();
 655                 node_keys.erase (it);
 656         } else {
 657                 it = node_keys.end ();
 658                 it--;
 659                 k = (*it)->getClave ()->Clone ();
 660                 node_keys.erase (it);
 661         }
 662
 663         WriteKeys (node, node_header, node_keys);
 664         WriteNodoHeader (node, &node_header);
 665         WriteBlock (node, node_num);
 666         DeleteKeys (node_keys);
 667
 668         delete [] node;
 669
 670         return k;
 671 }
 672
 673 void BTree::FindBrothers (uint node_num, uint padre, uint &left, uint &right)
 674 {
 675         uchar *node;
 676         BTreeNodeHeader node_header;
 677         std::list<BTreeData *> node_keys;
 678
 679         node = ReadBlock (padre);
 680         ReadNodoHeader (node, &node_header);
 681         node_keys = ReadKeys (node, node_header);
 682
 683         std::list<BTreeData *>::iterator it = node_keys.begin ();
 684         std::list<BTreeData *>::iterator anterior = node_keys.begin ();
 685         std::list<BTreeData *>::iterator siguiente;
 686
 687         BTreeData *lchild = (*it++);
 688
 689         if (lchild->getChild () == node_num) {
 690                 /* Solo tengo hermano derecho */
 691                 std::cout << "Hermano Izquierdo : NO TENGO" << std::endl;
 692                 left = 0;
 693                 std::cout << "Hermano Derecho   : " << (*it)->getChild () << std::endl;
 694                 right = (*it)->getChild ();
 695                 return;
 696         }
 697
 698         while (it != node_keys.end ()) {
 699                 if ((*it)->getChild () == node_num)
 700                         break;
 701                 anterior = it;
 702                 it++;
 703         }
 704         siguiente = it++;
 705
 706         std::cout << "Hermano Izquierdo : " << (*anterior)->getChild () << std::endl;
 707         left = (*anterior)->getChild ();
 708         if (siguiente != node_keys.end ()) {
 709                 right = (*siguiente)->getChild ();
 710                 std::cout << "Hermano Derecho   : " << (*siguiente)->getChild () << std::endl;
 711         } else {
 712                 right = 0;
 713                 std::cout << "Hermano Derecho   : NO TENGO" << std::endl;
 714         }
 715 }
 716
 717 Clave *BTree::ReplaceKeyInFather (uint node_num, uint padre, Clave *k)
 718 {
 719         uchar *node;
 720         BTreeNodeHeader node_header;
 721         std::list<BTreeData *> node_keys;
 722
 723         node = ReadBlock (padre);
 724         ReadNodoHeader (node, &node_header);
 725         node_keys = ReadKeys (node, node_header);
 726
 727         std::list<BTreeData *>::iterator it = node_keys.begin ();
 728         std::list<BTreeData *>::iterator anterior = node_keys.begin ();
 729         std::list<BTreeData *>::iterator siguiente;
 730
 731         BTreeData *lchild = (*it++);
 732
 733         if (lchild->getChild () == node_num) {
 734                 Clave *ret = (*it)->getClave ();
 735                 (*it)->setClave (k);
 736
 737                 WriteKeys (node, node_header, node_keys);
 738                 WriteNodoHeader (node, &node_header);
 739                 WriteBlock (node, padre);
 740                 DeleteKeys (node_keys);
 741
 742                 delete [] node;
 743                 return ret;
 744         }
 745
 746         while (it != node_keys.end ()) {
 747                 if ((*it)->getChild () == node_num)
 748                         break;
 749                 anterior = it;
 750                 it++;
 751         }
 752
 753         Clave *ret = (*it)->getClave ();
 754         (*it)->setClave (k);
 755
 756         WriteKeys (node, node_header, node_keys);
 757         WriteNodoHeader (node, &node_header);
 758         WriteBlock (node, padre);
 759         DeleteKeys (node_keys);
 760
 761         delete [] node;
 762         return ret;
 763 }
 764
 765 void BTree::DelKeyFromNode (Clave *k, uint node_num, uint padre, uint left, uint right)
 766 {
 767         uint padre_hijo;
 768         uchar *node;
 769         BTreeNodeHeader node_header;
 770         std::list<BTreeData *> node_keys;
 771
 772         node = ReadBlock (padre);
 773         ReadNodoHeader (node, &node_header);
 774         node_keys = ReadKeys (node, node_header);
 775
 776         if (right != 0) {
 777                 std::cout << "Busco para la derecha y luego todo a la izquierda\n";
 778                 uchar *node_r;
 779                 BTreeNodeHeader node_hr;
 780                 std::list<BTreeData *> node_keyr;
 781
 782                 /* Busco la clave inmediatamente superior en el arbol */
 783                 padre_hijo = node_num;
 784                 do {
 785                         node_r = ReadBlock (right);
 786                         ReadNodoHeader (node_r, &node_hr);
 787                         if (node_hr.level != 0) {
 788                                 BTreeData *data_r;
 789                                 node_keyr = ReadKeys (node_r, node_hr);
 790                                 data_r = *(node_keyr.begin ());
 791                                 padre_hijo = right;
 792                                 right = data_r->getChild ();
 793
 794                                 DeleteKeys (node_keyr);
 795                                 delete [] node_r;
 796                         }
 797                 } while (node_hr.level != 0);
 798
 799                 std::cout << "Voy a reemplazar en el nodo " << right << std::endl;
 800
 801                 /* Reemplazo la clave a borrar por la de la hoja */
 802                 node_keyr = ReadKeys (node_r, node_hr);
 803                 BTreeData *reemplazar = *(node_keyr.begin ());
 804
 805                 std::string ss = *reemplazar;
 806                 std::cout << "Voy a reemplazar por : " << ss << std::endl;
 807
 808                 BTreeData *data = new BTreeLeafData (k->Clone());
 809
 810                 std::list<BTreeData *>::iterator it = node_keys.begin ();
 811                 while (it != node_keys.end ()) {
 812                         if ((*data) == (*(*it))) {
 813                                 break;
 814                         }
 815                         it++;
 816                 }
 817                 (*it)->setClave (reemplazar->getClave ());
 818                 reemplazar->setClave (k->Clone ());
 819
 820                 std::cout << "Tengo todo reemplazado ...\n";
 821
 822                 /* Grabo los nodos */
 823                 WriteKeys (node, node_header, node_keys);
 824                 WriteNodoHeader (node, &node_header);
 825                 WriteBlock (node, padre);
 826                 DeleteKeys (node_keys);
 827                 delete [] node;
 828
 829                 WriteKeys (node_r, node_hr, node_keyr);
 830                 WriteNodoHeader (node_r, &node_hr);
 831                 WriteBlock (node_r, right);
 832                 DeleteKeys (node_keyr);
 833                 delete [] node_r;
 834
 835                 std::cout << "Grabe todo en disco ...\n";
 836
 837                 /* Ahora debo eliminar la clave que puse en el nodo hoja */
 838                 std::cout << "Borro la clave desde la hoja!\n";
 839
 840                 DelKeyFromLeaf (k, right, padre_hijo);
 841
 842                 std::cout << "Listo, Listo!\n";
 843         } else if (left != 0) {
 844         } else {
 845                 std::cout << "PANIC : No tengo hijos para reemplazar!!!!\n";
 846         }
 847 }
 848
 849 void BTree::ReadNodoHeader (uchar *node, BTreeNodeHeader *header)
 850 {
 851         memcpy (header, node, sizeof (BTreeNodeHeader));
 852 }
 853
 854 void BTree::WriteNodoHeader (uchar *node, BTreeNodeHeader *header)
 855 {
 856         memcpy (node, header, sizeof (BTreeNodeHeader));
 857 }
 858
 859 uchar *BTree::ReadBlock (uint num)
 860 {
 861         /* Como el bloque 0 se usa para el header, el Nodo "num"
 862          * está en el bloque "num+1"
 863          */
 864         num++;
 865
 866         uchar *out = new uchar[header.block_size];
 867
 868         fseek (fp, num*header.block_size, SEEK_SET);
 869         fread (out, 1, header.block_size, fp);
 870
 871         return out;
 872 }
 873
 874 std::list<BTreeData *> BTree::ReadKeys (uchar *node, BTreeNodeHeader &node_header)
 875 {
 876         std::list<BTreeData *> keys;
 877         node += sizeof (BTreeNodeHeader);
 878         uint count = node_header.item_count;
 879
 880         if (node_header.item_count == 0) return keys;
 881
 882         if (node_header.level != 0) {
 883                 /* Si no es una hoja, lo primero que tengo es un BTreeChildData */
 884                 BTreeChildData *d = new BTreeChildData (node);
 885                 node += d->Size ();
 886                 keys.push_back (d);
 887                 count--;
 888         }
 889
 890         for (uint i=0; i<count; i++) {
 891                 /* TODO : El tipo de clave deberia ser usado
 892                  * dependiendo de algun dato en el header del
 893                  * arbol
 894                  */
 895                 /* TODO : Detectar si estoy en una hoja */
 896                 BTreeData *data;
 897                 if (node_header.level == 0) {
 898                         data = new BTreeLeafData (node, key_type);
 899                 } else {
 900                         data = new BTreeData (node, key_type);
 901                 }
 902                 node += data->Size ();
 903                 keys.push_back (data);
 904         }
 905
 906         return keys;
 907 }
 908
 909 void BTree::WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<BTreeData *> &keys)
 910 {
 911         std::list<BTreeData *>::iterator it = keys.begin ();
 912
 913         node += sizeof (BTreeNodeHeader);
 914
 915         node_header.item_count = 0;
 916         node_header.free_space = header.block_size - sizeof (BTreeNodeHeader);
 917
 918         while (it != keys.end ()) {
 919                 BTreeData *d = (*it);
 920                 uchar *n = d->ToArray ();
 921                 memcpy (node, n, d->Size ());
 922                 delete [] n;
 923                 node += d->Size ();
 924                 node_header.free_space -= d->Size ();
 925                 node_header.item_count++;
 926                 it++;
 927         }
 928
 929         /* TODO : incrementar node_header.item_count aca o fuera de este metodo? */
 930 }
 931
 932 void BTree::PrintNode (uint num)
 933 {
 934         uchar *node = ReadBlock (num);
 935         BTreeNodeHeader node_header;
 936         ReadNodoHeader (node, &node_header);
 937
 938         std::list<BTreeData *> node_keys = ReadKeys (node, node_header);
 939         std::list<BTreeData *>::iterator it = node_keys.begin ();
 940
 941         std::cout << "Nodo  : " << num << std::endl;
 942         std::cout << "Level : " << node_header.level << std::endl;
 943         std::cout << "Items : " << node_header.item_count << std::endl;
 944         std::cout << "Free  : " << node_header.free_space << " (" << (header.block_size - sizeof (BTreeNodeHeader)) << ")" << std::endl;
 945         while (it != node_keys.end ()) {
 946                 std::string s = *(*it);
 947                 std::cout << s << " ";
 948                 it++;
 949         }
 950         std::cout << std::endl;
 951
 952         delete [] node;
 953         DeleteKeys (node_keys);
 954 }
 955
 956 uchar *BTree::NewBlock (uint &num)
 957 {
 958         long filelen;
 959         uchar *node;
 960         BTreeNodeHeader nh;
 961
 962         fseek (fp, 0, SEEK_END);
 963         filelen = ftell (fp);
 964
 965         num = filelen/header.block_size - 1;
 966
 967         node = new uchar[header.block_size];
 968         ReadNodoHeader (node, &nh);
 969         nh.level = 0;
 970         nh.free_space = header.block_size - sizeof (BTreeNodeHeader);
 971         nh.item_count = 0;
 972         WriteNodoHeader (node, &nh);
 973         WriteBlock (node, num);
 974
 975         return node;
 976 }
 977
 978 BTreeFindResult *BTree::FindKey (const Clave &k)
 979 {
 980         return FindKeyR (&k, 0);
 981 }
 982
 983 BTreeFindResult *BTree::FindKeyR (const Clave *k, uint node_num)
 984 {
 985         std::list<BTreeData *> node_keys;
 986         BTreeNodeHeader node_header;
 987
 988         /* Leo el nodo raiz para empezar a agregar */
 989         uchar *node = ReadBlock (node_num);
 990         ReadNodoHeader (node, &node_header);
 991         node_keys = ReadKeys (node, node_header);
 992
 993         std::list<BTreeData *>::iterator it = node_keys.begin ();
 994         std::list<BTreeData *>::iterator posterior;
 995         std::list<BTreeData *>::iterator ultima;
 996
 997         /* Se supone que la primera es un hijo :) */
 998         BTreeData *lchild;
 999         if (node_header.level != 0) {
1000                 lchild = (*it++);
1001         }
1002         posterior = it;
1003
1004         BTreeData *data;
1005         if (node_header.level == 0)
1006                 data = new BTreeLeafData ((Clave *)k);
1007         else
1008                 data = new BTreeData ((Clave *)k, 0);
1009
1010         while (it != node_keys.end ()) {
1011                 if ((*data) == (*(*it))) {
1012                         /* La encontre!, retorno */
1013                         delete [] node;
1014                         DeleteKeys (node_keys);
1015                         BTreeFindResult *result = new BTreeFindResult ();
1016                         result->node = node_num;
1017                         result->header = node_header;
1018                         return result;
1019                 }
1020
1021                 if ((*data) < (*(*it)))
1022                         break;
1023                 ultima = it;
1024                 it++;
1025         }
1026
1027         /* TODO: Aca faltaria liberar memoria */
1028         if (it == posterior)
1029                 return FindKeyR (k, lchild->getChild ());
1030
1031         return FindKeyR (k, (*ultima)->getChild ());
1032 }
1033
1034 void BTree::DeleteKeys (std::list<BTreeData *> &keys)
1035 {
1036         std::list<BTreeData *>::iterator it = keys.begin ();
1037
1038         while (it != keys.end ()) {
1039                 BTreeData *d = (*it);
1040                 delete d;
1041                 it++;
1042         }
1043 }