Agrego clave variable.

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

diff --git a/src/btree_data.h b/src/btree_data.h
index d920c9f208b58b064d0c856c360989c127a04bcb..ca2cd4dc2e0526fcd1d5adef1f9a52ee6e84d41c 100644 (file)
--- a/src/btree_data.h
+++ b/src/btree_data.h
@@ -6,16 +6,33 @@
 #include <stdlib.h>
 #include "clave.h"
 #include "clave_fija.h"
 #include <stdlib.h>
 #include "clave.h"
 #include "clave_fija.h"

+#include "clave_variable.h"

 
 /** Dato a guardar en los nodos */
 class BTreeData {
        public:
                BTreeData () {}
 
 /** Dato a guardar en los nodos */
 class BTreeData {
        public:
                BTreeData () {}

-               BTreeData (uchar *node);
+               BTreeData (uchar *node, int tree_type);
+               BTreeData (Clave *k, uint child);

                virtual ~BTreeData ();
 
                virtual ~BTreeData ();
 

-               virtual uint Size ();
-               virtual uchar *ToArray ();
+               virtual uint Size () const;
+               virtual uchar *ToArray () const;
+
+               Clave* getClave () { return clave; }
+               uint getChild () { return hijo; }
+               void setChild (uint c) { hijo = c; }
+
+               bool operator < (const BTreeData &data) const;
+               bool operator == (const BTreeData &data) const;
+               virtual operator std::string () const {
+                       std::string out = (*clave);
+                       std::stringstream ss;
+                       ss << "(" << out ;
+                       ss << ")" << "[" << hijo << "]";
+                       ss >> out;
+                       return out;
+               }

        protected:
                Clave *clave;
                uint hijo;
        protected:
                Clave *clave;
                uint hijo;


@@ -24,11 +41,35 @@ class BTreeData {
 /** Dato a guardar en las hojas */
 class BTreeLeafData:public BTreeData {
        public:
 /** Dato a guardar en las hojas */
 class BTreeLeafData:public BTreeData {
        public:

-               BTreeLeafData (Clave *k):BTreeData () { clave = k; }
+               BTreeLeafData (Clave *k) { clave = k; }
+               BTreeLeafData (uchar *node, int key_type);

                virtual ~BTreeLeafData ();
 
                virtual ~BTreeLeafData ();
 

-               virtual uint Size ();
-               virtual uchar *ToArray ();
+               virtual uint Size () const;
+               virtual uchar *ToArray () const;
+               virtual operator std::string () const {
+                       if (clave) {
+                               return (*clave);
+                       }
+                       return "NoKey";
+               }
+};
+
+class BTreeChildData:public BTreeData {
+       public:
+               BTreeChildData (uint child):BTreeData () { hijo = child; clave = NULL; }
+               BTreeChildData (uchar *node);
+               virtual ~BTreeChildData ();
+
+               virtual uint Size () const;
+               virtual uchar *ToArray () const;
+               virtual operator std::string () const {
+                       std::string out;
+                       std::stringstream ss;
+                       ss << "|" << hijo << "|";
+                       ss >> out;
+                       return out;
+               }

 };
 
 #endif
 };
 
 #endif