]> SALOME platform Git repositories - modules/shaper.git/blob - src/ModuleBase/ModuleBase_WidgetFileSelector.cpp
Salome HOME
Issue #2052: Modification of parameters don't work (sketch, extrusion)
[modules/shaper.git] / 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 }