...
[iramuteq] / Rscripts / simi.R
1 #from proxy package
2 #############################################################
3 #a, b, c, and d are the counts of all (TRUE, TRUE), (TRUE, FALSE), (FALSE, TRUE), and (FALSE, FALSE) 
4 # n <- a + b + c + d = nrow(x)
5
6 make.a <- function(x) {
7     a  <- t(x) %*% (x)
8     a
9 }
10
11 make.b <- function(x) {
12     b <- t(x) %*% (1 - x)
13     b
14 }
15
16 make.c <- function(x) {
17     c <- (1-t(x)) %*% x
18     c
19 }
20
21 make.d <- function(x, a, b, c) {
22 #??????????? ncol ?
23     d <- ncol(x) - a - b - c
24     d
25 }
26
27 ###########################################
28 #x, a
29 ###########################################
30 my.jaccard <- function(x) {
31     a <- make.a(x)
32     b <- make.b(x)
33     c <- make.c(x)
34     d <- make.d(x, a, b, c)
35     jac <- a / (a + b + c)
36     jac
37 }
38
39
40 prcooc <- function(x, a) {
41     prc <- (a / nrow(x)) 
42     prc
43 }
44
45 make.bin <- function(cs, a, i, j, nb) {
46     if (a[i, j] >= 1) {
47         ab <- a[i, j] - 1 
48         res <- binom.test(ab, nb, (cs[i]/nb) * (cs[j]/nb), "less")
49     } else {
50         res <- NULL
51         res$p.value <- 0
52     }
53     #res <- binom.test(ab, nb, (cs[i]/nb) * (cs[j]/nb), "less")
54     res$p.value
55     }
56
57 binom.sim <- function(x) {
58     a <- make.a(x)
59     n <- nrow(x)
60     cs <- colSums(x)
61     mat <- matrix(0,ncol(x),ncol(x))
62     colnames(mat)<-colnames(a)
63     rownames(mat)<-rownames(a)
64     for (i in 1:(ncol(x)-1)) {
65         for (j in (i+1):ncol(x)) {
66             mat[j,i] <- make.bin(cs, a, i, j , n)
67         }
68     }
69 #    print(mat)
70     mat
71 }
72
73
74 ############################################
75 # a, b, c
76 ############################################
77 # jaccard a, b, c   a / (a + b + c)
78 # Kulczynski1 a, b, c   a / (b + c)
79 # Kulczynski2 a, b, c   [a / (a + b) + a / (a + c)] / 2
80 # Mountford a, b, c    2a / (ab + ac + 2bc)
81 # Fager, McGowan a, b, c   a / sqrt((a + b)(a + c)) - 1 / 2 sqrt(a + c)
82 # Russel, Rao a (a/n)
83 # Dice, Czekanowski, Sorensen a, b, c   2a / (2a + b + c)
84 # Mozley, Margalef a, b, c  an / (a + b)(a + c)
85 # Ochiai a, b, c  a / sqrt[(a + b)(a + c)]
86 # Simpson a, b, c   a / min{(a + b), (a + c)}
87 # Braun-Blanquet a, b, c  a / max{(a + b), (a + c)}
88
89 #simple matching, Sokal/Michener a, b, c, d, ((a + d) /n)
90 # Hamman, a, b, c, d, ([a + d] - [b + c]) / n
91 # Faith , a, b, c, d, (a + d/2) / n
92 # Tanimoto, Rogers a, b, c, d, (a + d) / (a + 2b + 2c + d)
93 # Phi  a, b, c, d   (ad - bc) / sqrt[(a + b)(c + d)(a + c)(b + d)]
94 # Stiles a, b, c, d  log(n(|ad-bc| - 0.5n)^2 / [(a + b)(c + d)(a + c)(b + d)])
95 # Michael   a, b, c, d   4(ad - bc) / [(a + d)^2 + (b + c)^2]
96 # Yule a, b, c, d  (ad - bc) / (ad + bc)
97 # Yule2  a, b, c, d  (sqrt(ad) - sqrt(bc)) / (sqrt(ad) + sqrt(bc))
98
99 BuildProf01<-function(x,classes) {
100         #x : donnees en 0/1
101         #classes : classes de chaque lignes de x
102         dm<-cbind(x,cl=classes)
103         clnb=length(summary(as.data.frame(as.character(classes)),max=100))
104         print(clnb)
105         print(summary(as.data.frame(as.character(classes)),max=100))
106         mat<-matrix(0,ncol(x),clnb)
107         rownames(mat)<-colnames(x)
108         for (i in 1:clnb) {
109                 dtmp<-dm[which(dm$cl==i),]
110                 for (j in 1:(ncol(dtmp)-1)) {
111                         mat[j,i]<-sum(dtmp[,j])
112                 }
113         }
114         mat
115 }
116
117 do.simi <- function(x, method = 'cooc',seuil = NULL, p.type = 'tkplot',layout.type = 'frutch', max.tree = TRUE, coeff.vertex=NULL, coeff.edge = NULL, minmaxeff=c(NULL,NULL), vcexminmax= c(NULL,NULL), cex = 1, coords = NULL, communities = NULL, halo = FALSE) {
118         mat.simi <- x$mat
119     mat.eff <- x$eff
120     v.label <- colnames(mat.simi)
121         g1<-graph.adjacency(mat.simi,mode="lower",weighted=TRUE)
122         g.toplot<-g1
123         weori<-get.edge.attribute(g1,'weight')
124         if (max.tree) {
125         if (method == 'cooc') {
126                     invw <- 1 / weori
127         } else {
128             invw <- 1 - weori
129         }
130                 E(g1)$weight<-invw
131                 g.max<-minimum.spanning.tree(g1)
132         if (method == 'cooc') {
133                     E(g.max)$weight<-1 / E(g.max)$weight
134         } else {
135             E(g.max)$weight<-1 - E(g.max)$weight
136         }
137                 g.toplot<-g.max
138         }
139
140     if (!is.null(seuil)) {
141         if (seuil >= max(mat.simi)) seuil <- 0
142         vec<-vector()
143         w<-E(g.toplot)$weight
144         tovire <- which(w<=seuil)
145         g.toplot <- delete.edges(g.toplot,(tovire))
146         for (i in 0:(length(V(g.toplot)))) {
147             if (length(neighbors(g.toplot,i))==0) {
148                 vec<-append(vec,i)
149             }
150         }
151         g.toplot <- delete.vertices(g.toplot,vec)
152         v.label <- V(g.toplot)$name
153         if (!is.logical(vec)) mat.eff <- mat.eff[-(vec)]
154     }
155
156         if (!is.null(minmaxeff[1])) {
157         eff<-norm.vec(mat.eff,minmaxeff[1],minmaxeff[2])
158     } else {
159         eff<-coeff.vertex
160     }
161     if (!is.null(vcexminmax[1])) {
162         label.cex = norm.vec(mat.eff, vcexminmax[1], vcexminmax[2])
163     } else {
164         label.cex = cex
165     }
166     if (!is.null(coeff.edge)) {
167         we.width <- norm.vec(abs(E(g.toplot)$weight), coeff.edge[1], coeff.edge[2]) 
168             #we.width <- abs((E(g.toplot)$weight/max(abs(E(g.toplot)$weight)))*coeff.edge)
169     } else {
170         we.width <- NULL
171     }
172     if (method != 'binom') {
173         we.label <- round(E(g.toplot)$weight,1)
174     } else {
175         we.label <- round(E(g.toplot)$weight,3)
176     }
177         if (p.type=='rgl') {
178         nd<-3
179     } else {
180         nd<-2
181     }
182     if (is.null(coords)) {
183         if (layout.type == 'frutch')
184                 lo <- layout.fruchterman.reingold(g.toplot,dim=nd)#, weightsA=E(g.toplot)$weight)
185         if (layout.type == 'kawa')
186                 lo <- layout.kamada.kawai(g.toplot,dim=nd)
187         if (layout.type == 'random')
188                 lo <- layout.random(g.toplot,dim=nd)
189         if (layout.type == 'circle' & p.type != 'rgl')
190                 lo <- layout.circle(g.toplot)
191         if (layout.type == 'circle' & p.type == 'rgl')
192                 lo <- layout.sphere(g.toplot)
193         if (layout.type == 'graphopt')
194             lo <- layout.graphopt(g.toplot)
195     } else {
196         lo <- coords
197     }
198     if (!is.null(communities)) {
199         if (communities == 0 ){ #'edge.betweenness.community') {
200             com <- edge.betweenness.community(g.toplot)
201         } else if (communities == 1) {
202             com <- fastgreedy.community(g.toplot)
203         } else if (communities == 2) {
204             com <- label.propagation.community(g.toplot)
205         } else if (communities == 3) {
206             com <- leading.eigenvector.community(g.toplot)
207         } else if (communities == 4) {
208             com <- multilevel.community(g.toplot)
209         } else if (communities == 5) {
210             com <- optimal.community(g.toplot)
211         } else if (communities == 6) {
212             com <- spinglass.community(g.toplot)
213         } else if (communities == 7) {
214             com <- walktrap.community(g.toplot)
215         } 
216     } else {
217         com <- NULL
218     }
219     
220         out <- list(graph = g.toplot, mat.eff = mat.eff, eff = eff, mat = mat.simi, v.label = v.label, we.width = we.width, we.label=we.label, label.cex = label.cex, layout = lo, communities = com, halo = halo)
221 }
222         
223 plot.simi <- function(graph.simi, p.type = 'tkplot',filename=NULL, communities = NULL, vertex.col = 'red', edge.col = 'black', edge.label = TRUE, vertex.label=TRUE, vertex.label.color = 'black', vertex.label.cex= NULL, vertex.size=NULL, leg=NULL, width = 800, height = 800, alpha = 0.1, cexalpha = FALSE, movie = NULL, svg = FALSE) {
224         mat.simi <- graph.simi$mat
225         g.toplot <- graph.simi$graph
226     if (is.null(vertex.size)) {
227         vertex.size <- graph.simi$eff
228     } else {
229         vertex.size <- vertex.size
230     }
231         we.width <- graph.simi$we.width
232     if (vertex.label) {
233         #v.label <- vire.nonascii(graph.simi$v.label)
234         v.label <- graph.simi$v.label
235     } else {
236         v.label <- NA
237     }
238     if (edge.label) {
239         we.label <- graph.simi$we.label
240     } else {
241         we.label <- NA
242     }
243         lo <- graph.simi$layout
244     if (!is.null(vertex.label.cex)) {
245         label.cex<-vertex.label.cex
246     } else {
247         label.cex = graph.simi$label.cex
248     }
249     if (cexalpha) {
250         alphas <- norm.vec(label.cex, 0.5,1)
251         nvlc <- NULL
252         if (length(vertex.label.color) == 1) {
253             for (i in 1:length(alphas)) {
254              nvlc <- append(nvlc, adjustcolor(vertex.label.color, alpha=alphas[i]))
255             }
256         } else {
257             for (i in 1:length(alphas)) {
258                 nvlc <- append(nvlc, adjustcolor(vertex.label.color[i], alpha=alphas[i]))
259             }
260         }
261         vertex.label.color <- nvlc  
262     }
263     if (p.type=='nplot') {
264         #print('ATTENTION - PAS OPEN FILE')
265         open_file_graph(filename, width = width, height = height, svg = svg)
266         par(mar=c(2,2,2,2))
267         if (!is.null(leg)) {
268             layout(matrix(c(1,2),1,2, byrow=TRUE),widths=c(3,lcm(7)))
269             par(mar=c(2,2,1,0))
270         }
271         par(pch=' ')
272         if (is.null(graph.simi$com)) {
273             plot(g.toplot,vertex.label='', edge.width=we.width, vertex.size=vertex.size, vertex.color=vertex.col, vertex.label.color='white', edge.label=we.label, edge.label.cex=cex, edge.color=edge.col, vertex.label.cex = 0, layout=lo)#, rescale = FALSE)
274         } else {
275             if (graph.simi$halo) {
276                 mark.groups <- communities(graph.simi$com)
277             } else {
278                 mark.groups <- NULL
279             }
280             plot(com, g.toplot,vertex.label='', edge.width=we.width, vertex.size=vertex.size, vertex.color=vertex.col, vertex.label.color='white', edge.label=we.label, edge.label.cex=cex, edge.color=edge.col, vertex.label.cex = 0, layout=lo, mark.groups = mark.groups)
281         }
282         #txt.layout <- lo
283         txt.layout <- layout.norm(lo, -1, 1, -1, 1, -1, 1)
284         #txt.layout <- txt.layout[order(label.cex),]
285         #vertex.label.color <- vertex.label.color[order(label.cex)]
286         #v.label <- v.label[order(label.cex)]
287         #label.cex <- label.cex[order(label.cex)]
288         text(txt.layout[,1], txt.layout[,2], v.label, cex=label.cex, col=vertex.label.color)
289         if (!is.null(leg)) {
290             par(mar=c(0,0,0,0))
291             plot(0, axes = FALSE, pch = '')
292             legend(x = 'center' , leg$unetoile, fill = leg$gcol)
293         }
294         dev.off()
295         return(lo)
296     }
297         if (p.type=='tkplot') {
298                 id <- tkplot(g.toplot,vertex.label=v.label, edge.width=we.width, vertex.size=vertex.size, vertex.color=vertex.col, vertex.label.color=vertex.label.color, edge.label=we.label, edge.color=edge.col, layout=lo)
299         coords = tkplot.getcoords(id)
300         ok <- try(coords <- tkplot.getcoords(id), TRUE)
301                 while (is.matrix(ok)) {
302             ok <- try(coords <- tkplot.getcoords(id), TRUE)
303                         Sys.sleep(0.5)
304         }
305         tkplot.off()
306     return(coords)
307         }
308         
309         if (p.type == 'rgl') {
310                 library('rgl')
311         #rgl.open()
312         #par3d(cex=0.8)
313         lo <- layout.norm(lo, -10, 10, -10, 10, -10, 10)
314                 rglplot(g.toplot,vertex.label='', edge.width=we.width/10, vertex.size=0.01, vertex.color=vertex.col, vertex.label.color="black", edge.color = edge.col, layout=lo, rescale = FALSE)
315         #los <- layout.norm(lo, -1, 1, -1, 1, -1, 1)
316                 text3d(lo[,1], lo[,2], lo[,3], vire.nonascii(v.label), col = vertex.label.color, alpha = 1, cex = vertex.label.cex)
317         rgl.spheres(lo, col = vertex.col, radius = vertex.size/100, alpha = alpha)
318         rgl.bg(color = c('white','black'))
319         if (!is.null(movie)) {
320             require(tcltk)
321             ReturnVal <- tkmessageBox(title="RGL 3 D",message="Cliquez pour commencer le film",icon="info",type="ok")
322
323             movie3d(spin3d(axis=c(0,1,0),rpm=6), movie = 'film_graph', frames = "tmpfilm", duration=10, clean=TRUE, top = TRUE, dir = movie)
324             ReturnVal <- tkmessageBox(title="RGL 3 D",message="Film fini !",icon="info",type="ok")
325         }
326         #play3d(spin3d(axis=c(0,1,0),rpm=6))
327         require(tcltk)
328         ReturnVal <- tkmessageBox(title="RGL 3 D",message="Cliquez pour fermer",icon="info",type="ok")
329         rgl.close()
330         #       while (rgl.cur() != 0)
331         #               Sys.sleep(0.5)
332         }
333 }
334
335
336 graph.word <- function(mat.simi, index) {
337     nm <- matrix(0, ncol = ncol(mat.simi), nrow=nrow(mat.simi), dimnames=list(row.names(mat.simi), colnames(mat.simi)))
338     nm[,index] <- mat.simi[,index]
339     nm[index,] <- mat.simi[index,]
340     nm
341 }