Add 'zdd_offset(A, var)' and 'zdd_onset(A, var)'

src/adiar/zdd.h

@@ -659,6 +659,20 @@ namespace adiar
                    ForwardIt end)
   { return zdd_offset(ep, A, make_generator(begin, end)); }
 
+  //////////////////////////////////////////////////////////////////////////////
+  /// \brief      Subset that do \em not include the given element.
+  ///
+  /// \param A    Family of set
+  ///
+  /// \param var  Variable to include.
+  //////////////////////////////////////////////////////////////////////////////
+  __zdd zdd_offset(const zdd &A, zdd::label_type var);
+
+  //////////////////////////////////////////////////////////////////////////////
+  /// \brief Subset that do \em not include the given element.
+  //////////////////////////////////////////////////////////////////////////////
+  __zdd zdd_offset(const exec_policy &ep, const zdd &A, zdd::label_type var);
+
   //////////////////////////////////////////////////////////////////////////////
   /// \brief      Subset that \em do include the given set of variables.
   ///
@@ -707,6 +721,20 @@ namespace adiar
                   ForwardIt end)
   { return zdd_onset(ep, A, make_generator(begin, end)); }
 
+  //////////////////////////////////////////////////////////////////////////////
+  /// \brief      Subset that \em do include the given element.
+  ///
+  /// \param A    Family of set
+  ///
+  /// \param var  Variable to include.
+  //////////////////////////////////////////////////////////////////////////////
+  __zdd zdd_onset(const zdd &A, zdd::label_type var);
+
+  //////////////////////////////////////////////////////////////////////////////
+  /// \brief Subset that \em do include the given element.
+  //////////////////////////////////////////////////////////////////////////////
+  __zdd zdd_onset(const exec_policy &ep, const zdd &A, zdd::label_type var);
+
   //////////////////////////////////////////////////////////////////////////////
   /// \brief     Project family of sets onto a domain, i.e. remove from every
   ///            set all variables not within the domain.

src/adiar/zdd/subset.cpp

@@ -8,6 +8,20 @@
 
 namespace adiar
 {
+  // TODO: Merge with the generator in `bdd_restrict(f, var, val)`.
+  inline auto make_generator(const zdd::label_type &var, bool &gen_called)
+  {
+    adiar_assert(!gen_called);
+
+    return [&gen_called, &var]() -> zdd::label_type {
+      if (gen_called) {
+        return generator_end<zdd::label_type>::value;
+      }
+      gen_called = true;
+      return var;
+    };
+  }
+
   template<assignment FIX_VALUE>
   class zdd_subset_labels
   {
@@ -135,6 +149,17 @@ namespace adiar
     return zdd_offset(exec_policy(), A, vars);
   }
 
+  __zdd zdd_offset(const exec_policy &ep, const zdd &A, zdd::label_type var)
+  {
+    bool gen_called = false;
+    return zdd_offset(ep, A, make_generator(var, gen_called));
+  }
+
+  __zdd zdd_offset(const zdd &A, zdd::label_type var)
+  {
+    return zdd_offset(exec_policy(), A, var);
+  }
+
   //////////////////////////////////////////////////////////////////////////////
   template<typename assignment_mgr>
   class zdd_onset_policy : public zdd_policy
@@ -212,4 +237,15 @@ namespace adiar
   {
     return zdd_onset(exec_policy(), A, xs);
   }
+
+  __zdd zdd_onset(const exec_policy &ep, const zdd &A, zdd::label_type var)
+  {
+    bool gen_called = false;
+    return zdd_onset(ep, A, make_generator(var, gen_called));
+  }
+
+  __zdd zdd_onset(const zdd &A, zdd::label_type var)
+  {
+    return zdd_onset(exec_policy(), A, var);
+  }
 }