GNOM/src/plots/GMDpaper/transects_eNd.jl at main · MTEL-USC/GNOM · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#================================================
Transects
================================================#
include("../plots_setup_Nd.jl")
# Basin profiles where obs only
# Only keep obs part of the cost function
# and interpolate model onto their location
# Function to extract indices of those model boxes (or obs.) that go in comparison


# Full figure with every transect
ts = εNd_transects.transects[ε_t_sort] # skips GA04 because it has no water values? (why is it in Med??)
nt = length(ts)
fig = Figure(resolution=(1600, 1600))
#display(s)
axs = Axis[]
scs = Any[]
# colors for cruises
tcol = ColorSchemes.tableau_10.colors
tcol = [:black; tcol]
tcol = tcol[[1,2,4,6,7,8,9,10,11]] # keep same colors for transects (but εNd transects is a subset of [Nd] transects)
# Minimap in top left
subl = GridLayout(1,2)
ax = subl[1,1] = Axis(fig, backgroundcolor=water_color)
fig[1,1] = subl
mapit!(ax, clon, mypolys(clon), color=land_color)
cts = Any[]
maptransects!(ax, cts, ts, tcol, wlon)
Makie.xlims!(ax, clon .+ (-180,180))
Makie.ylims!(ax, (-90,90))
#mapit!(ax, clon, mypolys(clon), color=:transparent, strokecolor=:black)
text!(ax, panellabels[1], fontsize=20, align = (:left, :top), offset = (4, -2), space = :relative, font=labelfont, color=labelcol)
hidedecorations!(ax)
hidespines!(ax)
# Legend
idx = permutedims(reshape(1:nt+1, (nt÷2+1, 2)), (2,1))[:]
labels = [" "; [t.cruise for t in ts]][idx]
invisible_element = LineElement(color = :transparent, linewidth = 0, linestyle = nothing)
legelems = [invisible_element; [PolyElement(color=tcol[i], strokecolor=:transparent) for i in eachindex(ts)]][idx]
leg = Legend(fig, legelems, labels, nbanks=2, labelsize=20, framevisible=false)
subl[1,2] = leg
for it in eachindex(ts)
    i, j = Tuple(CartesianIndices((nt÷2+1, 2))[it+1])
    t = sort(OceanographyCruises.shiftlon(ts[it], baselon=wlon))
    ct = CruiseTrack(t)
    local ax = fig[i,j] = Axis(fig, xgridvisible=false, ygridvisible=false, backgroundcolor=land_color, halign=:left)
    push!(axs, ax)
    hm = makietransect!(ax, εNdmodel, ct, colormap=εcmap, nan_color=nan_color, levels=εlevels, colorrange=εclims,
                       extendhigh = :auto, extendlow = :auto) # note: added εlevels with contourf
    sc = makiescattertransect!(ax, t, markersize=markersize, colormap=εcmap)
    push!(scs, hm)
    sc.plots[end].colorrange = εclims # color of last overlaid scatter
    Makie.ylims!(ax, (6000,0))
    dis = ustrip(km, maximum(scattertransect(t)[1]))
    if dis > 1500
        Label(fig, bbox = axs[it].scene.px_area, t.cruise, fontsize=20, halign=:right, valign=:bottom, padding=(0,35,5,0), font=labelfont, color=labelcol)
        Label(fig, bbox = axs[it].scene.px_area, panellabels[it+1], fontsize=20, halign=:left, valign=:bottom, padding=(10,10,5,10), font=labelfont, color=labelcol)
        Label(fig, bbox = axs[it].scene.px_area, "■", fontsize=20, halign=:right, valign=:bottom, padding=(0,10,5,0), font=labelfont, color=tcol[it])
    else # special treatment for tiny transects (labels don't fit)
        Label(fig, bbox = axs[it].scene.px_area, string(panellabels[it+1], " ", t.cruise), fontsize=20, halign=:right, valign=:bottom, padding=(0,-115,5,0), font=labelfont, color=labelcol)
        Label(fig, bbox = axs[it].scene.px_area, "■", fontsize=20, halign=:right, valign=:bottom, padding=(0,-140,5,0), font=labelfont, color=tcol[it])
    end
    ax.xlabel="Distance (km)"
    ax.ylabel="Depth (m)"
    !((i==nt÷2+1) || (it==nt)) && hidexdecorations!(ax, ticks=false)
    j==2 && hideydecorations!(ax, ticks=false)
    tightlimits!(ax, Bottom())
    ax.xticks = 0:1000:maximum(dis)
    ax.width = 700 * (dis+2t_ext) / 8000
    Makie.xlims!(ax, (-t_ext, dis+t_ext))
    Makie.ylims!(ax, (6001, -1))
    hidespines!(ax, :t, :r)
end
# colorbar (need to use scs[1].plots[end] because doubly-overlaid-scatter plot)
cbar = fig[end+1,:] = Colorbar(fig, scs[1].plots[1], vertical=false, width=600, height=30,
                               flipaxis=false, ticklabelalign=(:center, :top), label="εNd (‱)",
                               labelsize=25, ticks=εlevels[1:10:end])
cbar.alignmode = Outside()
cbar.tellheight = true
# final touches
trim!(fig.layout)

if use_GLMakie
    display(fig) # show the output wiht GLMakie
else
    save(joinpath(archive_path, "eNd_transects_$(lastcommit)_run$(run_num).pdf"), fig)
    nothing # just so that no output is spat out
end