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

   1 # -*- coding: utf-8 -*-
   2 # Author: Pierre Ratinaud
   3 # lisence : GNU GPL
   4 # copyright : 2014 (c) Pierre Ratinaud
   5
   6 import os
   7 from time import time
   8 from analysetxt import AnalyseText
   9 from OptionAlceste import OptionAlc
  10 from PrintRScript import RchdTxt, ReinertTxtProf, TgenProfScript
  11 from layout import PrintRapport
  12 from chemins import ChdTxtPathOut, PathOut
  13 from functions import DoConf, print_liste, TGen
  14
  15
  16 class Reinert(AnalyseText) :
  17     def doanalyse(self) :
  18         self.parametres['type'] = 'alceste'
  19         self.pathout.basefiles(ChdTxtPathOut)
  20         self.actives, lim = self.corpus.make_actives_nb(self.parametres['max_actives'], 1)
  21         self.parametres['eff_min_forme'] = lim
  22         self.parametres['nbactives'] = len(self.actives)
  23         uci = False
  24         if self.parametres['classif_mode'] == 0 :
  25             lenuc1, lenuc2 = self.corpus.make_and_write_sparse_matrix_from_uc(self.actives, self.parametres['tailleuc1'], self.parametres['tailleuc2'], self.pathout['TableUc1'], self.pathout['TableUc2'], self.pathout['listeuce1'], self.pathout['listeuce2'])
  26             self.parametres['lenuc1'] = lenuc1
  27             self.parametres['lenuc2'] = lenuc2
  28         elif self.parametres['classif_mode'] == 1 :
  29             self.corpus.make_and_write_sparse_matrix_from_uces(self.actives, self.pathout['TableUc1'], self.pathout['listeuce1'])
  30         elif self.parametres['classif_mode'] == 2 :
  31             self.corpus.make_and_write_sparse_matrix_from_uci(self.actives, self.pathout['TableUc1'], self.pathout['listeuce1'])
  32             uci = True
  33         Rscript = self.printRscript()
  34         result = self.doR(Rscript, dlg=self.dlg, message='CHD...')
  35         if not result :
  36             return 'NOK'
  37         self.corpus.make_ucecl_from_R(self.pathout['uce'])
  38         self.corpus.make_and_write_profile(self.actives, self.corpus.lc, self.pathout['Contout'], uci = uci)
  39         self.sup, lim = self.corpus.make_actives_nb(self.parametres['max_actives'], 2)
  40         self.corpus.make_and_write_profile(self.sup, self.corpus.lc, self.pathout['ContSupOut'], uci = uci)
  41         self.corpus.make_and_write_profile_et(self.corpus.lc, self.pathout['ContEtOut'], uci = uci)
  42         self.clnb = len(self.corpus.lc)
  43         self.parametres['clnb'] = self.clnb
  44         Rscript = self.printRscript2()
  45         self.doR(Rscript, dlg=self.dlg, message='profils et A.F.C. ...')
  46         self.time = time() - self.t1
  47         minutes, seconds = divmod(self.time, 60)
  48         hours, minutes = divmod(minutes, 60)
  49         self.parametres['time'] = '%.0fh %.0fm %.0fs' % (hours, minutes, seconds)
  50         self.print_graph_files()
  51
  52     def preferences(self) :
  53         parametres = DoConf(self.parent.ConfigPath['reinert']).getoptions('ALCESTE')
  54         parametres['corpus'] = self.corpus
  55         parametres['pathout'] = self.pathout
  56         self.dial = OptionAlc(self.parent, parametres)
  57         self.dial.CenterOnParent()
  58         self.dialok = self.dial.ShowModal()
  59         if self.dialok == 5100 :
  60             parametres['classif_mode'] = self.dial.radio_box_2.GetSelection()
  61             parametres['tailleuc1'] = self.dial.spin_ctrl_1.GetValue()
  62             parametres['tailleuc2'] = self.dial.spin_ctrl_2.GetValue()
  63             parametres['mincl'] = self.dial.spin_ctrl_4.GetValue()
  64             parametres['minforme'] = self.dial.spin_ctrl_5.GetValue()
  65             parametres['nbcl_p1'] = self.dial.spin_nbcl.GetValue()
  66             parametres['max_actives'] = self.dial.spin_max_actives.GetValue()
  67             parametres['corpus'] = ''
  68             parametres['svdmethod'] = self.dial.svdmethod[self.dial.choicesvd.GetSelection()]
  69             parametres['pathout'] = self.pathout.dirout
  70             parametres['mode.patate'] = self.dial.check_patate.GetValue()
  71             DoConf(self.parent.ConfigPath['reinert']).makeoptions(['ALCESTE'], [parametres])
  72             self.dial.Destroy()
  73             print parametres
  74             return parametres
  75         else :
  76             self.dial.Destroy()
  77             return None
  78
  79     def printRscript(self) :
  80         RchdTxt(self.pathout, self.parent.RscriptsPath, self.parametres['mincl'], self.parametres['classif_mode'], nbt=self.parametres['nbcl_p1'] - 1, svdmethod=self.parametres['svdmethod'], libsvdc=self.parent.pref.getboolean('iramuteq', 'libsvdc'), libsvdc_path=self.parent.pref.get('iramuteq', 'libsvdc_path'), R_max_mem=False, mode_patate=self.parametres['mode.patate'])
  81         return self.pathout['Rchdtxt']
  82
  83     def printRscript2(self) :
  84         ReinertTxtProf(self.pathout, self.parent.RscriptsPath, self.clnb, 0.9)
  85         return self.pathout['RTxtProfGraph']
  86
  87     def print_graph_files(self) :
  88         mess_afc = u"La position des points n'est peut être pas exacte"
  89         afc_graph_list = [[os.path.basename(self.pathout['AFC2DL_OUT']), u'Variables actives - coordonnées - 30 points par classes - facteurs 1 / 2 - %s' % mess_afc],
  90                       [os.path.basename(self.pathout['AFC2DSL_OUT']), u'variables supplémentaires - coordonnées - 30 points par classes - facteurs 1 / 2 - %s' % mess_afc],
  91                       [os.path.basename(self.pathout['AFC2DEL_OUT']), u'Variables illustratives - Coordonnées - 30 points par classes - facteur 1 / 2 - %s' % mess_afc],
  92                       [os.path.basename(self.pathout['AFC2DCL_OUT']), u'Classes - Coordonnées - facteur 1 / 2']]
  93         chd_graph_list = [[os.path.basename(self.pathout['dendro1']), u'dendrogramme à partir de chd1']]
  94         if self.parametres['classif_mode'] == 0 :
  95             chd_graph_list.append([os.path.basename(self.pathout['dendro2']), u'dendrogramme à partir de chd2'])
  96         chd_graph_list.append([os.path.basename(self.pathout['arbre1']), u'chd1'])
  97         if self.parametres['classif_mode'] == 0 :
  98             chd_graph_list.append([os.path.basename(self.pathout['arbre2']), u'chd2'])
  99         print_liste(self.pathout['liste_graph_afc'], afc_graph_list)
 100         print_liste(self.pathout['liste_graph_chd'], chd_graph_list)
 101         PrintRapport(self, self.corpus, self.parametres)
 102
 103 class TgenProf(AnalyseText):
 104     def __init__(self, ira, corpus, parametres, cluster_size):
 105         self.ira = ira
 106         self.corpus = corpus
 107         self.parametres = parametres
 108         self.pathout = PathOut(dirout = self.parametres['pathout'])
 109         self.cluster_size = [len(classe) for classe in corpus.lc]
 110         print cluster_size
 111         self.doanalyse()
 112
 113     def doanalyse(self):
 114         self.tgen = TGen(path = self.parametres['tgenpath'], encoding = self.ira.syscoding)
 115         self.tgen.read(self.tgen.path)
 116         #self.parametres['etoiles'].sort()
 117         self.parametres['tgeneff'] = os.path.join(self.parametres['pathout'], 'tgeneff.csv')
 118         tgenst = self.corpus.make_tgen_profile(self.tgen.tgen, self.corpus.lc)
 119         clnames = ['cluster_%03d' % i for i in range(1, len(self.cluster_size) + 1)]
 120         et = dict(zip(clnames, self.cluster_size))
 121         tgenst = dict([[line[0], dict(zip(clnames, line[1:]))] for line in tgenst])
 122         self.tgen.writetable(self.parametres['tgeneff'], tgenst, et)
 123         self.parametres['tgenspec'] = os.path.join(self.parametres['pathout'], 'tgenchi2.csv')
 124         self.Rscript = TgenProfScript(self)
 125         self.Rscript.make_script()
 126         self.Rscript.write()
 127         self.doR(self.Rscript.scriptout, dlg = False, message = 'R...')
 128
 129
 130
 131
 132
 133