Add a method to overwrite a bitset.

This copies the value from one bitset to another one.

DESCRIPTION

@@ -65,7 +65,7 @@ Suggests:
     testthat (>= 2.1.0),
     xml2,
     bench
-RoxygenNote: 7.2.3
+RoxygenNote: 7.3.1
 VignetteBuilder: knitr
 LinkingTo:
   Rcpp,

R/RcppExports.R

@@ -41,6 +41,10 @@ bitset_or <- function(a, b) {
     invisible(.Call(`_individual_bitset_or`, a, b))
 }
 
+bitset_assign <- function(a, b) {
+    invisible(.Call(`_individual_bitset_assign`, a, b))
+}
+
 bitset_xor <- function(a, b) {
     invisible(.Call(`_individual_bitset_xor`, a, b))
 }

R/bitset.R

@@ -108,6 +108,17 @@ Bitset <- list(
         self
       },
 
+      #' ```{r echo=FALSE, results="asis"}
+      #' bitset_method_doc(
+      #'   "assign",
+      #'   "overwrite the value of a bitset from another bitset",
+      #'   other = "the other bitset.")
+      #' ```
+      assign = function(other) {
+        bitset_assign(self$.bitset, other$.bitset)
+        self
+      },
+
       #' ```{r echo=FALSE, results="asis"}
       #' bitset_method_doc(
       #'   "size",

src/RcppExports.cpp

@@ -124,6 +124,17 @@ BEGIN_RCPP
     return R_NilValue;
 END_RCPP
 }
+// bitset_assign
+void bitset_assign(const Rcpp::XPtr<individual_index_t> a, const Rcpp::XPtr<individual_index_t> b);
+RcppExport SEXP _individual_bitset_assign(SEXP aSEXP, SEXP bSEXP) {
+BEGIN_RCPP
+    Rcpp::RNGScope rcpp_rngScope_gen;
+    Rcpp::traits::input_parameter< const Rcpp::XPtr<individual_index_t> >::type a(aSEXP);
+    Rcpp::traits::input_parameter< const Rcpp::XPtr<individual_index_t> >::type b(bSEXP);
+    bitset_assign(a, b);
+    return R_NilValue;
+END_RCPP
+}
 // bitset_xor
 void bitset_xor(const Rcpp::XPtr<individual_index_t> a, const Rcpp::XPtr<individual_index_t> b);
 RcppExport SEXP _individual_bitset_xor(SEXP aSEXP, SEXP bSEXP) {
@@ -1464,6 +1475,7 @@ static const R_CallMethodDef CallEntries[] = {
     {"_individual_bitset_and", (DL_FUNC) &_individual_bitset_and, 2},
     {"_individual_bitset_not", (DL_FUNC) &_individual_bitset_not, 2},
     {"_individual_bitset_or", (DL_FUNC) &_individual_bitset_or, 2},
+    {"_individual_bitset_assign", (DL_FUNC) &_individual_bitset_assign, 2},
     {"_individual_bitset_xor", (DL_FUNC) &_individual_bitset_xor, 2},
     {"_individual_bitset_set_difference", (DL_FUNC) &_individual_bitset_set_difference, 2},
     {"_individual_bitset_sample", (DL_FUNC) &_individual_bitset_sample, 2},

src/bitset.cpp

@@ -88,6 +88,17 @@ void bitset_or(
     (*a) |= (*b);
 }
 
+//[[Rcpp::export]]
+void bitset_assign(
+    const Rcpp::XPtr<individual_index_t> a,
+    const Rcpp::XPtr<individual_index_t> b
+    ) {
+    if (a->max_size() != b->max_size()) {
+        Rcpp::stop("Incompatible bitmap sizes");
+    }
+    (*a) = (*b);
+}
+
 //[[Rcpp::export]]
 void bitset_xor(
     const Rcpp::XPtr<individual_index_t> a,

tests/testthat/test-bitset.R

@@ -80,6 +80,31 @@ test_that("bitset or works", {
   expect_equal(a$to_vector(), c(1, 3, 5, 6, 7))
 })
 
+test_that("bitset assign works", {
+  a <- Bitset$new(10)
+  a$insert(c(1, 5, 6))
+  b <- Bitset$new(10)
+  b$insert(c(1, 3, 7))
+
+  a$assign(b)
+  expect_equal(a$to_vector(), c(1, 3, 7))
+
+  # Check the two bitsets aren't aliases of each other.
+  b$clear()
+  expect_equal(a$to_vector(), c(1, 3, 7))
+  expect_equal(b$to_vector(), numeric(0))
+})
+
+test_that("bitset assign requires bitsets to have same max size", {
+  a <- Bitset$new(8)
+  a$insert(c(1, 5, 6))
+  b <- Bitset$new(10)
+  b$insert(c(1, 3, 7))
+
+  expect_error(a$assign(b), "Incompatible bitmap sizes")
+})
+
+
 test_that("bitset set difference works for sets with intersection", {
 
   a <- Bitset$new(20)