Skip to content

Try speedup getindex #382

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
charleskawczynski wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Try speedup getindex #382

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 29 additions & 7 deletions src/optics/CloudOptics.jl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,35 @@
This function computes the longwave TwoStream clouds optics properties and adds them
to the TwoStream longwave gas optics properties.
"""
function add_cloud_optics_2stream(op::TwoStream, as::AtmosphericState, lkp::LookUpLW, lkp_cld, glay, gcol, ibnd, igpt)
if as.cld_mask_lw isa AbstractArray
cld_mask = as.cld_mask_lw[igpt, glay, gcol]
if cld_mask
τ_cl, τ_cl_ssa, τ_cl_ssag = compute_cld_props(lkp_cld, as, cld_mask, glay, gcol, ibnd, igpt)
increment!(op, τ_cl, τ_cl_ssa, τ_cl_ssag, glay, gcol, igpt)
end
add_cloud_optics_2stream(op::TwoStream, as::AtmosphericState, lkp::LookUpLW, lkp_cld, glay, gcol, ibnd, igpt) =
add_cloud_optics_2stream(op, as, lkp, lkp_cld, glay, gcol, ibnd, igpt, as.cld_mask_lw)
add_cloud_optics_2stream(
op::TwoStream,
as::AtmosphericState,
lkp::LookUpLW,
lkp_cld,
glay,
gcol,
ibnd,
igpt,
cld_mask_lw,
) = nothing

function add_cloud_optics_2stream(
op::TwoStream,
as::AtmosphericState,
lkp::LookUpLW,
lkp_cld,
glay,
gcol,
ibnd,
igpt,
cld_mask_lw::AbstractArray,
)
@inbounds cld_mask = as.cld_mask_lw[igpt, glay, gcol]
if cld_mask
τ_cl, τ_cl_ssa, τ_cl_ssag = compute_cld_props(lkp_cld, as, cld_mask, glay, gcol, ibnd, igpt)
increment!(op, τ_cl, τ_cl_ssa, τ_cl_ssag, glay, gcol, igpt)
end
return nothing
end
Expand Down