R/subGroups.R

In joeburns06/hocuspocus: A Suite of Single-Cell RNA-Seq Analysis Tools Built for Dummy Biologists

#' Select a subset of groups in phenoData
#'
#' Select a subset of specified groups in the phenoData slot of a Hocus object while retaining associated color info in colorData.
#'
#' @export


subGroups <- function(cellData,groups,names,remove=FALSE){
  
  if (remove==FALSE){
    cellData<-cellData[,row.names(pData(cellData))[pData(cellData)[,groups]%in%names]]
    
    for (i in 1:length(cellData@colorData)){
      cellData@colorData[[i]] <- cellData@colorData[[i]][as.vector(unique(pData(cellData)[,names(cellData@colorData[i])]))]
    }
  }
  
  if (remove==TRUE){
    cellData<-cellData[,row.names(pData(cellData))[!(pData(cellData)[,groups]%in%names)]]
    
    for (i in 1:length(cellData@colorData)){
      cellData@colorData[[i]] <- cellData@colorData[[i]][as.vector(unique(pData(cellData)[,names(cellData@colorData[i])]))]
    }
  }
  
  cellData
  
}