added function clear\!

bmad-sim · Mar 29, 2024 · bd844d4 · bd844d4
1 parent ea3f35e
commit bd844d4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/GTPSA.jl b/src/GTPSA.jl
@@ -118,6 +118,7 @@ export
  scalar,
  setGTPSA!,
  getGTPSA,
+ clear!,
 
  # Temporaries:
  @FastGTPSA,

diff --git a/src/operators.jl b/src/operators.jl
@@ -1,3 +1,12 @@
+# -- clear --
+function clear!(t::TPS)
+ mad_tpsa_clear!(t.tpsa)
+end
+
+function clear!(t::ComplexTPS)
+ mad_ctpsa_clear!(t.tpsa)
+end
+
 # -- zero -- 
 function zero(t::TPS)::TPS
  return TPS(mad_tpsa_new(t.tpsa, MAD_TPSA_SAME))