Merge pull request #224 from JuliaDynamics/hw/ext_namespace

hexaeder · web-flow · commit d3fd3bfc6169 · 2025-03-21T22:09:08.000+01:00
namespace extensions
diff --git a/Project.toml b/Project.toml
@@ -40,9 +40,9 @@ ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"
 Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
 
 [extensions]
-CUDAExt = ["CUDA", "Adapt"]
-MTKExt = ["ModelingToolkit", "Symbolics"]
-SymbolicsExt = ["Symbolics", "MacroTools"]
+NetworkDynamicsCUDAExt = ["CUDA", "Adapt"]
+NetworkDynamicsMTKExt = ["ModelingToolkit", "Symbolics"]
+NetworkDynamicsSymbolicsExt = ["Symbolics", "MacroTools"]
 
 [compat]
 Adapt = "4.0.4"
diff --git a/docs/make.jl b/docs/make.jl
@@ -33,7 +33,7 @@ for example in filter(contains(r".jl$"), readdir(example_dir, join=true))
     Literate.script(example, outdir; keep_comments=true)
 end
 
-mtkext = Base.get_extension(NetworkDynamics, :MTKExt)
+mtkext = Base.get_extension(NetworkDynamics, :NetworkDynamicsMTKExt)
 kwargs = (;
     root=joinpath(pkgdir(NetworkDynamics), "docs"),
     sitename="NetworkDynamics",
diff --git a/ext/MTKExt_precomp_workload.jl b/ext/MTKExt_precomp_workload.jl
diff --git a/ext/MTKExt_utils.jl b/ext/MTKExt_utils.jl
diff --git a/ext/NetworkDynamicsCUDAExt.jl b/ext/NetworkDynamicsCUDAExt.jl
@@ -1,4 +1,4 @@
-module CUDAExt
+module NetworkDynamicsCUDAExt
 using NetworkDynamics: Network, NetworkLayer, ComponentBatch,
                        KAAggregator, AggregationMap, SparseAggregator,
                        LazyGBufProvider, EagerGBufProvider, LazyGBuf,
diff --git a/ext/NetworkDynamicsMTKExt.jl b/ext/NetworkDynamicsMTKExt.jl
@@ -1,4 +1,4 @@
-module MTKExt
+module NetworkDynamicsMTKExt
 
 using ModelingToolkit: Symbolic, iscall, operation, arguments, build_function
 using ModelingToolkit: ModelingToolkit, Equation, ODESystem, Differential
@@ -13,7 +13,7 @@ using NetworkDynamics: NetworkDynamics, set_metadata!,
                        PureFeedForward, FeedForward, NoFeedForward, PureStateMap
 import NetworkDynamics: VertexModel, EdgeModel, AnnotatedSym
 
-include("MTKUtils.jl")
+include("MTKExt_utils.jl")
 
 """
     VertexModel(sys::ODESystem, inputs, outputs;
@@ -478,7 +478,7 @@ _all_rhs_symbols(eqs) = mapreduce(eq->get_variables(eq.rhs), ∪, eqs, init=Set{
 
 using PrecompileTools: @compile_workload
 @compile_workload begin
-    include("precompile_workload.jl")
+    include("MTKExt_precomp_workload.jl")
 end
 
 end
diff --git a/ext/NetworkDynamicsSymbolicsExt.jl b/ext/NetworkDynamicsSymbolicsExt.jl
@@ -1,4 +1,4 @@
-module SymbolicsExt
+module NetworkDynamicsSymbolicsExt
 
 using Symbolics: Symbolics, @variables, Num
 using MacroTools: postwalk, @capture
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -55,7 +55,7 @@ haskey(ENV, "BUILDKITE") && @test CUDA.functional() # fail early in buildkite if
 
     # check on the precompile files
     @safetestset "Precompile workload" begin include("../src/precompile_workload.jl") end
-    @safetestset "MTK precompile workload" begin include("../ext/precompile_workload.jl") end
+    @safetestset "MTK precompile workload" begin include("../ext/MTKExt_precomp_workload.jl") end
 end
 
 @testset "Test Doc Examples" begin

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module CUDAExt`
	`1`	`+module NetworkDynamicsCUDAExt`
`2`	`2`	`using NetworkDynamics: Network, NetworkLayer, ComponentBatch,`
`3`	`3`	`KAAggregator, AggregationMap, SparseAggregator,`
`4`	`4`	`LazyGBufProvider, EagerGBufProvider, LazyGBuf,`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module SymbolicsExt`
	`1`	`+module NetworkDynamicsSymbolicsExt`
`2`	`2`
`3`	`3`	`using Symbolics: Symbolics, @variables, Num`
`4`	`4`	`using MacroTools: postwalk, @capture`