Extend show for NLS backends (#216)

JuliaSmoothOptimizers · Mar 1, 2024 · a0c1978 · a0c1978
1 parent 7778da6
commit a0c1978
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 18 deletions.
diff --git a/src/ad.jl b/src/ad.jl
@@ -48,24 +48,6 @@ struct ADModelBackend{GB, HvB, JvB, JtvB, JB, HB, GHJ, HvBLS, JvBLS, JtvBLS, JBL
   hessian_residual_backend::HBLS
 end
 
-function Base.show(
-  io::IO,
-  backend::ADModelBackend{GB, HvB, JvB, JtvB, JB, HB, GHJ},
-) where {GB, HvB, JvB, JtvB, JB, HB, GHJ}
-  print(io, replace(replace(
-    "ADModelBackend{
-  $GB,
-  $HvB,
-  $JvB,
-  $JtvB,
-  $JB,
-  $HB,
-  $GHJ,
-}",
-    "ADNLPModels." => "",
-  ), r"\{(.+)\}" => s""))
-end
-
 function ADModelBackend(
   nvar::Integer,
   f;

diff --git a/src/ad_api.jl b/src/ad_api.jl
@@ -6,6 +6,47 @@ abstract type InPlaceADbackend <: ADBackend end
 struct EmptyADbackend <: ADBackend end
 EmptyADbackend(args...; kwargs...) = EmptyADbackend()
 
+function Base.show(
+  io::IO,
+  backend::ADModelBackend{GB, HvB, JvB, JtvB, JB, HB, GHJ, HvBLS, JvBLS, JtvBLS, JBLS, HBLS},
+) where {GB, HvB, JvB, JtvB, JB, HB, GHJ, HvBLS <: EmptyADbackend, JvBLS <: EmptyADbackend, JtvBLS <: EmptyADbackend, JBLS <: EmptyADbackend, HBLS <: EmptyADbackend}
+  print(io, replace(replace(
+    "ADModelBackend{
+  $GB,
+  $HvB,
+  $JvB,
+  $JtvB,
+  $JB,
+  $HB,
+  $GHJ,
+}",
+    "ADNLPModels." => "",
+  ), r"\{(.+)\}" => s""))
+end
+
+function Base.show(
+  io::IO,
+  backend::ADModelBackend{GB, HvB, JvB, JtvB, JB, HB, GHJ, HvBLS, JvBLS, JtvBLS, JBLS, HBLS},
+) where {GB, HvB, JvB, JtvB, JB, HB, GHJ, HvBLS, JvBLS, JtvBLS, JBLS, HBLS}
+  print(io, replace(replace(
+    "ADModelBackend{
+  $GB,
+  $HvB,
+  $JvB,
+  $JtvB,
+  $JB,
+  $HB,
+  $GHJ,
+  $HvBLS,
+  $JvBLS,
+  $JtvBLS,
+  $JBLS,
+  $HBLS,
+}",
+    "ADNLPModels." => "",
+  ), r"\{(.+)\}" => s""))
+end
+
 """
     get_nln_nnzj(::ADBackend, nvar, ncon)
     get_nln_nnzj(b::ADModelBackend, nvar, ncon)