Mercurial > repos > public > sbplib_julia

comparison src/SbpOperators/quadrature/quadrature.jl @ 562:8f7919a9b398 feature/boundary_ops

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge with default
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Mon, 30 Nov 2020 18:30:24 +0100
parents 011ca1639153
children
comparison
equal deleted inserted replaced
544:884be64e82d9 562:8f7919a9b398
20 end 20 end
21 21
22 LazyTensors.range_size(H::Quadrature) = getindex.(range_size.(H.H),1) 22 LazyTensors.range_size(H::Quadrature) = getindex.(range_size.(H.H),1)
23 LazyTensors.domain_size(H::Quadrature) = getindex.(domain_size.(H.H),1) 23 LazyTensors.domain_size(H::Quadrature) = getindex.(domain_size.(H.H),1)
24 24
25 function LazyTensors.apply(Q::Quadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Index,Dim}) where {T,Dim} 25 function LazyTensors.apply(Q::Quadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Any,Dim}) where {T,Dim}
26 error("not implemented") 26 error("not implemented")
27 end 27 end
28 28
29 function LazyTensors.apply(Q::Quadrature{1,T}, v::AbstractVector{T}, I::Index) where T 29 function LazyTensors.apply(Q::Quadrature{1,T}, v::AbstractVector{T}, i) where T
30 @inbounds q = apply(Q.H[1], v , I) 30 @inbounds q = apply(Q.H[1], v , i)
31 return q 31 return q
32 end 32 end
33 33
34 function LazyTensors.apply(Q::Quadrature{2,T}, v::AbstractArray{T,2}, I::Index, J::Index) where T 34 function LazyTensors.apply(Q::Quadrature{2,T}, v::AbstractArray{T,2}, i, j) where T
35 # Quadrature in x direction 35 # Quadrature in x direction
36 @inbounds vx = view(v, :, Int(J)) 36 @inbounds vx = view(v, :, Int(j))
37 @inbounds qx = apply(Q.H[1], vx , I) 37 @inbounds qx = apply(Q.H[1], vx , i)
38 # Quadrature in y-direction 38 # Quadrature in y-direction
39 @inbounds vy = view(v, Int(I), :) 39 @inbounds vy = view(v, Int(i), :)
40 @inbounds qy = apply(Q.H[2], vy, J) 40 @inbounds qy = apply(Q.H[2], vy, j)
41 return qx*qy 41 return qx*qy
42 end 42 end
43 43
44 LazyTensors.apply_transpose(Q::Quadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Index,Dim}) where {Dim,T} = LazyTensors.apply(Q,v,I...) 44 LazyTensors.apply_transpose(Q::Quadrature{Dim,T}, v::AbstractArray{T,Dim}, I::Vararg{Any,Dim}) where {Dim,T} = LazyTensors.apply(Q,v,I...)