add some titles, labels for pyplot

WIAS-PDELib · Jul 14, 2021 · 8b2a3fa · 8b2a3fa · j-fu · Jul 14, 2021
1 parent e431365
commit 8b2a3fa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "GridVisualize"
 uuid = "5eed8a63-0fb0-45eb-886d-8d5a387d12b8"
 authors = ["Juergen Fuhrmann <[email protected]>"]
-version = "0.2.9"
+version = "0.2.10"
 
 [deps]
 ColorSchemes = "35d6a980-a343-548e-a6ea-1d62b119f2f4"

diff --git a/src/pyplot.jl b/src/pyplot.jl
@@ -310,7 +310,7 @@ end
 
 
 ### 1D Function
-function scalarplot!(ctx, TP::Type{PyPlotType}, ::Type{Val{1}},grid, func)
+function scalarplot!(ctx, TP::Type{PyPlotType}, ::Type{Val{1}},grid, func::AbstractVector)
     PyPlot=ctx[:Plotter]
     if !haskey(ctx,:ax)
         ctx[:ax]=ctx[:figure].add_subplot(ctx[:layout]...,ctx[:iplot])
@@ -446,6 +446,10 @@ function scalarplot!(ctx, TP::Type{PyPlotType}, ::Type{Val{2}},grid, func)
     if ctx[:colorbar]==:vertical
         ctx[:cbar]=fig.colorbar(cnt,ax=ax,ticks=isolines,boundaries=colorlevels, orientation="vertical")
     end
+
+    ax.set_xlabel(ctx[:xlabel])
+
+    ax.set_ylabel(ctx[:ylabel])
 
     reveal(ctx,TP)
 end
@@ -512,6 +516,7 @@ function scalarplot!(ctx, TP::Type{PyPlotType}, ::Type{Val{3}},grid,func)
     ax.set_zlim3d(xyzmin[3],xyzmax[3])
     ax.view_init(ctx[:elev],ctx[:azim])
 
+    ax.set_title(ctx[:title])
 
     if ctx[:legend]!=:none
         ax.legend(loc=leglocs[ctx[:legend]])