]> git.llucax.com Git - z.facultad/75.52/treemulator.git/blob - src/btree.cpp
Profiling
[z.facultad/75.52/treemulator.git] / 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                         }
423                         return;
424                 }
425
426                 if ((*k) < (*(*it)))
427                         break;
428                 ultima = it;
429                 it++;
430         }
431
432         /* Si llego aca y estoy en nivel 0 (una hoja) quiere
433          * decir que no lo encontre
434          */
435         if (node_header.level == 0) {
436                 std::cout << "Clave no encontrada\n";
437                 return;
438         }
439
440         /* TODO: Aca faltaria liberar memoria */
441         if (it == posterior) {
442                 DelKeyR (k, lchild->getChild (), node_num);
443         } else {
444                 DelKeyR (k, (*ultima)->getChild (), node_num);
445         }
446 }
447
448 void BTree::DelKeyFromLeaf (Clave *k, uint node_num, uint padre)
449 {
450         BTreeData *data;
451         uchar *node;
452         BTreeNodeHeader node_header;
453         std::list<BTreeData *> node_keys;
454
455         node = ReadBlock (node_num);
456         ReadNodoHeader (node, &node_header);
457         node_keys = ReadKeys (node, node_header);
458
459         data = new BTreeLeafData (k);
460
461         std::list<BTreeData *>::iterator it;
462         it = node_keys.begin ();
463         while (it != node_keys.end ()) {
464                 if ((*data) == (*(*it))) {
465                         node_keys.erase (it);
466                         break;
467                 }
468                 it++;
469         }
470
471         delete data;
472
473         /* TODO : que pasa cuando el nodo se queda vacio o con 
474          * muy pocas claveS??
475          */
476         WriteKeys (node, node_header, node_keys);
477         WriteNodoHeader (node, &node_header);
478         WriteBlock (node, node_num);
479
480         /* Veo si se cumple la condición de minimalidad */
481         if ((node_header.free_space <= (header.block_size/2)) && (node_num != 0)) {
482                 /* Oops! Debo pedir prestada clave */
483                 uint hi, hd;
484                 Clave *pedida;
485
486                 FindBrothers (node_num, padre, hi, hd);
487         
488                 if ((pedida = GetKey (hi, 1)) != NULL) {
489                         std::string s = *pedida;
490                         std::cout << "Clave Pedida : " << s << std::endl;
491
492                         pedida = ReplaceKeyInFather (node_num, padre, pedida);
493
494                         node_keys.insert (node_keys.begin (), new BTreeLeafData (pedida));
495                 } else if ((pedida = GetKey (hd, 0)) != NULL) {
496                         std::string s = *pedida;
497                         std::cout << "Clave Pedida : " << s << std::endl;
498
499                         pedida = ReplaceKeyInFather (node_num, padre, pedida);
500
501                         node_keys.push_back (new BTreeLeafData (pedida));
502                 } else {
503                         std::cout << "NADIE ME PUEDE PRESTAR, FUNDIR NODOS\n";
504                         uint join1, join2;
505                         if (hi != 0) {
506                                 join1 = hi;
507                                 join2 = node_num;
508                         } else {
509                                 join1 = node_num;
510                                 join2 = hd;
511                         }
512
513                         JoinNodes (join1, join2, padre);
514                         return;
515                 }
516
517                 WriteKeys (node, node_header, node_keys);
518                 WriteNodoHeader (node, &node_header);
519                 WriteBlock (node, node_num);
520         }
521
522         DeleteKeys (node_keys);
523         delete [] node;
524         std::cout << "Borrado de una hoja listo\n";
525 }
526
527 void BTree::JoinNodes (uint node1, uint node2, uint padre)
528 {
529         uchar *n1, *n2, *npadre;
530         BTreeNodeHeader nh1, nh2, nhp;
531         std::list<BTreeData *> nk1, nk2, nkpadre;
532
533         /* Leo los nodos */
534         n1 = ReadBlock (node1);
535         n2 = ReadBlock (node2);
536         npadre = ReadBlock (padre);
537
538         ReadNodoHeader (n1, &nh1);
539         ReadNodoHeader (n2, &nh2);
540         ReadNodoHeader (npadre, &nhp);
541
542         nk1 = ReadKeys (n1, nh1);
543         nk2 = ReadKeys (n2, nh2);
544         nkpadre = ReadKeys (npadre, nhp);
545
546         /* Busco la clave a juntar con los nodos */
547         std::list<BTreeData *>::iterator it = nkpadre.begin ();
548         std::list<BTreeData *>::iterator sig;
549         
550         Clave *cpadre;
551         BTreeData *lchild = (*it++);
552
553         if (lchild->getChild () == node1) {
554                 cpadre = (*it)->getClave ();
555                 nkpadre.erase (it);
556         } else {
557                 while (it != nkpadre.end ()) {
558                         if ((*it)->getChild () == node1)
559                                 break;
560                         it++;
561                 }
562                 cpadre = (*it)->getClave ();
563                 nkpadre.erase (it);
564         }
565         sig = it++;
566
567         std::list<BTreeData *> newkeys;
568         std::list<BTreeData *>::iterator i;
569
570         i = nk1.begin ();
571         while (i != nk1.end ()) {
572                 newkeys.push_back ( new BTreeLeafData ((*i)->getClave ()->Clone ()));
573                 i++;
574         }
575         if (cpadre)
576                 newkeys.push_back ( new BTreeLeafData (cpadre->Clone ()));
577         i = nk2.begin ();
578         while (i != nk2.end ()) {
579                 newkeys.push_back ( new BTreeLeafData ((*i)->getClave ()->Clone ()));
580                 i++;
581         }
582
583         if (padre == 0) {
584                 nhp.level = 0;
585                 WriteKeys (npadre, nhp, newkeys);
586                 WriteNodoHeader (npadre, &nhp);
587                 WriteBlock (npadre, padre);
588
589                 /* TODO: Recuperar nodo1 y nodo2 */
590         } else {
591                 WriteKeys (n1, nh1, newkeys);
592                 WriteNodoHeader (n1, &nh1);
593                 WriteBlock (n1, node1);
594
595                 /* TODO : Recuperar node2 */
596                 /* Actualizo punero al padre */
597                 (*sig)->setChild (node1);
598                 WriteKeys (npadre, nhp, nkpadre);
599                 WriteNodoHeader (npadre, &nhp);
600                 WriteBlock (npadre, padre);
601         }
602
603         DeleteKeys (nk1);
604         DeleteKeys (nk2);
605         DeleteKeys (nkpadre);
606         DeleteKeys (newkeys);
607
608         delete [] n1;
609         delete [] n2;
610         delete [] npadre;
611 }
612
613 Clave *BTree::GetKey (uint node_num, char maxmin)
614 {
615         if (node_num == 0) return NULL;
616
617         uchar *node;
618         BTreeNodeHeader node_header;
619         std::list<BTreeData *> node_keys;
620
621         node = ReadBlock (node_num);
622         ReadNodoHeader (node, &node_header);
623         node_keys = ReadKeys (node, node_header);
624
625         std::list<BTreeData *>::iterator it = node_keys.begin ();
626
627         if (node_header.level != 0) it++;
628
629         Clave *k;
630         uint free = node_header.free_space + (*it)->Size ();
631         if (free > (header.block_size/2)) {
632                 WriteKeys (node, node_header, node_keys);
633                 WriteNodoHeader (node, &node_header);
634                 WriteBlock (node, node_num);
635                 DeleteKeys (node_keys);
636         
637                 delete [] node;
638
639                 return NULL;
640         }
641
642         if (maxmin == 0) {
643                 k = (*it)->getClave ()->Clone ();
644                 node_keys.erase (it);
645         } else {
646                 it = node_keys.end ();
647                 it--;
648                 k = (*it)->getClave ()->Clone ();
649                 node_keys.erase (it);
650         }
651
652         WriteKeys (node, node_header, node_keys);
653         WriteNodoHeader (node, &node_header);
654         WriteBlock (node, node_num);
655         DeleteKeys (node_keys);
656
657         delete [] node;
658
659         return k;
660 }
661
662 void BTree::FindBrothers (uint node_num, uint padre, uint &left, uint &right)
663 {
664         uchar *node;
665         BTreeNodeHeader node_header;
666         std::list<BTreeData *> node_keys;
667
668         node = ReadBlock (padre);
669         ReadNodoHeader (node, &node_header);
670         node_keys = ReadKeys (node, node_header);
671
672         std::list<BTreeData *>::iterator it = node_keys.begin ();
673         std::list<BTreeData *>::iterator anterior = node_keys.begin ();
674         std::list<BTreeData *>::iterator siguiente;
675
676         BTreeData *lchild = (*it++);
677
678         if (lchild->getChild () == node_num) {
679                 /* Solo tengo hermano derecho */
680                 std::cout << "Hermano Izquierdo : NO TENGO" << std::endl;
681                 left = 0;
682                 std::cout << "Hermano Derecho   : " << (*it)->getChild () << std::endl;
683                 right = (*it)->getChild ();
684                 return;
685         }
686
687         while (it != node_keys.end ()) {
688                 if ((*it)->getChild () == node_num)
689                         break;
690                 anterior = it;
691                 it++;
692         }
693         siguiente = it++;
694
695         std::cout << "Hermano Izquierdo : " << (*anterior)->getChild () << std::endl;
696         left = (*anterior)->getChild ();
697         if (siguiente != node_keys.end ()) {
698                 right = (*siguiente)->getChild ();
699                 std::cout << "Hermano Derecho   : " << (*siguiente)->getChild () << std::endl;
700         } else {
701                 right = 0;
702                 std::cout << "Hermano Derecho   : NO TENGO" << std::endl;
703         }
704 }
705
706 Clave *BTree::ReplaceKeyInFather (uint node_num, uint padre, Clave *k)
707 {
708         uchar *node;
709         BTreeNodeHeader node_header;
710         std::list<BTreeData *> node_keys;
711
712         node = ReadBlock (padre);
713         ReadNodoHeader (node, &node_header);
714         node_keys = ReadKeys (node, node_header);
715
716         std::list<BTreeData *>::iterator it = node_keys.begin ();
717         std::list<BTreeData *>::iterator anterior = node_keys.begin ();
718         std::list<BTreeData *>::iterator siguiente;
719
720         BTreeData *lchild = (*it++);
721
722         if (lchild->getChild () == node_num) {
723                 Clave *ret = (*it)->getClave ();
724                 (*it)->setClave (k);
725
726                 WriteKeys (node, node_header, node_keys);
727                 WriteNodoHeader (node, &node_header);
728                 WriteBlock (node, padre);
729                 DeleteKeys (node_keys);
730
731                 delete [] node;
732                 return ret;
733         }
734
735         while (it != node_keys.end ()) {
736                 if ((*it)->getChild () == node_num)
737                         break;
738                 anterior = it;
739                 it++;
740         }
741
742         Clave *ret = (*it)->getClave ();
743         (*it)->setClave (k);
744
745         WriteKeys (node, node_header, node_keys);
746         WriteNodoHeader (node, &node_header);
747         WriteBlock (node, padre);
748         DeleteKeys (node_keys);
749
750         delete [] node;
751         return ret;
752 }
753
754 void BTree::DelKeyFromOther (const Clave &k, BTreeFindResult *r)
755 {
756 }
757
758 void BTree::ReadNodoHeader (uchar *node, BTreeNodeHeader *header)
759 {
760         memcpy (header, node, sizeof (BTreeNodeHeader));
761 }
762
763 void BTree::WriteNodoHeader (uchar *node, BTreeNodeHeader *header)
764 {
765         memcpy (node, header, sizeof (BTreeNodeHeader));
766 }
767
768 uchar *BTree::ReadBlock (uint num)
769 {
770         /* Como el bloque 0 se usa para el header, el Nodo "num"
771          * está en el bloque "num+1"
772          */
773         num++;
774
775         uchar *out = new uchar[header.block_size];
776
777         fseek (fp, num*header.block_size, SEEK_SET);    
778         fread (out, 1, header.block_size, fp);
779
780         return out;
781 }
782
783 std::list<BTreeData *> BTree::ReadKeys (uchar *node, BTreeNodeHeader &node_header)
784 {
785         std::list<BTreeData *> keys;
786         node += sizeof (BTreeNodeHeader);
787         uint count = node_header.item_count;
788
789         if (node_header.item_count == 0) return keys;
790
791         if (node_header.level != 0) {
792                 /* Si no es una hoja, lo primero que tengo es un BTreeChildData */
793                 BTreeChildData *d = new BTreeChildData (node);
794                 node += d->Size ();
795                 keys.push_back (d);
796                 count--;
797         }
798
799         for (uint i=0; i<count; i++) {
800                 /* TODO : El tipo de clave deberia ser usado 
801                  * dependiendo de algun dato en el header del
802                  * arbol
803                  */
804                 /* TODO : Detectar si estoy en una hoja */
805                 BTreeData *data;
806                 if (node_header.level == 0) {
807                         data = new BTreeLeafData (node, key_type);
808                 } else {
809                         data = new BTreeData (node, key_type);
810                 }
811                 node += data->Size ();
812                 keys.push_back (data);
813         }
814
815         return keys;
816 }
817
818 void BTree::WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<BTreeData *> &keys)
819 {
820         std::list<BTreeData *>::iterator it = keys.begin ();
821
822         node += sizeof (BTreeNodeHeader);
823
824         node_header.item_count = 0;
825         node_header.free_space = header.block_size - sizeof (BTreeNodeHeader);
826
827         while (it != keys.end ()) {
828                 BTreeData *d = (*it);
829                 uchar *n = d->ToArray ();
830                 memcpy (node, n, d->Size ());
831                 delete [] n;
832                 node += d->Size ();
833                 node_header.free_space -= d->Size ();
834                 node_header.item_count++;
835                 it++;
836         }
837
838         /* TODO : incrementar node_header.item_count aca o fuera de este metodo? */
839 }
840                 
841 void BTree::PrintNode (uint num)
842 {
843         uchar *node = ReadBlock (num);
844         BTreeNodeHeader node_header;
845         ReadNodoHeader (node, &node_header);
846                 
847         std::list<BTreeData *> node_keys = ReadKeys (node, node_header);
848         std::list<BTreeData *>::iterator it = node_keys.begin ();
849
850         std::cout << "Nodo  : " << num << std::endl;
851         std::cout << "Level : " << node_header.level << std::endl;
852         std::cout << "Items : " << node_header.item_count << std::endl;
853         std::cout << "Free  : " << node_header.free_space << " (" << (header.block_size - sizeof (BTreeNodeHeader)) << ")" << std::endl;
854         while (it != node_keys.end ()) {
855                 std::string s = *(*it);
856                 std::cout << s << " "; 
857                 it++;
858         }
859         std::cout << std::endl;
860
861         delete [] node;
862         DeleteKeys (node_keys);
863 }
864
865 uchar *BTree::NewBlock (uint &num)
866 {
867         long filelen;
868         uchar *node;
869         BTreeNodeHeader nh;
870
871         fseek (fp, 0, SEEK_END);
872         filelen = ftell (fp);
873
874         num = filelen/header.block_size - 1;
875
876         node = new uchar[header.block_size];
877         ReadNodoHeader (node, &nh);
878         nh.level = 0;
879         nh.free_space = header.block_size - sizeof (BTreeNodeHeader);
880         nh.item_count = 0;
881         WriteNodoHeader (node, &nh);
882         WriteBlock (node, num);
883
884         return node;
885 }
886
887 BTreeFindResult *BTree::FindKey (const Clave &k)
888 {
889         return FindKeyR (&k, 0);
890 }
891
892 BTreeFindResult *BTree::FindKeyR (const Clave *k, uint node_num)
893 {
894         std::list<BTreeData *> node_keys;
895         BTreeNodeHeader node_header;
896
897         /* Leo el nodo raiz para empezar a agregar */
898         uchar *node = ReadBlock (node_num);
899         ReadNodoHeader (node, &node_header);
900         node_keys = ReadKeys (node, node_header);
901
902         std::list<BTreeData *>::iterator it = node_keys.begin ();
903         std::list<BTreeData *>::iterator posterior;
904         std::list<BTreeData *>::iterator ultima;
905         
906         /* Se supone que la primera es un hijo :) */
907         BTreeData *lchild;
908         if (node_header.level != 0) {
909                 lchild = (*it++);
910         }
911         posterior = it;
912
913         BTreeData *data;
914         if (node_header.level == 0)
915                 data = new BTreeLeafData ((Clave *)k);
916         else
917                 data = new BTreeData ((Clave *)k, 0);
918
919         while (it != node_keys.end ()) {
920                 if ((*data) == (*(*it))) {
921                         /* La encontre!, retorno */
922                         delete [] node;
923                         DeleteKeys (node_keys);
924                         BTreeFindResult *result = new BTreeFindResult ();
925                         result->node = node_num;
926                         result->header = node_header;
927                         return result;
928                 }
929
930                 if ((*data) < (*(*it)))
931                         break;
932                 ultima = it;
933                 it++;
934         }
935
936         /* TODO: Aca faltaria liberar memoria */
937         if (it == posterior)
938                 return FindKeyR (k, lchild->getChild ());
939                 
940         return FindKeyR (k, (*ultima)->getChild ());
941 }
942
943 void BTree::DeleteKeys (std::list<BTreeData *> &keys)
944 {
945         std::list<BTreeData *>::iterator it = keys.begin ();
946
947         while (it != keys.end ()) {
948                 BTreeData *d = (*it);
949                 delete d;
950                 it++;
951         }
952 }