runtime/kcontrol/locale/countryselectordialog.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Albert Astals Cid <aacid@kde.org>               *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  ***************************************************************************/
   9
  10 #include "countryselectordialog.h"
  11
  12 #include "kcontrollocale.h"
  13
  14 #include <KStandardDirs>
  15
  16 #include <QAbstractItemModel>
  17 #include <QHBoxLayout>
  18 #include <QKeyEvent>
  19 #include <QListView>
  20 #include <QScrollBar>
  21
  22 struct CountryModelItem
  23 {
  24     CountryModelItem()
  25     {
  26     }
  27
  28     CountryModelItem( CountryModelItem *p, const QString &theText, const QString &theTag )
  29      : parent(p), text(theText), tag(theTag)
  30     {
  31     }
  32
  33     ~CountryModelItem()
  34     {
  35         qDeleteAll(children);
  36     }
  37
  38     int row() const
  39     {
  40         if (parent) return parent->children.indexOf(const_cast<CountryModelItem*>(this));
  41         return 0;
  42     }
  43
  44     CountryModelItem *parent;
  45     QList< CountryModelItem* > children;
  46
  47     QString text, tag;
  48     KIcon icon;
  49 };
  50
  51
  52
  53 bool CountryModelItemLessThan(CountryModelItem *s1, CountryModelItem *s2)
  54 {
  55     return s1->text < s2->text;
  56 }
  57
  58
  59
  60 class CountryModel : public QAbstractItemModel
  61 {
  62     public:
  63         CountryModel(QObject *parent) : QAbstractItemModel(parent)
  64         {
  65             m_rootItem = new CountryModelItem(NULL, QString(), QString());
  66         }
  67
  68         CountryModel(CountryModelItem *rootItem, QObject *parent) : QAbstractItemModel(parent)
  69         {
  70             m_rootItem = rootItem;
  71         }
  72
  73         ~CountryModel()
  74         {
  75             if (m_rootItem->parent == NULL) delete m_rootItem;
  76         }
  77
  78         void addRegion(const QString &name, const QString &tag)
  79         {
  80             CountryModelItem *cmi = new CountryModelItem(m_rootItem, name, tag);
  81             m_rootItem->children.append(cmi);
  82         }
  83
  84         void addSubRegion(const KIcon &flag, const QString &name, const QString &tag, const QString &superRegion)
  85         {
  86             CountryModelItem *parent = NULL;
  87             foreach(CountryModelItem *cm, m_rootItem->children)
  88             {
  89                 if (cm->tag == superRegion) parent = cm;
  90             }
  91             if (parent)
  92             {
  93                 CountryModelItem *cmi = new CountryModelItem(parent, name, tag);
  94                 cmi->icon = flag;
  95                 parent->children.append(cmi);
  96             }
  97         }
  98
  99         void sort()
 100         {
 101             qSort(m_rootItem->children.begin(), m_rootItem->children.end(), CountryModelItemLessThan);
 102             foreach(CountryModelItem *cm, m_rootItem->children)
 103             {
 104                 qSort(cm->children.begin(), cm->children.end(), CountryModelItemLessThan);
 105             }
 106         }
 107
 108         int rowCount(const QModelIndex &parent) const
 109         {
 110             if (parent.column() > 0)
 111                 return 0;
 112
 113             CountryModelItem *parentItem;
 114             if (!parent.isValid())
 115                 parentItem = m_rootItem;
 116             else
 117                 parentItem = static_cast<CountryModelItem*>(parent.internalPointer());
 118
 119             return parentItem->children.count();
 120         }
 121
 122         int columnCount(const QModelIndex &parent) const
 123         {
 124             if (!parent.isValid()) return 1;
 125             else
 126             {
 127                 CountryModelItem *p = static_cast<CountryModelItem*>(parent.internalPointer());
 128                 if (p->parent) return 1;
 129                 else return 0;
 130             }
 131         }
 132
 133         QModelIndex index(int row, int column, const QModelIndex &parent) const
 134         {
 135             if (!hasIndex(row, column, parent))
 136                 return QModelIndex();
 137
 138             CountryModelItem *parentItem;
 139             if (!parent.isValid())
 140                 parentItem = m_rootItem;
 141             else
 142                 parentItem = static_cast<CountryModelItem*>(parent.internalPointer());
 143
 144             CountryModelItem *childItem = parentItem->children.at(row);
 145             if (childItem)
 146                 return createIndex(row, column, childItem);
 147             else
 148                 return QModelIndex();
 149         }
 150
 151         QModelIndex parent(const QModelIndex &index) const
 152         {
 153             if (!index.isValid())
 154                 return QModelIndex();
 155
 156             CountryModelItem *childItem = static_cast<CountryModelItem*>(index.internalPointer());
 157             CountryModelItem *parentItem = childItem->parent;
 158
 159             if (parentItem == m_rootItem)
 160                 return QModelIndex();
 161
 162             return createIndex(parentItem->row(), 0, parentItem);
 163         }
 164
 165         QVariant data(const QModelIndex &index, int role) const
 166         {
 167             if (index.isValid())
 168             {
 169                 CountryModelItem *cmi = static_cast<CountryModelItem*>(index.internalPointer());
 170                 if (role == Qt::DisplayRole)
 171                 {
 172                     return cmi->text;
 173                 }
 174                 else if (role == Qt::DecorationRole)
 175                 {
 176                     if (cmi->parent->parent) return cmi->icon;
 177                 }
 178             }
 179             return QVariant();
 180         }
 181
 182     private:
 183         CountryModelItem *m_rootItem;
 184 };
 185
 186
 187
 188 class CSDListView : public QListView
 189 {
 190     public:
 191         CSDListView(QWidget *parent) : QListView(parent)
 192         {
 193         }
 194
 195         void setOtherWidget(QWidget *widget, Qt::Key key)
 196         {
 197             m_other = widget;
 198             m_key = key;
 199         }
 200
 201     protected:
 202         void keyPressEvent(QKeyEvent *event)
 203         {
 204             if (event->key() == m_key) m_other->setFocus();
 205             else QListView::keyPressEvent(event);
 206         }
 207
 208     private:
 209         QWidget *m_other;
 210         Qt::Key m_key;
 211 };
 212
 213
 214
 215 CountrySelectorDialog::CountrySelectorDialog(QWidget *parent) : KDialog(parent)
 216 {
 217     setCaption( i18n("Country Selector") );
 218     setButtons( KDialog::Ok | KDialog::Cancel );
 219
 220     QWidget *widget = new QWidget(this);
 221     setMainWidget(widget);
 222 }
 223
 224 bool CountrySelectorDialog::editCountry(KControlLocale *locale)
 225 {
 226     QHBoxLayout *hbl = new QHBoxLayout(mainWidget());
 227     CSDListView *lv1 = new CSDListView(mainWidget());
 228     m_countriesView = new CSDListView(mainWidget());
 229     hbl->addWidget(lv1);
 230     hbl->addWidget(m_countriesView);
 231     lv1->setOtherWidget(m_countriesView, Qt::Key_Right);
 232     m_countriesView->setOtherWidget(lv1, Qt::Key_Left);
 233
 234     QString country, region;
 235     country = locale->country();
 236
 237     CountryModel *cm = new CountryModel(this);
 238
 239     const QString &sub = QString::fromLatin1("l10n/");
 240
 241     QStringList regionlist = KGlobal::dirs()->findAllResources("locale",
 242                                  sub + QString::fromLatin1("*.desktop"),
 243                                  KStandardDirs::NoDuplicates);
 244
 245     QFontMetrics fm(lv1->font());
 246     int lv1Width = 0;
 247
 248     foreach(const QString &region, regionlist)
 249     {
 250         QString tag = region;
 251         int index;
 252
 253         index = tag.lastIndexOf('/');
 254         if (index != -1)
 255             tag = tag.mid(index + 1);
 256
 257         index = tag.lastIndexOf('.');
 258         if (index != -1)
 259             tag.truncate(index);
 260
 261         KConfig entry(region);
 262         KConfigGroup cg = entry.group("KCM Locale");
 263         QString name = cg.readEntry("Name", ki18n("without name").toString(locale));
 264
 265         cm->addRegion(name, tag);
 266         QString spacedName = name + "  ";
 267         lv1Width = qMax(lv1Width, fm.width(spacedName));
 268     }
 269
 270     // add all languages to the list
 271     QStringList countrylist = KGlobal::dirs()->findAllResources("locale",
 272                                  sub + QString::fromLatin1("*/entry.desktop"),
 273                                  KStandardDirs::NoDuplicates);
 274     foreach(const QString &countryFile, countrylist)
 275     {
 276         KConfig entry(countryFile);
 277         KConfigGroup cg = entry.group("KCM Locale");
 278         QString name = cg.readEntry("Name", ki18n("without name").toString(locale));
 279         QString parentRegion = cg.readEntry("Region");
 280
 281         QString tag = countryFile;
 282         int index = tag.lastIndexOf('/');
 283         tag.truncate(index);
 284         index = tag.lastIndexOf('/');
 285         tag = tag.mid(index + 1);
 286
 287         QString flag( KStandardDirs::locate( "locale", QString::fromLatin1( "l10n/%1/flag.png" ).arg(tag) ) );
 288         cm->addSubRegion(KIcon(flag), name, tag, parentRegion);
 289         if (tag == country) region = parentRegion;
 290     }
 291
 292     cm->sort();
 293
 294     cm->addRegion(i18nc("@item:inlistbox Country", "Not set (Generic English)"), "C");
 295     if (country == "C") region = "C";
 296
 297     lv1->setModel(cm);
 298     lv1->setFixedWidth(lv1Width + lv1->verticalScrollBar()->height());
 299     // + 2 because 1 is "Not set (Generic English)" and the other is for spacing
 300     lv1->setMinimumHeight((regionlist.count() + 2) * fm.height());
 301
 302     connect(lv1->selectionModel(), SIGNAL(currentChanged(const QModelIndex &, const QModelIndex &)), SLOT(regionChanged(const QModelIndex &)));
 303     connect(lv1, SIGNAL(activated(const QModelIndex &)), SLOT(regionActivated()));
 304     connect(m_countriesView, SIGNAL(activated(const QModelIndex &)), SLOT(accept()));
 305
 306     for(int i = 0; i < lv1->model()->rowCount(); ++i)
 307     {
 308         const QModelIndex &current = lv1->model()->index(i, 0);
 309         CountryModelItem *cmi = static_cast<CountryModelItem*>(current.internalPointer());
 310         if (cmi->tag == region)
 311         {
 312             lv1->selectionModel()->setCurrentIndex(current, QItemSelectionModel::SelectCurrent);
 313         }
 314     }
 315     if (m_countriesView->model() != NULL)
 316     {
 317         for(int i = 0; i < m_countriesView->model()->rowCount(); ++i)
 318         {
 319             const QModelIndex &current = m_countriesView->model()->index(i, 0);
 320             CountryModelItem *cmi = static_cast<CountryModelItem*>(current.internalPointer());
 321             if (cmi->tag == country)
 322             {
 323                 m_countriesView->selectionModel()->setCurrentIndex(current, QItemSelectionModel::SelectCurrent);
 324             }
 325         }
 326         m_countriesView->setFocus();
 327     }
 328     else lv1->setFocus();
 329
 330     if (exec() == QDialog::Accepted)
 331     {
 332         const QModelIndex &current = m_countriesView->currentIndex();
 333         if (current.isValid())
 334         {
 335             CountryModelItem *cmi = static_cast<CountryModelItem*>(current.internalPointer());
 336             return locale->setCountry(cmi->tag);
 337         }
 338         else if (m_countriesView->model() == NULL)
 339         {
 340             return locale->setCountry("C");
 341         }
 342     }
 343     return false;
 344 }
 345
 346 void CountrySelectorDialog::regionChanged(const QModelIndex &current)
 347 {
 348     delete m_countriesView->model();
 349     CountryModelItem *cmi = static_cast<CountryModelItem*>(current.internalPointer());
 350     if (!cmi->children.isEmpty())
 351     {
 352         CountryModel *cm = new CountryModel(cmi, this);
 353         m_countriesView->setModel(cm);
 354     }
 355     else
 356     {
 357         m_countriesView->setModel(NULL);
 358     }
 359 }
 360
 361 void CountrySelectorDialog::regionActivated()
 362 {
 363     if (m_countriesView->model() != NULL) m_countriesView->setFocus();
 364     else accept();
 365 }