#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();
}