1 # -*- coding: utf-8 -*-
2 # Author: Pierre Ratinaud
4 # copyright : 2012-2013 (c) Pierre Ratinaud
7 from chemins import PathOut
8 from functions import exec_rcode, check_Rresult, DoConf, ReadDicoAsDico, progressbar
9 from shutil import copy
10 from time import time, sleep
11 from uuid import uuid4
13 from openanalyse import OpenAnalyse
14 from dialog import StatDialog
16 log = logging.getLogger('iramuteq.analyse')
19 def __init__(self, ira, corpus, parametres=None, dlg=False, lemdial=True) :
25 self.parametres = parametres
26 self.lemdial = lemdial
28 self.keys = DoConf(self.ira.ConfigPath['key']).getoptions()
29 if not 'pathout' in self.parametres :
30 self.pathout = PathOut(corpus.parametres['originalpath'], analyse_type=parametres['type'], dirout=corpus.parametres['pathout'])
32 self.pathout = PathOut(filename=corpus.parametres['originalpath'], dirout=self.parametres['pathout'], analyse_type=self.parametres['type'])
33 self.parametres = self.lemparam()
35 if self.parametres is not None :
36 self.parametres = self.make_config(parametres)
37 log.info(self.pathout.dirout)
38 if self.parametres is not None :
39 self.keys = DoConf(self.ira.ConfigPath['key']).getoptions()
40 gramact = [k for k in self.keys if self.keys[k] == 1]
41 gramsup = [k for k in self.keys if self.keys[k] == 2]
42 self.parametres['pathout'] = self.pathout.mkdirout()
43 self.pathout = PathOut(dirout=self.parametres['pathout'])
44 self.pathout.createdir(self.parametres['pathout'])
45 self.parametres['corpus'] = self.corpus.parametres['uuid']
46 self.parametres['uuid'] = str(uuid4())
47 self.parametres['name'] = os.path.split(self.parametres['pathout'])[1]
48 self.parametres['type'] = parametres['type']
49 self.parametres['encoding'] = self.ira.syscoding
51 if not self.parametres.get('dictionary', False) :
52 self.corpus.make_lems(lem=self.parametres['lem'])
54 dico = ReadDicoAsDico(self.parametres['dictionary'])
55 self.corpus.make_lems_from_dict(dico, dolem=self.parametres['lem'])
56 dictname = os.path.basename(self.parametres['dictionary'])
57 dictpath = os.path.join(self.pathout.dirout, dictname)
58 copy(self.parametres['dictionary'], dictpath)
59 self.parametres['dictionary'] = dictpath
60 self.corpus.parse_active(gramact, gramsup)
62 self.dlg = progressbar(self.ira, dlg)
63 result_analyse = self.doanalyse()
64 if result_analyse is None :
65 self.time = time() - self.t1
66 minutes, seconds = divmod(self.time, 60)
67 hours, minutes = divmod(minutes, 60)
68 self.parametres['time'] = '%.0fh %.0fm %.0fs' % (hours, minutes, seconds)
69 self.parametres['ira'] = self.pathout['Analyse.ira']
70 DoConf().makeoptions([self.parametres['type']], [self.parametres], self.pathout['Analyse.ira'])
71 self.ira.history.add(self.parametres)
74 OpenAnalyse(self.parent, self.parametres['ira'])
75 self.ira.tree.AddAnalyse(self.parametres)
90 if self.dlg and self.lemdial:
91 dial = StatDialog(self.parent, self.keys)
93 val = dial.ShowModal()
95 if dial.radio_lem.GetSelection() == 0 :
99 self.parametres['lem'] = lem
100 if dial.radio_dictchoice.GetSelection() == 1 :
101 self.parametres['dictionary'] = dial.dictpath.GetValue()
103 return self.parametres
108 return self.parametres
110 def make_config(self, config) :
111 if config is not None :
115 return self.preferences()
119 def readconfig(self, config) :
122 def preferences(self) :
123 return self.parametres
125 def printRscript(self) :
128 def doR(self, Rscript, wait=False, dlg=None, message='') :
129 log.info('R code...')
130 pid = exec_rcode(self.ira.RPath, Rscript, wait=wait)
131 while pid.poll() is None :
133 self.dlg.Pulse(message)
137 return check_Rresult(self.ira, pid)
144 # keys = {'art_def' : 2,
173 # gramact = [k for k in keys if keys[k] == 1]
174 # gramsup = [k for k in keys if keys[k] == 2]