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),
allocatable :: chi(:,:)
115 real(real64) :: volume_element
116 real(real64),
allocatable :: vol_pp(:)
117 real(real64),
allocatable :: jacobian_inverse(:,:,:)
119 logical :: masked_periodic_boundaries
120 character(len=256) :: periodic_boundary_mask
126 procedure :: dmesh_allreduce_0, zmesh_allreduce_0, imesh_allreduce_0, lmesh_allreduce_0
127 procedure :: dmesh_allreduce_1, zmesh_allreduce_1, imesh_allreduce_1, lmesh_allreduce_1
128 procedure :: dmesh_allreduce_2, zmesh_allreduce_2, imesh_allreduce_2, lmesh_allreduce_2
129 procedure :: dmesh_allreduce_3, zmesh_allreduce_3, imesh_allreduce_3, lmesh_allreduce_3
130 procedure :: dmesh_allreduce_4, zmesh_allreduce_4, imesh_allreduce_4, lmesh_allreduce_4
131 procedure :: dmesh_allreduce_5, zmesh_allreduce_5, imesh_allreduce_5, lmesh_allreduce_5
132 procedure :: dmesh_allreduce_6, zmesh_allreduce_6, imesh_allreduce_6, lmesh_allreduce_6
133 generic :: allreduce => dmesh_allreduce_0, zmesh_allreduce_0, imesh_allreduce_0, lmesh_allreduce_0
134 generic :: allreduce => dmesh_allreduce_1, zmesh_allreduce_1, imesh_allreduce_1, lmesh_allreduce_1
135 generic :: allreduce => dmesh_allreduce_2, zmesh_allreduce_2, imesh_allreduce_2, lmesh_allreduce_2
136 generic :: allreduce => dmesh_allreduce_3, zmesh_allreduce_3, imesh_allreduce_3, lmesh_allreduce_3
137 generic :: allreduce => dmesh_allreduce_4, zmesh_allreduce_4, imesh_allreduce_4, lmesh_allreduce_4
138 generic :: allreduce => dmesh_allreduce_5, zmesh_allreduce_5, imesh_allreduce_5, lmesh_allreduce_5
139 generic :: allreduce => dmesh_allreduce_6, zmesh_allreduce_6, imesh_allreduce_6, lmesh_allreduce_6
156 real(real64) :: u(3), v(3)
157 real(real64) :: origin(3)
158 real(real64) :: spacing
159 integer :: nu, mu, nv, mv
169 real(real64) :: origin(2)
170 real(real64) :: spacing
177 class(mesh_t),
intent(inout) :: this
181 call this%set_time_dependent(.false.)
191 real(real64),
intent(in) :: alpha
192 integer,
intent(out) :: db(:)
201 do idir = 1, space%periodic_dim
202 db(idir) = mesh%idx%ll(idir)
204 do idir = space%periodic_dim + 1, space%dim
205 db(idir) = nint(alpha * (mesh%idx%ll(idir) - 1)) + 1
215 integer,
optional,
intent(in) :: iunit
216 type(
namespace_t),
optional,
intent(in) :: namespace
219 real(real64) :: cutoff
224 do ii = 1, this%box%dim
232 write(
message(2),
'(a, i10)')
' # inner mesh = ', this%np_global
233 write(
message(3),
'(a, i10)')
' # total mesh = ', this%np_part_global
245 pure subroutine mesh_r(mesh, ip, rr, origin, coords)
246 class(
mesh_t),
intent(in) :: mesh
247 integer,
intent(in) :: ip
248 real(real64),
intent(out) :: rr
249 real(real64),
intent(in),
optional :: origin(:)
250 real(real64),
intent(out),
optional :: coords(:)
252 real(real64) :: xx(1:mesh%box%dim)
255 if (
present(origin)) xx = xx - origin
258 if (
present(coords))
then
267 integer(int64),
intent(in) :: ipg
268 real(real64),
intent(out) :: rr
269 real(real64),
intent(in),
optional :: origin(:)
270 real(real64),
intent(out),
optional :: coords(:)
272 real(real64) :: xx(1:mesh%box%dim)
275 if (
present(origin)) xx = xx - origin
278 if (
present(coords))
then
290 class(
mesh_t),
intent(in) :: mesh
291 real(real64),
intent(in) :: pos(:)
292 real(real64),
intent(out) :: dmin
293 integer,
intent(out) :: rankmin
304 dd = sum((pos - mesh%x(ip, :))**2)
305 if ((dd < dmin) .or. (ip == 1))
then
312 call mesh%mpi_grp%bcast(imin, 1, mpi_integer, rankmin)
320 class(
mesh_t),
intent(in ) :: mesh
321 real(real64),
intent(inout) ::
values(:, :)
323 integer :: i, ip, ndim
325 real(real64) :: dummy
329 if (mesh%parallel_in_domains)
then
334 if (mesh%mpi_grp%rank == process)
values(:, i) = mesh%x(ip, :)
335 call mesh%mpi_grp%bcast(
values(:, i), ndim, mpi_double_precision, process)
354 class(
mesh_t),
intent(in) :: mesh
357 gmax =
m_pi / (maxval(mesh%spacing))
364 type(
mesh_t),
intent(in) :: mesh
365 character(len=*),
intent(in) :: dir
366 character(len=*),
intent(in) :: filename
367 type(mpi_grp_t),
intent(in) :: mpi_grp
368 type(namespace_t),
intent(in) :: namespace
369 integer,
intent(out) :: ierr
377 iunit = io_open(trim(dir)//
"/"//trim(filename), namespace, action=
'write', &
378 die=.false., grp=mpi_grp)
379 if (iunit == -1)
then
380 message(1) =
"Unable to open file '"//trim(dir)//
"/"//trim(filename)//
"'."
381 call messages_warning(1, namespace=namespace)
384 if (mpi_grp%is_root())
then
385 write(iunit,
'(a20,i21)')
'np_part_global= ', mesh%np_part_global
386 write(iunit,
'(a20,i21)')
'np_global= ', mesh%np_global
387 write(iunit,
'(a20,i21)')
'algorithm= ', 1
388 write(iunit,
'(a20,i21)')
'checksum= ', mesh%idx%checksum
389 write(iunit,
'(a20,i21)')
'bits= ', mesh%idx%bits
390 write(iunit,
'(a20,i21)')
'dim= ', mesh%idx%dim
391 write(iunit,
'(a20,i21)')
'type= ', mesh%idx%type
392 do ii = 1, mesh%idx%dim
393 write(iunit,
'(a7,i2,a11,i21)')
'offset(',ii,
')= ', mesh%idx%offset(ii)
395 do ii = 1, mesh%idx%dim
396 write(iunit,
'(a7,i2,a11,i21)')
'nn(',ii,
')= ', mesh%idx%nr(2, ii) - mesh%idx%nr(1, ii) + 1
399 call io_close(iunit, grp=mpi_grp)
412 read_np_part, read_np, bits, type, offset, nn, ierr)
413 type(
mesh_t),
intent(in) :: mesh
414 character(len=*),
intent(in) :: dir
415 character(len=*),
intent(in) :: filename
416 type(mpi_grp_t),
intent(in) :: mpi_grp
417 type(namespace_t),
intent(in) :: namespace
418 integer(int64),
intent(out) :: read_np_part
419 integer(int64),
intent(out) :: read_np
420 integer,
intent(out) :: bits
421 integer,
intent(out) :: type
422 integer,
intent(out) :: offset(1:mesh%idx%dim)
423 integer,
intent(out) :: nn(1:mesh%idx%dim)
424 integer,
intent(out) :: ierr
426 character(len=20) :: str
427 character(len=100) :: lines(7)
428 integer :: iunit, algorithm, dim, err, ii
429 integer(int64) :: checksum
435 read_np_part = 0_int64
438 iunit = io_open(trim(dir)//
"/"//trim(filename), namespace, action=
'read', &
439 status=
'old', die=.false., grp=mpi_grp)
440 if (iunit == -1)
then
442 message(1) =
"Unable to open file '"//trim(dir)//
"/"//trim(filename)//
"'."
443 call messages_warning(1, namespace=namespace)
445 call iopar_read(mpi_grp, iunit, lines, 7, err)
449 read(lines(1),
'(a20,i21)') str, read_np_part
450 read(lines(2),
'(a20,i21)') str, read_np
451 read(lines(3),
'(a20,i21)') str, algorithm
452 read(lines(4),
'(a20,i21)') str, checksum
453 read(lines(5),
'(a20,i21)') str, bits
454 read(lines(6),
'(a20,i21)') str, dim
455 read(lines(7),
'(a20,i21)') str,
type
457 if (dim /= mesh%idx%dim)
then
461 call iopar_read(mpi_grp, iunit, lines, dim, err)
466 read(lines(ii),
'(a20,i21)') str, offset(ii)
471 call iopar_read(mpi_grp, iunit, lines, dim, err)
476 read(lines(ii),
'(a20,i21)') str, nn(ii)
481 assert(read_np_part >= read_np)
483 if (read_np_part == mesh%np_part_global &
484 .and. read_np == mesh%np_global &
485 .and. algorithm == 1 &
486 .and. checksum == mesh%idx%checksum)
then
492 call io_close(iunit, grp=mpi_grp)
500 type(
mesh_t),
intent(in) :: mesh
501 character(len=*),
intent(in) :: dir
502 character(len=*),
intent(in) :: filename
503 type(namespace_t),
intent(in) :: namespace
504 type(mpi_grp_t),
intent(in) :: mpi_grp
505 logical,
intent(out) :: grid_changed
506 logical,
intent(out) :: grid_reordered
507 integer(int64),
allocatable,
intent(out) :: map(:)
508 integer,
intent(out) :: ierr
510 integer(int64) :: ipg, ipg_new, read_np_part, read_np
512 integer :: bits,
type, offset(mesh%idx%dim), point(mesh%idx%dim), nn(mesh%idx%dim)
513 integer(int64),
allocatable :: read_indices(:)
514 type(index_t) :: idx_old
520 grid_changed = .false.
521 grid_reordered = .false.
525 bits,
type, offset, nn, err)
528 message(1) =
"Unable to read mesh fingerprint from '"//trim(dir)//
"/"//trim(filename)//
"'."
529 call messages_warning(1, namespace=namespace)
531 else if (read_np > 0)
then
532 if (.not.
associated(mesh%box))
then
537 grid_changed = .
true.
542 grid_reordered = (read_np == mesh%np_global)
545 safe_allocate(read_indices(1:read_np_part))
546 call io_binary_read(trim(io_workpath(dir, namespace))//
"/indices.obf", read_np_part, &
550 message(1) =
"Unable to read index map from '"//trim(dir)//
"'."
551 call messages_warning(1, namespace=namespace)
554 call index_init(idx_old, mesh%idx%dim)
557 idx_old%nr(1, :) = -offset
558 idx_old%nr(2, :) = -offset + nn - 1
559 idx_old%offset = offset
560 idx_old%stride(1) = 1
561 do idir = 2, mesh%idx%dim
562 idx_old%stride(idir) = idx_old%stride(idir-1) * nn(idir-1)
565 safe_allocate(map(1:read_np))
568 call index_spatial_to_point(idx_old, mesh%idx%dim, read_indices(ipg), point)
573 if (map(ipg) > mesh%np_global) map(ipg) = 0
575 if (map(ipg) == 0) grid_reordered = .false.
577 call index_end(idx_old)
580 safe_deallocate_a(read_indices)
590 class(
mesh_t),
intent(inout) :: this
595 call lmpi_destroy_shared_memory_window(this%idx%window_grid_to_spatial)
596 call lmpi_destroy_shared_memory_window(this%idx%window_spatial_to_grid)
597 nullify(this%idx%grid_to_spatial_global)
598 nullify(this%idx%spatial_to_grid_global)
600 safe_deallocate_p(this%idx%grid_to_spatial_global)
601 safe_deallocate_p(this%idx%spatial_to_grid_global)
604 safe_deallocate_a(this%x)
605 safe_deallocate_a(this%chi)
606 safe_deallocate_a(this%vol_pp)
607 safe_deallocate_a(this%jacobian_inverse)
609 if (this%parallel_in_domains)
then
610 call par_vec_end(this%pv)
611 call partition_end(this%partition)
614 call index_end(this%idx)
615 safe_deallocate_a(this%spacing)
628 class(
mesh_t),
intent(in) :: mesh
629 class(space_t),
intent(in) :: space
630 integer,
intent(in) :: ip
632 integer :: ix(space%dim), nr(2, space%dim), idim
633 real(real64) :: xx(space%dim), rr, ufn_re, ufn_im
638 nr(1, :) = mesh%idx%nr(1, :) + mesh%idx%enlarge(:)
639 nr(2, :) = mesh%idx%nr(2, :) - mesh%idx%enlarge(:)
641 do idim = 1, space%periodic_dim
642 do while (ix(idim) < nr(1, idim))
643 ix(idim) = ix(idim) + mesh%idx%ll(idim)
645 do while (ix(idim) > nr(2, idim))
646 ix(idim) = ix(idim) - mesh%idx%ll(idim)
653 if (mesh%masked_periodic_boundaries)
then
654 call mesh_r(mesh, ip, rr, coords = xx)
655 call parse_expression(ufn_re, ufn_im, space%dim, xx, rr, m_zero, mesh%periodic_boundary_mask)
662 class(
mesh_t),
intent(in) :: mesh
663 class(space_t),
intent(in) :: space
664 integer(int64),
intent(in) :: ipg_in
666 integer :: ix(space%dim), nr(2, space%dim), idim
667 real(real64) :: xx(space%dim), rr, ufn_re, ufn_im
672 nr(1, :) = mesh%idx%nr(1, :) + mesh%idx%enlarge(:)
673 nr(2, :) = mesh%idx%nr(2, :) - mesh%idx%enlarge(:)
675 do idim = 1, space%periodic_dim
676 if (ix(idim) < nr(1, idim)) ix(idim) = ix(idim) + mesh%idx%ll(idim)
677 if (ix(idim) > nr(2, idim)) ix(idim) = ix(idim) - mesh%idx%ll(idim)
683 if (mesh%masked_periodic_boundaries)
then
685 call parse_expression(ufn_re, ufn_im, space%dim, xx, rr, m_zero, mesh%periodic_boundary_mask)
686 if (int(ufn_re) == 0) ipg = ipg_in
695 use iso_c_binding,
only: c_sizeof, c_long_long
696 class(
mesh_t),
intent(in) :: mesh
699 memory = c_sizeof(c_long_long) * real(mesh%np_part_global, real64) * 2
706 use iso_c_binding,
only: c_sizeof, c_long_long
707 class(
mesh_t),
intent(in) :: mesh
712 memory = memory + real64 * real(mesh%np_part, real64) * mesh%idx%dim
714 memory = memory + c_sizeof(c_long_long) * real(mesh%np_part, real64) * 2
721 class(
mesh_t),
intent(in) :: mesh
722 integer(int64),
intent(in) :: ipg
723 real(real64) :: xx(1:mesh%box%dim)
725 real(real64) :: chi(1:mesh%box%dim)
726 integer :: ix(1:mesh%box%dim)
731 chi = ix * mesh%spacing
732 xx = mesh%coord_system%to_cartesian(chi)
739 class(
mesh_t),
intent(in) :: mesh
740 type(symmetries_t),
intent(in) :: symm
741 integer,
intent(in) :: periodic_dim
743 integer :: iop, ip, idim, nops, ix(1:3)
744 real(real64) :: destpoint(1:3), srcpoint(1:3), lsize(1:3), offset(1:3)
745 real(real64),
parameter :: tol_spacing = 1e-12_real64
753 if (mesh%idx%ll(1) == mesh%idx%ll(2) .and. &
754 mesh%idx%ll(2) == mesh%idx%ll(3) .and. &
755 abs(mesh%spacing(1) - mesh%spacing(2)) < tol_spacing .and. &
756 abs(mesh%spacing(2) - mesh%spacing(3)) < tol_spacing)
return
760 message(1) =
"Checking if the real-space grid is symmetric"
761 call messages_info(1)
763 lsize(1:3) = real(mesh%idx%ll(1:3), real64)
764 offset(1:3) = real(mesh%idx%nr(1, 1:3) + mesh%idx%enlarge(1:3), real64)
766 nops = symmetries_number(symm)
773 destpoint(1:3) = real(ix(1:3), real64) - offset(1:3)
775 assert(all(destpoint >= 0))
776 assert(all(destpoint < lsize))
779 destpoint = destpoint - real(int(lsize)/2, real64)
783 destpoint(idim) = destpoint(idim)/lsize(idim)
788 srcpoint = symm_op_apply_red(symm%ops(iop), destpoint)
792 srcpoint(idim) = srcpoint(idim)*lsize(idim)
796 srcpoint = srcpoint + real(int(lsize)/2, real64)
799 do idim = 1, periodic_dim
800 if (nint(srcpoint(idim)) < 0 .or. nint(srcpoint(idim)) >= lsize(idim))
then
801 srcpoint(idim) = modulo(srcpoint(idim)+m_half*symprec, lsize(idim))
804 assert(all(srcpoint >= -symprec))
805 assert(all(srcpoint < lsize))
807 srcpoint(1:3) = srcpoint(1:3) + offset(1:3)
809 if (any(srcpoint-anint(srcpoint)> symprec*m_two))
then
810 message(1) =
"The real-space grid breaks at least one of the symmetries of the system."
811 message(2) =
"Change your spacing or use SymmetrizeDensity=no."
812 call messages_fatal(2)
823 class(
mesh_t),
intent(in) :: mesh
824 integer,
intent(in) :: ix(:)
826 index = index_from_coords(mesh%idx, ix)
832 type(
mesh_t),
intent(in) :: mesh
833 integer(int64),
intent(in) :: ipg
834 integer,
intent(out) :: ix(:)
836 call index_to_coords(mesh%idx, ipg, ix)
842 type(
mesh_t),
intent(in) :: mesh
843 integer,
intent(in) :: ix(:)
845 integer(int64) :: ipg
847 ipg = index_from_coords(mesh%idx, ix)
854 type(
mesh_t),
intent(in) :: mesh
855 integer,
intent(in) :: ip
856 integer,
intent(out) :: ix(:)
858 integer(int64) :: ipg
861 call index_to_coords(mesh%idx, ipg, ix)
866 type(
mesh_t),
intent(in) :: mesh
867 integer,
intent(in) :: ip
869 ipg = par_vec_local2global(mesh%pv, ip)
876 type(
mesh_t),
intent(in) :: mesh
877 integer(int64),
intent(in) :: ipg
879 ip = par_vec_global2local(mesh%pv, ipg)
885 type(
mesh_t),
intent(in) :: mesh
886 real(real64),
intent(inout) :: min_or_max
887 integer,
intent(out) :: rank_min_or_max
888 type(mpi_op),
intent(in) :: op
890 real(real64) :: loc_in(2), loc_out(2)
894 assert(op == mpi_minloc .or. op == mpi_maxloc)
897 if (mesh%parallel_in_domains)
then
898 loc_in(1) = min_or_max
899 loc_in(2) = mesh%mpi_grp%rank
900 call mesh%mpi_grp%allreduce(loc_in, loc_out, 1, mpi_2double_precision, op)
901 min_or_max = loc_out(1)
902 rank_min_or_max = nint(loc_out(2))
911#include "mesh_inc.F90"
914#include "complex.F90"
915#include "mesh_inc.F90"
918#include "integer.F90"
919#include "mesh_inc.F90"
922#include "integer8.F90"
923#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.
integer(int64) function, public mesh_periodic_point_global(mesh, space, ipg_in)
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)
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)
Given a global point index, this function returns the coordinates of the point.
subroutine mesh_r_global(mesh, ipg, rr, origin, coords)
return the distance to the origin for a given grid point
subroutine, public mesh_write_fingerprint(mesh, dir, filename, mpi_grp, namespace, ierr)
subroutine mesh_init(this)
character(len=256), dimension(max_lines), public message
to be output by fatal, warning
subroutine, public messages_info(no_lines, iunit, debug_only, stress, all_nodes, namespace)
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)