Salome HOME
DCQ : Merge with Ecole_Ete_a6.
[modules/smesh.git] / src / SMESH_SWIG / SMESH_mechanic.py
index 7d695d7a36f5c0ef6d5cd238fbd01e80806888c4..3011c0c827b1cbee036c2e7012f997e7e81088d1 100644 (file)
@@ -1,31 +1,44 @@
-#==============================================================================
-#  File      : SMESH_withHole.py
-#  Created   : 26 august 2002
-#  Author    : Lucien PIGNOLONI
-#  Project   : SALOME
-#  Copyright : Open CASCADE, 2002
+#  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+#  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS 
+# 
+#  This library is free software; you can redistribute it and/or 
+#  modify it under the terms of the GNU Lesser General Public 
+#  License as published by the Free Software Foundation; either 
+#  version 2.1 of the License. 
+# 
+#  This library is distributed in the hope that it will be useful, 
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of 
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
+#  Lesser General Public License for more details. 
+# 
+#  You should have received a copy of the GNU Lesser General Public 
+#  License along with this library; if not, write to the Free Software 
+#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA 
+# 
+#  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
+#
+#
+#
+#  File   : SMESH_withHole.py
+#  Author : Lucien PIGNOLONI
+#  Module : SMESH
 #  $Header$
-#==============================================================================
 
-import SMESH
-import smeshpy
 import salome
-from salome import sg
-import math
-#import SMESH_BasicHypothesis_idl
-
 import geompy
 
+geom  = salome.lcc.FindOrLoadComponent("FactoryServer", "GEOM")
+smesh = salome.lcc.FindOrLoadComponent("FactoryServer", "SMESH")
 
-
-# ---------------------------- GEOM --------------------------------------
-geom = salome.lcc.FindOrLoadComponent("FactoryServer", "GEOM")
-myBuilder = salome.myStudy.NewBuilder()
-#from geompy import gg
+geom.GetCurrentStudy(salome.myStudy._get_StudyId())
+smesh.SetCurrentStudy(salome.myStudy)
 
 smeshgui = salome.ImportComponentGUI("SMESH")
 smeshgui.Init(salome.myStudyId);
 
+import StdMeshers
+
+# ---------------------------- GEOM --------------------------------------
 ShapeTypeCompSolid = 1
 ShapeTypeSolid     = 2
 ShapeTypeShell     = 3
@@ -34,9 +47,7 @@ ShapeTypeWire      = 5
 ShapeTypeEdge      = 6
 ShapeTypeVertex    = 7
 
-
 # ---- define contigous arcs and segment to define a closed wire
-
 p1   = geom.MakePointStruct( 100.0,   0.0,  0.0 )
 p2   = geom.MakePointStruct(  50.0,  50.0,  0.0 )
 p3   = geom.MakePointStruct( 100.0, 100.0,  0.0 ) 
@@ -52,9 +63,7 @@ arc2 = geom.MakeArc( p4, p5, p6 )
 p7   = geom.MakePointStruct( 120.0, 30.0, 0.0 )
 arc3 = geom.MakeArc( p6, p7, p1 )
 
-
 # ---- define a closed wire with arcs and segment
-
 List1 = []
 List1.append( arc1 )
 List1.append( seg1 )
@@ -68,13 +77,11 @@ wire1 = geom.MakeWire( ListIOR1 )
 
 Id_wire1 = geompy.addToStudy( wire1, "wire1")
 
-
 # ---- define a planar face with wire
 WantPlanarFace = 1 #True
 face1 = geom.MakeFace( wire1, WantPlanarFace )
 Id_face1 = geompy.addToStudy( face1, "face1")
 
-
 # ---- create a shape by extrusion
 pO = geom.MakePointStruct( 0.0, 0.0,   0.0 )
 pz = geom.MakePointStruct( 0.0, 0.0, 100.0 )
