Mercurial > repos > public > sbplib_julia

view sbpPlot.jl @ 134:79699dda29be

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge in cell_based_test
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 21 Feb 2019 16:27:28 +0100
parents 631eb9b35d72
children d61bfc8cf6a5
line wrap: on
line source

module sbpPlot
using PyPlot, PyCall

function plotgridfunction(grid::EquidistantGrid, gridfunction)
    if dimension(grid) == 1
        plot(pointsalongdim(grid,1), gridfunction, linewidth=2.0)
    elseif dimension(grid) == 2
        mx = grid.size[1]
        my = grid.size[2]
        X = repeat(pointsalongdim(grid,1),1,my)
        Y = permutedims(repeat(pointsalongdim(grid,2),1,mx))
        plot_surface(X,Y,reshape(gridfunction,mx,my));
    else
        error(string("Plot not implemented for dimension ", string(dimension(grid))))
    end
end
end