CDMATH/tests/swig/test_PV_routines.py

   1 #!/usr/bin/env python
   2 # -*-coding:utf-8 -*-
   3
   4 from PV_routines import *
   5 import cdmath
   6
   7 #Meshes and fields initialisation
   8 #================================
   9
  10 #cell field on 2D structured mesh
  11 M1 = cdmath.Mesh(0.0, 1.0, 10, 0., 1., 5)
  12 fieldName1="testfield1"
  13 field1 = cdmath.Field(fieldName1, cdmath.CELLS, M1, 1)
  14 for j in range(field1.getNumberOfComponents()):
  15     for i in range(field1.getNumberOfElements()):
  16         field1[i, j] = i + j
  17
  18 fileNameVTK1 = "2D_structured_cell_field"
  19 field1.writeVTK(fileNameVTK1)
  20
  21 #node field on 2D unstructured mesh
  22 M2 = cdmath.Mesh("meshSquare.med")
  23 fieldName2="testfield2"
  24 field2 = cdmath.Field(fieldName2, 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 fieldName3="testfield3"
  35 field3 = cdmath.Field(fieldName3, cdmath.NODES, M3, 1)
  36 for j in range(field3.getNumberOfComponents()):
  37     for i in range(field3.getNumberOfElements()):
  38         field3[i, j] = i + j
  39
  40 fileNameVTK3 = "3D_unstructured_node_field"
  41 field3.writeVTK(fileNameVTK3)
  42
  43 #node field on sphere with unstructured mesh
  44 M4 = cdmath.Mesh("meshSphere.med")
  45 fieldName4="testfield4"
  46 field4 = cdmath.Field(fieldName4, cdmath.NODES, M4, 1)
  47 for j in range(field4.getNumberOfComponents()):
  48     for i in range(field4.getNumberOfElements()):
  49         field4[i, j] = i + j
  50
  51 fileNameVTK4 = "Sphere_unstructured_node_field"
  52 field4.writeVTK(fileNameVTK4)
  53
  54 #cell field on 3D structured mesh
  55 M5 = cdmath.Mesh(0.0, 1.0, 4, 0.0, 1.0, 4, 0.0, 1.0, 4)
  56 fieldName5="testfield5"
  57 field5 = cdmath.Field(fieldName5, cdmath.CELLS, M5, 1)
  58 for j in range(field5.getNumberOfComponents()):
  59     for i in range(field5.getNumberOfElements()):
  60         field5[i, j] = i + j
  61
  62 fileNameVTK5 = "3D_structured_cell_field"
  63 field5.writeVTK(fileNameVTK5)
  64
  65 #2D tests
  66 #===========================================
  67 point1=[1.,0.,0.]
  68 point2=[0.,1.,0.]
  69 resolution=100
  70
  71 outputFileName="Extract_PV_over_line_"+fileNameVTK1+".csv"
  72 Extract_PV_data_over_line_to_txt_file('2D_structured_cell_field_0.vtu', outputFileName, point1, point2, resolution)
  73 print( "Extract_VTK_over_line ok")
  74
  75 point=[0.5,0.5,0]
  76 normal=[1,1,0]
  77 outputFileName="Slice_PV_data_to_txt_file_"+fileNameVTK3+".csv"
  78 Slice_PV_data_to_txt_file(fileNameVTK3+'_0.vtu', outputFileName, point, normal,resolution )
  79 print( "Slice_PV_data_to_txt_file ok")
  80
  81 #outputFileName="Slice_field_data_to_txt_file"+fileNameVTK4+".csv"
  82 #Slice_PV_field_data_to_txt_file(field4, outputFileName, point, normal,resolution)
  83 #print "Slice_field_data_to_txt_file ok"
  84
  85 outputFileName="Clip_PV_data_to_VTK_"+fileNameVTK5
  86 inputFileName="Clip_VTK_data_to_VTK_"+fileNameVTK5
  87 Save_PV_data_to_picture_file(inputFileName+'_0.vtu',fieldName5,'CELLS',outputFileName)
  88 print( "Save_PV_Clip_data_to_picture_file ok")
  89
  90 outputFileName="Slice_PV_data_to_VTK_"+fileNameVTK5
  91 inputFileName="Slice_VTK_data_to_VTK_"+fileNameVTK5
  92 Save_PV_data_to_picture_file(inputFileName+'_0.vtu',fieldName5,'CELLS',outputFileName)
  93 print( "Save_PV_Slice_data_to_picture_file ok")
  94
  95 outputFileName="Save_PV_data_to_picture_file_"+fileNameVTK2
  96 Save_PV_data_to_picture_file(fileNameVTK2+'_0.vtu',fieldName2,'NODES',outputFileName)
  97 print( "Save_PV_data_to_picture_file " + fileNameVTK2+ " ok")
  98
  99 outputFileName="Save_PV_data_to_picture_file_"+fileNameVTK3
 100 Save_PV_data_to_picture_file(fileNameVTK3+'_0.vtu',fieldName3,'NODES',outputFileName)
 101 print( "Save_PV_data_to_picture_file " + fileNameVTK3+ " ok")
 102
 103 outputFileName="Save_PV_data_to_picture_file_"+fileNameVTK4
 104 Save_PV_data_to_picture_file(fileNameVTK4+'_0.vtu',fieldName4,'NODES',outputFileName)
 105 print( "Save_PV_data_to_picture_file " + fileNameVTK4+ " ok")
 106
 107 outputFileName="Save_PV_data_to_picture_file_"+fileNameVTK5
 108 Save_PV_data_to_picture_file(fileNameVTK5+'_0.vtu',fieldName5,'CELLS',outputFileName)
 109 print( "Save_PV_data_to_picture_file " + fileNameVTK5+ " ok")