Salome HOME
Merge remote branch 'origin/V8_5_asterstudy'
[modules/smesh.git] / src / StdMeshers_I / StdMeshers_ImportSource1D_i.cxx
1 // Copyright (C) 2007-2016  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, or (at your option) any later version.
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
23 //  SMESH SMESH_I : idl implementation based on 'SMESH' unit's classes
24 //  File   : StdMeshers_ImportSource1D_i.cxx
25 //  Module : SMESH
26 //
27 #include "StdMeshers_ImportSource1D_i.hxx"
28
29 #include "SMESH_Gen.hxx"
30 #include "SMESH_Gen_i.hxx"
31 #include "SMESH_Group_i.hxx"
32 #include "SMESH_PythonDump.hxx"
33 #include "StdMeshers_ObjRefUlils.hxx"
34
35 #include <Utils_CorbaException.hxx>
36 #include <utilities.h>
37 #include <SALOMEDS_wrap.hxx>
38
39 #include <TCollection_AsciiString.hxx>
40
41 #include CORBA_SERVER_HEADER(SMESH_Group)
42
43 using namespace std;
44
45 //=============================================================================
46 /*!
47  *  StdMeshers_ImportSource1D_i::StdMeshers_ImportSource1D_i
48  *
49  *  Constructor
50  */
51 //=============================================================================
52
53 StdMeshers_ImportSource1D_i::StdMeshers_ImportSource1D_i( PortableServer::POA_ptr thePOA,
54                                                           ::SMESH_Gen*            theGenImpl )
55   : SALOME::GenericObj_i( thePOA ), 
56     SMESH_Hypothesis_i( thePOA )
57 {
58   myBaseImpl = new ::StdMeshers_ImportSource1D( theGenImpl->GetANewId(),
59                                                 theGenImpl );
60   _groupEntries = new SMESH::string_array();
61 }
62
63 //=============================================================================
64 /*!
65  *  StdMeshers_ImportSource1D_i::~StdMeshers_ImportSource1D_i
66  *
67  *  Destructor
68  */
69 //=============================================================================
70
71 StdMeshers_ImportSource1D_i::~StdMeshers_ImportSource1D_i()
72 {
73 }
74
75 //=============================================================================
76 /*!
77  *  SetSourceEdges
78  */
79 //=============================================================================
80
81 void StdMeshers_ImportSource1D_i::SetSourceEdges(const SMESH::ListOfGroups& groups)
82 {
83   ASSERT( myBaseImpl );
84   try
85   {
86     std::vector<SMESH_Group*> smesh_groups;
87     std::vector<string> entries;
88     for ( CORBA::ULong i = 0; i < groups.length(); ++i )
89       if ( SMESH_GroupBase_i* gp_i = SMESH::DownCast<SMESH_GroupBase_i*>( groups[i] ))
90       {
91         if ( gp_i->GetType() != SMESH::EDGE )
92           THROW_SALOME_CORBA_EXCEPTION("Wrong group type", SALOME::BAD_PARAM);
93         smesh_groups.push_back( gp_i->GetSmeshGroup() );
94
95         SALOMEDS::SObject_wrap so = SMESH_Gen_i::ObjectToSObject(groups[i]);
96         if ( !so->_is_nil())
97         {
98           CORBA::String_var entry = so->GetID();
99           entries.push_back( entry.in() );
100         }
101       }
102     this->GetImpl()->SetGroups( smesh_groups );
103
104     _groupEntries = new SMESH::string_array;
105     _groupEntries->length( entries.size ());
106     for ( size_t i = 0; i < entries.size(); ++i )
107       _groupEntries[i] = entries[i].c_str();
108   }
109   catch ( SALOME_Exception& S_ex )
110   {
111     THROW_SALOME_CORBA_EXCEPTION( S_ex.what(), SALOME::BAD_PARAM );
112   }
113
114   // Update Python script
115   SMESH::TPythonDump() << _this() << ".SetSourceEdges( " << groups << " )";
116 }
117
118 //=============================================================================
119 /*!
120  * Return entries of groups
121  */
122 //=============================================================================
123
124 SMESH::string_array*  StdMeshers_ImportSource1D_i::GetSourceEdges()
125 {
126   SMESH::string_array_var res = new SMESH::string_array( _groupEntries );
127   return res._retn();
128 }
129
130 //================================================================================
131 /*!
132  * \brief Set to copy mesh and groups
133  */
134 //================================================================================
135
136 void StdMeshers_ImportSource1D_i::SetCopySourceMesh(CORBA::Boolean toCopyMesh,
137                                                     CORBA::Boolean toCopyGroups)
138 {
139   GetImpl()->SetCopySourceMesh(toCopyMesh,toCopyGroups);
140   SMESH::TPythonDump() << _this() << ".SetCopySourceMesh( "
141                        << toCopyMesh << ", " << toCopyGroups << " )";
142 }
143
144 //================================================================================
145 /*!
146  * \brief Return "to copy mesh and groups"
147  */
148 //================================================================================
149
150 void StdMeshers_ImportSource1D_i::GetCopySourceMesh(CORBA::Boolean& toCopyMesh,
151                                                     CORBA::Boolean& toCopyGroups)
152 {
153   GetImpl()->GetCopySourceMesh(toCopyMesh,toCopyGroups);
154 }
155
156 //================================================================================
157 /*!
158  * \brief Write parameters in a string
159   * \retval char* - resulting string
160  */
161 //================================================================================
162
163 char* StdMeshers_ImportSource1D_i::SaveTo()
164 {
165   std::ostringstream os;
166   os << " " << _groupEntries->length();
167
168   for ( size_t i = 0; i < _groupEntries->length(); ++i )
169   {
170     // entry
171     os << " " << _groupEntries[i];
172
173     // id
174     SALOMEDS::SObject_wrap groupSO = SMESH_Gen_i::getStudyServant()->FindObjectID( _groupEntries[i] );
175     CORBA::Object_var     groupObj;
176     if ( !groupSO->_is_nil() )
177       groupObj = groupSO->GetObject();
178     StdMeshers_ObjRefUlils::SaveToStream( groupObj, os );
179   }
180
181   myBaseImpl->SaveTo( os );
182
183   return CORBA::string_dup( os.str().c_str() );
184 }
185
186 //================================================================================
187 /*!
188  * \brief Retrieve parameters from the string
189   * \param theStream - the input string
190  */
191 //================================================================================
192
193 void StdMeshers_ImportSource1D_i::LoadFrom( const char* theStream )
194 {
195   std::istringstream is( theStream );
196
197   int nbGroups;
198   is >> nbGroups;
199
200   _groupEntries = new SMESH::string_array;
201   _groupEntries->length( nbGroups );
202   std::string id, entry;
203   for ( size_t i = 0; i < _groupEntries->length(); ++i )
204   {
205     if ( is >> entry )
206       _groupEntries[i] = entry.c_str();
207     else
208     {
209       _groupEntries->length( i );
210       is.clear(ios::badbit | is.rdstate());
211       break;
212     }
213     if ( is >> id )
214       _groupIDs.push_back( id );
215     else
216     {
217       is.clear(ios::badbit | is.rdstate());
218       break;
219     }
220   }
221
222   myBaseImpl->LoadFrom( is );
223 }
224
225 //================================================================================
226 /*!
227  * \brief Retrieve groups by their ids loaded by LoadFrom()
228  * This is possible only when all meshes are fully loaded
229  */
230 //================================================================================
231
232 void StdMeshers_ImportSource1D_i::UpdateAsMeshesRestored()
233 {
234   std::vector<SMESH_Group*> smesh_groups;
235   for ( unsigned i = 0; i < _groupIDs.size(); ++i )
236   {
237     std::istringstream is( _groupIDs[i].c_str() );
238     SMESH::SMESH_GroupBase_var group =
239       StdMeshers_ObjRefUlils::LoadObjectFromStream<SMESH::SMESH_GroupBase>( is );
240     if ( SMESH_GroupBase_i* gp_i = SMESH::DownCast<SMESH_GroupBase_i*>( group ))
241       smesh_groups.push_back( gp_i->GetSmeshGroup() );
242   }
243   GetImpl()->RestoreGroups(smesh_groups);
244 }
245
246 //=============================================================================
247 /*!
248  *  StdMeshers_ImportSource1D_i::GetImpl
249  *
250  *  Get implementation
251  */
252 //=============================================================================
253
254 ::StdMeshers_ImportSource1D* StdMeshers_ImportSource1D_i::GetImpl()
255 {
256   return ( ::StdMeshers_ImportSource1D* )myBaseImpl;
257 }
258
259 //================================================================================
260 /*!
261  * \brief Verify whether hypothesis supports given entity type 
262   * \param type - dimension (see SMESH::Dimension enumeration)
263   * \retval CORBA::Boolean - TRUE if dimension is supported, FALSE otherwise
264  * 
265  * Verify whether hypothesis supports given entity type (see SMESH::Dimension enumeration)
266  */
267 //================================================================================  
268 CORBA::Boolean StdMeshers_ImportSource1D_i::IsDimSupported( SMESH::Dimension type )
269 {
270   return type == SMESH::DIM_1D;
271 }
272