+class CreateTgenDialog ( wx.Frame ):
+
+ def __init__( self, parent, lemlist, tgen = None, tgens = None ):
+ wx.Frame.__init__ ( self, parent, id = wx.ID_ANY, title = u'Tgen Creator', pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE|wx.TAB_TRAVERSAL|wx.FRAME_FLOAT_ON_PARENT|wx.STAY_ON_TOP )
+
+ self.tgens = tgens
+ self.edit = False
+ self.parent = parent
+ self.SetSizeHintsSz( wx.DefaultSize, wx.DefaultSize )
+
+ bSizer2 = wx.BoxSizer( wx.VERTICAL )
+
+ fgSizer3 = wx.FlexGridSizer( 0, 2, 0, 0 )
+ fgSizer3.SetFlexibleDirection( wx.BOTH )
+ fgSizer3.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
+
+ self.m_staticText3 = wx.StaticText( self, wx.ID_ANY, u"Name", wx.DefaultPosition, wx.DefaultSize, 0 )
+ self.m_staticText3.Wrap( -1 )
+ fgSizer3.Add( self.m_staticText3, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALIGN_LEFT|wx.ALL, 5 )
+
+ self.m_textCtrl1 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0 )
+ self.m_textCtrl1.SetMinSize( wx.Size( 250,-1 ) )
+
+
+ fgSizer3.Add( self.m_textCtrl1, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+
+ bSizer2.Add( fgSizer3, 1, wx.ALIGN_CENTER_HORIZONTAL, 5 )
+
+ self.ip = VItemsPicker(self,-1, lemlist,'Forms', 'Selected')
+ self.ip._source.SetMinSize( wx.Size( 250, 400 ) )
+
+ bSizer2.Add( self.ip, 0, wx.ALL, 5 )
+
+ m_sdbSizer3 = wx.StdDialogButtonSizer()
+ self.m_sdbSizer3OK = wx.Button( self, wx.ID_OK )
+ m_sdbSizer3.AddButton( self.m_sdbSizer3OK )
+ self.m_sdbSizer3Cancel = wx.Button( self, wx.ID_CANCEL )
+ m_sdbSizer3.AddButton( self.m_sdbSizer3Cancel )
+ m_sdbSizer3.Realize();
+
+ bSizer2.Add( m_sdbSizer3, 1, wx.EXPAND, 5 )
+
+
+ self.SetSizer( bSizer2 )
+ self.Layout()
+ bSizer2.Fit( self )
+
+ self.Centre( wx.BOTH )
+
+ self.m_textCtrl1.Bind( wx.EVT_TEXT, self.OnTextEnter )
+ self.ip.Bind(EVT_IP_SELECTION_CHANGED, self.OnSelectionChange)
+ self.m_sdbSizer3OK.Bind(wx.EVT_BUTTON, self.OnClose)
+ self.m_sdbSizer3Cancel.Bind(wx.EVT_BUTTON, self.OnCancel)
+
+ #self.ip.SetItems(lemlist)
+ self.m_sdbSizer3OK.Enable(False)
+
+ if tgen is not None :
+ self.m_textCtrl1.SetValue(tgen)
+ self.ip._destData = dict([[i,[word,'']] for i, word in enumerate(tgens[tgen])])
+ self.ip._SetDestItems()
+ #self.ip.SetSelections(tgens[tgen])
+ self.m_sdbSizer3OK.Enable(True)
+ self.edit = True
+ else :
+ self.edit = False
+
+ def __del__( self ):
+ pass
+
+ def OnTextEnter(self, evt):
+ if self.m_textCtrl1.GetValue() != '' and self.m_textCtrl1.GetValue() not in self.tgens and self.ip.GetSelections() != []:
+ self.m_sdbSizer3OK.Enable(True)
+ else :
+ self.m_sdbSizer3OK.Enable(False)
+ if self.m_textCtrl1.GetValue() != '' and self.ip.GetSelections() and self.edit:
+ self.m_sdbSizer3OK.Enable(True)
+
+ def OnSelectionChange(self, evt):
+ if self.ip.GetSelections() != [] and self.m_textCtrl1.GetValue() != '' and self.m_textCtrl1.GetValue() not in self.tgens :
+ self.m_sdbSizer3OK.Enable(True)
+ else :
+ self.m_sdbSizer3OK.Enable(False)
+ if self.m_textCtrl1.GetValue() != '' and self.ip.GetSelections() and self.edit:
+ self.m_sdbSizer3OK.Enable(True)
+
+ def OnClose(self, evt):
+ self.Close()
+
+ def OnCancel(self, evt):
+ self.Destroy()
+
+class TGenFrame ( wx.Frame ):
+
+ def __init__( self, parent, corpus, Tgen ):
+ wx.Frame.__init__ ( self, parent, id = wx.ID_ANY, title = u"Tgen", pos = wx.DefaultPosition, size = wx.Size( 600,434 ), style = wx.DEFAULT_FRAME_STYLE|wx.TAB_TRAVERSAL|wx.FRAME_FLOAT_ON_PARENT|wx.STAY_ON_TOP )
+
+ self.Tgen = Tgen
+ self.parent = parent
+ self.corpus = corpus
+ self.activetgen = None
+
+ self.SetSizeHintsSz( wx.DefaultSize, wx.DefaultSize )
+
+ bSizer1 = wx.BoxSizer( wx.VERTICAL )
+
+ fgSizer1 = wx.FlexGridSizer( 0, 2, 0, 0 )
+ fgSizer1.SetFlexibleDirection( wx.BOTH )
+ fgSizer1.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
+
+ self.m_staticText1 = wx.StaticText( self, wx.ID_ANY, u"Tgen", wx.DefaultPosition, wx.DefaultSize, 0 )
+ self.m_staticText1.Wrap( -1 )
+ fgSizer1.Add( self.m_staticText1, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ self.m_staticText2 = wx.StaticText( self, wx.ID_ANY, u"Content", wx.DefaultPosition, wx.DefaultSize, 0 )
+ self.m_staticText2.Wrap( -1 )
+ fgSizer1.Add( self.m_staticText2, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ tgensChoices = Tgen.tgen.keys()
+ self.tgens = wx.ListBox( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, tgensChoices, 0 )
+ self.tgens.SetMinSize( wx.Size( 200,250 ) )
+
+ fgSizer1.Add( self.tgens, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ tgencontentChoices = []
+ self.tgencontent = wx.ListBox( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, tgencontentChoices, 0|wx.VSCROLL )
+ self.tgencontent.SetMinSize( wx.Size( 200,250 ) )
+
+ fgSizer1.Add( self.tgencontent, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ fgSizer2 = wx.FlexGridSizer( 0, 2, 0, 0 )
+ fgSizer2.SetFlexibleDirection( wx.BOTH )
+ fgSizer2.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
+
+ self.but_new = wx.Button( self, wx.ID_ANY, u"New...", wx.DefaultPosition, wx.DefaultSize, 0 )
+ fgSizer2.Add( self.but_new, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ self.but_del = wx.Button( self, wx.ID_ANY, u"Delete", wx.DefaultPosition, wx.DefaultSize, 0 )
+ fgSizer2.Add( self.but_del, 0, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+
+ fgSizer1.Add( fgSizer2, 0, wx.EXPAND, 5 )
+
+ fgSizer3 = wx.FlexGridSizer( 0, 2, 0, 0 )
+ fgSizer3.SetFlexibleDirection( wx.BOTH )
+ fgSizer3.SetNonFlexibleGrowMode( wx.FLEX_GROWMODE_SPECIFIED )
+ self.but_edit = wx.Button( self, wx.ID_ANY, u"Edit", wx.DefaultPosition, wx.DefaultSize, 0 )
+ fgSizer3.Add( self.but_edit, 1, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ self.but_compute = wx.Button( self, wx.ID_ANY, u"Compute", wx.DefaultPosition, wx.DefaultSize, 0 )
+ fgSizer3.Add( self.but_compute, 1, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ fgSizer1.Add( fgSizer3, 0, wx.EXPAND, 5 )
+
+ bSizer1.Add( fgSizer1, 1, wx.ALIGN_CENTER_HORIZONTAL|wx.ALIGN_CENTER_VERTICAL|wx.ALL, 5 )
+
+ m_sdbSizer2 = wx.StdDialogButtonSizer()
+ self.m_sdbSizer2OK = wx.Button( self, wx.ID_OK )
+ m_sdbSizer2.AddButton( self.m_sdbSizer2OK )
+ #self.m_sdbSizer2Cancel = wx.Button( self, wx.ID_CANCEL )
+ #m_sdbSizer2.AddButton( self.m_sdbSizer2Cancel )
+ m_sdbSizer2.Realize();
+
+ bSizer1.Add( m_sdbSizer2, 0, wx.EXPAND, 5 )
+
+
+ self.SetSizer( bSizer1 )
+ self.Layout()
+
+ self.Centre( wx.BOTH )
+
+ # Connect Events
+ self.tgens.Bind( wx.EVT_LISTBOX, self.GetContent )
+ self.but_new.Bind( wx.EVT_BUTTON, self.OnNewTgen )
+ self.but_del.Bind( wx.EVT_BUTTON, self.OnDeleteTgen )
+ self.but_edit.Bind( wx.EVT_BUTTON, self.OnEditTgen )
+ self.but_compute.Bind(wx.EVT_BUTTON, self.OnCompute)
+ self.m_sdbSizer2OK.Bind( wx.EVT_BUTTON, self.OnOK )
+
+ def __del__( self ):
+ pass
+
+ def GetContent( self, event ):
+ tgen = event.GetString()
+ if tgen != '' :
+ self.tgencontent.Clear()
+ for word in self.Tgen[tgen] :
+ self.tgencontent.Append(word)
+
+ def OnNewTgen( self, event, tgen = None ):
+ if tgen is None :
+ self.dial = CreateTgenDialog(self, dict([[i, [lem, self.corpus.lems[lem].freq]] for i, lem in enumerate(self.corpus.lems.keys())]), tgens = self.Tgen.tgen)
+ else :
+ self.dial = CreateTgenDialog(self, dict([[i, [lem, self.corpus.lems[lem].freq]] for i, lem in enumerate(self.corpus.lems.keys())]), tgen = tgen, tgens = self.Tgen.tgen)
+ self.dial.ip._source.selected = dict(zip(self.Tgen.tgen[tgen], self.Tgen.tgen[tgen]))
+ self.activetgen = tgen
+ self.dial.Show()
+ self.dial.Bind(wx.EVT_CLOSE, self.OnDialClose)
+
+ def OnDeleteTgen( self, event ):
+ if self.tgens.GetSelection() != -1 :
+ tgens = self.tgens.GetItems()
+ del self.Tgen.tgen[tgens[self.tgens.GetSelection()]]
+ self.Tgen.write()
+ self.tgens.Clear()
+ self.tgencontent.Clear()
+ for val in self.Tgen.tgen :
+ self.tgens.Append(val)
+ event.Skip()
+
+ def OnEditTgen( self, event ):
+ if self.tgens.GetSelection() != -1 :
+ tgens = self.tgens.GetItems()
+ tgen = tgens[self.tgens.GetSelection()]
+ self.activetgen = tgen
+ self.dial = CreateTgenDialog(self, dict([[i, [lem, self.corpus.lems[lem].freq]] for i, lem in enumerate(self.corpus.lems.keys())]), tgen = tgen, tgens = self.Tgen.tgen)
+ self.dial.Bind(wx.EVT_CLOSE, self.OnDialClose)
+ self.dial.m_textCtrl1.Enable(False)
+ self.dial.ip._source.selected = dict(zip(self.Tgen.tgen[tgen], self.Tgen.tgen[tgen]))
+ self.dial.Show()
+ event.Skip()
+
+ def OnCompute(self, evt):
+ ira = wx.GetApp().GetTopWindow()
+ ira.tree.OnTgenCompute(evt)
+
+ def OnOK(self, evt):
+ self.Destroy()
+
+ def OnDialClose(self, evt):
+ if self.dial.edit :
+ del self.Tgen.tgen[self.activetgen]
+ self.tgens.Clear()
+ self.tgencontent.Clear()
+ for val in self.Tgen.tgen :
+ self.tgens.Append(val)
+ self.Tgen.tgen[self.dial.m_textCtrl1.GetValue()] = self.dial.ip.GetSelections()
+ self.Tgen.write()
+ self.tgens.Append(self.dial.m_textCtrl1.GetValue())
+ self.dial.Destroy()
+ evt.Skip()