Agrego clave variable.

[z.facultad/75.52/treemulator.git] / src / btree_data.cpp

diff --git a/src/btree_data.cpp b/src/btree_data.cpp
index b95b86a2fd030f6fc1a5b1e27e327de0763fd510..a5d9b6805706311ca288279c5030f83f582adb22 100644 (file)
--- a/src/btree_data.cpp
+++ b/src/btree_data.cpp
@@ -1,12 +1,16 @@
 
 #include "btree_data.h"
+#include "btree.h"
 
-BTreeData::BTreeData (uchar *node)
+BTreeData::BTreeData (uchar *node, int tree_type)
 {
        /* TODO : Aca deberia detectar el tipo de clave (necesito
         * info desde el arbol
         */
-       clave = new ClaveFija (node);
+       if (tree_type == BTree::KEY_FIXED)
+               clave = new ClaveFija (node);
+       else
+               clave = new ClaveVariable (node);
        node += clave->Size ();
 
        memcpy (&hijo, node, sizeof (uint));
@@ -26,7 +30,7 @@ uint BTreeData::Size () const
 {
        uint s = sizeof (uint);
        if (clave) s += clave->Size ();
-
+       else std::cout << "No tengo clave\n";
        return s;
 }
 
@@ -43,9 +47,25 @@ uchar* BTreeData::ToArray () const
 
 bool BTreeData::operator < (const BTreeData &data) const
 {
+       if (!data.clave) return false;
+
        return (*clave) < (*(data.clave));
 }
 
+bool BTreeData::operator == (const BTreeData &data) const
+{
+       if (!data.clave) return false;
+
+       return (*clave) == (*(data.clave));
+}
+
+BTreeLeafData::BTreeLeafData (uchar *node, int key_type)
+{
+       if (key_type == BTree::KEY_FIXED)
+               clave = new ClaveFija (node);
+       else
+               clave = new ClaveVariable (node);
+}
 
 BTreeLeafData::~BTreeLeafData ()
 {
@@ -53,7 +73,6 @@ BTreeLeafData::~BTreeLeafData ()
 
 uint BTreeLeafData::Size () const
 {
-       std::cout << "BTreeLeafData::Size()" << std::endl;
        if (!clave) {
                std::cout << "BTreeLeafData::Size : No tengo clave!" << std::endl;
                return 0;
@@ -77,7 +96,6 @@ BTreeChildData::~BTreeChildData ()
 
 uint BTreeChildData::Size () const
 {
-       std::cout << "BTreeChildData::Size()" << std::endl;
        return sizeof (uint);
 }