#include "main_win.h"

#include "guipixbufs.h"

MainWin::MainWin()

{

  set_title ("no file");

  set_default_size(WIN_WIDTH,WIN_HEIGHT);

  add(vbox);

  // custom icons for the toolbar

  Glib::RefPtr<Gtk::IconFactory> p_icon_factory = Gtk::IconFactory::create();

  Glib::RefPtr<Gdk::Pixbuf> p_move_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_move);

  Glib::RefPtr<Gdk::Pixbuf> p_addnode_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_addnode);

  Glib::RefPtr<Gdk::Pixbuf> p_addlink_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_addlink);

  Glib::RefPtr<Gdk::Pixbuf> p_delete_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_delete);

  Glib::RefPtr<Gdk::Pixbuf> p_editlink_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_editlink);

  Glib::RefPtr<Gdk::Pixbuf> p_editnode_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_editnode);

  Glib::RefPtr<Gdk::Pixbuf> p_newmap_pixbuf = Gdk::Pixbuf::create_from_inline(

      2328, gui_icons_newmap);

  Gtk::IconSource move_icon_source;

  move_icon_source.set_pixbuf(p_move_pixbuf);

  Gtk::IconSet move_icon_set;

  move_icon_set.add_source(move_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-move"), move_icon_set);

  Gtk::IconSource addnode_icon_source;

  addnode_icon_source.set_pixbuf(p_addnode_pixbuf);

  Gtk::IconSet addnode_icon_set;

  addnode_icon_set.add_source(addnode_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-addnode"), addnode_icon_set);

  Gtk::IconSource addlink_icon_source;

  addlink_icon_source.set_pixbuf(p_addlink_pixbuf);

  Gtk::IconSet addlink_icon_set;

  addlink_icon_set.add_source(addlink_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-addlink"), addlink_icon_set);

  Gtk::IconSource delete_icon_source;

  delete_icon_source.set_pixbuf(p_delete_pixbuf);

  Gtk::IconSet delete_icon_set;

  delete_icon_set.add_source(delete_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-delete"), delete_icon_set);

  Gtk::IconSource editlink_icon_source;

  editlink_icon_source.set_pixbuf(p_editlink_pixbuf);

  Gtk::IconSet editlink_icon_set;

  editlink_icon_set.add_source(editlink_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-editlink"), editlink_icon_set);

  Gtk::IconSource editnode_icon_source;

  editnode_icon_source.set_pixbuf(p_editnode_pixbuf);

  Gtk::IconSet editnode_icon_set;

  editnode_icon_set.add_source(editnode_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-editnode"), editnode_icon_set);

  Gtk::IconSource newmap_icon_source;

  newmap_icon_source.set_pixbuf(p_newmap_pixbuf);

  Gtk::IconSet newmap_icon_set;

  newmap_icon_set.add_source(newmap_icon_source);

  p_icon_factory->add(Gtk::StockID("gd-newmap"), newmap_icon_set);

  p_icon_factory->add_default();

  ag=Gtk::ActionGroup::create();

  ag->add( Gtk::Action::create("FileMenu", "_File") );

  ag->add( Gtk::Action::create("FileNew", Gtk::Stock::NEW),

      sigc::mem_fun(*this, &MainWin::newTab));

  ag->add( Gtk::Action::create("FileOpen", Gtk::Stock::OPEN),

      sigc::mem_fun(*this, &MainWin::openFile));

  ag->add( Gtk::Action::create("FileClearTab", "Clear Tab"),

      sigc::mem_fun(*this, &MainWin::newFile));

  ag->add( Gtk::Action::create("FileSave", Gtk::Stock::SAVE),

      sigc::mem_fun(*this, &MainWin::saveFile));

  ag->add( Gtk::Action::create("FileSaveAs", Gtk::Stock::SAVE_AS),

      sigc::mem_fun(*this, &MainWin::saveFileAs));

  ag->add( Gtk::Action::create("Close", Gtk::Stock::CLOSE),

      sigc::mem_fun(*this, &MainWin::closeTab));

  ag->add( Gtk::Action::create("Quit", Gtk::Stock::QUIT),

      sigc::mem_fun(*this, &MainWin::hide));

  ag->add( Gtk::Action::create("ViewMenu", "_View") );

  ag->add( Gtk::Action::create("ViewZoomIn", Gtk::Stock::ZOOM_IN),

      sigc::mem_fun(*this, &MainWin::zoomIn));

  ag->add( Gtk::Action::create("ViewZoomOut", Gtk::Stock::ZOOM_OUT),

      sigc::mem_fun(*this, &MainWin::zoomOut));

  ag->add( Gtk::Action::create("ViewZoomFit", Gtk::Stock::ZOOM_FIT),

      sigc::mem_fun(*this, &MainWin::zoomFit));

  ag->add( Gtk::Action::create("ViewZoom100", Gtk::Stock::ZOOM_100),

      sigc::mem_fun(*this, &MainWin::zoom100));

  ag->add( Gtk::Action::create("ShowMenu", "_Show") );

  ag->add( Gtk::Action::create("ShowMaps", "_Maps"),

	   sigc::mem_fun(*this, &MainWin::createMapWin));

  ag->add( Gtk::Action::create("AlgoMenu", "_Algorithms") );

  ag->add( Gtk::Action::create("AlgoGeneral", "_General"),

	   sigc::bind( sigc::mem_fun ( *this, &MainWin::createAlgoWin ), 0) );

  ag->add( Gtk::Action::create("AlgoKruskal", "_Kruskal"),

	   sigc::bind( sigc::mem_fun ( *this, &MainWin::createAlgoWin ), 1) );

  Gtk::RadioAction::Group tool_group;

  ag->add( Gtk::RadioAction::create(tool_group, "MoveItem", Gtk::StockID("gd-move"), "Move"),

      sigc::bind( sigc::mem_fun ( *this, &MainWin::changeEditorialTool ), 0) );

  ag->add( Gtk::RadioAction::create(tool_group, "CreateNode", Gtk::StockID("gd-addnode"), "Create node"),

      sigc::bind( sigc::mem_fun ( *this, &MainWin::changeEditorialTool ), 1) );

  ag->add( Gtk::RadioAction::create(tool_group, "CreateEdge", Gtk::StockID("gd-addlink"), "Create edge"),

      sigc::bind( sigc::mem_fun ( *this, &MainWin::changeEditorialTool ), 2) );

  ag->add( Gtk::RadioAction::create(tool_group, "EraseItem", Gtk::StockID("gd-delete"), "Delete"),

      sigc::bind( sigc::mem_fun ( *this, &MainWin::changeEditorialTool ), 3) );

  ag->add( Gtk::RadioAction::create(tool_group, "EditEdgeMap", Gtk::StockID("gd-editlink"), "Edit edge map"),

      sigc::bind( sigc::mem_fun ( *this, &MainWin::changeEditorialTool ), 4) );

  ag->add( Gtk::RadioAction::create(tool_group, "EditNodeMap", Gtk::StockID("gd-editnode"), "Edit node map"),

      sigc::bind( sigc::mem_fun ( *this, &MainWin::changeEditorialTool ), 5) );

  ag->add( Gtk::Action::create("AddMap", Gtk::StockID("gd-newmap")),

      sigc::mem_fun ( *this , &MainWin::createNewMapWin ) );

  uim=Gtk::UIManager::create();

  uim->insert_action_group(ag);

  add_accel_group(uim->get_accel_group());

  try

  {

    Glib::ustring ui_info =

      "<ui>"

      "  <menubar name='MenuBar'>"

      "    <menu action='FileMenu'>"

      "      <menuitem action='FileNew'/>"

      "      <menuitem action='FileOpen'/>"

      "      <menuitem action='FileClearTab'/>"

      "      <menuitem action='FileSave'/>"

      "      <menuitem action='FileSaveAs'/>"

      "      <menuitem action='Close'/>"

      "      <menuitem action='Quit'/>"

      "    </menu>"

      "    <menu action='ViewMenu'>"

      "      <menuitem action='ViewZoomIn' />"

      "      <menuitem action='ViewZoomOut' />"

      "      <menuitem action='ViewZoom100' />"

      "      <menuitem action='ViewZoomFit' />"

      "    </menu>"

      "    <menu action='ShowMenu'>"

      "      <menuitem action='ShowMaps'/>"

      "    </menu>"

      "    <menu action='AlgoMenu'>"

      "      <menuitem action='AlgoGeneral'/>"

      "      <menuitem action='AlgoKruskal'/>"

      "    </menu>"

      "  </menubar>"

      "  <toolbar name='ToolBar'>"

      "    <toolitem action='FileNew' />"

      "    <toolitem action='FileOpen' />"

      "    <toolitem action='FileSave' />"

      "    <toolitem action='Close' />"

      "    <separator />"

      "    <toolitem action='ViewZoomIn' />"

      "    <toolitem action='ViewZoomOut' />"

      "    <toolitem action='ViewZoom100' />"

      "    <toolitem action='ViewZoomFit' />"

      "    <separator />"

      "    <toolitem action='MoveItem' />"

      "    <toolitem action='CreateNode' />"

      "    <toolitem action='CreateEdge' />"

      "    <toolitem action='EraseItem' />"

      "    <toolitem action='EditEdgeMap' />"

      "    <toolitem action='EditNodeMap' />"

      "    <separator />"

      "    <toolitem action='AddMap' />"

      "  </toolbar>"

      "</ui>";

    uim->add_ui_from_string(ui_info);

  }

  catch(const Glib::Error& ex)

  {

    std::cerr << "building menus failed: " <<  ex.what();

  }

  Gtk::Widget* menubar = uim->get_widget("/MenuBar");

  if (menubar){

    vbox.pack_start(*menubar, Gtk::PACK_SHRINK);

  }

  Gtk::Widget* toolbar = uim->get_widget("/ToolBar");

  if (toolbar)

  {

    static_cast<Gtk::Toolbar*>(toolbar)->set_toolbar_style(Gtk::TOOLBAR_ICONS);

    vbox.pack_start(*toolbar, Gtk::PACK_SHRINK);

  }

  tooltips.set_tip(*(uim->get_widget("/ToolBar/CreateNode")),"Create Node");

  tooltips.enable();

  active_tab=-1;

  notebook.signal_switch_page().connect(sigc::mem_fun(*this, &MainWin::onChangeTab));

  active_tool = MOVE;

  vbox.pack_start(notebook);

  show_all_children();

}

void MainWin::set_tabtitle(std::string name)

{

  if(strinst.find(name)==strinst.end())

    {

      tabnames[active_tab]=name;

      strinst[name]=1;

    }

  else

    {

      strinst[name]++;

      std::ostringstream o;

      o << strinst[name];

      tabnames[active_tab]=name+" - "+o.str();

    }

  set_title(tabnames[active_tab] + " - " + prog_name);

  notebook.set_tab_label_text((Widget&)*(tabs[active_tab]), tabnames[active_tab]);

  updateAlgoWinTabs();

}

void MainWin::readFile(const std::string & filename)

{

  newTab();

  tabs[active_tab]->readFile(filename);

}

void MainWin::newTab()

{

  int size=tabs.size();

  tabs.resize(size+1);

  tabnames.resize(size+1);

  active_tab=size;

  tabs[active_tab]=new NoteBookTab();

  tabs[active_tab]->signal_title_ch().connect(sigc::mem_fun(*this, &MainWin::set_tabtitle));

  tabs[active_tab]->signal_newmap_needed().connect(sigc::mem_fun(*this, &MainWin::createNewMapWinAfterSignal));

  tabs[active_tab]->gd_canvas->changeEditorialTool(active_tool);

  notebook.append_page((Gtk::Widget&)(*(tabs[active_tab])));

  notebook.set_current_page(size);

  set_tabtitle("unsaved file");

  updateAlgoWinTabs();

}

void MainWin::closeTab()

{

  if(active_tab!=-1)

    {

      if (tabs[active_tab]->mapstorage.modified)

	{

	  Gtk::MessageDialog mdialog("<b>Save changes before closing?</b>", true, 

				     Gtk::MESSAGE_WARNING, Gtk::BUTTONS_NONE);

	  mdialog.add_button("Close file _without Saving", Gtk::RESPONSE_REJECT);

	  mdialog.add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);

	  mdialog.add_button(Gtk::Stock::SAVE, Gtk::RESPONSE_ACCEPT);

	  switch (mdialog.run())

	    {

	    case Gtk::RESPONSE_CANCEL:

	      return;

	    case Gtk::RESPONSE_REJECT:

	      break;

	    case Gtk::RESPONSE_ACCEPT:

	      tabs[active_tab]->saveFile();

	      break;

	    }

	}

      //tabs vector will be decreased with the deleted value

      int size=tabs.size();

      if(size>1)

	{

	  for(int i=active_tab+1;i<size;i++)

	    {

	      tabnames[i-1]=tabnames[i];

	      tabs[i-1]=tabs[i];

	    }

	}

      //if size==1 resize will delete the only element

      tabs.resize(size-1);

      tabnames.resize(size-1);

      int old_active_tab=active_tab;

      notebook.remove_page(active_tab);

      //If the first tab was active, upon delete notebook

      //will first switch one tab upper and not lower like

      //in the case, when not the first tab was active.

      //But after deletion it will become the first tab,

      //and this should be registrated in tabs vector,

      //as well.

      if((old_active_tab==0)&&(size!=1))

	{

	  onChangeTab(NULL,0);

	}

      //if this was the last page in notebook, there is

      //no active_tab now

      if(size==1)

	{

	  active_tab=-1;

	}

      updateAlgoWinTabs();

    }

}

void MainWin::onChangeTab(GtkNotebookPage* page, guint page_num)

{

  page=page;

  active_tab=page_num;

  tabs[active_tab]->gd_canvas->changeEditorialTool(active_tool);

  set_title(tabnames[active_tab]);

}

void MainWin::onCloseTab()

{

}

void MainWin::newFile()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->newFile();

    }

}

void MainWin::openFile()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->openFile();

    }

}

