R/density_check.R

In SpatialFloor: Spatial Floor Simulation (Isotropic)

#' Internal function for density check of check layout
#'
#' This is an inernal function
#' @param layout.matrix The layout matrix of designed grid
#' @param by.density The check density in the layout matrix
#' @import reshape2
#' @export

density_check = function(layout.matrix, by.density){
  dta.tmp = reshape2::melt(layout.matrix)
  colnames(dta.tmp) = c('row.var', 'col.var', 'check')
  res = sum(dta.tmp$check==1)/nrow(dta.tmp) - by.density

  if(res > 0){
	check.id = round(abs(res)*nrow(dta.tmp))
    idc = sample(rownames(dta.tmp[dta.tmp$check==1, ]), check.id)
    dta.tmp[idc, ]$check = 0
    return(dta.tmp)
  } else if(res < 0){
	check.id = round(abs(res)*nrow(dta.tmp))
    idc = sample(rownames(dta.tmp[dta.tmp$check==0, ]), check.id)
    dta.tmp[idc, ]$check = 1
	return(dta.tmp)
  } else {
    return(dta.tmp)
  }
}