src/TEST_PY/test_unit/revolution.py

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