Fix add option and symmetric pairwise dispensing.

Add a private `set_block_` function to ease debugging.
oist · Sep 26, 2023 · 20a139e · 20a139e
1 parent 1a7abc6
commit 20a139e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 10 deletions.
diff --git a/R/dispensing.R b/R/dispensing.R
@@ -54,11 +54,11 @@ dispense_pairs <- function (plate, set1, set2 = set1, volume, start = "A01") {
   }
 
   for (Nth in seq_along(set1)) {
-    plate <- set_block(plate, dispense_pairs_first(Nth),  set1[Nth], volume)
+    plate <- set_block(plate, dispense_pairs_first(Nth),  set1[Nth], volume, add = TRUE)
   }
 
   for (Nth in seq_along(set2)) {
-    plate <- set_block(plate, dispense_pairs_second(Nth), set2[Nth], volume)
+    plate <- set_block(plate, dispense_pairs_second(Nth), set2[Nth], volume, add = TRUE)
   }
 
   plate

diff --git a/R/plates.R b/R/plates.R
@@ -379,23 +379,32 @@ setMethod( "setWell", c("Plate", "Well", "character", "logical")
 setGeneric("set_block", function(plate, block, what, value, add = FALSE)
   standardGeneric("set_block"))
 
-#' @rdname set_block
-#' @export
-
-setMethod( "set_block", c("Plate", "character", "character", "numeric")
-         , function(plate, block, what, value, add) {
+set_block_ <-  function(plate, block, what, value, add) {
   plate$well <- rownames(plate)
   if (isTRUE(add)) {
-    oldVolumes <- plate[,what]
-    oldVolumes[is.na(oldVolumes)] <- 0
+    if(what %in% colnames(plate)) {
+      oldVolumes <- plate[,what]
+      oldVolumes[is.na(oldVolumes)] <- 0
+    } else {
+      oldVolumes <- 0
+    }
+    plate[,what] <- 0
     plate <- platetools::set_block(plate, block, what, value)
     plate[,what] <- plate[,what] + oldVolumes
   } else {
     plate <- platetools::set_block(plate, block, what, value)
   }
   plate$well <- NULL
   plate
-})
+}
+
+
+#' @rdname set_block
+#' @export
+
+setMethod( "set_block"
+         , c("Plate", "character", "character", "numeric")
+         , set_block_)
 
 #' Get reagent name
 #'