X-Git-Url: https://git.llucax.com/z.facultad/75.52/treemulator.git/blobdiff_plain/1d533a2f726534cd02f1c7ba71f0fd454f70e16f..3981ffa23945f8764b5bc4c790ff0f0947858010:/src/btree_data.h

diff --git a/src/btree_data.h b/src/btree_data.h
index 141db7f..db8ecc1 100644
--- a/src/btree_data.h
+++ b/src/btree_data.h
@@ -10,23 +10,65 @@
 /** Dato a guardar en los nodos */
 class BTreeData {
 	public:
+		BTreeData () {}
 		BTreeData (uchar *node);
+		BTreeData (Clave *k, uint child);
 		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;
 };
 
 /** Dato a guardar en las hojas */
-class BTreeLeafData:BTreeData {
+class BTreeLeafData:public BTreeData {
 	public:
+		BTreeLeafData (Clave *k) { clave = k; }
+		BTreeLeafData (uchar *node) { clave = new ClaveFija (node); }
 		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