@@ -82,8 +89,6 @@ pz = geom.MakePointStruct( 0.0, 0.0, 100.0 )
 prism1    = geom.MakePrism( face1, pO, pz )
 Id_prism1 = geompy.addToStudy( prism1, "prism1")
 
-
-
 # ---- create two cylinders
 
 pc1 = geom.MakePointStruct(  90.0, 50.0, -40.0 )
@@ -97,7 +102,6 @@ cyl2  = geom.MakeCylinder( pc2, vz, radius, height )
 Id_Cyl1 = geompy.addToStudy( cyl1, "cyl1" )
 Id_Cyl2 = geompy.addToStudy( cyl2, "cyl2" )
 
-
 # ---- cut with cyl1 
 shape  = geom.MakeBoolean( prism1, cyl1, 2 )
 
@@ -106,203 +110,185 @@ shape1 =  geom.MakeBoolean( shape, cyl2, 3 )
 
 Id_shape1 = geompy.addToStudy( shape1, "shape1")
 
-
+#faces = geompy.SubShapeAllSorted( shape1, ShapeTypeFace)
+#i = 0
+#for face in faces:
+#    geompy.addToStudy(face,"face_" + str(i))
+#    i = i+1
+    
 # ---- add a face sub shape in study to be meshed different
 IdSubFaceList = []
-IdSubFaceList.append(10)
+IdSubFaceList.append(1)
 sub_face = geompy.SubShapeSorted( shape1, ShapeTypeFace, IdSubFaceList )
 name     = geompy.SubShapeName( sub_face._get_Name(), shape1._get_Name() )
 
 Id_SubFace = geompy.addToStudyInFather( shape1, sub_face, name )
 
-
 # ---- add a face sub shape in study to be meshed different
 IdSubFaceL = []
-IdSubFaceL.append(7)
+IdSubFaceL.append(2)
 sub_face2 = geompy.SubShapeSorted( shape1, ShapeTypeFace, IdSubFaceL )
 name      = geompy.SubShapeName( sub_face2._get_Name(), shape1._get_Name() )
 
 Id_SubFace2 = geompy.addToStudyInFather( shape1, sub_face2, name )
 
+# ---- add a face sub shape in study to be meshed different
+IdSubFaceL = []
+IdSubFaceL.append(3)
+sub_face3 = geompy.SubShapeSorted( shape1, ShapeTypeFace, IdSubFaceL )
+name      = geompy.SubShapeName( sub_face3._get_Name(), shape1._get_Name() )
 
+Id_SubFace3 = geompy.addToStudyInFather( shape1, sub_face3, name )
 
+# ---- add a face sub shape in study to be meshed different
+IdSubFaceL = []
+IdSubFaceL.append(6)
+sub_face4 = geompy.SubShapeSorted( shape1, ShapeTypeFace, IdSubFaceL )
+name      = geompy.SubShapeName( sub_face4._get_Name(), shape1._get_Name() )
 
-# ---------------------------- SMESH --------------------------------------
-
-# ---- launch SMESH, init a Mesh with shape 'shape1'
-gen  = smeshpy.smeshpy()
-mesh = gen.Init( Id_shape1 )
+Id_SubFace4 = geompy.addToStudyInFather( shape1, sub_face4, name )
 
-idmesh = smeshgui.AddNewMesh( salome.orb.object_to_string(mesh) )
-smeshgui.SetName( idmesh, "Mesh_meca" );
-smeshgui.SetShape( Id_shape1, idmesh );
 
+# ---------------------------- SMESH --------------------------------------
 
 # ------------------------------ Length Hypothesis
 
 print "-------------------------- create Hypothesis"
 print "-------------------------- LocalLength"
