src/TEST_PY/test_unit/revolution.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 # ================================================================= save_schema
  29 class FileVtk :
  30     def __init__ (self, doc, radical):
  31         self.document = doc
  32         self.count    = 0
  33         self.radical  = radical
  34
  35     def save (self) :
  36         """
  37         sauvegarde vtk du modele de bloc
  38         """
  39         self.count += 1
  40         file_name = os.path.join(self.radical + str(self.count) + '.vtk')
  41         self.document.saveVtk(file_name)
  42
  43 # ======================================================= make_grid
  44 def make_grid (doc, nr, na, nl) :
  45
  46     ori  = doc.addVertex ( 0, 0, 0)
  47     vx   = doc.addVector ( 1 ,0, 0)
  48     vz   = doc.addVector ( 0, 0, 1)
  49
  50     dr = 1
  51     da = 360
  52     dl = 1
  53
  54     grid = doc.makeCylindrical (ori, vx,vz, dr,da,dl, nr,na,nl, False)
  55     return grid
  56
  57
  58 # ========================================================== test_revolution
  59 def test_revolution () :
  60     doc  = hexablock.addDocument ("default")
  61     vtk  = FileVtk (doc, "Revolution");
  62     nr = 1
  63     na = 6
  64     nl = 1
  65     grid = make_grid (doc, nr, na, nl)
  66     vtk.save ()
  67
  68     liste = [ ]
  69     for nx in range (nr) :
  70         for ny in range (na) :
  71             cell = grid.getQuadIJ (nx, ny, nl)
  72             print " ... cell = ", cell
  73             liste.append (cell);
  74
  75     center = doc.addVertex (0, -10, 0);
  76     axis   = doc.addVector (1, 0, 0);
  77     angles = [5, 10, 15, 20, 30, 20, 15, 10, 5 ]
  78
  79     vtk.save ()
  80     bloc = doc.revolutionQuads  (liste, center, axis, angles);
  81     vtk.save ()
  82     return doc
  83
  84 # ========================================================== test_prism
  85 def test_prism () :
  86     doc  = hexablock.addDocument("prism")
  87     vtk  = FileVtk (doc, "prism");
  88     nr = 1
  89     na = 6
  90     nl = 1
  91     grid = make_grid (doc, nr, na, nl)
  92     vtk.save ()
  93
  94     liste = [ ]
  95     for nx in range (nr) :
  96         for ny in range (na) :
  97             cell = grid.getQuadIJ (nx, ny, nl)
  98             print " ... cell = ", cell
  99             liste.append (cell);
 100
 101     axis = doc.addVector (1, 1, 1);
 102
 103     bloc = doc.prismQuads  (liste, axis, 3)
 104     vtk.save ()
 105     return doc
 106
 107 # ========================================================== test_prism
 108 def test_prism_vec () :
 109     doc  = hexablock.addDocument("prism_vec")
 110     vtk  = FileVtk (doc, "prism_vec");
 111     nr = 1
 112     na = 6
 113     nl = 1
 114     grid = make_grid (doc, nr, na, nl)
 115     vtk.save ()
 116
 117     liste = [ ]
 118     for nx in range (nr) :
 119         for ny in range (na) :
 120             cell = grid.getQuadIJ (nx, ny, nl)
 121             print " ... cell = ", cell
 122             liste.append (cell);
 123
 124     axis = doc.addVector (1, 1, 1);
 125     hauteurs = [1, 3, 7, 15]
 126
 127     bloc = doc.prismQuadsVec (liste, axis, hauteurs, 0)
 128     vtk.save ()
 129     return doc
 130
 131 # ================================================================= Begin
 132
 133 ###  doc = test_revolution ()
 134 ###  doc = test_prism ()
 135 doc = test_prism_vec ()
 136
 137 law = doc.addLaw("Uniform", 3)
 138
 139 for j in range(doc.countPropagation()):
 140     propa = doc.getPropagation(j)
 141     propa.setLaw(law)
 142
 143 mesh_hexas = hexablock.mesh(doc, "maillage:hexas")
 144