doc/test_doc/cartesian_grid/cartesian_grid.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2009-2023  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 ####### Test Cartesian Grid #################
  22
  23 import hexablock
  24
  25
  26 doc  = hexablock.addDocument ("Cartesian Grid Test")
  27
  28 # Simple Cartesian Grid
  29
  30 nbX = 3
  31 nbY = 4
  32 nbZ = 5
  33
  34 grid0 = doc.makeCartesianTop(nbX, nbY, nbZ)
  35 grid0.saveVtk("makeCartesianTop.vtk")
  36
  37 # Uniform Cartesian Grid
  38
  39 orig1 = doc.addVertex(10, 0, 0)
  40
  41 vecX = doc.addVector(1, 0, 0)
  42 vecY = doc.addVector(0, 1, 0)
  43 vecZ = doc.addVector(0, 0, 1)
  44
  45 lenX = 5
  46 lenY = 3.5
  47 lenZ = 2
  48
  49 grid1 = doc.makeCartesianUni(orig1, vecX, vecY, vecZ, lenX, lenY, lenZ, nbX, nbY, nbZ)
  50 grid1.saveVtk("makeCartesianUni.vtk")
  51
  52 # Custom Cartesian Grid
  53
  54 orig2 = doc.addVertex(20, 0, 0)
  55
  56 tx = [] # a list of x coordinates
  57 ty = [] # a list of y coordinates
  58 tz = [] # a list of z coordinates
  59 for i in range(6):
  60   tx.append(i+1)
  61   ty.append(i+1)
  62   tz.append(i+1)
  63 # tx=ty=tz=[1,2,3,4,5,6]
  64
  65 grid2 = doc.makeCartesian(orig2, vecX, vecY, vecZ, tx, ty, tz)
  66 grid2.saveVtk("makeCartesian.vtk")