-hyp1    = gen.CreateHypothesis( "LocalLength" )
-hypLen1 = hyp1._narrow( SMESH.SMESH_LocalLength )
-hypLen1.SetLength( 100.0 )
-print hypLen1.GetName()
-print hypLen1.GetId()
-print hypLen1.GetLength()
-
-idlength = smeshgui.AddNewHypothesis( salome.orb.object_to_string(hypLen1) );
-smeshgui.SetName(idlength, "Local_Length_100");
+hyp1 = smesh.CreateHypothesis("LocalLength", "libStdMeshersEngine.so")
+hyp1.SetLength( 100.0 )
+print hyp1.GetName()
+print hyp1.GetId()
+print hyp1.GetLength()
 
+idlength = salome.ObjectToID(hyp1)
+smeshgui.SetName(idlength, "Local_Length_100");
 
 print "-------------------------- NumberOfSegments"
 
-hyp2      = gen.CreateHypothesis( "NumberOfSegments" )
-hypNbSeg1 = hyp2._narrow( SMESH.SMESH_NumberOfSegments )
-hypNbSeg1.SetNumberOfSegments( 10 )
-print hypNbSeg1.GetName()
-print hypNbSeg1.GetId()
-print hypNbSeg1.GetNumberOfSegments()
+hyp2 = smesh.CreateHypothesis("NumberOfSegments", "libStdMeshersEngine.so")
+hyp2.SetNumberOfSegments( 10 )
+print hyp2.GetName()
+print hyp2.GetId()
+print hyp2.GetNumberOfSegments()
 
-idseg = smeshgui.AddNewHypothesis( salome.orb.object_to_string(hypNbSeg1) );
+idseg = salome.ObjectToID(hyp2)
 smeshgui.SetName(idseg, "NumberOfSegments_12");
 
-
 print "-------------------------- MaxElementArea"
 
-hyp3     = gen.CreateHypothesis( "MaxElementArea" )
-hypArea1 = hyp3._narrow( SMESH.SMESH_MaxElementArea )
-hypArea1.SetMaxElementArea( 25 )
-print hypArea1.GetName()
-print hypArea1.GetId()
-print hypArea1.GetMaxElementArea()
+hyp3 = smesh.CreateHypothesis( "MaxElementArea", "libStdMeshersEngine.so" )
+hyp3.SetMaxElementArea( 25 )
+print hyp3.GetName()
+print hyp3.GetId()
+print hyp3.GetMaxElementArea()
 
-idarea1 = smeshgui.AddNewHypothesis( salome.orb.object_to_string(hypArea1) );
+idarea1 = salome.ObjectToID(hyp3)
 smeshgui.SetName(idarea1, "MaxElementArea_20");
 
-
-
 print "-------------------------- MaxElementArea"
 
-hyp4     = gen.CreateHypothesis( "MaxElementArea" )
-hypArea2 = hyp4._narrow( SMESH.SMESH_MaxElementArea )
-hypArea2.SetMaxElementArea( 35 )
-print hypArea2.        GetName()
-print hypArea2.GetId()
-print hypArea2.GetMaxElementArea()
+hyp4 = smesh.CreateHypothesis("MaxElementArea", "libStdMeshersEngine.so")
+hyp4.SetMaxElementArea( 35 )
+print hyp4.GetName()
+print hyp4.GetId()
+print hyp4.GetMaxElementArea()
 
-idarea2 = smeshgui.AddNewHypothesis( salome.orb.object_to_string(hypArea2) );
+idarea2 = salome.ObjectToID(hyp4)
 smeshgui.SetName(idarea2, "MaxElementArea_30");
 
 
 print "-------------------------- Regular_1D"
 
-alg1    = gen.CreateHypothesis( "Regular_1D" )
-algo1   = alg1._narrow( SMESH.SMESH_Algo )
-listHyp =algo1.GetCompatibleHypothesis()
+alg1 = smesh.CreateHypothesis("Regular_1D", "libStdMeshersEngine.so")
+listHyp =alg1.GetCompatibleHypothesis()
 for hyp in listHyp:
     print hyp
