R/rgbwt_to_rgba_int.R

In exaexa/scattermore: Scatterplots with More Points

# This file is part of scattermore.
#
# Copyright (C) 2022 Mirek Kratochvil <exa.exa@gmail.com>
#               2022-2023 Tereza Kulichova <kulichova.t@gmail.com>
#
# scattermore is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# scattermore is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with scattermore. If not, see <https://www.gnu.org/licenses/>.

#' rgbwt_to_rgba_int
#'
#' Convert a RGBWT matrix to an integer RGBA matrix.
#'
#' @param fRGBWT The RGBWT matrix.
#' @return A RGBA matrix. The output *is not premultiplied* by alpha.
#'
#' @export
#' @useDynLib scattermore, .registration=TRUE

rgbwt_to_rgba_int <- function(fRGBWT) {
  if (!is.array(fRGBWT) || dim(fRGBWT)[3] != 5) stop("not supported fRGBWT format")

  W <- 255 / pmax(scattermore.globals$epsilon, fRGBWT[, , 4])

  i32RGBA <- array(0L, c(dim(fRGBWT)[1:2], 4))
  i32RGBA[, , 1] <- as.integer(fRGBWT[, , 1] * W)
  i32RGBA[, , 2] <- as.integer(fRGBWT[, , 2] * W)
  i32RGBA[, , 3] <- as.integer(fRGBWT[, , 3] * W)
  i32RGBA[, , 4] <- as.integer(255 * (1 - fRGBWT[, , 5]))

  return(i32RGBA)
}