biomechanic_eggshell upgrade

R/biomechanic_eggshell.R

@@ -79,15 +79,6 @@ setMethod("biomechanic_eggshell",definition=function(object,input,remove_samples
     # extrazct Instron results
     instron<-methods::slot(object,"tables")$results
 
-    ###################
-    # remove samples
-    if(!base::is.null(remove_samples)){
-
-      ###################
-      # remove samples
-      Data<-Data[!samples%in%remove_samples]
-    }
-
     ###################
     # add Instron results
     Data<-merge(
@@ -96,11 +87,29 @@ setMethod("biomechanic_eggshell",definition=function(object,input,remove_samples
       by="samples",all.x=T,sort=F,
       incomparables=NA
     )
-
+    
     ###################
     # ordering columns
     Data<-Data[,base::c(2,1,3:base::ncol(Data)),with=F]
 
+    ###################
+    # remove samples
+    if(!base::is.null(remove_samples)){
+      
+      ###################
+      # add NA for samples to remove
+      Data<-Data[
+        samples%in%remove_samples,
+        `:=`(
+          mm=NA,
+          F=NA,
+          Sd=NA,
+          pearson_cor=NA,
+          broken_egg=NA
+        )         
+      ]
+    }
+    
     ###################
     # add Instron results
     if("class.x"%in%base::names(Data)){