diff --git a/src/CMBLensing.jl b/src/CMBLensing.jl index ca04cff9..eaaec89f 100644 --- a/src/CMBLensing.jl +++ b/src/CMBLensing.jl @@ -145,9 +145,6 @@ include("masking.jl") include("taylens.jl") include("bilinearlens.jl") -# plotting -include("plots.jl") - # PPL include("distributions.jl") include("simpleppl.jl") @@ -158,6 +155,9 @@ include("maximization.jl") include("sampling.jl") include("chains.jl") +# plotting +include("plots.jl") + # deprecated stuff include("deprecated.jl") diff --git a/src/plots.jl b/src/plots.jl index ac276402..f4dc9969 100644 --- a/src/plots.jl +++ b/src/plots.jl @@ -39,4 +39,24 @@ end @recipe function plot(Cℓ::Cℓs) (Cℓ.ℓ, Cℓ.Cℓ) +end + +@recipe function plot(k::GetDistKDE{2}; clevels=[0.95,0.68], filled=false) + + if filled + seriestype := :contourf + levels := [PyArray(k.kde.getContourLevels(clevels)); prevfloat(Inf)] + alpha := 0.5 + else + seriestype := :contour + levels := PyArray(k.kde.getContourLevels(clevels)) + end + cbar := false + + return PyArray(k.kde.x), PyArray(k.kde.y), PyArray(k.kde.P) + +end + +@recipe function plot(k::GetDistKDE{1}; transform=identity) + (PyArray(k.kde.x), transform.(PyArray(k.kde.P))) end \ No newline at end of file