src/btree.cpp

   1
   2 #include "btree.h"
   3
   4 BTree::BTree (const std::string &name, unsigned int block_size, bool create_new_file)
   5 {
   6         uchar *node;
   7         BTreeNodeHeader nh;
   8
   9         fp = fopen (name.c_str(), "wb");
  10         if (!fp) {
  11                 /* TODO : mandar una exception ? */
  12                 return;
  13         }
  14
  15         /* Nombre de archivo */
  16         filename = name;
  17
  18         /* Inicializo el header */
  19         header.block_size = block_size;
  20         WriteFileHeader ();
  21
  22         /* Creo el primer bloque vacio */
  23         node = new uchar[block_size];
  24         ReadNodoHeader (node, &nh);
  25         nh.level = 0;
  26         nh.free_space = block_size - sizeof (BTreeNodeHeader);
  27         nh.item_count = 0;
  28         WriteNodoHeader (node, &nh);
  29         WriteBlock (node, 0);
  30
  31         delete [] node;
  32 }
  33
  34 BTree::~BTree ()
  35 {
  36         fclose (fp);
  37 }
  38
  39 void BTree::WriteFileHeader ()
  40 {
  41         fseek (fp, 0L, SEEK_SET);
  42         fwrite (&header, 1, sizeof (BTreeFileHeader), fp);
  43 }
  44
  45 void BTree::WriteBlock (uchar *block, uint num)
  46 {
  47         fseek (fp, num*header.block_size + sizeof (BTreeFileHeader), SEEK_SET);
  48         fwrite (block, 1, header.block_size, fp);
  49 }
  50
  51 void BTree::AddKey (const Clave &k)
  52 {
  53         std::list<Clave *> node_keys;
  54
  55         /* Leo el nodo raiz para empezar a agregar */
  56         uchar *root = ReadBlock (0);
  57         BTreeNodeHeader root_header;
  58         ReadNodoHeader (root, &root_header);
  59
  60         node_keys = ReadKeys (root, root_header);
  61         std::cout << node_keys.size () << std::endl;
  62
  63         if (root_header.free_space >= k.Size ()) {
  64                 std::cout << "Hay lugar para meter la clave" << std::endl;
  65                 /* TODO : Insertar ordenado */
  66                 node_keys.push_back (k.Clone ());
  67         } else {
  68                 std::cout << "__NO__ hay lugar para meter la clave" << std::endl;
  69         }
  70
  71
  72         WriteKeys (root, root_header, node_keys);
  73         WriteNodoHeader (root, &root_header);
  74         WriteBlock (root, 0);
  75
  76         delete [] root;
  77 }
  78
  79 void BTree::DelKey (const Clave &k) {}
  80
  81 void BTree::ReadNodoHeader (uchar *node, BTreeNodeHeader *header)
  82 {
  83         memcpy (header, node, sizeof (BTreeNodeHeader));
  84 }
  85
  86 void BTree::WriteNodoHeader (uchar *node, BTreeNodeHeader *header)
  87 {
  88         memcpy (node, header, sizeof (BTreeNodeHeader));
  89 }
  90
  91 uchar *BTree::ReadBlock (uint num)
  92 {
  93         uchar *out = new uchar[header.block_size];
  94
  95         fseek (fp, num*header.block_size + sizeof (BTreeFileHeader), SEEK_SET);
  96         fread (out, 1, header.block_size, fp);
  97
  98         return out;
  99 }
 100
 101 std::list<Clave *> BTree::ReadKeys (uchar *node, BTreeNodeHeader &node_header)
 102 {
 103         std::list<Clave *> keys;
 104         node += sizeof (BTreeNodeHeader);
 105
 106         for (uint i=0; i<node_header.item_count; i++) {
 107                 /* TODO : El tipo de clave deberia ser usado
 108                  * dependiendo de algun dato en el header del
 109                  * arbol
 110                  */
 111                 ClaveFija *k = new ClaveFija (node);
 112                 node += k->Size ();
 113                 keys.push_back (k);
 114         }
 115
 116         return keys;
 117 }
 118
 119 void BTree::WriteKeys (uchar *node, BTreeNodeHeader &node_header, std::list<Clave *> &keys)
 120 {
 121         std::list<Clave *>::iterator it = keys.begin ();
 122
 123         node += sizeof (BTreeNodeHeader);
 124
 125         while (it != keys.end ()) {
 126                 Clave *c = (*it);
 127                 memcpy (node, c->ToArray(), c->Size ());
 128                 node += c->Size ();
 129                 it++;
 130         }
 131
 132         /* TODO : incrementar node_header.item_count aca o fuera de este metodo? */
 133 }