Skip to content

Commit b1457b0

Browse files
committed
Fix hiatusdist output
1 parent 51f2f10 commit b1457b0

File tree

2 files changed

+17
-22
lines changed

2 files changed

+17
-22
lines changed

src/StratMetropolis.jl

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@
125125
model_heights = model_heights[active_height_t]
126126
mdl = StratAgeModel(model_heights, agedist)
127127

128-
return mdl, agedist, lldist, hiatusdist
128+
return mdl, agedist, hiatusdist, lldist
129129
end
130130

131131
## --- Stratigraphic MCMC model without hiatus, with distribution LL # # # # # #
@@ -264,7 +264,7 @@
264264
model_heights = model_heights[active_height_t]
265265
mdl = StratAgeModel(model_heights, agedist)
266266

267-
return mdl, agedist, lldist, hiatusdist
267+
return mdl, agedist, hiatusdist, lldist
268268
end
269269

270270
## --- Stratigraphic MCMC model without hiatus, for radiocarbon ages # # # # # #
@@ -405,7 +405,7 @@
405405
model_heights = model_heights[active_height_t]
406406
mdl = StratAgeModel(model_heights, agedist)
407407

408-
return mdl, agedist, lldist, hiatusdist
408+
return mdl, agedist, hiatusdist, lldist
409409
end
410410

411411
## --- # Internals of the Markov chain
@@ -716,9 +716,13 @@
716716
if model_agesₚ[h-1] == model_agesₚ[h]
717717
n = findclosestunequal(model_agesₚ, h)
718718
if n < h
719-
model_agesₚ[n:h-1] .= model_agesₚ[n]
719+
@inbounds for i = n:h-1
720+
model_agesₚ[i] = model_agesₚ[n]
721+
end
720722
elseif n > h
721-
model_agesₚ[h:n] .= model_agesₚ[n]
723+
@inbounds for i = h:n
724+
model_agesₚ[i] = model_agesₚ[n]
725+
end
722726
end
723727
end
724728
end
@@ -740,7 +744,7 @@
740744

741745
# Add log likelihood for hiatus duration
742746
@. durationₚ = model_agesₚ[closest_hiatus_unique - 1] - model_agesₚ[closest_hiatus_unique]
743-
llₚ += normcdf_ll!(Hiatus_duration, Hiatus_duration_sigma, durationₚ)
747+
llₚ += normcdf_ll(Hiatus_duration, Hiatus_duration_sigma, durationₚ)
744748

745749
# Accept or reject proposal based on likelihood
746750
if log(rand(Float64)) < (llₚ - ll)
@@ -809,9 +813,13 @@
809813
if model_agesₚ[h-1] == model_agesₚ[h]
810814
n = findclosestunequal(model_agesₚ, h)
811815
if n < h
812-
model_agesₚ[n:h-1] .= model_agesₚ[n]
816+
@inbounds for i = n:h-1
817+
model_agesₚ[i] = model_agesₚ[n]
818+
end
813819
elseif n > h
814-
model_agesₚ[h:n] .= model_agesₚ[n]
820+
@inbounds for i = h:n
821+
model_agesₚ[i] = model_agesₚ[n]
822+
end
815823
end
816824
end
817825
end
@@ -832,7 +840,7 @@
832840

833841
# Add log likelihood for hiatus duration
834842
@. durationₚ = model_agesₚ[closest_hiatus_unique - 1] - model_agesₚ[closest_hiatus_unique]
835-
llₚ += normcdf_ll!(Hiatus_duration, Hiatus_duration_sigma, durationₚ)
843+
llₚ += normcdf_ll(Hiatus_duration, Hiatus_duration_sigma, durationₚ)
836844

837845
# Accept or reject proposal based on likelihood
838846
if log(rand(Float64)) < (llₚ - ll)

src/foo.jl

Lines changed: 0 additions & 13 deletions
This file was deleted.

0 commit comments

Comments
 (0)