Salome HOME
3359cdb2d8532900146032d048fe741c31b22b5e
[modules/hexablock.git] / doc / test_doc / hemisphere / rind.py
1 # -*- coding: utf-8 -*-
2 ####### Rind Grid Test ##########
3
4 import hexablock
5
6
7 doc  = hexablock.addDocument ("Rind Grid Test")
8
9
10 # Simple Rind -----
11
12
13 nr = 4
14 na = 8
15 nh = 12
16
17 rind0 = doc.makeRindTop(nr, na, nh)
18 rind0.saveVtk("makeRindTop.vtk")
19
20 # Uniform Rind -----
21
22 center  = doc.addVertex (0, 0, 0)
23 vplan = doc.addVertex (0, 0, -0.5)
24 vx    = doc.addVector (1, 0, 0)
25 vz    = doc.addVector (0, 0, 1)
26 rtrou = 1
27 rint  = 8
28 rext  = 10
29 angle = 180
30
31 rind1 = doc.makeRindUni(center, vx, vz, rtrou, rint, rext, angle, vplan, nr, na, nh)
32 rind1.saveVtk("makeRindUni.vtk")
33
34 # Custom Rind -----
35
36 dr     = (rext-rtrou)/nr
37 dtheta = angle/na
38 dphi   = 180.0/nh
39 trad = []
40 tang = []
41 tphi = []
42
43 trad.append(rtrou)
44 for nro in range(nr+1):
45   trad.append(rint + nro*dr)
46   
47 for nro in range(na+1):
48   tang.append(nro*dtheta)
49   
50 for nro in range(nh+1):
51   tphi.append(-90 + nro*dphi)
52
53 rind2 = doc.makeRind(center, vx, vz, trad, tang, tphi)
54 rind2.saveVtk("makeRind.vtk")