fix markerscene, precompile [skip ci]

Author: islent

Project.toml

@@ -20,6 +20,7 @@ ParallelOperations = "09c1cff2-b94d-4c31-85a1-721512e21c63"
 PhysicalMeshes = "97d9904f-034f-4fb7-aeaa-03a173434233"
 PhysicalParticles = "78724250-4eb2-11e9-0f34-13a7aab864c5"
 PhysicalTrees = "dc5d4990-50a5-4af2-bdaa-6f78dc9b9fb2"
+PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
@@ -47,6 +48,7 @@ ParallelOperations = "0.1"
 PhysicalMeshes = "0.1"
 PhysicalParticles = "1"
 PhysicalTrees = "0.1"
+PrecompileTools = "1"
 ProgressMeter = "1"
 Reexport = "1"
 StructArrays = "0.6"

src/AstroPlot.jl

@@ -1,5 +1,6 @@
 module AstroPlot
 
+using PrecompileTools
 using Reexport
 using Unitful, UnitfulAstro
 using LaTeXStrings
@@ -137,4 +138,5 @@ include("snapshots/profiling.jl")
 include("video.jl")
 include("images.jl")
 
+include("precompile.jl")
 end

src/PhysicalParticles.jl

@@ -3,7 +3,7 @@ function point3(p::PVector)
 end
 
 function estimate_markersize(S::Real)
-    return 0.0001 / sqrt(S)
+    return 0.1 / sqrt(S)
 end
 
 function estimate_markersize(data, u; xaxis = :x, yaxis = :y)
@@ -34,7 +34,7 @@ plot_makie(d, u"m")
 """
 function plot_makie(data::Array{T,1}, u::Union{Nothing, Unitful.FreeUnits} = u"kpc";
     markersize = estimate_markersize(data, u),
-    markerspace=SceneSpace,
+    markerspace=:data,
     size = (1000, 1000),
     kw...
 ) where T<:PVector
@@ -44,7 +44,7 @@ end
 
 function plot_makie(data::Array{T,1}, u::Union{Nothing, Unitful.FreeUnits} = u"kpc";
     markersize = estimate_markersize(data, u),
-    markerspace=SceneSpace,
+    markerspace=:data,
     size = (1000, 1000),
     kw...
 ) where T<:AbstractParticle3D
@@ -84,7 +84,7 @@ plot_makie!(fig, d, u"m")
 """
 function plot_makie!(fig, data::Array{T,1}, u::Union{Nothing, Unitful.FreeUnits} = u"kpc";
     markersize = estimate_markersize(data, u),
-    markerspace=SceneSpace,
+    markerspace=:data,
     size = (1000, 1000),
     kw...
 ) where T<:PVector
@@ -94,7 +94,7 @@ end
 
 function plot_makie!(fig, data::Array{T,1}, u::Union{Nothing, Unitful.FreeUnits} = u"kpc";
     markersize = estimate_markersize(data, u),
-    markerspace=SceneSpace,
+    markerspace=:data,
     size = (1000, 1000),
     kw...
 ) where T<:AbstractParticle3D

src/precompile.jl

