Mercurial > repos > public > sbplib_julia
diff grid.jl @ 26:d2896e70cd4f
Merge with default
author | Ylva Rydin <ylva.rydin@telia.com> |
---|---|
date | Mon, 17 Dec 2018 16:35:08 +0100 |
parents | af8469bc1cb3 |
children | 2dbdd00eaea0 aff8ea85ca70 |
line wrap: on
line diff
--- a/grid.jl Mon Dec 17 16:30:39 2018 +0100 +++ b/grid.jl Mon Dec 17 16:35:08 2018 +0100 @@ -1,1 +1,54 @@ -abstract type BoundaryID end +module grid + +abstract type Grid end + +function numberOfDimensions(grid::Grid) + error("Not yet implemented") +end + +function numberOfPoints(grid::Grid) + error("Not yet implemented") +end + +function points(grid::Grid) + error("Not yet implemented") +end + +abstract type BoundaryId end + +# Move to seperate file. +struct EquidistantGrid <: Grid + nPointsPerDim::Vector{Int} + limits::Vector{Pair{Real, Real}} + function EquidistantGrid(nPointsPerDim, lims) + @assert length(lims) == length(nPointsPerDim) + return new(nPointsPerDim, lims) + end +end + +function numberOfDimensions(grid::EquidistantGrid) + return length(grid.nPointsPerDim) +end + +function numberOfPoints(grid::EquidistantGrid) + numberOfPoints = grid.nPointsPerDim[1]; + for i = 2:length(grid.nPointsPerDim); + numberOfPoints = numberOfPoints*grid.nPointsPerDim[i] + end + return numberOfPoints +end + +function points(grid::EquidistantGrid) + points = Vector{Real}(undef, numberOfPoints(grid)) + for i = 1:numberOfDimensions(grid) + lims = limitsForDimension(grid,i) + points = range(lims.first, stop=lims.second, length=grid.nPointsPerDim[i]) + end + return points +end + +function limitsForDimension(grid::EquidistantGrid, dim::Int) + return grid.limits[dim] +end + +end