Mercurial > repos > public > sbplib_julia
comparison grid.jl @ 46:50c6c252d954
Fix issue in plotgridfunction in 2d. Remove 3d case. Will be added later
author | Vidar Stiernström <vidar.stiernstrom@it.uu.se> |
---|---|
date | Thu, 10 Jan 2019 17:39:42 +0100 |
parents | 3d8bfb695497 |
children | 614b56a017b9 |
comparison
equal
deleted
inserted
replaced
45:56722ba36604 | 46:50c6c252d954 |
---|---|
118 | 118 |
119 function plotgridfunction(grid::EquidistantGrid, gridfunction) | 119 function plotgridfunction(grid::EquidistantGrid, gridfunction) |
120 if numberOfDimensions(grid) == 1 | 120 if numberOfDimensions(grid) == 1 |
121 plot(pointsalongdim(grid,1), gridfunction, linewidth=2.0) | 121 plot(pointsalongdim(grid,1), gridfunction, linewidth=2.0) |
122 elseif numberOfDimensions(grid) == 2 | 122 elseif numberOfDimensions(grid) == 2 |
123 mx = grid.numberOfPointsPerDim[1]; | |
124 my = grid.numberOfPointsPerDim[2]; | |
123 x = pointsalongdim(grid,1) | 125 x = pointsalongdim(grid,1) |
124 X = repeat(x,1,grid.numberOfPointsPerDim[2]) | 126 X = repeat(x,1,my) |
125 y = pointsalongdim(grid,2) | 127 y = pointsalongdim(grid,2) |
126 Y = repeat(y,1,grid.numberOfPointsPerDim[1])' | 128 Y = repeat(y,1,mx)' |
127 surface(X,Y,reshape(gridfunction)) | 129 surface(X,Y,reshape(gridfunction,mx,my)) |
128 elseif numberOfDimensions(grid) == 3 | |
129 x = pointsalongdim(grid,1) | |
130 X = repeat(x,1,grid.numberOfPointsPerDim[2]) | |
131 y = pointsalongdim(grid,2) | |
132 Y = repeat(y,1,grid.numberOfPointsPerDim[1])' | |
133 y = pointsalongdim(grid,2) | |
134 Y = repeat(y,1,grid.numberOfPointsPerDim[1])' | |
135 surface(X,Y,gridfunction) | |
136 else | 130 else |
137 error(string("Plot not implemented for dimension ", string(dim))) | 131 error(string("Plot not implemented for dimension ", string(numberOfDimensions(grid)))) |
138 end | 132 end |
139 end | 133 end |
140 | 134 |
141 # Evaluate function f on the grid g | 135 # Evaluate function f on the grid g |
142 function evalOn(g::Grid, f::Function) | 136 function evalOn(g::Grid, f::Function) |