! Copyright (c) 2016 Tobias Schneider ! Copyright (c) 2016 Kannan Masilamani ! Copyright (c) 2016-2017 Peter Vitt ! Copyright (c) 2019 Harald Klimach ! ! Redistribution and use in source and binary forms, with or without ! modification, are permitted provided that the following conditions are met: ! ! 1. Redistributions of source code must retain the above copyright notice, this ! list of conditions and the following disclaimer. ! ! 2. Redistributions in binary form must reproduce the above copyright notice, ! this list of conditions and the following disclaimer in the documentation ! and/or other materials provided with the distribution. ! ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" ! AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE ! IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE ! DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE ! FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL ! DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR ! SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER ! CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, ! OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE ! OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. !> This UTEST is to test operation variables function pointers !! program tem_varSys_opVar_test use, intrinsic :: iso_c_binding, only: C_NEW_LINE, c_ptr, c_loc, c_f_pointer use env_module, only: rk, eps, solSpecLen, labelLen, newUnit, & & fin_env use tem_bc_prop_module, only: tem_bc_prop_type use tem_general_module, only: tem_general_type use treelmesh_module, only: treelmesh_type use tem_time_module, only: tem_time_type use tem_utestEnv_module, only: load_env use tem_tools_module, only: upper_to_lower, tem_PositionInSorted use tem_varSys_module, only: tem_varSys_init, tem_varSys_type use tem_variable_module, only: tem_variable_type, tem_variable_load use tem_grow_array_module, only: grw_intArray_type, init, append, destroy use tem_dyn_array_module, only: PositionOfVal use tem_spacetime_fun_module, only: tem_st_fun_linkedList_type, & & tem_create_subTree_of_st_funList use tem_derived_module, only: tem_varSys_append_luaVar use tem_logging_module, only: tem_logging_init use tem_subTree_module, only: tem_create_subTree_of use tem_geometry_module, only: tem_BaryOfId use aotus_module, only: open_config_chunk, close_config, flu_state use aot_table_module, only: aot_table_open, aot_table_close, & & aot_table_length, aot_get_val use aot_out_module, only: aot_out_type, aot_out_open, aot_out_close, & & aot_out_val, aot_out_toChunk !mpi!nprocs = 1 implicit none character, parameter :: nl = C_NEW_LINE character(len=solSpecLen), parameter :: sysConf = & & 'function lua_fun(x,y,z,t)' // nl & & // ' return x' // nl & & // 'end' // nl & & // 'shape = {' // nl & & // ' kind = "canoND",' // nl & & // ' object = {' // nl & & // ' origin = {0.0,0.0,0.0},' //nl & & // ' vec = {0.0,1.0,0.0},' // nl & & // ' segments = 2' // nl & & // ' }' // nl & & // '}' // nl & & // 'variable = {' // nl & & // ' {' // nl & & // ' name = "lua_fun",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "st_fun",' // nl & & // ' st_fun = lua_fun' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "luafun_shape",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "st_fun",' // nl & & // ' st_fun = {' // nl & & // ' fun = lua_fun,' // nl & & // ' shape = shape' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "const11",' // nl & & // ' ncomponents = 1, ' // nl & & // ' vartype = "st_fun",' // nl & & // ' st_fun = 11.0' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "const11_shape",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "st_fun",' // nl & & // ' st_fun = {' // nl & & // ' const = 11.0,' // nl & & // ' shape = shape' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "const11_lua_fun",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "st_fun",' // nl & & // ' st_fun = {' // nl & & // ' {' // nl & & // ' fun = lua_fun,' // nl & & // ' shape = shape' // nl & & // ' },' // nl & & // ' 100.0' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "add1",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "operation",' // nl & & // ' operation = {' // nl & & // ' kind = "addition",' // nl & & // ' input_varname = {' // nl & & // ' "lua_fun",' // nl & & // ' "const11"' // nl & & // ' }' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "add2",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "operation",' // nl & & // ' operation = {' // nl & & // ' kind = "addition",' // nl & & // ' input_varname = {' // nl & & // ' "luafun_shape",' // nl & & // ' "const11_shape"' // nl & & // ' }' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "diff1",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "operation",' // nl & & // ' operation= {' // nl & & // ' kind = "difference",' // nl & & // ' input_varname = {' // nl & & // ' "const11_lua_fun", ' // nl & & // ' "add1"' // nl & & // ' }' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "combine",' // nl & & // ' ncomponents = 2,' // nl & & // ' vartype = "operation",' // nl & & // ' operation = {' // nl & & // ' kind = "combine",' // nl & & // ' input_varname = {' // nl & & // ' "diff1",' // nl & & // ' "add2"' // nl & & // ' }' // nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "extract1",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "operation",' // nl & & // ' operation= {' // nl & & // ' kind = "extract", ' // nl & & // ' input_varname = {"combine"}, '// nl & & // ' input_varindex = {1} '// nl & & // ' }' // nl & & // ' },' // nl & & // ' {' // nl & & // ' name = "extract2",' // nl & & // ' ncomponents = 1,' // nl & & // ' vartype = "operation",' // nl & & // ' operation= {' // nl & & // ' kind = "extract",' // nl & & // ' input_varname = {"combine"},' //nl & & // ' input_varindex = {2}' //nl & & // ' }' // nl & & // ' },' // nl & & // '}' // nl & & // 'track_variable = {' // nl & & // ' "lua_fun",' // nl & & // ' "luafun_shape",' // nl & & // ' "const11",' // nl & & // ' "const11_shape",' // nl & & // ' "const11_lua_fun",' // nl & & // ' "add1",' // nl & & // ' "add2",' // nl & & // ' "diff1",' // nl & & // ' "combine",' // nl & & // ' "extract1",' // nl & & // ' "extract2"' // nl & & // '}' // nl ! write output to screen logical, parameter :: dumpRes = .true. ! number of elements to track integer, parameter :: nElems_track = 5 ! position in global treeID list to track integer, dimension(nElems_track), parameter :: elemPos = (/ 1, 3, 5, 7, 8 /) ! reference values real(kind=rk), dimension(nElems_track), parameter :: & & luaFun_ref = (/ 0.25_rk, 0.25_rk, 0.25_rk, 0.25_rk, 0.75_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & luaFunShape_ref = (/ 0.25_rk, 0.25_rk, 0.0_rk, 0.0_rk, 0.0_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & const11_ref = (/ 11.0_rk, 11.0_rk, 11.0_rk, 11.0_rk, 11.0_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & const11Shape_ref = (/ 11.0_rk, 11.0_rk, 0.0_rk, 0.0_rk, 0.0_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & const11LuaFun_ref = (/ 100.25_rk, 100.25_rk, 100.0_rk, 100.0_rk, 100.0_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & add1_ref = (/ 11.25_rk, 11.25_rk, 11.25_rk, 11.25_rk, 11.75_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & add2_ref = (/ 11.25_rk, 11.25_rk, 0.0_rk, 0.0_rk, 0.0_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & diff1_ref = (/ 89.0_rk, 89.0_rk, 88.75_rk, 88.75_rk, 88.25_rk /) real(kind=rk), dimension(nElems_track,2), parameter :: & & combine_ref = reshape((/ 89.0_rk, 89.0_rk, 88.75_rk, 88.75_rk, 88.25_rk, & & 11.25_rk, 11.25_rk, 0.0_rk, 0.0_rk, 0.0_rk /),& & (/nElems_track,2/)) real(kind=rk), dimension(nElems_track), parameter :: & & extract1_ref = (/ 89.0_rk, 89.0_rk, 88.75_rk, 88.75_rk, 88.25_rk /) real(kind=rk), dimension(nElems_track), parameter :: & & extract2_ref = (/ 11.25_rk, 11.25_rk, 0.0_rk, 0.0_rk, 0.0_rk /) type solver_type integer :: nDofs real(kind=rk), allocatable :: state(:) type(treelmesh_type) :: tree type(tem_bc_prop_type) :: boundary type(tem_general_type) :: general end type solver_type type tracking_type type(grw_intArray_type) :: varPos integer :: nRequestedVars character(len=labelLen), allocatable :: variable(:) end type tracking_type type var_index integer :: idx(nElems_track) end type type(tem_varSys_type) :: varSys type(tem_st_fun_linkedList_type) :: st_funList type(solver_type), target :: solver type(tracking_type) :: tracking type(tem_variable_type), allocatable :: newVar(:) integer :: addedPos integer :: iElem integer :: iVar real(kind=rk), allocatable :: res(:) integer, allocatable :: error(:) real(kind=rk) :: point(nElems_track,3) integer :: idx(nElems_track) type(var_index), allocatable :: varIdx(:) write(*,*) 'Hello from tem_varSys_test' ! load utest mesh call load_env( tree = solver%tree, & & boundary = solver%boundary, & & general = solver%general ) call tem_logging_init( level = 10, rank = 0 ) write(*,*) 'nElems ', solver%tree%nElems allocate(solver%general%solver%conf(1)) call load_config( conf = solver%general%solver%conf(1), & & chunk = trim(sysConf), & & stfun_linkedList = st_funList, & & tracking = tracking, & & newVar = newVar ) solver%nDofs = 1 ! initialize variable system write(*,*) 'calling varsys init' call tem_varSys_init(me = varSys, systemName = 'utest') write(*,*) 'add new variables. nVars:', size(newVar) call tem_varSys_append_luaVar( luaVar = newVar, & & varSys = varSys, & & st_funList = st_funList ) write(*,*) write(*,*) 'Create subtree for all space time functions stored ' write(*,*) 'in linked list of spacetime function' call tem_create_subTree_of_st_funList( & & me = st_funList, & & tree = solver%tree, & & bc_prop = solver%boundary ) write(*,*) 'Done creating subtree for all spacetime functions' write(*,*) write(*,*) 'Setup indices ' do iElem = 1, nElems_track point(iElem,:) = tem_BaryOfId( tree = solver%tree, & & treeID = solver%tree & & %treeID(elemPos(iElem)) ) end do ! idx is equal for all variables !do iVar = 1, tracking%nRequestedVars ! call varSys%method%val(iVar)%setup_indices( varSys = varSys, & ! & point = point, & ! & iLevel = 1, & ! & tree = solver%tree, & ! & nPnts = nElems_track, & ! & idx = idx ) !end do ! idx is different for all variables ! idx>0 only if point is part of subTree shape allocate(varIdx(tracking%nRequestedVars)) do iVar=1,tracking%varPos%nVals addedPos = tracking%varPos%val(iVar) call varSys%method%val(addedPos)%setup_indices( & & varSys = varSys, & & point = point, & & iLevel = 1, & & tree = solver%tree, & & nPnts = nElems_track, & & idx = varIdx(addedPos)%idx ) write(*,*) 'iVar ', iVar, ' name '//trim(newVar(addedPos)%label) write(*,*) 'idx ', varIdx(addedPos)%idx end do !write(*,*) 'Create varMap for tracking' !call init(tracking%varPos) !do iVar = 1, tracking%nRequestedVars ! addedPos = PositionOfVal( me = varSys%varname, & ! & val = trim(tracking%variable(iVar)) ) ! if (addedPos > 0) then ! call append( me = tracking%varPos, val = addedPos ) ! else ! write(*,*) 'Variable ', trim(tracking%variable(iVar)), & ! & ' not found in varSys' ! end if !end do !write(*,*) allocate(error(tracking%varPos%nVals)) write(*,*) 'Checking variable extraction ' do iVar=1,tracking%varPos%nVals if (allocated(res)) deallocate(res) addedPos = tracking%varPos%val(iVar) write(*,*) 'track var: ', trim(varSys%varname%val(addedPos)) allocate(res(nElems_track*varSys%method%val(addedPos)%nComponents & & *solver%nDofs)) ! access state array for given elemPos write(*,*) 'Checking get_element ' call varSys%method%val(addedPos)%get_element( & & varSys = varSys, & & elemPos = elemPos, & & time = solver%general & & %simControl%now, & & tree = solver%tree, & & nElems = nElems_track, & & nDofs = solver%nDofs, & & res = res ) call check_res( varname = varSys%varname%val(addedPos) , & & res = res, & & error = error(iVar) ) if (error(iVar) == -1) exit write(*,*) 'Checking get_point ' call varSys%method%val(addedPos)%get_point( & & varSys = varSys, & & point = point, & & time = solver%general & & %simControl%now, & & tree = solver%tree, & & nPnts = nElems_track, & & res = res ) call check_res( varname = varSys%varname%val(addedPos) , & & res = res, & & error = error(iVar) ) if (error(iVar) == -1) exit write(*,*) 'Checking get_valOfIndex ' call varSys%method%val(addedPos)%get_valOfIndex( & & varSys = varSys, & & time = solver%general & & %simControl%now, & & iLevel = 1, & & idx = idx, & & nVals = nElems_track, & & res = res ) call check_res( varname = varSys%varname%val(addedPos) , & & res = res, & & error = error(iVar) ) if (error(iVar) == -1) exit !do iElem = 1, nElems_track ! write(*,*) 'iElem ', iElem, 'elemPos ', elemPos(iElem) ! write(*,*) 'res ', & ! & res( (iElem-1)*varSys%method%val(addedPos)%nComponents + 1 : & ! & iElem*varSys%method%val(addedPos)%nComponents ) !end do end do if (any(error/=0)) then write(*,*) 'FAILED' else write(*,*) 'PASSED' end if call close_config(L=solver%general%solver%conf(1)) call fin_env() contains ! ***************************************************************************! !> load variables defined in sysConf subroutine load_config( conf, chunk, stfun_linkedList, tracking, newVar ) ! -------------------------------------------------------------------------! type(flu_state) :: conf character(len=*), intent(in) :: chunk type(tem_st_fun_linkedList_type), intent(out) :: stfun_linkedList type(tracking_type), intent(out) :: tracking type(tem_variable_type), allocatable, intent(out) :: newVar(:) ! -------------------------------------------------------------------------! integer :: nVars, iVar, thandle, iError integer, allocatable :: vError(:) ! -------------------------------------------------------------------------! call open_config_chunk(L=conf, chunk=trim(chunk)) call tem_variable_load( me = newVar, & & conf = conf, & & key = 'variable', & & vError = vError ) call aot_table_open( L = conf, & & thandle = thandle, & & key = 'track_variable' ) nVars = aot_table_length( L = conf, thandle = thandle ) tracking%nRequestedVars = nVars allocate(tracking%variable(nVars)) write(*,*) 'track_variables :' do iVar = 1, nVars ! Get the names of the variable call aot_get_val( L = conf, & & thandle = thandle, & & val = tracking%variable(iVar), & & ErrCode = iError, & & pos = iVar ) write(*,*) trim(tracking%variable(iVar)) end do write(*,*) call aot_table_close( L = conf, thandle = thandle) end subroutine load_config ! ***************************************************************************! ! ***************************************************************************! !> Check tracking results subroutine check_res( varname, res, error ) ! ---------------------------------------------------------------------- ! character(len=labelLen), intent(in) :: varname real(kind=rk), intent(in) :: res(:) integer, intent(out) :: error ! ---------------------------------------------------------------------- ! integer :: iElem real(kind=rk) :: diff_spt(nElems_track) real(kind=rk) :: diff_vec(nElems_track,2) ! ---------------------------------------------------------------------- ! error = 0 select case (trim(varname)) case ('lua_fun') diff_spt = res(:) - luaFun_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for lua_fun does not match' end if if (dumpRes) then write(*,*) 'reference: ', luaFun_ref write(*,*) ' output: ', res end if case ('luafun_shape') diff_spt = res(:) - luaFunShape_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for luafun_shape does not match' write(*,*) 'diff ', diff_spt end if if (dumpRes) then write(*,*) 'reference: ', luaFunShape_ref write(*,*) ' output: ', res end if case ('const11') diff_spt = res(:) - const11_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for const11 does not match' end if if (dumpRes) then write(*,*) 'reference: ', const11_ref write(*,*) ' output: ', res end if case ('const11_shape') diff_spt = res(:) - const11Shape_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for const11_shape does not match' end if if (dumpRes) then write(*,*) 'reference: ', const11Shape_ref write(*,*) ' output: ', res end if case ('const11_lua_fun') diff_spt = res(:) - const11LuaFun_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for const11_lua_fun does not match' end if if (dumpRes) then write(*,*) 'reference: ', const11LuaFun_ref write(*,*) ' output: ', res end if case ('add1') diff_spt = res(:) - add1_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for add1 does not match' end if if (dumpRes) then write(*,*) 'reference: ', add1_ref write(*,*) ' output: ', res end if case ('add2') diff_spt = res(:) - add2_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for add2 does not match' end if if (dumpRes) then write(*,*) 'reference: ', add2_ref write(*,*) ' output: ', res end if case ('diff1') diff_spt = res(:) - diff1_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for diff1 does not match' end if if (dumpRes) then write(*,*) 'reference: ', diff1_ref write(*,*) ' output: ', res end if case ('combine') do iElem = 1, nElems_track diff_vec(iElem,:) = res((iElem-1)*2 + 1 : (iElem-1)*2 + 2 ) & & - combine_ref(iElem,:) end do if ( any(diff_vec > eps) ) then error = -1 write(*,*) 'Reference value for combine does not match' end if if (dumpRes) then do iElem = 1, nElems_track write(*,*) 'reference: ', combine_ref(iElem, :) write(*,*) ' output: ', res( (iElem-1)*2 + 1 : (iElem-1)*2 + 2 ) end do end if case ('extract1') diff_spt = res(:) - extract1_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for extract1 does not match' end if if (dumpRes) then write(*,*) 'reference: ', extract1_ref write(*,*) ' output: ', res end if case ('extract2') diff_spt = res(:) - extract2_ref if ( any(diff_spt > eps) ) then error = -1 write(*,*) 'Reference value for extract2 does not match' end if if (dumpRes) then write(*,*) 'reference: ', extract2_ref write(*,*) ' output: ', res end if case default write(*,*) 'Unknown variable name', trim(varname) error = -1 end select end subroutine check_res ! ***************************************************************************! end program tem_varSys_opVar_test