doc/test_doc/cartesian_grid/cartesian_grid.py

   1 # -*- coding: utf-8 -*-
   2 ####### Test Cartesian Grid #################
   3
   4 import hexablock
   5
   6
   7 doc  = hexablock.addDocument ("Cartesian Grid Test")
   8
   9 # Simple Cartesian Grid
  10
  11 nbX = 3
  12 nbY = 4
  13 nbZ = 5
  14
  15 grid0 = doc.makeCartesianTop(nbX, nbY, nbZ)
  16 grid0.saveVtk("makeCartesianTop.vtk")
  17
  18 # Uniform Cartesian Grid
  19
  20 orig1 = doc.addVertex(10, 0, 0)
  21
  22 vecX = doc.addVector(1, 0, 0)
  23 vecY = doc.addVector(0, 1, 0)
  24 vecZ = doc.addVector(0, 0, 1)
  25
  26 lenX = 5
  27 lenY = 3.5
  28 lenZ = 2
  29
  30 grid1 = doc.makeCartesianUni(orig1, vecX, vecY, vecZ, lenX, lenY, lenZ, nbX, nbY, nbZ)
  31 grid1.saveVtk("makeCartesianUni.vtk")
  32
  33 # Custom Cartesian Grid
  34
  35 orig2 = doc.addVertex(20, 0, 0)
  36
  37 tx = [] # a list of x coordinates
  38 ty = [] # a list of y coordinates
  39 tz = [] # a list of z coordinates
  40 for i in range(6):
  41   tx.append(i+1)
  42   ty.append(i+1)
  43   tz.append(i+1)
  44 # tx=ty=tz=[1,2,3,4,5,6]
  45
  46 grid2 = doc.makeCartesian(orig2, vecX, vecY, vecZ, tx, ty, tz)
  47 grid2.saveVtk("makeCartesian.vtk")