Properly extend methods

trixi-framework · Nov 3, 2023 · fb91bd9 · fb91bd9
1 parent 5c663fc
commit fb91bd9
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 19 deletions.
diff --git a/src/TrixiSmartShockFinder.jl b/src/TrixiSmartShockFinder.jl
@@ -3,7 +3,7 @@ module TrixiSmartShockFinder
 using MuladdMacro: @muladd
 using Trixi
 using Trixi: AbstractIndicator, AbstractEquations, AbstractSemidiscretization, @threaded,
-             trixi_include
+             summary_box, trixi_include
 
 include("indicators.jl")
 include("indicators_1d.jl")

diff --git a/src/indicators_1d.jl b/src/indicators_1d.jl
@@ -7,14 +7,14 @@
 
 # this method is used when the indicator is constructed as for shock-capturing volume integrals
 # empty cache is default
-function create_cache(::Type{<:IndicatorNeuralNetwork},
-                      equations::AbstractEquations{1}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{<:IndicatorNeuralNetwork},
+                            equations::AbstractEquations{1}, basis::LobattoLegendreBasis)
     return NamedTuple()
 end
 
 # cache for NeuralNetworkPerssonPeraire-type indicator
-function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}},
-                      equations::AbstractEquations{1}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}},
+                            equations::AbstractEquations{1}, basis::LobattoLegendreBasis)
     alpha = Vector{real(basis)}()
     alpha_tmp = similar(alpha)
     A = Array{real(basis), ndims(equations)}
@@ -27,8 +27,8 @@ function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}
 end
 
 # cache for NeuralNetworkRayHesthaven-type indicator
-function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}},
-                      equations::AbstractEquations{1}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}},
+                            equations::AbstractEquations{1}, basis::LobattoLegendreBasis)
     alpha = Vector{real(basis)}()
     alpha_tmp = similar(alpha)
     A = Array{real(basis), ndims(equations)}
@@ -41,8 +41,8 @@ function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}},
 end
 
 # this method is used when the indicator is constructed as for AMR
-function create_cache(typ::Type{<:IndicatorNeuralNetwork},
-                      mesh, equations::AbstractEquations{1}, dg::DGSEM, cache)
+function Trixi.create_cache(typ::Type{<:IndicatorNeuralNetwork},
+                            mesh, equations::AbstractEquations{1}, dg::DGSEM, cache)
     create_cache(typ, equations, dg.basis)
 end
 

diff --git a/src/indicators_2d.jl b/src/indicators_2d.jl
@@ -7,14 +7,14 @@
 
 # this method is used when the indicator is constructed as for shock-capturing volume integrals
 # empty cache is default
-function create_cache(::Type{IndicatorNeuralNetwork},
-                      equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{IndicatorNeuralNetwork},
+                            equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
     return NamedTuple()
 end
 
 # cache for NeuralNetworkPerssonPeraire-type indicator
-function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}},
-                      equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}},
+                            equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
     alpha = Vector{real(basis)}()
     alpha_tmp = similar(alpha)
     A = Array{real(basis), ndims(equations)}
@@ -30,8 +30,8 @@ function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}
 end
 
 # cache for NeuralNetworkRayHesthaven-type indicator
-function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}},
-                      equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}},
+                            equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
     alpha = Vector{real(basis)}()
     alpha_tmp = similar(alpha)
     A = Array{real(basis), ndims(equations)}
@@ -50,8 +50,8 @@ function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}},
 end
 
 # cache for NeuralNetworkCNN-type indicator
-function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkCNN}},
-                      equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
+function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkCNN}},
+                            equations::AbstractEquations{2}, basis::LobattoLegendreBasis)
     alpha = Vector{real(basis)}()
     alpha_tmp = similar(alpha)
     A = Array{real(basis), ndims(equations)}
@@ -69,8 +69,8 @@ function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkCNN}},
 end
 
 # this method is used when the indicator is constructed as for AMR
-function create_cache(typ::Type{<:IndicatorNeuralNetwork},
-                      mesh, equations::AbstractEquations{2}, dg::DGSEM, cache)
+function Trixi.create_cache(typ::Type{<:IndicatorNeuralNetwork},
+                            mesh, equations::AbstractEquations{2}, dg::DGSEM, cache)
     create_cache(typ, equations, dg.basis)
 end