rebuildSPC: enforce aqp_df_class; possibly changed by aqp_df_class<- #…

…157
ncss-tech · Sep 30, 2020 · c39ecf1 · c39ecf1
1 parent 4a473fa
commit c39ecf1
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/R/Class-SoilProfileCollection.R b/R/Class-SoilProfileCollection.R
@@ -1316,6 +1316,19 @@ setMethod(f = 'aqp_df_class', signature(object = 'SoilProfileCollection'),
 
             return(u)
           })
+if (!isGeneric("aqp_df_class<-"))
+  setGeneric("aqp_df_class<-", function(object, ...)
+    standardGeneric("aqp_df_class<-")) 
+#' @param value "data.frame", "data.table" or "tibble"
+#' @rdname aqp_df_class
+setReplaceMethod("aqp_df_class", signature(object = "SoilProfileCollection"),
+                 function(object, value) {
+                   if (!value %in% c('data.frame',"data.table","tibble"))
+                     stop("aqp_df_class metadata entry must be data.frame, data.table or tibble", 
+                          call.=FALSE)
+                   metadata(object)$aqp_df_class <- value
+                   return(object)
+                 })
 
 #' Get depth units from metadata
 #'

diff --git a/R/rebuildSPC.R b/R/rebuildSPC.R
@@ -6,7 +6,7 @@ rebuildSPC <- function(x) {
   x.list <- suppressWarnings(as(x, 'list'))
 
   # seed object for new SPC
-  res <- x.list$horizons
+  res <- .as.data.frame.aqp(x.list$horizons, aqp_df_class(x))
 
   # init SPC from pieces
   # note: using depths<- because it will generate a horizon ID