R/hasGenotype.R

In nprcgenekeepr: Genetic Tools for Colony Management

#' Check for genotype data in dataframe
#'
## Copyright(c) 2017-2024 R. Mark Sharp
## This file is part of nprcgenekeepr
#' Checks to ensure the content and structure are appropriate for genotype
#' data are in the dataframe and ready for the \code{geneDrop} function by
#' already being mapped to integers and placed in columns named \code{first}
#' and \code{second}. These checks are simply based on expected columns
#' and legal domains.
#'
#' @return A logical value representing whether or not the data.frame passed in
#' contains genotypic data that can be used. Non-standard column names are
#' accepted for this assessment.
#'
#' @param genotype dataframe with genotype data
#' @importFrom stringi stri_detect_fixed
#' @export
#' @examples
#' library(nprcgenekeepr)
#' rhesusPedigree <- nprcgenekeepr::rhesusPedigree
#' rhesusGenotypes <- nprcgenekeepr::rhesusGenotypes
#' pedWithGenotypes <- addGenotype(
#'   ped = rhesusPedigree,
#'   genotype = rhesusGenotypes
#' )
#' hasGenotype(pedWithGenotypes)
hasGenotype <- function(genotype) {
  cols <- names(genotype)
  if (length(cols) < 3L) {
    FALSE # "Genotype file must have at least three columns
  } else if (!any(stri_detect_fixed(tolower(cols), "id"))) {
    FALSE # "Genotype must have 'id' as a column.")
  } else if (!any(tolower(cols) == "first")) {
    FALSE # "Genotype must have a column named 'first'
  } else if (!any(tolower(cols) == "second")) {
    FALSE # "Genotype  must have a column named 'second'
  } else {
    if (!any(is.numeric(genotype$first))) {
      FALSE # genotype representation (indirection) should be integer
      # at this point
    } else if (!any(is.numeric(genotype$second))) {
      FALSE # genotype representation (indirection) should be integer
    } else {
      TRUE
    }
  }
}