src/ModuleBase/ModuleBase_WidgetFileSelector.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 /*
   4  * ModuleBase_WidgetFileSelector.cpp
   5  *
   6  *  Created on: Aug 28, 2014
   7  *      Author: sbh
   8  */
   9
  10 #include <ModelAPI_AttributeString.h>
  11 #include <ModelAPI_Data.h>
  12 #include <ModelAPI_Object.h>
  13 #include <ModelAPI_Validator.h>
  14 #include <ModelAPI_Session.h>
  15 #include <ModuleBase_WidgetFileSelector.h>
  16 #include <ModuleBase_Tools.h>
  17
  18 #include <Config_WidgetAPI.h>
  19
  20 #include <QFileDialog>
  21 #include <QGridLayout>
  22 #include <QLabel>
  23 #include <QLineEdit>
  24 #include <QList>
  25 #include <QObject>
  26 #include <QPushButton>
  27 #include <QRegExp>
  28 #include <QString>
  29
  30 #include <memory>
  31 #include <string>
  32
  33 ModuleBase_WidgetFileSelector::ModuleBase_WidgetFileSelector(QWidget* theParent,
  34                                                              const Config_WidgetAPI* theData)
  35 : ModuleBase_ModelWidget(theParent, theData)
  36 {
  37   myTitle = translate(theData->getProperty("title"));
  38   myType = (theData->getProperty("type") == "save") ? WFS_SAVE : WFS_OPEN;
  39   myDefaultPath = QString::fromStdString(theData->getProperty("path"));
  40
  41   QGridLayout* aMainLay = new QGridLayout(this);
  42   ModuleBase_Tools::adjustMargins(aMainLay);
  43   QLabel* aTitleLabel = new QLabel(myTitle, this);
  44   aTitleLabel->setIndent(1);
  45   aMainLay->addWidget(aTitleLabel, 0, 0);
  46   myPathField = new QLineEdit(this);
  47   aMainLay->addWidget(myPathField, 1, 0);
  48   QPushButton* aSelectPathBtn = new QPushButton("...", this);
  49   aSelectPathBtn->setToolTip(tr("Select file..."));
  50   aSelectPathBtn->setMaximumWidth(20);
  51   aSelectPathBtn->setMaximumHeight(20);
  52   aMainLay->addWidget(aSelectPathBtn, 1, 1);
  53   aMainLay->setColumnStretch(0, 1);
  54   myPathField->setMinimumHeight(20);
  55   aMainLay->setHorizontalSpacing(1);
  56   this->setLayout(aMainLay);
  57
  58   connect(myPathField, SIGNAL(textChanged(const QString&)),
  59           this,        SLOT(onPathChanged()));
  60   connect(aSelectPathBtn, SIGNAL(clicked()),
  61           this,        SLOT(onPathSelectionBtn()));
  62 }
  63
  64 ModuleBase_WidgetFileSelector::~ModuleBase_WidgetFileSelector()
  65 {
  66 }
  67
  68 bool ModuleBase_WidgetFileSelector::storeValueCustom()
  69 {
  70   // A rare case when plugin was not loaded.
  71   if (!myFeature)
  72     return false;
  73   DataPtr aData = myFeature->data();
  74   AttributeStringPtr aStringAttr = aData->string(attributeID());
  75   QString aWidgetValue = myPathField->text();
  76   aStringAttr->setValue(aWidgetValue.toStdString());
  77   updateObject(myFeature);
  78   return true;
  79 }
  80
  81 bool ModuleBase_WidgetFileSelector::restoreValueCustom()
  82 {
  83   // A rare case when plugin was not loaded.
  84   if (!myFeature)
  85     return false;
  86   DataPtr aData = myFeature->data();
  87   AttributeStringPtr aStringAttr = aData->string(attributeID());
  88
  89   bool isBlocked = myPathField->blockSignals(true);
  90   QString aNewText = QString::fromStdString(aStringAttr->value());
  91   if( myPathField->text() != aNewText )
  92     myPathField->setText( aNewText );
  93   myPathField->blockSignals(isBlocked);
  94
  95   return true;
  96 }
  97
  98 QList<QWidget*> ModuleBase_WidgetFileSelector::getControls() const
  99 {
 100   QList<QWidget*> result;
 101   result << myPathField;
 102   return result;
 103 }
 104
 105 bool ModuleBase_WidgetFileSelector::isCurrentPathValid()
 106 {
 107   QFileInfo aFile(myPathField->text());
 108   return aFile.exists();
 109 }
 110
 111 void ModuleBase_WidgetFileSelector::onPathSelectionBtn()
 112 {
 113   QString aDefaultPath = myPathField->text().isEmpty()
 114       ? myDefaultPath
 115       : QFileInfo(myPathField->text()).absolutePath();
 116   QString aFilter = filterString();
 117   QString aFileName = (myType == WFS_SAVE)
 118       ? QFileDialog::getSaveFileName(this, myTitle, aDefaultPath, aFilter, &mySelectedFilter)
 119       : QFileDialog::getOpenFileName(this, myTitle, aDefaultPath, aFilter, &mySelectedFilter);
 120   if (!aFileName.isEmpty()) {
 121     if (myType == WFS_SAVE)
 122       aFileName = applyExtension(aFileName, mySelectedFilter);
 123     myPathField->setText(aFileName);
 124     emit focusOutWidget(this);
 125   }
 126 }
 127
 128 void ModuleBase_WidgetFileSelector::onPathChanged()
 129 {
 130   if (myType == WFS_OPEN && !isCurrentPathValid())
 131     return;
 132   storeValue();
 133   emit valuesChanged();
 134 }
 135
 136 QString ModuleBase_WidgetFileSelector::formatToFilter(const QString & theFormat)
 137 {
 138   if (theFormat.isEmpty() && !theFormat.contains(":"))
 139     return QString();
 140
 141   QStringList aExtesionList = theFormat.section(':', 0, 0).split("|");
 142   QString aFormat = theFormat.section(':', 1, 1);
 143   return QString("%1 files (%2)").arg(aFormat)
 144       .arg(QStringList(aExtesionList).replaceInStrings(QRegExp("^(.*)$"), "*.\\1").join(" "));
 145 }
 146
 147 QString ModuleBase_WidgetFileSelector::filterToShortFormat(const QString & theFilter)
 148 {
 149   // Simplified implementation.
 150   // It relies on theFilter was made by formatToFilter() function.
 151   return theFilter.section(' ', 0, 0);
 152 }
 153
 154 QStringList ModuleBase_WidgetFileSelector::filterToExtensions(const QString & theFilter)
 155 {
 156   // Simplified implementation.
 157   // It relies on theFilter was made by formatToFilter() function.
 158   QStringList anExtensions = theFilter.section("(", 1, 1).section(")", 0, 0).split(" ");
 159   return anExtensions;
 160 }
 161
 162 QStringList ModuleBase_WidgetFileSelector::getValidatorFormats() const
 163 {
 164   SessionPtr aMgr = ModelAPI_Session::get();
 165   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
 166
 167   ModelAPI_ValidatorsFactory::Validators allValidators;
 168   aFactory->validators(myFeature->getKind(), myAttributeID, allValidators);
 169
 170   QStringList aResult;
 171   std::list<std::string> anArgumentList = allValidators.front().second;
 172   std::list<std::string>::const_iterator it = anArgumentList.begin();
 173   for (; it != anArgumentList.end(); ++it) {
 174     QString aFormat = QString::fromStdString(*it);
 175     if (!aFormat.isEmpty())
 176       aResult << aFormat;
 177   }
 178   return aResult;
 179 }
 180
 181 QString ModuleBase_WidgetFileSelector::filterString() const
 182 {
 183   QStringList aResult;
 184   QStringList aValidatorFormats = getValidatorFormats();
 185
 186   foreach(const QString & eachFormat, aValidatorFormats) {
 187     aResult << formatToFilter(eachFormat);
 188   }
 189   if (myType == WFS_OPEN)
 190     aResult << QString("All files (*.*)");
 191   return aResult.join(";;");
 192 }
 193
 194 QString ModuleBase_WidgetFileSelector::applyExtension(const QString& theFileName,
 195                                                       const QString& theFilter)
 196 {
 197   QString aResult = theFileName;
 198   bool hasExtension = false;
 199   QStringList anExtensions = filterToExtensions(theFilter);
 200   foreach(const QString& anExtension, anExtensions) {
 201     if (theFileName.endsWith(anExtension.section(".", 1, 1), Qt::CaseInsensitive)) {
 202       hasExtension = true;
 203       break;
 204     }
 205   }
 206   if (!hasExtension && !anExtensions.isEmpty())
 207     aResult = QString("%1.%2").arg(theFileName).arg(anExtensions[0].section(".", 1, 1));
 208   return aResult;
 209 }