Mercurial > repos > public > sbplib_julia

comparison sbpPlot.jl @ 152:f54dd4408fa7 boundary_conditions

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge with default
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 15 Apr 2019 16:15:04 +0200
parents d61bfc8cf6a5
children
comparison
equal deleted inserted replaced
145:e0c8f5cf3a3f 152:f54dd4408fa7
1 module sbpPlot 1 include("sbp.jl")
2 using PyPlot, PyCall 2 using Makie
3 3 import .sbp.Grid
4 function plotgridfunction(grid::EquidistantGrid, gridfunction) 4 function plotgridfunction(grid::sbp.Grid.EquidistantGrid, gridfunction::AbstractArray)
5 if dimension(grid) == 1 5 if sbp.Grid.dimension(grid) == 1
6 plot(pointsalongdim(grid,1), gridfunction, linewidth=2.0) 6 plot(sbp.Grid.pointsalongdim(grid,1), gridfunction)
7 elseif dimension(grid) == 2 7 elseif sbp.Grid.dimension(grid) == 2
8 mx = grid.size[1] 8 scene = surface(sbp.Grid.pointsalongdim(grid,1),sbp.Grid.pointsalongdim(grid,2), gridfunction)
9 my = grid.size[2]
10 X = repeat(pointsalongdim(grid,1),1,my)
11 Y = permutedims(repeat(pointsalongdim(grid,2),1,mx))
12 plot_surface(X,Y,reshape(gridfunction,mx,my));
13 else 9 else
14 error(string("Plot not implemented for dimension ", string(dimension(grid)))) 10 error(string("Plot not implemented for dimension ", string(dimension(grid))))
15 end 11 end
16 end 12 end
17 end