void MainWin::saveFile()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->saveFile();

    }

}

void MainWin::saveFileAs()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->saveFileAs();

    }

}

void MainWin::close()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->close();

    }

}

void MainWin::zoomIn()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->gd_canvas->zoomIn();

    }

}

void MainWin::zoomOut()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->gd_canvas->zoomOut();

    }

}

void MainWin::zoomFit()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->gd_canvas->zoomFit();

    }

}

void MainWin::zoom100()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->gd_canvas->zoom100();

    }

}

void MainWin::createMapWin()

{

  if(active_tab!=-1)

    {

      tabs[active_tab]->createMapWin(tabnames[active_tab]);

    }

}

void MainWin::createAlgoWin(int algoid)

{

  AlgoWin * aw=new AlgoWin(algoid, tabnames);

  aw->signal_closing().connect(sigc::mem_fun(*this, &MainWin::deRegisterAlgoWin));

  aw->signal_maplist_needed().connect(sigc::mem_fun(*this, &MainWin::updateAlgoWinMaps));

  aw->signal_newmapwin_needed().connect(sigc::mem_fun(*this, &MainWin::createNewMapWinTabString));

  aws.insert(aw);

  aw->show();

}

void MainWin::updateAlgoWinTabs()

{

  std::set< AlgoWin* >::iterator awsi=aws.begin();

  for(;awsi!=aws.end();awsi++)

    {

      (*awsi)->update_tablist(tabnames);

    }

}

