Optional typed wrapper constructor where type parameters are provided

Weaken
AlgebraicJulia · Dec 12, 2024 · cc44137 · cc44137
1 parent ee14b2b
commit cc44137
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/syntax/TheoryInterface.jl b/src/syntax/TheoryInterface.jl
@@ -329,6 +329,9 @@ function wrapper(name::Symbol, t::GAT, mod)
       Xs = map(Ts) do s 
         :($(GlobalRef($(TheoryInterface), :impl_type))(x, $($(name)), $($(Meta.quot)(s))))
       end
+      XTs = map(zip(Ts,Xs)) do (T,X)
+        :($X <: $T || error("Mismatch $($($(Meta.quot)(T))): $($X) ⊄ $($T)"))
+      end
       esc(quote 
         # Catch any potential docs above the macro call
         const $(doctarget) = nothing
@@ -337,6 +340,14 @@ function wrapper(name::Symbol, t::GAT, mod)
         # Declare the wrapper struct
         struct $n{$(Ts...)} <: $abs
           val::Any
+          function $n{$(Ts...)}(x::Any) where {$(Ts...)}
+            $($(GlobalRef(TheoryInterface, :implements)))(x, $($name)) || error(
+              "Invalid $($($(name))) model: $x")
+            $(XTs...)
+            # CHECK THAT THE GIVEN PARAMETERS MATCH Xs
+            new{$(Ts...)}(x)
+          end
+
           function $n(x::Any)
             $($(GlobalRef(TheoryInterface, :implements)))(x, $($name)) || error(
               "Invalid $($($(name))) model: $x")

diff --git a/test/models/ModelInterface.jl b/test/models/ModelInterface.jl
@@ -262,6 +262,8 @@ ThCategory.Meta.@wrapper Cat2 <: MyAbsType
 ThCategory.Meta.@typed_wrapper TCat
 
 c = TCat(FinSetC())
+@test c == TCat{Int,Vector{Int}}(FinSetC())
+@test_throws ErrorException TCat{Bool,Symbol}(FinSetC()) # Ob: Int ⊄ Bool
 @test c isa TCat{Int, Vector{Int}}
 @test id(c, 2) == [1,2]