CDMATH/tests/swig/test_VTK_routines.py

   1 #!/usr/bin/env python
   2 # -*-coding:utf-8 -*-
   3
   4 from VTK_routines import *
   5 import cdmath
   6 from math import sqrt
   7
   8 #Meshes and fields initialisation
   9 #================================
  10
  11 #cell field on 2D structured mesh
  12 M1 = cdmath.Mesh(0.0, 1.0, 10, 0., 1., 5)
  13
  14 field1 = cdmath.Field("test field 1", cdmath.CELLS, M1, 1)
  15 for j in range(field1.getNumberOfComponents()):
  16     for i in range(field1.getNumberOfElements()):
  17         field1[i, j] = i + j
  18
  19 fileNameVTK1 = "2D_structured_cell_field"
  20 field1.writeVTK(fileNameVTK1)
  21
  22 #node field on 2D unstructured mesh
  23 M2 = cdmath.Mesh("meshSquare.med")
  24 field2 = cdmath.Field("test field 2", cdmath.NODES, M2, 1)
  25 for j in range(field2.getNumberOfComponents()):
  26     for i in range(field2.getNumberOfElements()):
  27         field2[i, j] = i + j
  28
  29 fileNameVTK2 = "2D_unstructured_node_field"
  30 field2.writeVTK(fileNameVTK2)
  31
  32 #node field on 3D unstructured mesh
  33 M3 = cdmath.Mesh("meshCube.med")
  34 field3 = cdmath.Field("test field 3", cdmath.NODES, M3, 1)
  35 for j in range(field3.getNumberOfComponents()):
  36     for i in range(field3.getNumberOfElements()):
  37         field3[i, j] = i + j
  38
  39 fileNameVTK3 = "3D_unstructured_node_field"
  40 field3.writeVTK(fileNameVTK3)
  41
  42 #node field on sphere with unstructured mesh
  43 M4 = cdmath.Mesh("meshSphere.med")
  44 field4 = cdmath.Field("test field 4", cdmath.NODES, M4, 1)
  45 for j in range(field4.getNumberOfComponents()):
  46     for i in range(field4.getNumberOfElements()):
  47         field4[i, j] = i + j
  48
  49 fileNameVTK4 = "Sphere_unstructured_node_field"
  50 field4.writeVTK(fileNameVTK4)
  51
  52 #cell field on 3D structured mesh
  53 M5 = cdmath.Mesh(0.0, 1.0, 4, 0.0, 1.0, 4, 0.0, 1.0, 4)
  54 field5 = cdmath.Field("testfield5", cdmath.CELLS, M5, 1)
  55 for j in range(field5.getNumberOfComponents()):
  56     for i in range(field5.getNumberOfElements()):
  57         field5[i, j] = i + j
  58
  59 fileNameVTK5 = "3D_structured_cell_field"
  60 field5.writeVTK(fileNameVTK5)
  61
  62 #2D tests
  63 #===========================================
  64 point1=[1.,0.,0.]
  65 point2=[0.,1.,0.]
  66 resolution=100
  67
  68 outputFileName="Extract_VTK_over_line_"+fileNameVTK1+".csv"
  69 Extract_VTK_data_over_line_to_txt_file('2D_structured_cell_field_0.vtu', outputFileName, point1, point2, resolution)
  70 print( "Extract_VTK_over_line ok")
  71
  72 outputFileName="Extract_field_over_line_"+fileNameVTK2+".csv"
  73 Extract_field_data_over_line_to_txt_file(field2, point1, point2, resolution, outputFileName)
  74 print( "Extract_field_over_line ok")
  75
  76 point=[0.5,0.5,0.5]
  77 normal=[-0.5,-0.5,-0.5]
  78 outputFileName="Slice_VTK_data_to_txt_file_"+fileNameVTK3+".csv"
  79 Slice_VTK_data_to_txt_file(fileNameVTK3+'_0.vtu', outputFileName, point, normal,resolution )
  80 print( "Slice_VTK_data_to_txt_file ok")
  81
  82 outputFileName="Slice_field_data_to_txt_file_"+fileNameVTK4+".csv"
  83 Slice_field_data_to_txt_file(field4, outputFileName, point, normal,resolution)
  84 print( "Slice_field_data_to_txt_file ok")
  85
  86 outputFileName="Slice_VTK_data_to_VTK_"+fileNameVTK5+'_0.vtu'
  87 Slice_VTK_data_to_VTK(fileNameVTK5+'_0.vtu',outputFileName,point, normal,resolution )
  88 print( "Slice_VTK_data_to_VTK ok")
  89
  90 outputFileName="Clip_VTK_data_to_VTK_"+fileNameVTK5+'_0.vtu'
  91 Clip_VTK_data_to_VTK(fileNameVTK5+'_0.vtu',outputFileName,point, normal,resolution )
  92 print( "Clip_VTK_data_to_VTK ok")
  93
  94 inputFileName="Slice_VTK_data_to_VTK_"+fileNameVTK5
  95 outputFileName="Save_Slice_VTK_data_to_VTK_"+fileNameVTK5
  96 Save_VTK_data_to_picture_file(inputFileName+'_0.vtu',"testfield5","CELLS",outputFileName)
  97 print( "Save_VTK_Slice_data_to_picture_file ok")
  98
  99 inputFileName="Clip_VTK_data_to_VTK_"+fileNameVTK5
 100 outputFileName="Save_Clip_VTK_data_to_VTK_"+fileNameVTK5
 101 Save_VTK_data_to_picture_file(inputFileName+'_0.vtu',"testfield5","CELLS",outputFileName)
 102 print( "Save_VTK_Clip_data_to_picture_file ok")
 103
 104 outputFileName="Save_VTK_data_to_picture_file_"+fileNameVTK5
 105 Save_VTK_data_to_picture_file(fileNameVTK5+'_0.vtu',"testfield5","CELLS",outputFileName)
 106 print( "Save_VTK_data_to_picture_file ok")