src/TEST_PY/test_unit/test_disco.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 # Hexa : Creation d'hexaedres
  22
  23 import hexablock
  24 import os
  25 import geompy
  26
  27 #---+----1----+----2----+----3----+----4----+----5----+----6----+----7----+----8
  28
  29 # ======================================================= test_disco_edges
  30 def test_disco_edges () :
  31
  32     nom  = "test_disco_edges"
  33     doc  = hexablock.addDocument(nom)
  34     doc.setLevel (0)
  35
  36     ori = doc.addVertex ( 0, 0, 0)
  37     vz  = doc.addVector ( 0, 0, 1)
  38     vy  = doc.addVector ( 0, 1, 1)
  39     vx  = doc.addVector ( 1 ,0, 0)
  40
  41     size_x = 1
  42     size_y = 2
  43     size_z = 2
  44
  45     grid = doc.makeCartesian (ori, vx, vy, vz, size_x, size_y, size_z)
  46
  47     t_hexas = []
  48     t_edges = []
  49     doc.setLevel (1)
  50
  51     for nk in range (size_z) :
  52         t_hexas.append (grid.getHexaIJK (0,0,nk))
  53         t_edges.append (grid.getEdgeK   (0,1,nk))
  54
  55     disco_edges =  doc.disconnectEdges (t_hexas, t_edges)
  56     return doc
  57
  58 # ================================================================= Begin
  59
  60 doc = test_disco_edges  ()
  61
  62 law = doc.addLaw("Uniform", 4)
  63
  64 for j in range(doc.countPropagation()):
  65     propa = doc.getPropagation(j)
  66     propa.setLaw(law)
  67
  68 mesh_hexas = hexablock.mesh(doc)
  69 muv, mue, muq, muh = hexablock.dump(doc, mesh_hexas)