emufs/indice_b.c

   1
   2 #include "indice_b.h"
   3 #include "common.h"
   4
   5 /* Cantidad de claves por nodo */
   6 #define CANT_HIJOS(x) ((x->tam_bloque-sizeof(B_NodoHeader))/sizeof(B_NodoEntry))
   7 #define CANT_NODOS(x) (CANT_HIJOS(x)+1)
   8 #define MIN_HIJOS(x) (CANT_HIJOS(x)/2)
   9
  10 /* Auxiliares */
  11 static void b_grabar_nodo(INDICE *idx, int id, char *data);
  12 static int b_ultimo_id(INDICE *idx);
  13 static char *b_leer_nodo(INDICE *idx, int id);
  14 static char *b_crear_nodo(INDICE *idx, int *i);
  15 static void b_leer_header(char *src, B_NodoHeader *header);
  16 static void b_actualizar_header(char *src, B_NodoHeader *header);
  17 static B_NodoEntry *b_leer_claves(char *src, B_NodoHeader *header);
  18 static void b_insertar_en_nodo(INDICE *idx, CLAVE clave, INDICE_DATO dato, int nodo_id, char *nodo, int hijo1, int hijo2);
  19 const int b_elegir_izquierdo(INDICE *idx, int a, int b);
  20
  21 void emufs_indice_b_crear(INDICE *idx)
  22 {
  23         FILE *fp;
  24         char *bloque;
  25         B_NodoHeader header;
  26
  27         header.cant = 0;
  28         header.nivel = 0;
  29         header.padre = -1;
  30         header.hijo_izquierdo = -1;
  31
  32         fp = fopen(idx->filename, "w");
  33         PERR("Creando indice");
  34         fprintf(stderr, "Archivo = (%s)\n", idx->filename);
  35         if (fp == NULL) {
  36                 PERR("Error al crear el archivo");
  37                 return;
  38         }
  39
  40         /* Creo el archivo con el Nodo raiz */
  41         bloque = (char *)malloc(idx->tam_bloque);
  42         memset(bloque, -1, idx->tam_bloque);
  43
  44         memcpy(bloque, &header, sizeof(B_NodoHeader));
  45
  46         fwrite(bloque, idx->tam_bloque, 1, fp);
  47         fclose(fp);
  48 }
  49
  50 int emufs_indice_b_insertar(INDICE *idx, CLAVE clave, INDICE_DATO dato)
  51 {
  52         int i, nodo_id, padre_id;
  53         B_NodoHeader header;
  54         B_NodoEntry *claves;
  55         char *nodo, *padre;
  56
  57         /* Leo la raiz */
  58         nodo = b_leer_nodo(idx, 0);
  59         padre_id = nodo_id = 0;
  60         padre = NULL;
  61         while (nodo) {
  62                 if (padre) free(padre);
  63                 padre = nodo;
  64                 padre_id = nodo_id;
  65                 b_leer_header(nodo, &header);
  66                 claves = b_leer_claves(nodo, &header);
  67                 i=0;
  68                 while ((i<header.cant) && (emufs_indice_es_menor(idx, claves[i].clave, clave))) i++;
  69                 if ((i<header.cant) && (emufs_indice_es_igual(idx, claves[i].clave, clave))) {
  70                         /* CLAVE DUPLICADA! */
  71                         return 0;
  72                 } else {
  73                         if (i == 0) {
  74                                 nodo = b_leer_nodo(idx, header.hijo_izquierdo);
  75                                 nodo_id = header.hijo_izquierdo;
  76                         } else {
  77                                 nodo = b_leer_nodo(idx, claves[i-1].hijo_derecho);
  78                                 nodo_id = claves[i-1].hijo_derecho;
  79                         }
  80                 }
  81         }
  82
  83         if (nodo) free(nodo);
  84         nodo = padre;
  85         nodo_id = padre_id;
  86         b_insertar_en_nodo(idx, clave, dato, nodo_id, nodo, -1, -1);
  87         return 1; /* Agregar OK! */
  88 }
  89
  90 INDICE_DATO emufs_indice_b_buscar(INDICE *idx, CLAVE clave)
  91 {
  92         int i;
  93         INDICE_DATO ret;
  94         B_NodoHeader header;
  95         B_NodoEntry *claves;
  96         char *nodo, *tmp;
  97
  98         /* Leo la raiz */
  99         nodo = b_leer_nodo(idx, 0);
 100         while (nodo) {
 101                 b_leer_header(nodo, &header);
 102                 claves = b_leer_claves(nodo, &header);
 103                 i=0;
 104                 while ((i<header.cant) && (emufs_indice_es_menor(idx, claves[i].clave, clave))) i++;
 105                 if (emufs_indice_es_igual(idx, claves[i].clave, clave)) {
 106                                 ret = claves[i].dato;
 107                                 free(nodo);
 108                                 return ret;
 109                 } else {
 110                         tmp = nodo;
 111                         if (i == 0) {
 112                                 nodo = b_leer_nodo(idx, header.hijo_izquierdo);
 113                         } else {
 114                                 nodo = b_leer_nodo(idx, claves[i-1].hijo_derecho);
 115                         }
 116                         free(tmp);
 117                 }
 118         }
 119
 120         /* Nodo no encontrado */
 121         ret.id = ret.bloque = -1;
 122         return ret;
 123 }
 124
 125 static int b_ultimo_id(INDICE *idx)
 126 {
 127         int i;
 128         FILE *fp;
 129         fp = fopen(idx->filename, "r");
 130         fseek(fp, 0, SEEK_END);
 131         i = ftell(fp)/idx->tam_bloque;
 132         fclose(fp);
 133
 134         return i;
 135 }
 136
 137 static char *b_crear_nodo(INDICE *idx, int *id)
 138 {
 139         char *bloque;
 140         B_NodoHeader header;
 141
 142         (*id) = b_ultimo_id(idx);
 143
 144         header.cant = 0;
 145         header.nivel = 0;
 146         header.hijo_izquierdo = -1;
 147         header.padre = -1;
 148
 149         bloque = (char *)malloc(idx->tam_bloque);
 150         memset(bloque, -1, idx->tam_bloque);
 151         memcpy(bloque, &header, sizeof(B_NodoHeader));
 152
 153         b_grabar_nodo(idx, *id, bloque);
 154
 155         return bloque;
 156 }
 157
 158 static char *b_leer_nodo(INDICE *idx, int id)
 159 {
 160         FILE *fp;
 161         char *out;
 162
 163         if (id < 0) return NULL;
 164
 165         fp = fopen(idx->filename, "r");
 166         if (fp == NULL) return NULL;
 167
 168         fseek(fp, id*idx->tam_bloque, SEEK_SET);
 169
 170         out = (char *)malloc(idx->tam_bloque);
 171         if (out == NULL) {
 172                 fclose(fp);
 173                 return NULL;
 174         }
 175
 176         if (fread(out, 1, idx->tam_bloque, fp) != idx->tam_bloque) {
 177                 free(out);
 178                 /* No se puso leer el nodo */
 179                 fclose(fp);
 180                 return NULL;
 181         }
 182
 183         fclose(fp);
 184         return out;
 185 }
 186
 187 static void b_grabar_nodo(INDICE *idx, int id, char *data)
 188 {
 189         FILE *fp;
 190
 191 /*      if (id > b_ultimo_id()) {
 192                 printf("AGREGANDO AL FINAL\n");
 193                 fp = fopen(FILENAME, "a");
 194                 if (fp == NULL) {
 195                 _("No se pudo abrir archivo\n");
 196                         return;
 197                 }
 198         } else {
 199                 fp = fopen(FILENAME, "w");
 200                 if (fp == NULL) {
 201                 _("No se pudo abrir archivo\n");
 202                         return;
 203                 }
 204                 fseek(fp, id*BLOCK_SIZE, SEEK_SET);
 205                 printf("SOLO GUARDO DATA\n");
 206         }*/
 207
 208         fp = fopen(idx->filename, "r+");
 209         fseek(fp, id*idx->tam_bloque, SEEK_SET);
 210         fwrite(data, 1, idx->tam_bloque, fp);
 211         fclose(fp);
 212 }
 213
 214 static void b_leer_header(char *src, B_NodoHeader *header)
 215 {
 216         if (!src) return;
 217
 218         memcpy(header, src, sizeof(B_NodoHeader));
 219 }
 220
 221 static void b_actualizar_header(char *src, B_NodoHeader *header)
 222 {
 223         if (!src) return;
 224         memcpy(src, header, sizeof(B_NodoHeader));
 225 }
 226
 227 static B_NodoEntry *b_leer_claves(char *src, B_NodoHeader *header)
 228 {
 229         return (B_NodoEntry *)(src+sizeof(B_NodoHeader));
 230 }
 231
 232 static void b_insertar_en_nodo(INDICE *idx, CLAVE clave, INDICE_DATO dato, int nodo_id, char *nodo, int hijo1, int hijo2)
 233 {
 234         char *padre, *nuevo;
 235         int nuevo_id;
 236         int i, j;
 237         static int rompi=0;
 238         char salir = 0;
 239         B_NodoHeader nodo_header, nuevo_header;
 240         B_NodoEntry *claves, *tmp_claves, *claves_nuevo;
 241
 242         do {
 243                 if (!nodo) {
 244                         /* CREAR NODO? */
 245                         nodo = b_crear_nodo(idx, &nodo_id);
 246                 }
 247                 b_leer_header(nodo, &nodo_header);
 248                 claves = b_leer_claves(nodo, &nodo_header);
 249
 250                 padre = b_leer_nodo(idx, nodo_header.padre);
 251
 252                 if (nodo_header.cant == CANT_HIJOS(idx)) {
 253                         int total;
 254                         nuevo = b_crear_nodo(idx, &nuevo_id);
 255                         i=0;
 256                         /* Creo una lista ordenada de los nodos a partir */
 257                         tmp_claves = (B_NodoEntry *)malloc(sizeof(B_NodoEntry)*(nodo_header.cant+1));
 258                         total = nodo_header.cant;
 259                         while ((i<nodo_header.cant) && (emufs_indice_es_menor(idx, claves[i].clave, clave))) {
 260                                 tmp_claves[i] = claves[i];
 261                                 i++;
 262                         }
 263                         tmp_claves[i].clave = clave;
 264                         tmp_claves[i].dato = dato;
 265                         tmp_claves[i].hijo_derecho = hijo1;
 266                         tmp_claves[i+1].hijo_derecho = hijo2;
 267                         while (i < nodo_header.cant) {
 268                                 tmp_claves[i+1] = claves[i];
 269                                 i++;
 270                         }
 271
 272                         /* Asigno a cada nodo lo que corresponde */
 273                         b_leer_header(nuevo, &nuevo_header);
 274
 275                         nuevo_header.nivel = nodo_header.nivel;
 276                         nodo_header.cant = total/2;
 277                         nuevo_header.cant = total - nodo_header.cant;
 278
 279                         memset(claves, '*', idx->tam_bloque-sizeof(B_NodoHeader));
 280                         for(j=0; j<nodo_header.cant; j++)
 281                                 claves[j] = tmp_claves[j];
 282
 283                         claves_nuevo = b_leer_claves(nuevo, &nuevo_header);
 284                         memset(claves_nuevo, '*', idx->tam_bloque-sizeof(B_NodoHeader));
 285                         for(j=0; j<nuevo_header.cant; j++)
 286                                 claves_nuevo[j] = tmp_claves[j+total/2+1];
 287
 288                         b_actualizar_header(nodo, &nodo_header);
 289                         b_actualizar_header(nuevo, &nuevo_header);
 290
 291                         if (nodo_id != 0) {
 292                                 clave = tmp_claves[total/2].clave;
 293                                 /* XXX dato.bloque = nuevo_id; */
 294
 295                                 b_grabar_nodo(idx, nodo_id, nodo);
 296                                 b_grabar_nodo(idx, nuevo_id, nuevo);
 297                                 free(nodo);
 298                                 free(nuevo);
 299                                 free(tmp_claves);
 300
 301                                 hijo1 = nodo_id;
 302                                 hijo2 = nuevo_id;
 303
 304                                 nodo = padre;
 305                                 nodo_id = nodo_header.padre;
 306                         } else {
 307                                 /* Oops, parti el raiz, y este debe quedar en 0, lo paso a otro bloque
 308                                  * y dejo el padre vacio
 309                                  */
 310                                 char *tmp_nuevo = b_crear_nodo(idx, &nodo_id);
 311                                 memcpy(tmp_nuevo, nodo, idx->tam_bloque);
 312                                 free(nodo);
 313                                 nodo = tmp_nuevo;
 314
 315                                 clave = tmp_claves[total/2].clave;
 316                                 /* XXX dato.bloque = nuevo_id; */
 317
 318                                 b_grabar_nodo(idx, nuevo_id+1, nodo);
 319                                 b_grabar_nodo(idx, nuevo_id, nuevo);
 320
 321                                 free(nodo);
 322                                 free(nuevo);
 323                                 free(tmp_claves);
 324
 325                                 hijo1 = nuevo_id+1;
 326                                 hijo2 = nuevo_id;
 327
 328                                 /* Limpio al padre */
 329                                 nuevo = b_leer_nodo(idx, 0);
 330
 331                                 b_leer_header(nuevo, &nuevo_header);
 332                                 nuevo_header.cant = 0;
 333                                 nuevo_header.padre = -1;
 334                                 nuevo_header.nivel = nodo_header.nivel+1;
 335                                 memset(nuevo, -1, idx->tam_bloque);
 336                                 b_actualizar_header(nuevo, &nuevo_header);
 337                                 b_grabar_nodo(idx, 0, nuevo);
 338
 339                                 nodo_id = 0;
 340                                 nodo = nuevo;
 341                                 rompi = 1;
 342                         }
 343                 } else {
 344                         /* La clave entra en este nodo!! */
 345                         i = 0;
 346                         if (nodo_header.cant > 0) {
 347                                 while ((emufs_indice_es_menor(idx, claves[i].clave, clave)) && (i < nodo_header.cant)) i++;
 348                                 for(j=nodo_header.cant; j > i; j--)
 349                                         claves[j] = claves[j-1];
 350                         }
 351                         nodo_header.cant++;
 352                         claves[i].clave = clave;
 353                         claves[i].dato = dato;
 354                         claves[i].hijo_derecho = hijo2;
 355                         nodo_header.hijo_izquierdo = b_elegir_izquierdo(idx, nodo_header.hijo_izquierdo, hijo1);
 356
 357                         b_actualizar_header(nodo, &nodo_header);
 358                         b_grabar_nodo(idx, nodo_id, nodo);
 359
 360                         /* Debo actualizar los punteros al padre de los hijos */
 361                         if (hijo1 != -1) {
 362                                 nuevo = b_leer_nodo(idx, hijo1);
 363                                 if (nuevo != NULL) {
 364                                         b_leer_header(nuevo, &nuevo_header);
 365                                         nuevo_header.padre = nodo_id;
 366                                         b_actualizar_header(nuevo, &nuevo_header);
 367                                         b_grabar_nodo(idx, hijo1, nuevo);
 368                                         free(nuevo);
 369                                 } else printf("FUCK! hijo1=%d no existe!\n", hijo1);
 370                         }
 371                         if (hijo2 != -1) {
 372                                 nuevo = b_leer_nodo(idx, hijo2);
 373                                 if (nuevo != NULL) {
 374                                         b_leer_header(nuevo, &nuevo_header);
 375                                         nuevo_header.padre = nodo_id;
 376                                         b_actualizar_header(nuevo, &nuevo_header);
 377                                         b_grabar_nodo(idx, hijo2, nuevo);
 378                                         free(nuevo);
 379                                 } else printf("FUCK! hijo2=%d no existe!\n", hijo2);
 380                         }
 381                         salir = 1;
 382                 }
 383         } while (!salir);
 384 }
 385
 386 const int b_elegir_izquierdo(INDICE *idx, int a, int b)
 387 {
 388         int cual;
 389         char *nodo1, *nodo2;
 390         B_NodoHeader header1, header2;
 391         B_NodoEntry *claves1, *claves2;
 392
 393         if (a==-1) return b;
 394         if (b==-1) return a;
 395
 396         nodo1 = b_leer_nodo(idx, a);
 397         nodo2 = b_leer_nodo(idx, b);
 398
 399         b_leer_header(nodo1, &header1);
 400         b_leer_header(nodo2, &header2);
 401
 402         claves1 = b_leer_claves(nodo1, &header1);
 403         claves2 = b_leer_claves(nodo2, &header2);
 404
 405         if (emufs_indice_es_menor(idx, claves1[0].clave, claves2[0].clave))
 406                 cual = a;
 407         else
 408                 cual = b;
 409
 410         free(nodo1);
 411         free(nodo2);
 412         return cual;
 413 }
 414
 415 INDICE_DATO *emufs_indice_b_buscar_muchos(INDICE *idx, CLAVE clave, int *cant)
 416 {
 417         /* Si el indice es primario no tiene sentido hacer nada */
 418         if (idx->funcion == IND_PRIMARIO) {
 419                 *cant = 0;
 420                 return NULL;
 421         }
 422
 423         /* TODO Implementar indices con repeticion */
 424         return NULL;
 425 }
 426