R/cosnn.R

In Directional: A Collection of Functions for Directional Data Analysis

################################
#### Discriminant analysis for directional data
#### using the k-NN alorithm,
#### Tsagris Michail 01/2016
#### mtsagris@yahoo.gr
################################
cosnn <- function(xnew, x, k = 5, index = FALSE, rann = FALSE) {
  xnew <- matrix(xnew, ncol = dim(x)[2])  ## makes sure xnew is a matrix
  nu <- dim(xnew)[1]
  if ( index ) {
    if (rann) {
      disa <-  t( RANN::nn2( data = x, query = xnew, k = k )$nn.idx )
    } else disa <- Rfast::dista(xnew, x, k = k, square = TRUE, index = TRUE )
  } else {
    if (rann) {
      disa <-  t( RANN::nn2(data = x, query = xnew, k = k )$nn.dists^2 )
    } else disa <- Rfast::dista(xnew, x, k = k, square = TRUE)
    disa <- Rfast::colSort( 0.5 * disa - 1)
    disa[ disa >= 1 ] <- 1
    disa[ disa <=  -1 ] <-  -1
    disa <- acos(disa)
  }
  disa
}