Salome HOME
0020523: String notebook support
[modules/kernel.git] / src / SALOMEDS / SALOMEDS_Driver_i.cxx
1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 //  This library is free software; you can redistribute it and/or
7 //  modify it under the terms of the GNU Lesser General Public
8 //  License as published by the Free Software Foundation; either
9 //  version 2.1 of the License.
10 //
11 //  This library is distributed in the hope that it will be useful,
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 //  Lesser General Public License for more details.
15 //
16 //  You should have received a copy of the GNU Lesser General Public
17 //  License along with this library; if not, write to the Free Software
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 #include "SALOMEDS_Driver_i.hxx"
23 #include "SALOMEDS_TMPFile_i.hxx"
24 #include "utilities.h"
25 #include "SALOMEDS_SObject_i.hxx"
26 #include "SALOMEDS_SComponent_i.hxx"
27 #include "SALOMEDS_Study_i.hxx"
28 #include "SALOMEDS.hxx"
29 #include <stdlib.h>
30
31 using namespace std;  
32
33 SALOMEDS_Driver_i::~SALOMEDS_Driver_i()
34 {
35 }
36
37 SALOMEDSImpl_TMPFile* SALOMEDS_Driver_i::Save(const SALOMEDSImpl_SComponent& theComponent,
38                                       const string& theURL,
39                                       long& theStreamLength,
40                                       bool isMultiFile)
41 {  
42   SALOMEDS::SComponent_var sco = SALOMEDS_SComponent_i::New (theComponent, _orb);
43   CORBA::String_var url = CORBA::string_dup(theURL.c_str());
44
45   SALOMEDS::unlock();
46   SALOMEDS::TMPFile_var aStream = _driver->Save(sco.in(), url, isMultiFile);
47   SALOMEDSImpl_TMPFile* aTMPFile = new SALOMEDS_TMPFile_i(aStream._retn());
48   theStreamLength = aTMPFile->Size();
49   SALOMEDS::lock();
50
51   return aTMPFile;
52 }
53
54 SALOMEDSImpl_TMPFile* SALOMEDS_Driver_i::SaveASCII(const SALOMEDSImpl_SComponent& theComponent,
55                                                    const string& theURL,
56                                                    long& theStreamLength,
57                                                    bool isMultiFile)
58 {
59   SALOMEDS::SComponent_var sco = SALOMEDS_SComponent_i::New (theComponent, _orb);
60   CORBA::String_var url = CORBA::string_dup(theURL.c_str());
61
62   SALOMEDS::unlock();
63   SALOMEDS::TMPFile_var aStream = _driver->SaveASCII(sco.in(), url, isMultiFile);
64   SALOMEDSImpl_TMPFile* aTMPFile = new SALOMEDS_TMPFile_i(aStream._retn());
65   theStreamLength = aTMPFile->Size();
66   SALOMEDS::lock();
67
68   return aTMPFile;
69 }
70   
71 bool SALOMEDS_Driver_i::Load(const SALOMEDSImpl_SComponent& theComponent,
72                              const unsigned char* theStream,
73                              const long theStreamLength,
74                              const string& theURL,
75                              bool isMultiFile)
76 {
77   SALOMEDS::SComponent_var sco = SALOMEDS_SComponent_i::New (theComponent, _orb);
78   CORBA::String_var url = CORBA::string_dup(theURL.c_str());
79   CORBA::Octet* anOctetBuf = (CORBA::Octet*)theStream;
80
81   SALOMEDS::TMPFile_var aStream;
82   if (theStreamLength > 0) 
83     aStream = new SALOMEDS::TMPFile(theStreamLength, theStreamLength, anOctetBuf, 0);  
84   else 
85     aStream = new SALOMEDS::TMPFile(0);
86
87   SALOMEDS::unlock();
88   bool isOk = _driver->Load(sco.in(), aStream.in(), url, isMultiFile);
89   SALOMEDS::lock();
90
91   return isOk;
92 }
93
94 bool SALOMEDS_Driver_i::LoadASCII(const SALOMEDSImpl_SComponent& theComponent,
95                                   const unsigned char* theStream,
96                                   const long theStreamLength,
97                                   const string& theURL,
98                                   bool isMultiFile)
99 {
100   SALOMEDS::SComponent_var sco = SALOMEDS_SComponent_i::New (theComponent, _orb);
101   CORBA::String_var url = CORBA::string_dup(theURL.c_str());
102   CORBA::Octet* anOctetBuf = (CORBA::Octet*)theStream;
103
104   SALOMEDS::TMPFile_var aStream;
105   if(theStreamLength > 0) 
106     aStream = new SALOMEDS::TMPFile(theStreamLength, theStreamLength, anOctetBuf, 0);  
107   else 
108     aStream = new SALOMEDS::TMPFile(0);
109
110   SALOMEDS::unlock();
111   bool isOk = _driver->LoadASCII(sco.in(), aStream.in(), url, isMultiFile);
112   SALOMEDS::lock();
113
114   return isOk;
115 }
116
117 void SALOMEDS_Driver_i::Close(const SALOMEDSImpl_SComponent& theComponent)
118 {
119   SALOMEDS::SComponent_var sco = SALOMEDS_SComponent_i::New (theComponent, _orb);
120
121   SALOMEDS::unlock();
122   _driver->Close(sco.in());
123   SALOMEDS::lock();
124 }
125
126
127
128 string SALOMEDS_Driver_i::IORToLocalPersistentID(const SALOMEDSImpl_SObject& theSObject,
129                                                  const string& IORString,
130                                                  bool isMultiFile,
131                                                  bool isASCII)
132 {
133   SALOMEDS::SObject_var so = SALOMEDS_SObject_i::New (theSObject, _orb);
134   CORBA::String_var ior = CORBA::string_dup(IORString.c_str());
135
136   SALOMEDS::unlock();
137   CORBA::String_var pers_string =_driver->IORToLocalPersistentID(so.in(), ior.in(), isMultiFile, isASCII);
138   SALOMEDS::lock();
139
140   return string(pers_string);
141 }
142
143
144 string SALOMEDS_Driver_i::LocalPersistentIDToIOR(const SALOMEDSImpl_SObject& theObject,
145                                                  const string& aLocalPersistentID,
146                                                  bool isMultiFile,
147                                                  bool isASCII)
148 {
149   SALOMEDS::SObject_var so = SALOMEDS_SObject_i::New (theObject, _orb);
150   CORBA::String_var pers_string = CORBA::string_dup(aLocalPersistentID.c_str());
151   SALOMEDS::unlock();
152   CORBA::String_var IOR = _driver->LocalPersistentIDToIOR(so.in(), pers_string.in(), isMultiFile, isASCII);
153   SALOMEDS::lock();
154   return string(IOR);
155 }
156
157 bool SALOMEDS_Driver_i::CanCopy(const SALOMEDSImpl_SObject& theObject)
158 {
159   SALOMEDS::SObject_var so = SALOMEDS_SObject_i::New (theObject, _orb);
160
161   SALOMEDS::unlock();
162   bool isOk = _driver->CanCopy(so.in());
163   so->Destroy();
164   SALOMEDS::lock();
165
166   return isOk;
167 }
168
169
170 SALOMEDSImpl_TMPFile* SALOMEDS_Driver_i::CopyFrom(const SALOMEDSImpl_SObject& theObject, 
171                                                   int& theObjectID,
172                                                   long& theStreamLength)
173 {
174   SALOMEDS::SObject_var so = SALOMEDS_SObject_i::New (theObject, _orb);
175
176   SALOMEDS::unlock();
177   CORBA::Long anObjectID;
178   SALOMEDS::TMPFile_var aStream = _driver->CopyFrom(so.in(), anObjectID);
179   SALOMEDSImpl_TMPFile* aTMPFile = new SALOMEDS_TMPFile_i(aStream._retn());
180   theStreamLength = aTMPFile->Size();
181   theObjectID = anObjectID;
182   SALOMEDS::lock();
183
184   return aTMPFile;
185 }
186
187 bool SALOMEDS_Driver_i::CanPaste(const string& theComponentName, int theObjectID)
188 {
189   SALOMEDS::unlock();
190   bool canPaste = _driver->CanPaste(theComponentName.c_str(), theObjectID);
191   SALOMEDS::lock();
192   return canPaste;
193 }
194
195 string SALOMEDS_Driver_i::PasteInto(const unsigned char* theStream,
196                                     const long theStreamLength,
197                                     int theObjectID,
198                                     const SALOMEDSImpl_SObject& theObject)
199 {
200   SALOMEDS::SObject_var so = SALOMEDS_SObject_i::New (theObject, _orb);
201   CORBA::Octet* anOctetBuf =  (CORBA::Octet*)theStream;
202
203   SALOMEDS::TMPFile_var aStream;
204   if(theStreamLength > 0) 
205     aStream = new SALOMEDS::TMPFile(theStreamLength, theStreamLength, anOctetBuf, 0);  
206   else 
207     aStream = new SALOMEDS::TMPFile(0);
208
209   SALOMEDS::unlock();
210   SALOMEDS::SObject_var ret_so = _driver->PasteInto(aStream.in(), theObjectID, so.in());
211   SALOMEDS::lock();
212
213   return string(ret_so->GetID());
214 }
215
216 SALOMEDSImpl_TMPFile* SALOMEDS_Driver_i::DumpPython(SALOMEDSImpl_Study* theStudy, 
217                                                     bool isPublished, 
218                                                     bool& isValidScript,
219                                                     long& theStreamLength)
220 {
221   SALOMEDS_Study_i *  st_servant = new SALOMEDS_Study_i (theStudy, _orb);
222   SALOMEDS::Study_var st  = SALOMEDS::Study::_narrow(st_servant->_this());
223   Engines::Component_ptr aComponent = Engines::Component::_narrow(_driver);
224
225   SALOMEDS::unlock();
226   CORBA::Boolean aValidScript, aPublished = isPublished;
227   Engines::TMPFile_var aStream = aComponent->DumpPython(st.in(), aPublished, aValidScript);
228   SALOMEDSImpl_TMPFile* aTMPFile = new Engines_TMPFile_i(aStream._retn());
229   theStreamLength = aTMPFile->Size();
230   isValidScript = aValidScript;
231   SALOMEDS::lock();
232
233   return aTMPFile;
234 }
235
236 //###############################################################################################################
237 //                                          SALOMEDS_DriverFactory
238 //###############################################################################################################
239
240 SALOMEDSImpl_Driver* SALOMEDS_DriverFactory_i::GetDriverByType(const string& theComponentType)
241 {
242   CORBA::Object_var obj;
243
244   string aFactoryType;
245   if (theComponentType == "SUPERV") aFactoryType = "SuperVisionContainer";
246   else aFactoryType = "FactoryServer";
247
248   SALOMEDS::unlock();
249   obj = SALOME_LifeCycleCORBA(_name_service).FindOrLoad_Component(aFactoryType.c_str(), theComponentType.c_str());
250   SALOMEDS::lock();
251
252   if (CORBA::is_nil(obj)) {
253     obj = SALOME_LifeCycleCORBA(_name_service).FindOrLoad_Component("FactoryServerPy", theComponentType.c_str());
254   }
255
256   if (!CORBA::is_nil(obj)) {
257     SALOMEDS::Driver_var aDriver = SALOMEDS::Driver::_narrow(obj);
258     return new SALOMEDS_Driver_i(aDriver, _orb);
259   }
260
261   return NULL;
262 }
263
264 SALOMEDSImpl_Driver* SALOMEDS_DriverFactory_i::GetDriverByIOR(const string& theIOR)
265 {
266   CORBA::Object_var obj;
267   obj = _orb->string_to_object(theIOR.c_str());
268  
269   if (!CORBA::is_nil(obj)) {
270     SALOMEDS::Driver_var aDriver = SALOMEDS::Driver::_narrow(obj);
271     return new SALOMEDS_Driver_i(aDriver, _orb);
272   }
273
274   return NULL;
275 }