R/cts_3s.R

In eead-csic-eesa/fingerPro_model: Sediment Source Fingerprinting

#' CTS method for 3 sources
#'
#' Compute the CTS for the selected pairs or triplets of tracers.
#'
#' @param source Data frame containing the sediment sources from a dataset
#' @param mixture Data frame containing one of the dataset mixtures
#' @param sol Selected solution from the pairs/triplet functions
#' 
#' @return Data frame containing the CR score for each tracer.
#'
#' @export
#'
cts_3s <- function(source, mixture, sol)
{
  source <- data.matrix(source[-1])
  mixture <- data.matrix(mixture[-1])
  
  # normalize
  cols <- (ncol(source)-1)/2
  for (col in c(1:cols))
  {
    mx <- max(source[,col]+source[,cols+col])
    mn <- min(source[,col]-source[,cols+col])
    source[,col] <- ( source[,col] - mn ) / ( mx - mn )
    source[,cols+col] <- source[,cols+col] / ( mx - mn )
    mixture[,col] <- ( mixture[,col] - mn ) / ( mx - mn )
  }
  
  tracer <- colnames(source)[1:cols]
  
  err <- c()
  for (col in c(1:cols))
  {
    err[col] <- abs(source[1,col] * sol[1] + source[2,col] * sol[2] + source[3,col] * sol[3] -  mixture[1,col])
  }
  
  return(data.frame(tracer, err))
}