From c6ea89e17981bce73a1cf639557d15cb624553fb Mon Sep 17 00:00:00 2001
From: Abs62 <ottomann@yandex.ru>
Date: Fri, 21 Dec 2012 15:48:41 +0400
Subject: [PATCH] Auto created groups combining

---
 groups_widgets.cc | 175 ++++++++++++++++++++++++++++++++++++++++++----
 groups_widgets.hh |   7 ++
 2 files changed, 168 insertions(+), 14 deletions(-)

diff --git a/groups_widgets.cc b/groups_widgets.cc
index ef8bd8a7..ac535fa0 100644
--- a/groups_widgets.cc
+++ b/groups_widgets.cc
@@ -520,6 +520,9 @@ DictGroupsWidget::DictGroupsWidget( QWidget * parent ):
 #if QT_VERSION >= 0x040500
   setMovable( true );
 #endif
+  setContextMenuPolicy( Qt::CustomContextMenu );
+  connect( this, SIGNAL( customContextMenuRequested( QPoint ) ),
+           this, SLOT( contextMenu( QPoint ) ) );
 }
 
 namespace {
@@ -627,6 +630,19 @@ void DictGroupsWidget::addNewGroup( QString const & name )
   setUsesScrollButtons( count() > 3 );
 }
 
+int DictGroupsWidget::addUniqueGroup( const QString & name )
+{
+  for( int n = 0; n < count(); n++ )
+    if( tabText( n ) == name )
+    {
+      setCurrentIndex( n );
+      return n;
+    }
+
+  addNewGroup( name );
+  return currentIndex();
+}
+
 void DictGroupsWidget::addAutoGroups()
 {
   if( !activeDicts )
@@ -707,21 +723,10 @@ void DictGroupsWidget::addAutoGroups()
 
   for( QStringList::ConstIterator gr = groupList.begin(); gr != groupList.end(); ++gr )
   {
-    int n;
-    for( n = 0; n < count(); n++ )
-      if( *gr == tabText( n ) )
-      {
-        setCurrentIndex( n );
-        break;
-      }
+    if( count() )
+      setCurrentIndex( count() - 1 );
 
-    if( n >= count() )
-    {
-      // group not found - add it
-      if( count() )
-        setCurrentIndex( count() - 1 );
-      addNewGroup( *gr );
-    }
+    addUniqueGroup( *gr );
 
     // add dictionaries into the current group
     QVector< sptr<Dictionary::Class> > vd = dictMap[ *gr ];
@@ -772,3 +777,145 @@ void DictGroupsWidget::removeAllGroups()
     delete w;
   }
 }
