Mercurial > repos > public > sbplib_julia
comparison sbpPlot.jl @ 125:22642722a8ec cell_based_test
Merge
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Wed, 13 Feb 2019 11:03:39 +0100 |
| parents | 631eb9b35d72 |
| children | d61bfc8cf6a5 |
comparison
equal
deleted
inserted
replaced
| 123:5df4ccb19476 | 125:22642722a8ec |
|---|---|
| 1 module sbpPlot | |
| 2 using PyPlot, PyCall | |
| 3 | |
| 4 function plotgridfunction(grid::EquidistantGrid, gridfunction) | |
| 5 if dimension(grid) == 1 | |
| 6 plot(pointsalongdim(grid,1), gridfunction, linewidth=2.0) | |
| 7 elseif dimension(grid) == 2 | |
| 8 mx = grid.size[1] | |
| 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 | |
| 14 error(string("Plot not implemented for dimension ", string(dimension(grid)))) | |
| 15 end | |
| 16 end | |
| 17 end |