doc/test_doc/joinQuad/join_quad.py

   1 # -*- coding: utf-8 -*-
   2 ####### Test for Join Quad ###############
   3
   4 import hexablock
   5
   6
   7
   8 doc = hexablock.addDocument("Join Quad Test")
   9
  10 dimx = 11
  11 dimy = 11
  12 dimz = 2
  13
  14 grid1 = doc.makeCartesianTop (dimx,dimy,dimz)
  15
  16 orig2 = doc.addVertex(dimx/2.0,0,8)
  17 vectj = doc.addVector (0,1,0);
  18 vecti = doc.addVector (1,0,0);
  19 grid2 = doc.makeCylinderUni (orig2, vecti, vectj, 1.0, 2.0, 180.0, 11.0, dimz,dimy,dimx)
  20
  21 mx = dimx/2
  22 my = dimy/2
  23 prems = grid1.getQuadIJ (mx, my, dimz)
  24 cible = grid2.getQuadJK (dimz, mx, my)
  25
  26 va1 = prems.getVertex (0)
  27 va2 = prems.getVertex (1)
  28
  29 vb1 = cible.getVertex (1)
  30 vb2 = cible.getVertex (2)
  31
  32 hauteur = 5
  33
  34 liste = []
  35
  36 liste.append(prems)
  37 for nx in range(mx):
  38     if nx!=mx:
  39         liste.append(grid1.getQuadIJ(nx, my, dimz))
  40
  41 for ny in range(dimy):
  42     if (ny!=my):
  43         liste.append(grid1.getQuadIJ (mx, ny, dimz))
  44
  45 doc.saveVtk ("jointQuad1.vtk")
  46 joint = doc.joinQuadsUni (liste, cible, va1,vb1,va2,vb2, hauteur)
  47 doc.saveVtk ("jointQuad2.vtk")