Agrego atributo de tipo de arbol.

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

diff --git a/src/main.cpp b/src/main.cpp
index c367a4c7e04c8da998b3e9361539aaeb8dda1e7e..069b3acd008a45fb0d16192768bc2a902be22550 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -2,23 +2,81 @@
 
 #include "btree.h"
 #include "clave_fija.h"
+#include "random.h"
+#include "keymanager.h"
 
 int main  (int argc, char *argv[])
 {
-       BTree tree ("test.idx", 64);
-       
-       if (argc != 2) {
-               printf ("Falta parametro cantidad de elementos a agregar\n");
+       int bloque, altas, bajas;
+
+       if (argc != 4) {
+               printf ("Uso : %s <block size> <cantidad de altas> <cantidad de bajas>\n", argv[0]);
                return 1;
        }
 
-       for (int i=0; i<=atoi(argv[1]); i++) {
-               ClaveFija c(i);
+       bloque = atoi (argv[1]);
+       altas = atoi (argv[2]);
+       bajas = atoi (argv[3]);
+
+       KeyManager<int> km;
+       BTree tree ("test.idx", bloque);
+       
+       std::list<int> lst;
+       std::list<int>::iterator it;
+       Random::Init ();
+       Random::Ints (lst, altas);
+
+       double paltas = bajas / (double)(altas + bajas);
+
+       it = lst.begin ();
+       while (it != lst.end ()) {
+               ClaveFija c(*it);
+
+               double l = Random::Double (0.0f, 1.0f);
+               std::cout << l << " >= " << paltas << std::endl;
+               if (!km.Size() // si no tenemos claves agregadas, mejor que hagamos un alta
+                               || l >= paltas) {
+                       tree.AddKey (c);
+                       km.AddValue (*it);
+                       it++;
+               } else {
+                       /* Tengo que borrar una clave entre 0 e "i" de la lista
+                        * porque son las que ya agregue. */
+                       ClaveFija c(km.GetRandom ());
+
+                       tree.DelKey (c);
+                       std::string sss = c;
+                       std::cout << "Clave Borrada " << sss << std::endl;
+               }
+       }
+
+       km.PrintInfo ();
+
+       std::list <int> l = km.GetList ();
+       it = l.begin ();
+       int bien = 0;
+       int mal = 0;
+       while (it != l.end ()) {
+               ClaveFija c(*it);
+               BTreeFindResult *r;
 
-               std::cout << "Agregando " << i << std::endl;
-               tree.AddKey (c);
+               r = tree.FindKey (c);
+               if (r != NULL) {
+                       std::cout << "** Clave " << (*it) << " encontrada.\n";
+                       bien++;
+                       delete r;
+               } else {
+                       std::cout << "** Clave " << (*it) << " NO encontrada.\n";
+                       mal++;
+               }
+               it++;
        }
 
+       std::cout << "\n\n === Resultados ===\n";
+       std::cout << "    Hist  : " << bien << "\n";
+       std::cout << "    Miss  : " << mal << "\n";
+       std::cout << "    Total : " << (bien+mal) << "\n";
+
        return 0;
 }