src/examples/itm_grid_example7_tetrahedra.f90

Go to the documentation of this file.

program itm_grid_example7_tetrahedra

  !> Triangle grid example, where for the node positions and for every triangles
  !> the node indices of the corners are given
  !> 
  !> The used grid is described in detail here on the ITM Documentation Website (https://www.efda-itm.eu/ITM/html/)
  !> Then go to IMP3 -> IMP3 general grid description & service library -> Example grids -> Example grid #2
  !> Direct link (might be broken): https://www.efda-itm.eu/ITM/html/imp3_gridexamples.html#imp3_gridexamples_5

  use itm_types         ! ITM standard Fortran data types, e.g. real(R8)
  use euITM_schemas     ! ITM data structure/CPO definitions
  use euITM_routines    ! ITM UAL routines
  use itm_grid_common   ! ITM grid service library: constant definitions like COORDTYPE_R, COORDTYPE_Z
  use itm_grid_simplex

  implicit none

  ! parameters
  integer, parameter :: SHOTNUM = 9007
  integer, parameter :: RUNNUM = 0

  integer, parameter :: NDIM = 3      ! We discretize the 3d x,y,z space
  integer, parameter :: NSPACE = 1    ! We use one space to do so

  integer, parameter :: NTETRA = 2    ! Number of tetrahedra
  integer, parameter :: NNODES = 5    ! Number of nodes/points


  ! variables

  ! The node positions as (R,Z) vectors
  real(R8) :: nodes(NNODES, NDIM) 
  ! The tetrahedra, defined by the indices of their three corners
  integer :: tetrahedra(NTETRA, NDIM+1)

  type (type_edge),pointer :: edgecpo(:) => null()
  type (type_complexgrid) :: grid
  integer :: idx, i
  real(R8) :: cellData(NTETRA)


  ! === 1. Set up CPO ===

  allocate( edgecpo(1) )    
  allocate( edgecpo(1)%datainfo%dataprovider(1) )
  edgecpo(1)%datainfo%dataprovider="IMP3"
  allocate( edgecpo(1)%codeparam%codename(1) )
  edgecpo(1)%codeparam%codename(1)="itm_grid_example2_tetrahedra_servicelibrary"
  edgecpo(1)%time = 0.0_R8

  ! === 2. Set up grid ===

  ! Node positions in the R,Z plane
  nodes( 1, : ) = (/ 0.0, 0.0, 0.0 /)  
  nodes( 2, : ) = (/ 1.0, 0.0, 0.0 /)
  nodes( 3, : ) = (/ 0.5, 1.0, 0.0 /)
  nodes( 4, : ) = (/ 0.5, 0.5, 1.0 /)
  nodes( 5, : ) = (/ 0.5, 0.5,-1.0 /)
  
  ! Tetrahedra defined by their corner nodes
  tetrahedra( 1, : ) = (/ 1, 2, 3, 4 /)
  tetrahedra( 2, : ) = (/ 1, 2, 3, 5 /)
  
  ! TODO: write grid to CPO
  call gridSetupSimplex( edgecpo(1)%grid, (/ COORDTYPE_X, COORDTYPE_Y, COORDTYPE_Z /), &
      & nodes, tetrahedra )

  ! === 4. Write some fake scalar data to the edge cpo ===

  ! Make up some data
  cellData = (/ (i,i=1,NTETRA) /)

  ! TODO: write data to CPO


  ! === 5. Write the edge CPO to the UAL ===

  write (*,*) "Example 7: writing to shot ", SHOTNUM, ", run ", RUNNUM
  call euitm_create( 'euitm', SHOTNUM, RUNNUM, 0, 0, idx)
  call euitm_put(idx, "edge", edgecpo)
  call euitm_deallocate(edgecpo)
  call euitm_close(idx)
  
end program itm_grid_example7_tetrahedra