WENO support for upwinding

Project.toml

@@ -1,7 +1,7 @@
 name = "Jutul"
 uuid = "2b460a1a-8a2b-45b2-b125-b5c536396eb9"
 authors = ["Olav Møyner <olav.moyner@gmail.com>"]
-version = "0.3.0"
+version = "0.3.1"
 
 [deps]
 AlgebraicMultigrid = "2169fc97-5a83-5252-b627-83903c6c433c"

src/Jutul.jl

@@ -107,6 +107,9 @@ module Jutul
     # Nonlinear finite-volume discretizations
     include("NFVM/NFVM.jl")
 
+    # Weighted essentially non-oscillatory (WENO) schemes
+    include("WENO/WENO.jl")
+
     # LBFGS suitable for PDE optimization
     include("LBFGS/LBFGS.jl")
     using .LBFGS

src/NFVM/decomposition.jl

@@ -128,6 +128,18 @@ function NFVMLinearDiscretization(decomp; left, right)
     return NFVMLinearDiscretization(left, right, t_l, t_r, t_mpfa)
 end
 
+function Jutul.discretization_stencil(d::NFVMLinearDiscretization, ::Cells)
+    (; left, right, T_left, T_right) = d
+    t_mpfa = d.mpfa
+
+    cells = Int[left, right]
+    for (c, trans) in t_mpfa
+        push!(cells, c)
+    end
+    unique!(cells)
+    return cells
+end
+
 function Jutul.subdiscretization(d::NFVMLinearDiscretization, subg, mapper::Jutul.FiniteVolumeGlobalMap, face)
     (; left, right, T_left, T_right) = d
     t_mpfa = d.mpfa

src/NFVM/types.jl

@@ -35,6 +35,12 @@ end
 Jutul.cell_pair(x::NFVMNonLinearDiscretization) = Jutul.cell_pair(x.ft_left)
 Jutul.cell_pair(x::NFVMLinearDiscretization) = (x.left, x.right)
 
+function Jutul.discretization_stencil(d::NFVMNonLinearDiscretization, e)
+    l = Jutul.discretization_stencil(d.ft_left, e)
+    r = Jutul.discretization_stencil(d.ft_right, e)
+    return unique!(vcat(l, r))
+end
+
 function Base.show(io::IO, ft::NFVMNonLinearDiscretization{T}) where T
     l, r = cell_pair(ft)
     print(io, "NFVMNonLinearDiscretization{$T} $(l)→$(r)")