src/TEST_PY/test_unit/test_disco.py

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