class MainWindow(QWidget):
def __init__(self, load_filepath, save_file_prefix, map_params):
super().__init__()
self.json_save_fpath = save_file_prefix + '.json'
self.world_save_fpath = save_file_prefix + '.world'
self.setWindowTitle(self.json_save_fpath)
self.show()
self.model = Model(map_params, load_filepath)
layout = QGridLayout()
layout.setSpacing(5)
self.setLayout(layout)
generateButton = QPushButton('Generate JSON/WORLD', self)
generateButton.setFixedSize(QSize(200, 25))
generateButton.pressed.connect(self.generateOutputFiles)
self.ctrl_grp = QButtonGroup()
self.ctrl_grp.setExclusive(True)
# TODO - maybe must be not "model" but "controller" connected to buttons
mode_buttons = [
(ModeButton('1. Create walls', Mode.WALLS, self.model,
self), GuiWallsMode(self.model, ObjectType.WALL)),
(ModeButton('2. Create doors', Mode.DOORS, self.model,
self), GuiDoorsMode(self.model, ObjectType.DOOR)),
(ModeButton('3. Create windows', Mode.WINDOWS, self.model,
self), GuiWindowsMode(self.model, ObjectType.WINDOW)),
(ModeButton('4. Create boxes', Mode.BOXES, self.model,
self), GuiBoxesMode(self.model, ObjectType.BOX)),
(ModeButton('5. Create signs', Mode.SIGNS, self.model,
self), GuiSignsMode(self.model, ObjectType.SIGN)),
(ModeButton('6. Create traffic-lights', Mode.TRAFFIC_LIGHTS,
self.model, self),
GuiTrafficLightsMode(self.model, ObjectType.TRAFFIC_LIGHT)),
(ModeButton('7. Create squares', Mode.SQUARES, self.model,
self), GuiSquaresMode(self.model, ObjectType.SQUARE)),
(ModeButton('8. Create cubes', Mode.CUBES, self.model,
self), GuiCubesMode(self.model, ObjectType.CUBE)),
(ModeButton('9. Create qr-cubes', Mode.QR_CUBES, self.model,
self), GuiQrCubesMode(self.model, ObjectType.QR_CUBE)),
]
# Layout fill
layout.addWidget(QLabel('To create the world:', self), 0, 1)
for idx, btn in enumerate(mode_buttons):
layout.addWidget(btn[0], idx + 1, 1)
self.ctrl_grp.addButton(btn[0])
self.model.add_mode(btn[0].mode, btn[1])
layout.addWidget(QLabel('Then press buttons below:', self),
len(mode_buttons) + 1, 1)
layout.addWidget(generateButton, len(mode_buttons) + 2, 1)
self.statusBar = QStatusBar()
self.statusBar.showMessage('Ready')
self.statusBar.setMaximumHeight(20)
layout.addWidget(self.statusBar, len(mode_buttons) + 3, 0, 1, 2)
self.canvas = Canvas(self.model, self)
layout.addWidget(self.canvas, 0, 0, len(mode_buttons) + 3, 1)
def paintEvent(self, event=None):
qp = QPainter(self)
self.drawPoints(qp)
def drawPoints(self, qp):
"""
@brief Draw window background with random points cloud
"""
qp.setPen(Qt.red)
windowSize = self.size()
for i in range(1000):
x = random.randint(1, windowSize.width() - 1)
y = random.randint(1, windowSize.height() - 1)
qp.drawPoint(x, y)
def generateOutputFiles(self):
converter.serialize_2_json(self.json_save_fpath, self.model.objects,
self.model.map_params)
converter.create_sdf(self.world_save_fpath, self.model.objects,
self.model.map_params)
log.info("Files generated!")
class TfrmBase(QMainWindow, TScreenStates):
recordCount = 0
def __init__(self, parent=None):
super(TfrmBase, self).__init__()
self.FOnStateChange = self.onStateChange
self.activeState = self.ssInactive
self._defaultSettings()
self._createWidgets()
self._setEvents()
def _defaultSettings(self):
self.setObjectName("frmBase")
self.resize(640, 480)
self.setMinimumSize(QSize(640, 480))
def _createWidgets(self):
self._createLayout()
self._createMenus()
self._createToolBar()
self._createStatusBar()
self._createPages()
self._setLayouts()
def _createLayout(self):
self.clientArea = QWidget()
self.clientArea.setSizePolicy(
QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding))
self.clientArea.setMinimumSize(QSize(640, 400))
self.clientArea.setBaseSize(QSize(640, 400))
self.clientArea.setLayoutDirection(Qt.LeftToRight)
self.clientArea.setObjectName("clientArea")
self.gridLayout = QGridLayout(self.clientArea)
self.gridLayout.setContentsMargins(0, 0, 0, 0)
self.gridLayout.setSpacing(0)
self.gridLayout.setObjectName("gridLayout")
def _createMenus(self):
# Create a Menu Bar
self.mnMenu = self.menuBar()
self.mnMenu.setObjectName("mnMenu")
# Create all Top Menus
self.mnApp = QMenu('&Aplicação')
self.mnApp.setObjectName('mnApp')
self.mnOpe = QMenu('&Operação')
self.mnOpe.setObjectName("mnOperations")
self.mnNav = QMenu('&Navegação')
self.mnNav.setObjectName("mnNav")
# Set Menus to MenuBar
self.mnMenu.addMenu(self.mnNav)
self.mnMenu.addMenu(self.mnOpe)
self.mnMenu.addMenu(self.mnApp)
# Crealte all Actions to Application Menu
self._createAppActions()
self._createOpeActions()
self._setMenuActions()
self.mnMenu.addAction(self.mnApp.menuAction())
self.mnMenu.addAction(self.mnOpe.menuAction())
self.mnMenu.addAction(self.mnNav.menuAction())
self._settingActionsEvents()
def _createAppActions(self):
# Exit Program Action
self.acExit = QAction(
self.getIcon("./resources/exit.ico", QSize(32, 32)), '&Sair')
self.acExit.setObjectName("acExit")
self.acExit.setShortcut('Ctrl+Q')
self.acExit.setStatusTip('Finalizar o Programa')
self.acExit.triggered.connect(self.closeApp)
def _createOpeActions(self):
# Search Action
self.acSearch = QAction(
self.getIcon("./resources/Search.ico", QSize(32, 32)),
'&Pesquisar')
self.acSearch.setObjectName("acSearch")
self.acSearch.setShortcut('F5,Ctrl+P')
self.acSearch.setStatusTip(
'Preenche o Filtro para Selecionar Registros')
# List Action
self.acList = QAction(
self.getIcon("./resources/list.ico", QSize(32, 32)), '&Listar')
self.acList.setShortcut('Ctrl+L')
self.acList.setStatusTip('Listar todos os Registros')
self.acList.setObjectName("acList")
# Insert Action
self.acInsert = QAction(
self.getIcon("./resources/db_add.ico", QSize(32, 32)), '&Inserir')
self.acInsert.setShortcut('F2,Ins')
self.acInsert.setStatusTip('Incluir Novo Registros')
self.acInsert.setObjectName("acInsert")
# Update Action
self.acUpdate = QAction(
self.getIcon("./resources/db_update.ico", QSize(32, 32)),
'&Editar')
self.acUpdate.setShortcut('Ctrl+U')
self.acUpdate.setStatusTip('Editar o Registro Atual')
self.acUpdate.setObjectName("acUpdate")
# Delete Action
self.acDelete = QAction(
self.getIcon("./resources/db_remove.ico", QSize(32, 32)),
'&Excluir')
self.acDelete.setShortcut('Ctrl+Del')
self.acDelete.setStatusTip('Exclui o Registro Atual')
self.acDelete.setObjectName("acDelete")
# Save Action
self.acSave = QAction(
self.getIcon("./resources/db_commit.ico", QSize(32, 32)),
'&Salvar')
self.acSave.setShortcut('F10,Ctrl+S')
self.acSave.setStatusTip('Salva as Alterações do Registro')
self.acSave.setObjectName("acSave")
# Cancel Action
self.acCancel = QAction(
self.getIcon("./resources/cancel.ico", QSize(32, 32)), '&Cancelar')
self.acCancel.setShortcut('Esc')
self.acCancel.setStatusTip('Cancela as Alterações do Registro')
self.acCancel.setObjectName("acCancel")
# First Action
self.acFirst = QAction(
self.getIcon("./resources/start.ico", QSize(32, 32)), '&Início')
self.acFirst.setShortcut('Ctrl+Left')
self.acFirst.setStatusTip('Vai para o Primeiro Registro')
self.acFirst.setObjectName("acFirst")
# Prior Action
self.acPrior = QAction(
self.getIcon("./resources/left.ico", QSize(32, 32)), '&Anterior')
self.acPrior.setShortcut('Left')
self.acPrior.setStatusTip('Vai para o Registro Anterior')
self.acPrior.setObjectName("acPrior")
# Next Action
self.acNext = QAction(
self.getIcon("./resources/right.ico", QSize(32, 32)), '&Próximo')
self.acNext.setShortcut('Right')
self.acNext.setStatusTip('Vai para o Próximo Registro')
self.acNext.setObjectName("acNext")
# Last Action
self.acLast = QAction(
self.getIcon("./resources/end.ico", QSize(32, 32)), '&Último')
self.acLast.setShortcut('Ctrl+Right')
self.acLast.setStatusTip('Vai para o Último Registro')
self.acLast.setObjectName("acLast")
# Form Title Action
self.dcTitle = QAction()
font = QFont()
font.setPointSize(14)
font.setBold(True)
font.setWeight(75)
self.dcTitle.setFont(font)
self.dcTitle.setObjectName("dcTitle")
def getIcon(self, res: str, size: QSize) -> QIcon:
icon = QIcon()
icon.addPixmap(
QPixmap(res).scaled(size.width(), size.height(),
Qt.KeepAspectRatio), QIcon.Active, QIcon.On)
return icon
def _setMenuActions(self):
# Set Menu Application Actions
self.mnApp.addAction(self.acExit)
# Set Menu Operations Actions
self.mnOpe.addAction(self.acSearch)
self.mnOpe.addAction(self.acList)
self.mnOpe.addSeparator()
self.mnOpe.addAction(self.acInsert)
self.mnOpe.addAction(self.acUpdate)
self.mnOpe.addAction(self.acDelete)
self.mnOpe.addSeparator()
self.mnOpe.addAction(self.acSave)
self.mnOpe.addAction(self.acCancel)
# Set Menu Navigation Actions
self.mnNav.addAction(self.acFirst)
self.mnNav.addAction(self.acPrior)
self.mnNav.addAction(self.acNext)
self.mnNav.addAction(self.acLast)
def _settingActionsEvents(self):
# Set Menu Operations Trigger onClick
self.acSearch.triggered.connect(
lambda: self.setFormStatus(self.ssSearch))
self.acList.triggered.connect(
lambda: self.setFormStatus(self.ssSearchAll))
self.acInsert.triggered.connect(
lambda: self.setFormStatus(self.ssInsert))
self.acUpdate.triggered.connect(
lambda: self.setFormStatus(self.ssUpdate))
self.acDelete.triggered.connect(
lambda: self.setFormStatus(self.ssDelete))
self.acSave.triggered.connect(lambda: self.setFormStatus(self.ssPost))
self.acCancel.triggered.connect(
lambda: self.setFormStatus(self.ssCancel))
# Set Menu Navigation Trigger onClick
self.acFirst.triggered.connect(
lambda: self.setFormStatus(self.ssFirst))
self.acPrior.triggered.connect(
lambda: self.setFormStatus(self.ssPrior))
self.acNext.triggered.connect(lambda: self.setFormStatus(self.ssNext))
self.acLast.triggered.connect(lambda: self.setFormStatus(self.ssLast))
def _createToolBar(self):
# Create a tbActions ToolBar
self.tbActions = QToolBar()
self.tbActions.setMinimumSize(QSize(300, 34))
self.tbActions.setMaximumSize(QSize(16777215, 34))
self.tbActions.setBaseSize(QSize(300, 34))
self.tbActions.setAcceptDrops(False)
self.tbActions.setToolTipDuration(3)
self.tbActions.setAllowedAreas(Qt.TopToolBarArea)
self.tbActions.setObjectName("tbActions")
self.addToolBar(Qt.TopToolBarArea, self.tbActions)
# Create a tbTitle ToolBar
self.tbTitle = QToolBar()
self.tbTitle.setMinimumSize(QSize(340, 34))
self.tbTitle.setMaximumSize(QSize(16777215, 34))
self.tbTitle.setBaseSize(QSize(341, 34))
self.tbTitle.setAllowedAreas(Qt.TopToolBarArea)
self.tbTitle.setToolButtonStyle(Qt.ToolButtonTextOnly)
self.tbTitle.setFloatable(False)
self.tbTitle.setObjectName("tbTitle")
# self.tbTitle.setLabelAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
self.addToolBar(Qt.TopToolBarArea, self.tbTitle)
# Call Add Actions to ToolBar
self._setToolBarActions()
def _setToolBarActions(self):
# Set ToolBar Actions
self.tbActions.addAction(self.acSearch)
self.tbActions.addAction(self.acInsert)
self.tbActions.addAction(self.acUpdate)
self.tbActions.addAction(self.acDelete)
self.tbActions.addSeparator()
self.tbActions.addAction(self.acSave)
self.tbActions.addAction(self.acExit)
self.tbTitle.addAction(self.dcTitle)
def _createStatusBar(self):
self.sbStatus = QStatusBar()
self.sbStatus.setMaximumHeight(24)
self.sbStatus.setObjectName("sbStatus")
self.sbStatus.setStyleSheet("""
.QLabel {
background-color: #FFFFFF;
color: #000000;
}
""")
self.lbStatus = QLabel(self.sbStatus)
self.lbStatus.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
self.lbStatus.setText('Inactive')
self.lbStatus.setMinimumSize(QSize(130, 15))
self.lbStatus.setFrameShape(QFrame.Panel)
self.lbStatus.setFrameShadow(QFrame.Sunken)
self.sbStatus.addPermanentWidget(self.lbStatus)
self.setStatusBar(self.sbStatus)
def _createPages(self):
self.tabMain = QTabWidget(self.clientArea)
self.tabMain.setSizePolicy(
QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding))
self.tabMain.setTabPosition(QTabWidget.South)
self.tabMain.setObjectName("tabMain")
self.pgList = QWidget(self.tabMain)
self.pgList.setObjectName("pgList")
self.pgDetail = QWidget(self.tabMain)
self.pgDetail.setObjectName("pgDetail")
self.tabMain.addTab(self.pgList, "")
self.tabMain.addTab(self.pgDetail, "")
self._createPageListContent()
def _createPageListContent(self):
self.treeWidget = QTreeWidget(self.pgList)
self.treeWidget.setSizePolicy(
QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding))
self.treeWidget.setFrameShape(QFrame.NoFrame)
self.treeWidget.setFrameShadow(QFrame.Plain)
self.treeWidget.setColumnCount(3)
self.treeWidget.setObjectName("treeWidget")
self.treeWidget.headerItem().setText(0, "Campo")
self.treeWidget.headerItem().setText(1, "Campo")
self.treeWidget.headerItem().setText(2, "Campo")
self.treeWidget.setGeometry(QRect(0, 0, 640, 370))
self.treeWidget.setMinimumSize(QSize(640, 370))
self.tabMain.setCurrentIndex(0)
def _setLayouts(self):
self.gridLayout.addWidget(self.tabMain, 0,
Qt.AlignBottom | Qt.AlignRight, 1, 1)
self.setCentralWidget(self.clientArea)
def translateForm(self):
self._translate = QCoreApplication.translate
self.setWindowTitle(
self._translate("TfrmBase", "Tela de Básica de Cadastros"))
self.mnApp.setTitle(self._translate("TfrmBase", "Aplicação"))
self.mnOpe.setTitle(self._translate("TfrmBase", "Operações"))
self.mnNav.setTitle(self._translate("TfrmBase", "Navegação"))
self.sbStatus.setToolTip(self._translate("TfrmBase",
"Barra de Status"))
self.tbActions.setWindowTitle(
self._translate("TfrmBase", "Ferramentas"))
self.tbActions.setToolTip(
self._translate("TfrmBase", "Barra de Ferramentas"))
self.tbTitle.setWindowTitle(self._translate("TfrmBase", "Descrição"))
self.acExit.setText(self._translate("TfrmBase", "&Sair"))
self.acExit.setToolTip(self._translate("TfrmBase", "Sair do Programa"))
self.acSearch.setText(self._translate("TfrmBase", "&Pesquisar"))
self.acSearch.setStatusTip(
self._translate("TfrmBase", "Procurar Por um Registro"))
self.acList.setText(self._translate("TfrmBase", "&Listar Todos"))
self.acList.setStatusTip(
self._translate("TfrmBase", "Lista todos os Registros"))
self.acInsert.setText(self._translate("TfrmBase", "&Inserir"))
self.acInsert.setStatusTip(
self._translate("TfrmBase", "Adicionar Registro"))
self.acUpdate.setText(self._translate("TfrmBase", "&Editar"))
self.acUpdate.setStatusTip(
self._translate("TfrmBase", "Editar Registro"))
self.acDelete.setText(self._translate("TfrmBase", "E&xcluir"))
self.acDelete.setStatusTip(
self._translate("TfrmBase", "Excluir Registro"))
self.acSave.setText(self._translate("TfrmBase", "&Salvar"))
self.acSave.setToolTip(self._translate("TfrmBase", "Salvar Registro"))
self.acCancel.setText(self._translate("TfrmBase", "&Cancelar"))
self.acCancel.setToolTip(
self._translate("TfrmBase", "Cencelar Alterações"))
self.dcTitle.setText(
self._translate("TfrmBase", "Título da Tela de Cadastros"))
self.dcTitle.setToolTip(
self._translate("TfrmBase", "Título da Tela de Cadastros"))
self.tabMain.setTabText(
self.tabMain.indexOf(self.pgList),
self._translate("TfrmBase", "Lista dos Registros"))
self.tabMain.setTabToolTip(
self.tabMain.indexOf(self.pgList),
self._translate("TfrmBase", "Listagem das Ferramentas"))
self.tabMain.setTabText(
self.tabMain.indexOf(self.pgDetail),
self._translate("TfrmBase", "Detalhes do Registro Selecionando"))
@property
def activeState(self):
return self._activeValue
@property
def activeStateColor(self):
return self.activeValue['FG']
@property
def activeStateBackgroud(self):
return self.activeValue['BG']
@activeState.setter # Seta a Propriedade _activeState
def activeState(self, value: int):
self.workValue = value
self._activeState = value
def setScreenState(self, stt: int):
self.acExit.setEnabled(self.inBrowse(stt))
# Set Menu Operations Actions
self.acSearch.setEnabled((self.inBrowse(stt)
or (self.recordCount == 0)))
self.acList.setEnabled((self.inBrowse(stt) or (self.recordCount == 0)))
self.acInsert.setEnabled(self.inBrowse(stt))
self.acUpdate.setEnabled((self.inBrowse(stt)
and (self.recordCount > 0)))
self.acDelete.setEnabled((self.inBrowse(stt)
and (self.recordCount > 0)))
self.acSave.setEnabled(self.inUpdate(stt))
self.acCancel.setEnabled(self.inUpdate(stt))
# Set Menu Navigation Actions
self.acFirst.setEnabled((self.inBrowse(stt)
and (self.recordCount > 0)))
self.acPrior.setEnabled((self.inBrowse(stt)
and (self.recordCount > 0)))
self.acNext.setEnabled((self.inBrowse(stt) and (self.recordCount > 0)))
self.acLast.setEnabled((self.inBrowse(stt) and (self.recordCount > 0)))
# Set tab Main if state in Browse enabled
self.tabMain.setEnabled(self.inBrowse(stt))
def _layoutWidgets(self):
return (self.frmLayout.itemAt(i)
for i in range(self.frmLayout.count()))
def _getAllFields(self):
arrFields = []
for w in self._layoutWidgets():
if (not (isinstance(w, QLabel))):
arrFields.append(w)
return arrFields
def setEnableFields(self, enable: bool = True):
# Enable All Fields
for controls in self._layoutWidgets():
QWidget(controls).setEnabled(enable)
def clearFields(self):
# cliar content of all fileds
for controls in self._getAllFields():
QWidget(controls).setText('')
def setColorFields(self):
# cliar content of all fileds
style = ".QWidget { backgroud-color: " + self.activeStateBackgroud + "; }"
for controls in self._getAllFields():
QWidget(controls).setStyle(style)
def showDataDetails(self):
# move data of selected record to fileds
if (self.tabMain.currentIndex() == 0):
self.tabMain.setCurrentIndex(1)
def filterRecord(self):
raise NotImplementedError(500)
def getFirstRecord(self):
raise NotImplementedError(500)
def getPriorRecord(self):
raise NotImplementedError(500)
def getNextRecord(self):
raise NotImplementedError(500)
def getLastRecord(self):
raise NotImplementedError(500)
def insertRecord(self):
raise NotImplementedError(500)
def deleteRecord(self):
raise NotImplementedError(500)
def updateRecord(self):
raise NotImplementedError(500)
def postRecord(self):
raise NotImplementedError(500)
def execOpertations(self, state: int):
if ((state == self.ssFilter) or (state == self.ssSearchAll)):
self.filterRecord()
elif (state == self.ssFirst):
self.getFirstRecord()
elif (state == self.ssPrior):
self.getPriorRecord()
elif (state == self.ssNext):
self.getNextRecord()
elif (state == self.ssLast):
self.getLastRecord()
elif (state == self.ssInsert):
self.insertRecord()
elif (state == self.ssDelete):
self.deleteRecord()
elif (state == self.ssUpdate):
self.updateRecord()
elif (state == self.ssPost):
self.postRecord()
else:
raise NotImplementedError(401, 'Operação não suportada')
@pyqtSlot(int)
def setFormStatus(self, state: int):
if ((state == self.ssSearch) and (self.activeState != state)):
self.clearFields()
self.setColorFields()
self.showDataDetails()
if (self.activeState != state):
self.activeState = state
if (state == self.ssCancel):
self.activeState = self.ssBrowse
@pyqtSlot(int, int, dict, str)
def onStateChange(self,
NewState: int,
OldState: int,
Result: dict = {},
Err: str = ''):
try:
# show screen state on status bar
state = self.getStateProperties(NewState)
style = '.QLabel { background-color: ' + state[
'BG'] + '; color: ' + state['FG'] + '; }'
self.sbStatus.setStyleSheet(style)
self.lbStatus.setText(state['Descr'])
# call operation into child screen
self.execOpertations(NewState)
# change buttons states
self.setScreenState(NewState)
# set result status code and result satatus Message
self.setResultStatusCode = 200
self.setResultStatusMessage = ''
except Exception as e:
self.ResultStatusCode = 200
self.ResultStatusMessage = str(e)
QMessageBox.critical(self, self.windowTitle(),
self.ResultStatusMessage)
return self.result
@pyqtSlot()
def tabMainChanged(self):
self.sbStatus.showMessage('TabMain change tabIndex to (' +
str(self.tabMain.currentIndex()) + ')!')
if (self.tabMain.currentIndex() == 1):
self.showDataDetails()
@pyqtSlot()
def InsertData(self):
# self.sbStatus.showMessage('Prepare to insert data....')
pass
def _setEvents(self):
self.tabMain.blockSignals(
True) # just for not showing the initial message
self.tabMain.currentChanged.connect(self.tabMainChanged) # changed!
self.tabMain.blockSignals(False) # wait signals now
@pyqtSlot()
def closeApp(self):
self.close()
