Mercurial > repos > public > sbplib_julia
annotate src/Grids/grid.jl @ 1516:8d64f8981bb0 feature/grids/componentview
Remove comments and add todo
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Thu, 21 Mar 2024 07:42:19 +0100 |
parents | 6309ca375dcd |
children | 7ee7df7d9b61 |
rev | line source |
---|---|
212
aa17d4d9d09e
Export some functions from AbstractGrid and move documentation to docstrings
Jonatan Werpers <jonatan@werpers.com>
parents:
211
diff
changeset
|
1 """ |
1257
198ccda331a6
Remove range dim as a type paratmeter on Grid as it is already encoded in T if available
Jonatan Werpers <jonatan@werpers.com>
parents:
1256
diff
changeset
|
2 Grid{T,D} |
1222 | 3 |
1331
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
4 A grid with coordinates of type `T`, e.g. `SVector{3,Float64}`, and dimension |
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
5 `D`. The grid can be embedded in a higher dimension in which case the number |
1336
52087a6c0682
Mention grids as grid functions
Jonatan Werpers <jonatan@werpers.com>
parents:
1334
diff
changeset
|
6 of indices and the number of components of the coordinate vectors will be |
1331
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
7 different. |
212
aa17d4d9d09e
Export some functions from AbstractGrid and move documentation to docstrings
Jonatan Werpers <jonatan@werpers.com>
parents:
211
diff
changeset
|
8 |
1336
52087a6c0682
Mention grids as grid functions
Jonatan Werpers <jonatan@werpers.com>
parents:
1334
diff
changeset
|
9 All grids are expected to behave as a grid function for the coordinates. |
52087a6c0682
Mention grids as grid functions
Jonatan Werpers <jonatan@werpers.com>
parents:
1334
diff
changeset
|
10 |
1331
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
11 `Grids` is top level abstract type for grids. A grid should implement Julia's interfaces for |
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
12 indexing and iteration. |
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
13 |
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
14 ## Note |
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
15 |
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
16 Importantly a grid does not have to be an `AbstractArray`. The reason is to |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
17 allow flexible handling of special types of grids like multi-block grids, or |
1331
ed3ea0630825
Add some docs for Grid
Jonatan Werpers <jonatan@werpers.com>
parents:
1330
diff
changeset
|
18 grids with special indexing. |
1222 | 19 """ |
1257
198ccda331a6
Remove range dim as a type paratmeter on Grid as it is already encoded in T if available
Jonatan Werpers <jonatan@werpers.com>
parents:
1256
diff
changeset
|
20 abstract type Grid{T,D} end |
1222 | 21 |
1262
5e28ae42caf2
Clean up IteratorSize and eltype for EquidistantGrid, Grid, and ZeroDimGrid
Jonatan Werpers <jonatan@werpers.com>
parents:
1257
diff
changeset
|
22 Base.ndims(::Grid{T,D}) where {T,D} = D |
5e28ae42caf2
Clean up IteratorSize and eltype for EquidistantGrid, Grid, and ZeroDimGrid
Jonatan Werpers <jonatan@werpers.com>
parents:
1257
diff
changeset
|
23 Base.eltype(::Type{<:Grid{T}}) where T = T |
1333
79a2193da5c1
Implement coordinate_size() and component_type() types
Jonatan Werpers <jonatan@werpers.com>
parents:
1332
diff
changeset
|
24 |
1386
4d628c83987e
Add method `getindex(::Grid,::CartesianIndex)` with a default implementation
Jonatan Werpers <jonatan@werpers.com>
parents:
1347
diff
changeset
|
25 Base.getindex(g::Grid, I::CartesianIndex) = g[Tuple(I)...] |
4d628c83987e
Add method `getindex(::Grid,::CartesianIndex)` with a default implementation
Jonatan Werpers <jonatan@werpers.com>
parents:
1347
diff
changeset
|
26 |
1334
47e309eac131
Add docs for coordinate_size and component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1333
diff
changeset
|
27 """ |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
28 coordinate_size(g) |
1334
47e309eac131
Add docs for coordinate_size and component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1333
diff
changeset
|
29 |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
30 The lenght of the coordinate vector of `Grid` `g`. |
1334
47e309eac131
Add docs for coordinate_size and component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1333
diff
changeset
|
31 """ |
1333
79a2193da5c1
Implement coordinate_size() and component_type() types
Jonatan Werpers <jonatan@werpers.com>
parents:
1332
diff
changeset
|
32 coordinate_size(::Type{<:Grid{T}}) where T = _ncomponents(T) |
79a2193da5c1
Implement coordinate_size() and component_type() types
Jonatan Werpers <jonatan@werpers.com>
parents:
1332
diff
changeset
|
33 coordinate_size(g::Grid) = coordinate_size(typeof(g)) # TBD: Name of this function?! |
79a2193da5c1
Implement coordinate_size() and component_type() types
Jonatan Werpers <jonatan@werpers.com>
parents:
1332
diff
changeset
|
34 |
1334
47e309eac131
Add docs for coordinate_size and component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1333
diff
changeset
|
35 """ |
1475
76b190ca9a27
Generalize component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1474
diff
changeset
|
36 component_type(gf) |
1334
47e309eac131
Add docs for coordinate_size and component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1333
diff
changeset
|
37 |
1507
de7b76b61ecd
Clearify docs of `component_type`
Jonatan Werpers <jonatan@werpers.com>
parents:
1491
diff
changeset
|
38 The type of the components of the elements of `gf`. |
1334
47e309eac131
Add docs for coordinate_size and component_type
Jonatan Werpers <jonatan@werpers.com>
parents:
1333
diff
changeset
|
39 """ |
1474
276c38a48aac
Start implementing componentview
Jonatan Werpers <jonatan@werpers.com>
parents:
1471
diff
changeset
|
40 component_type(T::Type) = eltype(eltype(T)) |
276c38a48aac
Start implementing componentview
Jonatan Werpers <jonatan@werpers.com>
parents:
1471
diff
changeset
|
41 component_type(t) = component_type(typeof(t)) |
276c38a48aac
Start implementing componentview
Jonatan Werpers <jonatan@werpers.com>
parents:
1471
diff
changeset
|
42 |
1476
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
43 componentview(gf, component_index...) = ArrayComponentView(gf, component_index) |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
44 |
1491
06dec7b68f82
REVIEW: Comment on dimensionality of input array to an ArrayComponentView, and on slicing ArrayComponetViews
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1476
diff
changeset
|
45 # REVIEW: Should this only be defined for vector-valued component types of the same dimension? |
06dec7b68f82
REVIEW: Comment on dimensionality of input array to an ArrayComponentView, and on slicing ArrayComponetViews
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1476
diff
changeset
|
46 # Now one can for instance do: v = [rand(2,2),rand(2,2), rand(2,1)] and cv = componentview(v,1,2) |
06dec7b68f82
REVIEW: Comment on dimensionality of input array to an ArrayComponentView, and on slicing ArrayComponetViews
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1476
diff
changeset
|
47 # resulting in #undef in the third component of cv. |
1508
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
48 # RESPONSE: I don't think it's possible to stop the user from |
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
49 # doing stupid things. My inclination is to just keep it simple and let the |
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
50 # user read the error messages they get. |
1476
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
51 struct ArrayComponentView{CT,T,D,AT <: AbstractArray{T,D}, IT} <: AbstractArray{CT,D} |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
52 v::AT |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
53 component_index::IT |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
54 |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
55 function ArrayComponentView(v, component_index) |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
56 CT = typeof(first(v)[component_index...]) |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
57 return new{CT, eltype(v), ndims(v), typeof(v), typeof(component_index)}(v,component_index) |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
58 end |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
59 end |
1474
276c38a48aac
Start implementing componentview
Jonatan Werpers <jonatan@werpers.com>
parents:
1471
diff
changeset
|
60 |
1476
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
61 Base.size(cv) = size(cv.v) |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
62 Base.getindex(cv::ArrayComponentView, i::Int) = cv.v[i][cv.component_index...] |
1491
06dec7b68f82
REVIEW: Comment on dimensionality of input array to an ArrayComponentView, and on slicing ArrayComponetViews
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1476
diff
changeset
|
63 Base.getindex(cv::ArrayComponentView, I::Vararg{Int}) = cv.v[I...][cv.component_index...] #REVIEW: Will this allocate if I... slices v? if so, we should probably use a view on v? |
1508
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
64 # RESPONSE: I imagine the values of `cv` will be small static vectors most of |
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
65 # the time for which this won't be a problem. I say we cross that bridge when |
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
66 # there is an obvoius need. (Just slapping a @view on there seems to be |
6309ca375dcd
Add responses to review comments
Jonatan Werpers <jonatan@werpers.com>
parents:
1507
diff
changeset
|
67 # changing the return tyoe to a 0-dimensional array. That's where i gave up.) |
1476
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
68 IndexStyle(::Type{<:ArrayComponentView{<:Any,<:Any,AT}}) where AT = IndexStyle(AT) |
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
69 |
1516
8d64f8981bb0
Remove comments and add todo
Jonatan Werpers <jonatan@werpers.com>
parents:
1508
diff
changeset
|
70 # TODO: Implement setindex!? |
1476
62f9d0387a2a
Introduce a view type and implement sliced indexing for the components.
Jonatan Werpers <jonatan@werpers.com>
parents:
1475
diff
changeset
|
71 # TODO: Implement a more general ComponentView that can handle non-AbstractArrays. |
1222 | 72 |
1264
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
73 """ |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
74 refine(g::Grid, r) |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
75 |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
76 The grid where `g` is refined by the factor `r`. |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
77 |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
78 See also: [`coarsen`](@ref). |
1264
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
79 """ |
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
80 function refine end |
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
81 |
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
82 """ |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
83 coarsen(g::Grid, r) |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
84 |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
85 The grid where `g` is coarsened by the factor `r`. |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
86 |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
87 See also: [`refine`](@ref). |
1264
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
88 """ |
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
89 function coarsen end |
1222 | 90 |
212
aa17d4d9d09e
Export some functions from AbstractGrid and move documentation to docstrings
Jonatan Werpers <jonatan@werpers.com>
parents:
211
diff
changeset
|
91 """ |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
92 boundary_identifiers(g::Grid) |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
93 |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
94 Identifiers for all the boundaries of `g`. |
1222 | 95 """ |
1233
3924c1f6ec6d
Fix empty function defs
Jonatan Werpers <jonatan@werpers.com>
parents:
1222
diff
changeset
|
96 function boundary_identifiers end |
1264
7a67935d3f3a
grid.jl: Fix function defs and add todos for documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1262
diff
changeset
|
97 |
1222 | 98 """ |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
99 boundary_grid(g::Grid, id::BoundaryIdentifier) |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
100 |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
101 The grid for the boundary specified by `id`. |
1222 | 102 """ |
1233
3924c1f6ec6d
Fix empty function defs
Jonatan Werpers <jonatan@werpers.com>
parents:
1222
diff
changeset
|
103 function boundary_grid end |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
104 # TBD: Can we implement a version here that accepts multiple ids and grouped boundaries? Maybe we need multiblock stuff? |
1222 | 105 |
1269
20f42cf0800c
Add test for Grid and make them pass. Start implementing eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1266
diff
changeset
|
106 """ |
1466
de4e15924d26
Move def of boundary_indices, remove some new lines
Jonatan Werpers <jonatan@werpers.com>
parents:
1461
diff
changeset
|
107 boundary_indices(g::Grid, id::BoundaryIdentifier) |
de4e15924d26
Move def of boundary_indices, remove some new lines
Jonatan Werpers <jonatan@werpers.com>
parents:
1461
diff
changeset
|
108 |
1471
a863aef1f799
Fix review comment about docstring
Jonatan Werpers <jonatan@werpers.com>
parents:
1470
diff
changeset
|
109 A collection of indices corresponding to the boundary with given id. For grids |
a863aef1f799
Fix review comment about docstring
Jonatan Werpers <jonatan@werpers.com>
parents:
1470
diff
changeset
|
110 with Cartesian indexing these collections will be tuples with elements of type |
a863aef1f799
Fix review comment about docstring
Jonatan Werpers <jonatan@werpers.com>
parents:
1470
diff
changeset
|
111 ``Union{Int,Colon}``. |
a863aef1f799
Fix review comment about docstring
Jonatan Werpers <jonatan@werpers.com>
parents:
1470
diff
changeset
|
112 |
a863aef1f799
Fix review comment about docstring
Jonatan Werpers <jonatan@werpers.com>
parents:
1470
diff
changeset
|
113 When implementing this method it is expected that the returned collection can |
a863aef1f799
Fix review comment about docstring
Jonatan Werpers <jonatan@werpers.com>
parents:
1470
diff
changeset
|
114 be used to index grid functions to obtain grid functions on the boundary grid. |
1466
de4e15924d26
Move def of boundary_indices, remove some new lines
Jonatan Werpers <jonatan@werpers.com>
parents:
1461
diff
changeset
|
115 """ |
de4e15924d26
Move def of boundary_indices, remove some new lines
Jonatan Werpers <jonatan@werpers.com>
parents:
1461
diff
changeset
|
116 function boundary_indices end |
de4e15924d26
Move def of boundary_indices, remove some new lines
Jonatan Werpers <jonatan@werpers.com>
parents:
1461
diff
changeset
|
117 |
de4e15924d26
Move def of boundary_indices, remove some new lines
Jonatan Werpers <jonatan@werpers.com>
parents:
1461
diff
changeset
|
118 """ |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
119 eval_on(g::Grid, f) |
1222 | 120 |
1397
86026367a9ff
Support evaluating scalars on grids
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1347
diff
changeset
|
121 Lazy evaluation of `f` on the grid. `f` can either be on the form `f(x,y,...)` |
1330
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
122 with each coordinate as an argument, or on the form `f(x̄)` taking a |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
123 coordinate vector. |
5f05a708d730
grid.l: More documentation
Jonatan Werpers <jonatan@werpers.com>
parents:
1324
diff
changeset
|
124 |
1347
08f06bfacd5c
Fix typos and formatting of documentation
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1344
diff
changeset
|
125 For concrete array grid functions `map(f,g)` can be used instead. |
1269
20f42cf0800c
Add test for Grid and make them pass. Start implementing eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1266
diff
changeset
|
126 """ |
1343
fa3695f634de
More clean up of todos and notes
Jonatan Werpers <jonatan@werpers.com>
parents:
1342
diff
changeset
|
127 eval_on(g::Grid, f) = eval_on(g, f, Base.IteratorSize(g)) |
1273
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
128 function eval_on(g::Grid, f, ::Base.HasShape) |
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
129 if hasmethod(f, (Any,)) |
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
130 return LazyTensors.LazyFunctionArray((I...)->f(g[I...]), size(g)) |
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
131 else |
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
132 return LazyTensors.LazyFunctionArray((I...)->f(g[I...]...), size(g)) |
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
133 end |
7fab13c07412
Allow multi-argument functions in eval_on
Jonatan Werpers <jonatan@werpers.com>
parents:
1272
diff
changeset
|
134 end |
1402
2d9eb2d07802
Fix formatting and delete stray "`"
Jonatan Werpers <jonatan@werpers.com>
parents:
1401
diff
changeset
|
135 |
1397
86026367a9ff
Support evaluating scalars on grids
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1347
diff
changeset
|
136 """ |
86026367a9ff
Support evaluating scalars on grids
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1347
diff
changeset
|
137 eval_on(g::Grid, f::Number) |
86026367a9ff
Support evaluating scalars on grids
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1347
diff
changeset
|
138 |
1402
2d9eb2d07802
Fix formatting and delete stray "`"
Jonatan Werpers <jonatan@werpers.com>
parents:
1401
diff
changeset
|
139 Lazy evaluation of a scalar `f` on the grid. |
1397
86026367a9ff
Support evaluating scalars on grids
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1347
diff
changeset
|
140 """ |
86026367a9ff
Support evaluating scalars on grids
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
1347
diff
changeset
|
141 eval_on(g::Grid, f::Number) = return LazyTensors.LazyConstantArray(f, size(g)) |
1222 | 142 |
1270
dcbac783e4c1
Factor out functions for getting the type and number of components in a type
Jonatan Werpers <jonatan@werpers.com>
parents:
1269
diff
changeset
|
143 _ncomponents(::Type{<:Number}) = 1 |
dcbac783e4c1
Factor out functions for getting the type and number of components in a type
Jonatan Werpers <jonatan@werpers.com>
parents:
1269
diff
changeset
|
144 _ncomponents(T::Type{<:SVector}) = length(T) |
1461
455e6b4c8b02
Add a function declaration for boundary_indices
Jonatan Werpers <jonatan@werpers.com>
parents:
1407
diff
changeset
|
145 |
455e6b4c8b02
Add a function declaration for boundary_indices
Jonatan Werpers <jonatan@werpers.com>
parents:
1407
diff
changeset
|
146 |