load rgl before graph 3d
[iramuteq] / Rscripts / Rgraph.R
index 4a23d3e..8407e31 100644 (file)
@@ -14,7 +14,7 @@
 #   dev.off()
 #}
 
-PlotDendroCut <- function(chd,filename,reso,clusternb) {   
+PlotDendroCut <- function(chd,filename,reso,clusternb) {
    h.chd <- as.hclust(chd)
    memb <- cutree(h.chd, k = clusternb)
    cent <- NULL
@@ -107,7 +107,7 @@ PlotAfc2dCoul<- function(afc,chisqrtable,filename, what='coord',col=FALSE, axeto
     classes <- classes[order(cex.par, decreasing = TRUE)]
     cex.par <- cex.par[order(cex.par, decreasing = TRUE)]
     table.out <- stopoverlap(table.in, cex.par=cex.par, xlim = c(xmin,xmax), ylim = c(ymin,ymax))
-    table.in <- table.out$toplot
+       table.in <- table.out$toplot
     notplot <- table.out$notplot
     if (! is.null(notplot)) {
         write.csv2(notplot, file = paste(filename, '_notplotted.csv', sep = ''))
@@ -183,7 +183,11 @@ overlap <- function(x1, y1, sw1, sh1, boxes) {
 }
 
 .overlap <- function(x11,y11,sw11,sh11,boxes1){
-    .Call("is_overlap",x11,y11,sw11,sh11,boxes1)
+       if (as.character(packageVersion('wordcloud')) >= '2.6') {
+               .Call("_wordcloud_is_overlap", x11,y11,sw11,sh11,boxes1)
+       } else {
+               .Call("is_overlap",x11,y11,sw11,sh11,boxes1)
+       }
 }
 ########################################################
 stopoverlap <- function(x, cex.par = NULL, xlim = NULL, ylim = NULL) {
@@ -226,7 +230,7 @@ stopoverlap <- function(x, cex.par = NULL, xlim = NULL, ylim = NULL) {
                                isOverlaped <- FALSE
                        } else {
                                if(r>sqrt(.5)){
-                                       print(paste(words[i], "could not be fit on page. It will not be plotted."))
+                                       #print(paste(words[i], "could not be fit on page. It will not be plotted."))
                     notplot <- rbind(notplot,c(words[i], x[i,1], x[i,2], size[i], i))
                                        isOverlaped <- FALSE
                                }
@@ -237,6 +241,8 @@ stopoverlap <- function(x, cex.par = NULL, xlim = NULL, ylim = NULL) {
                        }
                }
     }
+       nbnot <- nrow(notplot)
+       print(paste(nbnot, ' not plotted'))
     row.names(toplot) <- words[toplot[,4]]
     return(list(toplot = toplot, notplot = notplot))
 }
@@ -343,6 +349,19 @@ getwordcloudcoord <- function(words,freq,scale=c(4,.5),min.freq=3,max.words=Inf,
        return(toplot)
 }
 
+new_tree_tot <- function(chd) {
+       lf <- chd$list_fille
+       m <- matrix(0, ncol=2)
+       for (val in 1:length(lf)) {
+               if (! is.null(lf[[val]])) {
+                       print(c(val,lf[[val]][1]))
+                       m <- rbind(m, c(val,lf[[val]][1]))
+                       m <- rbind(m, c(val,lf[[val]][2]))
+               }
+       }
+       m[-1,]
+}
+
 make_tree_tot <- function (chd) {
        library(ape)
        lf<-chd$list_fille
@@ -582,7 +601,7 @@ del.yellow <- function(colors) {
     colors
 }
 
-make_afc_graph <- function(toplot, classes, clnb, xlab, ylab, cex.txt = NULL, leg = FALSE, cmd = FALSE, black = FALSE, xminmax=NULL, yminmax=NULL) {
+make_afc_graph <- function(toplot, classes, clnb, xlab, ylab, cex.txt = NULL, leg = FALSE, cmd = FALSE, black = FALSE, xminmax=NULL, yminmax=NULL, color=NULL) {
     
     rain <- rainbow(clnb)
     compt <- 1
@@ -607,6 +626,9 @@ make_afc_graph <- function(toplot, classes, clnb, xlab, ylab, cex.txt = NULL, le
     if (black) {
         cl.color <- 'black'
     }
+    if (!is.null(color)) {
+        cl.color <- color
+    }
        plot(toplot[,1],toplot[,2], pch='', xlab = xlab, ylab = ylab, xlim=xminmax, ylim = yminmax)
        abline(h=0, v=0, lty = 'dashed')
        if (is.null(cex.txt))
@@ -662,14 +684,14 @@ plot.dendro.prof <- function(tree, classes, chisqtable, nbbycl = 60, type.dendro
         colcloud <- rainbow(length(sum.cl))
         colcloud <- del.yellow(colcloud)
     }
-    label.ori<-tree[[2]]
+       label.ori<-tree$tip.label
     if (!is.null(lab)) {
         tree$tip.label <- lab
     } else {
-           tree[[2]]<-paste('classe ',tree[[2]])
-    }
+           tree$tip.label<-paste('classe ',tree$tip.label)
+       }
        par(mar=c(2,1,0,1))
-       plot.phylo(tree,label.offset=0, tip.col=col, type=type.dendro, direction = 'downwards', srt=90, adj = 0.5, cex = 1, y.lim=c(-0.3,tree$Nnode))
+       plot.phylo(tree,label.offset=0, tip.col=col, type=type.dendro, direction = 'downwards', srt=90, adj = 0.5, cex = 1.5, y.lim=c(-0.3,tree$Nnode))
        par(mar=c(0,0,0,0))
        d <- barplot(-sum.cl[tree.order], col=col, names.arg='', axes=FALSE, axisname=FALSE)
        text(x=d, y=(-sum.cl[tree.order]+3), label=paste(round(sum.cl[tree.order],1),'%'), cex=1)
@@ -678,7 +700,7 @@ plot.dendro.prof <- function(tree, classes, chisqtable, nbbycl = 60, type.dendro
         #wordcloud(names(lclasses[[i]]), lclasses[[i]], scale = c(1.5, 0.2), random.order=FALSE, colors = colcloud[i])
         yval <- 1.1
         plot(0,0,pch='', axes = FALSE)
-        vcex <- norm.vec(lclasses[[i]], 1, 3)
+        vcex <- norm.vec(lclasses[[i]], 2, 3)
         for (j in 1:length(lclasses[[i]])) {
             yval <- yval-(strheight( names(lclasses[[i]])[j],cex=vcex[j])+0.02)
             text(-0.9, yval, names(lclasses[[i]])[j], cex = vcex[j], col = colcloud[i], adj=0)
@@ -730,12 +752,12 @@ plot.dendro.cloud <- function(tree, classes, chisqtable, nbbycl = 60, type.dendr
         colcloud <- rainbow(length(sum.cl))
     }
     par(mar=c(0,0,0,0))
-    label.ori<-tree[[2]]
+       label.ori<-tree$tip.label
     if (!is.null(lab)) {
         tree$tip.label <- lab
     } else {
-           tree[[2]]<-paste('classe ',tree[[2]])
-    }
+           tree$tip.label<-paste('classe ',tree$tip.label)
+       }
        plot.phylo(tree,label.offset=0.1,tip.col=col, type=type.dendro)
     for (i in rev(tree.order)) {
         par(mar=c(0,0,1,0),cex=0.9)
@@ -781,11 +803,11 @@ plot.dendropr <- function(tree, classes, type.dendro="phylogram", histo=FALSE, f
         }
     }
        par(mar=c(0,0,0,0),cex=1)
-       label.ori<-tree[[2]]
+       label.ori<-tree$tip.label
     if (!is.null(lab)) {
         tree$tip.label <- lab
     } else {
-           tree[[2]]<-paste('classe ',tree[[2]])
+           tree$tip.label<-paste('classe ',tree$tip.label)
     }
        plot.phylo(tree,label.offset=0.1,tip.col=col, type=type.dendro)
     #cl.order <- as.numeric(label.ori)
@@ -844,11 +866,11 @@ plot.dendro.lex <- function(tree, to.plot, bw=FALSE, lab=NULL, lay.width=c(3,3,2
     }
     layout(matlay, widths=lay.width,TRUE)
        par(mar=c(3,0,2,4),cex=1)
-       label.ori<-tree[[2]]
+       label.ori<-tree$tip.label
     if (!is.null(lab)) {
-        tree$tip.label <- lab[tree.order]
+        tree$tip.label <- lab
     } else {
-           tree[[2]]<-paste('classe ',tree[[2]])
+           tree$tip.label<-paste('classe ',tree$tip.label)
     }
     to.plot <- matrix(to.plot[,tree.order], nrow=nrow(to.plot), dimnames=list(rownames(to.plot), colnames(to.plot)))
     if (!bw) {
@@ -900,13 +922,43 @@ plot.dendro.lex <- function(tree, to.plot, bw=FALSE, lab=NULL, lay.width=c(3,3,2
     }    
     par(mar=c(0,0,0,0))
     plot(0, axes = FALSE, pch = '')
-    legend(x = 'center' , rownames(to.plot), fill = col.bars)
+    legend(x = 'center' , rev(rownames(to.plot)), fill = rev(col.bars))
     if (!cmd) {
         dev.off()
     }
        tree[[2]]<-label.ori
 }
 
+plot.spec <- function(spec, nb.word = 20) {
+       word.to.plot <- NULL
+       word.size <- NULL
+       rno <- rownames(spec)
+       cn <- colnames(spec)
+       if (nb.word > length(rno)) {nb.word <- length(rno)}
+       for (val in 1:ncol(spec)) {
+               rn <- rno[order(spec[,val], decreasing=T)][1:nb.word]
+               score <- spec[order(spec[,val], decreasing=T),val][1:nb.word]
+               word.to.plot <- cbind(word.to.plot, rn)
+               word.size <- cbind(word.size, score)
+       }
+       mat.lay <- matrix(1:ncol(spec),nrow=1,ncol=ncol(spec))
+       layout(mat.lay)
+       for (i in 1:ncol(spec)) {
+               col <- ifelse((i %% 2) == 0, 'red', 'blue')
+               par(mar=c(0,0,1,0),cex=0.7)
+           yval <- 1.1
+           plot(0,0,pch='', axes = FALSE)
+           vcex <- norm.vec(word.size[,i], 2, 3)
+               text(-0.9, -0.5, cn[i], cex = 1, adj=0, srt=90, col='black')
+           for (j in 1:length(word.size[,i])) {
+               yval <- yval-(strheight(word.to.plot[j,i],cex=vcex[j])+0.01)
+               text(-0.9, yval, word.to.plot[j,i], cex = vcex[j], col = col, adj=0)
+           }
+       }
+
+
+}
+
 plot.alceste.graph <- function(rdata,nd=3,layout='fruke', chilim = 2) {
     load(rdata)
     if (is.null(debsup)) {
@@ -944,6 +996,7 @@ plot.alceste.graph <- function(rdata,nd=3,layout='fruke', chilim = 2) {
 
 make.simi.afc <- function(x,chitable,lim=0, alpha = 0.1, movie = NULL) {
     library(igraph)
+    library(rgl)
     chimax<-as.matrix(apply(chitable,1,max))
     chimax<-as.matrix(chimax[,1][1:nrow(x)])
     chimax<-cbind(chimax,1:nrow(x))
@@ -1136,20 +1189,24 @@ graphml.to.file <- function(graph.path) {
 }
 
 
+
 graph.to.file <- function(graph.simi, nodesfile = NULL, edgesfile = NULL, community = FALSE, color = NULL, sweight = NULL) {
        require(igraph)
        g <- graph.simi$graph
-    print(graph.simi$eff)
+    #print(graph.simi$eff)
     if (!is.null(graph.simi$eff)) {
            V(g)$weight <- graph.simi$eff
     } else {
         V(g)$weight <- graph.simi$label.cex
     }
-       V(g)$x <- graph.simi$layout[,1]
-       V(g)$y <- graph.simi$layout[,2]
-       if (ncol(graph.simi$layout) == 3) {
-               V(g)$z <- graph.simi$layout[,3]
+       layout <- layout.norm(graph.simi$layout,-10,10,-10,10,-10,10)
+       #print(layout)
+       V(g)$x <- layout[,1]
+       V(g)$y <- layout[,2]
+       if (ncol(layout) == 3) {
+               V(g)$z <- layout[,3]
        }
+    E(g)$weight <- graph.simi$we.width
        if (community) {
                member <- graph.simi$communities$membership
                col <- rainbow(max(member))
@@ -1163,7 +1220,7 @@ graph.to.file <- function(graph.simi, nodesfile = NULL, edgesfile = NULL, commun
                v.colors <- col2rgb(color)
                V(g)$r <- v.colors[1,]
                V(g)$g <- v.colors[2,]
-               V(g)$b <- v.colors[3,]          
+               V(g)$b <- v.colors[3,]
        }
        if (!is.null(sweight)) {
                V(g)$sweight <- sweight
@@ -1179,3 +1236,40 @@ graph.to.file <- function(graph.simi, nodesfile = NULL, edgesfile = NULL, commun
                df
        }
 }
+
+graph.to.file2 <- function(graph, layout, nodesfile = NULL, edgesfile = NULL, community = FALSE, color = NULL, sweight = NULL) {
+       require(igraph)
+       g <- graph
+    layout <- layout.norm(layout,-5,5,-5,5,-5,5)
+       V(g)$x <- layout[,1]
+       V(g)$y <- layout[,2]
+       if (ncol(layout) == 3) {
+               V(g)$z <- layout[,3]
+       }
+       v.colors <- col2rgb(V(g)$color)
+       V(g)$r <- v.colors[1,]
+       V(g)$g <- v.colors[2,]
+       V(g)$b <- v.colors[3,]          
+       
+       if (!is.null(sweight)) {
+               V(g)$sweight <- sweight
+       }
+       if (is.null(V(g)$weight)) {
+               if (!is.null(sweight)) {
+                       V(g)$weight <- sweight
+               } else {
+                       V(g)$weight <- 1
+               }
+       }
+       df <- get.data.frame(g, what='both')
+       if (!is.null(nodesfile)) {
+               write.table(df$vertices, nodesfile, sep='\t', row.names=FALSE)
+       }
+       if (!is.null(edgesfile)) {
+               write.table(df$edges, edgesfile, sep='\t', row.names=FALSE)
+       }
+       if (is.null(edgesfile) & is.null(nodesfile)) {
+               df
+       }
+}
+