R/resample_cifti_from_template.R

In mandymejia/ciftiTools: Tools for Reading, Writing, Viewing and Manipulating CIFTI Files

#' Resample a CIFTI from a template
#'
#' Resample a CIFTI from a template, ensuring the new CIFTI's resolution matches
#'  that of the template.
#' 
#' @param original_fname A CIFTI file to resample.
#' @param template_fname A CIFTI file to use as the template.
#' @param target_fname The file name to save the resampled CIFTI.
#'
#' @return The \code{target_fname}, invisibly
#' 
#' @family manipulating xifti
#' @export
#'
#' @section Connectome Workbench:
#' This function interfaces with the \code{"-cifti-resample"} Workbench command.
#' 
resample_cifti_from_template <- function(
  original_fname, template_fname, target_fname){
  
  # Check brainstructures.
  original_info <- info_cifti(original_fname)
  brainstructures <- original_info$cifti$brainstructures
  template_info <- info_cifti(template_fname)
  template_brainstructures <- template_info$cifti$brainstructures  
  for (b in brainstructures) {
    if (!(b %in% template_brainstructures)) {
      stop(paste(
        "The", b, 
        "brainstructure is in the original CIFTI but not in the template."
      ))
    }
  }

  # Build the Connectome Workbench command. 
  cmd <- paste(
    "-cifti-resample", 
    sys_path(original_fname),
    "COLUMN",
    sys_path(template_fname),
    "COLUMN", 
    "BARYCENTRIC",
    "CUBIC",
    sys_path(target_fname)
  )

  # Cortical spheres.
  if ("left" %in% brainstructures || "right" %in% brainstructures) {
    original_res <- infer_resolution(original_info)
    if (any(is.na(original_res))) { 
      stop("Could not infer original CIFTI's resolution from its metadata.")
    }

    resamp_res <- infer_resolution(template_info)
    if (any(is.na(resamp_res))) { 
      stop("Could not infer template CIFTI's resolution from its metadata.")
    }
    
    if ("left" %in% brainstructures) {
      if (original_res[1]==0) { stop("No left cortex data in original CIFTI.") }
      if (resamp_res[1]==0) { stop("No left cortex data in template CIFTI.") }
    }
    if ("right" %in% brainstructures) {
      if (original_res[2]==0) { stop("No left cortex data in original CIFTI.") }
      if (resamp_res[2]==0) { stop("No left cortex data in template CIFTI.") }
    }

    tdir <- tempdir()
    # Create original sphere.
    sphereL_original_fname <- format_path(paste0("sphereL_", original_res[1], ".surf.gii"), tdir, mode=2)
    sphereR_original_fname <- format_path(paste0("sphereR_", original_res[2], ".surf.gii"), tdir, mode=2)
    write_spheres(sphereL_original_fname, sphereR_original_fname, original_res)
    # Create target sphere.
    sphereL_target_fname <- format_path(paste0("sphereL_", resamp_res[1], ".surf.gii"), tdir, mode=2)
    sphereR_target_fname <- format_path(paste0("sphereR_", resamp_res[2], ".surf.gii"), tdir, mode=2)
    write_spheres(sphereL_target_fname, sphereR_target_fname, resamp_res)

    if ("left" %in% brainstructures) {
      cmd <- paste(
        cmd, "-left-spheres", 
        sys_path(sphereL_original_fname), sys_path(sphereL_target_fname)
      )
    }
    if ("right" %in% brainstructures) {
      cmd <- paste(
        cmd, "-right-spheres", 
        sys_path(sphereR_original_fname), sys_path(sphereR_target_fname)
      )
    }
  }

  # Run command.
  run_wb_cmd(cmd)

  invisible(target_fname)
}