void MainWin::updateAlgoWinMaps(AlgoWin * awp, std::string tabname)

{

  int i=0;

  for(;(i<(int)tabnames.size())&&(tabnames[i]!=tabname);i++)

    {

    }

  //awp->update_maplist(tabs[i]->mapstorage.getNodeMapList(),tabs[i]->mapstorage.getEdgeMapList());

  awp->update_maplist(&(tabs[i]->mapstorage));

}

void MainWin::deRegisterAlgoWin(AlgoWin * awp)

{

  aws.erase(awp);

}

void MainWin::changeEditorialTool(int tool)

{

  active_tool=tool;

  if(active_tab!=-1)

    {

      tabs[active_tab]->gd_canvas->changeEditorialTool(tool);

    }

}

void MainWin::createNewMapWin()

{

  if(active_tab!=-1)

    {

      NewMapWin * nmw=new NewMapWin("Create New Map - "+tabnames[active_tab], *(tabs[active_tab]));

      nmw->show();

    }

}

void MainWin::createNewMapWinTabString(std::string tabname, bool itisedge)

{

  int i=0;

  for(;((i<tabnames.size())&&(tabnames[i]!=tabname));i++)

    {

    }

  createNewMapWinAfterSignal(tabs[i], itisedge);

}

void MainWin::createNewMapWinAfterSignal(NoteBookTab * nbt, bool itisedge)

{

  std::vector<NoteBookTab*>::iterator nbti=tabs.begin();

  int i=0;

  for(;nbti!=tabs.end();nbti++)

    {

      if(*nbti!=nbt)

	{

	  i++;

	}

      else

	{

	  continue;

	}

    }

  NewMapWin * nmw=new NewMapWin("Create New Map - "+tabnames[i], *nbt, itisedge, false);

  nmw->run();

}