class RuleEditor(QMainWindow):
###########################################################################
def __init__(self, parent=None):
super(RuleEditor, self).__init__(parent)
self.currentElementFile = None
self.currentRuleFile = None
self.sceneElementList = []
self.connections = []
self.elementSet = ElementSet()
self.rule = Rule()
self.ruleSet = RuleSet()
# Frames
self.gBox1 = QFrame()
self.gBox1.setMaximumSize(QSize(250, 600))
self.scene = ChainScene(-200, -200, 400, 400)
self.scene.itemInserted.connect(self.itemInserted)
# Graphics view
self.graphicsView = QGraphicsView(self.scene)
self.graphicsView.setMinimumSize(QSize(800, 600))
#Toolbox
self.createToolBox()
self.loadToolBox()
# Left menu
self.leftLayout = QVBoxLayout(self.gBox1)
self.leftInnerLayout = QHBoxLayout()
self.leftInnerLayout2 = QHBoxLayout()
self.label1 = QLabel("Elements file:")
self.fileName = QLabel("--")
self.chooseFile = QPushButton("Choose file")
self.loadFile = QPushButton("Load")
self.ruleListWidget = QListWidget()
self.addRule = QPushButton("Add rule")
self.deleteRule = QPushButton("Delete")
self.leftLayout.addWidget(self.label1)
self.leftLayout.addWidget(self.fileName)
self.leftInnerLayout.addWidget(self.chooseFile)
self.leftInnerLayout.addWidget(self.loadFile)
self.leftLayout.addLayout(self.leftInnerLayout)
self.leftLayout.addWidget(self.ruleListWidget)
self.leftInnerLayout2.addWidget(self.addRule)
self.leftInnerLayout2.addWidget(self.deleteRule)
self.leftLayout.addLayout(self.leftInnerLayout2)
self.leftLayout.addWidget(self.toolBox)
self.chooseFile.clicked.connect(self.chooseElementsFile)
self.loadFile.clicked.connect(self.loadElementsFile)
self.addRule.clicked.connect(self.newRule)
self.deleteRule.clicked.connect(self.removeRule)
self.ruleListWidget.currentRowChanged[int].connect(
self.changeCurrentRule)
# Layout
self.centralWidget = QWidget()
self.centralLayout = QHBoxLayout()
self.centralLayout.addWidget(self.gBox1)
self.centralLayout.addWidget(self.graphicsView)
self.centralWidget.setLayout(self.centralLayout)
self.setCentralWidget(self.centralWidget)
self.resize(1200, 600)
self.setWindowTitle("WordChain : Rule editor")
# ToolBars
self.fileToolBar = self.addToolBar("File")
self.saveAction = QAction(QIcon(':/images/save.png'),
"Save to &File",
self,
shortcut="Ctrl+S",
triggered=self.saveToFile)
self.openAction = QAction(QIcon(':/images/open.png'),
"Open &File",
self,
shortcut="Ctrl+O",
triggered=self.loadFromFile)
self.fileToolBar.addAction(self.saveAction)
self.fileToolBar.addAction(self.openAction)
self.pointerToolbar = self.addToolBar("Pointer type")
pointerButton = QToolButton()
pointerButton.setCheckable(True)
pointerButton.setChecked(True)
pointerButton.setIcon(QIcon(":/images/arrow.png"))
lineButton = QToolButton()
lineButton.setCheckable(True)
lineButton.setIcon(QIcon(":/images/line.png"))
self.pointerTypeGroup = QButtonGroup()
self.pointerTypeGroup.addButton(pointerButton, ChainScene.MoveItem)
self.pointerTypeGroup.addButton(lineButton, ChainScene.InsertLine)
self.pointerTypeGroup.buttonClicked[int].connect(
self.pointerGroupClicked)
self.pointerToolbar.addWidget(pointerButton)
self.pointerToolbar.addWidget(lineButton)
self.testToolbar = self.addToolBar("Model testing")
testButton = QPushButton()
testButton.setIcon(QIcon(":/images/play.png"))
testButton.setText("Test rule")
testButton.clicked.connect(self.testRule)
self.testToolbar.addWidget(testButton)
###########################################################################
def createToolBox(self):
self.toolBox = QToolBox()
self.toolBox.setMinimumSize(QSize(250, 600))
self.buttonGroup = QButtonGroup()
self.buttonGroup.setExclusive(False)
self.buttonGroup.buttonClicked[int].connect(self.buttonGroupClicked)
self.toolBoxLayout = QGridLayout()
self.toolBoxLayout.setRowStretch(4, 10)
self.toolBoxLayout.setColumnStretch(2, 10)
self.toolBoxWidget = QWidget()
self.toolBoxWidget.setLayout(self.toolBoxLayout)
self.toolBox.addItem(self.toolBoxWidget, "Grammar elements")
###########################################################################
def buttonGroupClicked(self, i):
if i == len(self.elementSet.elementList):
pass
else:
el = self.elementSet.elementList[i]
self.scene.setToInert(el)
self.scene.setMode(ChainScene.InsertItem)
buttons = self.buttonGroup.buttons()
for button in buttons:
if self.buttonGroup.button(i) != button:
button.setChecked(False)
###########################################################################
def itemInserted(self, item):
self.pointerTypeGroup.button(ChainScene.MoveItem).setChecked(True)
self.scene.setMode(self.pointerTypeGroup.checkedId())
buttons = self.buttonGroup.buttons()
for button in buttons:
button.setChecked(False)
self.rule.fromElementList(self.scene.items())
###########################################################################
def pointerGroupClicked(self, i):
self.scene.setMode(self.pointerTypeGroup.checkedId())
###########################################################################
def layoutToolBox(self):
self.clearToolBox()
self.loadToolBox()
###########################################################################
def loadToolBox(self):
rowNum = 0
colNum = 0
num = 0
for elem in self.elementSet.elementList:
self.toolBoxLayout.addWidget(
self.createCellWidget(elem.elementName, elem.image(), num),
rowNum, colNum)
colNum += 1
num += 1
if colNum > 1:
colNum = 0
rowNum += 1
###########################################################################
def clearToolBox(self):
cnt = self.toolBoxLayout.count()
while cnt > 0:
itm = self.toolBoxLayout.itemAt(cnt - 1)
if itm is not None:
wdg = itm.widget()
self.toolBoxLayout.removeWidget(wdg)
wdg.hide()
wdg.deleteLater()
cnt -= 1
###########################################################################
def createCellWidget(self, text, elementImage, num):
icon = QIcon(elementImage)
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.buttonGroup.addButton(button, num)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignHCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
###########################################################################
def saveToFile(self):
self.rule.fromElementList(self.scene.items())
if self.currentRuleFile is not None:
saveFile = QSaveFile(self.currentRuleFile)
saveFile.open(QIODevice.WriteOnly)
data = self.ruleSet.toBytes()
saveFile.writeData(data)
saveFile.commit()
else:
self.saveAs()
self.setWindowTitle("WordChain : Rule editor : " +
self.currentRuleFile)
###########################################################################
def saveAs(self):
ret = QFileDialog.getSaveFileName(filter='*.rl')
self.currentRuleFile = ret[0]
self.saveToFile()
self.setWindowTitle("WordChain : Rule editor : " +
self.currentRuleFile)
###########################################################################
def loadFromFile(self):
ret = QFileDialog.getOpenFileName(filter='*.rl')
self.currentRuleFile = ret[0]
file = open(self.currentRuleFile, 'r')
st = '\n'.join(file.readlines())
self.ruleSet.fromString(st)
if len(self.scene.items()) > 0:
for item in self.scene.items():
self.scene.removeItem(item)
self.currentElementFile = self.ruleSet.elementsFileName
self.fileName.setText(os.path.basename(self.currentElementFile))
self.loadElementsFile()
self.scene.setMode(ChainScene.MoveItem)
self.updateRuleList()
self.changeCurrentRule(0)
self.setWindowTitle("WordChain : Rule editor : " +
self.currentRuleFile)
###########################################################################
def showWindow(self):
self.show()
self.updateRuleList()
def updateRuleList(self):
self.ruleListWidget.clear()
rowCount = 0
for rule in self.ruleSet.ruleList:
self.ruleListWidget.insertItem(rowCount, rule.name)
rowCount += 1
def newRule(self):
name, ok = QInputDialog.getText(None, "Create new rule",
"Enter new rule name:")
if name is not None and ok:
self.rule = Rule()
self.rule.setName(name)
self.ruleSet.addRule(self.rule)
self.updateRuleList()
self.changeCurrentRule(len(self.ruleSet.ruleList) - 1)
self.ruleListWidget.setCurrentRow(len(self.ruleSet.ruleList) - 1)
def changeCurrentRule(self, idx):
self.rule.fromElementList(self.scene.items())
self.rule = self.ruleSet.getRule(idx)
self.loadRuleToScene()
def removeRule(self):
self.ruleSet.deleteRule(self.ruleListWidget.currentRow())
self.updateRuleList()
def loadRuleToScene(self):
self.scene.clear()
for ruleElem in self.rule.elements:
for el in self.elementSet.elementList:
if el.elementName == ruleElem['elementName']:
item = MetaElement(el.leftConnectorType,
el.rightConnectorType, el.elementName,
el.color)
item.addProperties(el.properties)
self.scene.addItem(item)
item.setPos(QPoint(ruleElem['x'], ruleElem['y']))
for ruleConn in self.rule.connections:
el1 = self.scene.items()[ruleConn['p1']]
el2 = self.scene.items()[ruleConn['p2']]
name = ruleConn.get('name', '')
conn = self.scene.addConnection(el2, el1, name)
conn.p1Properties = ruleConn['p1Properties']
conn.p2Properties = ruleConn['p2Properties']
self.scene.addItem(conn)
###########################################################################
def chooseElementsFile(self):
ret = QFileDialog.getOpenFileName(filter='*.elms')
self.currentElementFile = ret[0]
self.ruleSet.setElementsFileName(self.currentElementFile)
self.fileName.setText(os.path.basename(self.currentElementFile))
###########################################################################
def loadElementsFile(self):
if self.currentElementFile is not None:
file = open(self.currentElementFile, 'r')
st = '\n'.join(file.readlines())
self.elementSet.fromString(st)
file.close()
self.layoutToolBox()
###########################################################################
def updateElementListView(self):
self.elemListWidget.clear()
for item in self.elementSet.elementList:
self.elemListWidget.addItem(item.elementName)
###########################################################################
def testRule(self):
self.rule.fromElementList(self.scene.items())
print(self.ruleSet.toString())
class RuleEditor(QMainWindow):
###########################################################################
def __init__(self, parent=None):
super(RuleEditor, self).__init__(parent)
self.currentElementFile = None
self.currentRuleFile = None
self.sceneElementList = []
self.connections = []
self.elementSet = ElementSet()
self.rule = Rule()
self.ruleSet = RuleSet()
# Frames
self.gBox1 = QFrame()
self.gBox1.setMaximumSize(QSize(250, 600))
self.scene = ChainScene(-200, -200, 400, 400)
self.scene.itemInserted.connect(self.itemInserted)
# Graphics view
self.graphicsView = QGraphicsView(self.scene)
self.graphicsView.setMinimumSize(QSize(800, 600))
#Toolbox
self.createToolBox()
self.loadToolBox()
# Left menu
self.leftLayout = QVBoxLayout(self.gBox1)
self.leftInnerLayout = QHBoxLayout()
self.leftInnerLayout2 = QHBoxLayout()
self.label1 = QLabel("Elements file:")
self.fileName = QLabel("--")
self.chooseFile = QPushButton("Choose file")
self.loadFile = QPushButton("Load")
self.ruleListWidget = QListWidget()
self.addRule = QPushButton("Add rule")
self.deleteRule = QPushButton("Delete")
self.leftLayout.addWidget(self.label1)
self.leftLayout.addWidget(self.fileName)
self.leftInnerLayout.addWidget(self.chooseFile)
self.leftInnerLayout.addWidget(self.loadFile)
self.leftLayout.addLayout(self.leftInnerLayout)
self.leftLayout.addWidget(self.ruleListWidget)
self.leftInnerLayout2.addWidget(self.addRule)
self.leftInnerLayout2.addWidget(self.deleteRule)
self.leftLayout.addLayout(self.leftInnerLayout2)
self.leftLayout.addWidget(self.toolBox)
self.chooseFile.clicked.connect(self.chooseElementsFile)
self.loadFile.clicked.connect(self.loadElementsFile)
self.addRule.clicked.connect(self.newRule)
self.deleteRule.clicked.connect(self.removeRule)
self.ruleListWidget.currentRowChanged[int].connect(self.changeCurrentRule)
# Layout
self.centralWidget = QWidget()
self.centralLayout = QHBoxLayout()
self.centralLayout.addWidget(self.gBox1)
self.centralLayout.addWidget(self.graphicsView)
self.centralWidget.setLayout(self.centralLayout)
self.setCentralWidget(self.centralWidget)
self.resize(1200, 600)
self.setWindowTitle("WordChain : Rule editor")
# ToolBars
self.fileToolBar = self.addToolBar("File")
self.saveAction = QAction(QIcon(':/images/save.png'), "Save to &File", self,
shortcut="Ctrl+S", triggered=self.saveToFile)
self.openAction = QAction(QIcon(':/images/open.png'), "Open &File", self,
shortcut="Ctrl+O", triggered=self.loadFromFile)
self.fileToolBar.addAction(self.saveAction)
self.fileToolBar.addAction(self.openAction)
self.pointerToolbar = self.addToolBar("Pointer type")
pointerButton = QToolButton()
pointerButton.setCheckable(True)
pointerButton.setChecked(True)
pointerButton.setIcon(QIcon(":/images/arrow.png"))
lineButton = QToolButton()
lineButton.setCheckable(True)
lineButton.setIcon(QIcon(":/images/line.png"))
self.pointerTypeGroup = QButtonGroup()
self.pointerTypeGroup.addButton(pointerButton, ChainScene.MoveItem)
self.pointerTypeGroup.addButton(lineButton, ChainScene.InsertLine)
self.pointerTypeGroup.buttonClicked[int].connect(self.pointerGroupClicked)
self.pointerToolbar.addWidget(pointerButton)
self.pointerToolbar.addWidget(lineButton)
self.testToolbar = self.addToolBar("Model testing")
testButton = QPushButton()
testButton.setIcon(QIcon(":/images/play.png"))
testButton.setText("Test rule")
testButton.clicked.connect(self.testRule)
self.testToolbar.addWidget(testButton)
###########################################################################
def createToolBox(self):
self.toolBox = QToolBox()
self.toolBox.setMinimumSize(QSize(250, 600))
self.buttonGroup = QButtonGroup()
self.buttonGroup.setExclusive(False)
self.buttonGroup.buttonClicked[int].connect(self.buttonGroupClicked)
self.toolBoxLayout = QGridLayout()
self.toolBoxLayout.setRowStretch(4, 10)
self.toolBoxLayout.setColumnStretch(2, 10)
self.toolBoxWidget = QWidget()
self.toolBoxWidget.setLayout(self.toolBoxLayout)
self.toolBox.addItem(self.toolBoxWidget, "Grammar elements")
###########################################################################
def buttonGroupClicked(self, i):
if i == len(self.elementSet.elementList):
pass
else:
el = self.elementSet.elementList[i]
self.scene.setToInert(el)
self.scene.setMode(ChainScene.InsertItem)
buttons = self.buttonGroup.buttons()
for button in buttons:
if self.buttonGroup.button(i) != button:
button.setChecked(False)
###########################################################################
def itemInserted(self, item):
self.pointerTypeGroup.button(ChainScene.MoveItem).setChecked(True)
self.scene.setMode(self.pointerTypeGroup.checkedId())
buttons = self.buttonGroup.buttons()
for button in buttons:
button.setChecked(False)
self.rule.fromElementList(self.scene.items())
###########################################################################
def pointerGroupClicked(self, i):
self.scene.setMode(self.pointerTypeGroup.checkedId())
###########################################################################
def layoutToolBox(self):
self.clearToolBox()
self.loadToolBox()
###########################################################################
def loadToolBox(self):
rowNum = 0
colNum = 0
num = 0
for elem in self.elementSet.elementList:
self.toolBoxLayout.addWidget(self.createCellWidget(elem.elementName, elem.image(), num), rowNum, colNum)
colNum += 1
num += 1
if colNum > 1:
colNum = 0
rowNum += 1
###########################################################################
def clearToolBox(self):
cnt = self.toolBoxLayout.count()
while cnt > 0:
itm = self.toolBoxLayout.itemAt(cnt - 1)
if itm is not None:
wdg = itm.widget()
self.toolBoxLayout.removeWidget(wdg)
wdg.hide()
wdg.deleteLater()
cnt -= 1
###########################################################################
def createCellWidget(self, text, elementImage, num):
icon = QIcon(elementImage)
button = QToolButton()
button.setIcon(icon)
button.setIconSize(QSize(50, 50))
button.setCheckable(True)
self.buttonGroup.addButton(button, num)
layout = QGridLayout()
layout.addWidget(button, 0, 0, Qt.AlignHCenter)
layout.addWidget(QLabel(text), 1, 0, Qt.AlignHCenter)
widget = QWidget()
widget.setLayout(layout)
return widget
###########################################################################
def saveToFile(self):
self.rule.fromElementList(self.scene.items())
if self.currentRuleFile is not None:
saveFile = QSaveFile(self.currentRuleFile)
saveFile.open(QIODevice.WriteOnly)
data = self.ruleSet.toBytes()
saveFile.writeData(data)
saveFile.commit()
else:
self.saveAs()
self.setWindowTitle("WordChain : Rule editor : " + self.currentRuleFile)
###########################################################################
def saveAs(self):
ret = QFileDialog.getSaveFileName(filter='*.rl')
self.currentRuleFile = ret[0]
self.saveToFile()
self.setWindowTitle("WordChain : Rule editor : " + self.currentRuleFile)
###########################################################################
def loadFromFile(self):
ret = QFileDialog.getOpenFileName(filter='*.rl')
self.currentRuleFile = ret[0]
file = open(self.currentRuleFile, 'r')
st = '\n'.join(file.readlines())
self.ruleSet.fromString(st)
if len(self.scene.items()) > 0:
for item in self.scene.items():
self.scene.removeItem(item)
self.currentElementFile = self.ruleSet.elementsFileName
self.fileName.setText(os.path.basename(self.currentElementFile))
self.loadElementsFile()
self.scene.setMode(ChainScene.MoveItem)
self.updateRuleList()
self.changeCurrentRule(0)
self.setWindowTitle("WordChain : Rule editor : " + self.currentRuleFile)
###########################################################################
def showWindow(self):
self.show()
self.updateRuleList()
def updateRuleList(self):
self.ruleListWidget.clear()
rowCount = 0
for rule in self.ruleSet.ruleList:
self.ruleListWidget.insertItem(rowCount, rule.name)
rowCount += 1
def newRule(self):
name, ok = QInputDialog.getText(None, "Create new rule", "Enter new rule name:")
if name is not None and ok:
self.rule = Rule()
self.rule.setName(name)
self.ruleSet.addRule(self.rule)
self.updateRuleList()
self.changeCurrentRule(len(self.ruleSet.ruleList) - 1)
self.ruleListWidget.setCurrentRow(len(self.ruleSet.ruleList) - 1)
def changeCurrentRule(self, idx):
self.rule.fromElementList(self.scene.items())
self.rule = self.ruleSet.getRule(idx)
self.loadRuleToScene()
def removeRule(self):
self.ruleSet.deleteRule(self.ruleListWidget.currentRow())
self.updateRuleList()
def loadRuleToScene(self):
self.scene.clear()
for ruleElem in self.rule.elements:
for el in self.elementSet.elementList:
if el.elementName == ruleElem['elementName']:
item = MetaElement(el.leftConnectorType,
el.rightConnectorType,
el.elementName,
el.color)
item.addProperties(el.properties)
self.scene.addItem(item)
item.setPos(QPoint(ruleElem['x'], ruleElem['y']))
for ruleConn in self.rule.connections:
el1 = self.scene.items()[ruleConn['p1']]
el2 = self.scene.items()[ruleConn['p2']]
name = ruleConn.get('name', '')
conn = self.scene.addConnection(el2, el1, name)
conn.p1Properties = ruleConn['p1Properties']
conn.p2Properties = ruleConn['p2Properties']
self.scene.addItem(conn)
###########################################################################
def chooseElementsFile(self):
ret = QFileDialog.getOpenFileName(filter='*.elms')
self.currentElementFile = ret[0]
self.ruleSet.setElementsFileName(self.currentElementFile)
self.fileName.setText(os.path.basename(self.currentElementFile))
###########################################################################
def loadElementsFile(self):
if self.currentElementFile is not None:
file = open(self.currentElementFile, 'r')
st = '\n'.join(file.readlines())
self.elementSet.fromString(st)
file.close()
self.layoutToolBox()
###########################################################################
def updateElementListView(self):
self.elemListWidget.clear()
for item in self.elementSet.elementList:
self.elemListWidget.addItem(item.elementName)
###########################################################################
def testRule(self):
self.rule.fromElementList(self.scene.items())
print(self.ruleSet.toString())
