iramuteq.org Git - iramuteq/blob - PrintRScript.py

   1 # -*- coding: utf-8 -*-
   2 #Author: Pierre Ratinaud
   3 #Copyright (c) 2008-2011 Pierre Ratinaud
   4 #Lisense: GNU/GPL
   5
   6 import tempfile
   7 from chemins import ffr
   8 import os
   9 import locale
  10 from datetime import datetime
  11 import logging
  12
  13 log = logging.getLogger('iramuteq.printRscript')
  14
  15 class PrintRScript :
  16     def __init__ (self, analyse):
  17         log.info('Rscript')
  18         self.pathout = analyse.pathout
  19         self.analyse = analyse
  20         self.parametres = analyse.parametres
  21         self.scriptout = self.pathout['temp']
  22         self.script =  u"#Script genere par IRaMuTeQ - %s" % datetime.now().ctime()
  23
  24     def add(self, txt) :
  25         self.script = '\n'.join([self.script, txt])
  26
  27     def defvar(self, name, value) :
  28         self.add(' <- '.join([name, value]))
  29
  30     def defvars(self, lvars) :
  31         for val in lvars :
  32             self.defvar(val[0],val[1])
  33
  34     def sources(self, lsources) :
  35         for source in lsources :
  36             self.add('source("%s")' % source)
  37
  38     def packages(self, lpks) :
  39         for pk in lpks :
  40             self.add('library(%s)' % pk)
  41
  42     def load(self, l) :
  43         for val in l :
  44             self.add('load("%s")' % val)
  45
  46     def write(self) :
  47         with open(self.scriptout, 'w') as f :
  48             f.write(self.script)
  49
  50
  51 class chdtxt(PrintRScript) :
  52     pass
  53
  54 def Rcolor(color) :
  55     return str(color).replace(')', ', max=255)')
  56
  57 class Alceste2(PrintRScript) :
  58     def doscript(self) :
  59         self.sources(['chdfunct'])
  60         self.load(['Rdata'])
  61         lvars = [['clnb', `self.analyse.clnb`],
  62                 ['Contout', '"%s"' % self.pathout['Contout']],
  63                 ['ContSupOut', '"%s"' % self.pathout['ContSupOut']],
  64                 ['ContEtOut', '"%s"' % self.pathout['ContEtOut']],
  65                 ['profileout', '"%s"' % self.pathout['profils.csv']],
  66                 ['antiout', '"%s"' % self.pathout['antiprofils.csv']],
  67                 ['chisqtable', '"%s"' % self.pathout['chisqtable.csv']],
  68                 ['ptable', '"%s"' % self.pathout['ptable.csv']]]
  69
  70         self.defvars(lvars)
  71
  72
  73
  74 #    txt = "clnb<-%i\n" % clnb
  75 #    txt += """
  76 #source("%s")
  77 #load("%s")
  78 #""" % (RscriptsPath['chdfunct'], DictChdTxtOut['RData'])
  79 #    txt += """
  80 #dataact<-read.csv2("%s", header = FALSE, sep = ';',quote = '\"', row.names = 1, na.strings = 'NA')
  81 #datasup<-read.csv2("%s", header = FALSE, sep = ';',quote = '\"', row.names = 1, na.strings = 'NA')
  82 #dataet<-read.csv2("%s", header = FALSE, sep = ';',quote = '\"', row.names = 1, na.strings = 'NA')
  83 #""" % (DictChdTxtOut['Contout'], DictChdTxtOut['ContSupOut'], DictChdTxtOut['ContEtOut'])
  84 #    txt += """
  85 #tablesqrpact<-BuildProf(as.matrix(dataact),n1,clnb)
  86 #tablesqrpsup<-BuildProf(as.matrix(datasup),n1,clnb)
  87 #tablesqrpet<-BuildProf(as.matrix(dataet),n1,clnb)
  88 #"""
  89 #    txt += """
  90 #PrintProfile(n1,tablesqrpact[4],tablesqrpet[4],tablesqrpact[5],tablesqrpet[5],clnb,"%s","%s",tablesqrpsup[4],tablesqrpsup[5])
  91 #""" % (DictChdTxtOut['PROFILE_OUT'], DictChdTxtOut['ANTIPRO_OUT'])
  92 #    txt += """
  93 #colnames(tablesqrpact[[2]])<-paste('classe',1:clnb,sep=' ')
  94 #colnames(tablesqrpact[[1]])<-paste('classe',1:clnb,sep=' ')
  95 #colnames(tablesqrpsup[[2]])<-paste('classe',1:clnb,sep=' ')
  96 #colnames(tablesqrpsup[[1]])<-paste('classe',1:clnb,sep=' ')
  97 #colnames(tablesqrpet[[2]])<-paste('classe',1:clnb,sep=' ')
  98 #colnames(tablesqrpet[[1]])<-paste('classe',1:clnb,sep=' ')
  99 #chistabletot<-rbind(tablesqrpact[2][[1]],tablesqrpsup[2][[1]])
 100 #chistabletot<-rbind(chistabletot,tablesqrpet[2][[1]])
 101 #ptabletot<-rbind(tablesqrpact[1][[1]],tablesqrpet[1][[1]])
 102 #"""
 103 #    txt += """
 104 #write.csv2(chistabletot,file="%s")
 105 #write.csv2(ptabletot,file="%s")
 106 #gbcluster<-n1
 107 #write.csv2(gbcluster,file="%s")
 108 #""" % (DictChdTxtOut['chisqtable'], DictChdTxtOut['ptable'], DictChdTxtOut['SbyClasseOut'])
 109 #
 110
 111
 112 def RchdTxt(DicoPath, RscriptPath, mincl, classif_mode, nbt = 9, libsvdc = False, libsvdc_path = None, R_max_mem = False):
 113     txt = """
 114     source("%s")
 115     source("%s")
 116     source("%s")
 117     source("%s")
 118     """ % (RscriptPath['CHD'], RscriptPath['chdtxt'], RscriptPath['anacor'], RscriptPath['Rgraph'])
 119     if R_max_mem :
 120         txt += """
 121     memory.limit(%i)
 122         """ % R_max_mem
 123
 124     txt += """
 125     nbt <- %i
 126     """ % nbt
 127     if libsvdc :
 128         txt += """
 129         libsvdc <- TRUE
 130         libsvdc.path <- "%s"
 131         """ % ffr(libsvdc_path)
 132     else :
 133         txt += """
 134         libsvdc <- FALSE
 135         libsvdc.path <- NULL
 136         """
 137
 138     txt +="""
 139     library(Matrix)
 140     data1 <- readMM("%s")
 141     data1 <- as(data1, "dgCMatrix")
 142     row.names(data1) <- 1:nrow(data1)
 143     """ % DicoPath['TableUc1']
 144
 145     if classif_mode == 0:
 146         txt += """
 147         data2 <- readMM("%s")
 148         data2 <- as(data2, "dgCMatrix")
 149         row.names(data2) <- 1:nrow(data2)
 150         """ % DicoPath['TableUc2']
 151     #log.info('ATTENTION ############# MODEPATATE ####################')
 152     txt += """
 153     chd1<-CHD(data1, x = nbt, mode.patate = FALSE, libsvdc = libsvdc, libsvdc.path = libsvdc.path)
 154     """
 155
 156     if classif_mode == 0:
 157         txt += """
 158     chd2<-CHD(data2, x = nbt, libsvdc = libsvdc, libsvdc.path = libsvdc.path)
 159     """
 160     else:
 161         txt += """
 162     chd2<-chd1
 163     """
 164
 165     txt += """
 166     #lecture des uce
 167     listuce1<-read.csv2("%s")
 168     """ % DicoPath['listeuce1']
 169
 170     if classif_mode == 0:
 171         txt += """
 172         listuce2<-read.csv2("%s")
 173         """ % DicoPath['listeuce2']
 174
 175     txt += """
 176 #    rm(data1)
 177     """
 178
 179     if classif_mode == 0:
 180         txt += """
 181 #        rm(data2)
 182         """
 183     txt += """
 184     chd.result <- Rchdtxt("%s",mincl=%i,classif_mode=%i, nbt = nbt)
 185     n1 <- chd.result$n1
 186     classeuce1 <- chd.result$cuce1
 187     classeuce2 <- chd.result$cuce2
 188     """ % (DicoPath['uce'], mincl, classif_mode)
 189
 190     txt += """
 191     tree.tot1 <- make_tree_tot(chd1)
 192 #    open_file_graph("%s", widt = 600, height=400)
 193 #    plot(tree.tot1$tree.cl)
 194 #    dev.off()
 195     """%DicoPath['arbre1']
 196
 197     if classif_mode == 0:
 198         txt += """
 199         tree.tot2 <- make_tree_tot(chd2)
 200 #        open_file_graph("%s", width = 600, height=400)
 201 #        plot(tree.tot2$tree.cl)
 202 #        dev.off()
 203         """ % DicoPath['arbre2']
 204
 205     txt += """
 206     tree.cut1 <- make_dendro_cut_tuple(tree.tot1$dendro_tuple, chd.result$coord_ok, classeuce1, 1, nbt)
 207     save(tree.cut1, file="%s")
 208     classes<-n1[,ncol(n1)]
 209     open_file_graph("%s", width = 600, height=400)
 210     plot.dendropr(tree.cut1$tree.cl,classes, histo=TRUE)
 211     open_file_graph("%s", width = 600, height=400)
 212     plot(tree.cut1$dendro_tot_cl)
 213     dev.off()
 214     """ % (DicoPath['Rdendro'], DicoPath['dendro1'], DicoPath['arbre1'])
 215
 216     if classif_mode == 0:
 217         txt += """
 218         tree.cut2 <- make_dendro_cut_tuple(tree.tot2$dendro_tuple, chd.result$coord_ok, classeuce2, 2, nbt)
 219         open_file_graph("%s", width = 600, height=400)
 220         plot(tree.cut2$tree.cl)
 221         dev.off()
 222         open_file_graph("%s", width = 600, height=400)
 223         plot(tree.cut1$dendro_tot_cl)
 224         dev.off()
 225         """ % (DicoPath['dendro2'], DicoPath['arbre2'])
 226
 227     txt += """
 228     save.image(file="%s")
 229     """ % DicoPath['RData']
 230     fileout = open(DicoPath['Rchdtxt'], 'w')
 231     fileout.write(txt)
 232     fileout.close()
 233
 234 def RPamTxt(corpus, RscriptPath):
 235     DicoPath = corpus.dictpathout
 236     param = corpus.parametre
 237     print param
 238     txt = """
 239     source("%s")
 240     """ % (RscriptPath['pamtxt'])
 241     txt += """
 242     source("%s")
 243     """ % (RscriptPath['Rgraph'])
 244     txt += """
 245     result <- pamtxt("%s", "%s", "%s", method = "%s", clust_type = "%s", clnb = %i)
 246     n1 <- result$uce
 247     """ % (DicoPath['TableUc1'], DicoPath['listeuce1'], DicoPath['uce'], param['method'], param['cluster_type'], param['nbcl'] )
 248     txt += """
 249     open_file_graph("%s", width=400, height=400)
 250     plot(result$cl)
 251     dev.off()
 252     """ % (DicoPath['arbre1'])
 253     txt += """
 254     save.image(file="%s")
 255     """ % DicoPath['RData']
 256     fileout = open(DicoPath['Rchdtxt'], 'w')
 257     fileout.write(txt)
 258     fileout.close()
 259
 260
 261 def RchdQuest(DicoPath, RscriptPath, nbcl = 10, mincl = 10):
 262     txt = """
 263     source("%s")
 264     source("%s")
 265     source("%s")
 266     source("%s")
 267     """ % (RscriptPath['CHD'], RscriptPath['chdquest'], RscriptPath['anacor'],RscriptPath['Rgraph'])
 268
 269     txt += """
 270     nbt <- %i - 1
 271     mincl <- %i
 272     """ % (nbcl, mincl)
 273
 274     txt += """
 275     chd.result<-Rchdquest("%s","%s","%s", nbt = nbt, mincl = mincl)
 276     n1 <- chd.result$n1
 277     classeuce1 <- chd.result$cuce1
 278     """ % (DicoPath['Act01'], DicoPath['listeuce1'], DicoPath['uce'])
 279
 280     txt += """
 281     tree_tot1 <- make_tree_tot(chd.result$chd)
 282     open_file_graph("%s", width = 600, height=400)
 283     plot(tree_tot1$tree.cl)
 284     dev.off()
 285     """%DicoPath['arbre1']
 286
 287     txt += """
 288     tree_cut1 <- make_dendro_cut_tuple(tree_tot1$dendro_tuple, chd.result$coord_ok, classeuce1, 1, nbt)
 289     tree.cut1 <- tree_cut1
 290     save(tree.cut1, file="%s")
 291     open_file_graph("%s", width = 600, height=400)
 292     classes<-n1[,ncol(n1)]
 293     plot.dendropr(tree_cut1$tree.cl,classes)
 294     """ % (DicoPath['Rdendro'],DicoPath['dendro1'])
 295
 296     txt += """
 297     save.image(file="%s")
 298     """ % DicoPath['RData']
 299     fileout = open(DicoPath['Rchdquest'], 'w')
 300     fileout.write(txt)
 301     fileout.close()
 302
 303 def AlcesteTxtProf(DictChdTxtOut, RscriptsPath, clnb, taillecar):
 304     txt = "clnb<-%i\n" % clnb
 305     txt += """
 306 source("%s")
 307 load("%s")
 308 """ % (RscriptsPath['chdfunct'], DictChdTxtOut['RData'])
 309     txt += """
 310 dataact<-read.csv2("%s", header = FALSE, sep = ';',quote = '\"', row.names = 1, na.strings = 'NA')
 311 datasup<-read.csv2("%s", header = FALSE, sep = ';',quote = '\"', row.names = 1, na.strings = 'NA')
 312 dataet<-read.csv2("%s", header = FALSE, sep = ';',quote = '\"', row.names = 1, na.strings = 'NA')
 313 """ % (DictChdTxtOut['Contout'], DictChdTxtOut['ContSupOut'], DictChdTxtOut['ContEtOut'])
 314     txt += """
 315 tablesqrpact<-BuildProf(as.matrix(dataact),n1,clnb)
 316 tablesqrpsup<-BuildProf(as.matrix(datasup),n1,clnb)
 317 tablesqrpet<-BuildProf(as.matrix(dataet),n1,clnb)
 318 """
 319     txt += """
 320 PrintProfile(n1,tablesqrpact[4],tablesqrpet[4],tablesqrpact[5],tablesqrpet[5],clnb,"%s","%s",tablesqrpsup[4],tablesqrpsup[5])
 321 """ % (DictChdTxtOut['PROFILE_OUT'], DictChdTxtOut['ANTIPRO_OUT'])
 322     txt += """
 323 colnames(tablesqrpact[[2]])<-paste('classe',1:clnb,sep=' ')
 324 colnames(tablesqrpact[[1]])<-paste('classe',1:clnb,sep=' ')
 325 colnames(tablesqrpsup[[2]])<-paste('classe',1:clnb,sep=' ')
 326 colnames(tablesqrpsup[[1]])<-paste('classe',1:clnb,sep=' ')
 327 colnames(tablesqrpet[[2]])<-paste('classe',1:clnb,sep=' ')
 328 colnames(tablesqrpet[[1]])<-paste('classe',1:clnb,sep=' ')
 329 chistabletot<-rbind(tablesqrpact[2][[1]],tablesqrpsup[2][[1]])
 330 chistabletot<-rbind(chistabletot,tablesqrpet[2][[1]])
 331 ptabletot<-rbind(tablesqrpact[1][[1]],tablesqrpet[1][[1]])
 332 """
 333     txt += """
 334 write.csv2(chistabletot,file="%s")
 335 write.csv2(ptabletot,file="%s")
 336 gbcluster<-n1
 337 write.csv2(gbcluster,file="%s")
 338 """ % (DictChdTxtOut['chisqtable'], DictChdTxtOut['ptable'], DictChdTxtOut['SbyClasseOut'])
 339     if clnb > 2 :
 340         txt += """
 341     library(ca)
 342     colnames(dataact)<-paste('classe',1:clnb,sep=' ')
 343     colnames(datasup)<-paste('classe',1:clnb,sep=' ')
 344     colnames(dataet)<-paste('classe',1:clnb,sep=' ')
 345     rowtot<-nrow(dataact)+nrow(dataet)+nrow(datasup)
 346     afctable<-rbind(as.matrix(dataact),as.matrix(datasup))
 347     afctable<-rbind(afctable,as.matrix(dataet))
 348     colnames(afctable)<-paste('classe',1:clnb,sep=' ')
 349     afc<-ca(afctable,suprow=((nrow(dataact)+1):rowtot),nd=(ncol(afctable)-1))
 350     debsup<-nrow(dataact)+1
 351     debet<-nrow(dataact)+nrow(datasup)+1
 352     fin<-rowtot
 353     afc<-AddCorrelationOk(afc)
 354     """
 355     #FIXME : split this!!!
 356         txt += """
 357     source("%s")
 358     """ % RscriptsPath['Rgraph']
 359
 360         txt += """
 361         afc <- summary.ca.dm(afc)
 362         afc_table <- create_afc_table(afc)
 363         write.csv2(afc_table$facteur, file = "%s")
 364         write.csv2(afc_table$colonne, file = "%s")
 365         write.csv2(afc_table$ligne, file = "%s")
 366         """ % (DictChdTxtOut['afc_facteur'], DictChdTxtOut['afc_col'], DictChdTxtOut['afc_row'])
 367
 368         txt += """
 369         #xlab <- paste('facteur 1 - ', round(afc$facteur[1,2],2), sep = '')
 370         #ylab <- paste('facteur 2 - ', round(afc$facteur[2,2],2), sep = '')
 371         #xlab <- paste(xlab, ' %', sep = '')
 372         #ylab <- paste(ylab, ' %', sep = '')
 373         """
 374
 375         txt += """
 376     PARCEX<-%s
 377     xmin <- min(afc$rowcoord[,1]) + (0.1 * min(afc$rowcoord[,1]))
 378     xmax <- max(afc$rowcoord[,1]) + (0.1 * max(afc$rowcoord[,1]))
 379     ymin <- min(afc$rowcoord[,2]) + (0.1 * min(afc$rowcoord[,2]))
 380     ymax <- max(afc$rowcoord[,2]) + (0.1 * max(afc$rowcoord[,2]))
 381     """ % taillecar
 382         txt += """
 383     PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", what='coord', deb=1, fin=(debsup-1), xlab = xlab, ylab = ylab, xmin=xmin, xmax=xmax, ymin = ymin, ymax=ymax)
 384     """ % (DictChdTxtOut['AFC2DL_OUT'])
 385         txt += """
 386     PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", what='coord', deb=debsup, fin=(debet-1), xlab = xlab, ylab = ylab, xmin=xmin, xmax=xmax, ymin = ymin, ymax=ymax)
 387     """ % (DictChdTxtOut['AFC2DSL_OUT'])
 388         txt += """
 389     PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", what='coord', deb=debet, fin=fin, xlab = xlab, ylab = ylab, xmin=xmin, xmax=xmax, ymin = ymin, ymax=ymax)
 390     """ % (DictChdTxtOut['AFC2DEL_OUT'])
 391         txt += """
 392     PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", col=TRUE, what='coord', xlab = xlab, ylab = ylab, xmin=xmin, xmax=xmax, ymin = ymin, ymax=ymax)
 393     """ % (DictChdTxtOut['AFC2DCL_OUT'])
 394 #        txt += """
 395  #   PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", what='crl', deb=1, fin=(debsup-1), xlab = xlab, ylab = ylab)
 396  #   PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", what='crl', deb=debsup, fin=(debet-1), xlab = xlab, ylab = ylab)
 397   #  PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", what='crl', deb=debet, fin=fin, xlab = xlab, ylab = ylab)
 398  #   PlotAfc2dCoul(afc, as.data.frame(chistabletot), "%s", col=TRUE, what='crl', xlab = xlab, ylab = ylab)
 399  #   """ % (DictChdTxtOut['AFC2DCoul'], DictChdTxtOut['AFC2DCoulSup'], DictChdTxtOut['AFC2DCoulEt'], DictChdTxtOut['AFC2DCoulCl'])
 400
 401     txt += """
 402 #rm(dataact)
 403 #rm(datasup)
 404 #rm(dataet)
 405 rm(tablesqrpact)
 406 rm(tablesqrpsup)
 407 rm(tablesqrpet)
 408 save.image(file="%s")
 409 """ % DictChdTxtOut['RData']
 410     file = open(DictChdTxtOut['RTxtProfGraph'], 'w')
 411     file.write(txt)
 412     file.close()
 413
 414
 415 def write_afc_graph(self):
 416     if self.param['over'] : over = 'TRUE'
 417     else : over = 'FALSE'
 418
 419     if self.param['do_select_nb'] : do_select_nb = 'TRUE'
 420     else : do_select_nb = 'FALSE'
 421
 422     if self.param['do_select_chi'] : do_select_chi = 'TRUE'
 423     else : do_select_chi = 'FALSE'
 424
 425     if self.param['do_select_chi_classe'] : do_select_chi_classe = 'TRUE'
 426     else : do_select_chi_classe = 'FALSE'
 427
 428     if self.param['cex_txt'] : cex_txt = 'TRUE'
 429     else : cex_txt = 'FALSE'
 430
 431     if self.param['tchi'] : tchi = 'TRUE'
 432     else : tchi = 'FALSE'
 433
 434     with open(self.RscriptsPath['afc_graph'], 'r') as f:
 435         txt = f.read()
 436
 437 #    self.DictPathOut['RData'], \
 438     scripts = txt % (self.RscriptsPath['Rgraph'],\
 439     self.param['typegraph'], \
 440     self.param['what'], \
 441     self.param['facteur'][0],\
 442     self.param['facteur'][1], \
 443     self.param['facteur'][2], \
 444     self.param['qui'], \
 445     over,  do_select_nb, \
 446     self.param['select_nb'],  \
 447     do_select_chi, \
 448     self.param['select_chi'], \
 449     do_select_chi_classe, \
 450     self.param['nbchic'], \
 451     cex_txt, \
 452     self.param['txt_min'], \
 453     self.param['txt_max'], \
 454     self.fileout, \
 455     self.param['width'], \
 456     self.param['height'],\
 457     self.param['taillecar'], \
 458     self.param['alpha'], \
 459     self.param['film'], \
 460     tchi,\
 461     self.param['tchi_min'],\
 462     self.param['tchi_max'],\
 463     ffr(os.path.dirname(self.fileout)))
 464     return scripts
 465
 466 def print_simi3d(self):
 467     simi3d = self.parent.simi3dpanel
 468     txt = '#Fichier genere par Iramuteq'
 469     if simi3d.movie.GetValue() :
 470         movie = "'" + ffr(os.path.dirname(self.DictPathOut['RData'])) + "'"
 471     else :
 472         movie = 'NULL'
 473
 474     #if self.corpus.parametres['type'] == 'corpus' :
 475     #    header = 'TRUE'
 476     #else :
 477     #    header = 'FALSE'
 478     header = 'FALSE'
 479     txt += """
 480     dm<-read.csv2("%s",row.names=1,header = %s)
 481     load("%s")
 482     """ % (self.DictPathOut['Contout'], header, self.DictPathOut['RData'])
 483
 484     txt += """
 485     source("%s")
 486     """ % self.parent.RscriptsPath['Rgraph']
 487
 488
 489     txt += """
 490     make.simi.afc(dm,chistabletot, lim=%i, alpha = %.2f, movie = %s)
 491     """ % (simi3d.spin_1.GetValue(), float(simi3d.slider_1.GetValue())/100, movie)
 492     tmpfile = tempfile.mktemp(dir=self.parent.TEMPDIR)
 493     tmp = open(tmpfile,'w')
 494     tmp.write(txt)
 495     tmp.close()
 496     return tmpfile
 497
 498 def dendroandbarplot(table, rownames, colnames, rgraph, tmpgraph, intxt = False, dendro=False) :
 499     if not intxt :
 500         txttable = 'c(' + ','.join([','.join(line) for line in table]) + ')'
 501     rownb = len(rownames)
 502     rownames = 'c("' + '","'.join(rownames) + '")'
 503     colnames = 'c("' + '","'.join(colnames) + '")'
 504     if not intxt :
 505         #FIXME
 506         txt = """
 507             di <- matrix(data=%s, nrow=%i, byrow = TRUE)
 508             rownames(di)<- %s
 509             colnames(di) <- %s
 510         """ % (txttable, rownb, rownames, colnames)
 511     else :
 512         txt = intxt
 513     txt += """
 514         load("%s")
 515         library(ape)
 516         source("%s")
 517         height <- (30*ncol(di)) + (15*nrow(di))
 518         height <- ifelse(height <= 400, 400, height)
 519         width <- 500
 520         open_file_graph("%s", width=width, height=height)
 521         plot.dendro.lex(tree.cut1$tree.cl, di)
 522         """ % (ffr(dendro),ffr(rgraph),  ffr(tmpgraph))
 523     return txt
 524
 525 def barplot(table, rownames, colnames, rgraph, tmpgraph, intxt = False) :
 526     if not intxt :
 527         txttable = 'c(' + ','.join([','.join(line) for line in table]) + ')'
 528     #width = 100 + (15 * len(rownames)) + (100 * len(colnames))
 529     #height =  len(rownames) * 15
 530     rownb = len(rownames)
 531     #if height < 400 :
 532     #    height = 400
 533     rownames = 'c("' + '","'.join(rownames) + '")'
 534     colnames = 'c("' + '","'.join(colnames) + '")'
 535     if not intxt :
 536         #FIXME
 537         txt = """
 538             inf <- NA
 539             di <- matrix(data=%s, nrow=%i, byrow = TRUE)
 540             di[is.na(di)] <- max(di, na.rm=TRUE) + 2
 541             rownames(di)<- %s
 542             colnames(di) <- %s
 543         """ % (txttable, rownb, rownames, colnames)
 544     else :
 545         txt = intxt
 546     txt += """
 547         source("%s")
 548         color = rainbow(nrow(di))
 549         width <- 100 + (20*length(rownames(di))) + (100 * length(colnames(di)))
 550         height <- nrow(di) * 15
 551         if (height < 400) { height <- 400}
 552         open_file_graph("%s",width = width, height = height)
 553         par(mar=c(0,0,0,0))
 554             layout(matrix(c(1,2),1,2, byrow=TRUE),widths=c(3,lcm(7)))
 555         par(mar=c(2,2,1,0))
 556         coord <- barplot(as.matrix(di), beside = TRUE, col = color, space = c(0.1,0.6))
 557         c <- colMeans(coord)
 558         c1 <- c[-1]
 559         c2 <- c[-length(c)]
 560         cc <- cbind(c1,c2)
 561         lcoord <- apply(cc, 1, mean)
 562         abline(v=lcoord)
 563         if (min(di) < 0) {
 564             amp <- abs(max(di) - min(di))
 565         } else {
 566             amp <- max(di)
 567         }
 568         if (amp < 10) {
 569             d <- 2
 570         } else {
 571             d <- signif(amp%%/%%10,1)
 572         }
 573         mn <- round(min(di))
 574         mx <- round(max(di))
 575         for (i in mn:mx) {
 576             if ((i/d) == (i%%/%%d)) {
 577                 abline(h=i,lty=3)
 578             }
 579         }
 580         par(mar=c(0,0,0,0))
 581         plot(0, axes = FALSE, pch = '')
 582         legend(x = 'center' , rownames(di), fill = color)
 583         dev.off()
 584         """ % (rgraph, ffr(tmpgraph))
 585     return txt
 586
 587 #def RAfcUci(DictAfcUciOut, nd=2, RscriptsPath='', PARCEX='0.8'):
 588 #    txt = """
 589 #    library(ca)
 590 #    nd<-%i
 591 #    """ % nd
 592 #    txt += """
 593 #    dataact<-read.csv2("%s")
 594 #    """ % (DictAfcUciOut['TableCont'])#, encoding)
 595 #    txt += """
 596 #    datasup<-read.csv2("%s")
 597 #    """ % (DictAfcUciOut['TableSup'])#, encoding)
 598 #    txt += """
 599 #    dataet<-read.csv2("%s")
 600 #    """ % (DictAfcUciOut['TableEt'])#, encoding)
 601 #    txt += """
 602 #    datatotsup<-cbind(dataact,datasup)
 603 #    datatotet<-cbind(dataact,dataet)
 604 #    afcact<-ca(dataact,nd=nd)
 605 #    afcsup<-ca(datatotsup,supcol=((ncol(dataact)+1):ncol(datatotsup)),nd=nd)
 606 #    afcet<-ca(datatotet,supcol=((ncol(dataact)+1):ncol(datatotet)),nd=nd)
 607 #    afctot<-afcsup$colcoord
 608 #    rownames(afctot)<-afcsup$colnames
 609 #    colnames(afctot)<-paste('coord. facteur',1:nd,sep=' ')
 610 #    afctot<-cbind(afctot,mass=afcsup$colmass)
 611 #    afctot<-cbind(afctot,distance=afcsup$coldist)
 612 #    afctot<-cbind(afctot,intertie=afcsup$colinertia)
 613 #    rcolet<-afcet$colsup
 614 #    afctmp<-afcet$colcoord[rcolet,]
 615 #    rownames(afctmp)<-afcet$colnames[rcolet]
 616 #    afctmp<-cbind(afctmp,afcet$colmass[rcolet])
 617 #    afctmp<-cbind(afctmp,afcet$coldist[rcolet])
 618 #    afctmp<-cbind(afctmp,afcet$colinertia[rcolet])
 619 #    afctot<-rbind(afctot,afctmp)
 620 #    write.csv2(afctot,file = "%s")
 621 #    source("%s")
 622 #    """ % (DictAfcUciOut['afc_row'], RscriptsPath['Rgraph'])
 623 #    txt += """
 624 #    PARCEX=%s
 625 #    """ % PARCEX
 626 #    #FIXME
 627 #    txt += """
 628 #    PlotAfc(afcet,filename="%s",toplot=c%s, PARCEX=PARCEX)
 629 #    """ % (DictAfcUciOut['AfcColAct'], "('none','active')")
 630 #    txt += """
 631 #    PlotAfc(afcsup,filename="%s",toplot=c%s, PARCEX=PARCEX)
 632 #    """ % (DictAfcUciOut['AfcColSup'], "('none','passive')")
 633 #    txt += """PlotAfc(afcet,filename="%s", toplot=c%s, PARCEX=PARCEX)
 634 #    """ % (DictAfcUciOut['AfcColEt'], "('none','passive')")
 635 #    txt += """
 636 #    PlotAfc(afcet,filename="%s", toplot=c%s, PARCEX=PARCEX)
 637 #    """ % (DictAfcUciOut['AfcRow'], "('all','none')")
 638 #    f = open(DictAfcUciOut['Rafcuci'], 'w')
 639 #    f.write(txt)
 640 #    f.close()
 641
 642 class PrintSimiScript(PrintRScript) :
 643     def make_script(self) :
 644         self.txtgraph = ''
 645         self.packages(['igraph', 'proxy', 'Matrix'])
 646         self.sources([self.analyse.parent.RscriptsPath['simi'], self.analyse.parent.RscriptsPath['Rgraph']])
 647         txt = """
 648         dm.path <- "%s"
 649         cn.path <- "%s"
 650         selected.col <- "%s"
 651         """ % (self.pathout['mat01.csv'], self.pathout['actives.csv'], self.pathout['selected.csv'])
 652         txt += """
 653         dm <-readMM(dm.path)
 654         cn <- read.table(cn.path, sep=';', quote='"')
 655         colnames(dm) <- cn[,1]
 656         sel.col <- read.csv2(selected.col)
 657         dm <- dm[, sel.col[,1] + 1]
 658         """
 659
 660         if self.parametres['coeff'] == 0 :
 661             method = 'cooc'
 662             txt += """
 663             method <- 'cooc'
 664             mat <- make.a(dm)
 665             """
 666         else :
 667             txt += """
 668             dm <- as.matrix(dm)
 669             """
 670         if self.parametres['coeff'] == 1 :
 671             method = 'prcooc'
 672             txt += """
 673             method <- 'Russel'
 674             mat <- simil(dm, method = 'Russel', diag = TRUE, upper = TRUE, by_rows = FALSE)
 675             """
 676         elif self.analyse.indices[self.parametres['coeff']] == 'binomial' :
 677             method = 'binomial'
 678             txt += """
 679             method <- 'binomial'
 680             mat <- binom.sim(dm)
 681             """
 682         elif self.parametres['coeff'] != 0 :
 683             method = self.analyse.indices[self.parametres['coeff']]
 684             txt += """
 685             method <-"%s"
 686             mat <- simil(dm, method = method, diag = TRUE, upper = TRUE, by_rows = FALSE)
 687             """ % self.analyse.indices[self.parametres['coeff']]
 688         txt += """
 689         mat <- as.matrix(stats::as.dist(mat,diag=TRUE,upper=TRUE))
 690         mat[is.na(mat)] <- 0
 691         mat[is.infinite(mat)] <- 0
 692         """
 693         if self.parametres['layout'] == 0 : layout = 'random'
 694         if self.parametres['layout'] == 1 : layout = 'circle'
 695         if self.parametres['layout'] == 2 : layout = 'frutch'
 696         if self.parametres['layout'] == 3 : layout = 'kawa'
 697         if self.parametres['layout'] == 4 : layout = 'graphopt'
 698
 699         self.filename=''
 700         if self.parametres['type_graph'] == 0 : type = 'tkplot'
 701         if self.parametres['type_graph'] == 1 :
 702             graphnb = 1
 703             type = 'nplot'
 704             dirout = os.path.dirname(self.pathout['mat01'])
 705             while os.path.exists(os.path.join(dirout,'graph_simi_'+str(graphnb)+'.png')):
 706                 graphnb +=1
 707             self.filename = ffr(os.path.join(dirout,'graph_simi_'+str(graphnb)+'.png'))
 708         if self.parametres['type_graph'] == 2 : type = 'rgl'
 709
 710         if self.parametres['arbremax'] :
 711             arbremax = 'TRUE'
 712             self.txtgraph += ' - arbre maximum'
 713         else : arbremax = 'FALSE'
 714
 715         if self.parametres['coeff_tv'] :
 716             coeff_tv = self.parametres['coeff_tv_nb']
 717             tvminmax = 'c(NULL,NULL)'
 718         elif not self.parametres['coeff_tv'] or self.parametres.get('sformchi', False) :
 719             coeff_tv = 'NULL'
 720             tvminmax = 'c(%i, %i)' %(self.parametres['tvmin'], self.parametres['tvmax'])
 721         if self.parametres['coeff_te'] : coeff_te = 'c(%i,%i)' % (self.parametres['coeff_temin'], self.parametres['coeff_temax'])
 722         else : coeff_te = 'NULL'
 723
 724         if self.parametres['vcex'] or self.parametres.get('cexfromchi', False) :
 725             vcexminmax = 'c(%i/10,%i/10)' % (self.parametres['vcexmin'],self.parametres['vcexmax'])
 726         else :
 727             vcexminmax = 'c(NULL,NULL)'
 728         if not self.parametres['label_v'] : label_v = 'FALSE'
 729         else : label_v = 'TRUE'
 730
 731         if not self.parametres['label_e'] : label_e = 'FALSE'
 732         else : label_e = 'TRUE'
 733
 734         if self.parametres['seuil_ok'] : seuil = str(self.parametres['seuil'])
 735         else : seuil = 'NULL'
 736
 737         cols = str(self.parametres['cols']).replace(')',', max=255)')
 738         cola = str(self.parametres['cola']).replace(')',',max=255)')
 739
 740         txt += """
 741         minmaxeff <- %s
 742         """ % tvminmax
 743         txt += """
 744         vcexminmax <- %s
 745         """ % vcexminmax
 746         txt += """
 747         cex = %i/10
 748         """ % self.parametres['cex']
 749
 750         if self.parametres['film'] :
 751             txt += """
 752             film <- "%s"
 753             """ % self.pathout['film']
 754         else :
 755             txt += """
 756             film <- NULL
 757             """
 758         txt += """
 759         seuil <- %s
 760         """ % seuil
 761
 762         txt += """
 763         label.v <- %s
 764         label.e <- %s
 765         """ % (label_v, label_e)
 766         txt += """
 767         cols <- rgb%s
 768         cola <- rgb%s
 769         """ % (cols, cola)
 770         txt += """
 771         width <- %i
 772         height <- %i
 773         """ % (self.parametres['width'], self.parametres['height'])
 774         if self.parametres['keep_coord'] :
 775             txt += """
 776             coords <- try(coords, TRUE)
 777             if (!is.matrix(coords)) {
 778                 coords<-NULL
 779             }
 780             """
 781         else :
 782             txt += """
 783             coords <- NULL
 784             """
 785         txt += """
 786         alpha <- %i/100
 787         """ % self.parametres['alpha']
 788         txt += """
 789         alpha <- %i/100
 790         """ % self.parametres['alpha']
 791 #############################################
 792         if  self.parametres.get('bystar',False) :
 793             txt += """
 794             et <- list()
 795             """
 796             print self.parametres
 797             for i,et in enumerate(self.parametres['stars']) :
 798                 txt+= """
 799                 et[[%i]] <- c(%s)
 800                 """ % (i+1, ','.join(et[1:]))
 801             txt+= """
 802             unetoile <- c('%s')
 803             """ % ("','".join([val[0] for val in self.tableau.etline]))
 804             txt += """
 805             fsum <- NULL
 806             rs <- rowSums(dm)
 807             for (i in 1:length(unetoile)) {
 808                 print(unetoile[i])
 809                 tosum <- et[[i]]
 810                 if (length(tosum) > 1) {
 811                     fsum <- cbind(fsum, colSums(dm[tosum,]))
 812                 } else {
 813                     fsum <- cbind(fsum, dm[tosum,])
 814                 }
 815             }
 816             source("%s")
 817             lex <- AsLexico2(fsum, chip=TRUE)
 818             dcol <- apply(lex[[4]],1,which.max)
 819             toblack <- apply(lex[[4]],1,max)
 820             gcol <- rainbow(length(unetoile))
 821             #gcol[2] <- 'orange'
 822             vertex.label.color <- gcol[dcol]
 823             vertex.label.color[which(toblack <= 3.84)] <- 'black'
 824             leg <- list(unetoile=unetoile, gcol=gcol)
 825             cols <- vertex.label.color
 826             chivertex.size <- norm.vec(toblack, vcexminmax[1],  vcexminmax[2])
 827
 828             """ % (self.parent.RscriptsPath['chdfunct'])
 829         else :
 830             txt += """
 831             vertex.label.color <- 'black'
 832             chivertex.size <- 1
 833             leg<-NULL
 834             """
 835 #############################################
 836
 837 #        txt += """
 838 #        eff <- colSums(dm)
 839 #        g.ori <- graph.adjacency(mat, mode='lower', weighted = TRUE)
 840 #        w.ori <- E(g.ori)$weight
 841 #        if (max.tree) {
 842 #            if (method == 'cooc') {
 843 #                E(g.ori)$weight <- 1 / w.ori
 844 #            } else {
 845 #                E(g.ori)$weigth <- 1 - w.ori
 846 #            }
 847 #            g.max <- minimum.spanning.tree(g.ori)
 848 #            if (method == 'cooc') {
 849 #                E(g.max)$weight <- 1 / E(g.max)$weight
 850 #            } else {
 851 #                E(g.max)$weight <- 1 - E(g.max)$weight
 852 #            }
 853 #            g.toplot <- g.max
 854 #        } else {
 855 #            g.toplot <- g.ori
 856 #        }
 857 #        """
 858         txt += """
 859         eff <- colSums(dm)
 860         x <- list(mat = mat, eff = eff)
 861         graph.simi <- do.simi(x, method='%s', seuil = seuil, p.type = '%s', layout.type = '%s', max.tree = %s, coeff.vertex=%s, coeff.edge = %s, minmaxeff = minmaxeff, vcexminmax = vcexminmax, cex = cex, coords = coords)
 862         """ % (method, type, layout, arbremax, coeff_tv, coeff_te)
 863
 864         if self.parametres.get('bystar',False) :
 865             if self.parametres.get('cexfromchi', False) :
 866                 txt+="""
 867                     label.cex<-chivertex.size
 868                     """
 869             else :
 870                 txt+="""
 871                 label.cex <- NULL
 872                 """
 873             if self.parametres.get('sfromchi', False) :
 874                 txt += """
 875                 vertex.size <- norm.vec(toblack, minmaxeff[1], minmaxeff[2])
 876                 """
 877             else :
 878                 txt += """
 879                 vertex.size <- NULL
 880                 """
 881         else :
 882             #FIXME
 883             tmpchi = False
 884             if tmpchi :
 885                 txt += """
 886                 lchi <- read.table("%s")
 887                 lchi <- lchi[,1]
 888                 """ % ffr(tmpchi)
 889                 if 'selected_col' in dir(self.tableau) :
 890                     txt += """
 891                     lchi <- lchi[c%s+1]
 892                     """ % datas
 893             if tmpchi and self.parametres.get('cexfromchi', False) :
 894                 txt += """
 895                 label.cex <- norm.vec(lchi, vcexminmax[1], vcexminmax[2])
 896                 """
 897             else :
 898                 txt += """
 899             if (is.null(vcexminmax[1])) {
 900                 label.cex <- NULL
 901             } else {
 902                 label.cex <- graph.simi$label.cex
 903             }
 904             """
 905             if tmpchi and self.parametres.get('sfromchi', False) :
 906                 txt += """
 907                 vertex.size <- norm.vec(lchi, minmaxeff[1], minmaxeff[2])
 908                 """
 909             else :
 910                 txt += """
 911             if (is.null(minmaxeff[1])) {
 912                 vertex.size <- NULL
 913             } else {
 914                 vertex.size <- graph.simi$eff
 915             }
 916             """
 917         txt += """ vertex.size <- NULL """
 918         txt += """
 919         coords <- plot.simi(graph.simi, p.type='%s',filename="%s", vertex.label = label.v, edge.label = label.e, vertex.col = cols, vertex.label.color = vertex.label.color, vertex.label.cex=label.cex, vertex.size = vertex.size, edge.col = cola, leg=leg, width = width, height = height, alpha = alpha, movie = film)
 920         save.image(file="%s")
 921         """ % (type, self.filename, self.pathout['RData'])
 922
 923         self.add(txt)
 924         self.write()
 925
 926 class WordCloudRScript(PrintRScript) :
 927     def make_script(self) :
 928         self.sources([self.analyse.parent.RscriptsPath['Rgraph']])
 929         self.packages(['wordcloud'])
 930         bg_col = Rcolor(self.parametres['col_bg'])
 931         txt_col = Rcolor(self.parametres['col_text'])
 932         txt = """
 933         act <- read.csv2("%s", header = FALSE, row.names=1, sep='\t')
 934         selected.col <- read.table("%s")
 935         toprint <- as.matrix(act[selected.col[,1] + 1,])
 936         rownames(toprint) <- rownames(act)[selected.col[,1] + 1]
 937         maxword <- %i
 938         if (nrow(toprint) > maxword) {
 939             toprint <- as.matrix(toprint[order(toprint[,1], decreasing=TRUE),])
 940             toprint <- as.matrix(toprint[1:maxword,])
 941         }
 942         open_file_graph("%s", width = %i, height = %i)
 943         par(bg=rgb%s)
 944         wordcloud(row.names(toprint), toprint[,1], scale=c(%f,%f), random.order=FALSE, colors=rgb%s)
 945         dev.off()
 946         """ % (ffr(self.analyse.pathout['actives_eff.csv']), ffr(self.analyse.pathout['selected.csv']), self.parametres['maxword'], ffr(self.parametres['graphout']), self.parametres['width'], self.parametres['height'], bg_col, self.parametres['maxcex'], self.parametres['mincex'], txt_col)
 947         self.add(txt)
 948         self.write()