src/TEST_PY/test_unit/test_cylinders.py

   1 # -*- coding: latin-1 -*-
   2 # Copyright (C) 2009-2013  CEA/DEN, EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20
  21 # Hexa : Creation d'hexaedres
  22
  23 import hexablock
  24 import os
  25 import geompy
  26
  27 #---+----1----+----2----+----3----+----4----+----5----+----6----+----7----+----8
  28
  29
  30 # ======================================================= test_cylindricals
  31 def test_cylindricals () :
  32
  33     doc  = hexablock.addDocument ("cyl:hexas")
  34
  35     orig = doc.addVertex (0, 0, 0)
  36     vz   = doc.addVector (0, 0, 1)
  37     vx   = doc.addVector (1 ,0, 0)
  38
  39
  40     tdr = [ 1, 1, 1, 1 ]
  41     tda = [ 45, 45,  45, 45,  45, 45,  45, 45 ]
  42     tdl = [ 1, 1, 1  ]
  43
  44     tdr = [ 1, 2, 0.5 ]
  45     tda = [ 30, 30, 30, 10, 10, 10,  10, 10, 10,  10, 10, 10,
  46             30, 30, 30, 10, 10, 10,  10, 10, 10,  10, 10, 10 ]
  47     tdl = [ 1, 2, 3  ]
  48
  49
  50     c1 = doc.makeCylindricals (orig, vx,vz, tdr, tda, tdl, False)
  51
  52     doc.saveVtk ("cylindricals.vtk")
  53     return doc
  54
  55 # ======================================================= test_cylindrical
  56 def test_cylindrical () :
  57
  58     doc  = hexablock.addDocument ("cyl:hexas")
  59
  60     orig = doc.addVertex (0, 0, 0)
  61     vz   = doc.addVector (0, 0, 1)
  62     vx   = doc.addVector (1 ,0, 0)
  63
  64     nr = 3
  65     na = 8
  66     nl = 3
  67     c1 = doc.makeCylindrical (orig, vx,vz, 1.0, 360.0, 1.0, nr,na,nl, False)
  68
  69
  70     doc.saveVtk ("cylindricals.vtk")
  71     return doc
  72
  73
  74 # ======================================================= test_cone
  75 def test_cone () :
  76
  77     doc  = hexablock.addDocument ("cyl:hexas")
  78
  79     orig = doc.addVertex (0, 0, 0)
  80     ori2 = doc.addVertex (0, 0, 10)
  81     vz   = doc.addVector (0, 0, 1)
  82     vx   = doc.addVector (1 ,0, 0)
  83
  84     nr = 2
  85     na = 8
  86     nl = 1
  87     c1 = doc.makeCylindrical (orig, vx,vz, 2.0, 360.0, 5.0, nr,na,nl, False)
  88     c2 = doc.makeCylindrical (ori2, vx,vz, 1.0, 360.0, 5.0, nr,na,nl, False)
  89
  90     vh0 = c2.getVertexIJK (0, 0, 0)
  91     vh1 = c2.getVertexIJK (1, 0, 0)
  92
  93     vb0 = c1.getVertexIJK (0, 0, nl)
  94     vb1 = c1.getVertexIJK (1, 0, nl)
  95
  96     qcible = c2.getQuadIJ (0, 0, 0)
  97     qliste = []
  98     for ni in range (nr) :
  99         for nj in range (na) :
 100             quad = c1.getQuadIJ (ni, nj, nl)
 101             qliste.append (quad)
 102             quad .setColor (3);
 103
 104     qcible.setColor (3);
 105     vb0.setColor (5);
 106     vh0.setColor (5);
 107     vb1.setColor (5);
 108     vh1.setColor (5);
 109
 110     doc.joinQuads  (qliste, qcible, vb0, vh0, vb1, vh1, 1)
 111
 112     doc.saveVtk ("cone.vtk")
 113     return doc
 114
 115 # ================================================================= Begin
 116
 117 ####  doc = test_cylindricals  ()
 118 ####  doc = test_cylindrical  ()
 119 doc = test_cone  ()
 120
 121 law = doc.addLaw("Uniform", 4)
 122
 123 for j in range(doc.countPropagation()):
 124     propa = doc.getPropagation(j)
 125     propa.setLaw(law)
 126
 127 mesh_hexas = hexablock.mesh(doc)
 128
 129