feSpace_visu.f90

Go to the documentation of this file.

program fespace_visu

  use choral
  use choral_constants

  implicit none

  !! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  !! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  !!
  !!      VARAIBLE DEFINITION : BEGIN
  !!
  !!  

  !! verb    = verbosity level
  !!
  integer, parameter :: verb = 2

  !! fe        = finite element type
  !!
  integer, parameter      :: fe = fe_p2_2d

  !! msh_file  = mesh file name
  !! msh       = mesh 
  !! X_h       = finite element space
  !!
  character(len=150)      :: msh_file
  type(mesh)              :: msh
  type(fespace)           :: X_h


  !! Variable to store a finite element function
  !!
  real(RP), dimension(:), allocatable :: f_h

  !! output files
  !!
  character(len=150)      :: feSpace_file, str
  !!
  !!   VARAIBLE DEFINITION : END
  !!
  !! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  !! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

  call choral_init(verb=verb)

  write(*,*)
  write(*,*)'mesh_visu: start'
  write(*,*)""
  write(*,*)"    ASSEMBLE A MESH "
  write(*,*)""
  write(*,*)"    and visualise it with gmsh"
  write(*,*)""

  !!
  !!  settings: choose a mesh
  !!
  ! msh_file = trim(GMSH_DIR)//"disk/disk1_1.msh"
  msh_file = trim(gmsh_dir)//"testMesh/square.msh"
  ! msh_file = trim(GMSH_DIR)//"sphere/sphere1_1.msh"
  ! msh_file = trim(GMSH_DIR)//"sphere/sphere2_1.msh"


  write(*,*) ""
  write(*,*)"================================== ASSEMBLING THE MESH "
  msh = mesh(trim(msh_file), "gmsh")
  if (verb>1) call print(msh)


  write(*,*) ""
  write(*,*)"================================== ASSEMBLING THE&
       & FINITE ELEMENT SPACE X_h"
  x_h = fespace(msh)
  call set(x_h, fe)
  call assemble(x_h)


  write(*,*) ""
  write(*,*)"================================== DISPLAY THE&
       & FINITE ELEMENT NODES"
  fespace_file = "feSpace.msh"
  call write(x_h, fespace_file, "gmsh")
  call system("gmsh -option &
       &"//trim(gmsh_dir)//"gmsh-options-mesh "//trim(fespace_file))


  write(*,*) ""
  write(*,*)"================================== DISPLAY A&
       & FUNCTION INTERPOLATED ON X_h"
  !!
  !! 1. Interpolate 'f' to the finite element function f_h \in X_h
  call interp_scal_func(f_h, f, x_h)            
  !!
  !! 2. Add the finite element function 'f_h' to the file 'feSpace_file'
  call gmsh_addview(x_h, f_h, fespace_file, &
       & 'Interpolant of the Function f', 0._rp, 0)
  !!
  !! 3. Visualise
  str='gmsh -option '//trim(gmsh_dir)//'gmsh-options-view '&
       &//trim(fespace_file)
  call system(trim(str))


contains

 function f(x)
    real(RP), dimension(3), intent(in) :: x
    real(RP) :: f

    f = sin(2*pi*x(1)) * sin(2*pi*x(2))

  end function f


  
end program fespace_visu