#include <gtkmm.h>
#include <libgnomecanvasmm.h>
+#include "random.h"
#include "view_btree.h"
#include "view_properties.h"
+#include "new_tree_dialog.h"
+#include "key_dialog.h"
+#include "view_debug.h"
using namespace Gnome::Canvas;
+ Glib::ustring ui_info =
+"<ui>"
+" <menubar name='MenuBar'>"
+" <menu action='MenuFile'>"
+" <menuitem action='Nuevo'/>"
+" <menuitem action='Abrir'/>"
+" <separator/>"
+" <menuitem action='Salir'/>"
+" </menu>"
+" <menu action='MenuNode'>"
+" <menuitem action='Ir al Padre'/>"
+" </menu>"
+" <menu action='MenuKey'>"
+" <menuitem action='Agregar Clave'/>"
+" <menuitem action='Borrar Clave'/>"
+" <menuitem action='Buscar Clave'/>"
+" </menu>"
+" <menu action='MenuZoom'>"
+" <menuitem action='Alejar'/>"
+" <menuitem action='Acercar'/>"
+" <separator/>"
+" <menuitem action='100 %'/>"
+" </menu>"
+" </menubar>"
+"</ui>";
+
+void nuevo_arbol ();
+void abrir_arbol ();
+void agregar_clave ();
+void borrar_clave ();
+void ir_al_padre ();
+void buscar_clave ();
+void zoom_out ();
+void zoom_in ();
+void zoom_normal ();
+
+Glib::RefPtr<ViewBTree> tree;
+Gnome::Canvas::Canvas *real_canvas;
+ViewProperties *real_frame;
+
int main(int argc, char *argv[])
{
Gtk::Main kit(argc, argv);
Gnome::Canvas::init ();
Gtk::Window window;
- Gtk::HBox hbox;
- ViewProperties frame;
+ Gtk::VBox hbox;
+ Gtk::VBox vbox;
+
Gtk::ScrolledWindow area;
Gnome::Canvas::Canvas canvas;
+ ViewProperties frame;
+
+ real_canvas = &canvas;
+ real_frame = &frame;
- ViewBTree canvas_grp (canvas.root (), "test.idx");
- canvas.set_scroll_region (0, 0, 5000, 5000);
+ canvas.set_scroll_region (0, 0, 100, 100);
area.add (canvas);
- hbox.pack_start (frame, false, false, 10);
hbox.pack_start (area);
+ hbox.pack_start (frame, false, false, 10);
+ frame.set_size_request (200, 200);
+
+ Glib::RefPtr<Gtk::ActionGroup> actiongroup = Gtk::ActionGroup::create();
+
+ actiongroup->add( Gtk::Action::create("MenuFile", "_Arbol") );
+ actiongroup->add( Gtk::Action::create("Nuevo", Gtk::Stock::NEW), &nuevo_arbol);
+ actiongroup->add( Gtk::Action::create("Abrir", Gtk::Stock::OPEN), &abrir_arbol);
+ actiongroup->add( Gtk::Action::create("Salir", Gtk::Stock::QUIT), Gtk::AccelKey ("<control>q"), &Gtk::Main::quit);
+ actiongroup->add( Gtk::Action::create("MenuNode", "_Nodo") );
+ actiongroup->add( Gtk::Action::create("Ir al Padre", Gtk::Stock::GO_BACK), Gtk::AccelKey ("<control>b"), &ir_al_padre);
+ actiongroup->add( Gtk::Action::create("MenuKey", "_Clave") );
+ actiongroup->add( Gtk::Action::create("Agregar Clave", Gtk::Stock::ADD), Gtk::AccelKey ("<control>a"), &agregar_clave);
+ actiongroup->add( Gtk::Action::create("Borrar Clave", Gtk::Stock::REMOVE), Gtk::AccelKey ("<control>d"), &borrar_clave);
+ actiongroup->add( Gtk::Action::create("Buscar Clave", Gtk::Stock::FIND), Gtk::AccelKey ("<control>f"), &buscar_clave);
+ actiongroup->add( Gtk::Action::create("MenuZoom", "_Zoom"));
+ actiongroup->add( Gtk::Action::create("Alejar", Gtk::Stock::ZOOM_OUT), Gtk::AccelKey ("<control>z"), &zoom_out );
+ actiongroup->add( Gtk::Action::create("Acercar", Gtk::Stock::ZOOM_IN), Gtk::AccelKey ("<control>x"), &zoom_in);
+ actiongroup->add( Gtk::Action::create("100 %", Gtk::Stock::ZOOM_100), Gtk::AccelKey ("<control>1"), &zoom_normal);
+
+ Glib::RefPtr<Gtk::UIManager> m_refUIManager = Gtk::UIManager::create();
+ m_refUIManager->insert_action_group (actiongroup);
+
+ m_refUIManager->add_ui_from_string(ui_info);
+ Gtk::Widget* menubar = m_refUIManager->get_widget("/MenuBar");
+ menubar->show_all ();
+
+ vbox.pack_start (*menubar, false, true, 0);
+ vbox.pack_end (hbox, true, true, 5);
- window.add (hbox);
+ window.add_accel_group (m_refUIManager->get_accel_group ());
+ window.add (vbox);
window.set_size_request (640, 480);
window.show_all ();
/* Conecto el Canvas con el Frame */
- canvas_grp.signal_selected ().connect ( sigc::mem_fun (frame, &ViewProperties::ShowItem) );
Gtk::Main::run(window);
return 0;
}
+
+void abrir_arbol ()
+{
+ Gtk::FileChooserDialog dlg ("Abrir Arbol");
+
+ dlg.add_button (Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
+ dlg.add_button (Gtk::Stock::OPEN, Gtk::RESPONSE_ACCEPT);
+ if (dlg.run () == Gtk::RESPONSE_ACCEPT) {
+ Glib::ustring s = dlg.get_filename ();
+ std::string filename = s;
+ ViewBTree *ptree = ViewBTree::Open (real_canvas->root (), filename);
+ if (ptree != NULL) {
+ tree = Glib::RefPtr<ViewBTree>(ptree);
+ tree->signal_selected ().connect ( sigc::mem_fun (*real_frame, &ViewProperties::ShowItem) );
+ tree->AddNode (0);
+ double x1, x2, y1, y2;
+ tree->get_bounds (x1, y1, x2, y2);
+ real_canvas->set_scroll_region (0, 0, 5000, 5000);
+ real_canvas->scroll_to (0, 0);
+ }
+ }
+}
+
+void nuevo_arbol ()
+{
+ real_canvas->set_scroll_region (0, 0, 5000, 5000);
+
+ NewTreeDialog d;
+ if (d.run () == Gtk::RESPONSE_OK) {
+ uint altas = d.getAdds ();
+ uint bajas = d.getDels ();
+
+ double paltas = bajas / (double)altas;
+
+ int type = d.getKeyType ();
+ int atype = d.getTreeType ();
+ std::string name = d.getName ();
+ tree = Glib::RefPtr<ViewBTree>(new ViewBTree (real_canvas->root(), name+".idx", d.getBlockSize (), atype, type));
+ real_frame->SetTree (tree);
+ tree->signal_selected ().connect ( sigc::mem_fun (*real_frame, &ViewProperties::ShowItem) );
+ if (type == BTree::KEY_FIXED) {
+ std::list<int> lst;
+ std::list<int>::iterator it;
+ Random::Init ();
+ Random::Ints (lst, altas);
+
+ it = lst.begin ();
+ uint i = 0;
+ while (it != lst.end ()) {
+ ClaveFija c(*it);
+
+ double l = Random::Double (0.0f, 1.0f);
+ std::cout << l << " >= " << paltas << std::endl;
+ if (l >= paltas) {
+ try {
+ tree->AddKey (c);
+ } catch (Exception *e) {
+ std::cout << "====== " << (std::string)c << e->Message () << std::endl;
+ }
+ i++;
+ } else {
+ /* Tengo que borrar una clave entre 0 e "i" de la lista
+ * porque son las que ya agregue. */
+ int aborrar = (int)Random::Double (0, i);
+ std::list<int>::iterator otro = lst.begin ();
+ int j = 0;
+ while (j < aborrar) {
+ otro++;
+ j++;
+ }
+ ClaveFija c(*otro);
+
+ tree->DelKey (c);
+ std::string sss = c;
+ std::cout << "Clave Borrada " << sss << std::endl;
+ }
+
+ it++;
+ }
+ } else {
+ std::list<std::string> lst;
+ std::list<std::string>::iterator it;
+ Random::Init ();
+ Random::Strings (lst, altas);
+
+ it = lst.begin ();
+ while (it != lst.end ()) {
+ ClaveVariable c(*it);
+
+ try {
+ tree->AddKey (c);
+ } catch (Exception *e) {
+ std::cout << "====== " << (std::string)c << e->Message () << std::endl;
+ }
+ it++;
+ }
+ }
+ tree->AddNode (0);
+ double x1, x2, y1, y2;
+ tree->get_bounds (x1, y1, x2, y2);
+ real_canvas->scroll_to (0, 0);
+ }
+}
+
+void agregar_clave ()
+{
+ if (!tree)
+ {
+ Gtk::MessageDialog d("No hay un arbol creado, por favor primero cree un arbol!",
+ false, Gtk::MESSAGE_ERROR, Gtk::BUTTONS_OK, true);
+ d.run();
+ return;
+ }
+ KeyDialog d("Agregar");
+ if (d.run () == Gtk::RESPONSE_OK)
+ {
+ Glib::ustring str_key = d.key();
+ Glib::ustring str_val = d.val();
+ if (tree->type() == BTree::KEY_FIXED)
+ {
+ ClaveFija c(atoi(str_key.c_str()));
+ tree->AddKey(c);
+ }
+ else
+ {
+ ClaveVariable c(str_key);
+ tree->AddKey(c);
+ }
+
+ /* Muestro la clave agregada */
+ BTreeFindResult* result = 0;
+ Clave *c = NULL;
+ if (tree->type() == BTree::KEY_FIXED) {
+ c = new ClaveFija (atoi(str_key.c_str()));
+ result = tree->FindKey(*c);
+ } else {
+ c = new ClaveVariable (str_key);
+ result = tree->FindKey(*c);
+ }
+ if (result) {
+ tree->Clear ();
+ tree->AddNode(result->node);
+ tree->HighliteKey (*c);
+ delete result;
+ } else {
+ delete tree->last_selected;
+ tree->last_selected = NULL;
+ tree->AddNode (0);
+ }
+ if (c) delete c;
+ real_canvas->scroll_to (0, 0);
+ }
+}
+
+void borrar_clave ()
+{
+ if (!tree)
+ {
+ Gtk::MessageDialog d("No hay un arbol creado, por favor primero cree un arbol!",
+ false, Gtk::MESSAGE_ERROR, Gtk::BUTTONS_OK, true);
+ d.run();
+ return;
+ }
+ KeyDialog d("Borrar");
+ if (d.run () == Gtk::RESPONSE_OK)
+ {
+ Glib::ustring str_key = d.key();
+ if (tree->type() == BTree::KEY_FIXED)
+ {
+ ClaveFija c(atoi(str_key.c_str()));
+ tree->DelKey(c);
+ }
+ else
+ {
+ ClaveVariable c(str_key);
+ tree->DelKey(c);
+ }
+ delete tree->last_selected;
+ tree->AddNode (0);
+ }
+}
+
+void ir_al_padre ()
+{
+ if (!tree)
+ {
+ Gtk::MessageDialog d("No hay un arbol creado, por favor primero cree un arbol!",
+ false, Gtk::MESSAGE_ERROR, Gtk::BUTTONS_OK, true);
+ d.run();
+ return;
+ }
+ tree->GoBack ();
+}
+
+void buscar_clave ()
+{
+ if (!tree)
+ {
+ Gtk::MessageDialog d("No hay un arbol creado, por favor primero cree un arbol!",
+ false, Gtk::MESSAGE_ERROR, Gtk::BUTTONS_OK, true);
+ d.run();
+ return;
+ }
+ KeyDialog d("Buscar");
+ while (true) // Repite hasta que se encuentre algo o se cancele
+ {
+ if (d.run () == Gtk::RESPONSE_OK)
+ {
+ BTreeFindResult* result = 0;
+ Clave *c = NULL;
+ Glib::ustring str_key = d.key();
+ if (tree->type() == BTree::KEY_FIXED)
+ {
+ c = new ClaveFija (atoi(str_key.c_str()));
+ result = tree->FindKey(*c);
+ }
+ else
+ {
+ c = new ClaveVariable (str_key);
+ result = tree->FindKey(*c);
+ }
+ if (result)
+ {
+ tree->Clear ();
+ tree->AddNode(result->node);
+ tree->HighliteKey (*c);
+ delete result;
+ delete c;
+ real_canvas->scroll_to (0, 0);
+ return; // Encontramos, salimos
+ }
+ if (c) delete c;
+ Gtk::MessageDialog msg("Clave no encontrada!", false,
+ Gtk::MESSAGE_ERROR, Gtk::BUTTONS_OK, true);
+ msg.run();
+ // Seguimos intentando
+ }
+ else return; // Cancelaron, salimos
+ }
+}
+
+void zoom_out ()
+{
+ double r = real_canvas->get_pixels_per_unit ();
+ r *= 0.9f;
+ if (r < 0.0001)
+ r = 0.0001;
+ real_canvas->set_pixels_per_unit (r);
+}
+
+void zoom_in ()
+{
+ double r = real_canvas->get_pixels_per_unit ();
+ r *= 1.1f;
+ if (r > 10)
+ r = 10;
+ real_canvas->set_pixels_per_unit (r);
+}
+
+void zoom_normal ()
+{
+ real_canvas->set_pixels_per_unit (1.0f);
+}
+