@@ -0,0 +1,58 @@
+@setup_workload begin
+    @compile_workload begin
+        foldertest = joinpath(@__DIR__, "../test")
+        header, data = read_gadget2(joinpath(foldertest, "plummer/snapshot_0000.gadget2"), uAstro, uGadget2, type=Star)
+        h, d = read_gadget2(joinpath(foldertest, "plummer_unitless.gadget2"), nothing, uGadget2, type=Star)
+
+        c = Cube(PVector(0.0,0.0,0.0), PVector(1.0,1.0,1.0))
+        fig = plot_makie(c, nothing)
+        f = Figure()
+        ax = Axis3(f[1,1])
+        plot_makie!(ax, c, nothing)
+
+        m = MeshCartesianStatic(d)
+        unicode_projection_density(m)
+        # f = projection_density(m) #TODO
+        unicode_slice(m, :rho, 5)
+        f = plot_slice(m, :pos, 5)
+
+        #! This is breaking. Anyway, tree plot is not often used
+        # fig = plot_peano(3)
+        # d = randn_pvector(15)
+        # t = octree(d)
+        # figure, axis, plot = plot_makie(t)
+
+        fig = plot_profiling(joinpath(foldertest, "profiling.csv"))
+        fig, df = plot_energy(joinpath(foldertest, "energy.csv"))
+        fig, df = plot_energy_delta(joinpath(foldertest, "energy.csv"))
+        fig = plot_densitycurve(data)
+        fig = plot_rotationcurve(data)
+
+        #TODO: there is something wrong???
+        plot_positionslice(
+            joinpath(foldertest, "plummer"), "snapshot_", collect(0:20:200), ".gadget2", gadget2(),
+            xlims = (-0.05, +0.05), ylims = (-0.05, +0.05),
+            times = collect(0.0:0.01:0.1) * u"Gyr",
+        )
+        fig, pos = plot_trajectory(
+            joinpath(foldertest, "plummer/"), "snapshot_", collect(0:20:200), [1,2,3], ".gadget2", gadget2(),
+        )
+        FigScale, FigLagrange, df = plot_radii(
+            joinpath(foldertest, "plummer/"), "snapshot_", collect(0:20:200), ".gadget2", gadget2(),
+            times = collect(0.0:0.01:0.1) * u"Gyr", title = "Direct Sum const",
+        )
+
+        plot_positionslice(joinpath(foldertest, "mosaic/"), "snapshot_", collect(1:9:100), ".gadget2", gadget2(),
+            dpi = 300, size = (800,800),
+            xlims = (-0.06, +0.06), ylims = (-0.06, +0.06),
+            times = collect(0.0:0.00005:0.005) * u"Gyr",
+        )
+        plt = mosaicview(joinpath(foldertest, "mosaic/"), "pos_", collect(1:9:100), ".png"; fillvalue = 0.5, npad = 3, ncol = 4, rowmajor = true)
+
+        png2video(joinpath(foldertest, "mosaic/"), "pos_", ".png", "TDE.mp4")
+
+        unicode_scatter(d, nothing)
+        unicode_density(d, nothing)
+        unicode_rotationcurve(d, nothing)
+    end
+end

src/snapshots/positions.jl

@@ -12,7 +12,7 @@ function plot_positionslice!(fig, data, u::Union{Nothing, Unitful.FreeUnits} = n
                              xaxis = :x,
                              yaxis = :y,
                              markersize = estimate_markersize(data, u; xaxis, yaxis),
-                             markerspace=SceneSpace,
+                             markerspace=:data,
                              kw...)
     xu, yu = pack_xy(data; xaxis, yaxis)
     x = ustrip.(u, xu)
@@ -55,7 +55,7 @@ function plot_positionslice(data, u::Union{Nothing, Unitful.FreeUnits} = nothing
                             aspect_ratio = 1.0,
                             title = "Positions",
                             markersize = (isnothing(xlims) && isnothing(ylims)) ? estimate_markersize(data, u; xaxis, yaxis) : estimate_markersize((xlims[2] - xlims[1]) * (ylims[2] - ylims[1])),
-                            markerspace=SceneSpace,
+                            markerspace=:data,
                             size = (1000, 1000),
                             kw...)
     xu, yu = pack_xy(data; xaxis, yaxis)
@@ -173,12 +173,12 @@ function plot_positionslice_adapt(data, u::Union{Nothing, Unitful.FreeUnits} = n
                                   yaxis = :y,
                                   xlabel = "",
                                   ylabel = "",
-                                  xlen::Float64 = 1.0,
-                                  ylen::Float64 = 1.0,
+                                  xlen = 1.0,
+                                  ylen = 1.0,
                                   aspect_ratio = 1.0,
                                   title = "Positions",
                                   markersize = estimate_markersize(xlen * ylen),
-                                  markerspace=SceneSpace,
+                                  markerspace=:data,
                                   size = (1000, 1000),
                                   kw...)
     xu, yu = pack_xy(data; xaxis, yaxis)
@@ -241,8 +241,8 @@ function plot_positionslice_adapt(folder::String, filenamebase::String, Counts::
                                   collection::Union{Nothing, Collection} = nothing,
                                   xlabel = "$(xaxis)$(axisunit(getuLength(units)))",
                                   ylabel = "$(yaxis)$(axisunit(getuLength(units)))",
-                                  xlen::Float64 = 0.2,
-                                  ylen::Float64 = 0.2,
+                                  xlen = 0.2,
+                                  ylen = 0.2,
                                   formatstring = "%04d",
                                   type = Star,
                                   kw...)