DAMASK_EICMD/src/grid/discretization_grid.f90

801 lines
36 KiB
Fortran
Raw Normal View History

!--------------------------------------------------------------------------------------------------
!> @author Franz Roters, Max-Planck-Institut für Eisenforschung GmbH
!> @author Philip Eisenlohr, Max-Planck-Institut für Eisenforschung GmbH
!> @author Martin Diehl, Max-Planck-Institut für Eisenforschung GmbH
2019-06-07 15:51:12 +05:30
!> @brief Parse geometry file to set up discretization and geometry for nonlocal model
!--------------------------------------------------------------------------------------------------
2020-03-20 11:28:55 +05:30
module discretization_grid
#include <petsc/finclude/petscsys.h>
2019-06-07 15:51:12 +05:30
use PETScsys
2019-06-08 14:08:10 +05:30
2019-06-07 15:51:12 +05:30
use prec
2020-09-13 13:49:38 +05:30
use parallelization
2019-06-07 15:51:12 +05:30
use system_routines
use base64
use zlib
2019-06-07 15:51:12 +05:30
use DAMASK_interface
use IO
2020-08-15 19:32:10 +05:30
use config
2019-11-24 12:14:17 +05:30
use results
2019-06-07 15:51:12 +05:30
use discretization
use geometry_plastic_nonlocal
use FEsolving
2020-01-29 19:18:15 +05:30
2019-06-07 15:51:12 +05:30
implicit none
private
2020-01-29 19:18:15 +05:30
2019-06-07 15:51:12 +05:30
integer, dimension(3), public, protected :: &
grid !< (global) grid
integer, public, protected :: &
grid3, & !< (local) grid in 3rd direction
grid3Offset !< (local) grid offset in 3rd direction
real(pReal), dimension(3), public, protected :: &
geomSize !< (global) physical size
2019-06-07 15:51:12 +05:30
real(pReal), public, protected :: &
size3, & !< (local) size in 3rd direction
size3offset !< (local) size offset in 3rd direction
2020-01-29 19:18:15 +05:30
2019-06-07 15:51:12 +05:30
public :: &
2020-03-20 11:28:55 +05:30
discretization_grid_init
2020-01-29 19:18:15 +05:30
contains
!--------------------------------------------------------------------------------------------------
2019-06-08 14:08:10 +05:30
!> @brief reads the geometry file to obtain information on discretization
!--------------------------------------------------------------------------------------------------
subroutine discretization_grid_init(restart)
logical, intent(in) :: restart
2020-01-29 19:18:15 +05:30
2019-06-07 15:51:12 +05:30
include 'fftw3-mpi.f03'
2019-06-08 14:08:10 +05:30
real(pReal), dimension(3) :: &
mySize, & !< domain size of this process
origin !< (global) distance to origin
2019-06-08 14:08:10 +05:30
integer, dimension(3) :: &
myGrid !< domain grid of this process
integer, dimension(:), allocatable :: &
2020-09-24 00:55:14 +05:30
materialAt
2019-06-08 14:08:10 +05:30
2020-06-18 21:13:25 +05:30
integer :: &
j, &
2020-07-02 02:09:44 +05:30
debug_element, &
debug_ip
2019-06-08 14:08:10 +05:30
integer(C_INTPTR_T) :: &
devNull, z, z_offset
2019-06-07 15:51:12 +05:30
2020-09-22 16:39:12 +05:30
print'(/,a)', ' <<<+- discretization_grid init -+>>>'; flush(IO_STDOUT)
2019-06-07 15:51:12 +05:30
if(index(interface_geomFile,'.vtr') /= 0) then
2020-09-24 00:55:14 +05:30
call readVTR(grid,geomSize,origin,materialAt)
else
2020-09-24 00:55:14 +05:30
call readGeom(grid,geomSize,origin,materialAt)
endif
2019-06-07 15:51:12 +05:30
2020-09-14 00:51:55 +05:30
print'(/,a,3(i12 ))', ' grid a b c: ', grid
print'(a,3(es12.5))', ' size x y z: ', geomSize
print'(a,3(es12.5))', ' origin x y z: ', origin
2019-06-08 14:45:01 +05:30
!--------------------------------------------------------------------------------------------------
! grid solver specific quantities
2019-06-08 14:08:10 +05:30
if(worldsize>grid(3)) call IO_error(894, ext_msg='number of processes exceeds grid(3)')
2019-06-07 15:51:12 +05:30
2019-06-08 14:08:10 +05:30
call fftw_mpi_init
2019-06-07 15:51:12 +05:30
devNull = fftw_mpi_local_size_3d(int(grid(3),C_INTPTR_T), &
int(grid(2),C_INTPTR_T), &
int(grid(1),C_INTPTR_T)/2+1, &
PETSC_COMM_WORLD, &
2019-06-08 14:08:10 +05:30
z, & ! domain grid size along z
z_offset) ! domain grid offset along z
grid3 = int(z)
grid3Offset = int(z_offset)
2019-06-07 15:51:12 +05:30
size3 = geomSize(3)*real(grid3,pReal) /real(grid(3),pReal)
size3Offset = geomSize(3)*real(grid3Offset,pReal)/real(grid(3),pReal)
2019-06-08 14:08:10 +05:30
myGrid = [grid(1:2),grid3]
mySize = [geomSize(1:2),size3]
2020-06-18 21:13:25 +05:30
!-------------------------------------------------------------------------------------------------
! debug parameters
debug_element = config_debug%get_asInt('element',defaultVal=1)
debug_ip = config_debug%get_asInt('integrationpoint',defaultVal=1)
2020-06-18 21:13:25 +05:30
2019-06-08 14:45:01 +05:30
!--------------------------------------------------------------------------------------------------
! general discretization
2020-09-24 00:55:14 +05:30
materialAt = materialAt(product(grid(1:2))*grid3Offset+1:product(grid(1:2))*(grid3Offset+grid3)) ! reallocate/shrink in case of MPI
2019-06-07 03:34:20 +05:30
2020-09-24 00:55:14 +05:30
call discretization_init(materialAt, &
2019-09-28 03:24:02 +05:30
IPcoordinates0(myGrid,mySize,grid3Offset), &
Nodes0(myGrid,mySize,grid3Offset),&
merge((grid(1)+1) * (grid(2)+1) * (grid3+1),& ! write top layer...
(grid(1)+1) * (grid(2)+1) * grid3,& ! ...unless not last process
worldrank+1==worldsize))
2019-06-07 15:51:12 +05:30
2019-06-08 14:08:10 +05:30
FEsolving_execElem = [1,product(myGrid)] ! parallel loop bounds set to comprise all elements
FEsolving_execIP = [1,1] ! parallel loop bounds set to comprise the only IP
2019-06-07 15:51:12 +05:30
2019-11-24 12:14:17 +05:30
!--------------------------------------------------------------------------------------------------
! store geometry information for post processing
if(.not. restart) then
call results_openJobFile
call results_closeGroup(results_addGroup('geometry'))
call results_addAttribute('grid', grid, 'geometry')
call results_addAttribute('size', geomSize,'geometry')
call results_addAttribute('origin',origin, 'geometry')
call results_closeJobFile
endif
2020-07-14 00:43:53 +05:30
2019-06-08 14:45:01 +05:30
!--------------------------------------------------------------------------------------------------
! geometry information required by the nonlocal CP model
call geometry_plastic_nonlocal_setIPvolume(reshape([(product(mySize/real(myGrid,pReal)),j=1,product(myGrid))], &
[1,product(myGrid)]))
2020-01-29 19:18:15 +05:30
call geometry_plastic_nonlocal_setIParea (cellSurfaceArea(mySize,myGrid))
call geometry_plastic_nonlocal_setIPareaNormal (cellSurfaceNormal(product(myGrid)))
2019-06-08 14:45:01 +05:30
call geometry_plastic_nonlocal_setIPneighborhood(IPneighborhood(myGrid))
!--------------------------------------------------------------------------------------------------
! sanity checks for debugging
2020-07-02 02:09:44 +05:30
if (debug_element < 1 .or. debug_element > product(myGrid)) call IO_error(602,ext_msg='element') ! selected element does not exist
if (debug_ip /= 1) call IO_error(602,ext_msg='IP') ! selected IP does not exist
2019-05-14 09:36:01 +05:30
2020-03-20 11:28:55 +05:30
end subroutine discretization_grid_init
!--------------------------------------------------------------------------------------------------
!> @brief Parses geometry file
2020-01-29 19:18:15 +05:30
!> @details important variables have an implicit "save" attribute. Therefore, this function is
! supposed to be called only once!
!--------------------------------------------------------------------------------------------------
2020-09-24 00:55:14 +05:30
subroutine readGeom(grid,geomSize,origin,material)
2019-06-07 15:51:12 +05:30
integer, dimension(3), intent(out) :: &
2020-09-13 22:02:49 +05:30
grid ! grid (across all processes!)
real(pReal), dimension(3), intent(out) :: &
2020-09-13 22:02:49 +05:30
geomSize, & ! size (across all processes!)
origin ! origin (across all processes!)
2019-06-07 15:51:12 +05:30
integer, dimension(:), intent(out), allocatable :: &
2020-09-24 00:55:14 +05:30
material
2020-01-29 19:18:15 +05:30
2019-06-08 14:45:01 +05:30
character(len=:), allocatable :: rawData
character(len=65536) :: line
integer, allocatable, dimension(:) :: chunkPos
integer :: &
headerLength = -1, & !< length of header (in lines)
fileLength, & !< length of the geom file (in characters)
fileUnit, &
startPos, endPos, &
myStat, &
l, & !< line counter
2020-09-24 00:55:14 +05:30
c, & !< counter for # materials in line
o, & !< order of "to" packing
2020-01-29 19:18:15 +05:30
e, & !< "element", i.e. spectral collocation point
i, j
2020-01-29 19:18:15 +05:30
2019-06-08 14:45:01 +05:30
grid = -1
geomSize = -1.0_pReal
!--------------------------------------------------------------------------------------------------
2019-06-08 14:45:01 +05:30
! read raw data as stream
inquire(file = trim(interface_geomFile), size=fileLength)
open(newunit=fileUnit, file=trim(interface_geomFile), access='stream',&
status='old', position='rewind', action='read',iostat=myStat)
if(myStat /= 0) call IO_error(100,ext_msg=trim(interface_geomFile))
allocate(character(len=fileLength)::rawData)
read(fileUnit) rawData
close(fileUnit)
2020-01-29 19:18:15 +05:30
!--------------------------------------------------------------------------------------------------
! get header length
2020-01-27 01:45:21 +05:30
endPos = index(rawData,IO_EOL)
2019-12-26 19:54:51 +05:30
if(endPos <= index(rawData,'head')) then ! ToDo: Should be 'header'
startPos = len(rawData)
2019-06-08 14:45:01 +05:30
call IO_error(error_ID=841, ext_msg='readGeom')
else
chunkPos = IO_stringPos(rawData(1:endPos))
2019-06-08 14:45:01 +05:30
if (chunkPos(1) < 2) call IO_error(error_ID=841, ext_msg='readGeom')
headerLength = IO_intValue(rawData(1:endPos),chunkPos,1)
startPos = endPos + 1
endif
!--------------------------------------------------------------------------------------------------
2020-09-13 22:02:49 +05:30
! read and interpret header
origin = 0.0_pReal
l = 0
do while (l < headerLength .and. startPos < len(rawData))
2020-01-27 01:45:21 +05:30
endPos = startPos + index(rawData(startPos:),IO_EOL) - 1
if (endPos < startPos) endPos = len(rawData) ! end of file without new line
line = rawData(startPos:endPos)
2019-06-08 14:45:01 +05:30
startPos = endPos + 1
l = l + 1
chunkPos = IO_stringPos(trim(line))
if (chunkPos(1) < 2) cycle ! need at least one keyword value pair
2020-01-29 19:18:15 +05:30
select case (IO_lc(IO_StringValue(trim(line),chunkPos,1)) )
case ('grid')
if (chunkPos(1) > 6) then
2019-06-08 14:45:01 +05:30
do j = 2,6,2
select case (IO_lc(IO_stringValue(line,chunkPos,j)))
case('a')
2019-06-08 14:45:01 +05:30
grid(1) = IO_intValue(line,chunkPos,j+1)
case('b')
2019-06-08 14:45:01 +05:30
grid(2) = IO_intValue(line,chunkPos,j+1)
case('c')
2019-06-08 14:45:01 +05:30
grid(3) = IO_intValue(line,chunkPos,j+1)
end select
enddo
endif
2020-01-29 19:18:15 +05:30
case ('size')
if (chunkPos(1) > 6) then
2019-06-08 14:45:01 +05:30
do j = 2,6,2
select case (IO_lc(IO_stringValue(line,chunkPos,j)))
case('x')
2019-06-08 14:45:01 +05:30
geomSize(1) = IO_floatValue(line,chunkPos,j+1)
case('y')
2019-06-08 14:45:01 +05:30
geomSize(2) = IO_floatValue(line,chunkPos,j+1)
case('z')
2019-06-08 14:45:01 +05:30
geomSize(3) = IO_floatValue(line,chunkPos,j+1)
end select
enddo
endif
2020-01-29 19:18:15 +05:30
case ('origin')
if (chunkPos(1) > 6) then
do j = 2,6,2
select case (IO_lc(IO_stringValue(line,chunkPos,j)))
case('x')
origin(1) = IO_floatValue(line,chunkPos,j+1)
case('y')
origin(2) = IO_floatValue(line,chunkPos,j+1)
case('z')
origin(3) = IO_floatValue(line,chunkPos,j+1)
end select
enddo
endif
end select
enddo
!--------------------------------------------------------------------------------------------------
! sanity checks
2019-06-08 14:45:01 +05:30
if(any(grid < 1)) &
call IO_error(error_ID = 842, ext_msg='grid (readGeom)')
if(any(geomSize < 0.0_pReal)) &
2019-06-08 14:45:01 +05:30
call IO_error(error_ID = 842, ext_msg='size (readGeom)')
2020-09-24 00:55:14 +05:30
allocate(material(product(grid)), source = -1) ! too large in case of MPI (shrink later, not very elegant)
2020-01-29 19:18:15 +05:30
!--------------------------------------------------------------------------------------------------
2019-05-14 02:48:02 +05:30
! read and interpret content
2019-06-08 14:45:01 +05:30
e = 1
do while (startPos < len(rawData))
2020-01-27 01:45:21 +05:30
endPos = startPos + index(rawData(startPos:),IO_EOL) - 1
if (endPos < startPos) endPos = len(rawData) ! end of file without new line
line = rawData(startPos:endPos)
2019-06-08 14:45:01 +05:30
startPos = endPos + 1
l = l + 1
chunkPos = IO_stringPos(trim(line))
2020-01-29 19:18:15 +05:30
noCompression: if (chunkPos(1) /= 3) then
c = chunkPos(1)
2020-09-24 00:55:14 +05:30
material(e:e+c-1) = [(IO_intValue(line,chunkPos,i+1), i=0, c-1)]
else noCompression
compression: if (IO_lc(IO_stringValue(line,chunkPos,2)) == 'of') then
c = IO_intValue(line,chunkPos,1)
2020-09-24 00:55:14 +05:30
material(e:e+c-1) = [(IO_intValue(line,chunkPos,3),i = 1,IO_intValue(line,chunkPos,1))]
2019-12-30 11:18:55 +05:30
else if (IO_lc(IO_stringValue(line,chunkPos,2)) == 'to') then compression
2019-06-08 14:45:01 +05:30
c = abs(IO_intValue(line,chunkPos,3) - IO_intValue(line,chunkPos,1)) + 1
o = merge(+1, -1, IO_intValue(line,chunkPos,3) > IO_intValue(line,chunkPos,1))
2020-09-24 00:55:14 +05:30
material(e:e+c-1) = [(i, i = IO_intValue(line,chunkPos,1),IO_intValue(line,chunkPos,3),o)]
else compression
c = chunkPos(1)
2020-09-24 00:55:14 +05:30
material(e:e+c-1) = [(IO_intValue(line,chunkPos,i+1), i=0, c-1)]
endif compression
endif noCompression
e = e+c
end do
2019-06-08 14:45:01 +05:30
if (e-1 /= product(grid)) call IO_error(error_ID = 843, el=e)
2019-06-08 14:45:01 +05:30
end subroutine readGeom
!--------------------------------------------------------------------------------------------------
!> @brief Parse vtk rectilinear grid (.vtr)
!> @details https://vtk.org/Wiki/VTK_XML_Formats
!--------------------------------------------------------------------------------------------------
2020-09-24 00:55:14 +05:30
subroutine readVTR(grid,geomSize,origin,material)
integer, dimension(3), intent(out) :: &
2020-09-13 22:02:49 +05:30
grid ! grid (across all processes!)
real(pReal), dimension(3), intent(out) :: &
2020-09-13 22:02:49 +05:30
geomSize, & ! size (across all processes!)
origin ! origin (across all processes!)
integer, dimension(:), intent(out), allocatable :: &
2020-09-24 00:55:14 +05:30
material
character(len=:), allocatable :: fileContent, dataType, headerType
2020-09-13 22:02:49 +05:30
logical :: inFile,inGrid,gotCoordinates,gotCellData,compressed
integer :: fileUnit, myStat, coord
integer(pI64) :: &
fileLength, & !< length of the geom file (in characters)
startPos, endPos, &
s
grid = -1
geomSize = -1.0_pReal
!--------------------------------------------------------------------------------------------------
! read raw data as stream
inquire(file = trim(interface_geomFile), size=fileLength)
open(newunit=fileUnit, file=trim(interface_geomFile), access='stream',&
status='old', position='rewind', action='read',iostat=myStat)
if(myStat /= 0) call IO_error(100,ext_msg=trim(interface_geomFile))
allocate(character(len=fileLength)::fileContent)
read(fileUnit) fileContent
close(fileUnit)
2020-09-13 22:02:49 +05:30
inFile = .false.
inGrid = .false.
gotCoordinates = .false.
gotCelldata = .false.
!--------------------------------------------------------------------------------------------------
2020-09-13 22:02:49 +05:30
! interpret XML file
startPos = 1_pI64
do while (startPos < len(fileContent,kind=pI64))
endPos = startPos + index(fileContent(startPos:),IO_EOL,kind=pI64) - 2_pI64
if (endPos < startPos) endPos = len(fileContent,kind=pI64) ! end of file without new line
if(.not. inFile) then
if(index(fileContent(startPos:endPos),'<VTKFile',kind=pI64) /= 0_pI64) then
inFile = .true.
2020-09-13 22:02:49 +05:30
if(.not. fileFormatOk(fileContent(startPos:endPos))) call IO_error(error_ID = 844, ext_msg='file format')
headerType = merge('UInt64','UInt32',getXMLValue(fileContent(startPos:endPos),'header_type')=='UInt64')
compressed = getXMLValue(fileContent(startPos:endPos),'compressor') == 'vtkZLibDataCompressor'
endif
else
if(.not. inGrid) then
if(index(fileContent(startPos:endPos),'<RectilinearGrid',kind=pI64) /= 0_pI64) inGrid = .true.
else
if(index(fileContent(startPos:endPos),'<CellData>',kind=pI64) /= 0_pI64) then
2020-09-13 22:02:49 +05:30
gotCellData = .true.
do while (index(fileContent(startPos:endPos),'</CellData>',kind=pI64) == 0_pI64)
if(index(fileContent(startPos:endPos),'<DataArray',kind=pI64) /= 0_pI64 .and. &
getXMLValue(fileContent(startPos:endPos),'Name') == 'material' ) then
if(getXMLValue(fileContent(startPos:endPos),'format') /= 'binary') &
call IO_error(error_ID = 844, ext_msg='format (materialpoint)')
dataType = getXMLValue(fileContent(startPos:endPos),'type')
startPos = endPos + 2_pI64
endPos = startPos + index(fileContent(startPos:),IO_EOL,kind=pI64) - 2_pI64
s = startPos + verify(fileContent(startPos:endPos),IO_WHITESPACE,kind=pI64) -1_pI64 ! start (no leading whitespace)
2020-09-24 00:55:14 +05:30
material = as_Int(fileContent(s:endPos),headerType,compressed,dataType)
exit
endif
startPos = endPos + 2_pI64
endPos = startPos + index(fileContent(startPos:),IO_EOL,kind=pI64) - 2_pI64
enddo
elseif(index(fileContent(startPos:endPos),'<Coordinates>',kind=pI64) /= 0_pI64) then
2020-09-13 22:02:49 +05:30
gotCoordinates = .true.
startPos = endPos + 2_pI64
coord = 0
do while (startPos<fileLength)
endPos = startPos + index(fileContent(startPos:),IO_EOL,kind=pI64) - 2_pI64
if(index(fileContent(startPos:endPos),'<DataArray',kind=pI64) /= 0_pI64) then
if(getXMLValue(fileContent(startPos:endPos),'format') /= 'binary') &
call IO_error(error_ID = 844, ext_msg='format (coordinates)')
dataType = getXMLValue(fileContent(startPos:endPos),'type')
startPos = endPos + 2_pI64
endPos = startPos + index(fileContent(startPos:),IO_EOL,kind=pI64) - 2_pI64
s = startPos + verify(fileContent(startPos:endPos),IO_WHITESPACE,kind=pI64) -1_pI64 ! start (no leading whitespace)
coord = coord + 1
call gridSizeOrigin(fileContent(s:endPos),headerType,compressed,dataType,coord)
endif
if(index(fileContent(startPos:endPos),'</Coordinates>',kind=pI64) /= 0_pI64) exit
startPos = endPos + 2_pI64
enddo
endif
endif
endif
2020-09-13 22:02:49 +05:30
if(gotCellData .and. gotCoordinates) exit
startPos = endPos + 2_pI64
end do
2020-09-24 00:55:14 +05:30
if(.not. allocated(material)) call IO_error(error_ID = 844, ext_msg='material data not found')
if(size(material) /= product(grid)) call IO_error(error_ID = 844, ext_msg='size(material)')
2020-09-24 13:06:19 +05:30
if(any(geomSize<=0)) call IO_error(error_ID = 844, ext_msg='size')
if(any(grid<1)) call IO_error(error_ID = 844, ext_msg='grid')
contains
!------------------------------------------------------------------------------------------------
!> @brief determine size and origin from coordinates
!------------------------------------------------------------------------------------------------
subroutine gridSizeOrigin(base64_str,headerType,compressed,dataType,direction)
character(len=*), intent(in) :: base64_str, & ! base64 encoded string of 1D coordinates
headerType, & ! header type (UInt32 or Uint64)
dataType ! data type (Int32, Int64, Float32, Float64)
logical, intent(in) :: compressed ! indicate whether data is zlib compressed
integer, intent(in) :: direction ! direction (1=x,2=y,3=z)
real(pReal), dimension(:), allocatable :: coords,delta
coords = as_pReal(base64_str,headerType,compressed,dataType)
delta = coords(2:) - coords(:size(coords)-1)
if(any(delta<0.0_pReal) .or. dNeq(maxval(delta),minval(delta),1.0e-8_pReal*maxval(abs(coords)))) &
call IO_error(error_ID = 844, ext_msg = 'grid spacing')
grid(direction) = size(coords)-1
origin(direction) = coords(1)
geomSize(direction) = coords(size(coords)) - coords(1)
end subroutine
!------------------------------------------------------------------------------------------------
!> @brief Interpret Base64 string in vtk XML file as integer of default kind
!------------------------------------------------------------------------------------------------
function as_Int(base64_str,headerType,compressed,dataType)
character(len=*), intent(in) :: base64_str, & ! base64 encoded string
headerType, & ! header type (UInt32 or Uint64)
dataType ! data type (Int32, Int64, Float32, Float64)
logical, intent(in) :: compressed ! indicate whether data is zlib compressed
integer, dimension(:), allocatable :: as_Int
select case(dataType)
case('Int32')
as_Int = int(prec_bytesToC_INT32_T(asBytes(base64_str,headerType,compressed)))
case('Int64')
as_Int = int(prec_bytesToC_INT64_T(asBytes(base64_str,headerType,compressed)))
case('Float32')
as_Int = int(prec_bytesToC_FLOAT (asBytes(base64_str,headerType,compressed)))
case('Float64')
as_Int = int(prec_bytesToC_DOUBLE (asBytes(base64_str,headerType,compressed)))
case default
call IO_error(844_pInt,ext_msg='unknown data type: '//trim(dataType))
end select
end function as_Int
!------------------------------------------------------------------------------------------------
!> @brief Interpret Base64 string in vtk XML file as integer of pReal kind
!------------------------------------------------------------------------------------------------
function as_pReal(base64_str,headerType,compressed,dataType)
character(len=*), intent(in) :: base64_str, & ! base64 encoded string
headerType, & ! header type (UInt32 or Uint64)
dataType ! data type (Int32, Int64, Float32, Float64)
logical, intent(in) :: compressed ! indicate whether data is zlib compressed
real(pReal), dimension(:), allocatable :: as_pReal
select case(dataType)
case('Int32')
as_pReal = real(prec_bytesToC_INT32_T(asBytes(base64_str,headerType,compressed)),pReal)
case('Int64')
as_pReal = real(prec_bytesToC_INT64_T(asBytes(base64_str,headerType,compressed)),pReal)
case('Float32')
as_pReal = real(prec_bytesToC_FLOAT (asBytes(base64_str,headerType,compressed)),pReal)
case('Float64')
as_pReal = real(prec_bytesToC_DOUBLE (asBytes(base64_str,headerType,compressed)),pReal)
case default
call IO_error(844_pInt,ext_msg='unknown data type: '//trim(dataType))
end select
end function as_pReal
!------------------------------------------------------------------------------------------------
!> @brief Interpret Base64 string in vtk XML file as bytes
!------------------------------------------------------------------------------------------------
function asBytes(base64_str,headerType,compressed) result(bytes)
character(len=*), intent(in) :: base64_str, & ! base64 encoded string
headerType ! header type (UInt32 or Uint64)
logical, intent(in) :: compressed ! indicate whether data is zlib compressed
integer(C_SIGNED_CHAR), dimension(:), allocatable :: bytes
if(compressed) then
bytes = asBytes_compressed(base64_str,headerType)
else
bytes = asBytes_uncompressed(base64_str,headerType)
endif
end function asBytes
!------------------------------------------------------------------------------------------------
!> @brief Interpret compressed Base64 string in vtk XML file as bytes
!> @details A compressed Base64 string consists of a header block and a data block
! [#blocks/#u-size/#p-size/#c-size-1/#c-size-2/.../#c-size-#blocks][DATA-1/DATA-2...]
! #blocks = Number of blocks
! #u-size = Block size before compression
! #p-size = Size of last partial block (zero if it not needed)
! #c-size-i = Size in bytes of block i after compression
!------------------------------------------------------------------------------------------------
function asBytes_compressed(base64_str,headerType) result(bytes)
character(len=*), intent(in) :: base64_str, & ! base64 encoded string
headerType ! header type (UInt32 or Uint64)
integer(C_SIGNED_CHAR), dimension(:), allocatable :: bytes, bytes_inflated
integer(pI64), dimension(:), allocatable :: temp, size_inflated, size_deflated
integer(pI64) :: headerLen, nBlock, b,s,e
if (headerType == 'UInt32') then
temp = int(prec_bytesToC_INT32_T(base64_to_bytes(base64_str(:base64_nChar(4_pI64)))),pI64)
nBlock = int(temp(1),pI64)
headerLen = 4_pI64 * (3_pI64 + nBlock)
temp = int(prec_bytesToC_INT32_T(base64_to_bytes(base64_str(:base64_nChar(headerLen)))),pI64)
elseif(headerType == 'UInt64') then
temp = int(prec_bytesToC_INT64_T(base64_to_bytes(base64_str(:base64_nChar(8_pI64)))),pI64)
nBlock = int(temp(1),pI64)
headerLen = 8_pI64 * (3_pI64 + nBlock)
temp = int(prec_bytesToC_INT64_T(base64_to_bytes(base64_str(:base64_nChar(headerLen)))),pI64)
endif
allocate(size_inflated(nBlock),source=temp(2))
size_inflated(nBlock) = merge(temp(3),temp(2),temp(3)/=0_pI64)
size_deflated = temp(4:)
bytes_inflated = base64_to_bytes(base64_str(base64_nChar(headerLen)+1_pI64:))
allocate(bytes(0))
e = 0_pI64
do b = 1, nBlock
s = e + 1_pI64
e = s + size_deflated(b) - 1_pI64
bytes = [bytes,zlib_inflate(bytes_inflated(s:e),size_inflated(b))]
enddo
end function asBytes_compressed
!------------------------------------------------------------------------------------------------
!> @brief Interprete uncompressed Base64 string in vtk XML file as bytes
!> @details An uncompressed Base64 string consists of N headers blocks and a N data blocks
![#bytes-1/DATA-1][#bytes-2/DATA-2]...
!------------------------------------------------------------------------------------------------
function asBytes_uncompressed(base64_str,headerType) result(bytes)
character(len=*), intent(in) :: base64_str, & ! base64 encoded string
headerType ! header type (UInt32 or Uint64)
integer(pI64) :: s
integer(pI64), dimension(1) :: nByte
integer(C_SIGNED_CHAR), dimension(:), allocatable :: bytes
allocate(bytes(0))
s=0_pI64
if (headerType == 'UInt32') then
do while(s+base64_nChar(4_pI64)<(len(base64_str,pI64)))
nByte = int(prec_bytesToC_INT32_T(base64_to_bytes(base64_str(s+1_pI64:s+base64_nChar(4_pI64)))),pI64)
bytes = [bytes,base64_to_bytes(base64_str(s+1_pI64:s+base64_nChar(4_pI64+nByte(1))),5_pI64)]
s = s + base64_nChar(4_pI64+nByte(1))
enddo
elseif(headerType == 'UInt64') then
do while(s+base64_nChar(8_pI64)<(len(base64_str,pI64)))
nByte = int(prec_bytesToC_INT64_T(base64_to_bytes(base64_str(s+1_pI64:s+base64_nChar(8_pI64)))),pI64)
bytes = [bytes,base64_to_bytes(base64_str(s+1_pI64:s+base64_nChar(8_pI64+nByte(1))),9_pI64)]
s = s + base64_nChar(8_pI64+nByte(1))
enddo
endif
end function asBytes_uncompressed
!------------------------------------------------------------------------------------------------
!> @brief Get XML string value for given key
!------------------------------------------------------------------------------------------------
pure function getXMLValue(line,key)
character(len=*), intent(in) :: line, key
character(len=:), allocatable :: getXMLValue
integer :: s,e
#ifdef __INTEL_COMPILER
character :: q
#endif
s = index(line," "//key,back=.true.)
if(s==0) then
getXMLValue = ''
else
e = s + 1 + scan(line(s+1:),"'"//'"')
if(scan(line(s:e-2),'=') == 0) then
getXMLValue = ''
else
s = e
! https://community.intel.com/t5/Intel-Fortran-Compiler/ICE-for-merge-with-strings/m-p/1207204#M151657
#ifdef __INTEL_COMPILER
q = line(s-1:s-1)
e = s + index(line(s:),q) - 1
#else
e = s + index(line(s:),merge("'",'"',line(s-1:s-1)=="'")) - 1
#endif
getXMLValue = line(s:e-1)
endif
endif
end function
!------------------------------------------------------------------------------------------------
2020-09-13 22:02:49 +05:30
!> @brief check for supported file format
!------------------------------------------------------------------------------------------------
2020-09-13 22:02:49 +05:30
pure function fileFormatOk(line)
character(len=*),intent(in) :: line
2020-09-13 22:02:49 +05:30
logical :: fileFormatOk
2020-09-13 22:02:49 +05:30
fileFormatOk = getXMLValue(line,'type') == 'RectilinearGrid' .and. &
getXMLValue(line,'byte_order') == 'LittleEndian' .and. &
getXMLValue(line,'compressor') /= 'vtkLZ4DataCompressor' .and. &
getXMLValue(line,'compressor') /= 'vtkLZMADataCompressor'
2020-09-13 22:02:49 +05:30
end function fileFormatOk
end subroutine readVTR
2019-05-14 02:48:02 +05:30
!---------------------------------------------------------------------------------------------------
2020-01-12 01:03:29 +05:30
!> @brief Calculate undeformed position of IPs/cell centers (pretend to be an element)
2019-05-14 02:48:02 +05:30
!---------------------------------------------------------------------------------------------------
2019-09-28 03:18:51 +05:30
function IPcoordinates0(grid,geomSize,grid3Offset)
2019-05-14 02:24:18 +05:30
2019-06-08 14:45:01 +05:30
integer, dimension(3), intent(in) :: grid ! grid (for this process!)
real(pReal), dimension(3), intent(in) :: geomSize ! size (for this process!)
integer, intent(in) :: grid3Offset ! grid(3) offset
2019-05-14 02:24:18 +05:30
2019-09-28 03:24:02 +05:30
real(pReal), dimension(3,product(grid)) :: ipCoordinates0
2019-06-08 14:45:01 +05:30
integer :: &
a,b,c, &
i
2020-01-29 19:18:15 +05:30
2019-06-08 14:45:01 +05:30
i = 0
do c = 1, grid(3); do b = 1, grid(2); do a = 1, grid(1)
i = i + 1
2019-09-28 03:24:02 +05:30
IPcoordinates0(1:3,i) = geomSize/real(grid,pReal) * (real([a,b,grid3Offset+c],pReal) -0.5_pReal)
2019-06-08 14:45:01 +05:30
enddo; enddo; enddo
2019-09-28 03:18:51 +05:30
end function IPcoordinates0
2019-05-14 02:48:02 +05:30
!---------------------------------------------------------------------------------------------------
2019-09-28 03:18:51 +05:30
!> @brief Calculate position of undeformed nodes (pretend to be an element)
2019-05-14 02:48:02 +05:30
!---------------------------------------------------------------------------------------------------
2019-09-28 03:18:51 +05:30
pure function nodes0(grid,geomSize,grid3Offset)
2019-05-14 02:48:02 +05:30
2019-06-08 14:45:01 +05:30
integer, dimension(3), intent(in) :: grid ! grid (for this process!)
real(pReal), dimension(3), intent(in) :: geomSize ! size (for this process!)
integer, intent(in) :: grid3Offset ! grid(3) offset
2019-05-14 02:48:02 +05:30
2019-09-28 03:24:02 +05:30
real(pReal), dimension(3,product(grid+1)) :: nodes0
2019-05-14 02:48:02 +05:30
2019-06-08 14:45:01 +05:30
integer :: &
a,b,c, &
n
2019-05-14 02:48:02 +05:30
2019-06-08 14:45:01 +05:30
n = 0
do c = 0, grid3; do b = 0, grid(2); do a = 0, grid(1)
n = n + 1
2019-09-28 03:18:51 +05:30
nodes0(1:3,n) = geomSize/real(grid,pReal) * real([a,b,grid3Offset+c],pReal)
2019-06-08 14:45:01 +05:30
enddo; enddo; enddo
2019-09-28 03:18:51 +05:30
end function nodes0
2019-06-07 15:51:12 +05:30
!--------------------------------------------------------------------------------------------------
2019-06-08 14:45:01 +05:30
!> @brief Calculate IP interface areas
2019-06-07 15:51:12 +05:30
!--------------------------------------------------------------------------------------------------
2020-01-29 19:18:15 +05:30
pure function cellSurfaceArea(geomSize,grid)
2019-06-08 14:45:01 +05:30
real(pReal), dimension(3), intent(in) :: geomSize ! size (for this process!)
integer, dimension(3), intent(in) :: grid ! grid (for this process!)
2019-06-07 15:51:12 +05:30
2020-01-29 19:18:15 +05:30
real(pReal), dimension(6,1,product(grid)) :: cellSurfaceArea
cellSurfaceArea(1:2,1,:) = geomSize(2)/real(grid(2)) * geomSize(3)/real(grid(3))
cellSurfaceArea(3:4,1,:) = geomSize(3)/real(grid(3)) * geomSize(1)/real(grid(1))
cellSurfaceArea(5:6,1,:) = geomSize(1)/real(grid(1)) * geomSize(2)/real(grid(2))
end function cellSurfaceArea
2019-06-07 15:51:12 +05:30
!--------------------------------------------------------------------------------------------------
2019-06-08 14:45:01 +05:30
!> @brief Calculate IP interface areas normals
2019-06-07 15:51:12 +05:30
!--------------------------------------------------------------------------------------------------
2020-01-29 19:18:15 +05:30
pure function cellSurfaceNormal(nElems)
2019-06-07 15:51:12 +05:30
integer, intent(in) :: nElems
2019-06-08 14:45:01 +05:30
2020-01-29 19:18:15 +05:30
real(pReal), dimension(3,6,1,nElems) :: cellSurfaceNormal
cellSurfaceNormal(1:3,1,1,:) = spread([+1.0_pReal, 0.0_pReal, 0.0_pReal],2,nElems)
cellSurfaceNormal(1:3,2,1,:) = spread([-1.0_pReal, 0.0_pReal, 0.0_pReal],2,nElems)
cellSurfaceNormal(1:3,3,1,:) = spread([ 0.0_pReal,+1.0_pReal, 0.0_pReal],2,nElems)
cellSurfaceNormal(1:3,4,1,:) = spread([ 0.0_pReal,-1.0_pReal, 0.0_pReal],2,nElems)
cellSurfaceNormal(1:3,5,1,:) = spread([ 0.0_pReal, 0.0_pReal,+1.0_pReal],2,nElems)
cellSurfaceNormal(1:3,6,1,:) = spread([ 0.0_pReal, 0.0_pReal,-1.0_pReal],2,nElems)
end function cellSurfaceNormal
2019-06-07 15:51:12 +05:30
2019-06-08 14:45:01 +05:30
!--------------------------------------------------------------------------------------------------
!> @brief Build IP neighborhood relations
!--------------------------------------------------------------------------------------------------
pure function IPneighborhood(grid)
integer, dimension(3), intent(in) :: grid ! grid (for this process!)
2020-01-29 19:18:15 +05:30
2020-01-29 22:45:49 +05:30
integer, dimension(3,6,1,product(grid)) :: IPneighborhood !< 6 neighboring IPs as [element ID, IP ID, face ID]
2020-01-29 19:18:15 +05:30
2019-06-08 14:45:01 +05:30
integer :: &
x,y,z, &
e
e = 0
do z = 0,grid(3)-1; do y = 0,grid(2)-1; do x = 0,grid(1)-1
e = e + 1
2020-01-27 02:26:08 +05:30
! element ID
2019-06-08 14:45:01 +05:30
IPneighborhood(1,1,1,e) = z * grid(1) * grid(2) &
+ y * grid(1) &
+ modulo(x+1,grid(1)) &
+ 1
IPneighborhood(1,2,1,e) = z * grid(1) * grid(2) &
+ y * grid(1) &
+ modulo(x-1,grid(1)) &
+ 1
IPneighborhood(1,3,1,e) = z * grid(1) * grid(2) &
+ modulo(y+1,grid(2)) * grid(1) &
+ x &
+ 1
IPneighborhood(1,4,1,e) = z * grid(1) * grid(2) &
+ modulo(y-1,grid(2)) * grid(1) &
+ x &
+ 1
IPneighborhood(1,5,1,e) = modulo(z+1,grid(3)) * grid(1) * grid(2) &
+ y * grid(1) &
+ x &
+ 1
IPneighborhood(1,6,1,e) = modulo(z-1,grid(3)) * grid(1) * grid(2) &
+ y * grid(1) &
+ x &
+ 1
2020-01-27 02:26:08 +05:30
! IP ID
IPneighborhood(2,:,1,e) = 1
! face ID
IPneighborhood(3,1,1,e) = 2
IPneighborhood(3,2,1,e) = 1
IPneighborhood(3,3,1,e) = 4
IPneighborhood(3,4,1,e) = 3
IPneighborhood(3,5,1,e) = 6
IPneighborhood(3,6,1,e) = 5
2019-06-08 14:45:01 +05:30
enddo; enddo; enddo
2020-01-29 19:18:15 +05:30
2019-06-08 14:45:01 +05:30
end function IPneighborhood
2019-06-07 15:51:12 +05:30
2020-03-20 11:28:55 +05:30
end module discretization_grid