src/KERNEL_PY/salome_test.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   3 #
   4 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   6 #
   7 # This library is free software; you can redistribute it and/or
   8 # modify it under the terms of the GNU Lesser General Public
   9 # License as published by the Free Software Foundation; either
  10 # version 2.1 of the License.
  11 #
  12 # This library is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 # Lesser General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU Lesser General Public
  18 # License along with this library; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  20 #
  21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  22 #
  23
  24 print
  25 print "Perform quick test of the application by loading of the GEOM, SMESH, VISU, MED"
  26 print "components and doing some operation within the components."
  27 print
  28
  29 import salome
  30 import SALOME
  31 import SALOMEDS
  32 import os
  33 import sys
  34 import SALOME_ModuleCatalog
  35
  36 step = 1
  37
  38 print "======================================================================"
  39 print "           %d. Initialize study " % step; step+=1
  40 print "======================================================================"
  41
  42 # initialize study
  43 salome.salome_init()
  44 # get study builder
  45 builder = salome.myStudy.NewBuilder()
  46 print "OK"
  47
  48 print
  49
  50 print "======================================================================"
  51 print "           %d. Retrieve module catalog " % step; step+=1
  52 print "======================================================================"
  53
  54 obj = salome.naming_service.Resolve('Kernel/ModulCatalog')
  55 catalog = obj._narrow(SALOME_ModuleCatalog.ModuleCatalog)
  56 if not catalog:
  57     raise RuntimeError, "Can't accesss module catalog"
  58 print "OK"
  59
  60 print
  61
  62 print "======================================================================"
  63 print "           %d. Check modules availability in the module catalog " % step; step+=1
  64 print "======================================================================"
  65
  66 print
  67 print "--- Check GEOM ..."
  68 comp = catalog.GetComponent("GEOM")
  69 if not comp:
  70     raise RuntimeError, "Component GEOM is not found in Module Catalog."
  71 print "OK"
  72
  73 print
  74 print "--- Check SMESH ..."
  75 comp = catalog.GetComponent("SMESH")
  76 if not comp:
  77     raise RuntimeError, "Component SMESH is not found in Module Catalog."
  78 print "OK"
  79
  80 print
  81 print "--- Check MED ..."
  82 comp = catalog.GetComponent("MED")
  83 if not comp:
  84     raise RuntimeError, "Component MED is not found in Module Catalog."
  85 print "OK"
  86
  87 print
  88 print "--- Check VISU ..."
  89 comp = catalog.GetComponent("VISU")
  90 if not comp:
  91     raise RuntimeError, "Component VISU is not found in Module Catalog."
  92 print "OK"
  93
  94 print
  95
  96 print "======================================================================"
  97 print "           %d. Test Data Server " % step; step+=1
  98 print "======================================================================"
  99
 100 print
 101 print "--- Create new component ..."
 102 comp = builder.NewComponent("TEST")
 103 if not comp:
 104     raise RuntimeError, "Can't create new component"
 105 print "OK"
 106
 107 print
 108 print "--- Create AttributeName ..."
 109 A = builder.FindOrCreateAttribute(comp, "AttributeName")
 110 if not A:
 111     raise RuntimeError, "Can't create AttributeName attribute"
 112 A.SetValue("TEST")
 113 if A.Value() != "TEST":
 114     raise RuntimeError, "Error : wrong value of  AttributeName"
 115 print "OK"
 116
 117 print
 118 print "--- Create AttributeReal ..."
 119 A = builder.FindOrCreateAttribute(comp, "AttributeReal")
 120 if not A:
 121     raise RuntimeError, "Can't create AttributeReal attribute"
 122 A.SetValue(0.0001)
 123 if A.Value() != 0.0001:
 124     raise RuntimeError, "Error : wrong value of  AttributeReal"
 125 print "OK"
 126
 127 print
 128
 129 print "======================================================================"
 130 print "           %d. Test Geometry " % step; step+=1
 131 print "======================================================================"
 132
 133 import geompy
 134
 135 ShapeTypeCompSolid = 1
 136 ShapeTypeSolid = 2
 137 ShapeTypeShell = 3
 138 ShapeTypeFace = 4
 139 ShapeTypeWire = 5
 140 ShapeTypeEdge = 6
 141 ShapeTypeVertex = 7
 142
 143 print
 144 print "--- Create a box ..."
 145 box = geompy.MakeBox(0., 0., 0., 100., 200., 300.)
 146 idbox = geompy.addToStudy(box, "box")
 147 box_obj = salome.myStudy.FindObjectByPath("/Geometry/box")
 148 if not box_obj:
 149     raise RuntimeError, "Error : wrong value of  AttributeReal"
 150 print "OK"
 151
 152 # ---- add shell from box in study
 153 print
 154 print "--- Extract shell ..."
 155 subShellList = geompy.SubShapeAll(box, ShapeTypeShell)
 156 shell = subShellList[0]
 157 name = geompy.SubShapeName(shell, box)
 158 idshell = geompy.addToStudyInFather(box, shell, name)
 159 print name
 160 print "OK"
 161
 162 # ---- add first face of box in study
 163 print
 164 print "--- Extract face ..."
 165 subShapeList = geompy.SubShapeAll(box, ShapeTypeFace)
 166 face = subShapeList[0]
 167 name = geompy.SubShapeName(face, box)
 168 idface = geompy.addToStudyInFather(box, face, name)
 169 print name
 170 print "OK"
 171
 172 # ---- add first edge of face in study
 173 print
 174 print "--- Extract edge ..."
 175 edgeList = geompy.SubShapeAll(face, ShapeTypeEdge)
 176 edge = edgeList[0];
 177 name = geompy.SubShapeName(edge, face)
 178 idedge = geompy.addToStudyInFather(face, edge, name)
 179 print name
 180 print "OK"
 181
 182 # ---- update object browser
 183 if salome.hasDesktop():
 184     salome.sg.updateObjBrowser(1);
 185
 186 print
 187
 188 print "======================================================================"
 189 print "           %d. Test Mesh " % step; step+=1
 190 print "======================================================================"
 191
 192 import StdMeshers
 193 import SMESH
 194
 195 smesh = salome.lcc.FindOrLoadComponent("FactoryServer", "SMESH")
 196 if salome.hasDesktop():
 197     smeshgui = salome.ImportComponentGUI("SMESH")
 198     smeshgui.Init(salome.myStudyId);
 199 else:
 200     smesh.SetCurrentStudy(salome.myStudy)
 201
 202 # ---- create hypotheses
 203
 204 if sys.platform == "win32":
 205   stdMeshersEngine = "StdMeshersEngine"
 206 else:
 207   stdMeshersEngine = "libStdMeshersEngine.so"
 208
 209 print
 210 print "--- Create hypotheses ..."
 211
 212 print
 213 print "------ LocalLength ..."
 214 hypLen1 = smesh.CreateHypothesis( "LocalLength", stdMeshersEngine )
 215 hypLen1.SetLength(100)
 216 print hypLen1.GetName()
 217 print hypLen1.GetId()
 218 print hypLen1.GetLength()
 219 if salome.hasDesktop():
 220     smeshgui.SetName(salome.ObjectToID(hypLen1), "Local_Length_100")
 221 print "OK"
 222
 223 print
 224 print "------ NumberOfSegments ..."
 225 hypNbSeg1= smesh.CreateHypothesis( "NumberOfSegments", stdMeshersEngine )
 226 hypNbSeg1.SetNumberOfSegments(7)
 227 print hypNbSeg1.GetName()
 228 print hypNbSeg1.GetId()
 229 print hypNbSeg1.GetNumberOfSegments()
 230 if salome.hasDesktop():
 231     smeshgui.SetName(salome.ObjectToID(hypNbSeg1), "NumberOfSegments_7")
 232 print "OK"
 233
 234 print
 235 print "------ MaxElementArea [1] ..."
 236 hypArea1 = smesh.CreateHypothesis( "MaxElementArea", stdMeshersEngine )
 237 hypArea1.SetMaxElementArea(2500)
 238 print hypArea1.GetName()
 239 print hypArea1.GetId()
 240 print hypArea1.GetMaxElementArea()
 241 if salome.hasDesktop():
 242     smeshgui.SetName(salome.ObjectToID(hypArea1), "MaxElementArea_2500")
 243 print "OK"
 244
 245 print
 246 print "------ MaxElementArea [2] ..."
 247 hypArea2 = smesh.CreateHypothesis( "MaxElementArea", stdMeshersEngine )
 248 hypArea2.SetMaxElementArea(500)
 249 print hypArea2.GetName()
 250 print hypArea2.GetId()
 251 print hypArea2.GetMaxElementArea()
 252 if salome.hasDesktop():
 253     smeshgui.SetName(salome.ObjectToID(hypArea2), "MaxElementArea_500")
 254 print "OK"
 255
 256 # ---- create algorithms
 257
 258 print
 259 print "--- Create algorithms ..."
 260
 261 print
 262 print "------ Regular_1D ..."
 263 algoReg = smesh.CreateHypothesis( "Regular_1D", stdMeshersEngine )
 264 listHyp = algoReg.GetCompatibleHypothesis()
 265 for hyp in listHyp:
 266     print hyp
 267 print algoReg.GetName()
 268 print algoReg.GetId()
 269 if salome.hasDesktop():
 270     smeshgui.SetName(salome.ObjectToID(algoReg), "Regular_1D" )
 271 print "OK"
 272
 273 print
 274 print "------ MEFISTO_2D ..."
 275 algoMef = smesh.CreateHypothesis( "MEFISTO_2D", stdMeshersEngine )
 276 listHyp=algoMef.GetCompatibleHypothesis()
 277 for hyp in listHyp:
 278     print hyp
 279 print algoMef.GetName()
 280 print algoMef.GetId()
 281 if salome.hasDesktop():
 282     smeshgui.SetName(salome.ObjectToID(algoMef), "MEFISTO_2D" )
 283 print "OK"
 284
 285 # ---- create mesh on the box, apply hypotheses / algorithms
 286
 287 print
 288 print "--- Create mesh on the box ..."
 289 mesh = smesh.CreateMesh(box)
 290 if salome.hasDesktop():
 291     smeshgui.SetName( salome.ObjectToID(mesh), "MeshBox" );
 292 ret = mesh.AddHypothesis(box, algoReg)
 293 ret = mesh.AddHypothesis(box, algoMef)
 294 ret = mesh.AddHypothesis(box, hypNbSeg1)
 295 ret = mesh.AddHypothesis(box, hypArea1)
 296 print "OK"
 297
 298 # ---- create submesh on the edge, add hypothesis
 299
 300 print
 301 print "--- Add 1D sub-mesh on the edge ..."
 302 submesh = mesh.GetSubMesh(edge, "SubMeshEdge")
 303 ret = mesh.AddHypothesis(edge, algoReg)
 304 ret = mesh.AddHypothesis(edge, hypLen1)
 305 print "OK"
 306
 307 # ---- create submesh on the edge, add hypothesis
 308
 309 print
 310 print "--- Add 2D sub-mesh on the face ..."
 311 submesh = mesh.GetSubMesh(face, "SubMeshFace")
 312 ret = mesh.AddHypothesis(face, hypArea2)
 313 print "OK"
 314
 315 # ---- compute mesh
 316 print
 317 print "--- Compute mesh ..."
 318 smesh.Compute(mesh, box)
 319 print "OK"
 320
 321 # ---- update object browser
 322 if salome.hasDesktop():
 323     salome.sg.updateObjBrowser(1);
 324
 325 print
 326
 327 print "======================================================================"
 328 print "           %d. Test Med " % step; step+=1
 329 print "======================================================================"
 330
 331 import xmed
 332 from xmed import properties
 333 from xmed.fieldproxy import FieldProxy
 334
 335 xmed.setConsoleGlobals(globals())
 336
 337 # Load some test data in the MedDataManager
 338 filepath  = properties.testFilePath
 339 xmed.dataManager.addDatasource(filepath)
 340 fieldHandlerList = xmed.dataManager.getFieldHandlerList()
 341
 342 fieldHandler0 = fieldHandlerList[0]
 343 print "---Field Handler 0:\n%s" % fieldHandler0
 344 fieldHandler1 = fieldHandlerList[1]
 345 print "---Field Handler 1:\n%s" % fieldHandler1
 346
 347 print "--- The addition of two fields can be done using field handler directly."
 348 addFieldHandler = xmed.calculator.add(fieldHandler0, fieldHandler1)
 349 print "--- Result handler:\n%s" % addFieldHandler
 350
 351 print "--- Or with a field proxy that easy the writing of operations."
 352 fieldProxy0 = FieldProxy(fieldHandler0)
 353 fieldProxy1 = FieldProxy(fieldHandler1)
 354
 355 resHandler = fieldProxy0 + fieldProxy1
 356 if resHandler is None:
 357     print "Error: result handler is None!"
 358 else:
 359     print "--- Result handler:\n%s" % resHandler
 360     print "OK"
 361
 362 print
 363
 364 print "======================================================================"
 365 print "           %d. Test Post-Pro " % step; step+=1
 366 print "======================================================================"
 367
 368 import VISU
 369
 370 medFileName = "pointe.med"
 371 medFile = os.path.join(os.getenv('DATA_DIR'), 'MedFiles', medFileName)
 372
 373 aMeshName = "maa1"
 374 anEntity = VISU.NODE
 375 field_name = "fieldnodedouble"
 376
 377 if salome.hasDesktop(): # in gui mode
 378
 379     import visu_gui
 380     visu = salome.lcc.FindOrLoadComponent("FactoryServer", "VISU")
 381
 382     print
 383     print "--- Import med file %s to the VISU ..." % medFile
 384     result2 = visu.ImportFile(medFile);
 385     if not result2:
 386         raise RuntimeError, "Can't import file"
 387     print "OK"
 388
 389     print
 390     print "--- Create mesh presentation ..."
 391     mesh2 = visu.MeshOnEntity(result2, aMeshName, anEntity);
 392     if not mesh2:
 393         raise RuntimeError, "Can't create mesh presentation"
 394     print "OK"
 395
 396     print
 397     print "--- Create scalar map ..."
 398     scalarMap2 = visu.ScalarMapOnField(result2, aMeshName, anEntity, field_name, 3)
 399     if not scalarMap2:
 400         raise RuntimeError, "Can't create scalar map"
 401     print "OK"
 402     pass
 403
 404 else: # not in gui mode, visu can not be tested
 405
 406     print
 407     print "VISU module requires SALOME to be running in GUI mode."
 408     print
 409     print "Skipping test for VISU..."
 410     pass
 411
 412 # ---- update object browser
 413 if salome.hasDesktop():
 414     salome.sg.updateObjBrowser(1);