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)