+
+void DictGroupsWidget::contextMenu( QPoint const & pos )
+{
+  int clickedGroup = tabBar()->tabAt( pos );
+  if( clickedGroup < 0 )
+    return;
+  QString name = tabText( clickedGroup );
+  if( name.length() != 7 || name.mid( 2, 3 ) != " - " )
+    return;
+
+  QMenu menu( this );
+
+  QAction *combineSourceAction = new QAction( QString( tr( "Combine groups by source language to \"%1->\"" ) )
+                                              .arg( name.left( 2 ) ), &menu );
+  combineSourceAction->setEnabled( false );
+  for( int i = 0; i < count(); i++ )
+  {
+    QString str = tabText( i );
+    if( i != clickedGroup && str.length() == 7 && str.mid( 2, 3 ) == " - " && str.startsWith( name.left( 2 ) ) )
+    {
+      combineSourceAction->setEnabled( true );
+      break;
+    }
+  }
+  menu.addAction( combineSourceAction );
+
+  QAction *combineTargetAction = new QAction( QString( tr( "Combine groups by target language to \"->%1\"" ) )
+                                              .arg( name.right( 2 ) ), &menu );
+  combineTargetAction->setEnabled( false );
+  for( int i = 0; i < count(); i++ )
+  {
+    QString str = tabText( i );
+    if( i != clickedGroup && str.length() == 7 && str.mid( 2, 3 ) == " - " && str.endsWith( name.right( 2 ) ) )
+    {
+      combineTargetAction->setEnabled( true );
+      break;
+    }
+  }
+  menu.addAction( combineTargetAction );
+
+  QAction *combineTwoSidedAction = NULL;
+  int secondGroup = -1;
+  if( name.left( 2 ) != name.right( 2 ) )
+  {
+     combineTwoSidedAction = new QAction( QString( tr( "Make two-side translate group \"%1-%2-%1\"" ) )
+                                          .arg( name.left( 2 ) ).arg( name.right( 2 ) ), &menu );
+
+    combineTwoSidedAction->setEnabled( false );
+
+    QString str = name.right( 2 ) + " - " + name.left( 2 );
+    for( int i = 0; i < count(); i++ )
+    {
+      if( str == tabText( i ) )
+      {
+        secondGroup = i;
+        combineTwoSidedAction->setEnabled( true );
+        break;
+      }
+    }
+
+    menu.addAction( combineTwoSidedAction );
+  }
+
+  QAction *result = menu.exec( mapToGlobal( pos ) );
+
+  setUpdatesEnabled( false );
+  int targetGroup;
+
+  if( result && result == combineSourceAction )
+  {
+    QString grBase = name.left( 2 );
+    setCurrentIndex( clickedGroup );
+    targetGroup = addUniqueGroup( grBase + "->" );
+
+    for( int i = 0; i < count(); i++ )
+    {
+      QString str = tabText( i );
+      if( str.length() == 7 && str.mid( 2, 3 ) == " - " && str.startsWith( grBase ) )
+      {
+        setCurrentIndex( i );
+        vector< sptr< Dictionary::Class > > const & dicts = getCurrentModel()->getCurrentDictionaries();
+
+        setCurrentIndex( targetGroup );
+        DictListModel *model = getCurrentModel();
+
+        for( unsigned j = 0; j < dicts.size(); j++ )
+          model->addRow( QModelIndex(), dicts[ j ] );
+      }
+    }
+    setCurrentIndex( targetGroup );
+  }
+  else
+  if( result && result == combineTargetAction )
+  {
+    QString grBase = name.right( 2 );
+
+    setCurrentIndex( clickedGroup );
+    targetGroup = addUniqueGroup( "->" + grBase );
+
+    for( int i = 0; i < count(); i++ )
+    {
+      QString str = tabText( i );
+      if( str.length() == 7 && str.mid( 2, 3 ) == " - " && str.endsWith( grBase ) )
+      {
+        setCurrentIndex( i );
+        vector< sptr< Dictionary::Class > > const & dicts = getCurrentModel()->getCurrentDictionaries();
+
+        setCurrentIndex( targetGroup );
+        DictListModel *model = getCurrentModel();
+
+        for( unsigned j = 0; j < dicts.size(); j++ )
+          model->addRow( QModelIndex(), dicts[ j ] );
+      }
+    }
+    setCurrentIndex( targetGroup );
+  }
+  else
+  if( result && result == combineTwoSidedAction )
+  {
+    setCurrentIndex( clickedGroup );
+    targetGroup = addUniqueGroup( name + " - " + name.left( 2 ) );
+    QString str = name.right( 2 ) + " - " + name.left( 2 );
+
+    for( int i = 0; i < count(); i++ )
+    {
+      if( tabText( i ) == name || tabText( i ) == str )
+      {
+        setCurrentIndex( i );
+        vector< sptr< Dictionary::Class > > const & dicts = getCurrentModel()->getCurrentDictionaries();
+
+        setCurrentIndex( targetGroup );
+        DictListModel *model = getCurrentModel();
+
+        for( unsigned j = 0; j < dicts.size(); j++ )
+          model->addRow( QModelIndex(), dicts[ j ] );
+      }
+    }
+    setCurrentIndex( targetGroup );
+  }
+
+  setUpdatesEnabled( true );
+}
diff --git a/groups_widgets.hh b/groups_widgets.hh
index e4741cd1..9b1e4355 100644
--- a/groups_widgets.hh
+++ b/groups_widgets.hh
@@ -146,6 +146,10 @@ public:
   /// Creates new empty group with the given name
   void addNewGroup( QString const & );
 
+  /// Creates new empty group with the given name if no such group
+  /// and return it index
+  int addUniqueGroup( QString const & name );
+
   void addAutoGroups();
 
   /// Returns currently chosen group's name
@@ -173,6 +177,9 @@ private:
   std::vector< sptr< Dictionary::Class > > const * allDicts;
   std::vector< sptr< Dictionary::Class > > const * activeDicts;
 
+private slots:
+  void contextMenu( QPoint const & );
+
 signals:
   void showDictionaryInfo( QString const & id );
 };