Skip to content

Commit

Permalink
Add helper methods
Browse files Browse the repository at this point in the history
  • Loading branch information
huiyuxie committed Aug 7, 2024
1 parent bf15bbe commit 51a12f7
Show file tree
Hide file tree
Showing 4 changed files with 22 additions and 3 deletions.
6 changes: 4 additions & 2 deletions src/TrixiGPU.jl
Original file line number Diff line number Diff line change
@@ -1,16 +1,18 @@
module TrixiGPU

# Include other packages that are used in TrixiGPU.jl
# using Reexport: @reexport

using CUDA: @cuda, CuArray, HostKernel, launch_configuration, threadIdx
using Trixi: AbstractEquations

# Include other source files
import Trixi: get_node_vars, get_node_coords, get_surface_node_vars

# Include other source files
include("function.jl")
include("auxiliary/auxiliary.jl")

# Export the public APIs
export configurator_1d, configurator_2d, configurator_3d
# export configurator_1d, configurator_2d, configurator_3d

end
3 changes: 2 additions & 1 deletion src/auxiliary/auxiliary.jl
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
include("configurator.jl")
include("configurators.jl")
include("methods.jl")
File renamed without changes.
16 changes: 16 additions & 0 deletions src/auxiliary/methods.jl
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
# Extend common helper methods from Trixi.jl

@inline function get_node_vars(u, equations, indices...)
SVector(ntuple(@inline(v->u[v, indices...]), Val(nvariables(equations))))
end

@inline function get_node_coords(x, equations, indices...)
SVector(ntuple(@inline(idx->x[idx, indices...]), Val(ndims(equations))))
end

@inline function get_surface_node_vars(u, equations, indices...)
u_ll = SVector(ntuple(@inline(v->u[1, v, indices...]), Val(nvariables(equations))))
u_rr = SVector(ntuple(@inline(v->u[2, v, indices...]), Val(nvariables(equations))))

return u_ll, u_rr
end

0 comments on commit 51a12f7

Please sign in to comment.