#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "main_win.h"
#include "guipixbufs.h"

#include "i18n.h"

MainWin::MainWin()
{
  set_title ("no file");
  set_default_size(WIN_WIDTH,WIN_HEIGHT);
  //add(vbox);
  add(table);

  // 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_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 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("ShowDesign", _("_Design")),
	   sigc::mem_fun(*this, &MainWin::createDesignWin));

  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) );
  ag->add( Gtk::Action::create("AlgoDijkstra", _("_Dijkstra")),
	   sigc::bind( sigc::mem_fun ( *this, &MainWin::createAlgoWin ), 2) );
  ag->add( Gtk::Action::create("AlgoSuurballe", _("_Suurballe")),
	   sigc::bind( sigc::mem_fun ( *this, &MainWin::createAlgoWin ), 3) );

  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::Action::create("AddMap", Gtk::StockID("gd-newmap")),
      sigc::mem_fun ( *this , &MainWin::createNewMapWin ) );

  ag->add( Gtk::Action::create("DesignGraph", Gtk::Stock::REFRESH),
      sigc::mem_fun ( *this , &MainWin::reDesignGraph ) );

  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'/>"
      "      <menuitem action='ShowDesign'/>"
      "    </menu>"
      "    <menu action='AlgoMenu'>"
      "      <menuitem action='AlgoGeneral'/>"
      "      <menuitem action='AlgoKruskal'/>"
      "      <menuitem action='AlgoDijkstra'/>"
      "      <menuitem action='AlgoSuurballe'/>"
      "    </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' />"
      "    <separator />"
      "    <toolitem action='AddMap' />"
      "    <toolitem action='DesignGraph' />"
      "  </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);
    table.attach(*menubar, 0, 1, 0, 1, Gtk::EXPAND|Gtk::FILL, Gtk::SHRINK);
  }

  Gtk::Widget* toolbar = uim->get_widget("/ToolBar");
  if (toolbar)
  {
    static_cast<Gtk::Toolbar*>(toolbar)->set_toolbar_style(Gtk::TOOLBAR_ICONS);
    //hbox.pack_start(*toolbar, Gtk::PACK_EXPAND_WIDGET);

    table.attach(*toolbar, 0, 1, 1, 2, Gtk::EXPAND|Gtk::FILL, Gtk::SHRINK);

  }
 
  table2.set_row_spacings(10);
  table2.set_col_spacings(5);

  auto_scale = new Gtk::CheckButton("Autoscale");
  auto_scale->set_active(false);
  auto_scale->signal_toggled().connect(sigc::mem_fun(*this, &MainWin::nodeViewChanged));
  table2.attach(*auto_scale, 0,2,0,1);

  Gtk::Label * width_label= new Gtk::Label("Edge Width:");
  table2.attach(*width_label, 0,1,1,2);
  
  Gtk::Adjustment * adjustment_width=new Gtk::Adjustment(20, 1, 200, 5, 10);
  
  edge_width = new Gtk::SpinButton(*adjustment_width, 5,0);
  edge_width->signal_value_changed().connect(sigc::mem_fun(*this, &MainWin::nodeViewChanged));
  table2.attach(*edge_width, 1,2,1,2);

  Gtk::Label * radius_label= new Gtk::Label("Node Radius:");
  table2.attach(*radius_label, 2,3,1,2);
  
  Gtk::Adjustment * adjustment_radius=new Gtk::Adjustment(20, 0, 500, 5, 10);

  radius_size = new Gtk::SpinButton(*adjustment_radius, 5,0);
  radius_size->signal_value_changed().connect(sigc::mem_fun(*this, &MainWin::nodeViewChanged));
  table2.attach(*radius_size, 3,4,1,2);

  zoom_track = new Gtk::CheckButton("Zoom tracking");
  zoom_track->set_active(false);
  zoom_track->signal_toggled().connect(sigc::mem_fun(*this, &MainWin::nodeViewChanged));
  table2.attach(*zoom_track, 2,4,0,1);


  table.attach(table2, 1, 2, 0, 2, Gtk::SHRINK, Gtk::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;

  table.attach(notebook,0,2,2,3, Gtk::EXPAND|Gtk::FILL, Gtk::EXPAND|Gtk::FILL);
  
  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]);
  bool autoscale;
  bool zoomtrack;
  double width;
  double radius;
  tabs[active_tab]->getView(autoscale, zoomtrack, width, radius);
  edge_width->set_value(width);
  radius_size->set_value(radius);
  zoom_track->set_active(zoomtrack);
  auto_scale->set_active(autoscale);

}

void MainWin::newFile()
{
  if(active_tab!=-1)
    {
      tabs[active_tab]->newFile();
    }
}
 
void MainWin::openFile()
{
  if(active_tab==-1)
    {
      newTab();
    }
  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::createDesignWin()
{
  if(active_tab!=-1)
    {
      tabs[active_tab]->createDesignWin(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));
}

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


void MainWin::nodeViewChanged()
{
  double width=edge_width->get_value();
  double radius=radius_size->get_value();
  bool zoomtrack=zoom_track->get_active();
  bool autoscale=auto_scale->get_active();
  tabs[active_tab]->setView(autoscale, zoomtrack, width, radius);
}

void MainWin::reDesignGraph()
{
  tabs[active_tab]->reDesignGraph();
}