-algoReg1d = alg1._narrow( SMESH.SMESH_Regular_1D )
-print algoReg1d.GetName()
-print algoReg1d.GetId()
+print alg1.GetName()
+print alg1.GetId()
 
-idreg1d = smeshgui.AddNewAlgorithms( salome.orb.object_to_string(algoReg1d) );
+idreg1d = salome.ObjectToID(alg1)
 smeshgui.SetName( idreg1d, "Regular_1D" );
 
-
-
 print "-------------------------- MEFISTO_2D"
 
-alg2    = gen.CreateHypothesis( "MEFISTO_2D" )
-algo2   = alg2._narrow( SMESH.SMESH_Algo )
-listHyp = algo2.GetCompatibleHypothesis()
+alg2 = smesh.CreateHypothesis("MEFISTO_2D", "libStdMeshersEngine.so")
+listHyp = alg2.GetCompatibleHypothesis()
 for hyp in listHyp:
     print hyp
-algoMef = alg2._narrow( SMESH.SMESH_MEFISTO_2D )
-print algoMef.GetName()
-print algoMef.GetId()
+print alg2.GetName()
+print alg2.GetId()
 
-idmef = smeshgui.AddNewAlgorithms( salome.orb.object_to_string(algoMef) );
+idmef = salome.ObjectToID(alg2)
 smeshgui.SetName( idmef, "MEFISTO_2D" );
 
-
-
 print "-------------------------- SMESH_Quadrangle_2D"
 
-alg3    = gen.CreateHypothesis( "Quadrangle_2D" )
-algo3   = alg3._narrow( SMESH.SMESH_2D_Algo )
-listHyp = algo3.GetCompatibleHypothesis()
+alg3 = smesh.CreateHypothesis( "Quadrangle_2D", "libStdMeshersEngine.so" )
+listHyp = alg3.GetCompatibleHypothesis()
 for hyp in listHyp:
     print hyp
-algoQad2 = alg3._narrow( SMESH.SMESH_Quadrangle_2D )
-print algoQad2.GetName()
-print algoQad2.GetId()
+print alg3.GetName()
+print alg3.GetId()
 
-idqad2 = smeshgui.AddNewAlgorithms( salome.orb.object_to_string(algoQad2) );
+idqad2 =  salome.ObjectToID(alg3)
 smeshgui.SetName( idqad2, "SMESH_Quadrangle_2D" );
 
 
-
 print "-------------------------- add hypothesis to main shape1"
 
 shape_mesh = salome.IDToObject( Id_shape1  )
-submesh    = mesh.GetElementsOnShape( shape_mesh )
-
-ret = mesh.AddHypothesis( shape_mesh, algoReg1d )   # Regular 1D/wire discretisation
-print ret
-ret = mesh.AddHypothesis( shape_mesh, algoMef )     # MEFISTO 2D
-print ret
-ret = mesh.AddHypothesis( shape_mesh, hypNbSeg1 )   # nb segments
-print ret
-ret = mesh.AddHypothesis( shape_mesh, hypArea1 )    # max area
-print ret
 
-smeshgui.SetAlgorithms( idmesh, idreg1d );  # Regular 1D/wire discretisation
-smeshgui.SetAlgorithms( idmesh, idmef );    # MEFISTO 2D
-smeshgui.SetHypothesis( idmesh, idseg );    # nb segments
-smeshgui.SetHypothesis( idmesh, idarea1 );  # max area
+mesh = smesh.CreateMesh(shape_mesh)
 
+idmesh = salome.ObjectToID(mesh) 
+smeshgui.SetName( idmesh, "Mesh_mechanic" );
 
+mesh.AddHypothesis( shape_mesh, alg1 )   # Regular 1D/wire discretisation
+mesh.AddHypothesis( shape_mesh, alg2 )     # MEFISTO 2D
 
-print "-------------------------- add hypothesis and algorith to sub face"        
+ret = mesh.AddHypothesis( shape_mesh, hyp2 )   # nb segments
+ret = mesh.AddHypothesis( shape_mesh, hyp3 )    # max area
 
