Line data Source code
1 : !--------------------------------------------------------------------------------------------------!
2 : ! CP2K: A general program to perform molecular dynamics simulations !
3 : ! Copyright 2000-2024 CP2K developers group <https://cp2k.org> !
4 : ! !
5 : ! SPDX-License-Identifier: GPL-2.0-or-later !
6 : !--------------------------------------------------------------------------------------------------!
7 : MODULE qs_rho_atom_methods
8 :
9 : USE atomic_kind_types, ONLY: atomic_kind_type,&
10 : get_atomic_kind,&
11 : get_atomic_kind_set
12 : USE basis_set_types, ONLY: get_gto_basis_set,&
13 : gto_basis_set_p_type,&
14 : gto_basis_set_type
15 : USE cp_control_types, ONLY: dft_control_type,&
16 : gapw_control_type
17 : USE cp_dbcsr_api, ONLY: dbcsr_get_block_p,&
18 : dbcsr_p_type
19 : USE kinds, ONLY: dp
20 : USE kpoint_types, ONLY: get_kpoint_info,&
21 : kpoint_type
22 : USE lebedev, ONLY: deallocate_lebedev_grids,&
23 : get_number_of_lebedev_grid,&
24 : init_lebedev_grids,&
25 : lebedev_grid
26 : USE mathconstants, ONLY: fourpi,&
27 : pi
28 : USE memory_utilities, ONLY: reallocate
29 : USE message_passing, ONLY: mp_para_env_type
30 : USE orbital_pointers, ONLY: indso,&
31 : nsoset
32 : USE paw_basis_types, ONLY: get_paw_basis_info
33 : USE qs_environment_types, ONLY: get_qs_env,&
34 : qs_environment_type
35 : USE qs_grid_atom, ONLY: create_grid_atom,&
36 : grid_atom_type
37 : USE qs_harmonics_atom, ONLY: create_harmonics_atom,&
38 : get_maxl_CG,&
39 : get_none0_cg_list,&
40 : harmonics_atom_type
41 : USE qs_kind_types, ONLY: get_qs_kind,&
42 : get_qs_kind_set,&
43 : qs_kind_type
44 : USE qs_neighbor_list_types, ONLY: get_iterator_info,&
45 : neighbor_list_iterate,&
46 : neighbor_list_iterator_create,&
47 : neighbor_list_iterator_p_type,&
48 : neighbor_list_iterator_release,&
49 : neighbor_list_set_p_type
50 : USE qs_oce_methods, ONLY: proj_blk
51 : USE qs_oce_types, ONLY: oce_matrix_type
52 : USE qs_rho_atom_types, ONLY: deallocate_rho_atom_set,&
53 : rho_atom_coeff,&
54 : rho_atom_type
55 : USE sap_kind_types, ONLY: alist_pre_align_blk,&
56 : alist_type,&
57 : get_alist
58 : USE spherical_harmonics, ONLY: clebsch_gordon,&
59 : clebsch_gordon_deallocate,&
60 : clebsch_gordon_init
61 : USE util, ONLY: get_limit
62 : USE whittaker, ONLY: whittaker_c0a,&
63 : whittaker_ci
64 :
65 : !$ USE OMP_LIB, ONLY: omp_get_max_threads, &
66 : !$ omp_get_thread_num, &
67 : !$ omp_lock_kind, &
68 : !$ omp_init_lock, omp_set_lock, &
69 : !$ omp_unset_lock, omp_destroy_lock
70 :
71 : #include "./base/base_uses.f90"
72 :
73 : IMPLICIT NONE
74 :
75 : PRIVATE
76 :
77 : ! *** Global parameters (only in this module)
78 :
79 : CHARACTER(len=*), PARAMETER, PRIVATE :: moduleN = 'qs_rho_atom_methods'
80 :
81 : ! *** Public subroutines ***
82 :
83 : PUBLIC :: allocate_rho_atom_internals, &
84 : calculate_rho_atom, &
85 : calculate_rho_atom_coeff, &
86 : init_rho_atom
87 :
88 : CONTAINS
89 :
90 : ! **************************************************************************************************
91 : !> \brief ...
92 : !> \param para_env ...
93 : !> \param rho_atom_set ...
94 : !> \param qs_kind ...
95 : !> \param atom_list ...
96 : !> \param natom ...
97 : !> \param nspins ...
98 : !> \param tot_rho1_h ...
99 : !> \param tot_rho1_s ...
100 : ! **************************************************************************************************
101 43966 : SUBROUTINE calculate_rho_atom(para_env, rho_atom_set, qs_kind, atom_list, &
102 43966 : natom, nspins, tot_rho1_h, tot_rho1_s)
103 :
104 : TYPE(mp_para_env_type), POINTER :: para_env
105 : TYPE(rho_atom_type), DIMENSION(:), POINTER :: rho_atom_set
106 : TYPE(qs_kind_type), INTENT(IN) :: qs_kind
107 : INTEGER, DIMENSION(:), INTENT(IN) :: atom_list
108 : INTEGER, INTENT(IN) :: natom, nspins
109 : REAL(dp), DIMENSION(:), INTENT(INOUT) :: tot_rho1_h, tot_rho1_s
110 :
111 : CHARACTER(len=*), PARAMETER :: routineN = 'calculate_rho_atom'
112 :
113 : INTEGER :: damax_iso_not0_local, handle, i, i1, i2, iat, iatom, icg, ipgf1, ipgf2, ir, &
114 : iset1, iset2, iso, iso1, iso1_first, iso1_last, iso2, iso2_first, iso2_last, j, l, l1, &
115 : l2, l_iso, l_sub, l_sum, lmax12, lmax_expansion, lmin12, m1s, m2s, max_iso_not0, &
116 : max_iso_not0_local, max_s_harm, maxl, maxso, mepos, n1s, n2s, nr, nset, num_pe, size1, &
117 : size2
118 43966 : INTEGER, ALLOCATABLE, DIMENSION(:) :: cg_n_list, dacg_n_list
119 43966 : INTEGER, ALLOCATABLE, DIMENSION(:, :, :) :: cg_list, dacg_list
120 : INTEGER, DIMENSION(2) :: bo
121 43966 : INTEGER, DIMENSION(:), POINTER :: lmax, lmin, npgf, o2nindex
122 43966 : LOGICAL, ALLOCATABLE, DIMENSION(:, :) :: done_vgg
123 : REAL(dp) :: c1, c2, rho_h, rho_s, root_zet12, zet12
124 43966 : REAL(dp), ALLOCATABLE, DIMENSION(:) :: erf_zet12, g1, g2, gg0, int1, int2
125 43966 : REAL(dp), ALLOCATABLE, DIMENSION(:, :) :: CPCH_sphere, CPCS_sphere, dgg, gg, gg_lm1
126 43966 : REAL(dp), ALLOCATABLE, DIMENSION(:, :, :) :: vgg
127 43966 : REAL(dp), DIMENSION(:, :), POINTER :: coeff, zet
128 : REAL(dp), DIMENSION(:, :, :), POINTER :: my_CG
129 : REAL(dp), DIMENSION(:, :, :, :), POINTER :: my_CG_dxyz
130 : TYPE(grid_atom_type), POINTER :: grid_atom
131 : TYPE(gto_basis_set_type), POINTER :: basis_1c
132 : TYPE(harmonics_atom_type), POINTER :: harmonics
133 :
134 43966 : CALL timeset(routineN, handle)
135 :
136 : !Note: tau is taken care of separately in qs_vxc_atom.F
137 :
138 43966 : NULLIFY (basis_1c)
139 43966 : NULLIFY (harmonics, grid_atom)
140 43966 : NULLIFY (lmin, lmax, npgf, zet, my_CG, my_CG_dxyz, coeff)
141 :
142 43966 : CALL get_qs_kind(qs_kind, grid_atom=grid_atom, harmonics=harmonics)
143 43966 : CALL get_qs_kind(qs_kind, basis_set=basis_1c, basis_type="GAPW_1C")
144 :
145 : CALL get_gto_basis_set(gto_basis_set=basis_1c, lmax=lmax, lmin=lmin, &
146 : maxl=maxl, npgf=npgf, nset=nset, zet=zet, &
147 43966 : maxso=maxso)
148 :
149 43966 : CALL get_paw_basis_info(basis_1c, o2nindex=o2nindex)
150 :
151 43966 : max_iso_not0 = harmonics%max_iso_not0
152 43966 : max_s_harm = harmonics%max_s_harm
153 :
154 43966 : nr = grid_atom%nr
155 43966 : lmax_expansion = indso(1, max_iso_not0)
156 : ! Distribute the atoms of this kind
157 43966 : num_pe = para_env%num_pe
158 43966 : mepos = para_env%mepos
159 43966 : bo = get_limit(natom, num_pe, mepos)
160 :
161 43966 : my_CG => harmonics%my_CG
162 43966 : my_CG_dxyz => harmonics%my_CG_dxyz
163 :
164 175864 : ALLOCATE (CPCH_sphere(nsoset(maxl), nsoset(maxl)))
165 131898 : ALLOCATE (CPCS_sphere(nsoset(maxl), nsoset(maxl)))
166 527592 : ALLOCATE (g1(nr), g2(nr), gg0(nr), gg(nr, 0:2*maxl), dgg(nr, 0:2*maxl), gg_lm1(nr, 0:2*maxl))
167 263796 : ALLOCATE (erf_zet12(nr), vgg(nr, 0:2*maxl, 0:indso(1, max_iso_not0)))
168 175864 : ALLOCATE (done_vgg(0:2*maxl, 0:indso(1, max_iso_not0)))
169 131898 : ALLOCATE (int1(nr), int2(nr))
170 : ALLOCATE (cg_list(2, nsoset(maxl)**2, max_s_harm), cg_n_list(max_s_harm), &
171 395694 : dacg_list(2, nsoset(maxl)**2, max_s_harm), dacg_n_list(max_s_harm))
172 :
173 77544 : DO iat = bo(1), bo(2)
174 33578 : iatom = atom_list(iat)
175 116242 : DO i = 1, nspins
176 72276 : IF (.NOT. ASSOCIATED(rho_atom_set(iatom)%rho_rad_h(i)%r_coef)) THEN
177 4114 : CALL allocate_rho_atom_rad(rho_atom_set, iatom, i, nr, max_iso_not0)
178 : ELSE
179 34584 : CALL set2zero_rho_atom_rad(rho_atom_set, iatom, i)
180 : END IF
181 : END DO
182 : END DO
183 :
184 43966 : m1s = 0
185 158074 : DO iset1 = 1, nset
186 : m2s = 0
187 509776 : DO iset2 = 1, nset
188 :
189 : CALL get_none0_cg_list(my_CG, lmin(iset1), lmax(iset1), lmin(iset2), lmax(iset2), &
190 395668 : max_s_harm, lmax_expansion, cg_list, cg_n_list, max_iso_not0_local)
191 395668 : CPASSERT(max_iso_not0_local .LE. max_iso_not0)
192 : CALL get_none0_cg_list(my_CG_dxyz, lmin(iset1), lmax(iset1), lmin(iset2), lmax(iset2), &
193 395668 : max_s_harm, lmax_expansion, dacg_list, dacg_n_list, damax_iso_not0_local)
194 395668 : n1s = nsoset(lmax(iset1))
195 :
196 1286458 : DO ipgf1 = 1, npgf(iset1)
197 890790 : iso1_first = nsoset(lmin(iset1) - 1) + 1 + n1s*(ipgf1 - 1) + m1s
198 890790 : iso1_last = nsoset(lmax(iset1)) + n1s*(ipgf1 - 1) + m1s
199 890790 : size1 = iso1_last - iso1_first + 1
200 890790 : iso1_first = o2nindex(iso1_first)
201 890790 : iso1_last = o2nindex(iso1_last)
202 890790 : i1 = iso1_last - iso1_first + 1
203 890790 : CPASSERT(size1 == i1)
204 890790 : i1 = nsoset(lmin(iset1) - 1) + 1
205 :
206 49986050 : g1(1:nr) = EXP(-zet(ipgf1, iset1)*grid_atom%rad2(1:nr))
207 :
208 890790 : n2s = nsoset(lmax(iset2))
209 3770482 : DO ipgf2 = 1, npgf(iset2)
210 2484024 : iso2_first = nsoset(lmin(iset2) - 1) + 1 + n2s*(ipgf2 - 1) + m2s
211 2484024 : iso2_last = nsoset(lmax(iset2)) + n2s*(ipgf2 - 1) + m2s
212 2484024 : size2 = iso2_last - iso2_first + 1
213 2484024 : iso2_first = o2nindex(iso2_first)
214 2484024 : iso2_last = o2nindex(iso2_last)
215 2484024 : i2 = iso2_last - iso2_first + 1
216 2484024 : CPASSERT(size2 == i2)
217 2484024 : i2 = nsoset(lmin(iset2) - 1) + 1
218 :
219 140279444 : g2(1:nr) = EXP(-zet(ipgf2, iset2)*grid_atom%rad2(1:nr))
220 2484024 : lmin12 = lmin(iset1) + lmin(iset2)
221 2484024 : lmax12 = lmax(iset1) + lmax(iset2)
222 :
223 2484024 : zet12 = zet(ipgf1, iset1) + zet(ipgf2, iset2)
224 2484024 : root_zet12 = SQRT(zet(ipgf1, iset1) + zet(ipgf2, iset2))
225 140279444 : DO ir = 1, nr
226 140279444 : erf_zet12(ir) = erf(root_zet12*grid_atom%rad(ir))
227 : END DO
228 :
229 646822452 : gg = 0.0_dp
230 646822452 : dgg = 0.0_dp
231 646822452 : gg_lm1 = 0.0_dp
232 3247902380 : vgg = 0.0_dp
233 71784040 : done_vgg = .FALSE.
234 : ! reduce the number of terms in the expansion local densities
235 2484024 : IF (lmin12 .LE. lmax_expansion) THEN
236 2481534 : IF (lmin12 == 0) THEN
237 84126938 : gg(1:nr, lmin12) = g1(1:nr)*g2(1:nr)
238 84126938 : gg_lm1(1:nr, lmin12) = 0.0_dp
239 84126938 : gg0(1:nr) = gg(1:nr, lmin12)
240 : ELSE
241 56025516 : gg0(1:nr) = g1(1:nr)*g2(1:nr)
242 56025516 : gg(1:nr, lmin12) = grid_atom%rad2l(1:nr, lmin12)*g1(1:nr)*g2(1:nr)
243 56025516 : gg_lm1(1:nr, lmin12) = grid_atom%rad2l(1:nr, lmin12 - 1)*g1(1:nr)*g2(1:nr)
244 : END IF
245 :
246 : ! reduce the number of terms in the expansion local densities
247 2481534 : IF (lmax12 .GT. lmax_expansion) lmax12 = lmax_expansion
248 :
249 3814398 : DO l = lmin12 + 1, lmax12
250 78557184 : gg(1:nr, l) = grid_atom%rad(1:nr)*gg(1:nr, l - 1)
251 78557184 : gg_lm1(1:nr, l) = gg(1:nr, l - 1)
252 81038718 : dgg(1:nr, l - 1) = -2.0_dp*(zet(ipgf1, iset1) + zet(ipgf2, iset2))*gg(1:nr, l)
253 :
254 : END DO
255 : dgg(1:nr, lmax12) = -2.0_dp*(zet(ipgf1, iset1) + &
256 140152454 : zet(ipgf2, iset2))*grid_atom%rad(1:nr)*gg(1:nr, lmax12)
257 :
258 2481534 : c2 = SQRT(pi*pi*pi/(zet12*zet12*zet12))
259 :
260 19664804 : DO iso = 1, max_iso_not0_local
261 17183270 : l_iso = indso(1, iso)
262 17183270 : c1 = fourpi/(2._dp*REAL(l_iso, dp) + 1._dp)
263 52987690 : DO icg = 1, cg_n_list(iso)
264 33322886 : iso1 = cg_list(1, icg, iso)
265 33322886 : iso2 = cg_list(2, icg, iso)
266 :
267 33322886 : l = indso(1, iso1) + indso(1, iso2)
268 33322886 : CPASSERT(l <= lmax_expansion)
269 33322886 : IF (done_vgg(l, l_iso)) CYCLE
270 4826418 : L_sum = l + l_iso
271 4826418 : L_sub = l - l_iso
272 :
273 4826418 : IF (l_sum == 0) THEN
274 84126938 : vgg(1:nr, l, l_iso) = erf_zet12(1:nr)*grid_atom%oorad2l(1:nr, 1)*c2
275 : ELSE
276 3404700 : CALL whittaker_c0a(int1, grid_atom%rad, gg0, erf_zet12, zet12, l, l_iso, nr)
277 3404700 : CALL whittaker_ci(int2, grid_atom%rad, gg0, zet12, L_sub, nr)
278 :
279 189935060 : DO ir = 1, nr
280 186530360 : int2(ir) = grid_atom%rad2l(ir, l_iso)*int2(ir)
281 189935060 : vgg(ir, l, l_iso) = c1*(int1(ir) + int2(ir))
282 : END DO
283 : END IF
284 50506156 : done_vgg(l, l_iso) = .TRUE.
285 : END DO
286 : END DO
287 : END IF ! lmax_expansion
288 :
289 5024343 : DO iat = bo(1), bo(2)
290 1649529 : iatom = atom_list(iat)
291 :
292 6150736 : DO i = 1, nspins
293 172301513 : CPCH_sphere = 0.0_dp
294 172301513 : CPCS_sphere = 0.0_dp
295 :
296 2017183 : coeff => rho_atom_set(iatom)%cpc_h(i)%r_coef
297 21196105 : CPCH_sphere(i1:i1 + size1 - 1, i2:i2 + size2 - 1) = coeff(iso1_first:iso1_last, iso2_first:iso2_last)
298 :
299 2017183 : coeff => rho_atom_set(iatom)%cpc_s(i)%r_coef
300 21196105 : CPCS_sphere(i1:i1 + size1 - 1, i2:i2 + size2 - 1) = coeff(iso1_first:iso1_last, iso2_first:iso2_last)
301 :
302 14087701 : DO iso = 1, max_iso_not0_local
303 12070518 : l_iso = indso(1, iso)
304 36560643 : DO icg = 1, cg_n_list(iso)
305 22472942 : iso1 = cg_list(1, icg, iso)
306 22472942 : iso2 = cg_list(2, icg, iso)
307 :
308 22472942 : l1 = indso(1, iso1)
309 22472942 : l2 = indso(1, iso2)
310 :
311 22472942 : l = indso(1, iso1) + indso(1, iso2)
312 22472942 : CPASSERT(l <= lmax_expansion)
313 :
314 : rho_atom_set(iatom)%rho_rad_h(i)%r_coef(1:nr, iso) = &
315 : rho_atom_set(iatom)%rho_rad_h(i)%r_coef(1:nr, iso) + &
316 1254406272 : gg(1:nr, l)*CPCH_sphere(iso1, iso2)*my_CG(iso1, iso2, iso)
317 :
318 : rho_atom_set(iatom)%rho_rad_s(i)%r_coef(1:nr, iso) = &
319 : rho_atom_set(iatom)%rho_rad_s(i)%r_coef(1:nr, iso) + &
320 1254406272 : gg(1:nr, l)*CPCS_sphere(iso1, iso2)*my_CG(iso1, iso2, iso)
321 :
322 : rho_atom_set(iatom)%drho_rad_h(i)%r_coef(1:nr, iso) = &
323 : rho_atom_set(iatom)%drho_rad_h(i)%r_coef(1:nr, iso) + &
324 1254406272 : dgg(1:nr, l)*CPCH_sphere(iso1, iso2)*my_CG(iso1, iso2, iso)
325 :
326 : rho_atom_set(iatom)%drho_rad_s(i)%r_coef(1:nr, iso) = &
327 : rho_atom_set(iatom)%drho_rad_s(i)%r_coef(1:nr, iso) + &
328 1254406272 : dgg(1:nr, l)*CPCS_sphere(iso1, iso2)*my_CG(iso1, iso2, iso)
329 :
330 : rho_atom_set(iatom)%vrho_rad_h(i)%r_coef(1:nr, iso) = &
331 : rho_atom_set(iatom)%vrho_rad_h(i)%r_coef(1:nr, iso) + &
332 1254406272 : vgg(1:nr, l, l_iso)*CPCH_sphere(iso1, iso2)*my_CG(iso1, iso2, iso)
333 :
334 : rho_atom_set(iatom)%vrho_rad_s(i)%r_coef(1:nr, iso) = &
335 : rho_atom_set(iatom)%vrho_rad_s(i)%r_coef(1:nr, iso) + &
336 1266476790 : vgg(1:nr, l, l_iso)*CPCS_sphere(iso1, iso2)*my_CG(iso1, iso2, iso)
337 :
338 : END DO ! icg
339 :
340 : END DO ! iso
341 :
342 46103095 : DO iso = 1, max_iso_not0 !damax_iso_not0_local
343 42436383 : l_iso = indso(1, iso)
344 80169969 : DO icg = 1, dacg_n_list(iso)
345 35716403 : iso1 = dacg_list(1, icg, iso)
346 35716403 : iso2 = dacg_list(2, icg, iso)
347 35716403 : l = indso(1, iso1) + indso(1, iso2)
348 35716403 : CPASSERT(l <= lmax_expansion)
349 185301995 : DO j = 1, 3
350 : rho_atom_set(iatom)%rho_rad_h_d(j, i)%r_coef(1:nr, iso) = &
351 : rho_atom_set(iatom)%rho_rad_h_d(j, i)%r_coef(1:nr, iso) + &
352 5860856229 : gg_lm1(1:nr, l)*CPCH_sphere(iso1, iso2)*my_CG_dxyz(j, iso1, iso2, iso)
353 :
354 : rho_atom_set(iatom)%rho_rad_s_d(j, i)%r_coef(1:nr, iso) = &
355 : rho_atom_set(iatom)%rho_rad_s_d(j, i)%r_coef(1:nr, iso) + &
356 5896572632 : gg_lm1(1:nr, l)*CPCS_sphere(iso1, iso2)*my_CG_dxyz(j, iso1, iso2, iso)
357 : END DO
358 : END DO ! icg
359 :
360 : END DO ! iso
361 :
362 : END DO ! i
363 : END DO ! iat
364 :
365 : END DO ! ipgf2
366 : END DO ! ipgf1
367 1301112 : m2s = m2s + maxso
368 : END DO ! iset2
369 158074 : m1s = m1s + maxso
370 : END DO ! iset1
371 :
372 77544 : DO iat = bo(1), bo(2)
373 33578 : iatom = atom_list(iat)
374 :
375 116242 : DO i = 1, nspins
376 :
377 576574 : DO iso = 1, max_iso_not0
378 : rho_s = 0.0_dp
379 : rho_h = 0.0_dp
380 28443918 : DO ir = 1, nr
381 27939620 : rho_h = rho_h + rho_atom_set(iatom)%rho_rad_h(i)%r_coef(ir, iso)*grid_atom%wr(ir)
382 28443918 : rho_s = rho_s + rho_atom_set(iatom)%rho_rad_s(i)%r_coef(ir, iso)*grid_atom%wr(ir)
383 : END DO ! ir
384 504298 : tot_rho1_h(i) = tot_rho1_h(i) + rho_h*harmonics%slm_int(iso)
385 542996 : tot_rho1_s(i) = tot_rho1_s(i) + rho_s*harmonics%slm_int(iso)
386 : END DO ! iso
387 :
388 : END DO ! ispin
389 :
390 : END DO ! iat
391 :
392 43966 : DEALLOCATE (CPCH_sphere, CPCS_sphere)
393 43966 : DEALLOCATE (g1, g2, gg0, gg, gg_lm1, dgg, vgg, done_vgg, erf_zet12, int1, int2)
394 43966 : DEALLOCATE (cg_list, cg_n_list, dacg_list, dacg_n_list)
395 43966 : DEALLOCATE (o2nindex)
396 :
397 43966 : CALL timestop(handle)
398 :
399 87932 : END SUBROUTINE calculate_rho_atom
400 :
401 : ! **************************************************************************************************
402 : !> \brief ...
403 : !> \param qs_env QuickStep environment
404 : !> (accessed components: atomic_kind_set, dft_control%nimages,
405 : !> dft_control%nspins, kpoints%cell_to_index)
406 : !> \param rho_ao density matrix in atomic basis set
407 : !> \param rho_atom_set ...
408 : !> \param qs_kind_set list of QuickStep kinds
409 : !> \param oce one-centre expansion coefficients
410 : !> \param sab neighbour pair list
411 : !> \param para_env parallel environment
412 : !> \par History
413 : !> Add OpenMP [Apr 2016, EPCC]
414 : !> Use automatic arrays [Sep 2016, M Tucker]
415 : !> Allow for external non-default kind_set, oce and sab [Dec 2019, A Bussy]
416 : !> \note Consider to declare 'rho_ao' dummy argument as a pointer to the two-dimensional
417 : !> (1:nspins, 1:nimages) set of matrices.
418 : ! **************************************************************************************************
419 23926 : SUBROUTINE calculate_rho_atom_coeff(qs_env, rho_ao, rho_atom_set, qs_kind_set, oce, sab, para_env)
420 :
421 : TYPE(qs_environment_type), POINTER :: qs_env
422 : TYPE(dbcsr_p_type), DIMENSION(*) :: rho_ao
423 : TYPE(rho_atom_type), DIMENSION(:), POINTER :: rho_atom_set
424 : TYPE(qs_kind_type), DIMENSION(:), POINTER :: qs_kind_set
425 : TYPE(oce_matrix_type), POINTER :: oce
426 : TYPE(neighbor_list_set_p_type), DIMENSION(:), &
427 : POINTER :: sab
428 : TYPE(mp_para_env_type), POINTER :: para_env
429 :
430 : CHARACTER(len=*), PARAMETER :: routineN = 'calculate_rho_atom_coeff'
431 :
432 : INTEGER :: bo(2), handle, i, iac, iatom, ibc, icol, ikind, img, irow, ispin, jatom, jkind, &
433 : kac, katom, kbc, kkind, len_CPC, len_PC1, max_gau, max_nsgf, mepos, n_cont_a, n_cont_b, &
434 : nat_kind, natom, nimages, nkind, nsatbas, nsoctot, nspins, num_pe
435 23926 : INTEGER, ALLOCATABLE, DIMENSION(:) :: kind_of
436 : INTEGER, DIMENSION(3) :: cell_b
437 23926 : INTEGER, DIMENSION(:), POINTER :: a_list, list_a, list_b
438 23926 : INTEGER, DIMENSION(:, :, :), POINTER :: cell_to_index
439 : LOGICAL :: dista, distab, distb, found, paw_atom
440 23926 : REAL(dp), ALLOCATABLE, DIMENSION(:, :) :: p_matrix
441 : REAL(KIND=dp) :: eps_cpc, factor, pmax
442 : REAL(KIND=dp), DIMENSION(3) :: rab
443 23926 : REAL(KIND=dp), DIMENSION(:, :), POINTER :: C_coeff_hh_a, C_coeff_hh_b, &
444 23926 : C_coeff_ss_a, C_coeff_ss_b, r_coef_h, &
445 23926 : r_coef_s
446 : TYPE(alist_type), POINTER :: alist_ac, alist_bc
447 23926 : TYPE(atomic_kind_type), DIMENSION(:), POINTER :: atomic_kind_set
448 : TYPE(dft_control_type), POINTER :: dft_control
449 23926 : TYPE(gto_basis_set_p_type), DIMENSION(:), POINTER :: basis_set_list
450 : TYPE(gto_basis_set_type), POINTER :: basis_1c, basis_set_a, basis_set_b
451 : TYPE(kpoint_type), POINTER :: kpoints
452 : TYPE(neighbor_list_iterator_p_type), &
453 23926 : DIMENSION(:), POINTER :: nl_iterator
454 23926 : TYPE(rho_atom_coeff), DIMENSION(:), POINTER :: p_block_spin
455 :
456 23926 : !$ INTEGER(kind=omp_lock_kind), ALLOCATABLE, DIMENSION(:) :: locks
457 : !$ INTEGER :: lock, number_of_locks
458 :
459 23926 : CALL timeset(routineN, handle)
460 :
461 : CALL get_qs_env(qs_env=qs_env, &
462 : dft_control=dft_control, &
463 23926 : atomic_kind_set=atomic_kind_set)
464 :
465 23926 : eps_cpc = dft_control%qs_control%gapw_control%eps_cpc
466 :
467 23926 : CPASSERT(ASSOCIATED(qs_kind_set))
468 23926 : CPASSERT(ASSOCIATED(rho_atom_set))
469 23926 : CPASSERT(ASSOCIATED(oce))
470 23926 : CPASSERT(ASSOCIATED(sab))
471 :
472 23926 : nspins = dft_control%nspins
473 23926 : nimages = dft_control%nimages
474 :
475 23926 : NULLIFY (cell_to_index)
476 23926 : IF (nimages > 1) THEN
477 228 : CALL get_qs_env(qs_env=qs_env, kpoints=kpoints)
478 228 : CALL get_kpoint_info(kpoint=kpoints, cell_to_index=cell_to_index)
479 : END IF
480 :
481 23926 : CALL get_atomic_kind_set(atomic_kind_set, natom=natom)
482 23926 : CALL get_qs_kind_set(qs_kind_set, maxsgf=max_nsgf, maxgtops=max_gau, basis_type='GAPW_1C')
483 :
484 23926 : nkind = SIZE(atomic_kind_set)
485 : ! Initialize to 0 the CPC coefficients and the local density arrays
486 73740 : DO ikind = 1, nkind
487 49814 : CALL get_atomic_kind(atomic_kind_set(ikind), atom_list=a_list, natom=nat_kind)
488 49814 : CALL get_qs_kind(qs_kind_set(ikind), paw_atom=paw_atom)
489 :
490 49814 : IF (.NOT. paw_atom) CYCLE
491 115126 : DO i = 1, nat_kind
492 69614 : iatom = a_list(i)
493 195568 : DO ispin = 1, nspins
494 31208442 : rho_atom_set(iatom)%cpc_h(ispin)%r_coef = 0.0_dp
495 31278056 : rho_atom_set(iatom)%cpc_s(ispin)%r_coef = 0.0_dp
496 : END DO ! ispin
497 : END DO ! i
498 :
499 45512 : num_pe = para_env%num_pe
500 45512 : mepos = para_env%mepos
501 45512 : bo = get_limit(nat_kind, num_pe, mepos)
502 154059 : DO i = bo(1), bo(2)
503 34807 : iatom = a_list(i)
504 124842 : DO ispin = 1, nspins
505 89617425 : rho_atom_set(iatom)%ga_Vlocal_gb_h(ispin)%r_coef = 0.0_dp
506 89652232 : rho_atom_set(iatom)%ga_Vlocal_gb_s(ispin)%r_coef = 0.0_dp
507 : END DO ! ispin
508 : END DO ! i
509 : END DO ! ikind
510 :
511 121592 : ALLOCATE (basis_set_list(nkind))
512 73740 : DO ikind = 1, nkind
513 49814 : CALL get_qs_kind(qs_kind_set(ikind), basis_set=basis_set_a)
514 73740 : IF (ASSOCIATED(basis_set_a)) THEN
515 49814 : basis_set_list(ikind)%gto_basis_set => basis_set_a
516 : ELSE
517 0 : NULLIFY (basis_set_list(ikind)%gto_basis_set)
518 : END IF
519 : END DO
520 :
521 23926 : len_PC1 = max_nsgf*max_gau
522 23926 : len_CPC = max_gau*max_gau
523 :
524 : num_pe = 1
525 23926 : !$ num_pe = omp_get_max_threads()
526 23926 : CALL neighbor_list_iterator_create(nl_iterator, sab, nthread=num_pe)
527 :
528 : !$OMP PARALLEL DEFAULT( NONE ) &
529 : !$OMP SHARED( max_nsgf, max_gau &
530 : !$OMP , len_PC1, len_CPC &
531 : !$OMP , nl_iterator, basis_set_list &
532 : !$OMP , nimages, cell_to_index &
533 : !$OMP , nspins, rho_ao &
534 : !$OMP , nkind, qs_kind_set &
535 : !$OMP , oce, eps_cpc &
536 : !$OMP , rho_atom_set &
537 : !$OMP , natom, locks, number_of_locks &
538 : !$OMP ) &
539 : !$OMP PRIVATE( p_block_spin, ispin &
540 : !$OMP , p_matrix, mepos &
541 : !$OMP , ikind, jkind, iatom, jatom &
542 : !$OMP , cell_b, rab, basis_1c &
543 : !$OMP , basis_set_a, basis_set_b &
544 : !$OMP , pmax, irow, icol, img &
545 : !$OMP , found &
546 : !$OMP , kkind &
547 : !$OMP , paw_atom, nsatbas &
548 : !$OMP , nsoctot, katom &
549 : !$OMP , iac , alist_ac, kac, n_cont_a, list_a &
550 : !$OMP , ibc , alist_bc, kbc, n_cont_b, list_b &
551 : !$OMP , C_coeff_hh_a, C_coeff_ss_a, dista &
552 : !$OMP , C_coeff_hh_b, C_coeff_ss_b, distb &
553 : !$OMP , distab &
554 : !$OMP , factor, r_coef_h, r_coef_s &
555 23926 : !$OMP )
556 :
557 : ALLOCATE (p_block_spin(nspins))
558 : ALLOCATE (p_matrix(max_nsgf, max_nsgf))
559 :
560 : !$OMP SINGLE
561 : !$ number_of_locks = nspins*natom
562 : !$ ALLOCATE (locks(number_of_locks))
563 : !$OMP END SINGLE
564 :
565 : !$OMP DO
566 : !$ DO lock = 1, number_of_locks
567 : !$ call omp_init_lock(locks(lock))
568 : !$ END DO
569 : !$OMP END DO
570 :
571 : mepos = 0
572 : !$ mepos = omp_get_thread_num()
573 : DO WHILE (neighbor_list_iterate(nl_iterator, mepos=mepos) == 0)
574 :
575 : CALL get_iterator_info(nl_iterator, mepos=mepos, &
576 : ikind=ikind, jkind=jkind, &
577 : iatom=iatom, jatom=jatom, &
578 : cell=cell_b, r=rab)
579 :
580 : basis_set_a => basis_set_list(ikind)%gto_basis_set
581 : IF (.NOT. ASSOCIATED(basis_set_a)) CYCLE
582 : basis_set_b => basis_set_list(jkind)%gto_basis_set
583 : IF (.NOT. ASSOCIATED(basis_set_b)) CYCLE
584 :
585 : pmax = 0._dp
586 : IF (iatom <= jatom) THEN
587 : irow = iatom
588 : icol = jatom
589 : ELSE
590 : irow = jatom
591 : icol = iatom
592 : END IF
593 :
594 : IF (nimages > 1) THEN
595 : img = cell_to_index(cell_b(1), cell_b(2), cell_b(3))
596 : CPASSERT(img > 0)
597 : ELSE
598 : img = 1
599 : END IF
600 :
601 : DO ispin = 1, nspins
602 : CALL dbcsr_get_block_p(matrix=rho_ao(nspins*(img - 1) + ispin)%matrix, &
603 : row=irow, col=icol, BLOCK=p_block_spin(ispin)%r_coef, &
604 : found=found)
605 : pmax = pmax + MAXVAL(ABS(p_block_spin(ispin)%r_coef))
606 : END DO
607 :
608 : DO kkind = 1, nkind
609 : CALL get_qs_kind(qs_kind_set(kkind), basis_set=basis_1c, basis_type="GAPW_1C", &
610 : paw_atom=paw_atom)
611 :
612 : IF (.NOT. paw_atom) CYCLE
613 :
614 : CALL get_paw_basis_info(basis_1c, nsatbas=nsatbas)
615 : nsoctot = nsatbas
616 :
617 : iac = ikind + nkind*(kkind - 1)
618 : ibc = jkind + nkind*(kkind - 1)
619 : IF (.NOT. ASSOCIATED(oce%intac(iac)%alist)) CYCLE
620 : IF (.NOT. ASSOCIATED(oce%intac(ibc)%alist)) CYCLE
621 :
622 : CALL get_alist(oce%intac(iac), alist_ac, iatom)
623 : CALL get_alist(oce%intac(ibc), alist_bc, jatom)
624 : IF (.NOT. ASSOCIATED(alist_ac)) CYCLE
625 : IF (.NOT. ASSOCIATED(alist_bc)) CYCLE
626 :
627 : DO kac = 1, alist_ac%nclist
628 : DO kbc = 1, alist_bc%nclist
629 : IF (alist_ac%clist(kac)%catom /= alist_bc%clist(kbc)%catom) CYCLE
630 : IF (ALL(cell_b + alist_bc%clist(kbc)%cell - alist_ac%clist(kac)%cell == 0)) THEN
631 : IF (pmax*alist_bc%clist(kbc)%maxac*alist_ac%clist(kac)%maxac < eps_cpc) CYCLE
632 :
633 : n_cont_a = alist_ac%clist(kac)%nsgf_cnt
634 : n_cont_b = alist_bc%clist(kbc)%nsgf_cnt
635 : IF (n_cont_a .EQ. 0 .OR. n_cont_b .EQ. 0) CYCLE
636 :
637 : list_a => alist_ac%clist(kac)%sgf_list
638 : list_b => alist_bc%clist(kbc)%sgf_list
639 :
640 : katom = alist_ac%clist(kac)%catom
641 :
642 : IF (iatom == katom .AND. ALL(alist_ac%clist(kac)%cell == 0)) THEN
643 : C_coeff_hh_a => alist_ac%clist(kac)%achint(:, :, 1)
644 : C_coeff_ss_a => alist_ac%clist(kac)%acint(:, :, 1)
645 : dista = .FALSE.
646 : ELSE
647 : C_coeff_hh_a => alist_ac%clist(kac)%acint(:, :, 1)
648 : C_coeff_ss_a => alist_ac%clist(kac)%acint(:, :, 1)
649 : dista = .TRUE.
650 : END IF
651 : IF (jatom == katom .AND. ALL(alist_bc%clist(kbc)%cell == 0)) THEN
652 : C_coeff_hh_b => alist_bc%clist(kbc)%achint(:, :, 1)
653 : C_coeff_ss_b => alist_bc%clist(kbc)%acint(:, :, 1)
654 : distb = .FALSE.
655 : ELSE
656 : C_coeff_hh_b => alist_bc%clist(kbc)%acint(:, :, 1)
657 : C_coeff_ss_b => alist_bc%clist(kbc)%acint(:, :, 1)
658 : distb = .TRUE.
659 : END IF
660 :
661 : distab = dista .AND. distb
662 :
663 : DO ispin = 1, nspins
664 :
665 : IF (iatom <= jatom) THEN
666 : CALL alist_pre_align_blk(p_block_spin(ispin)%r_coef, &
667 : SIZE(p_block_spin(ispin)%r_coef, 1), p_matrix, SIZE(p_matrix, 1), &
668 : list_a, n_cont_a, list_b, n_cont_b)
669 : ELSE
670 : CALL alist_pre_align_blk(p_block_spin(ispin)%r_coef, &
671 : SIZE(p_block_spin(ispin)%r_coef, 1), p_matrix, SIZE(p_matrix, 1), &
672 : list_b, n_cont_b, list_a, n_cont_a)
673 : END IF
674 :
675 : factor = 1.0_dp
676 : IF (iatom == jatom) factor = 0.5_dp
677 :
678 : r_coef_h => rho_atom_set(katom)%cpc_h(ispin)%r_coef
679 : r_coef_s => rho_atom_set(katom)%cpc_s(ispin)%r_coef
680 :
681 : !$ CALL omp_set_lock(locks((katom - 1)*nspins + ispin))
682 : IF (iatom <= jatom) THEN
683 : CALL proj_blk(C_coeff_hh_a, C_coeff_ss_a, n_cont_a, &
684 : C_coeff_hh_b, C_coeff_ss_b, n_cont_b, &
685 : p_matrix, max_nsgf, r_coef_h, r_coef_s, nsoctot, &
686 : len_PC1, len_CPC, factor, distab)
687 : ELSE
688 : CALL proj_blk(C_coeff_hh_b, C_coeff_ss_b, n_cont_b, &
689 : C_coeff_hh_a, C_coeff_ss_a, n_cont_a, &
690 : p_matrix, max_nsgf, r_coef_h, r_coef_s, nsoctot, &
691 : len_PC1, len_CPC, factor, distab)
692 : END IF
693 : !$ CALL omp_unset_lock(locks((katom - 1)*nspins + ispin))
694 :
695 : END DO
696 : EXIT !search loop over jatom-katom list
697 : END IF
698 : END DO
699 : END DO
700 : END DO
701 : END DO
702 : ! Wait for all threads to finish the loop before locks can be freed
703 : !$OMP BARRIER
704 :
705 : !$OMP DO
706 : !$ DO lock = 1, number_of_locks
707 : !$ call omp_destroy_lock(locks(lock))
708 : !$ END DO
709 : !$OMP END DO
710 : !$OMP SINGLE
711 : !$ DEALLOCATE (locks)
712 : !$OMP END SINGLE NOWAIT
713 :
714 : DEALLOCATE (p_block_spin, p_matrix)
715 : !$OMP END PARALLEL
716 :
717 23926 : CALL neighbor_list_iterator_release(nl_iterator)
718 :
719 23926 : CALL get_atomic_kind_set(atomic_kind_set, kind_of=kind_of)
720 :
721 103780 : DO iatom = 1, natom
722 79854 : ikind = kind_of(iatom)
723 :
724 195660 : DO ispin = 1, nspins
725 171734 : IF (ASSOCIATED(rho_atom_set(iatom)%cpc_h(ispin)%r_coef)) THEN
726 62336442 : CALL para_env%sum(rho_atom_set(iatom)%cpc_h(ispin)%r_coef)
727 62336442 : CALL para_env%sum(rho_atom_set(iatom)%cpc_s(ispin)%r_coef)
728 80442 : r_coef_h => rho_atom_set(iatom)%cpc_h(ispin)%r_coef
729 80442 : r_coef_s => rho_atom_set(iatom)%cpc_s(ispin)%r_coef
730 62416884 : r_coef_h(:, :) = r_coef_h(:, :) + TRANSPOSE(r_coef_h(:, :))
731 62416884 : r_coef_s(:, :) = r_coef_s(:, :) + TRANSPOSE(r_coef_s(:, :))
732 : END IF
733 : END DO
734 :
735 : END DO
736 :
737 23926 : DEALLOCATE (kind_of, basis_set_list)
738 :
739 23926 : CALL timestop(handle)
740 :
741 71778 : END SUBROUTINE calculate_rho_atom_coeff
742 :
743 : ! **************************************************************************************************
744 : !> \brief ...
745 : !> \param rho_atom_set the type to initialize
746 : !> \param atomic_kind_set list of atomic kinds
747 : !> \param qs_kind_set the kind set from which to take quantum numbers and basis info
748 : !> \param dft_control DFT control type
749 : !> \param para_env parallel environment
750 : !> \par History:
751 : !> - Generalised by providing the rho_atom_set and the qs_kind_set 12.2019 (A.Bussy)
752 : ! **************************************************************************************************
753 1004 : SUBROUTINE init_rho_atom(rho_atom_set, atomic_kind_set, qs_kind_set, dft_control, para_env)
754 :
755 : TYPE(rho_atom_type), DIMENSION(:), POINTER :: rho_atom_set
756 : TYPE(atomic_kind_type), DIMENSION(:), POINTER :: atomic_kind_set
757 : TYPE(qs_kind_type), DIMENSION(:), POINTER :: qs_kind_set
758 : TYPE(dft_control_type), POINTER :: dft_control
759 : TYPE(mp_para_env_type), POINTER :: para_env
760 :
761 : CHARACTER(len=*), PARAMETER :: routineN = 'init_rho_atom'
762 :
763 : INTEGER :: handle, ikind, il, iso, iso1, iso2, l1, l1l2, l2, la, lc1, lc2, lcleb, ll, llmax, &
764 : lmax_sphere, lp, m1, m2, max_s_harm, max_s_set, maxl, maxlgto, maxs, mm, mp, na, nat, &
765 : natom, nr, nspins, quadrature
766 1004 : INTEGER, DIMENSION(:), POINTER :: atom_list
767 : LOGICAL :: paw_atom
768 : REAL(dp), ALLOCATABLE, DIMENSION(:, :) :: rga
769 1004 : REAL(dp), DIMENSION(:, :, :), POINTER :: my_CG
770 : TYPE(gapw_control_type), POINTER :: gapw_control
771 : TYPE(grid_atom_type), POINTER :: grid_atom
772 : TYPE(gto_basis_set_type), POINTER :: basis_1c_set
773 : TYPE(harmonics_atom_type), POINTER :: harmonics
774 :
775 1004 : CALL timeset(routineN, handle)
776 :
777 1004 : NULLIFY (basis_1c_set)
778 1004 : NULLIFY (my_CG, grid_atom, harmonics, atom_list)
779 :
780 1004 : CPASSERT(ASSOCIATED(atomic_kind_set))
781 1004 : CPASSERT(ASSOCIATED(dft_control))
782 1004 : CPASSERT(ASSOCIATED(para_env))
783 1004 : CPASSERT(ASSOCIATED(qs_kind_set))
784 :
785 1004 : CALL get_atomic_kind_set(atomic_kind_set, natom=natom)
786 :
787 1004 : CALL get_qs_kind_set(qs_kind_set, maxlgto=maxlgto, basis_type="GAPW_1C")
788 :
789 1004 : nspins = dft_control%nspins
790 1004 : gapw_control => dft_control%qs_control%gapw_control
791 :
792 1004 : lmax_sphere = gapw_control%lmax_sphere
793 :
794 1004 : llmax = MIN(lmax_sphere, 2*maxlgto)
795 1004 : max_s_harm = nsoset(llmax)
796 1004 : max_s_set = nsoset(maxlgto)
797 :
798 1004 : lcleb = MAX(llmax, 2*maxlgto, 1)
799 :
800 : ! *** allocate calculate the CG coefficients up to the maxl ***
801 1004 : CALL clebsch_gordon_init(lcleb)
802 1004 : CALL reallocate(my_CG, 1, max_s_set, 1, max_s_set, 1, max_s_harm)
803 :
804 3012 : ALLOCATE (rga(lcleb, 2))
805 3658 : DO lc1 = 0, maxlgto
806 11152 : DO iso1 = nsoset(lc1 - 1) + 1, nsoset(lc1)
807 7494 : l1 = indso(1, iso1)
808 7494 : m1 = indso(2, iso1)
809 32302 : DO lc2 = 0, maxlgto
810 97838 : DO iso2 = nsoset(lc2 - 1) + 1, nsoset(lc2)
811 68190 : l2 = indso(1, iso2)
812 68190 : m2 = indso(2, iso2)
813 68190 : CALL clebsch_gordon(l1, m1, l2, m2, rga)
814 68190 : IF (l1 + l2 > llmax) THEN
815 : l1l2 = llmax
816 : ELSE
817 : l1l2 = l1 + l2
818 : END IF
819 68190 : mp = m1 + m2
820 68190 : mm = m1 - m2
821 68190 : IF (m1*m2 < 0 .OR. (m1*m2 == 0 .AND. (m1 < 0 .OR. m2 < 0))) THEN
822 30348 : mp = -ABS(mp)
823 30348 : mm = -ABS(mm)
824 : ELSE
825 37842 : mp = ABS(mp)
826 37842 : mm = ABS(mm)
827 : END IF
828 250032 : DO lp = MOD(l1 + l2, 2), l1l2, 2
829 159688 : il = lp/2 + 1
830 159688 : IF (ABS(mp) <= lp) THEN
831 112392 : IF (mp >= 0) THEN
832 73840 : iso = nsoset(lp - 1) + lp + 1 + mp
833 : ELSE
834 38552 : iso = nsoset(lp - 1) + lp + 1 - ABS(mp)
835 : END IF
836 112392 : my_CG(iso1, iso2, iso) = rga(il, 1)
837 : END IF
838 227878 : IF (mp /= mm .AND. ABS(mm) <= lp) THEN
839 55840 : IF (mm >= 0) THEN
840 37116 : iso = nsoset(lp - 1) + lp + 1 + mm
841 : ELSE
842 18724 : iso = nsoset(lp - 1) + lp + 1 - ABS(mm)
843 : END IF
844 55840 : my_CG(iso1, iso2, iso) = rga(il, 2)
845 : END IF
846 : END DO
847 : END DO ! iso2
848 : END DO ! lc2
849 : END DO ! iso1
850 : END DO ! lc1
851 1004 : DEALLOCATE (rga)
852 1004 : CALL clebsch_gordon_deallocate()
853 :
854 : ! *** initialize the Lebedev grids ***
855 1004 : CALL init_lebedev_grids()
856 1004 : quadrature = gapw_control%quadrature
857 :
858 2932 : DO ikind = 1, SIZE(atomic_kind_set)
859 1928 : CALL get_atomic_kind(atomic_kind_set(ikind), atom_list=atom_list, natom=nat)
860 : CALL get_qs_kind(qs_kind_set(ikind), &
861 : paw_atom=paw_atom, &
862 : grid_atom=grid_atom, &
863 : harmonics=harmonics, &
864 1928 : ngrid_rad=nr, ngrid_ang=na)
865 :
866 : ! *** determine the Lebedev grid for this kind ***
867 :
868 1928 : ll = get_number_of_lebedev_grid(n=na)
869 1928 : na = lebedev_grid(ll)%n
870 1928 : la = lebedev_grid(ll)%l
871 1928 : grid_atom%ng_sphere = na
872 1928 : grid_atom%nr = nr
873 :
874 1928 : IF (llmax > la) THEN
875 0 : WRITE (*, '(/,72("*"))')
876 : WRITE (*, '(T2,A,T66,I4)') &
877 0 : "WARNING: the lebedev grid is built for angular momentum l up to ", la, &
878 0 : " the max l of spherical harmonics is larger, l_max = ", llmax, &
879 0 : " good integration is guaranteed only for l <= ", la
880 0 : WRITE (*, '(72("*"),/)')
881 : END IF
882 :
883 : ! *** calculate the radial grid ***
884 1928 : CALL create_grid_atom(grid_atom, nr, na, llmax, ll, quadrature)
885 :
886 : ! *** calculate the spherical harmonics on the grid ***
887 :
888 1928 : CALL get_qs_kind(qs_kind_set(ikind), basis_set=basis_1c_set, basis_type="GAPW_1C")
889 1928 : CALL get_gto_basis_set(gto_basis_set=basis_1c_set, maxl=maxl)
890 1928 : maxs = nsoset(maxl)
891 : CALL create_harmonics_atom(harmonics, &
892 : my_CG, na, llmax, maxs, max_s_harm, ll, grid_atom%wa, &
893 1928 : grid_atom%azi, grid_atom%pol)
894 6788 : CALL get_maxl_CG(harmonics, basis_1c_set, llmax, max_s_harm)
895 :
896 : END DO
897 :
898 1004 : CALL deallocate_lebedev_grids()
899 1004 : DEALLOCATE (my_CG)
900 :
901 1004 : CALL allocate_rho_atom_internals(rho_atom_set, atomic_kind_set, qs_kind_set, dft_control, para_env)
902 :
903 1004 : CALL timestop(handle)
904 :
905 2008 : END SUBROUTINE init_rho_atom
906 :
907 : ! **************************************************************************************************
908 : !> \brief ...
909 : !> \param rho_atom_set ...
910 : !> \param atomic_kind_set list of atomic kinds
911 : !> \param qs_kind_set the kind set from which to take quantum numbers and basis info
912 : !> \param dft_control DFT control type
913 : !> \param para_env parallel environment
914 : ! **************************************************************************************************
915 2616 : SUBROUTINE allocate_rho_atom_internals(rho_atom_set, atomic_kind_set, qs_kind_set, dft_control, para_env)
916 :
917 : TYPE(rho_atom_type), DIMENSION(:), POINTER :: rho_atom_set
918 : TYPE(atomic_kind_type), DIMENSION(:), POINTER :: atomic_kind_set
919 : TYPE(qs_kind_type), DIMENSION(:), POINTER :: qs_kind_set
920 : TYPE(dft_control_type), POINTER :: dft_control
921 : TYPE(mp_para_env_type), POINTER :: para_env
922 :
923 : CHARACTER(len=*), PARAMETER :: routineN = 'allocate_rho_atom_internals'
924 :
925 : INTEGER :: bo(2), handle, iat, iatom, ikind, ispin, &
926 : max_iso_not0, maxso, mepos, nat, &
927 : natom, nsatbas, nset, nsotot, nspins, &
928 : num_pe
929 2616 : INTEGER, DIMENSION(:), POINTER :: atom_list
930 : LOGICAL :: paw_atom
931 : TYPE(gto_basis_set_type), POINTER :: basis_1c
932 : TYPE(harmonics_atom_type), POINTER :: harmonics
933 :
934 2616 : CALL timeset(routineN, handle)
935 :
936 2616 : CPASSERT(ASSOCIATED(atomic_kind_set))
937 2616 : CPASSERT(ASSOCIATED(dft_control))
938 2616 : CPASSERT(ASSOCIATED(para_env))
939 2616 : CPASSERT(ASSOCIATED(qs_kind_set))
940 :
941 2616 : CALL get_atomic_kind_set(atomic_kind_set, natom=natom)
942 :
943 2616 : nspins = dft_control%nspins
944 :
945 2616 : IF (ASSOCIATED(rho_atom_set)) THEN
946 0 : CALL deallocate_rho_atom_set(rho_atom_set)
947 : END IF
948 16558 : ALLOCATE (rho_atom_set(natom))
949 :
950 8078 : DO ikind = 1, SIZE(atomic_kind_set)
951 :
952 5462 : NULLIFY (atom_list, harmonics)
953 5462 : CALL get_atomic_kind(atomic_kind_set(ikind), atom_list=atom_list, natom=nat)
954 : CALL get_qs_kind(qs_kind_set(ikind), &
955 : paw_atom=paw_atom, &
956 5462 : harmonics=harmonics)
957 :
958 5462 : IF (paw_atom) THEN
959 4912 : CALL get_qs_kind(qs_kind_set(ikind), basis_set=basis_1c, basis_type="GAPW_1C")
960 4912 : CALL get_gto_basis_set(gto_basis_set=basis_1c, nset=nset, maxso=maxso)
961 4912 : nsotot = nset*maxso
962 4912 : CALL get_paw_basis_info(basis_1c, nsatbas=nsatbas)
963 : END IF
964 :
965 5462 : max_iso_not0 = harmonics%max_iso_not0
966 14172 : DO iat = 1, nat
967 8710 : iatom = atom_list(iat)
968 : ! *** allocate the radial density for each LM,for each atom ***
969 :
970 35986 : ALLOCATE (rho_atom_set(iatom)%rho_rad_h(nspins))
971 27276 : ALLOCATE (rho_atom_set(iatom)%rho_rad_s(nspins))
972 27276 : ALLOCATE (rho_atom_set(iatom)%vrho_rad_h(nspins))
973 27276 : ALLOCATE (rho_atom_set(iatom)%vrho_rad_s(nspins))
974 :
975 : ALLOCATE (rho_atom_set(iatom)%cpc_h(nspins), &
976 : rho_atom_set(iatom)%cpc_s(nspins), &
977 : rho_atom_set(iatom)%drho_rad_h(nspins), &
978 : rho_atom_set(iatom)%drho_rad_s(nspins), &
979 : rho_atom_set(iatom)%rho_rad_h_d(3, nspins), &
980 187952 : rho_atom_set(iatom)%rho_rad_s_d(3, nspins))
981 : ALLOCATE (rho_atom_set(iatom)%int_scr_h(nspins), &
982 45842 : rho_atom_set(iatom)%int_scr_s(nspins))
983 :
984 14172 : IF (paw_atom) THEN
985 16102 : DO ispin = 1, nspins
986 : ALLOCATE (rho_atom_set(iatom)%cpc_h(ispin)%r_coef(1:nsatbas, 1:nsatbas), &
987 51588 : rho_atom_set(iatom)%cpc_s(ispin)%r_coef(1:nsatbas, 1:nsatbas))
988 : ALLOCATE (rho_atom_set(iatom)%int_scr_h(ispin)%r_coef(1:nsatbas, 1:nsatbas), &
989 42990 : rho_atom_set(iatom)%int_scr_s(ispin)%r_coef(1:nsatbas, 1:nsatbas))
990 :
991 3358490 : rho_atom_set(iatom)%cpc_h(ispin)%r_coef = 0.0_dp
992 3365994 : rho_atom_set(iatom)%cpc_s(ispin)%r_coef = 0.0_dp
993 : END DO
994 : END IF
995 :
996 : END DO ! iat
997 :
998 5462 : num_pe = para_env%num_pe
999 5462 : mepos = para_env%mepos
1000 5462 : bo = get_limit(nat, num_pe, mepos)
1001 17895 : DO iat = bo(1), bo(2)
1002 4355 : iatom = atom_list(iat)
1003 : ALLOCATE (rho_atom_set(iatom)%ga_Vlocal_gb_h(nspins), &
1004 27276 : rho_atom_set(iatom)%ga_Vlocal_gb_s(nspins))
1005 9817 : IF (paw_atom) THEN
1006 8051 : DO ispin = 1, nspins
1007 : CALL reallocate(rho_atom_set(iatom)%ga_Vlocal_gb_h(ispin)%r_coef, &
1008 4299 : 1, nsotot, 1, nsotot)
1009 : CALL reallocate(rho_atom_set(iatom)%ga_Vlocal_gb_s(ispin)%r_coef, &
1010 4299 : 1, nsotot, 1, nsotot)
1011 :
1012 10801681 : rho_atom_set(iatom)%ga_Vlocal_gb_h(ispin)%r_coef = 0.0_dp
1013 10805433 : rho_atom_set(iatom)%ga_Vlocal_gb_s(ispin)%r_coef = 0.0_dp
1014 : END DO
1015 : END IF
1016 :
1017 : END DO ! iat
1018 :
1019 : END DO
1020 :
1021 2616 : CALL timestop(handle)
1022 :
1023 5232 : END SUBROUTINE allocate_rho_atom_internals
1024 :
1025 : ! **************************************************************************************************
1026 : !> \brief ...
1027 : !> \param rho_atom_set ...
1028 : !> \param iatom ...
1029 : !> \param ispin ...
1030 : !> \param nr ...
1031 : !> \param max_iso_not0 ...
1032 : ! **************************************************************************************************
1033 4114 : SUBROUTINE allocate_rho_atom_rad(rho_atom_set, iatom, ispin, nr, max_iso_not0)
1034 :
1035 : TYPE(rho_atom_type), DIMENSION(:), POINTER :: rho_atom_set
1036 : INTEGER, INTENT(IN) :: iatom, ispin, nr, max_iso_not0
1037 :
1038 : CHARACTER(len=*), PARAMETER :: routineN = 'allocate_rho_atom_rad'
1039 :
1040 : INTEGER :: handle, j
1041 :
1042 4114 : CALL timeset(routineN, handle)
1043 :
1044 : ALLOCATE (rho_atom_set(iatom)%rho_rad_h(ispin)%r_coef(1:nr, 1:max_iso_not0), &
1045 : rho_atom_set(iatom)%rho_rad_s(ispin)%r_coef(1:nr, 1:max_iso_not0), &
1046 : rho_atom_set(iatom)%vrho_rad_h(ispin)%r_coef(1:nr, 1:max_iso_not0), &
1047 53482 : rho_atom_set(iatom)%vrho_rad_s(ispin)%r_coef(1:nr, 1:max_iso_not0))
1048 :
1049 3121938 : rho_atom_set(iatom)%rho_rad_h(ispin)%r_coef = 0.0_dp
1050 3121938 : rho_atom_set(iatom)%rho_rad_s(ispin)%r_coef = 0.0_dp
1051 3121938 : rho_atom_set(iatom)%vrho_rad_h(ispin)%r_coef = 0.0_dp
1052 3121938 : rho_atom_set(iatom)%vrho_rad_s(ispin)%r_coef = 0.0_dp
1053 :
1054 : ALLOCATE (rho_atom_set(iatom)%drho_rad_h(ispin)%r_coef(nr, max_iso_not0), &
1055 28798 : rho_atom_set(iatom)%drho_rad_s(ispin)%r_coef(nr, max_iso_not0))
1056 3121938 : rho_atom_set(iatom)%drho_rad_h(ispin)%r_coef = 0.0_dp
1057 3121938 : rho_atom_set(iatom)%drho_rad_s(ispin)%r_coef = 0.0_dp
1058 :
1059 16456 : DO j = 1, 3
1060 : ALLOCATE (rho_atom_set(iatom)%rho_rad_h_d(j, ispin)%r_coef(nr, max_iso_not0), &
1061 86394 : rho_atom_set(iatom)%rho_rad_s_d(j, ispin)%r_coef(nr, max_iso_not0))
1062 9365814 : rho_atom_set(iatom)%rho_rad_h_d(j, ispin)%r_coef = 0.0_dp
1063 9369928 : rho_atom_set(iatom)%rho_rad_s_d(j, ispin)%r_coef = 0.0_dp
1064 : END DO
1065 :
1066 4114 : CALL timestop(handle)
1067 :
1068 4114 : END SUBROUTINE allocate_rho_atom_rad
1069 :
1070 : ! **************************************************************************************************
1071 : !> \brief ...
1072 : !> \param rho_atom_set ...
1073 : !> \param iatom ...
1074 : !> \param ispin ...
1075 : ! **************************************************************************************************
1076 34584 : SUBROUTINE set2zero_rho_atom_rad(rho_atom_set, iatom, ispin)
1077 :
1078 : TYPE(rho_atom_type), DIMENSION(:), POINTER :: rho_atom_set
1079 : INTEGER, INTENT(IN) :: iatom, ispin
1080 :
1081 : INTEGER :: j
1082 :
1083 25360678 : rho_atom_set(iatom)%rho_rad_h(ispin)%r_coef = 0.0_dp
1084 25360678 : rho_atom_set(iatom)%rho_rad_s(ispin)%r_coef = 0.0_dp
1085 :
1086 25360678 : rho_atom_set(iatom)%vrho_rad_h(ispin)%r_coef = 0.0_dp
1087 25360678 : rho_atom_set(iatom)%vrho_rad_s(ispin)%r_coef = 0.0_dp
1088 :
1089 25360678 : rho_atom_set(iatom)%drho_rad_h(ispin)%r_coef = 0.0_dp
1090 25360678 : rho_atom_set(iatom)%drho_rad_s(ispin)%r_coef = 0.0_dp
1091 :
1092 138336 : DO j = 1, 3
1093 76082034 : rho_atom_set(iatom)%rho_rad_h_d(j, ispin)%r_coef = 0.0_dp
1094 76116618 : rho_atom_set(iatom)%rho_rad_s_d(j, ispin)%r_coef = 0.0_dp
1095 : END DO
1096 :
1097 34584 : END SUBROUTINE set2zero_rho_atom_rad
1098 :
1099 : END MODULE qs_rho_atom_methods
|
