Version: 8.3.0
SalomeWrap_DataModel.cxx
Go to the documentation of this file.
1 // Copyright (C) 2006-2016 CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19 
20 #include "SalomeWrap_DataModel.hxx"
21 
22 #include <SalomeApp_DataObject.h>
23 #include <SalomeApp_Study.h>
24 #include <SalomeApp_Module.h>
25 #include <SalomeApp_Application.h>
26 #include <LightApp_SelectionMgr.h>
27 #include <SALOME_ListIO.hxx>
28 #include <SUIT_DataBrowser.h>
29 #include <SUIT_ViewManager.h>
30 #include <QxScene_ViewWindow.h>
31 
32 #include <QFileInfo>
33 
34 //#define _DEVDEBUG_
35 #include "YacsTrace.hxx"
36 
37 using namespace std;
38 
39 SalomeWrap_DataModel::SalomeWrap_DataModel(CAM_Module* theModule)
40  : SalomeApp_DataModel(theModule)
41 {
42  DEBTRACE("SalomeWrap_DataModel::SalomeWrap_DataModel");
43  _viewEntryMap.clear();
44  _entryViewMap.clear();
45 }
46 
47 SalomeWrap_DataModel::~SalomeWrap_DataModel()
48 {
49  DEBTRACE("SalomeWrap_DataModel::~SalomeWrap_DataModel");
50 }
51 
52 void SalomeWrap_DataModel::createNewSchema(const QString& schemaName,
53  QWidget* viewWindow)
54 {
55  DEBTRACE("SalomeWrap_DataModel::createNewSchema : " << schemaName.toStdString());
56  SalomeApp_ModuleObject* aRoot = dynamic_cast<SalomeApp_ModuleObject*>(root());
57  if (!aRoot)
58  {
59  DEBTRACE("SalomeApp_ModuleObject* aRoot =0");
60  return;
61  }
62  DEBTRACE("aRoot OK");
63  _PTR(SComponent) aSComp(aRoot->object());
64  _PTR(Study) aStudy = getStudy()->studyDS();
65  _PTR(StudyBuilder) aBuilder(aStudy->NewBuilder());
66  _PTR(GenericAttribute) anAttr;
67  _PTR(AttributeName) aName;
68  _PTR(AttributePixMap) aPixmap;
69  _PTR(AttributeParameter) aType;
70 
71  // --- create a new schema SObject
72 
73  _PTR(SObject) aSObj;
74  aSObj = aBuilder->NewObject(aSComp);
75 
76  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributeName");
77  aName = _PTR(AttributeName)(anAttr);
78  aName->SetValue(QFileInfo(schemaName).baseName().toStdString());
79 
80  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributePixMap");
81  aPixmap = _PTR(AttributePixMap)(anAttr);
82  aPixmap->SetPixMap("schema.png");
83 
84  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributeParameter");
85  aType = _PTR(AttributeParameter)(anAttr);
86  aType->SetInt("ObjectType", SchemaObject);
87  string filePath = schemaName.toStdString();
88  aType->SetString("FilePath", filePath.c_str());
89  DEBTRACE("DataModel : FilePath = " << filePath);
90 
91  _viewEntryMap[viewWindow] = aSObj->GetID();
92  DEBTRACE(viewWindow);
93  _entryViewMap[aSObj->GetID()] = viewWindow;
94  DEBTRACE("--- " << viewWindow << " "<< aSObj->GetID());
95  SalomeApp_Module *mod = dynamic_cast<SalomeApp_Module*>(module());
96  if (mod) mod->updateObjBrowser();
97 
98  QxScene_ViewWindow *swv = dynamic_cast<QxScene_ViewWindow*>(viewWindow);
99  if (!swv) return;
100  QString tabName = QFileInfo(schemaName).baseName();
101  swv->getViewManager()->setTitle(tabName);
102 }
103 
104 bool SalomeWrap_DataModel::renameSchema(const QString& oldName,
105  const QString& newName,
106  QWidget* viewWindow)
107 {
108  DEBTRACE("SalomeWrap_DataModel::renameSchema " << oldName.toStdString()
109  << " " << newName.toStdString());
110  SalomeApp_ModuleObject* aRoot = dynamic_cast<SalomeApp_ModuleObject*>(root());
111  if (!aRoot) return false;
112  if (!_viewEntryMap.count(viewWindow)) return false;
113 
114  _PTR(SComponent) aSComp(aRoot->object());
115  _PTR(Study) aStudy = getStudy()->studyDS();
116 
117  string id = _viewEntryMap[viewWindow];
118  _PTR(SObject) aSObj = aStudy->FindObjectID(id);
119 
120  _PTR(StudyBuilder) aBuilder(aStudy->NewBuilder());
121  _PTR(GenericAttribute) anAttr;
122  _PTR(AttributeName) aName;
123  _PTR(AttributeParameter) aType;
124 
125  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributeName");
126  aName = _PTR(AttributeName)(anAttr);
127  aName->SetValue(QFileInfo(newName).baseName().toStdString());
128 
129  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributeParameter");
130  aType = _PTR(AttributeParameter)(anAttr);
131  aType->SetInt("ObjectType", SchemaObject);
132  string filePath = newName.toStdString();
133  aType->SetString("FilePath", filePath.c_str());
134  DEBTRACE("DataModel : FilePath = " << filePath);
135 
136  SalomeApp_Module *mod = dynamic_cast<SalomeApp_Module*>(module());
137  if (mod) mod->updateObjBrowser();
138 
139  QxScene_ViewWindow *swv = dynamic_cast<QxScene_ViewWindow*>(viewWindow);
140  QString tabName = QFileInfo(newName).baseName();
141  if (swv) swv->getViewManager()->setTitle(tabName);
142  return true;
143 }
144 
145 bool SalomeWrap_DataModel::deleteSchema(QWidget* viewWindow)
146 {
147  DEBTRACE("SalomeWrap_DataModel::deleteSchema");
148  SalomeApp_ModuleObject* aRoot = dynamic_cast<SalomeApp_ModuleObject*>(root());
149  if (!aRoot) return false;
150  if (!_viewEntryMap.count(viewWindow)) return false;
151 
152  _PTR(SComponent) aSComp(aRoot->object());
153  _PTR(Study) aStudy = getStudy()->studyDS();
154 
155  string id = _viewEntryMap[viewWindow];
156  _PTR(SObject) aSObj = aStudy->FindObjectID(id);
157 
158  _PTR(StudyBuilder) aBuilder(aStudy->NewBuilder());
159  aBuilder->RemoveObject(aSObj);
160 
161  SalomeApp_Module *mod = dynamic_cast<SalomeApp_Module*>(module());
162  if (mod) mod->updateObjBrowser();
163  return true;
164 }
165 
166 void SalomeWrap_DataModel::createNewRun(const QString& schemaName,
167  const QString& runName,
168  QWidget* refWindow,
169  QWidget* viewWindow)
170 {
171  DEBTRACE("SalomeWrap_DataModel::createNewRun");
172  SalomeApp_ModuleObject* aRoot = dynamic_cast<SalomeApp_ModuleObject*>(root());
173  if (!aRoot) return;
174  DEBTRACE(refWindow);
175  if (!_viewEntryMap.count(refWindow)) return;
176  DEBTRACE("---");
177 
178  _PTR(SComponent) aSComp(aRoot->object());
179  _PTR(Study) aStudy = getStudy()->studyDS();
180 
181  _PTR(StudyBuilder) aBuilder(aStudy->NewBuilder());
182  _PTR(GenericAttribute) anAttr;
183  _PTR(AttributeName) aName;
184  _PTR(AttributePixMap) aPixmap;
185  _PTR(AttributeParameter) aType;
186 
187  string id = _viewEntryMap[refWindow];
188  _PTR(SObject) aRefObj = aStudy->FindObjectID(id);
189 
190  // --- create a new schema SObject
191 
192  _PTR(SObject) aSObj;
193  aSObj = aBuilder->NewObject(aRefObj);
194 
195  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributeName");
196  aName = _PTR(AttributeName)(anAttr);
197  aName->SetValue(QFileInfo(runName).baseName().toStdString());
198 
199  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributePixMap");
200  aPixmap = _PTR(AttributePixMap)(anAttr);
201  aPixmap->SetPixMap("schema.png");
202 
203  anAttr = aBuilder->FindOrCreateAttribute(aSObj, "AttributeParameter");
204  aType = _PTR(AttributeParameter)(anAttr);
205  aType->SetInt("ObjectType", SchemaObject);
206  string filePath = runName.toStdString();
207  aType->SetString("FilePath", filePath.c_str());
208  DEBTRACE("DataModel : FilePath = " << filePath);
209 
210  _viewEntryMap[viewWindow] = aSObj->GetID();
211  _entryViewMap[aSObj->GetID()] = viewWindow;
212  DEBTRACE("--- " << viewWindow << " "<< aSObj->GetID());
213  SalomeApp_Module *mod = dynamic_cast<SalomeApp_Module*>(module());
214  if (mod) mod->updateObjBrowser();
215 
216  QxScene_ViewWindow *swv = dynamic_cast<QxScene_ViewWindow*>(viewWindow);
217  if (!swv) return;
218 
219  int count = 0;
220  if (_runCountMap.count(schemaName.toStdString()))
221  count = ++_runCountMap[schemaName.toStdString()];
222  else
223  _runCountMap[schemaName.toStdString()] = count;
224 
225  QString tabName = QFileInfo(schemaName).baseName() +QString("_run%1").arg(count);
226  swv->getViewManager()->setTitle(tabName);
227 
228  QPixmap pixmap;
229  pixmap.load("icons:run_active.png");
230  swv->getViewManager()->setIcon(pixmap);
231 }
232 
233 void SalomeWrap_DataModel::setSelected(QWidget* viewWindow)
234 {
235  DEBTRACE("SalomeWrap_DataModel::setSelected");
236  if (!_viewEntryMap.count(viewWindow)) return;
237  string entry = _viewEntryMap[viewWindow];
238  LightApp_SelectionMgr* selMgr = getModule()->getApp()->selectionMgr();
239  SALOME_ListIO ioList;
240  ioList.Append( new SALOME_InteractiveObject( entry.c_str(), "", "" ) );
241  selMgr->setSelectedObjects( ioList, false );
242 }
243 
244 SUIT_DataObject* SalomeWrap_DataModel::getDataObject(std::string entry)
245 {
246  DEBTRACE("SalomeWrap_DataModel::getDataObject " << entry);
247  QString refEntry = entry.c_str();
248  SUIT_DataObject* sdo = 0;
249  SalomeApp_ModuleObject* aRoot = dynamic_cast<SalomeApp_ModuleObject*>(root());
250  if (!aRoot) return 0;
251 
252  QList<SUIT_DataObject*> objList = root()->children(true);
253  for (int i = 0; i < objList.size(); ++i)
254  {
255  SalomeApp_DataObject* obj = dynamic_cast<SalomeApp_DataObject*>(objList.at(i));
256  if (obj && (!QString::compare(refEntry, obj->entry())))
257  {
258  sdo = obj;
259  break;
260  }
261  }
262  return sdo;
263 }
264 
265 QWidget* SalomeWrap_DataModel::getViewWindow(std::string entry)
266 {
267  if (! _entryViewMap.count(entry)) return 0;
268  return _entryViewMap[entry];
269 }