Skip to content

Commit b1e0a46

Browse files
committed
Calculate force density for t=0 inside data handler initialization
1 parent 95d3cca commit b1e0a46

File tree

5 files changed

+13
-6
lines changed

5 files changed

+13
-6
lines changed

src/core/data_handler.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
function initialize!(dh, solver)
2+
calc_force_density!(dh, 0.0, solver.Δt)
23
return nothing
34
end
45

src/core/mpi_body_data_handler.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -421,7 +421,8 @@ function export_reference_results(dh::MPIBodyDataHandler, options::AbstractJobOp
421421
return nothing
422422
end
423423

424-
function initialize!(::AbstractMPIBodyDataHandler, ::AbstractTimeSolver)
424+
function initialize!(dh::AbstractMPIBodyDataHandler, solver::AbstractTimeSolver)
425+
calc_force_density!(dh, 0.0, solver.Δt)
425426
return nothing
426427
end
427428

src/core/threads_multibody_data_handler.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,9 @@ function update_caches!(dh::ThreadsMultibodyDataHandler)
8585
return nothing
8686
end
8787

88-
function initialize!(dh::AbstractThreadsMultibodyDataHandler, ::AbstractTimeSolver)
88+
function initialize!(dh::AbstractThreadsMultibodyDataHandler, solver::AbstractTimeSolver)
8989
init_contact_nhs!(dh)
90+
calc_force_density!(dh, 0.0, solver.Δt)
9091
return nothing
9192
end
9293

src/discretization/bond_associated_system.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,20 +105,22 @@ end
105105
@inline get_hood_volume(chunk::AbstractBodyChunk) = chunk.system.hood_volume
106106

107107
function initialize!(dh::AbstractThreadsBodyDataHandler{<:BondAssociatedSystem},
108-
::AbstractTimeSolver)
108+
solver::AbstractTimeSolver)
109109
@threads :static for chunk in dh.chunks
110110
calc_hood_volumes!(chunk)
111111
end
112112
@threads :static for chunk_id in eachindex(dh.chunks)
113113
exchange_loc_to_halo!(get_hood_volume, dh, chunk_id)
114114
end
115+
calc_force_density!(dh, 0.0, solver.Δt)
115116
return nothing
116117
end
117118

118119
function initialize!(dh::AbstractMPIBodyDataHandler{<:BondAssociatedSystem},
119-
::AbstractTimeSolver)
120+
solver::AbstractTimeSolver)
120121
calc_hood_volumes!(dh.chunk)
121122
exchange_loc_to_halo!(get_hood_volume, dh)
123+
calc_force_density!(dh, 0.0, solver.Δt)
122124
return nothing
123125
end
124126

src/discretization/bond_system_corrections.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,22 +53,24 @@ end
5353
end
5454

5555
function initialize!(dh::AbstractThreadsBodyDataHandler{BondSystem{EnergySurfaceCorrection}},
56-
::AbstractTimeSolver)
56+
solver::AbstractTimeSolver)
5757
@threads :static for chunk in dh.chunks
5858
calc_mfactor!(chunk)
5959
end
6060
@threads :static for chunk_id in eachindex(dh.chunks)
6161
exchange_loc_to_halo!(get_mfactor, dh, chunk_id)
6262
calc_scfactor!(dh.chunks[chunk_id])
6363
end
64+
calc_force_density!(dh, 0.0, solver.Δt)
6465
return nothing
6566
end
6667

6768
function initialize!(dh::AbstractMPIBodyDataHandler{BondSystem{EnergySurfaceCorrection}},
68-
::AbstractTimeSolver)
69+
solver::AbstractTimeSolver)
6970
calc_mfactor!(dh.chunk)
7071
exchange_loc_to_halo!(get_mfactor, dh)
7172
calc_scfactor!(dh.chunk)
73+
calc_force_density!(dh, 0.0, solver.Δt)
7274
return nothing
7375
end
7476

0 commit comments

Comments
 (0)