94 class(box_t),
pointer :: box
95 class(coordinate_system_t),
pointer :: coord_system
97 logical :: use_curvilinear
99 real(real64),
allocatable :: spacing(:)
106 integer(int64) :: np_global
107 integer(int64) :: np_part_global
108 logical :: parallel_in_domains
109 type(mpi_grp_t) :: mpi_grp
110 type(par_vec_t) :: pv
111 type(partition_t) :: partition
113 real(real64),
allocatable :: x(:,:)
114 real(real64) :: volume_element
115 real(real64),
allocatable :: vol_pp(:)
117 logical :: masked_periodic_boundaries
118 character(len=256) :: periodic_boundary_mask
124 procedure :: dmesh_allreduce_0, zmesh_allreduce_0, imesh_allreduce_0
125 procedure :: dmesh_allreduce_1, zmesh_allreduce_1, imesh_allreduce_1
126 procedure :: dmesh_allreduce_2, zmesh_allreduce_2, imesh_allreduce_2
127 procedure :: dmesh_allreduce_3, zmesh_allreduce_3, imesh_allreduce_3
128 procedure :: dmesh_allreduce_4, zmesh_allreduce_4, imesh_allreduce_4
129 procedure :: dmesh_allreduce_5, zmesh_allreduce_5, imesh_allreduce_5
130 generic :: allreduce => dmesh_allreduce_0, zmesh_allreduce_0, imesh_allreduce_0
131 generic :: allreduce => dmesh_allreduce_1, zmesh_allreduce_1, imesh_allreduce_1
132 generic :: allreduce => dmesh_allreduce_2, zmesh_allreduce_2, imesh_allreduce_2
133 generic :: allreduce => dmesh_allreduce_3, zmesh_allreduce_3, imesh_allreduce_3
134 generic :: allreduce => dmesh_allreduce_4, zmesh_allreduce_4, imesh_allreduce_4
135 generic :: allreduce => dmesh_allreduce_5, zmesh_allreduce_5, imesh_allreduce_5
152 real(real64) :: u(3), v(3)
153 real(real64) :: origin(3)
154 real(real64) :: spacing
155 integer :: nu, mu, nv, mv
165 real(real64) :: origin(2)
166 real(real64) :: spacing
173 class(mesh_t),
intent(inout) :: this
177 call this%set_time_dependent(.false.)
185 class(space_t),
intent(in) :: space
186 type(mesh_t),
intent(in) :: mesh
187 real(real64),
intent(in) :: alpha
188 integer,
intent(out) :: db(:)
197 do idir = 1, space%periodic_dim
198 db(idir) = mesh%idx%ll(idir)
200 do idir = space%periodic_dim + 1, space%dim
201 db(idir) = nint(alpha * (mesh%idx%ll(idir) - 1)) + 1
211 integer,
optional,
intent(in) :: iunit
212 type(
namespace_t),
optional,
intent(in) :: namespace
215 real(real64) :: cutoff
220 do ii = 1, this%box%dim
228 write(
message(2),
'(a, i10)')
' # inner mesh = ', this%np_global
229 write(
message(3),
'(a, i10)')
' # total mesh = ', this%np_part_global
241 pure subroutine mesh_r(mesh, ip, rr, origin, coords)
242 class(mesh_t),
intent(in) :: mesh
243 integer,
intent(in) :: ip
244 real(real64),
intent(out) :: rr
245 real(real64),
intent(in),
optional :: origin(:)
246 real(real64),
intent(out),
optional :: coords(:)
248 real(real64) :: xx(1:mesh%box%dim)
251 if (
present(origin)) xx = xx - origin
254 if (
present(coords))
then
266 class(
mesh_t),
intent(in) :: mesh
267 real(real64),
intent(in) :: pos(:)
268 real(real64),
intent(out) :: dmin
269 integer,
intent(out) :: rankmin
280 dd = sum((pos - mesh%x(ip, :))**2)
281 if ((dd < dmin) .or. (ip == 1))
then
288 call mesh%mpi_grp%bcast(imin, 1, mpi_integer, rankmin)
296 class(
mesh_t),
intent(in ) :: mesh
297 real(real64),
intent(inout) ::
values(:, :)
299 integer :: i, ip, ndim
301 real(real64) :: dummy
305 if (mesh%parallel_in_domains)
then
310 if (mesh%mpi_grp%rank == process)
values(:, i) = mesh%x(ip, :)
311 call mesh%mpi_grp%bcast(
values(:, i), ndim, mpi_double_precision, process)
316 values(:, i) = mesh%x(ip, :)
330 class(
mesh_t),
intent(in) :: mesh
333 gmax =
m_pi / (maxval(mesh%spacing))
340 type(
mesh_t),
intent(in) :: mesh
341 character(len=*),
intent(in) :: dir
342 character(len=*),
intent(in) :: filename
343 type(mpi_grp_t),
intent(in) :: mpi_grp
344 type(namespace_t),
intent(in) :: namespace
345 integer,
intent(out) :: ierr
353 iunit = io_open(trim(dir)//
"/"//trim(filename), namespace, action=
'write', &
354 die=.false., grp=mpi_grp)
356 message(1) =
"Unable to open file '"//trim(dir)//
"/"//trim(filename)//
"'."
357 call messages_warning(1, namespace=namespace)
360 if (mpi_grp_is_root(mpi_grp))
then
361 write(iunit,
'(a20,i21)')
'np_part_global= ', mesh%np_part_global
362 write(iunit,
'(a20,i21)')
'np_global= ', mesh%np_global
363 write(iunit,
'(a20,i21)')
'algorithm= ', 1
364 write(iunit,
'(a20,i21)')
'checksum= ', mesh%idx%checksum
365 write(iunit,
'(a20,i21)')
'bits= ', mesh%idx%bits
366 write(iunit,
'(a20,i21)')
'dim= ', mesh%idx%dim
367 write(iunit,
'(a20,i21)')
'type= ', mesh%idx%type
368 do ii = 1, mesh%idx%dim
369 write(iunit,
'(a7,i2,a11,i21)')
'offset(',ii,
')= ', mesh%idx%offset(ii)
371 do ii = 1, mesh%idx%dim
372 write(iunit,
'(a7,i2,a11,i21)')
'nn(',ii,
')= ', mesh%idx%nr(2, ii) - mesh%idx%nr(1, ii) + 1
375 call io_close(iunit, grp=mpi_grp)
388 read_np_part, read_np, bits, type, offset, nn, ierr)
389 type(
mesh_t),
intent(in) :: mesh
390 character(len=*),
intent(in) :: dir
391 character(len=*),
intent(in) :: filename
392 type(mpi_grp_t),
intent(in) :: mpi_grp
393 type(namespace_t),
intent(in) :: namespace
394 integer(int64),
intent(out) :: read_np_part
395 integer(int64),
intent(out) :: read_np
396 integer,
intent(out) :: bits
397 integer,
intent(out) :: type
398 integer,
intent(out) :: offset(1:mesh%idx%dim)
399 integer,
intent(out) :: nn(1:mesh%idx%dim)
400 integer,
intent(out) :: ierr
402 character(len=20) :: str
403 character(len=100) :: lines(7)
404 integer :: iunit, algorithm, dim, err, ii
405 integer(int64) :: checksum
411 read_np_part = 0_int64
414 iunit = io_open(trim(dir)//
"/"//trim(filename), namespace, action=
'read', &
415 status=
'old', die=.false., grp=mpi_grp)
418 message(1) =
"Unable to open file '"//trim(dir)//
"/"//trim(filename)//
"'."
419 call messages_warning(1, namespace=namespace)
421 call iopar_read(mpi_grp, iunit, lines, 7, err)
425 read(lines(1),
'(a20,i21)') str, read_np_part
426 read(lines(2),
'(a20,i21)') str, read_np
427 read(lines(3),
'(a20,i21)') str, algorithm
428 read(lines(4),
'(a20,i21)') str, checksum
429 read(lines(5),
'(a20,i21)') str, bits
430 read(lines(6),
'(a20,i21)') str, dim
431 read(lines(7),
'(a20,i21)') str,
type
433 if (dim /= mesh%idx%dim)
then
437 call iopar_read(mpi_grp, iunit, lines, dim, err)
442 read(lines(ii),
'(a20,i21)') str, offset(ii)
447 call iopar_read(mpi_grp, iunit, lines, dim, err)
452 read(lines(ii),
'(a20,i21)') str, nn(ii)
457 assert(read_np_part >= read_np)
459 if (read_np_part == mesh%np_part_global &
460 .and. read_np == mesh%np_global &
461 .and. algorithm == 1 &
462 .and. checksum == mesh%idx%checksum)
then
468 call io_close(iunit, grp=mpi_grp)
476 type(
mesh_t),
intent(in) :: mesh
477 character(len=*),
intent(in) :: dir
478 character(len=*),
intent(in) :: filename
479 type(namespace_t),
intent(in) :: namespace
480 type(mpi_grp_t),
intent(in) :: mpi_grp
481 logical,
intent(out) :: grid_changed
482 logical,
intent(out) :: grid_reordered
483 integer(int64),
allocatable,
intent(out) :: map(:)
484 integer,
intent(out) :: ierr
486 integer(int64) :: ipg, ipg_new, read_np_part, read_np
488 integer :: bits,
type, offset(mesh%idx%dim), point(mesh%idx%dim), nn(mesh%idx%dim)
489 integer(int64),
allocatable :: read_indices(:)
490 type(index_t) :: idx_old
496 grid_changed = .false.
497 grid_reordered = .false.
501 bits,
type, offset, nn, err)
504 message(1) =
"Unable to read mesh fingerprint from '"//trim(dir)//
"/"//trim(filename)//
"'."
505 call messages_warning(1, namespace=namespace)
507 else if (read_np > 0)
then
508 if (.not.
associated(mesh%box))
then
513 grid_changed = .
true.
518 grid_reordered = (read_np == mesh%np_global)
521 safe_allocate(read_indices(1:read_np_part))
522 call io_binary_read(trim(io_workpath(dir, namespace))//
"/indices.obf", read_np_part, &
526 message(1) =
"Unable to read index map from '"//trim(dir)//
"'."
527 call messages_warning(1, namespace=namespace)
530 call index_init(idx_old, mesh%idx%dim)
533 idx_old%nr(1, :) = -offset
534 idx_old%nr(2, :) = -offset + nn - 1
535 idx_old%offset = offset
536 idx_old%stride(1) = 1
537 do idir = 2, mesh%idx%dim
538 idx_old%stride(idir) = idx_old%stride(idir-1) * nn(idir-1)
541 safe_allocate(map(1:read_np))
544 call index_spatial_to_point(idx_old, mesh%idx%dim, read_indices(ipg), point)
549 if (map(ipg) > mesh%np_global) map(ipg) = 0
551 if (map(ipg) == 0) grid_reordered = .false.
553 call index_end(idx_old)
556 safe_deallocate_a(read_indices)
566 class(
mesh_t),
intent(inout) :: this
571 call lmpi_destroy_shared_memory_window(this%idx%window_grid_to_spatial)
572 call lmpi_destroy_shared_memory_window(this%idx%window_spatial_to_grid)
573 nullify(this%idx%grid_to_spatial_global)
574 nullify(this%idx%spatial_to_grid_global)
576 safe_deallocate_p(this%idx%grid_to_spatial_global)
577 safe_deallocate_p(this%idx%spatial_to_grid_global)
580 safe_deallocate_a(this%x)
581 safe_deallocate_a(this%vol_pp)
583 if (this%parallel_in_domains)
then
584 call par_vec_end(this%pv)
585 call partition_end(this%partition)
588 call index_end(this%idx)
589 safe_deallocate_a(this%spacing)
602 class(
mesh_t),
intent(in) :: mesh
603 class(space_t),
intent(in) :: space
604 integer,
intent(in) :: ip
606 integer :: ix(space%dim), nr(2, space%dim), idim
607 real(real64) :: xx(space%dim), rr, ufn_re, ufn_im
612 nr(1, :) = mesh%idx%nr(1, :) + mesh%idx%enlarge(:)
613 nr(2, :) = mesh%idx%nr(2, :) - mesh%idx%enlarge(:)
615 do idim = 1, space%periodic_dim
616 if (ix(idim) < nr(1, idim)) ix(idim) = ix(idim) + mesh%idx%ll(idim)
617 if (ix(idim) > nr(2, idim)) ix(idim) = ix(idim) - mesh%idx%ll(idim)
623 if (mesh%masked_periodic_boundaries)
then
624 call mesh_r(mesh, ip, rr, coords = xx)
625 call parse_expression(ufn_re, ufn_im, space%dim, xx, rr, m_zero, mesh%periodic_boundary_mask)
634 use iso_c_binding,
only: c_sizeof, c_long_long
635 class(
mesh_t),
intent(in) :: mesh
638 memory = c_sizeof(c_long_long) * real(mesh%np_part_global, real64) * 2
645 use iso_c_binding,
only: c_sizeof, c_long_long
646 class(
mesh_t),
intent(in) :: mesh
651 memory = memory + real64 * real(mesh%np_part, real64) * mesh%idx%dim
653 memory = memory + c_sizeof(c_long_long) * real(mesh%np_part, real64) * 2
659 class(
mesh_t),
intent(in) :: mesh
660 integer(int64),
intent(in) :: ipg
661 real(real64) :: xx(1:mesh%box%dim)
663 real(real64) :: chi(1:mesh%box%dim)
664 integer :: ix(1:mesh%box%dim)
669 chi = ix * mesh%spacing
670 xx = mesh%coord_system%to_cartesian(chi)
677 type(
mesh_t),
intent(in) :: mesh
679 cb = .not. mesh%use_curvilinear .and. &
680 .not. mesh%parallel_in_domains
687 class(
mesh_t),
intent(in) :: mesh
688 type(symmetries_t),
intent(in) :: symm
689 integer,
intent(in) :: periodic_dim
691 integer :: iop, ip, idim, nops, ix(1:3)
692 real(real64) :: destpoint(1:3), srcpoint(1:3), lsize(1:3), offset(1:3)
693 real(real64),
parameter :: tol_spacing = 1e-12_real64
701 if (mesh%idx%ll(1) == mesh%idx%ll(2) .and. &
702 mesh%idx%ll(2) == mesh%idx%ll(3) .and. &
703 abs(mesh%spacing(1) - mesh%spacing(2)) < tol_spacing .and. &
704 abs(mesh%spacing(2) - mesh%spacing(3)) < tol_spacing)
return
708 message(1) =
"Checking if the real-space grid is symmetric"
709 call messages_info(1)
711 lsize(1:3) = real(mesh%idx%ll(1:3), real64)
712 offset(1:3) = real(mesh%idx%nr(1, 1:3) + mesh%idx%enlarge(1:3), real64)
714 nops = symmetries_number(symm)
721 destpoint(1:3) = real(ix(1:3), real64) - offset(1:3)
723 assert(all(destpoint >= 0))
724 assert(all(destpoint < lsize))
727 destpoint = destpoint - real(int(lsize)/2, real64)
731 destpoint(idim) = destpoint(idim)/lsize(idim)
736 srcpoint = symm_op_apply_red(symm%ops(iop), destpoint)
740 srcpoint(idim) = srcpoint(idim)*lsize(idim)
744 srcpoint = srcpoint + real(int(lsize)/2, real64)
747 do idim = 1, periodic_dim
748 if (nint(srcpoint(idim)) < 0 .or. nint(srcpoint(idim)) >= lsize(idim))
then
749 srcpoint(idim) = modulo(srcpoint(idim)+m_half*symprec, lsize(idim))
752 assert(all(srcpoint >= -symprec))
753 assert(all(srcpoint < lsize))
755 srcpoint(1:3) = srcpoint(1:3) + offset(1:3)
757 if (any(srcpoint-anint(srcpoint)> symprec*m_two))
then
758 message(1) =
"The real-space grid breaks at least one of the symmetries of the system."
759 message(2) =
"Change your spacing or use SymmetrizeDensity=no."
760 call messages_fatal(2)
771 class(
mesh_t),
intent(in) :: mesh
772 integer,
intent(in) :: ix(:)
774 index = index_from_coords(mesh%idx, ix)
780 type(
mesh_t),
intent(in) :: mesh
781 integer(int64),
intent(in) :: ipg
782 integer,
intent(out) :: ix(:)
784 call index_to_coords(mesh%idx, ipg, ix)
790 type(
mesh_t),
intent(in) :: mesh
791 integer,
intent(in) :: ix(:)
793 integer(int64) :: ipg
795 ipg = index_from_coords(mesh%idx, ix)
802 type(
mesh_t),
intent(in) :: mesh
803 integer,
intent(in) :: ip
804 integer,
intent(out) :: ix(:)
806 integer(int64) :: ipg
809 call index_to_coords(mesh%idx, ipg, ix)
814 type(
mesh_t),
intent(in) :: mesh
815 integer,
intent(in) :: ip
817 ipg = par_vec_local2global(mesh%pv, ip)
822 type(
mesh_t),
intent(in) :: mesh
823 integer(int64),
intent(in) :: ipg
825 ip = par_vec_global2local(mesh%pv, ipg)
831 type(
mesh_t),
intent(in) :: mesh
832 real(real64),
intent(inout) :: min_or_max
833 integer,
intent(out) :: rank_min_or_max
834 type(mpi_op),
intent(in) :: op
836 real(real64) :: loc_in(2), loc_out(2)
840 assert(op == mpi_minloc .or. op == mpi_maxloc)
843 if (mesh%parallel_in_domains)
then
844 loc_in(1) = min_or_max
845 loc_in(2) = mesh%mpi_grp%rank
846 call mesh%mpi_grp%allreduce(loc_in, loc_out, 1, mpi_2double_precision, op)
847 min_or_max = loc_out(1)
848 rank_min_or_max = nint(loc_out(2))
857#include "mesh_inc.F90"
860#include "complex.F90"
861#include "mesh_inc.F90"
864#include "integer.F90"
865#include "mesh_inc.F90"
real(real64), parameter, public m_two
real(real64), parameter, public m_zero
real(real64), parameter, public m_pi
some mathematical constants
This module implements a simple hash table for non-negative integer keys and integer values.
This module implements the index, used for the mesh points.
This module defines the meshes, which are used in Octopus.
integer(int64) function, public mesh_periodic_point(mesh, space, ip)
This function returns the point inside the grid corresponding to a boundary point when PBCs are used....
subroutine, public mesh_global_index_to_coords(mesh, ipg, ix)
Given a global point index, this function returns the set of integer coordinates of the point.
integer function, public mesh_local_index_from_coords(mesh, ix)
This function returns the local index of the point for a given vector of integer coordinates.
subroutine, public mesh_double_box(space, mesh, alpha, db)
finds the dimension of a box doubled in the non-periodic dimensions
integer(int64) function, public mesh_global_index_from_coords(mesh, ix)
This function returns the true global index of the point for a given vector of integer coordinates.
subroutine, public mesh_check_symmetries(mesh, symm, periodic_dim)
subroutine, public mesh_write_info(this, iunit, namespace)
subroutine, public mesh_local_index_to_coords(mesh, ip, ix)
Given a local point index, this function returns the set of integer coordinates of the point.
subroutine, public mesh_discretize_values_to_mesh(mesh, values)
Assign a set of values to their nearest discrete points on the mesh.
integer function, public mesh_nearest_point(mesh, pos, dmin, rankmin)
Returns the index of the point which is nearest to a given vector position pos.
real(real64) pure function, public mesh_global_memory(mesh)
subroutine, public mesh_read_fingerprint(mesh, dir, filename, mpi_grp, namespace, read_np_part, read_np, bits, type, offset, nn, ierr)
This function reads the fingerprint of a mesh written in filename. If the meshes are equal (same fing...
pure subroutine, public mesh_r(mesh, ip, rr, origin, coords)
return the distance to the origin for a given grid point
subroutine, public mesh_check_dump_compatibility(mesh, dir, filename, namespace, mpi_grp, grid_changed, grid_reordered, map, ierr)
real(real64) function, public mesh_gcutoff(mesh)
mesh_gcutoff returns the "natural" band limitation of the grid mesh, in terms of the maximum G vector...
subroutine, public mesh_minmaxloc(mesh, min_or_max, rank_min_or_max, op)
Given a local min/max this returns the global min/max and the rank where this is located.
integer function, public mesh_global2local(mesh, ipg)
This function returns the local mesh index for a given global index.
recursive subroutine, public mesh_end(this)
logical pure function, public mesh_compact_boundaries(mesh)
real(real64) pure function, public mesh_local_memory(mesh)
integer(int64) function, public mesh_local2global(mesh, ip)
This function returns the global mesh index for a given local index.
real(real64) function, dimension(1:mesh%box%dim), public mesh_x_global(mesh, ipg)
subroutine, public mesh_write_fingerprint(mesh, dir, filename, mpi_grp, namespace, ierr)
subroutine mesh_init(this)
subroutine, public messages_info(no_lines, iunit, verbose_limit, stress, all_nodes, namespace)
character(len=256), dimension(max_lines), public message
to be output by fatal, warning
This module contains some common usage patterns of MPI routines.
Some general things and nomenclature:
brief This module defines the class unit_t which is used by the unit_systems_oct_m module.
character(len=20) pure function, public units_abbrev(this)
This module defines the unit system, used for input and output.
type(unit_system_t), public units_out
abstract class for basis sets
This data type defines a line, and a regular grid defined on this line (or rather,...
define a grid on a plane.
Describes mesh distribution to nodes.
real(real64) function values(xx)