doc/test_doc/quad_revolution/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 #### Quad Revolution Test #####
  22
  23 import hexablock
  24 import os
  25
  26
  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         Save the model in a vtk file
  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             liste.append (cell);
  72
  73     center = doc.addVertex (0, -10, 0);
  74     axis   = doc.addVector (1, 0, 0);
  75     angles = [5, 10, 15, 20, 30, 20, 15, 10, 5 ]
  76
  77     vtk.save ()
  78     bloc = doc.revolutionQuads  (liste, center, axis, angles);
  79     vtk.save ()
  80     return doc
  81
  82 # ================================================================= Test
  83 doc = test_revolution ()
  84