R/plottrans.tSNE.R

In jianan/qtlpvl: Test of Pleiotropy vs. Close Linkage of QTL

##' tsne plot for a transband.
##'
##' @inheritParams plottrans.PCA
##' @export
plottrans.tSNE <- function(Y, geno, nonrecomb, ...){

  SNE <- Rtsne::Rtsne(Y)$Y # Run TSNE

  blue <- rgb(123, 104, 238, maxColorValue = 256)
  orange <- rgb(230, 159, 0, maxColorValue = 256)
  green <- rgb(27, 159, 120, maxColorValue = 256)
  yellow <- rgb(255, 255, 0, maxColorValue = 256)
  genecolor <- c(blue, orange, green, yellow)

  Class <- geno
  Class[-nonrecomb] <- 4
  xlim <- range(SNE[, 1])
  ylim <- range(SNE[, 2])
  px <- pretty(xlim)
  py <- pretty(ylim)
  broman::grayplot(x=SNE[,1],y=SNE[,2],
                   pch=21,xat=px,yat=py,col="black",bg=genecolor[Class],
                   hlines=py, vlines=px,
                   xaxt="n", yaxt="n",
                   xaxs="r", yaxs="r",
                   xlim=xlim, ylim=ylim,
                   xlab="tSNE-x", ylab="tSNE-y",
                   mgp=c(1.6,0.2,0), cex=0.8, ...)

  u <- par("usr")
  x <- u[1] + diff(u[1:2])*((2:5)*0.1+0.05)
  points(x, rep(u[4]+diff(u[3:4])*0.035, 4), pch=21, col="black",
         bg=genecolor, xpd=TRUE, cex=0.8)
  x <- u[1] + diff(u[1:2])*((2:5)*0.1+0.05)
  text(x, rep(u[4]+diff(u[3:4])*0.035, 3), c("BB","BR","RR","Recombinants"),
       col=c(genecolor[1:3],"black"), xpd=TRUE, cex=0.8, pos=4)
}