Add support for removing shapes from WF specifications (#146)

microsoft · Sep 26, 2024 · 58d4807 · 58d4807
1 parent 02fcf57
commit 58d4807
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/include/trieste/wf.h b/include/trieste/wf.h
@@ -1071,6 +1071,30 @@ namespace trieste
         wf.append(shape2);
         return wf;
       }
+
+      inline Wellformed operator-(const Wellformed& wf, const Token& token) {
+        Wellformed wf2;
+        wf2.shapes.insert(wf.shapes.begin(), wf.shapes.end());
+        wf2.shapes.erase(token);
+        return wf2;
+      }
+
+      inline Wellformed operator-(const Wellformed& wf, Token&& token) {
+        Wellformed wf2;
+        wf2.shapes.insert(wf.shapes.begin(), wf.shapes.end());
+        wf2.shapes.erase(token);
+        return wf2;
+      }
+
+      inline Wellformed operator-(Wellformed&& wf, const Token& token) {
+        wf.shapes.erase(token);
+        return std::move(wf);
+      }
+
+      inline Wellformed operator-(Wellformed&& wf, Token&& token) {
+        wf.shapes.erase(token);
+        return std::move(wf);
+      }
     }
 
     namespace detail