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 :
8 : ! **************************************************************************************************
9 : !> \brief Quickstep force driver routine
10 : !> \author MK (12.06.2002)
11 : ! **************************************************************************************************
12 : MODULE qs_force
13 : USE atomic_kind_types, ONLY: atomic_kind_type,&
14 : get_atomic_kind_set
15 : USE cp_control_types, ONLY: dft_control_type
16 : USE cp_dbcsr_operations, ONLY: dbcsr_allocate_matrix_set,&
17 : dbcsr_deallocate_matrix_set
18 : USE cp_dbcsr_output, ONLY: cp_dbcsr_write_sparse_matrix
19 : USE cp_log_handling, ONLY: cp_get_default_logger,&
20 : cp_logger_get_default_io_unit,&
21 : cp_logger_type
22 : USE cp_output_handling, ONLY: cp_p_file,&
23 : cp_print_key_finished_output,&
24 : cp_print_key_should_output,&
25 : cp_print_key_unit_nr
26 : USE dbcsr_api, ONLY: dbcsr_copy,&
27 : dbcsr_p_type,&
28 : dbcsr_set
29 : USE dft_plus_u, ONLY: plus_u
30 : USE ec_env_types, ONLY: energy_correction_type
31 : USE efield_utils, ONLY: calculate_ecore_efield,&
32 : efield_potential_lengh_gauge
33 : USE energy_corrections, ONLY: energy_correction
34 : USE excited_states, ONLY: excited_state_energy
35 : USE hfx_exx, ONLY: calculate_exx
36 : USE input_constants, ONLY: ri_mp2_laplace,&
37 : ri_mp2_method_gpw,&
38 : ri_rpa_method_gpw
39 : USE input_section_types, ONLY: section_vals_get,&
40 : section_vals_get_subs_vals,&
41 : section_vals_type,&
42 : section_vals_val_get
43 : USE kinds, ONLY: dp
44 : USE lri_environment_types, ONLY: lri_environment_type
45 : USE message_passing, ONLY: mp_para_env_type
46 : USE mp2_cphf, ONLY: update_mp2_forces
47 : USE mulliken, ONLY: mulliken_restraint
48 : USE particle_types, ONLY: particle_type
49 : USE qs_core_energies, ONLY: calculate_ecore_overlap,&
50 : calculate_ecore_self
51 : USE qs_core_hamiltonian, ONLY: build_core_hamiltonian_matrix
52 : USE qs_dftb_dispersion, ONLY: calculate_dftb_dispersion
53 : USE qs_dftb_matrices, ONLY: build_dftb_matrices
54 : USE qs_energy, ONLY: qs_energies
55 : USE qs_energy_types, ONLY: qs_energy_type
56 : USE qs_environment_methods, ONLY: qs_env_rebuild_pw_env
57 : USE qs_environment_types, ONLY: get_qs_env,&
58 : qs_environment_type
59 : USE qs_external_potential, ONLY: external_c_potential,&
60 : external_e_potential
61 : USE qs_force_types, ONLY: allocate_qs_force,&
62 : qs_force_type,&
63 : replicate_qs_force,&
64 : zero_qs_force
65 : USE qs_ks_methods, ONLY: qs_ks_update_qs_env
66 : USE qs_ks_types, ONLY: qs_ks_env_type,&
67 : set_ks_env
68 : USE qs_rho_types, ONLY: qs_rho_get,&
69 : qs_rho_type
70 : USE qs_scf_post_scf, ONLY: qs_scf_compute_properties
71 : USE qs_subsys_types, ONLY: qs_subsys_set,&
72 : qs_subsys_type
73 : USE ri_environment_methods, ONLY: build_ri_matrices
74 : USE rt_propagation_forces, ONLY: calc_c_mat_force,&
75 : rt_admm_force
76 : USE rt_propagation_velocity_gauge, ONLY: velocity_gauge_ks_matrix,&
77 : velocity_gauge_nl_force
78 : USE se_core_core, ONLY: se_core_core_interaction
79 : USE se_core_matrix, ONLY: build_se_core_matrix
80 : USE virial_types, ONLY: symmetrize_virial,&
81 : virial_type
82 : USE xtb_matrices, ONLY: build_xtb_matrices
83 : #include "./base/base_uses.f90"
84 :
85 : IMPLICIT NONE
86 :
87 : PRIVATE
88 :
89 : ! *** Global parameters ***
90 :
91 : CHARACTER(len=*), PARAMETER, PRIVATE :: moduleN = 'qs_force'
92 :
93 : ! *** Public subroutines ***
94 :
95 : PUBLIC :: qs_calc_energy_force
96 :
97 : CONTAINS
98 :
99 : ! **************************************************************************************************
100 : !> \brief ...
101 : !> \param qs_env ...
102 : !> \param calc_force ...
103 : !> \param consistent_energies ...
104 : !> \param linres ...
105 : ! **************************************************************************************************
106 19923 : SUBROUTINE qs_calc_energy_force(qs_env, calc_force, consistent_energies, linres)
107 : TYPE(qs_environment_type), POINTER :: qs_env
108 : LOGICAL :: calc_force, consistent_energies, linres
109 :
110 19923 : qs_env%linres_run = linres
111 19923 : IF (calc_force) THEN
112 10227 : CALL qs_forces(qs_env)
113 : ELSE
114 : CALL qs_energies(qs_env, calc_forces=.FALSE., &
115 9696 : consistent_energies=consistent_energies)
116 : END IF
117 :
118 19923 : END SUBROUTINE qs_calc_energy_force
119 :
120 : ! **************************************************************************************************
121 : !> \brief Calculate the Quickstep forces.
122 : !> \param qs_env ...
123 : !> \date 29.10.2002
124 : !> \author MK
125 : !> \version 1.0
126 : ! **************************************************************************************************
127 10227 : SUBROUTINE qs_forces(qs_env)
128 :
129 : TYPE(qs_environment_type), POINTER :: qs_env
130 :
131 : CHARACTER(len=*), PARAMETER :: routineN = 'qs_forces'
132 :
133 : INTEGER :: after, handle, i, iatom, ic, ikind, &
134 : ispin, iw, natom, nkind, nspin, &
135 : output_unit
136 10227 : INTEGER, ALLOCATABLE, DIMENSION(:) :: atom_of_kind, kind_of, natom_of_kind
137 : LOGICAL :: do_admm, do_exx, do_gw, do_im_time, &
138 : has_unit_metric, omit_headers, &
139 : perform_ec, reuse_hfx
140 : REAL(dp) :: dummy_real, dummy_real2(2)
141 10227 : TYPE(atomic_kind_type), DIMENSION(:), POINTER :: atomic_kind_set
142 : TYPE(cp_logger_type), POINTER :: logger
143 10227 : TYPE(dbcsr_p_type), DIMENSION(:), POINTER :: matrix_s, matrix_w, rho_ao
144 10227 : TYPE(dbcsr_p_type), DIMENSION(:, :), POINTER :: matrix_w_kp
145 : TYPE(dft_control_type), POINTER :: dft_control
146 : TYPE(energy_correction_type), POINTER :: ec_env
147 : TYPE(lri_environment_type), POINTER :: lri_env
148 : TYPE(mp_para_env_type), POINTER :: para_env
149 10227 : TYPE(particle_type), DIMENSION(:), POINTER :: particle_set
150 : TYPE(qs_energy_type), POINTER :: energy
151 10227 : TYPE(qs_force_type), DIMENSION(:), POINTER :: force
152 : TYPE(qs_ks_env_type), POINTER :: ks_env
153 : TYPE(qs_rho_type), POINTER :: rho
154 : TYPE(qs_subsys_type), POINTER :: subsys
155 : TYPE(section_vals_type), POINTER :: hfx_sections, print_section
156 : TYPE(virial_type), POINTER :: virial
157 :
158 10227 : CALL timeset(routineN, handle)
159 10227 : NULLIFY (logger)
160 10227 : logger => cp_get_default_logger()
161 :
162 : ! rebuild plane wave environment
163 10227 : CALL qs_env_rebuild_pw_env(qs_env)
164 :
165 : ! zero out the forces in particle set
166 10227 : CALL get_qs_env(qs_env, particle_set=particle_set)
167 10227 : natom = SIZE(particle_set)
168 75814 : DO iatom = 1, natom
169 272575 : particle_set(iatom)%f = 0.0_dp
170 : END DO
171 :
172 : ! get atom mapping
173 10227 : NULLIFY (atomic_kind_set)
174 10227 : CALL get_qs_env(qs_env, atomic_kind_set=atomic_kind_set)
175 : CALL get_atomic_kind_set(atomic_kind_set=atomic_kind_set, &
176 : atom_of_kind=atom_of_kind, &
177 10227 : kind_of=kind_of)
178 :
179 10227 : NULLIFY (force, subsys, dft_control)
180 : CALL get_qs_env(qs_env, &
181 : force=force, &
182 : subsys=subsys, &
183 10227 : dft_control=dft_control)
184 10227 : IF (.NOT. ASSOCIATED(force)) THEN
185 : ! *** Allocate the force data structure ***
186 2765 : nkind = SIZE(atomic_kind_set)
187 2765 : CALL get_atomic_kind_set(atomic_kind_set=atomic_kind_set, natom_of_kind=natom_of_kind)
188 2765 : CALL allocate_qs_force(force, natom_of_kind)
189 2765 : DEALLOCATE (natom_of_kind)
190 2765 : CALL qs_subsys_set(subsys, force=force)
191 : END IF
192 10227 : CALL zero_qs_force(force)
193 :
194 : ! Check if CDFT potential is needed and save it until forces have been calculated
195 10227 : IF (dft_control%qs_control%cdft) THEN
196 118 : dft_control%qs_control%cdft_control%save_pot = .TRUE.
197 : END IF
198 :
199 : ! recalculate energy with forces
200 10227 : CALL qs_energies(qs_env, calc_forces=.TRUE.)
201 :
202 10227 : NULLIFY (para_env)
203 : CALL get_qs_env(qs_env, &
204 10227 : para_env=para_env)
205 :
206 : ! Now we handle some special cases
207 : ! Maybe some of these would be better dealt with in qs_energies?
208 10227 : IF (qs_env%run_rtp) THEN
209 1222 : NULLIFY (matrix_w, matrix_s, ks_env)
210 : CALL get_qs_env(qs_env, &
211 : ks_env=ks_env, &
212 : matrix_w=matrix_w, &
213 1222 : matrix_s=matrix_s)
214 1222 : CALL dbcsr_allocate_matrix_set(matrix_w, dft_control%nspins)
215 2686 : DO ispin = 1, dft_control%nspins
216 1464 : ALLOCATE (matrix_w(ispin)%matrix)
217 : CALL dbcsr_copy(matrix_w(ispin)%matrix, matrix_s(1)%matrix, &
218 1464 : name="W MATRIX")
219 2686 : CALL dbcsr_set(matrix_w(ispin)%matrix, 0.0_dp)
220 : END DO
221 1222 : CALL set_ks_env(ks_env, matrix_w=matrix_w)
222 :
223 1222 : CALL calc_c_mat_force(qs_env)
224 1222 : IF (dft_control%do_admm) CALL rt_admm_force(qs_env)
225 1222 : IF (dft_control%rtp_control%velocity_gauge .AND. dft_control%rtp_control%nl_gauge_transform) &
226 22 : CALL velocity_gauge_nl_force(qs_env, particle_set)
227 : END IF
228 : ! from an eventual Mulliken restraint
229 10227 : IF (dft_control%qs_control%mulliken_restraint) THEN
230 6 : NULLIFY (matrix_w, matrix_s, rho)
231 : CALL get_qs_env(qs_env, &
232 : matrix_w=matrix_w, &
233 : matrix_s=matrix_s, &
234 6 : rho=rho)
235 6 : NULLIFY (rho_ao)
236 6 : CALL qs_rho_get(rho, rho_ao=rho_ao)
237 : CALL mulliken_restraint(dft_control%qs_control%mulliken_restraint_control, &
238 6 : para_env, matrix_s(1)%matrix, rho_ao, w_matrix=matrix_w)
239 : END IF
240 : ! Add non-Pulay contribution of DFT+U to W matrix, since it has also to be
241 : ! digested with overlap matrix derivatives
242 10227 : IF (dft_control%dft_plus_u) THEN
243 72 : NULLIFY (matrix_w)
244 72 : CALL get_qs_env(qs_env, matrix_w=matrix_w)
245 72 : CALL plus_u(qs_env=qs_env, matrix_w=matrix_w)
246 : END IF
247 :
248 : ! Write W Matrix to output (if requested)
249 10227 : CALL get_qs_env(qs_env, has_unit_metric=has_unit_metric)
250 10227 : IF (.NOT. has_unit_metric) THEN
251 7225 : NULLIFY (matrix_w_kp)
252 7225 : CALL get_qs_env(qs_env, matrix_w_kp=matrix_w_kp)
253 7225 : nspin = SIZE(matrix_w_kp, 1)
254 15442 : DO ispin = 1, nspin
255 8217 : IF (BTEST(cp_print_key_should_output(logger%iter_info, &
256 7225 : qs_env%input, "DFT%PRINT%AO_MATRICES/W_MATRIX"), cp_p_file)) THEN
257 : iw = cp_print_key_unit_nr(logger, qs_env%input, "DFT%PRINT%AO_MATRICES/W_MATRIX", &
258 8 : extension=".Log")
259 8 : CALL section_vals_val_get(qs_env%input, "DFT%PRINT%AO_MATRICES%NDIGITS", i_val=after)
260 8 : CALL section_vals_val_get(qs_env%input, "DFT%PRINT%AO_MATRICES%OMIT_HEADERS", l_val=omit_headers)
261 8 : after = MIN(MAX(after, 1), 16)
262 16 : DO ic = 1, SIZE(matrix_w_kp, 2)
263 : CALL cp_dbcsr_write_sparse_matrix(matrix_w_kp(ispin, ic)%matrix, 4, after, qs_env, &
264 16 : para_env, output_unit=iw, omit_headers=omit_headers)
265 : END DO
266 : CALL cp_print_key_finished_output(iw, logger, qs_env%input, &
267 8 : "DFT%PRINT%AO_MATRICES/W_MATRIX")
268 : END IF
269 : END DO
270 : END IF
271 :
272 : ! Check if energy correction should be skipped
273 10227 : perform_ec = .FALSE.
274 10227 : IF (qs_env%energy_correction) THEN
275 432 : CALL get_qs_env(qs_env, ec_env=ec_env)
276 432 : IF (.NOT. ec_env%do_skip) perform_ec = .TRUE.
277 : END IF
278 :
279 : ! Compute core forces (also overwrites matrix_w)
280 10227 : IF (dft_control%qs_control%semi_empirical) THEN
281 : CALL build_se_core_matrix(qs_env=qs_env, para_env=para_env, &
282 3002 : calculate_forces=.TRUE.)
283 3002 : CALL se_core_core_interaction(qs_env, para_env, calculate_forces=.TRUE.)
284 7225 : ELSEIF (dft_control%qs_control%dftb) THEN
285 : CALL build_dftb_matrices(qs_env=qs_env, para_env=para_env, &
286 724 : calculate_forces=.TRUE.)
287 : CALL calculate_dftb_dispersion(qs_env=qs_env, para_env=para_env, &
288 724 : calculate_forces=.TRUE.)
289 6501 : ELSEIF (dft_control%qs_control%xtb) THEN
290 : CALL build_xtb_matrices(qs_env=qs_env, para_env=para_env, &
291 428 : calculate_forces=.TRUE.)
292 6073 : ELSEIF (perform_ec) THEN
293 : ! Calculates core and grid based forces
294 432 : CALL energy_correction(qs_env, ec_init=.FALSE., calculate_forces=.TRUE.)
295 : ELSE
296 : ! Dispersion energy and forces are calculated in qs_energy?
297 5641 : CALL build_core_hamiltonian_matrix(qs_env=qs_env, calculate_forces=.TRUE.)
298 : ! The above line reset the core H, which should be re-updated in case a TD field is applied:
299 5641 : IF (qs_env%run_rtp) THEN
300 810 : IF (dft_control%apply_efield_field) &
301 160 : CALL efield_potential_lengh_gauge(qs_env)
302 810 : IF (dft_control%rtp_control%velocity_gauge) &
303 22 : CALL velocity_gauge_ks_matrix(qs_env, subtract_nl_term=.FALSE.)
304 :
305 : END IF
306 5641 : CALL calculate_ecore_self(qs_env)
307 5641 : CALL calculate_ecore_overlap(qs_env, para_env, calculate_forces=.TRUE.)
308 5641 : CALL calculate_ecore_efield(qs_env, calculate_forces=.TRUE.)
309 : !swap external_e_potential before external_c_potential, to ensure
310 : !that external potential on grid is loaded before calculating energy of cores
311 5641 : CALL external_e_potential(qs_env)
312 5641 : IF (.NOT. dft_control%qs_control%gapw) THEN
313 5217 : CALL external_c_potential(qs_env, calculate_forces=.TRUE.)
314 : END IF
315 : ! RIGPW matrices
316 5641 : IF (dft_control%qs_control%rigpw) THEN
317 0 : CALL get_qs_env(qs_env=qs_env, lri_env=lri_env)
318 0 : CALL build_ri_matrices(lri_env, qs_env, calculate_forces=.TRUE.)
319 : END IF
320 : END IF
321 :
322 : ! MP2 Code
323 10227 : IF (ASSOCIATED(qs_env%mp2_env)) THEN
324 310 : NULLIFY (energy)
325 310 : CALL get_qs_env(qs_env, energy=energy)
326 310 : CALL qs_scf_compute_properties(qs_env, wf_type='MP2 ', do_mp2=.TRUE.)
327 310 : CALL qs_ks_update_qs_env(qs_env, just_energy=.TRUE.)
328 310 : energy%total = energy%total + energy%mp2
329 :
330 : IF ((qs_env%mp2_env%method == ri_mp2_method_gpw .OR. qs_env%mp2_env%method == ri_mp2_laplace .OR. &
331 : qs_env%mp2_env%method == ri_rpa_method_gpw) &
332 310 : .AND. .NOT. qs_env%mp2_env%do_im_time) THEN
333 260 : CALL update_mp2_forces(qs_env)
334 : END IF
335 :
336 : !RPA EXX energy and forces
337 310 : IF (qs_env%mp2_env%method == ri_rpa_method_gpw) THEN
338 : do_exx = .FALSE.
339 48 : hfx_sections => section_vals_get_subs_vals(qs_env%input, "DFT%XC%WF_CORRELATION%RI_RPA%HF")
340 48 : CALL section_vals_get(hfx_sections, explicit=do_exx)
341 48 : IF (do_exx) THEN
342 26 : do_gw = qs_env%mp2_env%ri_rpa%do_ri_g0w0
343 26 : do_admm = qs_env%mp2_env%ri_rpa%do_admm
344 26 : reuse_hfx = qs_env%mp2_env%ri_rpa%reuse_hfx
345 26 : do_im_time = qs_env%mp2_env%do_im_time
346 26 : output_unit = cp_logger_get_default_io_unit()
347 26 : dummy_real = 0.0_dp
348 :
349 : CALL calculate_exx(qs_env=qs_env, &
350 : unit_nr=output_unit, &
351 : hfx_sections=hfx_sections, &
352 : x_data=qs_env%mp2_env%ri_rpa%x_data, &
353 : do_gw=do_gw, &
354 : do_admm=do_admm, &
355 : calc_forces=.TRUE., &
356 : reuse_hfx=reuse_hfx, &
357 : do_im_time=do_im_time, &
358 : E_ex_from_GW=dummy_real, &
359 : E_admm_from_GW=dummy_real2, &
360 26 : t3=dummy_real)
361 : END IF
362 : END IF
363 9917 : ELSEIF (.NOT. perform_ec) THEN
364 : ! Compute grid-based forces
365 9485 : CALL qs_ks_update_qs_env(qs_env, calculate_forces=.TRUE.)
366 : END IF
367 :
368 : ! Excited state forces
369 10227 : CALL excited_state_energy(qs_env, calculate_forces=.TRUE.)
370 :
371 : ! replicate forces (get current pointer)
372 10227 : NULLIFY (force)
373 10227 : CALL get_qs_env(qs_env=qs_env, force=force)
374 10227 : CALL replicate_qs_force(force, para_env)
375 :
376 75814 : DO iatom = 1, natom
377 65587 : ikind = kind_of(iatom)
378 65587 : i = atom_of_kind(iatom)
379 : ! XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
380 : ! the force is - dE/dR, what is called force is actually the gradient
381 : ! Things should have the right name
382 : ! The minus sign below is a hack
383 : ! XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
384 524696 : force(ikind)%other(1:3, i) = -particle_set(iatom)%f(1:3) + force(ikind)%ch_pulay(1:3, i)
385 262348 : force(ikind)%total(1:3, i) = force(ikind)%total(1:3, i) + force(ikind)%other(1:3, i)
386 534923 : particle_set(iatom)%f = -force(ikind)%total(1:3, i)
387 : END DO
388 :
389 10227 : NULLIFY (virial, energy)
390 10227 : CALL get_qs_env(qs_env=qs_env, virial=virial, energy=energy)
391 10227 : IF (virial%pv_availability) THEN
392 784 : CALL para_env%sum(virial%pv_overlap)
393 784 : CALL para_env%sum(virial%pv_ekinetic)
394 784 : CALL para_env%sum(virial%pv_ppl)
395 784 : CALL para_env%sum(virial%pv_ppnl)
396 784 : CALL para_env%sum(virial%pv_ecore_overlap)
397 784 : CALL para_env%sum(virial%pv_ehartree)
398 784 : CALL para_env%sum(virial%pv_exc)
399 784 : CALL para_env%sum(virial%pv_exx)
400 784 : CALL para_env%sum(virial%pv_vdw)
401 784 : CALL para_env%sum(virial%pv_mp2)
402 784 : CALL para_env%sum(virial%pv_nlcc)
403 784 : CALL para_env%sum(virial%pv_gapw)
404 784 : CALL para_env%sum(virial%pv_lrigpw)
405 784 : CALL para_env%sum(virial%pv_virial)
406 784 : CALL symmetrize_virial(virial)
407 : ! Add the volume terms of the virial
408 784 : IF ((.NOT. virial%pv_numer) .AND. &
409 : (.NOT. (dft_control%qs_control%dftb .OR. &
410 : dft_control%qs_control%xtb .OR. &
411 : dft_control%qs_control%semi_empirical))) THEN
412 :
413 : ! Harris energy correction requires volume terms from
414 : ! 1) Harris functional contribution, and
415 : ! 2) Linear Response solver
416 592 : IF (perform_ec) THEN
417 168 : CALL get_qs_env(qs_env, ec_env=ec_env)
418 168 : energy%hartree = ec_env%ehartree
419 168 : energy%exc = ec_env%exc
420 168 : IF (dft_control%do_admm) THEN
421 38 : energy%exc_aux_fit = ec_env%exc_aux_fit
422 : END IF
423 : END IF
424 2368 : DO i = 1, 3
425 : virial%pv_ehartree(i, i) = virial%pv_ehartree(i, i) &
426 1776 : - 2.0_dp*(energy%hartree + energy%sccs_pol)
427 : virial%pv_virial(i, i) = virial%pv_virial(i, i) - energy%exc &
428 1776 : - 2.0_dp*(energy%hartree + energy%sccs_pol)
429 1776 : virial%pv_exc(i, i) = virial%pv_exc(i, i) - energy%exc
430 2368 : IF (dft_control%do_admm) THEN
431 198 : virial%pv_exc(i, i) = virial%pv_exc(i, i) - energy%exc_aux_fit
432 198 : virial%pv_virial(i, i) = virial%pv_virial(i, i) - energy%exc_aux_fit
433 : END IF
434 : ! The factor 2 is a hack. It compensates the plus sign in h_stress/pw_poisson_solve.
435 : ! The sign in pw_poisson_solve is correct for FIST, but not for QS.
436 : ! There should be a more elegant solution to that ...
437 : END DO
438 : END IF
439 : END IF
440 :
441 : output_unit = cp_print_key_unit_nr(logger, qs_env%input, "DFT%PRINT%DERIVATIVES", &
442 10227 : extension=".Log")
443 10227 : print_section => section_vals_get_subs_vals(qs_env%input, "DFT%PRINT%DERIVATIVES")
444 10227 : IF (dft_control%qs_control%semi_empirical) THEN
445 : CALL write_forces(force, atomic_kind_set, 2, output_unit=output_unit, &
446 3002 : print_section=print_section)
447 7225 : ELSE IF (dft_control%qs_control%dftb) THEN
448 : CALL write_forces(force, atomic_kind_set, 4, output_unit=output_unit, &
449 724 : print_section=print_section)
450 6501 : ELSE IF (dft_control%qs_control%xtb) THEN
451 : CALL write_forces(force, atomic_kind_set, 4, output_unit=output_unit, &
452 428 : print_section=print_section)
453 6073 : ELSE IF (dft_control%qs_control%gapw .OR. dft_control%qs_control%gapw_xc) THEN
454 : CALL write_forces(force, atomic_kind_set, 1, output_unit=output_unit, &
455 496 : print_section=print_section)
456 : ELSE
457 : CALL write_forces(force, atomic_kind_set, 0, output_unit=output_unit, &
458 5577 : print_section=print_section)
459 : END IF
460 : CALL cp_print_key_finished_output(output_unit, logger, qs_env%input, &
461 10227 : "DFT%PRINT%DERIVATIVES")
462 :
463 : ! deallocate W Matrix:
464 10227 : NULLIFY (ks_env, matrix_w_kp)
465 : CALL get_qs_env(qs_env=qs_env, &
466 : matrix_w_kp=matrix_w_kp, &
467 10227 : ks_env=ks_env)
468 10227 : CALL dbcsr_deallocate_matrix_set(matrix_w_kp)
469 10227 : NULLIFY (matrix_w_kp)
470 10227 : CALL set_ks_env(ks_env, matrix_w_kp=matrix_w_kp)
471 :
472 10227 : DEALLOCATE (atom_of_kind, kind_of)
473 :
474 10227 : CALL timestop(handle)
475 :
476 20454 : END SUBROUTINE qs_forces
477 :
478 : ! **************************************************************************************************
479 : !> \brief Write a Quickstep force data structure to output unit
480 : !> \param qs_force ...
481 : !> \param atomic_kind_set ...
482 : !> \param ftype ...
483 : !> \param output_unit ...
484 : !> \param print_section ...
485 : !> \date 05.06.2002
486 : !> \author MK
487 : !> \version 1.0
488 : ! **************************************************************************************************
489 10227 : SUBROUTINE write_forces(qs_force, atomic_kind_set, ftype, output_unit, &
490 : print_section)
491 :
492 : TYPE(qs_force_type), DIMENSION(:), POINTER :: qs_force
493 : TYPE(atomic_kind_type), DIMENSION(:), POINTER :: atomic_kind_set
494 : INTEGER, INTENT(IN) :: ftype, output_unit
495 : TYPE(section_vals_type), POINTER :: print_section
496 :
497 : CHARACTER(LEN=13) :: fmtstr5
498 : CHARACTER(LEN=15) :: fmtstr4
499 : CHARACTER(LEN=20) :: fmtstr3
500 : CHARACTER(LEN=35) :: fmtstr2
501 : CHARACTER(LEN=48) :: fmtstr1
502 : INTEGER :: i, iatom, ikind, my_ftype, natom, ndigits
503 10227 : INTEGER, ALLOCATABLE, DIMENSION(:) :: atom_of_kind, kind_of
504 : REAL(KIND=dp), DIMENSION(3) :: grand_total
505 :
506 10227 : IF (output_unit > 0) THEN
507 :
508 152 : IF (.NOT. ASSOCIATED(qs_force)) THEN
509 : CALL cp_abort(__LOCATION__, &
510 : "The qs_force pointer is not associated "// &
511 0 : "and cannot be printed")
512 : END IF
513 :
514 : CALL get_atomic_kind_set(atomic_kind_set=atomic_kind_set, atom_of_kind=atom_of_kind, &
515 152 : kind_of=kind_of, natom=natom)
516 :
517 : ! Variable precision output of the forces
518 : CALL section_vals_val_get(print_section, "NDIGITS", &
519 152 : i_val=ndigits)
520 :
521 152 : fmtstr1 = "(/,/,T2,A,/,/,T3,A,T11,A,T23,A,T40,A1,2( X,A1))"
522 152 : WRITE (UNIT=fmtstr1(41:42), FMT="(I2)") ndigits + 5
523 :
524 152 : fmtstr2 = "(/,(T2,I5,4X,I4,T18,A,T34,3F . ))"
525 152 : WRITE (UNIT=fmtstr2(32:33), FMT="(I2)") ndigits
526 152 : WRITE (UNIT=fmtstr2(29:30), FMT="(I2)") ndigits + 6
527 :
528 152 : fmtstr3 = "(/,T3,A,T34,3F . )"
529 152 : WRITE (UNIT=fmtstr3(18:19), FMT="(I2)") ndigits
530 152 : WRITE (UNIT=fmtstr3(15:16), FMT="(I2)") ndigits + 6
531 :
532 152 : fmtstr4 = "((T34,3F . ))"
533 152 : WRITE (UNIT=fmtstr4(12:13), FMT="(I2)") ndigits
534 152 : WRITE (UNIT=fmtstr4(9:10), FMT="(I2)") ndigits + 6
535 :
536 : fmtstr5 = "(/T2,A//T3,A)"
537 :
538 : WRITE (UNIT=output_unit, FMT=fmtstr1) &
539 152 : "FORCES [a.u.]", "Atom", "Kind", "Component", "X", "Y", "Z"
540 :
541 152 : grand_total(:) = 0.0_dp
542 :
543 152 : my_ftype = ftype
544 :
545 0 : SELECT CASE (my_ftype)
546 : CASE DEFAULT
547 0 : DO iatom = 1, natom
548 0 : ikind = kind_of(iatom)
549 0 : i = atom_of_kind(iatom)
550 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
551 0 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
552 0 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
553 : END DO
554 : CASE (0)
555 469 : DO iatom = 1, natom
556 337 : ikind = kind_of(iatom)
557 337 : i = atom_of_kind(iatom)
558 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
559 1348 : iatom, ikind, " overlap", qs_force(ikind)%overlap(1:3, i), &
560 1348 : iatom, ikind, " overlap_admm", qs_force(ikind)%overlap_admm(1:3, i), &
561 1348 : iatom, ikind, " kinetic", qs_force(ikind)%kinetic(1:3, i), &
562 1348 : iatom, ikind, " gth_ppl", qs_force(ikind)%gth_ppl(1:3, i), &
563 1348 : iatom, ikind, " gth_nlcc", qs_force(ikind)%gth_nlcc(1:3, i), &
564 1348 : iatom, ikind, " gth_ppnl", qs_force(ikind)%gth_ppnl(1:3, i), &
565 1348 : iatom, ikind, " core_overlap", qs_force(ikind)%core_overlap(1:3, i), &
566 1348 : iatom, ikind, " rho_core", qs_force(ikind)%rho_core(1:3, i), &
567 1348 : iatom, ikind, " rho_elec", qs_force(ikind)%rho_elec(1:3, i), &
568 1348 : iatom, ikind, " rho_lri_elec", qs_force(ikind)%rho_lri_elec(1:3, i), &
569 1348 : iatom, ikind, " ch_pulay", qs_force(ikind)%ch_pulay(1:3, i), &
570 1348 : iatom, ikind, " dispersion", qs_force(ikind)%dispersion(1:3, i), &
571 1348 : iatom, ikind, " gCP", qs_force(ikind)%gcp(1:3, i), &
572 1348 : iatom, ikind, " other", qs_force(ikind)%other(1:3, i), &
573 1348 : iatom, ikind, " fock_4c", qs_force(ikind)%fock_4c(1:3, i), &
574 1348 : iatom, ikind, " ehrenfest", qs_force(ikind)%ehrenfest(1:3, i), &
575 1348 : iatom, ikind, " efield", qs_force(ikind)%efield(1:3, i), &
576 1348 : iatom, ikind, " eev", qs_force(ikind)%eev(1:3, i), &
577 1348 : iatom, ikind, " mp2_non_sep", qs_force(ikind)%mp2_non_sep(1:3, i), &
578 1685 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
579 1480 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
580 : END DO
581 : CASE (1)
582 0 : DO iatom = 1, natom
583 0 : ikind = kind_of(iatom)
584 0 : i = atom_of_kind(iatom)
585 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
586 0 : iatom, ikind, " overlap", qs_force(ikind)%overlap(1:3, i), &
587 0 : iatom, ikind, " overlap_admm", qs_force(ikind)%overlap_admm(1:3, i), &
588 0 : iatom, ikind, " kinetic", qs_force(ikind)%kinetic(1:3, i), &
589 0 : iatom, ikind, " gth_ppl", qs_force(ikind)%gth_ppl(1:3, i), &
590 0 : iatom, ikind, " gth_nlcc", qs_force(ikind)%gth_nlcc(1:3, i), &
591 0 : iatom, ikind, " gth_ppnl", qs_force(ikind)%gth_ppnl(1:3, i), &
592 0 : iatom, ikind, " all_potential", qs_force(ikind)%all_potential(1:3, i), &
593 0 : iatom, ikind, " core_overlap", qs_force(ikind)%core_overlap(1:3, i), &
594 0 : iatom, ikind, " rho_core", qs_force(ikind)%rho_core(1:3, i), &
595 0 : iatom, ikind, " rho_elec", qs_force(ikind)%rho_elec(1:3, i), &
596 0 : iatom, ikind, " rho_lri_elec", qs_force(ikind)%rho_lri_elec(1:3, i), &
597 0 : iatom, ikind, " vhxc_atom", qs_force(ikind)%vhxc_atom(1:3, i), &
598 0 : iatom, ikind, " g0s_Vh_elec", qs_force(ikind)%g0s_Vh_elec(1:3, i), &
599 0 : iatom, ikind, " ch_pulay", qs_force(ikind)%ch_pulay(1:3, i), &
600 0 : iatom, ikind, " dispersion", qs_force(ikind)%dispersion(1:3, i), &
601 0 : iatom, ikind, " gCP", qs_force(ikind)%gcp(1:3, i), &
602 0 : iatom, ikind, " fock_4c", qs_force(ikind)%fock_4c(1:3, i), &
603 0 : iatom, ikind, " ehrenfest", qs_force(ikind)%ehrenfest(1:3, i), &
604 0 : iatom, ikind, " efield", qs_force(ikind)%efield(1:3, i), &
605 0 : iatom, ikind, " eev", qs_force(ikind)%eev(1:3, i), &
606 0 : iatom, ikind, " mp2_non_sep", qs_force(ikind)%mp2_non_sep(1:3, i), &
607 0 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
608 0 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
609 : END DO
610 : CASE (2)
611 75 : DO iatom = 1, natom
612 73 : ikind = kind_of(iatom)
613 73 : i = atom_of_kind(iatom)
614 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
615 292 : iatom, ikind, " all_potential", qs_force(ikind)%all_potential(1:3, i), &
616 292 : iatom, ikind, " rho_elec", qs_force(ikind)%rho_elec(1:3, i), &
617 365 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
618 294 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
619 : END DO
620 : CASE (3)
621 0 : DO iatom = 1, natom
622 0 : ikind = kind_of(iatom)
623 0 : i = atom_of_kind(iatom)
624 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
625 0 : iatom, ikind, " overlap", qs_force(ikind)%overlap(1:3, i), &
626 0 : iatom, ikind, "overlap_admm", qs_force(ikind)%overlap_admm(1:3, i), &
627 0 : iatom, ikind, " kinetic", qs_force(ikind)%kinetic(1:3, i), &
628 0 : iatom, ikind, " gth_ppl", qs_force(ikind)%gth_ppl(1:3, i), &
629 0 : iatom, ikind, " gth_nlcc", qs_force(ikind)%gth_nlcc(1:3, i), &
630 0 : iatom, ikind, " gth_ppnl", qs_force(ikind)%gth_ppnl(1:3, i), &
631 0 : iatom, ikind, " core_overlap", qs_force(ikind)%core_overlap(1:3, i), &
632 0 : iatom, ikind, " rho_core", qs_force(ikind)%rho_core(1:3, i), &
633 0 : iatom, ikind, " rho_elec", qs_force(ikind)%rho_elec(1:3, i), &
634 0 : iatom, ikind, " ch_pulay", qs_force(ikind)%ch_pulay(1:3, i), &
635 0 : iatom, ikind, " fock_4c", qs_force(ikind)%fock_4c(1:3, i), &
636 0 : iatom, ikind, " mp2_non_sep", qs_force(ikind)%mp2_non_sep(1:3, i), &
637 0 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
638 0 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
639 : END DO
640 : CASE (4)
641 152 : DO iatom = 1, natom
642 134 : ikind = kind_of(iatom)
643 134 : i = atom_of_kind(iatom)
644 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
645 536 : iatom, ikind, " all_potential", qs_force(ikind)%all_potential(1:3, i), &
646 536 : iatom, ikind, " overlap", qs_force(ikind)%overlap(1:3, i), &
647 536 : iatom, ikind, " rho_elec", qs_force(ikind)%rho_elec(1:3, i), &
648 536 : iatom, ikind, " repulsive", qs_force(ikind)%repulsive(1:3, i), &
649 536 : iatom, ikind, " dispersion", qs_force(ikind)%dispersion(1:3, i), &
650 536 : iatom, ikind, " efield", qs_force(ikind)%efield(1:3, i), &
651 536 : iatom, ikind, " ehrenfest", qs_force(ikind)%ehrenfest(1:3, i), &
652 670 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
653 554 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
654 : END DO
655 : CASE (5)
656 152 : DO iatom = 1, natom
657 0 : ikind = kind_of(iatom)
658 0 : i = atom_of_kind(iatom)
659 : WRITE (UNIT=output_unit, FMT=fmtstr2) &
660 0 : iatom, ikind, " overlap", qs_force(ikind)%overlap(1:3, i), &
661 0 : iatom, ikind, " kinetic", qs_force(ikind)%kinetic(1:3, i), &
662 0 : iatom, ikind, " rho_elec", qs_force(ikind)%rho_elec(1:3, i), &
663 0 : iatom, ikind, " dispersion", qs_force(ikind)%dispersion(1:3, i), &
664 0 : iatom, ikind, " all potential", qs_force(ikind)%all_potential(1:3, i), &
665 0 : iatom, ikind, " other", qs_force(ikind)%other(1:3, i), &
666 0 : iatom, ikind, " total", qs_force(ikind)%total(1:3, i)
667 0 : grand_total(1:3) = grand_total(1:3) + qs_force(ikind)%total(1:3, i)
668 : END DO
669 : END SELECT
670 :
671 152 : WRITE (UNIT=output_unit, FMT=fmtstr3) "Sum of total", grand_total(1:3)
672 :
673 152 : DEALLOCATE (atom_of_kind)
674 152 : DEALLOCATE (kind_of)
675 :
676 : END IF
677 :
678 10227 : END SUBROUTINE write_forces
679 :
680 : END MODULE qs_force
|