emufs/b_plus.c

   1 /** Arbol B+ */
   2 #include "b_plus.h"
   3
   4 /* Private prototypes */
   5 int b_plus_grabar_nodo(INDEXSPECS *idx, NODO_B_PLUS *nodo, int num_node);
   6 NODO_B_PLUS *b_plus_leer_nodo(INDEXSPECS *idx, int num_node);
   7 NODO_B_PLUS *b_plus_crearnodo(INDEXSPECS *idx);
   8
   9
  10 /** Crea un nuevo nodo y lo inicializa */
  11 NODO_B_PLUS *b_plus_crearnodo(INDEXSPECS *idx) {
  12
  13         NODO_B_PLUS *nodo = (NODO_B_PLUS*)malloc(sizeof(NODO_B_PLUS));
  14         nodo->nivel = 0;
  15         nodo->cant_claves = 0;
  16
  17     /* Calculamos lo que ocupan las cadenas de bytes claves + hijos */
  18         nodo->claves = (int*)malloc(idx->size_claves);
  19         nodo->hijos = (int*)malloc(idx->size_hijos);
  20         memset(nodo->claves,-1,idx->size_claves);
  21         memset(nodo->hijos,-1,idx->size_hijos);
  22
  23     return nodo;
  24 }
  25
  26 /** Crea el archivo indice B+ */
  27 int emufs_b_plus_crear(INDEXSPECS *idx) {
  28
  29         FILE *fp;
  30         NODO_B_PLUS *raiz;
  31         int error = 0;
  32
  33         /* Creamos el archivo que contendra el indice */
  34         fp = fopen(idx->filename, "w");
  35         PERR("Creando indice con nodo raiz");
  36         if (fp == NULL) {
  37                 PERR("Error al crear el archivo");
  38                 return -1;
  39         }
  40         fclose(fp);
  41
  42         /* Creamos el nodo raiz y lo guardamos el en indice */
  43         raiz = b_plus_crearnodo(idx);
  44         error = b_plus_grabar_nodo(idx,raiz,0);
  45
  46         /* Liberamos areas de memoria reservadas */
  47         free(raiz->claves);
  48         free(raiz->hijos);
  49         free(raiz);
  50
  51         return error;
  52 }
  53
  54 /** Busca el nro de bloque donde se debe guardar un reg con clave X */
  55 int emufs_b_plus_get_bloque(INDEXSPECS *idx, INDEX_DAT *query) {
  56
  57     NODO_B_PLUS *curnode;
  58
  59         /* Comienzo leyendo la raiz, entry point de toda funcion */
  60         printf ("Buscando donde insertar clave: %i\n\n",query->clave.i_clave);
  61         curnode = b_plus_leer_nodo(idx,0);
  62         if (curnode == NULL) return -1;
  63
  64         /* Mientras no encontre la hoja con la clave, busco.. */
  65         while ((curnode->nivel > 0) && curnode)
  66         {
  67
  68         }
  69
  70         free(curnode);
  71
  72         return 0;
  73 }
  74
  75 NODO_B_PLUS *b_plus_leer_nodo(INDEXSPECS *idx, int num_node) {
  76
  77         int i = 0;
  78         FILE *fp;
  79         NODO_B_PLUS *memnode = b_plus_crearnodo(idx);
  80         char *disknode = (char*)malloc(idx->tam_bloque);
  81
  82         if (disknode == NULL) return NULL;
  83         if (memnode == NULL) return NULL;
  84
  85     /* Open up file */
  86         fp = fopen(idx->filename, "r+");
  87         if (fp == NULL) {
  88                 free(disknode);
  89                 free(memnode);
  90                 return NULL;
  91         }
  92
  93         /* Intentamos leer un nodo, sino podemos error! */
  94         fseek(fp, num_node*idx->tam_bloque, SEEK_SET);
  95         if (fread(disknode, idx->tam_bloque, 1, fp) != 1) {
  96                 free(disknode);
  97                 fclose(fp);
  98                 return NULL;
  99         }
 100         fclose(fp);
 101
 102         /* Pudimos leer un nodo de disco, ahora lo transformamos a nodo mem */
 103         memcpy(memnode,disknode,SIZE_B_PLUS_HEADER);
 104         memcpy(memnode->claves,disknode+SIZE_B_PLUS_HEADER,idx->size_claves);
 105         memcpy(memnode->hijos,disknode+SIZE_B_PLUS_HEADER+idx->size_claves,idx->size_hijos);
 106         free(disknode);
 107
 108         printf("Dumping Node_%i\n",num_node);
 109         printf("Nivel: %i  Cant Claves: %i\n",memnode->nivel,memnode->cant_claves);
 110         printf("Claves:");
 111         for (i = 0; i < idx->size_claves/sizeof(int); ++i) printf(" %i",memnode->claves[i]);
 112         printf("\nHijos:");
 113         for (i = 0; i < idx->size_hijos/sizeof(int); ++i) printf(" %i",memnode->hijos[i]);
 114         printf("\nEnd Dump\n");
 115
 116         return memnode;
 117
 118 }
 119
 120 int b_plus_grabar_nodo(INDEXSPECS *idx, NODO_B_PLUS *nodo, int num_node)
 121 {
 122         FILE *fp;
 123
 124         fp = fopen(idx->filename, "r+");
 125         if (fp == NULL) return -1;
 126
 127         fseek(fp,num_node*sizeof(NODO_B_PLUS),SEEK_SET);
 128         fwrite(nodo,SIZE_B_PLUS_HEADER,1,fp);
 129         fwrite(nodo->claves,idx->size_claves,1,fp);
 130         fwrite(nodo->hijos,idx->size_hijos,1,fp);
 131         fclose(fp);
 132
 133         return 0;
 134 }