基于trie树的具有联想功能的文本编辑器
之前的软件设计与开发实践课程中，自己构思的大作业题目。做的具有核心功能，但是还欠缺边边角角的小功能和持久化数据结构，先放出来，有机会一点点改。github：https://github.com/chuxiuhong/smarteditor
数据结构，使用过程截图以及源代码如下：
#数据结构
**trie树**
trie树相应的介绍点击链接 https://en.wikipedia.org/wiki/Trie
trie树在python文件中的类型定义
Node定义
#GUI设计界面
首先，用较大的文本文件进行训练，完成trie的树的训练。本文件夹下提供了一个big2.txt文件作为示例。
在之后的输入会弹出提示框
#!/usr/bin/python # -*- coding: utf-8 -*- import PyQt4 from PyQt4.QtCore import * from PyQt4.QtGui import * import PyQt4.QtCore import sys import pickle print '===================start=======================' class Node: def __init__(self): self.value = None self.children = {} # children is of type {char, Node} self.fre = 0 self.father = None def CMP(a, b): return b.fre - a.fre class Trie: def __init__(self): self.root = Node() self.choose = [] self.__OpenCorrect__ = 0 def insert(self, key): # key is of type string # key should be a low-case string, this must be checked here! node = self.root for char in key: if char not in node.children: child = Node() node.children[char] = child child.value = char child.father = node node = child else: node = node.children[char] # node.value = key node.fre += 1 def search(self, key): node = self.root for char in key: if char not in node.children: return None else: node = node.children[char] return node.value def display_node(self, node): if (node.value != None): print node.value for char in 'abcdefghijklmnopqrstuvwxyz': if char in node.children: self.display_node(node.children[char]) return def fallback(self, node): f_char = '' while node != self.root: f_char = node.value + f_char node = node.father # print f_char return f_char def display(self): self.display_node(self.root) def find_node(self, string): res_node = self.root for i in string: res_node = res_node.children[i] return res_node def association(self, node): # 调用此函数前应该先将self.choose恢复成空列表 if (node.value != None): if node.fre > 0: self.choose.append(node) for char in 'abcdefghijklmnopqrstuvwxyz': if char in node.children: self.association(node.children[char]) def output_association(self, char): char = str(char).lower() self.choose = [] result_list = [] self.association(self.find_node(char)) self.choose.sort(cmp=CMP) if len(self.choose) > 0: for i in self.choose: result_list.append(self.fallback(i)) if self.__OpenCorrect__ == 0: result_list.insert(1, self.correct(char)) # print 'result_list',result_list return result_list def correct(self, string): self.choose = [] p = self.find_node(string[:-1]) self.association(p) self.choose.sort(cmp=CMP) if len(self.choose) > 1: return self.fallback(self.choose[0]) def train(trie, path): # f = open(r'big2.txt') f = open(path) word = f.read() f.close() word = word.split(' ') for i in word: trie.insert(i) trie = Trie() trie.__OpenCorrect__ = 0 def save_model(T): f1 = open("trie.pkl", 'wb') pickle.dump(T, f1) f1.close() def load_model(path): f2 = open(path, 'rb') trie = pickle.load(f2) f2.close() print '================= END =====================' class UI(QDialog): def __init__(self, parent=None): super(UI, self).__init__(parent) QSList = QStringList() # default # QSList<<'One'<<'Tow'<<'Three'<<'Four'<<'Five' # instance of Completer class cmp = Completer(QSList) global edit edit = TextEdit() edit.setFontPointSize(80) edit.setCompleter(cmp) self.setWindowTitle(u"智能文本编辑器") button1 = QPushButton(u"训练模型") button2 = QPushButton(u"保存文本文件") button3 = QPushButton(u"打开文本文件") ''' buttons = QushButton() ''' ''' 定义按钮，（）内为按钮名称 ''' self.connect(button1, SIGNAL("clicked()"), self.get_file) self.connect(button2, SIGNAL("clicked()"), self.func2) self.connect(button3, SIGNAL("clicked()"), self.func3) ''' 关联按钮与函数格式同上，触发为clicked()单击，最后一个参数为 类内函数 ''' layout = QGridLayout() layout.addWidget(edit, 0, 1, 1, 5) layout.addWidget(button1, 2, 1) layout.addWidget(button2, 2, 3) layout.addWidget(button3, 2, 5) ''' 按钮布局 ''' self.setLayout(layout) self.center() def center(self): screen = QDesktopWidget().screenGeometry() size = self.geometry() self.move((screen.width() - size.width()) / 2, (screen.height() - size.height()) / 2) def get_file(self): s = QFileDialog.getOpenFileName(self, "Open file dialog", "/", "TXT Files(*.txt)") train(trie, s) def func2(self): s = QFileDialog.getSaveFileName(self, "文件保存", "C:/", "All Files (*);Text Files (*.txt)") f = open(s, 'w') f.write(edit.toPlainText()) f.close() def func3(self): s = QFileDialog.getOpenFileName(self, "Open file dialog", "/", "TXT Files(*.txt)") f = open(s) edit.setText(PyQt4.QtCore.QString(f.read())) f.close() class TextEdit(QTextEdit): def __init__(self, parent=None): super(TextEdit, self).__init__(parent) self.cmp = None self.p = '' self.count = 0 def setCompleter(self, completer): if self.cmp: self.disconnect(self.cmp, 0, 0) self.cmp = completer if (not self.cmp): return self.cmp.setWidget(self) self.cmp.setCompletionMode(QCompleter.PopupCompletion) self.cmp.setCaseSensitivity(Qt.CaseInsensitive) self.connect(self.cmp, SIGNAL('activated(QString)'), self.insertCompletion) def completer(self): return self.cmp def insertCompletion(self, string): # get cursor position tc = self.textCursor() # selectd ranges tc.movePosition(QTextCursor.StartOfWord, QTextCursor.KeepAnchor) # replace selected ranges tc.insertText(string) self.p += str(string) # set cursor pos back to original pos self.setTextCursor(tc) def textUnderCursor(self): tc = self.textCursor() tc.select(QTextCursor.WordUnderCursor) return tc.selectedText() def keyPressEvent(self, e): ''' if e.key() != Qt.Key_Backspace: self.p = self.p + e.text() self.count+=1 print 'yes' else: self.p = self.p[:-1] self.count-=1 ''' print 'p ', self.p print 'pressed >> ', e.text() if (self.cmp and self.cmp.popup().isVisible()): if e.key() in (Qt.Key_Enter, Qt.Key_Return, Qt.Key_Escape, Qt.Key_Tab, Qt.Key_Backtab): e.ignore() return isShortcut = ((e.modifiers() & Qt.ControlModifier) and e.key() == Qt.Key_E) if (not self.cmp or not isShortcut): super(TextEdit, self).keyPressEvent(e) ctrlOrShift = e.modifiers() & (Qt.ControlModifier | Qt.ShiftModifier) if (not self.cmp or (ctrlOrShift and e.text().isEmpty())): return eow = QString("~!@#$%^&*()_+{}|:\"<>?,./;'[]\\-=") hasModifier = (e.modifiers() != Qt.NoModifier) and not ctrlOrShift completionPrefix = self.textUnderCursor() # hide popup while matching invalid cases if (not isShortcut and (hasModifier or e.text().isEmpty() or completionPrefix.length() < 1 or eow.contains(e.text().right(1)))): self.cmp.popup().hide() return self.cmp.update(completionPrefix) self.cmp.popup().setCurrentIndex(self.cmp.completionModel().index(0, 0)) cr = self.cursorRect() cr.setWidth(self.cmp.popup().sizeHintForColumn(0) + self.cmp.popup().verticalScrollBar().sizeHint().width()) word = self.p.split(' ')[:-1] self.p = '' for k in xrange(len(word)): self.p += word[k] print 'self.p=', self.p self.cmp.complete(cr) class Completer(QCompleter): def __init__(self, stringlist, parent=None): super(Completer, self).__init__(parent) self.stringlist = stringlist self.setModel(QStringListModel()) # update function will trigger while the text has been modified def update(self, completionText): # generate a new QStringList instance qsList = QStringList() # generate hint lists which returns by customatic definitions newList = genMyStrList(completionText) for item in newList: qsList.append(item) self.stringlist = qsList # filteredList = self.stringlist.filter(completionText, Qt.CaseInsensitive) self.model().setStringList(self.stringlist) self.popup().setCurrentIndex(self.model().index(0, 0)) # the function below defined a way to generate a string list def genMyStrList(key): my_str_list = trie.output_association(key) return my_str_list def main(): app = QApplication(sys.argv) app.processEvents() form = UI() form.show() # splash.finish(form) app.exec_() window = QMainWindow() window.statusBar().showMessage('Yuan Ziqi') menubar = window.menuBar window.setWindowTitle('Auto Complete Demo') window.resize(400, 200) window.move(400, 100) edit = TextEdit() edit.setFontPointSize(40) edit.setCompleter(cmp) # bt = addbutton() # window.addDockWidget(bt) window.setCentralWidget(edit) window.show() sys.exit(app.exec_()) if __name__ == '__main__': print 'app is running successfully' main()
新博客迁移至https://chuxiuhong.com/