-sub_face  = salome.IDToObject( Id_SubFace )
-submesh   = mesh.GetElementsOnShape( sub_face )
+print "--------Add hypothesis and algorith to sub face"        
 
-ret = mesh.AddHypothesis( sub_face, algoQad2 )  # Quadrangle 2D
-print ret
-ret = mesh.AddHypothesis( sub_face, hypArea2 )  # max area
-print ret
+#sub_face  = salome.IDToObject( Id_SubFace )
+submesh   = mesh.GetSubMesh(sub_face, "SubMeshFace")
 
-idsm2 = smeshgui.AddSubMeshOnShape( idmesh,
-                                    Id_SubFace,
-                                    salome.orb.object_to_string(submesh),
-                                    ShapeTypeFace )
+mesh.AddHypothesis( sub_face, alg3 )  # Quadrangle 2D
+mesh.AddHypothesis( sub_face, hyp4 )  # max area
 
-smeshgui.SetName(idsm2, "SubMeshFace")
-smeshgui.SetAlgorithms( idsm2, idqad2 );  # Quadrangle 2D
-smeshgui.SetHypothesis( idsm2, idarea2 ); # max area
+print "--------Add hypothesis and algorith to sub face 2"        
 
+#sub_face2  = salome.IDToObject( Id_SubFace2 )
+submesh    = mesh.GetSubMesh(sub_face2, "SubMeshFace2")
 
+mesh.AddHypothesis( sub_face2, alg3 )  # Quadrangle 2D
+ret = mesh.AddHypothesis( sub_face2, hyp4 )  # max area
 
-print "-------------------------- add hypothesis and algorith to sub face"        
+print "--------Add hypothesis and algorith to sub face 3"        
 
-sub_face2  = salome.IDToObject( Id_SubFace2 )
-submesh    = mesh.GetElementsOnShape( sub_face2 )
+#sub_face3  = salome.IDToObject( Id_SubFace3 )
+submesh    = mesh.GetSubMesh(sub_face3, "SubMeshFace3")
 
-ret = mesh.AddHypothesis( sub_face2, algoQad2 )  # Quadrangle 2D
-print ret
-ret = mesh.AddHypothesis( sub_face2, hypArea2 )  # max area
-print ret
+mesh.AddHypothesis( sub_face3, alg3 )  # Quadrangle 2D
+ret = mesh.AddHypothesis( sub_face3, hyp4 )  # max area
 
-idsm3 = smeshgui.AddSubMeshOnShape( idmesh,
-                                    Id_SubFace2,
-                                    salome.orb.object_to_string(submesh),
-                                    ShapeTypeFace )
+print "--------Add hypothesis and algorith to sub face 4"        
 
-smeshgui.SetName(idsm3, "SubMeshFace2")
-smeshgui.SetAlgorithms( idsm3, idqad2 );  # Quadrangle 2D
-smeshgui.SetHypothesis( idsm3, idarea2 ); # max area
+#sub_face4  = salome.IDToObject( Id_SubFace4 )
+submesh    = mesh.GetSubMesh(sub_face4, "SubMeshFace4")
 
+mesh.AddHypothesis( sub_face4, alg3 )  # Quadrangle 2D
+ret = mesh.AddHypothesis( sub_face4, hyp4 )  # max area
 
+smesh.Compute(mesh,shape_mesh)
 
+print "Information about the Mesh_mechanic:"
+print "Number of nodes      : ", mesh.NbNodes()
+print "Number of edges      : ", mesh.NbEdges()
+print "Number of faces      : ", mesh.NbFaces()
+print "Number of triangles  : ", mesh.NbTriangles()
+print "Number of quadrangles: ", mesh.NbQuadrangles()
 
-sg.updateObjBrowser(1);
+salome.sg.updateObjBrowser(1);