class SideBar(QScrollArea):
def __init__(self):
QScrollArea.__init__(self)
self.mainWidget = QListWidget()
self.setWidget(self.mainWidget)
self.setWidgetResizable(True)
self.setFixedWidth(200)
self.setTitle()
def setProperties(self, signal):
self.mainWidget.clear()
self.setTitle()
for x in signal.properties:
self.mainWidget.addItem(" " + x[0] + ": " + x[1])
def setTitle(self):
self.mainWidget.addItem("Signal Properties")
class PropertyBar(QScrollArea):
def __init__(self, parent):
QScrollArea.__init__(self)
self.parent = parent
self.mainWidget = QListWidget()
self.setWidget(self.mainWidget)
self.setWidgetResizable(True)
self.setFixedHeight(130)
self.setTitle()
def setProperties(self):
self.mainWidget.clear()
self.setTitle()
if self.parent:
for x in self.parent.project.experimentProperties:
self.mainWidget.addItem(" " + x[0] + ": " + x[1])
def setTitle(self):
self.mainWidget.addItem("Experiment Properties")
def initUI(self):
mainlay = QVBoxLayout()
mainlay.addWidget(QLabel('<b>Deformers ({0})</b>'.format(len(self.mdl.deformers))))
def_list = QListWidget()
for index, el in enumerate(self.mdl.deformers):
def_list.addItem('{0} - {1}'.format(index, el))
#def_list.addItems(self.mdl.deformers)
mainlay.addWidget(def_list)
self.def_list = def_list
btns = QHBoxLayout()
btns.addStretch()
save = QPushButton('Save')
save.clicked.connect(self.save)
close = QPushButton('Close')
close.clicked.connect(self.close)
btns.addWidget(save)
btns.addWidget(close)
mainlay.addLayout(btns)
self.setLayout(mainlay)
self.setGeometry(340, 340, 200, 400)
self.setWindowTitle('MSH Suite - {0} Deformers'.format(self.mdl.name))
self.show()
class HiddenDialog(QDialog):
lifelines = None
def __init__(self, hiddenLifeline, parent = None):
super(HiddenDialog, self).__init__(parent)
self.lifelines = hiddenLifeline
layout = QVBoxLayout(self)
listTitle = QLabel('Hidden Life-lines')
layout.addWidget(listTitle)
self.listHiddenLifelines = QListWidget()
self.listHiddenLifelines.setFixedWidth(400)
self.listHiddenLifelines.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
for text in self.lifelines:
item = QtGui.QListWidgetItem(text)
self.listHiddenLifelines.addItem(item)
layout.addWidget(self.listHiddenLifelines)
buttons = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Ok | QtGui.QDialogButtonBox.Cancel, QtCore.Qt.Horizontal, self)
buttons.button(QDialogButtonBox.Ok).setText('Show')
buttons.accepted.connect(self.accept)
buttons.rejected.connect(self.reject)
layout.addWidget(buttons)
@staticmethod
def getSelectedItems(lifelines, parent = None):
dialog = HiddenDialog(lifelines,parent)
result = dialog.exec_()
return (result, [str(x.text()) for x in dialog.listHiddenLifelines.selectedItems()])
class Console():
def __init__(self, targetLayoutContainer):
self.textarea = QTextEdit()
self.commits = QListWidget()
self.commits.addAction(QAction('Rollback to this revision', self.commits, triggered=self.rollback))
self.commits.setContextMenuPolicy(Qt.ActionsContextMenu)
self.widget = QTabWidget()
self.widget.addTab(self.textarea, 'Log')
self.widget.addTab(self.commits, 'Commits')
targetLayoutContainer.addWidget(self.widget)
def color(self, module, function, color, *args):
print module, function, args
prettyString = '<font color="' + color + '"><b>', module, '</b><i>::', function, '</i> --> ', ''.join(args), '</font>'
self.textarea.append(''.join(prettyString))
def info(self, module, function, *args):
print module, function, args
prettyString = '<font color="black"><b>', module, '</b><i>::', function, '</i> --> ', ''.join(args), '</font>'
self.textarea.append(''.join(prettyString))
def error(self, module, function, *args):
print module, function, args
prettyString = '<font color="red"><b>', module, '</b><i>::', function, '</i> --> ', ''.join(args), '</font>'
self.textarea.append(''.join(prettyString))
def warn(self, module, function, *args):
print module, function, args
prettyString = '<font color="#BE9900"><b>', module, '</b><i>::', function, '</i> --> ', ''.join(args), '</font>'
self.textarea.append(''.join(prettyString))
def set_commits(self, commits):
self.commits.clear()
for commit in commits:
self.commits.addItem(commit)
def clear(self):
self.textarea.clear()
def rollback(self):
targetCommit = self.commits.currentItem().text().split(' ')[0]
if QMessageBox.warning(None, 'Rollback to commit?', 'All commits after ' + targetCommit + ' will be lost, proceed?', QMessageBox.Yes, QMessageBox.No) == QMessageBox.Yes:
rollback(targetCommit)
def __init__(self, parent=None):
super(EditPreferencesDlg, self).__init__(parent)
self.setWindowTitle("Preferences")
# define widgets
pref_list = QListWidget()
pref_list.addItem("General")
pref_list.addItem("Display")
pref_list.setMaximumWidth(150)
pref_list.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)
button_box = QDialogButtonBox(QDialogButtonBox.Ok|QDialogButtonBox.Cancel)
general_page = QWidget()
general_layout = QGridLayout()
general_layout.setAlignment(Qt.AlignTop)
general_layout.addWidget(QLabel("<b>General</b>"), 0, 0)
general_page.setLayout(general_layout)
display_page = QWidget()
display_layout = QGridLayout()
display_layout.setAlignment(Qt.AlignTop)
display_layout.addWidget(QLabel("<b>Display Options</b>"), 0, 0)
self.multitabs_checkbox = QCheckBox("Limit the display of tabs to one relief device (and the device's scenarios) at a time")
if parent.limit_tabs is True:
self.multitabs_checkbox.setChecked(True)
display_layout.addWidget(self.multitabs_checkbox, 1, 0)
display_page.setLayout(display_layout)
stacked_widget = QStackedWidget()
for page in general_page, display_page:
stacked_widget.addWidget(page)
main_layout = QVBoxLayout()
widgets_layout = QHBoxLayout()
widgets_layout.addWidget(pref_list)
widgets_layout.addWidget(stacked_widget)
buttons_layout = QHBoxLayout()
buttons_layout.addStretch()
buttons_layout.addWidget(button_box)
main_layout.addLayout(widgets_layout)
main_layout.addLayout(buttons_layout)
self.setLayout(main_layout)
pref_list.currentRowChanged.connect(stacked_widget.setCurrentIndex)
button_box.accepted.connect(self.accept)
button_box.rejected.connect(self.reject)
def display(self):
layout = QHBoxLayout()
layout.setSpacing(10)
# Left panel - repositories
qlistWidget = QListWidget()
for repo in self.args.user.repositories():
qlistWidget.addItem(repo.name)
layout.addWidget (qlistWidget)
# Middle panel - commits
commitsLabel = QLabel("There will be the commits")
layout.addWidget (commitsLabel)
# Something
somethingLabel = QLabel("There will be something")
layout.addWidget (somethingLabel)
self.setLayout(layout)
self.show()
class UiMain(QMainWindow):
""" The main gui interface, invokes all windows and ties everything
together
"""
def __init__(self):
""" automatically called __init__ function """
super(UiMain, self).__init__()
# initialize all the variables that are going to be defined in the
# future
self.update_dialog = None
self.update_dialog_lbl = None
self.app_select_box = None
self.selector_lbl = None
self.current_playing_lbl = None
self.current_playing = None
self.misc_messages = None
self.start_btn = None
self.output_dir_lbl = None
self.select_output_dir_btn = None
self.output_cur_dir_lbl = None
self.active_items_list = None
self.inactive_items_list = None
self.switch_active_item_button_off = None
self.switch_active_item_button_on = None
self.switch_output_split_btn = None
self.switch_output_split_lbl = None
# initialize the system tray
# self.system_tray = QSystemTrayIcon(self)
# self.system_tray.setIcon(QIcon(resource_path('icon.png')))
# self.system_tray.show()
# self.system_tray.setToolTip('SMG')
# self.system_tray.activated.connect(self.on_systray_activated)
# initialize the main window
self.setObjectName('self')
self.setWindowTitle('SMG - By Azeirah')
self.resize(400, 250)
# Gives the self an icon
self.setWindowIcon(QIcon(resource_path('icon.png')))
# create the tabs
# the tab widget itself
self.tabbed_windows = QTabWidget(self)
self.tabbed_windows.resize(400, 300)
# tab 1, contains the music player selection
self.music_players = QFrame()
# tab 2, contains options
self.options = QFrame()
self.tabbed_windows.addTab(self.music_players, 'Music players')
self.tabbed_windows.addTab(self.options, 'Options')
# initializes the two tabs, with all the code down below
self.tab_music_players()
self.tab_options()
# shows the main window
self.show()
def closeEvent(self, event):
""" an automatically called function when the program is about to
close.
"""
# Stops all Threads. These would continue to run in the background
# Even if the window was closed.
Main.running = False
# close the ZuneNowPlaying.exe process
if Constants.SUBP:
Constants.SUBP.kill()
def changeEvent(self, event):
# if event.type() == QEvent.WindowStateChange:
# if self.isMinimized():
# event.ignore()
# self.hide()
# self.system_tray.showMessage('Running', 'Running in the
# background.')
# return
super(UiMain, self).changeEvent(event)
def on_systray_activated(self, reason):
if reason == QSystemTrayIcon.DoubleClick:
self.show()
@staticmethod
def toggle_split(event):
# 0 = Qt.Unchecked The item is unchecked.
# 1 = Qt.PartiallyChecked The item is partially checked. Items in
# hierarchical models may be partially checked if some, but not all,
# of
# their children are checked.
# 2 = Qt.Checked The item is checked.
if event == 0:
Constants.OPTIONS['splitText'] = False
elif event == 2:
Constants.OPTIONS['splitText'] = True
def update(self):
""" Checks a webpage for current version, compares this to built-in
current versions, and shows update dialog if necessary
"""
try:
ver = urlopen('http://league-insanity.tk/Azeirah_content/version')\
.read()
except IOError:
# if for some reason it couldn't retrieve the version, set it to
# automatically ignore the update: False
ver = False
if not float(VERSION) >= float(ver):
self.popup = QDialog(self)
self.popup.setModal(True)
self.popup.setGeometry(200, 100, 500, 100)
self.popup.show()
self.popup_text = QLabel(self.popup)
self.popup_text.setGeometry(5, 5, 500, 30)
self.popup_text.setOpenExternalLinks(True)
self.popup_text.show()
self.popup_text.setText(
"""There is an update available. Run update.exe or <a href='https://sourceforge.net/projects/obsmusicstreamd'>download the update manually</a>""")
# reply = QMessageBox.question(Constants.UI, 'Message',
# "Do you want to update?", QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
# if reply == QMessageBox.Yes:
# import atexit
# import subprocess
# def runUpdater():
# import time
# time.sleep(3)
# subprocess.Popen(resource_path('update.exe'))
# atexit.register(runUpdater)
# sys.exit()
# Constants.update_dialog = QWidget()
# Constants.update_dialog.resize(350, 100)
# Constants.update_dialog.setWindowIcon(QIcon(resource_path\
# ('icon.png')))
# Constants.update_dialog.setWindowTitle('Updater')
# Constants.update_dialog_lbl = QLabel(Constants.update_dialog)
# Constants.update_dialog_lbl.setGeometry(10, 40, 340, 12)
# Constants.update_dialog.show()
# updateThread = Thread(target = update.update)
# updateThread.setName('updateThread')
# updateThread.start()
def tab_music_players(self):
""" Everything inside the Music players tab gets created here."""
# self.music_players
# Creates the box with all the music players inside of it
self.app_select_box = QComboBox(self.music_players)
self.app_select_box.setGeometry(135, 10, 150, 25)
# Whenever you change the application, it runs the selectnewapp func
self.app_select_box.activated[str].connect(self.select_new_app)
# Creates the label for the selection combobox
self.selector_lbl = QLabel(self.music_players)
self.selector_lbl.setGeometry(10, 10, 150, 25)
self.selector_lbl.setText('Select your music player: ')
# Creates the label for the current playing song (and the current
# playing song label)
self.current_playing_lbl = QLabel(self.music_players)
self.current_playing_lbl.setGeometry(10, 45, 150, 25)
self.current_playing_lbl.setText('Current playing song: ')
self.current_playing = QLabel(self.music_players)
self.current_playing.setGeometry(117, 45, 250, 25)
self.current_playing.setText(Misc.noSongPlaying)
# Creates a label which displays any additional messages
self.misc_messages = QLabel(self.music_players)
self.misc_messages.setGeometry(10, 80, 390, 24)
self.misc_messages.setText(Misc.misc_message())
self.misc_messages.setOpenExternalLinks(True)
# adds all the music players into the combobox
self.app_select_box.addItem(None)
for item in Constants.ACTIVEITEMS:
if item == '__name__' or item == 'active':
continue
self.app_select_box.addItem(item)
# creates the start button
self.start_btn = QPushButton(self.music_players)
self.start_btn.setGeometry(75, 120, 250, 35)
self.start_btn.setText('Start')
# links the start button to the self.start function
QObject.connect(self.start_btn, SIGNAL("clicked()"),
lambda: Thread(target=self.start, name='startbutton').start())
def tab_options(self):
""" Everything inside the Options tab gets created here. """
# self.options
# This section is for selecting output dir
# Creates the output dir label
self.output_dir_lbl = QLabel(self.options)
self.output_dir_lbl.setGeometry(10, 10, 125, 15)
self.output_dir_lbl.setText('Change Output Directory: ')
# Creates the output dir button
self.select_output_dir_btn = QPushButton(self.options)
self.select_output_dir_btn.setGeometry(137, 8, 30, 20)
self.select_output_dir_btn.setText('...')
# Creates the output dir currentdir Lineedit
self.output_cur_dir_lbl = QLineEdit(self.options)
self.output_cur_dir_lbl.setGeometry(170, 6, 210, 25)
self.output_cur_dir_lbl.setReadOnly(True)
self.output_cur_dir_lbl.setText(Constants.CONFIG.
get('directories', 'current_song'))
# when the '...' button is clicked, show a dialog (fire func
# disp_dialog)
QObject.connect(self.select_output_dir_btn, SIGNAL("clicked()"),
self.disp_dialog)
# This section is for selecting what players you use
# The box with all the active players
self.active_items_list = QListWidget(self.options)
self.active_items_list.setGeometry(10, 40, 150, 100)
# The box with all the inactive players
self.inactive_items_list = QListWidget(self.options)
self.inactive_items_list.setGeometry(230, 40, 150, 100)
# Populate the two boxes with active and inactive items
for item in Constants.ACTIVEITEMS:
if item == '__name__' or item == 'active':
continue
self.active_items_list.addItem(item)
for item in Constants.INACTIVEITEMS:
if item == '__name__' or item == 'active':
continue
self.inactive_items_list.addItem(item)
# The buttons responsible for switching
# off button
self.switch_active_item_button_off = QPushButton(self.options)
self.switch_active_item_button_off.setText('->'.decode('utf-8'))
# Makes the -> readable and clear
self.switch_active_item_button_off.setFont(QFont('SansSerif', 17))
self.switch_active_item_button_off.setGeometry(175, 55, 40, 30)
# on button
self.switch_active_item_button_on = QPushButton(self.options)
self.switch_active_item_button_on.setText('<-'.decode('utf-8'))
# makes <- readable and clear
self.switch_active_item_button_on.setFont(QFont('SansSerif', 17))
self.switch_active_item_button_on.setGeometry(175, 90, 40, 30)
QObject.connect(self.switch_active_item_button_on, SIGNAL
("clicked()"), self.switch_item_on)
QObject.connect(self.switch_active_item_button_off, SIGNAL
("clicked()"), self.switch_item_off)
# A button to toggle the split output in half option. It's a temporary
# fix for the Foobar double output problem.
self.switch_output_split_btn = QCheckBox(self.options)
self.switch_output_split_btn.setCheckState(Qt.CheckState.Unchecked)
self.switch_output_split_btn.setGeometry(10, 140, 40, 30)
self.switch_output_split_btn.stateChanged.connect(self.toggle_split)
# The label for the split toggle
self.switch_output_split_lbl = QLabel(self.options)
self.switch_output_split_lbl.setText(
"Split the output text in half (don't use this if you don't need it)")
self.switch_output_split_lbl.setGeometry(30, 140, 300, 30)
def switch_item_on(self):
""" Switches items (musicapps) on """
try:
# If an item from the active box is selected
# Remove it and place it inside the inactive box
item_taken = self.inactive_items_list.takeItem(
self.inactive_items_list.currentRow())
self.active_items_list.addItem(item_taken)
active_items = {}
inactive_items = {}
for i in range(self.active_items_list.count()):
active_items[self.active_items_list.item(i).text()] =\
ITEMS[self.active_items_list.item(i).text()
.encode('utf-8')]
for i in range(self.inactive_items_list.count()):
inactive_items[self.inactive_items_list.item(i).text()] =\
ITEMS[self.inactive_items_list.item(i).text()
.encode('utf-8')]
Constants.ACTIVE_ITEMS = active_items
Constants.INACTIVE_ITEMS = inactive_items
# clear the selection combobox
self.app_select_box.clear()
# Repopulate the combobox
self.app_select_box.addItem(None)
for item in active_items:
self.app_select_box.addItem(item)
Constants.CONFIG.set('active', item_taken.text(),
ITEMS[item_taken.text()])
Constants.CONFIG.remove_option('inactive', item_taken.text())
# Updates the config file to be up to date with activeItems
Constants.CONFIG.update()
except:
raise
def switch_item_off(self):
""" Switches items (musicapps) off """
try:
# If an item from the inactive box is selected.
# Remove it and place it inside the active box
item_taken = self.active_items_list.takeItem(
self.active_items_list.currentRow())
self.inactive_items_list.addItem(item_taken)
# update activeItems
active_items = {}
inactive_items = {}
for i in range(self.active_items_list.count()):
active_items[self.active_items_list.item(i).text()] =\
ITEMS[self.active_items_list.item(i).text()
.encode('utf-8')]
for i in range(self.inactive_items_list.count()):
inactive_items[self.inactive_items_list.item(i).text()] =\
ITEMS[self.inactive_items_list.item(i).text()
.encode('utf-8')]
Constants.ACTIVE_ITEMS = active_items
Constants.INACTIVE_ITEMS = inactive_items
# clear the selection combobox
self.app_select_box.clear()
# Repopulate the combobox
self.app_select_box.addItem(None)
for item in active_items:
self.app_select_box.addItem(item)
# Updates the active items Constants property
Constants.CONFIG.set('inactive', item_taken.text(),
ITEMS[item_taken.text()])
Constants.CONFIG.remove_option('active', item_taken.text())
# Updates the config file to be up to date with activeItems
Constants.CONFIG.update()
except:
raise
def disp_dialog(self):
""" displays the dialog which select a directory for output. """
fname = QFileDialog.getExistingDirectory()
Constants.CONFIG.set('directories', 'current_song', fname)
self.output_cur_dir_lbl.setText(Constants.CONFIG.
get('directories', 'current_song'))
def select_new_app(self, text):
""" Sets the new application to check for """
try:
Main.selectedProgram = ITEMS[text]
except KeyError:
# catches the empty option, it's obviously not in the dict
pass
# custom message for zune
if Main.selectedProgram == 'zune':
self.misc_messages.setText(Misc.ZuneNotification)
# custom message for webplayers which require the groovemarklet
elif text.find('*'):
self.misc_messages.setText(Misc.GetGroovemarklet)
def start(self):
""" When the start button is pressed, start the main program loop """
if Main.selectedProgram:
if not Main.running:
self.start_btn.setText('Stop')
Main.running = True
try:
pythoncom.CoInitializeEx(pythoncom.COINIT_MULTITHREADED)
except pythoncom.com_error:
# already initialized.
pass
thread = Thread(
target=Main.enumWindows, name='enumWindows')
thread.run()
else:
self.start_btn.setText('Start')
Main.running = False
self.set_playing(Misc.noSongPlaying)
Wr.write('')
def set_playing(self, title=''):
""" Sets the text of the label of what song is playing """
# print 'setting title: ', title
self.current_playing.setText(title)
class ServiceBrowser(QWidget):
def __init__(self, parent = None):
QWidget.__init__(self, parent)
self.serviceManager = QServiceManager(self)
self.registerExampleServices()
self.initWidgets()
self.reloadServicesList()
self.setWindowTitle(self.tr("Services Browser"))
def __del__(self):
self.unregisterExampleServices()
def currentInterfaceImplChanged(self, current, previous):
if not current:
return
descriptor = current.data(Qt.UserRole)
self.reloadAttributesList()
self.reloadAttributesRadioButtonText()
if descriptor.isValid():
self.defaultInterfaceButton.setText(self.tr("Set as default implementation for %s" % str(descriptor.interfaceName())))
self.defaultInterfaceButton.setEnabled(True)
def reloadServicesList(self):
self.servicesListWidget.clear()
services = self.serviceManager.findServices()
for serv in services:
self.servicesListWidget.addItem(serv)
self.servicesListWidget.addItem(self.showAllServicesItem)
self._services = services
def reloadInterfaceImplementationsList(self):
serviceName = None
allServices = self.servicesListWidget.currentItem().text() == self.showAllServicesItem.text()
if self.servicesListWidget.currentItem() and not allServices:
serviceName = self.servicesListWidget.currentItem().text()
self.interfacesGroup.setTitle(self.tr("Interfaces implemented by %s" % str(serviceName)))
else:
self.interfacesGroup.setTitle(self.tr("All interface implementations"))
descriptors = self.serviceManager.findInterfaces(serviceName)
self.attributesListWidget.clear()
self.interfacesListWidget.clear()
self._i = []
for desc in descriptors:
text = "%s %d.%d" % (desc.interfaceName(), desc.majorVersion(), desc.minorVersion())
if not serviceName:
text += " (" + desc.serviceName() + ")"
defaultInterfaceImpl = self.serviceManager.interfaceDefault(desc.interfaceName())
if desc == defaultInterfaceImpl:
text += self.tr(" (default)")
item = QListWidgetItem(text)
item.setData(Qt.UserRole, desc)
item._data = desc
self.interfacesListWidget.addItem(item)
self.defaultInterfaceButton.setEnabled(False)
def reloadAttributesList(self):
item = self.interfacesListWidget.currentItem()
if not item:
return
selectedImpl = item.data(Qt.UserRole)
implementationRef = None
if self.selectedImplRadioButton.isChecked():
implementationRef = self.serviceManager.loadInterface(selectedImpl)
else:
implementationRef = self.serviceManager.loadInterface(selectedImpl.interfaceName())
self.attributesListWidget.clear()
if not implementationRef:
self.attributesListWidget.addItem(self.tr("(Error loading service plugin)"))
return
metaObject = implementationRef.metaObject()
self.attributesGroup.setTitle(self.tr("Invokable attributes for %s class" % metaObject.className()))
for i in range(metaObject.methodCount()):
method = metaObject.method(i)
self.attributesListWidget.addItem("[METHOD] " + method.signature())
for i in range(metaObject.propertyCount()):
p = metaObject.property(i)
self.attributesListWidget.addItem("[PROPERTY] " + p.name())
def setDefaultInterfaceImplementation(self):
item = self.interfacesListWidget.currentItem()
if not item:
return
descriptor = item.data(Qt.UserRole)
if descriptor.isValid():
if self.serviceManager.setInterfaceDefault(descriptor):
currentIndex = self.interfacesListWidget.row(item)
self.reloadInterfaceImplementationsList()
self.interfacesListWidget.setCurrentRow(currentIndex)
else:
print "Unable to set default service for interface:", descriptor.interfaceName()
def registerExampleServices(self):
exampleXmlFiles = ["filemanagerservice.xml", "bluetoothtransferservice.xml"]
for fileName in exampleXmlFiles:
path = "./xmldata/" + fileName
self.serviceManager.addService(path)
def unregisterExampleServices(self):
self.serviceManager.removeService("FileManagerService")
self.serviceManager.removeService("BluetoothTransferService")
def reloadAttributesRadioButtonText(self):
item = self.interfacesListWidget.currentItem()
if not item:
return
selectedImpl = item.data(Qt.UserRole)
defaultImpl = self.serviceManager.interfaceDefault(selectedImpl.interfaceName())
self.defaultImplRadioButton.setText(self.tr("Default implementation for %s\n(currently provided by %s)" % (str(defaultImpl.interfaceName()), str(defaultImpl.serviceName()))))
def initWidgets(self):
self.showAllServicesItem = QListWidgetItem(self.tr("(All registered services)"))
self.servicesListWidget = QListWidget()
self.interfacesListWidget = QListWidget()
self.interfacesListWidget.addItem(self.tr("(Select a service)"))
self.attributesListWidget = QListWidget()
self.attributesListWidget.addItem(self.tr("(Select an interface implementation)"))
self.interfacesListWidget.setMinimumWidth(450)
self.servicesListWidget.currentItemChanged.connect(self.reloadInterfaceImplementationsList)
self.interfacesListWidget.currentItemChanged.connect(self.currentInterfaceImplChanged)
self.defaultInterfaceButton = QPushButton(self.tr("Set as default implementation"))
self.defaultInterfaceButton.setEnabled(False)
self.defaultInterfaceButton.clicked.connect(self.setDefaultInterfaceImplementation)
self.selectedImplRadioButton = QRadioButton(self.tr("Selected interface implementation"))
self.defaultImplRadioButton = QRadioButton(self.tr("Default implementation"))
self.selectedImplRadioButton.setChecked(True)
self.radioButtons = QButtonGroup(self)
self.radioButtons.addButton(self.selectedImplRadioButton)
self.radioButtons.addButton(self.defaultImplRadioButton)
self.radioButtons.buttonClicked.connect(self.reloadAttributesList)
self.servicesGroup = QGroupBox(self.tr("Show services for:"))
servicesLayout = QVBoxLayout()
servicesLayout.addWidget(self.servicesListWidget)
self.servicesGroup.setLayout(servicesLayout)
self.interfacesGroup = QGroupBox(self.tr("Interface implementations"))
interfacesLayout = QVBoxLayout()
interfacesLayout.addWidget(self.interfacesListWidget)
interfacesLayout.addWidget(self.defaultInterfaceButton)
self.interfacesGroup.setLayout(interfacesLayout)
self.attributesGroup = QGroupBox(self.tr("Invokable attributes"))
attributesLayout = QVBoxLayout()
self.attributesGroup.setLayout(attributesLayout)
attributesLayout.addWidget(self.attributesListWidget)
attributesLayout.addWidget(QLabel(self.tr("Show attributes for:")))
attributesLayout.addWidget(self.selectedImplRadioButton)
attributesLayout.addWidget(self.defaultImplRadioButton)
self.attributesGroup.setLayout(attributesLayout)
layout = QGridLayout()
layout.addWidget(self.servicesGroup, 0, 0)
layout.addWidget(self.attributesGroup, 0, 1, 2, 1)
layout.addWidget(self.interfacesGroup, 1, 0)
self.setLayout(layout)
class beso_gui(QDialog):
def __init__(self):
super().__init__()
self.title = 'BESO Topology Optimization (experimental)'
self.left = 250
self.top = 30
self.width = 550
self.height = 730
beso_gui.inp_file = ""
beso_gui.beso_dir = os.path.dirname(__file__)
self.initUI()
# def closeEvent(self, event):
# self.close()
def initUI(self):
self.setWindowTitle(self.title)
self.setGeometry(self.left, self.top, self.width, self.height)
# Select analysis file button
button = QPushButton('Select analysis file', self)
button.setToolTip('*.inp CalculiX analysis file.')
button.move(10, 10)
button.clicked.connect(self.on_click)
# Text box - file path and name
self.textbox_file_name = QLineEdit(self)
self.textbox_file_name.move(120, 15)
self.textbox_file_name.resize(420, 20)
self.textbox_file_name.setText("None analysis file selected")
self.textbox_file_name.setToolTip('Analysis file.')
# Update button
button1 = QPushButton('Update domains', self)
button1.setToolTip('Update naming inputs and material data from FreeCAD.')
button1.move(10, 50)
button1.clicked.connect(self.on_click1)
# Domains definition
# Label above domains definition
label21 = QLabel('Domain 0', self)
label21.setStyleSheet("font-weight: bold")
label21.move(120, 50)
label21 = QLabel('Domain 1', self)
label21.setStyleSheet("font-weight: bold")
label21.move(260, 50)
label21 = QLabel('Domain 2', self)
label21.setStyleSheet("font-weight: bold")
label21.move(400, 50)
label24 = QLabel('Material object', self)
label24.move(20, 80)
label25 = QLabel('Thickness object', self)
label25.move(20, 110)
label26 = QLabel('As design domain', self)
label26.move(20, 140)
label27 = QLabel('Stress limit [MPa]', self)
label27.move(20, 170)
# Combo box - select domain by material object
self.combo = QComboBox(self)
self.combo.setToolTip('Material object to define the domain.')
self.combo.move(120, 80)
self.combo.resize(140, 30)
self.combo.currentIndexChanged.connect(self.on_change)
self.combo1 = QComboBox(self)
self.combo1.setToolTip('Material object to define the domain.')
self.combo1.move(260, 80)
self.combo1.resize(140, 30)
self.combo1.currentIndexChanged.connect(self.on_change1)
self.combo2 = QComboBox(self)
self.combo2.setToolTip('Material object to define the domain.')
self.combo2.move(400, 80)
self.combo2.resize(140, 30)
self.combo2.currentIndexChanged.connect(self.on_change2)
# Combo box - select thickness object
self.combo0t = QComboBox(self)
self.combo0t.setToolTip('Thickness object to specify if domain is for shells.')
self.combo0t.move(120, 110)
self.combo0t.resize(140, 30)
self.combo1t = QComboBox(self)
self.combo1t.setToolTip('Thickness object to specify if domain is for shells.')
self.combo1t.move(260, 110)
self.combo1t.resize(140, 30)
self.combo2t = QComboBox(self)
self.combo2t.setToolTip('Thickness object to specify if domain is for shells.')
self.combo2t.move(400, 110)
self.combo2t.resize(140, 30)
self.textbox3 = QLineEdit(self)
self.textbox3.move(120, 170)
self.textbox3.resize(40, 20)
# self.textbox3.setText("")
self.textbox3.setToolTip('Thickness [mm] of shell elements in the domain.\n'
'This value overwrites thickness defined in FreeCAD')
self.textbox4 = QLineEdit(self)
self.textbox4.move(260, 170)
self.textbox4.resize(40, 20)
# self.textbox4.setText("")
self.textbox4.setToolTip('Thickness [mm] of shell elements in the domain.\n'
'This value overwrites thickness defined in FreeCAD')
self.textbox5 = QLineEdit(self)
self.textbox5.move(400, 170)
self.textbox5.resize(40, 20)
# self.textbox5.setText("")
self.textbox5.setToolTip('Thickness [mm] of shell elements in the domain.\n'
'This value overwrites thickness defined in FreeCAD')
# Check box - design or nondesign
self.checkbox = QCheckBox('', self)
self.checkbox.setChecked(True)
self.checkbox.setToolTip('Check to be the design domain.')
self.checkbox.move(120, 140)
self.checkbox1 = QCheckBox('', self)
self.checkbox1.setChecked(True)
self.checkbox1.setToolTip('Check to be the design domain.')
self.checkbox1.move(260, 140)
self.checkbox2 = QCheckBox('', self)
self.checkbox2.setChecked(True)
self.checkbox2.setToolTip('Check to be the design domain.')
self.checkbox2.move(400, 140)
# Text box - stress limit
self.textbox = QLineEdit(self)
self.textbox.move(120, 170)
self.textbox.resize(40, 20)
# self.textbox.setText("")
self.textbox.setToolTip('Von Mises stress [MPa] limit, when reached, material removing will stop.')
self.textbox1 = QLineEdit(self)
self.textbox1.move(260, 170)
self.textbox1.resize(40, 20)
# self.textbox1.setText("")
self.textbox1.setToolTip('Von Mises stress [MPa] limit, when reached, material removing will stop.')
self.textbox2 = QLineEdit(self)
self.textbox2.move(400, 170)
self.textbox2.resize(40, 20)
# self.textbox2.setText("")
self.textbox2.setToolTip('Von Mises stress [MPa] limit, when reached, material removing will stop.')
# Filters
# Label above filter definition
label31 = QLabel('Filter 0', self)
label31.setStyleSheet("font-weight: bold")
label31.move(120, 210)
label32 = QLabel('Filter 1', self)
label32.setStyleSheet("font-weight: bold")
label32.move(260, 210)
label33 = QLabel('Filter 2', self)
label33.setStyleSheet("font-weight: bold")
label33.move(400, 210)
label34 = QLabel('Type', self)
label34.move(20, 240)
label35 = QLabel('Range [mm]', self)
label35.move(20, 270)
label36 = QLabel('Direction vector', self)
label36.move(20, 300)
label37 = QLabel('Apply to', self)
label37.move(20, 330)
# Combo box - select filter type
self.combo6 = QComboBox(self)
self.combo6.setToolTip('Filters:\n'
'"simple" to suppress checkerboard effect,\n'
'"casting" to prescribe casting direction (opposite to milling direction)\n'
'Recommendation: for casting use as first "casting" and as second "simple"')
self.combo6.addItem("None")
self.combo6.addItem("simple")
self.combo6.addItem("casting")
self.combo6.setCurrentIndex(1)
self.combo6.move(120, 240)
self.combo6.currentIndexChanged.connect(self.on_change6)
self.combo7 = QComboBox(self)
self.combo7.setToolTip('Filters:\n'
'"simple" to suppress checkerboard effect,\n'
'"casting" to prescribe casting direction (opposite to milling direction)\n'
'Recommendation: for casting use as first "casting" and as second "simple"')
self.combo7.addItem("None")
self.combo7.addItem("simple")
self.combo7.addItem("casting")
self.combo7.move(260, 240)
self.combo7.currentIndexChanged.connect(self.on_change7)
self.combo8 = QComboBox(self)
self.combo8.setToolTip('Filters:\n'
'"simple" to suppress checkerboard effect,\n'
'"casting" to prescribe casting direction (opposite to milling direction)\n'
'Recommendation: for casting use as first "casting" and as second "simple"')
self.combo8.addItem("None")
self.combo8.addItem("simple")
self.combo8.addItem("casting")
self.combo8.move(400, 240)
self.combo8.currentIndexChanged.connect(self.on_change8)
# Text box - filter range
self.textbox6 = QLineEdit(self)
self.textbox6.move(120, 270)
self.textbox6.resize(50, 20)
# self.textbox6.setText("")
self.textbox6.setToolTip('Filter range [mm], recommended two times mesh size.')
self.textbox7 = QLineEdit(self)
self.textbox7.move(260, 270)
self.textbox7.resize(50, 20)
# self.textbox7.setText("")
self.textbox7.setToolTip('Filter range [mm], recommended two times mesh size.')
self.textbox7.setEnabled(False)
self.textbox8 = QLineEdit(self)
self.textbox8.move(400, 270)
self.textbox8.resize(50, 20)
# self.textbox8.setText("")
self.textbox8.setToolTip('Filter range [mm], recommended two times mesh size.')
self.textbox8.setEnabled(False)
# Text box - casting direction
self.textbox9 = QLineEdit(self)
self.textbox9.move(120, 300)
self.textbox9.resize(80, 20)
self.textbox9.setText("0, 0, 1")
self.textbox9.setEnabled(False)
self.textbox9.setToolTip('Casting direction vector, e.g. direction in z axis:\n'
'0, 0, 1\n\n'
'solid void\n'
'XXXXXX.................\n'
'XXX........................\n'
'XX........................... --> z axis\n'
'XXXXX....................\n'
'XXXXXXXXXXX......')
self.textbox10 = QLineEdit(self)
self.textbox10.move(260, 300)
self.textbox10.resize(80, 20)
self.textbox10.resize(80, 20)
self.textbox10.setText("0, 0, 1")
self.textbox10.setEnabled(False)
self.textbox10.setToolTip('Casting direction vector, e.g. direction in z axis:\n'
'0, 0, 1\n\n'
'solid void\n'
'XXXXXX.................\n'
'XXX........................\n'
'XX........................... --> z axis\n'
'XXXXX....................\n'
'XXXXXXXXXXX......')
self.textbox11 = QLineEdit(self)
self.textbox11.move(400, 300)
self.textbox11.resize(80, 20)
self.textbox11.setText("0, 0, 1")
self.textbox11.setEnabled(False)
self.textbox11.setToolTip('Casting direction vector, e.g. direction in z axis:\n'
'0, 0, 1\n\n'
'solid void\n'
'XXXXXX.................\n'
'XXX........................\n'
'XX........................... --> z axis\n'
'XXXXX....................\n'
'XXXXXXXXXXX......')
# list widget - select domains
self.widget = QListWidget(self)
self.widget.setToolTip('Domains affected by the filter.\n'
'Select only from domains which you defined above.')
self.widget.move(120, 330)
self.widget.resize(140, 120)
self.widget.setSelectionMode(QAbstractItemView.MultiSelection)
self.widget1 = QListWidget(self)
self.widget1.setToolTip('Domains affected by the filter.\n'
'Select only from domains which you defined above.')
self.widget1.move(260, 330)
self.widget1.resize(140, 120)
self.widget1.setSelectionMode(QAbstractItemView.MultiSelection)
self.widget1.setEnabled(False)
self.widget2 = QListWidget(self)
self.widget2.setToolTip('Domains affected by the filter.\n'
'Select only from domains which you defined above.')
self.widget2.move(400, 330)
self.widget2.resize(140, 120)
self.widget2.setSelectionMode(QAbstractItemView.MultiSelection)
self.widget2.setEnabled(False)
# Other settings
label40 = QLabel('Other settings', self)
label40.setStyleSheet("font-weight: bold")
label40.move(10, 470)
# AR, RR slider
label41 = QLabel('Change per iteration: low', self)
label41.setFixedWidth(160)
label41.move(10, 500)
label42 = QLabel('high', self)
label42.move(240, 500)
self.slider = QSlider(Qt.Horizontal, self)
self.slider.setRange(1, 3)
self.slider.setSingleStep(1)
self.slider.setValue(2)
self.slider.move(150, 500)
self.slider.resize(80, 30)
self.slider.setToolTip('Sets mass change per iteration, which is controlled as\n'
'slow: mass_addition_ratio=0.01, mass_removal_ratio=0.02\n'
'middle: mass_addition_ratio=0.015, mass_removal_ratio=0.03\n'
'fast: mass_addition_ratio=0.03, mass_removal_ratio=0.06')
# optimization base combobox
label51 = QLabel('Optimization base', self)
label51.move(10, 530)
self.combo51 = QComboBox(self)
self.combo51.setToolTip('Basic principle to determine if element should remain or be removed:\n'
'"stiffness" to maximize stiffness (minimize compliance),\n'
'"heat" to maximize heat flow.')
self.combo51.addItem("stiffness")
self.combo51.addItem("heat")
self.combo51.move(120, 530)
# mass goal ratio
label52 = QLabel('Mass goal ratio', self)
label52.move(10, 560)
self.textbox52 = QLineEdit(self)
self.textbox52.move(120, 560)
self.textbox52.resize(50, 20)
self.textbox52.setText("0.4")
self.textbox52.setToolTip('Fraction of all design domains masses to be achieved;\n'
'between 0 and 1.')
# generate conf. file button
button21 = QPushButton('Generate conf. file', self)
button21.setToolTip('Writes configuration file with optimization parameters.')
button21.move(10, 600)
button21.clicked.connect(self.on_click21)
# edit conf. file button
button22 = QPushButton('Edit conf. file', self)
button22.setToolTip('Opens configuration file for hand modifications.')
button22.move(10, 630)
button22.clicked.connect(self.on_click22)
# run optimization button
button23 = QPushButton('Run optimization', self)
button23.setToolTip('Writes configuration file and runs optimization.')
button23.move(10, 660)
button23.clicked.connect(self.on_click23)
# generate conf file and run optimization button
button24 = QPushButton('Generate conf.\nfile and run\noptimization', self)
button24.setToolTip('Writes configuration file and runs optimization.')
button24.move(120, 600)
button24.resize(100, 90)
button24.clicked.connect(self.on_click24)
# help buttons
label41 = QLabel('Help', self)
label41.move(440, 560)
button31 = QPushButton('Example', self)
button31.setToolTip('https://github.com/fandaL/beso/wiki/Example-4:-GUI-in-FreeCAD')
button31.move(440, 590)
# button31.resize(80, 50)
button31.clicked.connect(self.on_click31)
button32 = QPushButton('Conf. comments', self)
button32.setToolTip('https://github.com/fandaL/beso/blob/master/beso_conf.py')
button32.move(440, 620)
# button32.resize(80, 50)
button32.clicked.connect(self.on_click32)
button33 = QPushButton('Close', self)
button33.move(440, 690)
# button33.resize(80, 50)
button33.clicked.connect(self.on_click33)
# open log file
button40 = QPushButton('Open log file', self)
button40.setToolTip('Opens log file in your text editor.\n'
'(Does not refresh automatically.)')
button40.move(10, 690)
button40.clicked.connect(self.on_click40)
self.on_click1() # first update
self.show()
# @pyqtSlot()
def on_click(self):
ex2 = SelectFile()
self.show()
self.textbox_file_name.setText(self.inp_file)
def on_click1(self):
# get material objects
self.materials = []
self.thicknesses = []
try:
App.ActiveDocument.Objects
except AttributeError:
App.newDocument("Unnamed")
print("Warning: Missing active document with FEM analysis. New document have been created.")
for obj in App.ActiveDocument.Objects:
if obj.Name[:23] == "MechanicalSolidMaterial":
self.materials.append(obj)
elif obj.Name[:13] == "MaterialSolid":
self.materials.append(obj)
elif obj.Name[:13] == "SolidMaterial":
self.materials.append(obj)
elif obj.Name[:17] == "ElementGeometry2D":
self.thicknesses.append(obj)
# update materials combo boxes
self.combo.clear()
self.combo.addItem("None")
self.combo1.clear()
self.combo1.addItem("None")
self.combo2.clear()
self.combo2.addItem("None")
self.combo0t.clear()
self.combo0t.addItem("None")
self.combo1t.clear()
self.combo1t.addItem("None")
self.combo2t.clear()
self.combo2t.addItem("None")
self.widget.clear()
self.widget.addItem("All defined")
self.widget.addItem("Domain 0")
self.widget.addItem("Domain 1")
self.widget.addItem("Domain 2")
self.widget.setCurrentItem(self.widget.item(0))
self.widget1.clear()
self.widget1.addItem("All defined")
self.widget1.addItem("Domain 0")
self.widget1.addItem("Domain 1")
self.widget1.addItem("Domain 2")
self.widget1.setCurrentItem(self.widget1.item(0))
self.widget2.clear()
self.widget2.addItem("All defined")
self.widget2.addItem("Domain 0")
self.widget2.addItem("Domain 1")
self.widget2.addItem("Domain 2")
self.widget2.setCurrentItem(self.widget2.item(0))
for mat in self.materials:
self.combo.addItem(mat.Label)
self.combo1.addItem(mat.Label)
self.combo2.addItem(mat.Label)
if self.materials:
self.combo.setCurrentIndex(1)
for th in self.thicknesses:
self.combo0t.addItem(th.Label)
self.combo1t.addItem(th.Label)
self.combo2t.addItem(th.Label)
def on_click21(self):
"""Overwrite beso_conf.py file in the macro directory"""
file_name = os.path.split(self.textbox_file_name.text())[1]
path = os.path.split(self.textbox_file_name.text())[0]
fea = ccxtools.FemToolsCcx()
fea.setup_ccx()
path_calculix = fea.ccx_binary
optimization_base = self.combo51.currentText()
elset_id = self.combo.currentIndex() - 1
thickness_id = self.combo0t.currentIndex() - 1
if elset_id != -1:
if thickness_id != -1:
elset = self.materials[elset_id].Name + self.thicknesses[thickness_id].Name
else: # 0 means None thickness selected
elset = self.materials[elset_id].Name + "Solid"
modulus = float(self.materials[elset_id].Material["YoungsModulus"].split()[0]) # MPa
if self.materials[elset_id].Material["YoungsModulus"].split()[1] != "MPa":
raise Exception(" units not recognised in " + self.materials[elset_id].Name)
poisson = float(self.materials[elset_id].Material["PoissonRatio"].split()[0])
try:
density = float(self.materials[elset_id].Material["Density"].split()[0]) * 1e-12 # kg/m3 -> t/mm3
if self.materials[elset_id].Material["Density"].split()[1] not in ["kg/m^3", "kg/m3"]:
raise Exception(" units not recognised in " + self.materials[elset_id].Name)
except KeyError:
density = 0.
try:
conductivity = float(self.materials[elset_id].Material["ThermalConductivity"].split()[0]) # W/m/K
if self.materials[elset_id].Material["ThermalConductivity"].split()[1] != "W/m/K":
raise Exception(" units not recognised in " + self.materials[elset_id].Name)
except KeyError:
conductivity = 0.
try:
if self.materials[elset_id].Material["ThermalExpansionCoefficient"].split()[1] == "um/m/K":
expansion = float(self.materials[elset_id].Material["ThermalExpansionCoefficient"].split()[
0]) * 1e-6 # um/m/K -> mm/mm/K
elif self.materials[elset_id].Material["ThermalExpansionCoefficient"].split()[1] == "m/m/K":
expansion = float(self.materials[elset_id].Material["ThermalExpansionCoefficient"].split()[
0]) # m/m/K -> mm/mm/K
else:
raise Exception(" units not recognised in " + self.materials[elset_id].Name)
except KeyError:
expansion = 0.
try:
specific_heat = float(self.materials[elset_id].Material["SpecificHeat"].split()[
0]) * 1e6 # J/kg/K -> mm^2/s^2/K
if self.materials[elset_id].Material["SpecificHeat"].split()[1] != "J/kg/K":
raise Exception(" units not recognised in " + self.materials[elset_id].Name)
except KeyError:
specific_heat = 0.
if thickness_id != -1:
thickness = str(self.thicknesses[thickness_id].Thickness).split()[0] # mm
if str(self.thicknesses[thickness_id].Thickness).split()[1] != "mm":
raise Exception(" units not recognised in " + self.thicknesses[thickness_id].Name)
else:
thickness = 0.
optimized = self.checkbox.isChecked()
if self.textbox.text():
von_mises = float(self.textbox.text())
else:
von_mises = 0.
elset_id1 = self.combo1.currentIndex() - 1
thickness_id1 = self.combo1t.currentIndex() - 1
if elset_id1 != -1:
if thickness_id1 != -1:
elset1 = self.materials[elset_id1].Name + self.thicknesses[thickness_id1].Name
else: # 0 means None thickness selected
elset1 = self.materials[elset_id1].Name + "Solid"
modulus1 = float(self.materials[elset_id1].Material["YoungsModulus"].split()[0]) # MPa
if self.materials[elset_id1].Material["YoungsModulus"].split()[1] != "MPa":
raise Exception(" units not recognised in " + self.materials[elset_id1].Name)
poisson1 = float(self.materials[elset_id1].Material["PoissonRatio"].split()[0])
try:
density1 = float(self.materials[elset_id1].Material["Density"].split()[0]) * 1e-12 # kg/m3 -> t/mm3
if self.materials[elset_id1].Material["Density"].split()[1] not in ["kg/m^3", "kg/m3"]:
raise Exception(" units not recognised in " + self.materials[elset_id1].Name)
except KeyError:
density1 = 0.
try:
conductivity1 = float(self.materials[elset_id1].Material["ThermalConductivity"].split()[0]) # W/m/K
if self.materials[elset_id1].Material["ThermalConductivity"].split()[1] != "W/m/K":
raise Exception(" units not recognised in " + self.materials[elset_id1].Name)
except KeyError:
conductivity1 = 0.
try:
if self.materials[elset_id1].Material["ThermalExpansionCoefficient"].split()[1] == "um/m/K":
expansion1 = float(self.materials[elset_id1].Material["ThermalExpansionCoefficient"].split()[
0]) * 1e-6 # um/m/K -> mm/mm/K
elif self.materials[elset_id1].Material["ThermalExpansionCoefficient"].split()[1] == "m/m/K":
expansion1 = float(self.materials[elset_id1].Material["ThermalExpansionCoefficient"].split()[
0]) # m/m/K -> mm/mm/K
else:
raise Exception(" units not recognised in " + self.materials[elset_id1].Name)
except KeyError:
expansion1 = 0.
try:
specific_heat1 = float(self.materials[elset_id1].Material["SpecificHeat"].split()[
0]) * 1e6 # J/kg/K -> mm^2/s^2/K
if self.materials[elset_id1].Material["SpecificHeat"].split()[1] != "J/kg/K":
raise Exception(" units not recognised in " + self.materials[elset_id1].Name)
except KeyError:
specific_heat1 = 0.
if thickness_id1 != -1:
thickness1 = str(self.thicknesses[thickness_id1].Thickness).split()[0] # mm
if str(self.thicknesses[thickness_id1].Thickness).split()[1] != "mm":
raise Exception(" units not recognised in " + self.thicknesses[thickness_id1].Name)
else:
thickness1 = 0.
optimized1 = self.checkbox1.isChecked()
if self.textbox1.text():
von_mises1 = float(self.textbox1.text())
else:
von_mises1 = 0.
elset_id2 = self.combo2.currentIndex() - 1
thickness_id2 = self.combo2t.currentIndex() - 1
if elset_id2 != -1:
if thickness_id2 != -1:
elset2 = self.materials[elset_id2].Name + self.thicknesses[thickness_id2].Name
else: # 0 means None thickness selected
else2t = self.materials[elset_id2].Name + "Solid"
modulus2 = float(self.materials[elset_id2].Material["YoungsModulus"].split()[0]) # MPa
if self.materials[elset_id2].Material["YoungsModulus"].split()[1] != "MPa":
raise Exception(" units not recognised in " + self.materials[elset_id2].Name)
poisson2 = float(self.materials[elset_id2].Material["PoissonRatio"].split()[0])
try:
density2 = float(self.materials[elset_id2].Material["Density"].split()[0]) * 1e-12 # kg/m3 -> t/mm3
if self.materials[elset_id2].Material["Density"].split()[1] not in ["kg/m^3", "kg/m3"]:
raise Exception(" units not recognised in " + self.materials[elset_id2].Name)
except KeyError:
density2 = 0.
try:
conductivity2 = float(self.materials[elset_id2].Material["ThermalConductivity"].split()[0]) # W/m/K
if self.materials[elset_id2].Material["ThermalConductivity"].split()[1] != "W/m/K":
raise Exception(" units not recognised in " + self.materials[elset_id2].Name)
except KeyError:
conductivity2 = 0.
try:
if self.materials[elset_id2].Material["ThermalExpansionCoefficient"].split()[1] == "um/m/K":
expansion2 = float(self.materials[elset_id2].Material["ThermalExpansionCoefficient"].split()[
0]) * 1e-6 # um/m/K -> mm/mm/K
elif self.materials[elset_id2].Material["ThermalExpansionCoefficient"].split()[1] == "m/m/K":
expansion2 = float(self.materials[elset_id2].Material["ThermalExpansionCoefficient"].split()[
0]) # m/m/K -> mm/mm/K
else:
raise Exception(" units not recognised in " + self.materials[elset_id2].Name)
except KeyError:
expansion2 = 0.
try:
specific_heat2 = float(self.materials[elset_id2].Material["SpecificHeat"].split()[
0]) * 1e6 # J/kg/K -> mm^2/s^2/K
if self.materials[elset_id2].Material["SpecificHeat"].split()[1] != "J/kg/K":
raise Exception(" units not recognised in " + self.materials[elset_id2].Name)
except KeyError:
specific_heat2 = 0.
if thickness_id2 != -1:
thickness2 = str(self.thicknesses[thickness_id2].Thickness).split()[0] # mm
if str(self.thicknesses[thickness_id2].Thickness).split()[1] != "mm":
raise Exception(" units not recognised in " + self.thicknesses[thickness_id2].Name)
else:
thickness2 = 0.
optimized2 = self.checkbox2.isChecked()
if self.textbox2.text():
von_mises2 = float(self.textbox2.text())
else:
von_mises2 = 0.
with open(os.path.join(self.beso_dir, "beso_conf.py"), "w") as f:
f.write("# This is the configuration file with input parameters. It will be executed as python commands\n")
f.write("# Written by beso_fc_gui.py at {}\n".format(datetime.datetime.now()))
f.write("\n")
f.write("path_calculix = '{}'\n".format(path_calculix))
f.write("path = '{}'\n".format(path))
f.write("file_name = '{}'\n".format(file_name))
f.write("\n")
if elset_id != -1:
f.write("elset_name = '{}'\n".format(elset))
f.write("domain_optimized[elset_name] = {}\n".format(optimized))
f.write("domain_density[elset_name] = [{}, {}]\n".format(density * 1e-6, density))
if thickness:
f.write("domain_thickness[elset_name] = [{}, {}]\n".format(thickness, thickness))
if von_mises:
f.write("domain_FI[elset_name] = [[('stress_von_Mises', {:.6})],\n".format(von_mises * 1e6))
f.write(" [('stress_von_Mises', {:.6})]]\n".format(von_mises))
f.write("domain_material[elset_name] = ['*ELASTIC\\n{:.6}, {}\\n*DENSITY\\n{:.6}\\n*CONDUCTIVITY\\n"
"{:.6}\\n*EXPANSION\\n{:.6}\\n*SPECIFIC HEAT\\n{:.6}\\n',\n".format(modulus * 1e-6, poisson,
density * 1e-6, conductivity * 1e-6, expansion * 1e-6, specific_heat * 1e-6))
f.write(" '*ELASTIC\\n{:.6}, {:.6}\\n*DENSITY\\n{:.6}\\n*CONDUCTIVITY\\n"
"{:.6}\\n*EXPANSION\\n{:.6}\\n*SPECIFIC HEAT\\n{:.6}\\n']\n".format(modulus, poisson, density,
conductivity, expansion, specific_heat))
f.write("\n")
if elset_id1 != -1:
f.write("elset_name = '{}'\n".format(elset1))
f.write("domain_optimized[elset_name] = {}\n".format(optimized1))
f.write("domain_density[elset_name] = [{}, {}]\n".format(density1 * 1e-6, density1))
if thickness1:
f.write("domain_thickness[elset_name] = [{}, {}]\n".format(thickness1, thickness1))
if von_mises1:
f.write("domain_FI[elset_name] = [[('stress_von_Mises', {:.6})],\n".format(von_mises1 * 1e6))
f.write(" [('stress_von_Mises', {:.6})]]\n".format(von_mises1))
f.write("domain_material[elset_name] = ['*ELASTIC\\n{:.6}, {:.6}\\n*DENSITY\\n{:.6}\\n*CONDUCTIVITY"
"\\n{:.6}\\n*EXPANSION\\n{:.6}\\n*SPECIFIC HEAT\\n{:.6}\\n',\n".format(modulus1 * 1e-6,
poisson1, density1 * 1e-6, conductivity1 * 1e-6, expansion1 * 1e-6, specific_heat1 * 1e-6))
f.write(" '*ELASTIC\\n{:.6}, {:.6}\\n*DENSITY\\n{:.6}\\n*CONDUCTIVITY\\n"
"{:.6}\\n" "*EXPANSION\\n{:.6}\\n*SPECIFIC HEAT\\n{:.6}\\n']\n".format(modulus1, poisson1,
density1, conductivity1, expansion1, specific_heat1))
f.write("\n")
if elset_id2 != -1:
f.write("elset_name = '{}'\n".format(elset2))
f.write("domain_optimized[elset_name] = {}\n".format(optimized2))
f.write("domain_density[elset_name] = [{}, {}]\n".format(density2 * 1e-6, density2))
if thickness2:
f.write("domain_thickness[elset_name] = [{}, {}]\n".format(thickness2, thickness2))
if von_mises2:
f.write("domain_FI[elset_name] = [[('stress_von_Mises', {:.6})],\n".format(von_mises2 * 1e6))
f.write(" [('stress_von_Mises', {:.6})]]\n".format(von_mises2))
f.write("domain_material[elset_name] = ['*ELASTIC\\n{:.6}, {:.6}\\n*DENSITY\\n{:.6}\\n*CONDUCTIVITY"
"\\n{:.6}\\n*EXPANSION\\n{:.6}\\n*SPECIFIC HEAT\\n{:.6}\\n',\n".format(modulus2 * 1e-6,
poisson2, density2 * 1e-6, conductivity2 * 1e-6, expansion2 * 1e-6, specific_heat2 * 1e-6))
f.write(" '*ELASTIC\\n{:.6}, {:.6}\\n*DENSITY\\n{:.6}\\n*CONDUCTIVITY\\n"
"{:.6}\\n*EXPANSION\\n{:.6}\\n*SPECIFIC HEAT\\n{:.6}\\n']\n".format(modulus2, poisson2,
density2, conductivity2, expansion2, specific_heat2))
f.write("\n")
f.write("mass_goal_ratio = " + self.textbox52.text())
f.write("\n")
f.write("filter_list = [")
filter = self.combo6.currentText()
range = self.textbox6.text()
direction = self.textbox9.text()
selection = [item.text() for item in self.widget.selectedItems()]
filter_domains = []
if "All defined" not in selection:
if "Domain 0" in selection:
filter_domains.append(elset)
if "Domain 1" in selection:
filter_domains.append(elset1)
if "Domain 2" in selection:
filter_domains.append(elset2)
if filter == "simple":
f.write("['simple', {}".format(range))
for dn in filter_domains:
f.write(", '{}'".format(dn))
f.write("],\n")
elif filter == "casting":
f.write("['casting', {}, ({})".format(range, direction))
for dn in filter_domains:
f.write(", '{}'".format(dn))
f.write("],\n")
filter1 = self.combo7.currentText()
range1 = self.textbox7.text()
direction1 = self.textbox10.text()
selection = [item.text() for item in self.widget1.selectedItems()]
filter_domains1 = []
if "All defined" not in selection:
if "Domain 0" in selection:
filter_domains1.append(elset)
if "Domain 1" in selection:
filter_domains1.append(elset1)
if "Domain 2" in selection:
filter_domains1.append(elset2)
if filter1 == "simple":
f.write(" ['simple', {}".format(range1))
for dn in filter_domains1:
f.write(", '{}'".format(dn))
f.write("],\n")
elif filter1 == "casting":
f.write(" ['casting', {}, ({})".format(range1, direction1))
for dn in filter_domains1:
f.write(", '{}'".format(dn))
f.write("],\n")
filter2 = self.combo8.currentText()
range2 = self.textbox8.text()
direction2 = self.textbox11.text()
selection = [item.text() for item in self.widget2.selectedItems()]
filter_domains2 = []
if "All defined" not in selection:
if "Domain 0" in selection:
filter_domains2.append(elset)
if "Domain 1" in selection:
filter_domains2.append(elset1)
if "Domain 2" in selection:
filter_domains2.append(elset2)
if filter2 == "simple":
f.write(" ['simple', {}".format(range2))
for dn in filter_domains2:
f.write(", '{}'".format(dn))
f.write("],\n")
elif filter2 == "casting":
f.write(" ['casting', {}, ({})".format(range2, direction2))
for dn in filter_domains2:
f.write(", '{}'".format(dn))
f.write("],\n")
f.write(" ]\n")
f.write("\n")
f.write("optimization_base = '{}'\n".format(optimization_base))
f.write("\n")
slider_position = self.slider.value()
if slider_position == 1:
f.write("mass_addition_ratio = 0.01\n")
f.write("mass_removal_ratio = 0.02\n")
if slider_position == 2:
f.write("mass_addition_ratio = 0.015\n")
f.write("mass_removal_ratio = 0.03\n")
if slider_position == 3:
f.write("mass_addition_ratio = 0.03\n")
f.write("mass_removal_ratio = 0.06\n")
f.write("ratio_type = 'relative'\n")
f.write("\n")
def on_click22(self):
"""Open beso_conf.py in FreeCAD editor"""
FreeCADGui.insert(os.path.join(self.beso_dir, "beso_conf.py"))
def on_click23(self):
""""Run optimization"""
# run in own thread (not freezing FreeCAD): needs also to comment "plt.show()" on the end of beso_main.py
# self.optimization_thread = RunOptimization("beso_main")
# self.optimization_thread.start()
# run in foreground (freeze FreeCAD)
exec(open(os.path.join(beso_gui.beso_dir, "beso_main.py")).read())
def on_click24(self):
self.on_click21() # generate beso_conf.py
self.on_click23() # run optimization
def on_click31(self):
webbrowser.open_new_tab("https://github.com/fandaL/beso/wiki/Example-4:-GUI-in-FreeCAD")
def on_click32(self):
webbrowser.open_new_tab("https://github.com/fandaL/beso/blob/master/beso_conf.py")
def on_click33(self):
self.close()
def on_click40(self):
"""Open log file"""
if self.textbox_file_name.text() in ["None analysis file selected", ""]:
print("None analysis file selected")
else:
log_file = os.path.normpath(self.textbox_file_name.text()[:-4] + ".log")
webbrowser.open(log_file)
def on_change(self):
if self.combo.currentText() == "None":
self.combo0t.setEnabled(False)
self.checkbox.setEnabled(False)
self.textbox.setEnabled(False)
else:
self.combo0t.setEnabled(True)
self.checkbox.setEnabled(True)
self.textbox.setEnabled(True)
def on_change1(self):
if self.combo1.currentText() == "None":
self.combo1t.setEnabled(False)
self.checkbox1.setEnabled(False)
self.textbox1.setEnabled(False)
else:
self.combo1t.setEnabled(True)
self.checkbox1.setEnabled(True)
self.textbox1.setEnabled(True)
def on_change2(self):
if self.combo2.currentText() == "None":
self.combo2t.setEnabled(False)
self.checkbox2.setEnabled(False)
self.textbox2.setEnabled(False)
else:
self.combo2t.setEnabled(True)
self.checkbox2.setEnabled(True)
self.textbox2.setEnabled(True)
def on_change6(self):
if self.combo6.currentText() == "None":
self.textbox6.setEnabled(False)
self.textbox9.setEnabled(False)
self.widget.setEnabled(False)
elif self.combo6.currentText() == "casting":
self.textbox6.setEnabled(True)
self.textbox9.setEnabled(True)
self.widget.setEnabled(True)
else:
self.textbox6.setEnabled(True)
self.textbox9.setEnabled(False)
self.widget.setEnabled(True)
def on_change7(self):
if self.combo7.currentText() == "None":
self.textbox7.setEnabled(False)
self.textbox10.setEnabled(False)
self.widget1.setEnabled(False)
elif self.combo7.currentText() == "casting":
self.textbox7.setEnabled(True)
self.textbox10.setEnabled(True)
self.widget1.setEnabled(True)
else:
self.textbox7.setEnabled(True)
self.textbox10.setEnabled(False)
self.widget1.setEnabled(True)
def on_change8(self):
if self.combo8.currentText() == "None":
self.textbox8.setEnabled(False)
self.textbox11.setEnabled(False)
self.widget2.setEnabled(False)
elif self.combo8.currentText() == "casting":
self.textbox8.setEnabled(True)
self.textbox11.setEnabled(True)
self.widget2.setEnabled(True)
else:
self.textbox8.setEnabled(True)
self.textbox11.setEnabled(False)
self.widget2.setEnabled(True)
class mainwin(QMainWindow):
def __init__(self, parent=None):
super(mainwin, self).__init__(parent)
self.setWindowTitle("Nigandu English to Tamil Dictionary")
self.setGeometry(200, 50, 650, 600)
self.setMinimumHeight(620)
self.setMinimumWidth(650)
self.setMaximumHeight(660)
self.setMaximumWidth(800)
#Setting up status bar
self.myStatusBar = QStatusBar()
self.myStatusBar.showMessage('Ready', 7000)
self.setStatusBar(self.myStatusBar)
#Setting up application icon
appIcon = QIcon(":/icons/njnlogo.png")
self.setWindowIcon(appIcon)
# defining the central widget
self.central = QWidget(self)
#combobox plus search button
self.whole = QVBoxLayout(self.central)
self.gridlayout = QGridLayout()
self.comboBox = QLineEdit(self)
#self.comboBox.setEditable(True)
self.comboBox.setObjectName("comboBox")
self.completer = QCompleter(self.comboBox)
self.completer.setCompletionMode(QCompleter.UnfilteredPopupCompletion)
self.completer.setCaseSensitivity(Qt.CaseInsensitive)
self.completer.setMaxVisibleItems(10)
self.comboBox.setCompleter(self.completer)
#self.comboBox.setCompleter()
self.gridlayout.addWidget(self.comboBox, 1, 1, 1, 2)
self.searchbtn = QPushButton()
self.searchbtn.setObjectName("searchbtn")
self.searchbtn.setText("&Search")
self.gridlayout.addWidget(self.searchbtn, 1, 3)
vbox = QVBoxLayout()
self.tamtext = QTextBrowser()
self.listview = QListWidget(self)
#self.listview.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.listview.setWindowTitle("Suggested words")
self.tamtext.setMinimumHeight(100)
self.tamtext.setMaximumHeight(150)
vbox.addWidget(self.tamtext)
self.suglbl = QLabel(self)
self.suglbl.setText("Suggested Words:")
vbox.addWidget(self.suglbl)
vbox.addWidget(self.listview)
self.whole.addLayout(self.gridlayout)
self.whole.addLayout(vbox)
self.setCentralWidget(self.central)
#setting docks
self.histdockwidg = QDockWidget("History", self)
self.bkmdockwidg = QDockWidget("Book Marks", self)
self.histdockwidg.setObjectName("self.histdockwidg")
self.bkmdockwidg.setObjectName("self.bkmdockwidg")
#self.histdockwidg.setMaximumWidth(histwidth)
self.histdockwidg.setAllowedAreas(Qt.RightDockWidgetArea)
self.bkmdockwidg.setAllowedAreas(Qt.RightDockWidgetArea)
self.histdockwidg.setMaximumWidth(250)
self.bkmdockwidg.setMaximumWidth(250)
self.histdockwidg.setMinimumWidth(200)
self.bkmdockwidg.setMinimumWidth(200)
#self.bkmdockwidg.setMaximumWidth(histwidth)
self.histli = QListWidget()
self.bkmli = QListWidget()
self.histlis = [0]
self.bkmlistfromfile = []
self.histdockwidg.setWidget(self.histli)
self.bkmdockwidg.setWidget(self.bkmli)
self.addDockWidget(Qt.RightDockWidgetArea, self.histdockwidg)
self.addDockWidget(Qt.RightDockWidgetArea, self.bkmdockwidg)
#file menu
fi_addwrd = self.createactions("&Add a word...", self.addwrdf, "Alt+A", ":/icons/add.png",
"Add a word to the dictionary. . .")
fi_options = self.createactions("&Options", self.optionsf, "None", ":/icons/options.png",
"Change the default settings. . .")
fi_help = self.createactions("&Help", self.helpf, QKeySequence.HelpContents, ":/icons/help.png",
"Help contents. . .")
fi_quit = self.createactions("&Quit", self.close, QKeySequence.Close, ":/icons/quit.png",
"Close the application. . .")
fplus = self.createactions("FontPlus", self.fplusf, "None", ":/icons/fplus.png", "Increase the font size")
fminus = self.createactions("FontMinus", self.fminusf, "None", ":/icons/fminus.png", "Decrease the font size")
#list of file actions
fi_menu = (fi_addwrd, fi_options, fi_help, None, fi_quit)
#go menu
self.go_prev = self.createactions("&Previous Word", self.prevf, "Alt+Z", ":/icons/prev.png",
"Previous Word")
self.go_next = self.createactions("&Next Word", self.nextf, "Alt+X", ":/icons/next.png", "Next Word")
self.go_rand = self.createactions("&Random Word", self.randf, "Ctrl+R", ":/icons/rand.png",
"Select a random word")
#list of go actions
go_menu = (self.go_prev, self.go_next, self.go_rand )
self.go_next.setEnabled(False)
self.go_prev.setEnabled(False)
#book mark menu
self.bkm_addfav = self.createactions("&Bookmark", self.addfavf, "Ctrl+B", ":/icons/bookmark.png",
"Book mark this word")
self.bkm_viewbkm = self.createactions("&View Bookmarks", self.viewbkmf, "Alt+V", ":/icons/viewbkm.png",
"View bookmarked words")
#list of book mark items
bkm_menu = (self.bkm_addfav, self.bkm_viewbkm)
#help menu
hlp_about = self.createactions("Abo&ut", self.aboutf, "Ctrl+U", ":/icons/about.png", "About")
hlp_visitblog = self.createactions("&Visit Blog", self.visitblogf, "None", ":/icons/visitblog.png",
"Visit our blog")
hlp_help = self.createactions("&Help", self.helpf, "Ctrl+H", ":/icons/help.png", "Help Contents")
#list of help menu items
hlp_menu = (hlp_about, hlp_visitblog, hlp_help)
#Setting up the menubar
filemenu = self.menuBar().addMenu("&File")
self.addmenu(filemenu, fi_menu)
gomenu = self.menuBar().addMenu("&Go")
self.addmenu(gomenu, go_menu)
bkmmenu = self.menuBar().addMenu("&Book Mark")
self.addmenu(bkmmenu, bkm_menu)
helpmenu = self.menuBar().addMenu("&Help")
self.addmenu(helpmenu, hlp_menu)
intn = QSize(40, 40)
self.setIconSize(intn)
#Setting up the tool bar
filetools = self.addToolBar("File")
filetools.setObjectName("filetools")
self.addmenu(filetools, (fi_addwrd, fplus, fminus))
gotools = self.addToolBar("Go")
gotools.setObjectName("gotools")
self.addmenu(gotools, go_menu)
bkmtools = self.addToolBar("Bkm")
bkmtools.setObjectName("bkmtools")
self.addmenu(bkmtools, bkm_menu)
hlptools = self.addToolBar("Help")
hlptools.setObjectName("helptools")
self.addmenu(hlptools, hlp_menu)
self.loadfiles()
self.returncount = 0
self.bkm_addfav.setEnabled(False)
#clipboard function
if self.clipauto:
clip = QApplication.clipboard()
cliptxt = clip.text()
self.comboBox.setText(cliptxt)
self.setevent()
#connections
self.connect(self.comboBox, SIGNAL("textChanged(QString)"), self.search)
self.connect(self.comboBox, SIGNAL("returnPressed()"), self.returnpressedevent)
self.connect(self.searchbtn, SIGNAL("clicked()"), self.onenter)
self.connect(self.listview, SIGNAL("itemDoubleClicked(QListWidgetItem*)"), self.listwidcall)
self.connect(self.histli, SIGNAL("itemDoubleClicked(QListWidgetItem*)"), self.listwidcall)
self.connect(self.bkmli, SIGNAL("itemDoubleClicked(QListWidgetItem*)"), self.listwidcall)
def writehistlis(self, lis):
if len(lis) >= 2:
for i in range(1, len(lis)):
cur.execute("insert into HISTORY values(?)", (lis[i], ))
def writebkmlis(self, lis):
cur.execute("delete from BOOKMARKS")
if len(lis) > 0:
for i in range(len(lis)):
cur.execute("insert into BOOKMARKS values(?)", (lis[i], ))
def listwidcall(self, item):
self.comboBox.setText(item.text())
self.setevent()
def search(self, text, *args):
li = []
tplus = text + "%"
cur.execute("select ENGW from ENGTAM where ENGW like ? limit 20", (tplus, ))
cuf = cur.fetchall()
model = QStringListModel()
for i in range(len(cuf)):
k = cuf[i][0]
li.append(k)
model.setStringList(li)
self.completer.setModel(model)
def returnpressedevent(self, *args):
self.comboBox.selectAll()
self.returncount += 1
if self.returncount % 2 == 0:
self.setevent()
else:
self.comboBox.selectAll()
def setevent(self):
self.comboBox.selectAll()
self.bkm_addfav.setEnabled(True)
lis = []
eng = self.comboBox.text()
cur.execute("SELECT rowid, TAMW FROM ENGTAM WHERE ENGW like ? limit 1", (eng,))
cuf = cur.fetchall()
if len(cuf) == 0:
self.tamtext.setText("No words found. . . ")
self.listview.addItem("No Suggestions. . .")
else:
for i in range(len(cuf)):
tam = cuf[0][1]
rowid = cuf[0][0]
self.tamtext.setText(tam)
if rowid <= 25:
start = 0
end = 50
elif rowid >= 190513:
start = rowid - 190487
end = rowid + 190537
else:
start = rowid - 25
end = rowid + 25
cur.execute("SELECT ENGW FROM ENGTAM WHERE rowid>=? and rowid<=?", (start, end, ))
cuff = cur.fetchall()
for i in range(len(cuff)):
engw = cuff[i][0]
lis.append(engw)
if self.listview.count() is not None:
self.listview.clear()
self.listview.addItems(lis)
self.addtoli(eng, self.histlis)
if self.histlis[0] >= 2:
self.go_prev.setEnabled(True)
self.comboBox.setFocus()
if self.histdock:
self.histli.addItem(eng)
def addtoli(self, addw, lis, c=1):
if len(lis) > 0:
if type(lis[0]) == int:
if len(lis) >= 2:
for i in range(1, len(lis)):
if lis[i] == addw:
c = 0
pass
if c == 1:
lis.append(addw)
else:
lis.append(addw)
lis[0] = len(lis) - 1
def addtobkmli(self, addw, lis, nc=1):
for i in range(len(lis)):
if lis[i] == addw:
nc = 0
pass
if nc == 1:
lis.append(addw)
def onenter(self, *args):
self.comboBox.selectAll()
self.setevent()
def loadfiles(self):
self.loadsettings()
self.loadhistlis()
self.loadbkm()
self.setfontsize(int(self.fontsize))
self.setdocks()
def setdocks(self):
ist = str(self.histdock)
jst = str(self.bkmdock)
if ist == "False":
self.removedock(self.histdockwidg)
else:
self.adddock(self.histdockwidg)
if jst == "False":
self.removedock(self.bkmdockwidg)
else:
self.adddock(self.bkmdockwidg)
def loadsettings(self):
cur.execute("select * from SETTINGS")
cuffun = cur.fetchall()
fn = int(cuffun[0][1])
self.fontsize = fn
self.clipauto = cuffun[1][1]
self.histdock = cuffun[2][1]
self.savehist = cuffun[3][1]
self.bkmdock = cuffun[4][1]
self.delhist = cuffun[5][1]
self.delbkm = cuffun[6][1]
def loadhistlis(self):
histtodockli = []
cur.execute("select * from HISTORY")
historyfetch = cur.fetchall()
for i in range(len(historyfetch)):
self.addtobkmli(historyfetch[i][0], histtodockli)
for i in histtodockli:
self.histli.addItem(i)
def loadbkm(self):
cur.execute("select * from BOOKMARKS")
bkmfetch = cur.fetchall()
for i in range(len(bkmfetch)):
self.addtobkmli(bkmfetch[i][0], self.bkmlistfromfile)
for i in self.bkmlistfromfile:
self.bkmli.addItem(i)
def createactions(self, text, slot=None, shortcut="None", icon=None, tip=None, checkable=False,
signal="triggered()"):
action = QAction(text, self)
if icon is not None:
action.setIcon(QIcon(icon))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
self.connect(action, SIGNAL(signal), slot)
if checkable:
action.setCheckable(True)
return action
def addmenu(self, target, actions):
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
#Actions
def addwrdf(self):
self.dlg = addawrd()
self.dlg.show()
self.connect(self.dlg.buttonBox, SIGNAL("rejected()"), self.dlg.close)
self.connect(self.dlg.buttonBox, SIGNAL("accepted()"), self.addawordtodb)
def addawordtodb(self):
eng = self.dlg.lineEdit.text()
tam = self.dlg.lineEdit_2.text()
if len(eng) != 0 and len(tam) != 0:
cur.execute("INSERT INTO ENGTAM(ENGW, TAMW) VALUES(?, ?)", (eng, tam, ))
self.dlg.close()
QMessageBox.information(self, "Nigandu Eng -> Tam Dictionary", "Added Successfully. . .")
else:
self.dlg.lineEdit.setFocus()
self.dlg.close()
QMessageBox.warning(self, "Nigandu Eng -> Tam Dictionary", "Invalid Entry. . .")
def optionsf(self):
self.opt = optdlg(self)
self.opt.spinBox.setProperty("value", int(self.fontsize))
font = QFont()
font.setPixelSize(int(self.fontsize))
self.opt.sampletxt.setFont(font)
if str(self.clipauto) == "True":
self.opt.checkclip.setChecked(True)
elif str(self.clipauto) == "False":
self.opt.checkclip.setChecked(False)
if str(self.histdock) == "True":
self.opt.checkshowhistdock.setChecked(True)
elif str(self.histdock) == "False":
self.opt.checkshowhistdock.setChecked(False)
if str(self.bkmdock) == "True":
self.opt.checkshowbkmdock.setChecked(True)
elif str(self.bkmdock) == "False":
self.opt.checkshowbkmdock.setChecked(False)
self.opt.show()
self.connect(self.opt.buttonBox, SIGNAL("accepted()"), self.optok)
self.connect(self.opt.buttonBox.button(QDialogButtonBox.Apply), SIGNAL("clicked()"), self.optapply)
self.connect(self.opt.checkdelhist, SIGNAL("stateChanged(int)"), self.deleteallhist)
self.connect(self.opt.checkshowhistdock, SIGNAL("stateChanged(int)"), self.shownexttime)
self.connect(self.opt.checkshowbkmdock, SIGNAL("stateChanged(int)"), self.shownexttime)
def shownexttime(self, i):
if i == 0:
pass
if i == 2:
QMessageBox.information(self, self.windowTitle(), "Click Apply or Ok \n The Dock window will be added, \n the next time you start the application. . .")
def optok(self):
self.optapply()
self.opt.close()
def optapply(self):
self.updatesettings()
self.applyopt()
def updatesettings(self):
self.fontsize = self.opt.spinBox.value()
self.clipauto = self.opt.checkclip.isChecked()
self.histdock = self.opt.checkshowhistdock.isChecked()
self.bkmdock = self.opt.checkshowbkmdock.isChecked()
self.delhist = self.opt.checkdelhist.isChecked()
for i, j in [("fontsize", self.fontsize),("clipauto", str(self.clipauto)),("histdock", str(self.histdock)),
("bkmdock", str(self.bkmdock)),("delhist", str(self.delhist))]:
cur.execute("UPDATE SETTINGS SET setting=? WHERE field=?", (j, i, ))
def applyopt(self):
self.loadsettings()
self.setfontsize(int(self.fontsize))
if str(self.bkmdock) == "False" or str(self.histdock) == "False":
self.setdocks()
def removedock(self, dock):
self.removeDockWidget(dock)
def adddock(self, dock):
self.addDockWidget(Qt.RightDockWidgetArea, dock)
def deleteallhist(self, i):
if i == 0:
pass
elif i == 2:
self.histli.clear()
self.histlis = [0]
cur.execute("delete from HISTORY")
QMessageBox.information(self, self.windowTitle(), "All the History Records are deleted. . .")
def setfontsize(self, i):
if i >= 8 or i <= 24:
font = QFont()
font.setPixelSize(i)
self.comboBox.setFont(font)
self.searchbtn.setFont(font)
self.bkmli.setFont(font)
self.histli.setFont(font)
self.listview.setFont(font)
self.tamtext.setFont(font)
def helpf(self):
form = helpform.HelpForm("index.html", self)
form.show()
def closeEvent(self, *args, **kwargs):
self.writehistlis(self.histlis)
self.writebkmlis(self.bkmlistfromfile)
for i, j in [("fontsize", int(self.fontsize)),("clipauto", str(self.clipauto)),("histdock", str(self.histdock)),
("bkmdock", str(self.bkmdock)),("delhist", str(self.delhist))]:
cur.execute("UPDATE SETTINGS SET setting=? WHERE field=?", (j, i, ))
con.commit()
con.close()
def fplusf(self):
self.fontsize += 1
if self.fontsize <= 24:
self.setfontsize(self.fontsize)
def fminusf(self):
self.fontsize -= 1
if self.fontsize >= 10:
self.setfontsize(self.fontsize)
def prevf(self):
pr = self.histlis[0] - 1
if pr > 1:
self.comboBox.setText(self.histlis[pr])
self.setevent()
self.histlis[0] = pr
self.go_next.setEnabled(True)
elif pr == 1:
self.comboBox.setText(self.histlis[pr])
self.setevent()
self.histlis[0] = pr
self.go_next.setEnabled(True)
self.go_prev.setEnabled(False)
else:
pass
def nextf(self):
pr = self.histlis[0] + 1
if pr < len(self.histlis) - 1:
self.comboBox.setText(self.histlis[pr])
self.setevent()
self.histlis[0] = pr
self.go_prev.setEnabled(True)
elif pr == len(self.histlis) - 1:
self.comboBox.setText(self.histlis[pr])
self.setevent()
self.histlis[0] = pr
self.go_prev.setEnabled(True)
self.go_next.setEnabled(False)
else:
pass
def randf(self):
import random
n = random.randrange(190538)
cur.execute("select ENGW from ENGTAM where rowid = ?", (n, ))
cuf = cur.fetchone()
self.comboBox.setText(cuf[0])
self.setevent()
def addfavf(self):
txt = self.comboBox.text()
if len(txt) != 0:
self.addtobkmli(txt, self.bkmlistfromfile)
self.writetolistwidget(self.bkmlistfromfile, self.bkmli)
def sortit(self):
self.bkmlistfromfile.sort()
self.writetolistwidget(self.bkmlistfromfile, self.form.listWidget)
self.writetolistwidget(self.bkmlistfromfile, self.bkmli)
cur.execute("delete from BOOKMARKS")
def writetolistwidget(self, lis, liswid):
liswid.clear()
for i in lis:
liswid.addItem(i)
def deletecurrentbkm(self):
ct = self.form.listWidget.currentItem().text()
self.bkmlistfromfile.remove(ct)
self.writetolistwidget(self.bkmlistfromfile, self.bkmli)
self.writetolistwidget(self.bkmlistfromfile, self.form.listWidget)
cur.execute("delete from BOOKMARKS")
def deleteallbkm(self):
self.form.listWidget.clear()
self.bkmli.clear()
self.bkmlistfromfile = []
cur.execute("delete from BOOKMARKS")
def viewbkmf(self):
self.form = managebkm(self)
self.writetolistwidget(self.bkmlistfromfile, self.form.listWidget)
self.form.show()
self.connect(self.form.closebtn, SIGNAL("clicked()"), self.form.close)
self.connect(self.form.sortbtn, SIGNAL("clicked()"), self.sortit)
self.connect(self.form.deletebtn, SIGNAL("clicked()"), self.deletecurrentbkm)
self.connect(self.form.deleteallbtn, SIGNAL("clicked()"), self.deleteallbkm)
def aboutf(self):
QMessageBox.about(self, "About Nigandu English to Tamil Dictionary",
"""<b>Nigandu English to Tamil Dictionary</b> v %s
<p>This is the first corss-platform English to Tamil
bilingual dictionary; Free to use.</p>
<p>Copyright © 2014 NJN Private Ltd.
All rights reserved.</p>
<p>Thanks to Python and PySide Project.</p>
<p>Using Python 3.3, Qt 4.8 and PySide 1.2.1</p>""" % (__version__))
def visitblogf(self):
webbrowser.open("http://www.e-nool.blogspot.com")
class MultiListDialog(QDialog):
def __init__(self, items): # , parent=None
super(MultiListDialog, self).__init__() # parent
self.resize(800, 550)
self.entries = items
layout = QVBoxLayout(self)
self.listWidget = QListWidget(self)
self.listWidget.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.HlineInputfiles = ConfigLineDir(u"Verzeichnis NI XML:", "dir")
self.HlineOutputfiles = ConfigLineDir(u"Verzeichnis Report:", "dir")
if os.path.isfile(PICKLE_TEMP_FILE):
pkl_file = open(PICKLE_TEMP_FILE, 'rb')
inPath = pickle.load(pkl_file)
outPath = pickle.load(pkl_file)
pkl_file.close()
if os.path.isdir(inPath):
self.HlineInputfiles.editText.setText(inPath)
if os.path.isdir(outPath):
self.HlineOutputfiles.editText.setText(outPath)
layout.addLayout(self.HlineInputfiles)
layout.addLayout(self.HlineOutputfiles)
layout.addWidget(self.listWidget)
layout.addWidget(QLabel("Mehrfachauswahl -> Strg + Mausklick"))
# OK and Cancel buttons
buttons = QDialogButtonBox(QDialogButtonBox.Apply | QDialogButtonBox.Close, Qt.Horizontal, self)
# Apply before Close
buttons.setStyleSheet("* { button-layout: 2 }")
layout.addWidget(buttons)
buttons.rejected.connect(self.close)
buttons.button(QDialogButtonBox.Apply).clicked.connect(self.Apply_Clicked)
self.SetListItems()
self.HlineInputfiles.editText.textChanged.connect(self.RefreshList)
self.HlineInputfiles.editText.textChanged.connect(self.SetListItems)
# def SetOutputDir(self):
# self.HlineOutputfiles.editText.setText(self.HlineInputfiles.getText())
def SetListItems(self, parent=None):
itemslist = os.listdir(self.HlineInputfiles.getText())
for it in itemslist:
if it.endswith(".xml"):
self.listWidget.addItem(it)
def GetMarkedItems(self):
results = []
try:
for el in self.listWidget.selectedItems():
results.append(str(el.text()))
return (results)
except:
return (None)
def RefreshList(self):
self.listWidget.clear()
print "refresh"
def Apply_Clicked(self):
SelectedItems = self.GetMarkedItems()
textMsg = "Selektierte Dateie(n):\r\n"
if len(SelectedItems) > 0:
for file in SelectedItems:
filePath = os.path.join(self.HlineInputfiles.getText(), file)
textMsg += (file + "\r\n")
XmlAnalysis.XmlAnalysis(filePath, self.HlineOutputfiles.getText())
#QMessageBox.information(self, "Report", textMsg)
textMsg += "\r\nReport-Verzeichnis öffnen?\r\n"
msgBox = QMessageBox.StandardButton.Yes
msgBox |= QMessageBox.StandardButton.No
if QMessageBox.question(self, "Report", textMsg, msgBox) == QMessageBox.Yes:
os.startfile(self.HlineOutputfiles.getText())
else:
QMessageBox.warning(self, "Achtung", "Keine Datei selektiert!")
pass
class Panel(QWidget):
def __init__(self, state, parent=None):
super().__init__(parent=parent)
self.state = state
self.createWidgets()
self.layoutWidgets()
self.createConnections()
self.tooltips.append((self, """<p><b>Groups panel</b></p>
<p>This panel shows the groups the current entry belongs to.</p>"""))
def createWidgets(self):
self.label = QLabel("Gro&ups")
self.groupsList = QListWidget()
self.tooltips.append((self.groupsList, """
<p><b>Groups</b> (Alt+U)</p>
<p>A (possibly empty) list of the current entry's groups.</p>"""))
self.label.setBuddy(self.groupsList)
self.closeButton = QToolButton()
self.closeButton.setIcon(QIcon(":/hide.svg"))
self.closeButton.setFocusPolicy(Qt.NoFocus)
self.tooltips.append((self.closeButton, """<p><b>Hide</b></p>
<p>Hide the Groups panel.</p>
<p>Use <b>Spelling→Show Suggestions and Groups</b> to show it
again.</p>"""))
self.helpButton = QToolButton()
self.helpButton.setIcon(QIcon(":/help.svg"))
self.helpButton.setFocusPolicy(Qt.NoFocus)
self.tooltips.append((self.helpButton, "Help on the Groups panel."))
def layoutWidgets(self):
layout = QVBoxLayout()
hbox = QHBoxLayout()
hbox.addWidget(self.label)
hbox.addStretch()
hbox.addWidget(self.closeButton)
hbox.addWidget(self.helpButton)
layout.addLayout(hbox)
layout.addWidget(self.groupsList)
self.setLayout(layout)
def createConnections(self):
self.closeButton.clicked.connect(self.state.window.closeGroups)
self.helpButton.clicked.connect(self.help)
self.groupsList.itemDoubleClicked.connect(
self.state.viewFilteredPanel.setGroup)
def updateUi(self):
enable = bool(self.state.model) and self.state.mode not in {
ModeKind.NO_INDEX, ModeKind.CHANGE
}
self.setEnabled(enable)
if enable:
(self.state.window.entryActions.addToNormalGroupAction.setEnabled(
self.state.model.normalGroupCount()))
(self.state.window.entryActions.addToLinkedGroupAction.setEnabled(
self.state.model.linkedGroupCount()))
self.state.window.entryActions.removeFromGroupAction.setEnabled(
self.groupsList.currentItem() is not None)
def updateGroups(self):
self.groupsList.clear()
eid = self.state.viewAllPanel.view.selectedEid
if eid is not None:
for gid, name, linked in self.state.model.groupsForEid(
eid, withLinks=True):
item = QListWidgetItem(name)
item.setData(Qt.UserRole, gid)
item.setIcon(
QIcon(":/grouplink.svg" if linked else ":/groups.svg"))
self.groupsList.addItem(item)
if self.groupsList.count():
self.groupsList.setCurrentRow(0)
self.updateUi()
self.state.viewFilteredPanel.groupChanged()
def clear(self):
self.groupsList.clear()
self.updateUi()
def help(self):
self.state.help("xix_ref_panel_grp.html")
def __getattr__(self, name):
return getattr(self.groupsList, name)
class MainControl(QWidget):
toChangeChunk = Signal(tuple)
toChangeSlice = Signal(int)
def __init__(self, parent):
super(MainControl, self).__init__(parent)
self.parent = parent
self.selectedMinerals = parent.selectedMinerals
self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
lb0 = QLabel('Minecraft Prism chunk minerals finder')
lb1 = QLabel('Chunk X')
lb2 = QLabel('Chunk Z')
lb3 = QLabel('Show deposits of:')
lb4 = QLabel('Slice Elevation')
#lb0.setFrameStyle(QFrame.Box)
lb0.setAlignment(Qt.AlignVCenter | Qt.AlignCenter)
lb2.setAlignment(Qt.AlignVCenter | Qt.AlignRight)
self.txtChunkX = QLineEdit()
self.txtChunkZ = QLineEdit()
self.txtElevation = QLineEdit(str(self.parent.presenter.elevation))
self.txtChunkX.setAlignment(Qt.AlignVCenter | Qt.AlignCenter)
self.txtChunkZ.setAlignment(Qt.AlignVCenter | Qt.AlignCenter)
self.txtElevation.setAlignment(Qt.AlignVCenter | Qt.AlignCenter)
self.btnViewChunk = QPushButton('View Chunk')
self.btnViewChunk.clicked.connect(self.btnViewChunk_OnClick)
self.btnSliceUp = QPushButton('Slice Up')
self.btnSliceUp.clicked.connect(self.btnSliceUp_OnClick)
self.btnSliceDown = QPushButton('Slice Down')
self.btnSliceDown.clicked.connect(self.btnSliceDown_OnClick)
self.lstMinerals = QListWidget()
self.lstMinerals.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.lstMinerals.addItem(SelectedMineral('Gold', 14))
self.lstMinerals.addItem(SelectedMineral('Iron', 15))
self.lstMinerals.addItem(SelectedMineral('Diamond', 56))
self.lstMinerals.addItem(SelectedMineral('Redstone', 73))
self.lstMinerals.addItem(SelectedMineral('Obsidian', 49))
self.lstMinerals.addItem(SelectedMineral('Coal', 16))
self.lstMinerals.addItem(SelectedMineral('Lazurit', 21))
self.lstMinerals.setCurrentItem(self.lstMinerals.item(0))
self.lstMinerals.itemSelectionChanged.connect(self.onSelectedMinerals)
elevBox = QHBoxLayout()
elevBox.addWidget(lb4)
elevBox.addWidget(self.txtElevation)
elevBox.addWidget(self.btnSliceUp)
elevBox.addWidget(self.btnSliceDown)
ly = QGridLayout()
ly.addWidget(lb0, 0, 0, 1, 4)
ly.addWidget(lb1, 1, 0, 1, 1)
ly.addWidget(self.txtChunkX, 1, 1, 1, 1)
ly.addWidget(lb2, 1, 2, 1, 1)
ly.addWidget(self.txtChunkZ, 1, 3, 1, 1)
ly.addWidget(lb3, 2, 0, 1, 2)
ly.addWidget(self.lstMinerals, 2, 2, 1, 2)
ly.addWidget(self.btnViewChunk, 3, 1, 1, 2)
ly.addLayout(elevBox, 4, 0, 1, 4)
ly.setColumnStretch(0, 20)
ly.setColumnStretch(1, 15)
ly.setColumnStretch(2, 46)
ly.setColumnStretch(3, 19)
self.setLayout(ly)
def sizeHint(self):
return QSize(self.parent.controlsWidth, 180)
def btnViewChunk_OnClick(self):
chX = int(self.txtChunkX.text())
chZ = int(self.txtChunkZ.text())
self.toChangeChunk.emit((chX, chZ))
def btnSliceUp_OnClick(self):
elev = int(self.txtElevation.text()) + 1
if elev < 64:
self.toChangeSlice.emit(elev)
self.txtElevation.setText(str(elev))
def btnSliceDown_OnClick(self):
elev = int(self.txtElevation.text()) - 1
if elev > -1:
self.toChangeSlice.emit(elev)
self.txtElevation.setText(str(elev))
def onSelectedMinerals(self):
selMin = []
for item in self.lstMinerals.selectedItems():
selMin.append(item.blockCode)
self.parent.selectedMinerals.setSelectedMinerals(selMin)
class Form(QDialog):
def __init__(self, term, state, parent=None):
super().__init__(parent)
Lib.prepareModalDialog(self)
self.addingText = term
self.state = state
ListWidgetItem.model = state.model
ListWidgetItem.refresh = self.refresh
self.buttons = []
self.createWidgets()
self.layoutWidgets()
self.createConnections()
self.setWindowTitle("Edit Groups — {}".format(
QApplication.applicationName()))
self.refresh()
settings = QSettings()
self.updateToolTips(
bool(
int(
settings.value(Gopt.Key.ShowDialogToolTips,
Gopt.Default.ShowDialogToolTips))))
def createWidgets(self):
self.listWidget = QListWidget()
self.buttonLayout = QVBoxLayout()
self.linkButton = None
for icon, text, slot, tip in ((":/add.svg", "&Add", self.add, """\
<p><b>Add</b> (Alt+A)</p><p>Add a new Normal Group.</p>"""), (":/edit.svg",
"&Rename",
self.rename, """\
<p><b>Rename</b> (Alt+R)</p><p>Rename the current Group.</p>"""),
(":/grouplink.svg", "&Link", self.link,
"""\
<p><b>Link</b> (Alt+L)</p><p>Change the current group into a Linked
Group.</p>
<p>This means that the pages of every entry in this group will be merged
and synchronized, and any future changes to the pages of any entries in
this group will be propagated to all the other entries in this
group to keep them all synchronized.</p>"""), (":/groups.svg", "&Unlink",
self.unlink, """\
<p><b>Unlink</b> (Alt+U)</p><p>Change the current group into a Normal
(unlinked) Group. If the linked group
has entries, the <b>Delete Linked Group</b> dialog will pop up.</p>"""),
(":/groupset.svg", "&View",
self.viewGroup, """\
<p><b>View</b> (Alt+V)</p><p>View the current group in the Filtered
View.</p>"""), (":/delete.svg", "&Delete", self.delete, """\
<p><b>Delete</b> (Alt+D)</p><p>Remove entries from the current normal group
and then delete the group. If the current group is a linked group that
has entries, the <b>Delete Linked Group</b> dialog will pop up.</p>"""),
(":/help.svg", "Help", self.help, """\
Help on the Groups dialog"""), (":/dialog-close.svg", "&Close", self.accept,
"""\
<p><b>Close</b></p><p>Close the dialog.</p>""")):
button = QPushButton(QIcon(icon), text)
button.setFocusPolicy(Qt.NoFocus)
if text in {"&Close", "Help"}:
self.buttonLayout.addStretch()
else:
self.buttons.append(button)
self.buttonLayout.addWidget(button)
button.clicked.connect(slot)
self.tooltips.append((button, tip))
if text == "&Link":
self.linkButton = button
button.setEnabled(False)
elif text == "&Unlink":
self.unlinkButton = button
button.setEnabled(False)
self.tooltips.append((self.listWidget, "List of Groups"))
def layoutWidgets(self):
layout = QHBoxLayout()
layout.addWidget(self.listWidget)
layout.addLayout(self.buttonLayout)
self.setLayout(layout)
def createConnections(self):
self.listWidget.currentRowChanged.connect(self.updateUi)
def updateUi(self):
for button in self.buttons:
button.setEnabled(True)
item = self.listWidget.currentItem()
if item is not None:
gid = item.data(Qt.UserRole)
self.linkButton.setEnabled(self.state.model.safeToLinkGroup(gid))
self.unlinkButton.setEnabled(self.state.model.isLinkedGroup(gid))
def refresh(self, *, text=None, index=None):
self.listWidget.clear()
for row, (gid, name,
linked) in enumerate(self.state.model.allGroups()):
item = ListWidgetItem(name)
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEditable
| Qt.ItemIsEnabled)
item.setBackground(self.palette().base() if row %
2 else self.palette().alternateBase())
item.setData(Qt.UserRole, gid)
item.setIcon(
QIcon(":/grouplink.svg" if linked else ":/groups.svg"))
self.listWidget.addItem(item)
if self.listWidget.count():
self.listWidget.setCurrentRow(0)
if index is not None:
self.listWidget.setCurrentRow(index)
elif text is not None:
for i in range(self.listWidget.count()):
if self.listWidget.item(i).text() == text:
self.listWidget.setCurrentRow(i)
break
self.updateUi()
def add(self):
item = ListWidgetItem(self.addingText, adding=True)
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEditable
| Qt.ItemIsEnabled)
self.listWidget.insertItem(0, item)
self.listWidget.setCurrentRow(0)
for button in self.buttons:
button.setEnabled(False)
self.listWidget.editItem(item)
def rename(self):
item = self.listWidget.currentItem()
if item is not None:
for button in self.buttons:
button.setEnabled(False)
self.listWidget.editItem(item)
def link(self):
item = self.listWidget.currentItem()
if item is not None:
gid = item.data(Qt.UserRole)
if not self.state.model.safeToLinkGroup(gid):
return # Should never happen
self.state.model.linkGroup(gid)
self.refresh(index=self.listWidget.currentRow())
def unlink(self):
item = self.listWidget.currentItem()
if item is not None:
gid = item.data(Qt.UserRole)
if not self.state.model.groupMemberCount(gid):
self.state.model.unlinkGroup(gid)
else:
with Lib.Qt.DisableUI(self, forModalDialog=True):
form = Forms.DeleteOrUnlink.Form("Unlink", gid, self.state,
self)
form.exec_()
self.refresh(index=self.listWidget.currentRow())
def viewGroup(self):
item = self.listWidget.currentItem()
if item:
self.state.viewFilteredPanel.setGroup(item)
def delete(self): # No need to restore focus widget
row = self.listWidget.currentRow()
item = self.listWidget.item(row)
if item is None:
return
gid = item.data(Qt.UserRole)
for button in self.buttons:
button.setEnabled(False)
deleteItem = False
if (not self.state.model.isLinkedGroup(gid)
or not self.state.model.groupMemberCount(gid)):
with Lib.Qt.DisableUI(self, forModalDialog=True):
reply = QMessageBox.question(
self,
"Delete Group — {}".format(QApplication.applicationName()),
"<p>Delete Group “{}”?</p>".format(item.text()),
QMessageBox.Yes | QMessageBox.No)
if reply == QMessageBox.Yes:
self.state.model.deleteGroup(gid)
deleteItem = True
else:
with Lib.Qt.DisableUI(self, forModalDialog=True):
form = Forms.DeleteOrUnlink.Form("Delete", gid, self.state,
self)
deleteItem = form.exec_()
if deleteItem:
item = self.listWidget.takeItem(row)
del item
self.updateUi()
def help(self):
self.state.help("xix_ref_dlg_groups.html")
class SubscriberDialog(QDialog):
if USE_MAEMO_5:
switchRequested = Signal()
def __init__(self, parent=None):
QDialog.__init__(self, parent)
self.ui = Ui_SubscriberDialog()
self.ui.setupUi(self)
self.subscriber = None
self.tableWidget = None
self.listWidget = None
if USE_MAEMO_5:
switchButton = self.ui.buttonBox.addButton(
self.tr('Switch'), QDialogButtonBox.ActionRole)
switchButton.clicked.connect(self.switchRequested)
self.tableWidget = self.ui.tableWidget
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
verticalHeader = self.tableWidget.verticalHeader()
verticalHeader.setVisible(False)
self.tableWidget.setColumnWidth(0, 200)
self.tableWidget.setColumnWidth(1, 400)
else:
desktopWidget = QDesktopWidget()
if desktopWidget.availableGeometry().width() < 400:
# Screen is too small to fit a table widget without scrolling, use a list widget instead.
self.listWidget = QListWidget()
self.listWidget.setAlternatingRowColors(True)
self.ui.verticalLayout.insertWidget(2, self.listWidget)
else:
self.tableWidget = QTableWidget()
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
self.tableWidget.verticalHeader()
self.setVisible(False)
self.ui.verticalLayout.insertWidget(2, self.tableWidget)
self.ui.connectButton.clicked.connect(self.changeSubscriberPath)
self.changeSubscriberPath()
# if the default path does not exist reset it to /
value = self.subscriber.value()
subPaths = self.subscriber.subPaths()
if not value and not subPaths:
self.ui.basePath.setText('/')
self.changeSubscriberPath()
def changeEvent(self, e):
QDialog.changeEvent(self, e)
if e.type() == QEvent.LanguageChange:
self.ui.retranslateUi(self)
def changeSubscriberPath(self):
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
if not self.subscriber:
self.subscriber = QValueSpaceSubscriber(self.ui.basePath.text(),
self)
else:
self.subscriber.setPath(self.ui.basePath.text())
self.subscriber.contentsChanged.connect(self.subscriberChanged)
self.subscriber.connectNotify("contentsChanged()")
self.subscriberChanged()
def subscriberChanged(self):
subPaths = self.subscriber.subPaths()
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
self.tableWidget.setRowCount(len(subPaths))
for i in xrange(len(subPaths)):
v = self.subscriber.value(subPaths[i])
if self.listWidget:
item = QListWidgetItem('%s (%s)\n%s' %
(subPaths[i], str(type(v)), str(v)))
item.setFlags(item.flags() & ~Qt.ItemIsEditable)
self.listWidget.addItem(item)
elif self.tableWidget:
pathItem = QTableWidgetItem(subPaths[i])
pathItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
valueItem = QTableWidgetItem(str(v))
valueItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
typeItem = QTableWidgetItem(str(type(v)))
typeItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
self.tableWidget.setItem(i, 0, pathItem)
self.tableWidget.setItem(i, 1, valueItem)
self.tableWidget.setItem(i, 2, typeItem)
class ListEditor ( BaseEditor ):
""" Enumeration editor, used for the "custom" style, that displays a list
box.
"""
#-- Public Methods ---------------------------------------------------------
def init ( self, parent ):
""" Finishes initializing the editor by creating the underlying toolkit
widget.
"""
super( ListEditor, self ).init( parent )
self.control = QListWidget( parent )
QObject.connect( self.control, SIGNAL( 'currentTextChanged(QString)' ),
self.update_object )
self.rebuild_editor()
self.set_tooltip()
def update_object ( self, text ):
""" Handles the user selecting a list box item.
"""
value = unicode( text )
try:
value = self.mapping[ value ]
except:
try:
value = self.factory.evaluate( value )
except:
pass
try:
self.value = value
except:
pass
def update_editor ( self ):
""" Updates the editor when the object facet changes externally to the
editor.
"""
control = self.control
try:
value = self.inverse_mapping[ self.value ]
for row in range( control.count() ):
itm = control.item( row )
if itm.text() == value:
control.setCurrentItem( itm )
control.scrollToItem( itm )
break
except:
pass
def rebuild_editor ( self ):
""" Rebuilds the contents of the editor whenever the original factory
object's **values** facet changes.
"""
self.control.clear()
for name in self.names:
self.control.addItem( name )
#-- EOF ------------------------------------------------------------------------
class ToolBox(QVBoxLayout):
sig = QtCore.Signal(object)
listThread = None
groupBoxThreadInfo = None
threadvbox = None
mode = None
def __init__(self, mode, parentQWidget=None):
QVBoxLayout.__init__(self)
self.sig.connect(self.addThreadList)
self.mode = mode
self.sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed,
QtGui.QSizePolicy.Expanding)
self.groupBoxSearch = QGroupBox()
self.groupBoxSearch.setStyleSheet(
"QGroupBox {border: 1px solid gray; border-radius: 4px; };")
vboxSearch = QVBoxLayout()
self.searchTitle = QLabel("Search Messages")
vboxSearch.addWidget(self.searchTitle)
self.searchHLayout = QHBoxLayout()
self.editTextSearch = QTextEdit('')
self.editTextSearch.setFixedSize(200, 30)
self.buttonSearch = QPushButton('Search')
self.buttonSearch.setFixedSize(100, 30)
self.buttonSearch.clicked.connect(self.searchMsg)
vboxSearch.addWidget(self.editTextSearch)
self.searchHLayout.addWidget(self.buttonSearch)
self.searchCursor = QLabel()
self.searchHLayout.addWidget(self.searchCursor)
vboxSearch.addLayout(self.searchHLayout)
self.browseHLayout = QHBoxLayout()
self.buttonLookUp = QPushButton('\u21e7') #Arrow up
self.buttonLookUp.setFixedWidth(100)
self.buttonLookUp.clicked.connect(self.moveToPrev)
self.buttonLookDown = QPushButton('\u21e9') #Arrow down
self.buttonLookDown.setFixedWidth(100)
self.buttonLookDown.clicked.connect(self.moveToNext)
self.browseHLayout.addWidget(self.buttonLookUp)
self.browseHLayout.addWidget(self.buttonLookDown)
vboxSearch.addLayout(self.browseHLayout)
self.groupBoxSearch.setLayout(vboxSearch)
self.addWidget(self.groupBoxSearch)
self.groupBoxSearch.setSizePolicy(
QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed,
QtGui.QSizePolicy.Fixed))
self.buttonHiddenLifelines = QPushButton('Show hidden life-lines')
self.buttonHiddenLifelines.setFixedWidth(200)
self.buttonHiddenLifelines.clicked.connect(self.showHiddenLifelines)
self.addWidget(self.buttonHiddenLifelines)
self.buttonHiddenMessages = QPushButton('Show hidden Messages')
self.buttonHiddenMessages.setFixedWidth(200)
self.buttonHiddenMessages.clicked.connect(self.showHiddenMessages)
self.addWidget(self.buttonHiddenMessages)
if const.mode_interactive == mode:
self.buttonCapture = QPushButton('Capture')
self.buttonCapture.setFixedWidth(200)
self.buttonCapture.clicked.connect(self.notifyCapture)
self.addWidget(self.buttonCapture)
self.msgRcv = []
self.msgInfo = QLabel("Message Info.")
self.groupBoxMessageInfo = QGroupBox()
self.groupBoxMessageInfo.setStyleSheet(
"QGroupBox {border: 1px solid gray; border-radius: 9px; margin-top: 0.5em} QGroupBox::title {subcontrol-origin: margin; left: 10px; padding: 0 3px 0 3px;"
)
vbox = QVBoxLayout()
vbox.addWidget(self.msgInfo)
self.tableTime = QtGui.QTableWidget(3, 2)
self.tableTime.setHorizontalHeaderLabels(['-', 'time'])
self.tableTime.setColumnWidth(0, 80)
self.tableTime.setColumnWidth(1, 150)
vwidth = self.tableTime.verticalHeader().length()
hwidth = self.tableTime.horizontalHeader().height()
fwidth = self.tableTime.frameWidth() * 2
self.tableTime.setFixedHeight(vwidth + hwidth + fwidth)
self.tableTime.horizontalHeader().setStretchLastSection(True)
self.tableTime.setItem(0, 0, QTableWidgetItem('begin'))
self.tableTime.setItem(0, 1, QTableWidgetItem(' - '))
self.tableTime.setItem(1, 0, QTableWidgetItem('end'))
self.tableTime.setItem(1, 1, QTableWidgetItem(' - '))
self.tableTime.setItem(2, 0, QTableWidgetItem('duration'))
self.tableTime.setItem(2, 1, QTableWidgetItem(' - '))
vbox.addWidget(self.tableTime)
self.titleArg = QLabel('Argument List')
vbox.addWidget(self.titleArg)
max_arg_num = 10
self.tableArgs = QtGui.QTableWidget(max_arg_num, 2)
self.tableArgs.setHorizontalHeaderLabels(['type', 'value'])
for idx in range(0, max_arg_num):
self.tableArgs.setItem(idx, 0, QTableWidgetItem())
self.tableArgs.setItem(idx, 1, QTableWidgetItem())
self.tableArgs.horizontalHeader().setStretchLastSection(True)
vbox.addWidget(self.tableArgs)
self.titleArg = QLabel('Return Value List')
vbox.addWidget(self.titleArg)
max_ret_num = 4
self.tableRet = QtGui.QTableWidget(max_ret_num, 2)
self.tableRet.setHorizontalHeaderLabels(['type', 'value'])
for idx in range(0, max_ret_num):
self.tableRet.setItem(idx, 0, QTableWidgetItem())
self.tableRet.setItem(idx, 1, QTableWidgetItem())
self.tableRet.horizontalHeader().setStretchLastSection(True)
vwidth = self.tableRet.verticalHeader().length()
hwidth = self.tableRet.horizontalHeader().height()
fwidth = self.tableRet.frameWidth() * 2
self.tableRet.setFixedHeight(vwidth + hwidth + fwidth)
vbox.addWidget(self.tableRet)
self.buttonSrcView = QPushButton('view code')
self.buttonSrcView.setFixedWidth(200)
self.buttonSrcView.clicked.connect(self.openSourceViewer)
self.buttonHide = QPushButton('Hide')
self.buttonHide.setFixedWidth(200)
self.buttonHide.clicked.connect(self.notifyHide)
self.buttonHideAllMsg = QPushButton('Hide All')
self.buttonHideAllMsg.setFixedWidth(200)
self.buttonHideAllMsg.clicked.connect(self.hideAllMsgNamedAsSelected)
self.groupBoxMessageInfo.setLayout(vbox)
self.checkHideCircular = QCheckBox('Hide Circular Messages')
self.checkHideCircular.setCheckState(QtCore.Qt.Unchecked)
self.checkHideCircular.stateChanged.connect(
self.changeHideCircularMessage)
self.addWidget(self.checkHideCircular)
self.addWidget(self.groupBoxMessageInfo)
self.groupBoxMessageInfo.setSizePolicy(self.sizePolicy)
def reset(self):
for idx in reversed(range(0, self.listThread.count())):
self.listThread.takeItem(idx)
def setMsgInfoMessage(self, msg):
self.strMessage = msg
def changeHideCircularMessage(self, state):
if state == QtCore.Qt.Unchecked:
self.diagramView.hideCircularChanged(False)
elif state == QtCore.Qt.Checked:
self.diagramView.hideCircularChanged(True)
def setMsgInfoModule(self, module):
self.strModule = module
def updateSearchStatus(self, curr, number):
self.searchCursor.setText("%d/%d" % (curr, number))
def connectSourceViewer(self, viewer):
self.srcViewer = viewer
def openSourceViewer(self):
self.srcViewer.openViewer(self.strModule, self.strMessage)
def setMessageInfoTime(self, begin, end, duration):
self.tableTime.item(0, 1).setText(begin)
self.tableTime.item(1, 1).setText(end)
self.tableTime.item(2, 1).setText(duration + ' msec')
def setMessageInfoArg(self, listParam, listArg):
if listArg:
for idx, text in enumerate(listArg):
self.tableArgs.item(idx, 1).setText(text)
for idx, text in enumerate(listParam):
self.tableArgs.item(idx, 0).setText(text)
else:
for idx in range(0, self.tableArgs.rowCount()):
self.tableArgs.item(idx, 1).setText('')
self.tableArgs.item(idx, 0).setText('')
def setMessageInfoRet(self, listRet):
if listRet:
for idx, text in enumerate(listRet):
self.tableRet.item(idx, 1).setText(text)
else:
for idx in range(0, self.tableRet.rowCount()):
self.tableRet.item(idx, 1).setText('')
self.tableRet.item(idx, 0).setText('')
def notifyInteractiveStateChanged(self, state):
if const.mode_interactive != self.mode:
return
if const.STATE_INTERACTIVE_CAPTURING == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Stop Capture')
if const.STATE_INTERACTIVE_PROCESSING == state:
self.buttonCapture.setEnabled(False)
if const.STATE_INTERACTIVE_IDLE == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Capture')
if const.STATE_INTERACTIVE_RESET == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Capture')
elif const.STATE_INTERACTIVE_ACTIVE == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Capture')
def setMessageInfo(self, info):
self.msgInfo.setText(info)
def setAvailable(self, threads):
self.sig.emit(threads)
def toggleThreadDisplay(self, item):
print(self.listThread.currentRow())
#if item.isSelected():
# print(item.text() + " is selected")
#else:
# print(item.text() + " is not selected")
self.diagramView.showThread(self.listThread.currentRow(),
item.isSelected())
def hideAllMsgNamedAsSelected(self):
self.diagramView.hideAllMessageSelected()
def addThreadList(self, threads):
if not self.groupBoxThreadInfo:
self.groupBoxThreadInfo = QGroupBox()
self.threadInfo = QLabel("Thread Info.")
self.groupBoxThreadInfo.setStyleSheet(
"QGroupBox {border: 1px solid gray; border-radius: 9px; margin-top: 0.5em} QGroupBox::title {subcontrol-origin: margin; left: 10px; padding: 0 3px 0 3px;"
)
if not self.threadvbox:
self.threadvbox = QVBoxLayout()
if not self.listThread:
self.listThread = QListWidget()
self.listThread.setFixedWidth(200)
self.listThread.setSelectionMode(
QtGui.QAbstractItemView.MultiSelection)
QtCore.QObject.connect(self.listThread,
QtCore.SIGNAL("itemClicked(QListWidgetItem *)"),
self.toggleThreadDisplay)
self.threadvbox.addWidget(self.threadInfo)
self.threadvbox.addWidget(self.listThread)
self.groupBoxThreadInfo.setLayout(self.threadvbox)
self.addWidget(self.groupBoxThreadInfo)
self.groupBoxThreadInfo.setSizePolicy(self.sizePolicy)
for id in threads:
item = QtGui.QListWidgetItem(id)
self.listThread.addItem(item)
def connectController(self, controller):
self.controller = controller
self.connect(controller, QtCore.SIGNAL('setAvailable()'),
self.setAvailable)
def connectDiagramView(self, view):
self.diagramView = view
def disconnectMsgRcv(self, receiver):
print("Implement this method !!! disconnectMsgRcv")
def connectMsgRcv(self, receiver):
self.msgRcv.append(receiver)
def notifyHide(self):
for rcv in self.msgRcv:
rcv.activateHide(True)
def showHiddenLifelines(self):
response, selected_items = HiddenDialog.HiddenDialog.getSelectedItems(
self.diagramView.getHiddenLifeLines())
if response:
self.diagramView.showLifelines(selected_items)
def showHiddenMessages(self):
response, selected_items = HiddenMessageDialog.HiddenMessageDialog.getSelectedItems(
self.diagramView.getHiddenMessages(),
self.diagramView.getHiddenLifeLines())
if response:
if selected_items[3] in self.diagramView.getHiddenLifeLines():
confirmation = ShowLifeLineDialog.ShowLifeLineDialog.confirmToShowLifeLine(
selected_items[3])
if confirmation:
self.diagramView.showLifelines([selected_items[3]])
self.diagramView.showMessages(selected_items)
else:
self.diagramView.showMessages(selected_items)
def notifyCapture(self):
for rcv in self.msgRcv:
rcv.activateCapture(True)
def moveToPrev(self):
for rcv in self.msgRcv:
rcv.moveToPrev()
def moveToNext(self):
for rcv in self.msgRcv:
rcv.moveToNext()
def searchMsg(self):
str = self.editTextSearch.toPlainText()
for rcv in self.msgRcv:
rcv.searchMessage(str)
class ToolBox(QVBoxLayout):
sig = QtCore.Signal(object)
listThread = None
groupBoxThreadInfo = None
threadvbox = None
mode = None
def __init__(self, mode, parentQWidget = None):
QVBoxLayout.__init__(self)
self.sig.connect(self.addThreadList)
self.mode = mode
self.sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed, QtGui.QSizePolicy.Expanding)
self.groupBoxSearch = QGroupBox()
self.groupBoxSearch.setStyleSheet("QGroupBox {border: 1px solid gray; border-radius: 4px; };")
vboxSearch = QVBoxLayout()
self.searchTitle = QLabel("Search Messages")
vboxSearch.addWidget(self.searchTitle)
self.searchHLayout = QHBoxLayout()
self.editTextSearch = QTextEdit('')
self.editTextSearch.setFixedSize(200,30)
self.buttonSearch = QPushButton('Search')
self.buttonSearch.setFixedSize(100,30)
self.buttonSearch.clicked.connect(self.searchMsg)
vboxSearch.addWidget(self.editTextSearch)
self.searchHLayout.addWidget(self.buttonSearch)
self.searchCursor = QLabel()
self.searchHLayout.addWidget(self.searchCursor)
vboxSearch.addLayout(self.searchHLayout)
self.browseHLayout = QHBoxLayout()
self.buttonLookUp = QPushButton('\u21e7') #Arrow up
self.buttonLookUp.setFixedWidth(100)
self.buttonLookUp.clicked.connect(self.moveToPrev)
self.buttonLookDown = QPushButton('\u21e9') #Arrow down
self.buttonLookDown.setFixedWidth(100)
self.buttonLookDown.clicked.connect(self.moveToNext)
self.browseHLayout.addWidget(self.buttonLookUp)
self.browseHLayout.addWidget(self.buttonLookDown)
vboxSearch.addLayout(self.browseHLayout)
self.groupBoxSearch.setLayout(vboxSearch)
self.addWidget(self.groupBoxSearch)
self.groupBoxSearch.setSizePolicy(QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed, QtGui.QSizePolicy.Fixed))
self.buttonHiddenLifelines = QPushButton('Show hidden life-lines')
self.buttonHiddenLifelines.setFixedWidth(200)
self.buttonHiddenLifelines.clicked.connect(self.showHiddenLifelines)
self.addWidget(self.buttonHiddenLifelines)
self.buttonHiddenMessages = QPushButton('Show hidden Messages')
self.buttonHiddenMessages.setFixedWidth(200)
self.buttonHiddenMessages.clicked.connect(self.showHiddenMessages)
self.addWidget(self.buttonHiddenMessages)
if const.mode_interactive == mode:
self.buttonCapture = QPushButton('Capture')
self.buttonCapture.setFixedWidth(200)
self.buttonCapture.clicked.connect(self.notifyCapture)
self.addWidget(self.buttonCapture)
self.msgRcv = []
self.msgInfo = QLabel("Message Info.")
self.groupBoxMessageInfo = QGroupBox()
self.groupBoxMessageInfo.setStyleSheet("QGroupBox {border: 1px solid gray; border-radius: 9px; margin-top: 0.5em} QGroupBox::title {subcontrol-origin: margin; left: 10px; padding: 0 3px 0 3px;")
vbox = QVBoxLayout()
vbox.addWidget(self.msgInfo)
self.tableTime = QtGui.QTableWidget(3,2)
self.tableTime.setHorizontalHeaderLabels(['-','time'])
self.tableTime.setColumnWidth(0,80)
self.tableTime.setColumnWidth(1,150)
vwidth = self.tableTime.verticalHeader().length()
hwidth = self.tableTime.horizontalHeader().height()
fwidth = self.tableTime.frameWidth() * 2
self.tableTime.setFixedHeight(vwidth + hwidth + fwidth)
self.tableTime.horizontalHeader().setStretchLastSection(True)
self.tableTime.setItem(0,0,QTableWidgetItem('begin'))
self.tableTime.setItem(0,1,QTableWidgetItem(' - '))
self.tableTime.setItem(1,0,QTableWidgetItem('end'))
self.tableTime.setItem(1,1,QTableWidgetItem(' - '))
self.tableTime.setItem(2,0,QTableWidgetItem('duration'))
self.tableTime.setItem(2,1,QTableWidgetItem(' - '))
vbox.addWidget(self.tableTime)
self.titleArg = QLabel('Argument List')
vbox.addWidget(self.titleArg)
max_arg_num = 10
self.tableArgs = QtGui.QTableWidget(max_arg_num,2)
self.tableArgs.setHorizontalHeaderLabels(['type','value'])
for idx in range(0,max_arg_num):
self.tableArgs.setItem(idx,0,QTableWidgetItem())
self.tableArgs.setItem(idx,1,QTableWidgetItem())
self.tableArgs.horizontalHeader().setStretchLastSection(True)
vbox.addWidget(self.tableArgs)
self.titleArg = QLabel('Return Value List')
vbox.addWidget(self.titleArg)
max_ret_num = 4
self.tableRet = QtGui.QTableWidget(max_ret_num,2)
self.tableRet.setHorizontalHeaderLabels(['type','value'])
for idx in range(0,max_ret_num):
self.tableRet.setItem(idx,0,QTableWidgetItem())
self.tableRet.setItem(idx,1,QTableWidgetItem())
self.tableRet.horizontalHeader().setStretchLastSection(True)
vwidth = self.tableRet.verticalHeader().length()
hwidth = self.tableRet.horizontalHeader().height()
fwidth = self.tableRet.frameWidth() * 2
self.tableRet.setFixedHeight(vwidth + hwidth + fwidth)
vbox.addWidget(self.tableRet)
self.buttonSrcView = QPushButton('view code')
self.buttonSrcView.setFixedWidth(200)
self.buttonSrcView.clicked.connect(self.openSourceViewer)
self.buttonHide = QPushButton('Hide')
self.buttonHide.setFixedWidth(200)
self.buttonHide.clicked.connect(self.notifyHide)
self.buttonHideAllMsg = QPushButton('Hide All')
self.buttonHideAllMsg.setFixedWidth(200)
self.buttonHideAllMsg.clicked.connect(self.hideAllMsgNamedAsSelected)
self.groupBoxMessageInfo.setLayout(vbox)
self.checkHideCircular = QCheckBox('Hide Circular Messages')
self.checkHideCircular.setCheckState(QtCore.Qt.Unchecked)
self.checkHideCircular.stateChanged.connect(self.changeHideCircularMessage)
self.addWidget(self.checkHideCircular)
self.addWidget(self.groupBoxMessageInfo)
self.groupBoxMessageInfo.setSizePolicy(self.sizePolicy)
def reset(self):
for idx in reversed(range(0,self.listThread.count())):
self.listThread.takeItem(idx)
def setMsgInfoMessage(self,msg):
self.strMessage = msg
def changeHideCircularMessage(self,state):
if state == QtCore.Qt.Unchecked:
self.diagramView.hideCircularChanged(False)
elif state == QtCore.Qt.Checked:
self.diagramView.hideCircularChanged(True)
def setMsgInfoModule(self,module):
self.strModule = module
def updateSearchStatus(self,curr,number):
self.searchCursor.setText("%d/%d" % (curr,number))
def connectSourceViewer(self,viewer):
self.srcViewer = viewer
def openSourceViewer(self):
self.srcViewer.openViewer(self.strModule,self.strMessage)
def setMessageInfoTime(self,begin,end,duration):
self.tableTime.item(0,1).setText(begin)
self.tableTime.item(1,1).setText(end)
self.tableTime.item(2,1).setText(duration + ' msec')
def setMessageInfoArg(self,listParam,listArg):
if listArg:
for idx, text in enumerate(listArg):
self.tableArgs.item(idx,1).setText(text)
for idx, text in enumerate(listParam):
self.tableArgs.item(idx,0).setText(text)
else:
for idx in range(0,self.tableArgs.rowCount()):
self.tableArgs.item(idx,1).setText('')
self.tableArgs.item(idx,0).setText('')
def setMessageInfoRet(self,listRet):
if listRet:
for idx, text in enumerate(listRet):
self.tableRet.item(idx,1).setText(text)
else:
for idx in range(0,self.tableRet.rowCount()):
self.tableRet.item(idx,1).setText('')
self.tableRet.item(idx,0).setText('')
def notifyInteractiveStateChanged(self,state):
if const.mode_interactive != self.mode:
return
if const.STATE_INTERACTIVE_CAPTURING == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Stop Capture')
if const.STATE_INTERACTIVE_PROCESSING == state:
self.buttonCapture.setEnabled(False)
if const.STATE_INTERACTIVE_IDLE == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Capture')
if const.STATE_INTERACTIVE_RESET == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Capture')
elif const.STATE_INTERACTIVE_ACTIVE == state:
self.buttonCapture.setEnabled(True)
self.buttonCapture.setText('Capture')
def setMessageInfo(self,info):
self.msgInfo.setText(info)
def setAvailable(self,threads):
self.sig.emit(threads)
def toggleThreadDisplay(self,item):
print(self.listThread.currentRow())
#if item.isSelected():
# print(item.text() + " is selected")
#else:
# print(item.text() + " is not selected")
self.diagramView.showThread(self.listThread.currentRow(),item.isSelected())
def hideAllMsgNamedAsSelected(self):
self.diagramView.hideAllMessageSelected()
def addThreadList(self,threads):
if not self.groupBoxThreadInfo:
self.groupBoxThreadInfo = QGroupBox()
self.threadInfo = QLabel("Thread Info.")
self.groupBoxThreadInfo.setStyleSheet("QGroupBox {border: 1px solid gray; border-radius: 9px; margin-top: 0.5em} QGroupBox::title {subcontrol-origin: margin; left: 10px; padding: 0 3px 0 3px;")
if not self.threadvbox:
self.threadvbox = QVBoxLayout()
if not self.listThread:
self.listThread = QListWidget()
self.listThread.setFixedWidth(200)
self.listThread.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
QtCore.QObject.connect(self.listThread, QtCore.SIGNAL("itemClicked(QListWidgetItem *)"), self.toggleThreadDisplay)
self.threadvbox.addWidget(self.threadInfo)
self.threadvbox.addWidget(self.listThread)
self.groupBoxThreadInfo.setLayout(self.threadvbox)
self.addWidget(self.groupBoxThreadInfo)
self.groupBoxThreadInfo.setSizePolicy(self.sizePolicy)
for id in threads:
item = QtGui.QListWidgetItem(id)
self.listThread.addItem(item)
def connectController(self,controller):
self.controller = controller
self.connect(controller,QtCore.SIGNAL('setAvailable()'),self.setAvailable)
def connectDiagramView(self,view):
self.diagramView = view
def disconnectMsgRcv(self,receiver):
print("Implement this method !!! disconnectMsgRcv")
def connectMsgRcv(self,receiver):
self.msgRcv.append(receiver)
def notifyHide(self):
for rcv in self.msgRcv:
rcv.activateHide(True)
def showHiddenLifelines(self):
response, selected_items = HiddenDialog.HiddenDialog.getSelectedItems(self.diagramView.getHiddenLifeLines())
if response:
self.diagramView.showLifelines(selected_items)
def showHiddenMessages(self):
response, selected_items = HiddenMessageDialog.HiddenMessageDialog.getSelectedItems(self.diagramView.getHiddenMessages(),self.diagramView.getHiddenLifeLines())
if response:
if selected_items[3] in self.diagramView.getHiddenLifeLines():
confirmation = ShowLifeLineDialog.ShowLifeLineDialog.confirmToShowLifeLine(selected_items[3])
if confirmation:
self.diagramView.showLifelines([selected_items[3]])
self.diagramView.showMessages(selected_items)
else:
self.diagramView.showMessages(selected_items)
def notifyCapture(self):
for rcv in self.msgRcv:
rcv.activateCapture(True)
def moveToPrev(self):
for rcv in self.msgRcv:
rcv.moveToPrev()
def moveToNext(self):
for rcv in self.msgRcv:
rcv.moveToNext()
def searchMsg(self):
str = self.editTextSearch.toPlainText()
for rcv in self.msgRcv:
rcv.searchMessage(str)
class ConfigEditor(QMainWindow):
def __init__(self, app, cfg, title='Config Editor'):
super(ConfigEditor, self).__init__()
self.app = app
self.config = cfg
self.title = title
def setup(self):
self.dirty = False
self.def_cfg = copy.deepcopy(self.config)
self.config.update_from_user_file()
self.base_cfg = copy.deepcopy(self.config)
self.categories = QListWidget()
#self.categories.setSizePolicy(QSizePolicy.Fixed,QSizePolicy.Expanding)
self.settings = QStackedWidget()
#self.categories.setSizePolicy(QSizePolicy.MinimumExpanding,QSizePolicy.Expanding)
QObject.connect(self.categories, SIGNAL('itemSelectionChanged()'), self.category_selected)
self.widget_list = {}
for cat in self.config.get_categories():
self.widget_list[cat] = {}
longest_cat = 0
for cat in self.config.get_categories():
if len(cat) > longest_cat:
longest_cat = len(cat)
self.categories.addItem(cat)
settings_layout = QGridLayout()
r = 0
c = 0
for setting in self.config.get_settings(cat):
info = self.config.get_setting(cat, setting, True)
s = QWidget()
s.setSizePolicy(QSizePolicy.Expanding,QSizePolicy.Fixed)
sl = QVBoxLayout()
label = QLabel()
if info.has_key('alias'):
label.setText(info['alias'])
else:
label.setText(setting)
if info.has_key('about'):
label.setToolTip(info['about'])
sl.addWidget(label)
if info['type'] == constants.CT_LINEEDIT:
w = LineEdit(self, self.config,cat,setting,info)
elif info['type'] == constants.CT_CHECKBOX:
w = CheckBox(self, self.config,cat,setting,info)
elif info['type'] == constants.CT_SPINBOX:
w = SpinBox(self, self.config,cat,setting,info)
elif info['type'] == constants.CT_DBLSPINBOX:
w = DoubleSpinBox(self, self.config,cat,setting,info)
elif info['type'] == constants.CT_COMBO:
w = ComboBox(self, self.config,cat,setting,info)
w.setSizePolicy(QSizePolicy.Expanding,QSizePolicy.Fixed)
self.widget_list[cat][setting] = w
sl.addWidget(w)
s.setLayout(sl)
c = self.config.config[cat].index(setting) % 2
settings_layout.addWidget(s, r, c)
if c == 1:
r += 1
settings = QWidget()
settings.setLayout(settings_layout)
settings_scroller = QScrollArea()
settings_scroller.setWidget(settings)
settings_scroller.setWidgetResizable(True)
self.settings.addWidget(settings_scroller)
font = self.categories.font()
fm = QFontMetrics(font)
self.categories.setMaximumWidth(fm.widthChar('X')*(longest_cat+4))
self.main = QWidget()
self.main_layout = QVBoxLayout()
self.config_layout = QHBoxLayout()
self.config_layout.addWidget(self.categories)
self.config_layout.addWidget(self.settings)
self.mainButtons = QDialogButtonBox(QDialogButtonBox.RestoreDefaults | QDialogButtonBox.Reset | QDialogButtonBox.Apply)
self.main_apply = self.mainButtons.button(QDialogButtonBox.StandardButton.Apply)
self.main_reset = self.mainButtons.button(QDialogButtonBox.StandardButton.Reset)
self.main_defaults = self.mainButtons.button(QDialogButtonBox.StandardButton.LastButton)
QObject.connect(self.mainButtons, SIGNAL('clicked(QAbstractButton *)'), self.mainbutton_clicked)
self.dirty_check()
self.main_layout.addLayout(self.config_layout)
self.main_layout.addWidget(self.mainButtons)
self.main.setLayout(self.main_layout)
self.setCentralWidget(self.main)
self.setWindowTitle(self.title)
self.setUnifiedTitleAndToolBarOnMac(True)
self.categories.setCurrentItem(self.categories.item(0))
self.menuBar = QMenuBar()
self.filemenu = QMenu('&File')
self.quitAction = QAction(self)
self.quitAction.setText('&Quit')
if platform.system() != 'Darwin':
self.quitAction.setShortcut(QKeySequence(Qt.CTRL + Qt.Key_Q))
QObject.connect(self.quitAction, SIGNAL('triggered()'), self.quitApp)
self.filemenu.addAction(self.quitAction)
self.menuBar.addMenu(self.filemenu)
self.setMenuBar(self.menuBar)
self.show()
self.activateWindow()
self.raise_()
self.setMinimumWidth(self.geometry().width()*1.2)
screen = QDesktopWidget().screenGeometry()
size = self.geometry()
self.move((screen.width()-size.width())/2, (screen.height()-size.height())/2)
def category_selected(self):
self.settings.setCurrentIndex(self.config.config.index(self.categories.selectedItems()[0].text()))
def mainbutton_clicked(self, button):
if button == self.main_reset:
for cat in self.base_cfg.get_categories():
for setting in self.base_cfg.get_settings(cat):
self.widget_list[cat][setting].updateValue(self.base_cfg.get_setting(cat,setting))
elif button == self.main_defaults:
for cat in self.def_cfg.get_categories():
for setting in self.def_cfg.get_settings(cat):
self.widget_list[cat][setting].updateValue(self.def_cfg.get_setting(cat,setting))
elif button == self.main_apply:
bad_settings = self.validate_settings()
if bad_settings == []:
self.save_settings()
self.main_apply.setEnabled(False)
self.main_reset.setEnabled(False)
else:
msgBox = QMessageBox()
msgBox.setText("Must fix the following invalid settings before quitting:")
msgBox.setStandardButtons(QMessageBox.Ok)
info = ''
for setting in bad_settings:
new = '%s,%s<br>' % setting
info = '%s%s' % (info, new)
msgBox.setInformativeText(info)
msgBox.exec_()
def quitApp(self):
self.app.closeAllWindows()
def get_changes(self):
enc = MyEncoder()
if enc.encode(self.def_cfg.config) == enc.encode(self.config.config):
return False
if enc.encode(self.base_cfg.config) != enc.encode(self.config.config):
newC = Config()
for c in self.config.config.keys():
for s in self.config.config[c].keys():
if self.config.config[c][s]['value'] != self.def_cfg.config[c][s]['value']:
newC.add_setting(c, s, self.config.config[c][s]['value'], stub=True)
return json.dumps(newC.config, separators=(',',': '), indent=4, sort_keys=True)
else:
return None
def validate_settings(self):
ret = []
for cat in self.config.get_categories():
for setting in self.config.get_settings(cat):
info = self.config.get_setting(cat, setting, True)
if info.has_key('validate'):
if not info['validate'](info):
ret.append((cat,setting))
return ret
def dirty_check(self):
if str(self.base_cfg) != str(self.config):
self.dirty = True
self.main_apply.setEnabled(True)
self.main_reset.setEnabled(True)
else:
self.dirty = False
self.main_apply.setEnabled(False)
self.main_reset.setEnabled(False)
if str(self.def_cfg) == str(self.config):
self.main_defaults.setEnabled(False)
else:
self.main_defaults.setEnabled(True)
def save_settings(self):
config = self.get_changes()
if config == False:
if os.path.isfile(self.config.user_file):
os.remove(self.config.user_file)
elif config != None:
with open(self.config.user_file, 'w+') as f:
f.write(config)
self.base_cfg = copy.deepcopy(self.config)
def closeEvent(self, event=None):
self.quitApp()
class SubscriberDialog(QDialog):
if USE_MAEMO_5:
switchRequested = Signal()
def __init__(self, parent=None):
QDialog.__init__(self, parent)
self.ui = Ui_SubscriberDialog()
self.ui.setupUi(self)
self.subscriber = None
self.tableWidget = None
self.listWidget = None
if USE_MAEMO_5:
switchButton = self.ui.buttonBox.addButton(self.tr('Switch'), QDialogButtonBox.ActionRole)
switchButton.clicked.connect(self.switchRequested)
self.tableWidget = self.ui.tableWidget
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
verticalHeader = self.tableWidget.verticalHeader()
verticalHeader.setVisible(False)
self.tableWidget.setColumnWidth(0, 200)
self.tableWidget.setColumnWidth(1, 400)
else:
desktopWidget = QDesktopWidget()
if desktopWidget.availableGeometry().width() < 400:
# Screen is too small to fit a table widget without scrolling, use a list widget instead.
self.listWidget = QListWidget()
self.listWidget.setAlternatingRowColors(True)
self.ui.verticalLayout.insertWidget(2, self.listWidget)
else:
self.tableWidget = QTableWidget()
headerLabels = ('Key', 'Value', 'Type')
self.tableWidget.setColumnCount(3)
self.tableWidget.setHorizontalHeaderLabels(headerLabels)
horizontalHeader = self.tableWidget.horizontalHeader()
horizontalHeader.setStretchLastSection(True)
self.tableWidget.verticalHeader()
self.setVisible(False)
self.ui.verticalLayout.insertWidget(2, self.tableWidget)
self.ui.connectButton.clicked.connect(self.changeSubscriberPath)
self.changeSubscriberPath()
# if the default path does not exist reset it to /
value = self.subscriber.value()
subPaths = self.subscriber.subPaths()
if not value and not subPaths:
self.ui.basePath.setText('/')
self.changeSubscriberPath()
def changeEvent(self, e):
QDialog.changeEvent(self, e)
if e.type() == QEvent.LanguageChange:
self.ui.retranslateUi(self)
def changeSubscriberPath(self):
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
if not self.subscriber:
self.subscriber = QValueSpaceSubscriber(self.ui.basePath.text(), self)
else:
self.subscriber.setPath(self.ui.basePath.text())
self.subscriber.contentsChanged.connect(self.subscriberChanged)
self.subscriber.connectNotify("contentsChanged()")
self.subscriberChanged()
def subscriberChanged(self):
subPaths = self.subscriber.subPaths()
if self.listWidget:
self.listWidget.clear()
elif self.tableWidget:
self.tableWidget.clearContents()
self.tableWidget.setRowCount(len(subPaths))
for i in xrange(len(subPaths)):
v = self.subscriber.value(subPaths[i])
if self.listWidget:
item = QListWidgetItem('%s (%s)\n%s' % (subPaths[i], str(type(v)), str(v)))
item.setFlags(item.flags() & ~Qt.ItemIsEditable)
self.listWidget.addItem(item)
elif self.tableWidget:
pathItem = QTableWidgetItem(subPaths[i])
pathItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
valueItem = QTableWidgetItem(str(v))
valueItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
typeItem = QTableWidgetItem(str(type(v)))
typeItem.setFlags(pathItem.flags() & ~Qt.ItemIsEditable)
self.tableWidget.setItem(i, 0, pathItem)
self.tableWidget.setItem(i, 1, valueItem)
self.tableWidget.setItem(i, 2, typeItem)
class PysideGui(generic.GenericGui):
def __init__(self):
generic.GenericGui.__init__(self)
window = QWidget()
window.setWindowTitle('quichem-pyside')
self.compiler_view = QListWidget()
self.compiler_view.currentRowChanged.connect(self.show_source)
self.stacked_widget = QStackedWidget()
self.stacked_widget.setFrameStyle(QFrame.StyledPanel | QFrame.Raised)
self.edit = QLineEdit()
self.edit.setPlaceholderText('Type quichem input...')
self.edit.textChanged.connect(self.change_value)
self.view = QWebView()
self.view.page().mainFrame().setScrollBarPolicy(Qt.Vertical,
Qt.ScrollBarAlwaysOff)
self.view.page().action(QWebPage.Reload).setVisible(False)
self.view.setMaximumHeight(0)
self.view.setUrl('qrc:/web/page.html')
self.view.setZoomFactor(2)
self.view.page().mainFrame().contentsSizeChanged.connect(
self._resize_view)
# For debugging JS:
## from PySide.QtWebKit import QWebSettings
## QWebSettings.globalSettings().setAttribute(
## QWebSettings.DeveloperExtrasEnabled, True)
button_image = QPushButton('Copy as Image')
button_image.clicked.connect(self.set_clipboard_image)
button_image.setToolTip('Then paste into any graphics program')
button_word = QPushButton('Copy as MS Word Equation')
button_word.clicked.connect(self.set_clipboard_word)
button_html = QPushButton('Copy as Formatted Text')
button_html.clicked.connect(self.set_clipboard_html)
line = QFrame()
line.setFrameShape(QFrame.HLine)
line.setFrameShadow(QFrame.Sunken)
button_layout = QHBoxLayout()
button_layout.addStretch()
button_layout.addWidget(button_image)
button_layout.addWidget(button_word)
button_layout.addWidget(button_html)
source_layout = QHBoxLayout()
source_layout.addWidget(self.compiler_view)
source_layout.addWidget(self.stacked_widget, 1)
QVBoxLayout(window)
window.layout().addWidget(self.edit)
window.layout().addWidget(self.view)
window.layout().addLayout(button_layout)
window.layout().addWidget(line)
window.layout().addLayout(source_layout, 1)
window.show()
window.resize(window.minimumWidth(), window.height())
# To prevent garbage collection of internal Qt object.
self._window = window
def show_source(self, index):
if not self.sources:
return
self.stacked_widget.setCurrentIndex(index)
self.change_value(self.edit.text())
def _resize_view(self):
"""Set the QWebView's minimum height based on its current
contents.
"""
div = self.view.page().mainFrame().findFirstElement('.output')
scrollbar_width = QApplication.style().pixelMetric(
QStyle.PM_ScrollBarExtent)
self.view.setMaximumHeight(
div.geometry().height() + scrollbar_width + 16)
def make_source(self, name):
self.compiler_view.addItem(name)
self.compiler_view.setCurrentRow(0)
scrollbar_width = QApplication.style().pixelMetric(
QStyle.PM_ScrollBarExtent)
self.compiler_view.setMaximumWidth(
self.compiler_view.sizeHintForColumn(0) + scrollbar_width + 16)
page = QWidget()
QHBoxLayout(page)
page.layout().setContentsMargins(*(0,) * 4)
source = QTextEdit()
source.setStyleSheet('min-width: 0; min-height: 0')
source.setReadOnly(True)
QVBoxLayout(source)
button = QPushButton('Copy')
button.clicked.connect(functools.partial(self.set_clipboard, source))
page.layout().addWidget(source)
source.layout().addWidget(button, 0, Qt.AlignRight | Qt.AlignBottom)
self.stacked_widget.addWidget(page)
return source
def run_script(self, js):
self.view.page().mainFrame().evaluateJavaScript(js)
def set_source(self, widget, source_factory):
if widget.isVisible():
widget.setPlainText(source_factory())
def set_clipboard_image(self):
"""Export the formatted output to an image and store it in the
clipboard.
The image stored in the clipboard is a PNG file with alpha
transparency.
"""
div = self.view.page().mainFrame().findFirstElement('.output')
images = {}
for background in (Qt.transparent, Qt.white):
image = QImage(div.geometry().size(),
QImage.Format_ARGB32_Premultiplied)
image.fill(background)
painter = QPainter(image)
div.render(painter)
painter.end()
images[background] = image
# Windows needs this buffer hack to get alpha transparency in
# the copied PNG.
buffer_ = QBuffer()
buffer_.open(QIODevice.WriteOnly)
images[Qt.transparent].save(buffer_, 'PNG')
buffer_.close()
data = QMimeData()
data.setData('PNG', buffer_.data())
data.setImageData(images[Qt.white])
QApplication.clipboard().setMimeData(data)
def set_clipboard_word(self):
"""Store the formatted output in the clipboard in a Microsoft
Word friendly format.
Microsoft Word interprets the clipboard contents as an
equation. Other programs will see it as plain text containing
XML.
"""
QApplication.clipboard().setText(generic.word_equation_from_mathml(
self.view.page().mainFrame().evaluateJavaScript(generic.MML_JS)))
def set_clipboard_html(self):
"""Place the HTML displayed in the HTML view widget into the
system clipboard.
"""
data = QMimeData()
data.setText(self.plain)
data.setHtml(self.html)
QApplication.clipboard().setMimeData(data)
def set_clipboard(self, source):
"""Place the text displayed in the given source widget into the
system clipboard.
"""
QApplication.clipboard().setText(source.toPlainText())
class Form(QDialog):
def __init__(self, state, info, parent=None):
super().__init__(parent)
Lib.prepareModalDialog(self)
self.state = state
ListWidgetItem.info = info
ListWidgetItem.refresh = self.refresh
self.info = info
self.helpPage = info.help
self.createWidgets()
self.layoutWidgets()
self.setWindowTitle("{} — {}".format(self.info.name,
QApplication.applicationName()))
self.refresh()
settings = QSettings()
self.updateToolTips(
bool(
int(
settings.value(Gopt.Key.ShowDialogToolTips,
Gopt.Default.ShowDialogToolTips))))
def createWidgets(self):
self.listWidget = QListWidget()
self.buttonLayout = QVBoxLayout()
for icon, text, slot, tip in ((":/add.svg", "&Add", self.add, """\
<p><b>Add</b></p><p>Add an item to the {}
list.</p>""".format(self.info.name)), (":/edit.svg", "&Edit", self.edit, """\
<p><b>Edit</b></p><p>Edit the {} list's current
item.</p>""".format(self.info.name)), (":/delete.svg", "&Remove...",
self.remove, """\
<p><b>Remove</b></p><p>Remove the {} list's current
item.</p>""".format(self.info.name)), (":/help.svg", "Help", self.help, """\
Help on the {} dialog""".format(self.info.name)), (":/dialog-close.svg",
"&Close", self.accept, """\
<p><b>Close</b></p><p>Close the dialog.</p>""")):
button = QPushButton(QIcon(icon), text)
button.setFocusPolicy(Qt.NoFocus)
if text in {"&Close", "Help"}:
self.buttonLayout.addStretch()
self.buttonLayout.addWidget(button)
button.clicked.connect(slot)
self.tooltips.append((button, tip))
self.tooltips.append((self.listWidget, self.info.desc))
def layoutWidgets(self):
layout = QHBoxLayout()
layout.addWidget(self.listWidget)
layout.addLayout(self.buttonLayout)
self.setLayout(layout)
def refresh(self, *, text=None):
self.listWidget.clear()
cursor = self.info.db.cursor()
for row, record in enumerate(cursor.execute(self.info.SELECT)):
item = ListWidgetItem(record[0])
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEditable
| Qt.ItemIsEnabled)
item.setBackground(self.palette().base() if row %
2 else self.palette().alternateBase())
self.listWidget.addItem(item)
if self.listWidget.count():
self.listWidget.setCurrentRow(0)
if text is not None:
for i in range(self.listWidget.count()):
if self.listWidget.item(i).text() == text:
self.listWidget.setCurrentRow(i)
break
def add(self):
item = ListWidgetItem(ADDING)
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEditable
| Qt.ItemIsEnabled)
self.listWidget.insertItem(0, item)
self.listWidget.setCurrentRow(0)
self.listWidget.editItem(item)
def edit(self):
item = self.listWidget.currentItem()
if item is not None:
self.listWidget.editItem(item)
def remove(self): # No need to restore focus widget
row = self.listWidget.currentRow()
item = self.listWidget.item(row)
if item is None:
return
with Lib.Qt.DisableUI(self, forModalDialog=True):
reply = QMessageBox.question(
self, "Remove {} — {}".format(self.info.name,
QApplication.applicationName()),
"Remove {} “{}”?".format(self.info.name, item.text()),
QMessageBox.Yes | QMessageBox.No)
if reply == QMessageBox.Yes:
cursor = self.info.db.cursor()
cursor.execute(self.info.DELETE, (item.text(), ))
item = self.listWidget.takeItem(row)
del item
def help(self):
self.state.help(self.helpPage)
class CC_window(QWidget):
def __init__(self):
super(CC_window, self).__init__()
self.version = '0.2'
glo = QVBoxLayout(self)
self.Runningo = None
self.Looping = None
self.P = CC_thread()
if name == 'nt':
ficon = r_path('images\\favicon.png')
licon = r_path('images\\logo.png')
else:
ficon = r_path('images/favicon.png')
licon = r_path('images/logo.png')
self.tf = QFont("", 13, QFont.Bold)
self.sf = QFont("", 10, QFont.Bold)
self.SelfIinit(QIcon(ficon))
self.center()
self.a_btn(licon, glo)
self.ss_btns(glo)
self.i_list(glo)
self.cl_btn(glo)
self.l_btns(glo)
self.f_btns(glo)
self.ds_bar(glo)
self.setLayout(glo)
self.activateWindow()
self.show()
def SelfIinit(self, icon):
self.setWindowTitle('chrome-cut ' + self.version)
self.setGeometry(300, 300, 200, 150)
self.setMinimumWidth(600)
self.setMaximumWidth(600)
self.setMinimumHeight(500)
self.setMaximumHeight(500)
# Setting Icon
self.setWindowIcon(icon)
QToolTip.setFont(self.sf)
def msgApp(self, title, msg):
uinfo = QMessageBox.question(self, title, msg,
QMessageBox.Yes | QMessageBox.No)
if uinfo == QMessageBox.Yes:
return 'y'
if uinfo == QMessageBox.No:
return 'n'
def closeEvent(self, event=None):
if self.P.isRunning():
response = self.msgApp("Making sure",
"Sure, you want to exit while looping ?")
if response == 'y':
if event is not None:
event.accept()
self.P.stop()
exit(0)
else:
if event is not None:
event.ignore()
else:
if event is not None:
event.accept()
exit(0)
def center(self):
qrect = self.frameGeometry()
cp = QDesktopWidget().availableGeometry().center()
qrect.moveCenter(cp)
self.move(qrect.topLeft())
def a_btn(self, icon, glo):
def show_about():
Amsg = "<center>All credit reserved to the author of chrome-cut "
Amsg += " version " + self.version
Amsg += ", This work is a free, open-source project licensed "
Amsg += " under Mozilla Public License version 2.0 . <br><br>"
Amsg += " visit us for more infos and how-tos :<br> "
Amsg += "<b><a href='https://github.io/mrf345/chrome-cut'> "
Amsg += "https://github.io/mrf345/chrome-cut </a> </b></center>"
Amsgb = "About chrome-cut"
return QMessageBox.about(self, Amsgb, Amsg)
self.abutton = QPushButton('', self)
self.abutton.setIcon(QPixmap(icon))
self.abutton.setIconSize(QSize(500, 100))
self.abutton.setToolTip('About chrome-cut')
self.abutton.clicked.connect(show_about)
glo.addWidget(self.abutton)
def ss_btns(self, glo):
self.lebutton = QPushButton('Scan IP address', self)
self.lebutton.setToolTip(
'Insert and check if a specific IP is a valid chrome cast')
self.lebutton.setFont(self.tf)
self.lebutton.clicked.connect(self.sip_input)
glo.addWidget(self.lebutton)
def sip_input(self):
self.lebutton.setEnabled(False)
self.s_bar.setStyleSheet(self.s_norm)
self.s_bar.showMessage('> Checking out the IP ..')
msg = "<center>Enter suspected IP to check : <br><i><u>Usually it is "
msg += "on 192.168.what_have_you.188</i></center>"
text, okPressed = QInputDialog.getText(self, "Scan IP", msg)
if okPressed and text != '':
self.setCursor(Qt.BusyCursor)
ch = is_ccast(text)
if ch:
self.il_add([text])
self.unsetCursor()
self.lebutton.setEnabled(True)
return True
else:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage(
'! Error: inserted IP is not chrome cast device')
self.unsetCursor()
self.lebutton.setEnabled(True)
return True
self.s_bar.clearMessage()
self.lebutton.setEnabled(True)
return True
def i_list(self, glo):
self.ci_list = QListWidget()
self.ci_list.setToolTip("List of discovered devices")
self.ci_list.setEnabled(False)
self.setFont(self.sf)
glo.addWidget(self.ci_list)
def cl_btn(self, glo):
self.cle_btn = QPushButton('Clear devices list')
self.cle_btn.setToolTip("Remove all devices from the list")
self.cle_btn.setFont(self.sf)
self.cle_btn.setEnabled(False)
self.cle_btn.clicked.connect(self.il_add)
glo.addWidget(self.cle_btn)
def l_btns(self, glo):
hlayout = QHBoxLayout()
self.llo = QCheckBox("Looping")
self.llo.setToolTip(
'Automate repeating the smae command on the smae device selected ')
self.llo.setEnabled(False)
self.nlo = QLineEdit()
self.nlo.setPlaceholderText("duration in seconds. Default is 10")
self.nlo.setEnabled(False)
self.nlo.setToolTip("Duration of sleep after each loop")
self.lbtn = QPushButton('Stop')
self.lbtn.setEnabled(False)
self.lbtn.setFont(self.tf)
self.llo.setFont(self.sf)
self.lbtn.setToolTip("stop on-going command or loop")
self.llo.toggled.connect(self.loped)
self.lbtn.clicked.connect(partial(self.inloop_state, out=True))
hlayout.addWidget(self.llo)
hlayout.addWidget(self.nlo)
hlayout.addWidget(self.lbtn)
glo.addLayout(hlayout)
def loped(self):
if self.Looping:
self.Looping = None
self.llo.setChecked(False)
self.nlo.setEnabled(False)
else:
self.Looping = True
self.llo.setChecked(True)
self.nlo.setEnabled(True)
return True
def f_btns(self, glo):
hlayout = QHBoxLayout()
self.ksbutton = QPushButton('Kill stream', self)
self.ksbutton.setToolTip('Kill whetever been streamed')
self.ksbutton.setFont(self.tf)
self.ksbutton.clicked.connect(self.k_act)
self.sbutton = QPushButton('Stream', self)
self.sbutton.setFont(self.tf)
self.sbutton.setToolTip('Stream a youtube video')
self.fbutton = QPushButton('Factory reset', self)
self.fbutton.setFont(self.tf)
self.fbutton.setToolTip('Factory reset the device')
self.fbutton.clicked.connect(self.fr_act)
self.sbutton.clicked.connect(self.yv_input)
self.ksbutton.setEnabled(False)
self.sbutton.setEnabled(False)
self.fbutton.setEnabled(False)
hlayout.addWidget(self.ksbutton)
hlayout.addWidget(self.sbutton)
hlayout.addWidget(self.fbutton)
glo.addLayout(hlayout)
def ch_dur(th=None, dur=10):
if len(dur) >= 1:
try:
dur = int(dur)
except:
dur = None
if dur is None or not isinstance(dur, int):
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage(
'! Error: wrong duration entery, only integers')
return None
else:
dur = 10
return dur
def k_act(self):
if self.ci_list.count() >= 1:
cr = self.ci_list.currentRow()
if cr is None or cr <= 0:
cr = 0
ip = self.ci_list.item(cr).text()
if self.Looping is not None:
dur = self.ch_dur(self.nlo.text())
if dur is None:
return True
self.P = CC_thread(cancel_app, dur, ip)
self.P.start()
self.P.somesignal.connect(self.handleStatusMessage)
self.P.setTerminationEnabled(True)
self.inloop_state()
return True
else:
ch = cancel_app(ip)
if ch is None:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: failed to kill stream ..')
else:
self.s_bar.setStyleSheet(self.s_norm)
self.s_bar.showMessage('# Stream got killed ')
else:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: No items selected from the list')
self.eout()
return True
def fr_act(self):
if self.ci_list.count() >= 1:
cr = self.ci_list.currentRow()
if cr is None or cr <= 0:
cr = 0
ip = self.ci_list.item(cr).text()
if self.Looping is not None:
dur = self.ch_dur(self.nlo.text())
if dur is None:
return True
self.P = CC_thread(reset_cc, dur, ip)
self.P.start()
self.P.somesignal.connect(self.handleStatusMessage)
self.P.setTerminationEnabled(True)
self.inloop_state()
return True
else:
ch = reset_cc(ip)
if ch is None:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage(
'! Error: failed to factory reset ..')
else:
self.s_bar.setStyleSheet(self.s_norm)
self.s_bar.showMessage('# Got factory reseted ')
else:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: No items selected from the list')
self.eout()
return True
def y_act(self, link=None):
if self.ci_list.count() >= 1:
cr = self.ci_list.currentRow()
if cr is None or cr <= 0:
cr = 0
ip = self.ci_list.item(cr).text()
if self.Looping is not None:
dur = self.ch_dur(self.nlo.text())
if dur is None:
return True
self.P = CC_thread(send_app, dur, ip, link)
self.P.start()
self.P.somesignal.connect(self.handleStatusMessage)
self.P.setTerminationEnabled(True)
self.inloop_state()
return True
else:
ch = reset_cc(ip)
if ch is None:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: failed to stream link ..')
else:
self.s_bar.setStyleSheet(self.s_norm)
self.s_bar.showMessage('# Streamed the link ')
else:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: No items selected from the list')
self.eout()
return True
def yv_input(self):
text, okPressed = QInputDialog.getText(
self, "Stream youtube",
"Enter youtube video link to be streamed :")
if okPressed and text != '':
ntext = text.split('?')
if len(ntext) <= 1:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: Invalid youtube linkd ')
return False
ntext = ntext[1]
if ntext != '' and len(ntext) > 1 and "youtube" in text:
self.y_act(ntext)
return True
else:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage('! Error: Invalid youtube linkd ')
return False
@Slot(object)
def handleStatusMessage(self, message):
self.s_bar.setStyleSheet(self.s_loop)
self.s_bar.showMessage(message)
def il_add(self, items=[]):
if len(items) >= 1:
self.s_bar.setStyleSheet(self.s_norm)
for i in items:
fitem = self.ci_list.findItems(i, Qt.MatchExactly)
if len(fitem) >= 1:
self.s_bar.setStyleSheet(self.s_error)
self.s_bar.showMessage(
'! Error: Device exists in the list')
else:
self.s_bar.setStyleSheet(self.s_norm)
self.ci_list.addItem(i)
self.s_bar.showMessage('# Device was found and added')
if not self.P.isRunning():
self.cle_btn.setEnabled(True)
self.ci_list.setEnabled(True)
self.llo.setEnabled(True)
self.nlo.setEnabled(True)
self.ksbutton.setEnabled(True)
self.sbutton.setEnabled(True)
self.fbutton.setEnabled(True)
else:
self.s_bar.setStyleSheet(self.s_norm)
self.s_bar.showMessage('# Cleard devices list')
self.ci_list.clear()
self.cle_btn.setEnabled(False)
self.ci_list.setEnabled(False)
self.ksbutton.setEnabled(False)
self.llo.setEnabled(False)
self.nlo.setEnabled(False)
self.sbutton.setEnabled(False)
self.fbutton.setEnabled(False)
return True
def inloop_state(self, out=False):
if not out:
self.lbtn.setEnabled(True)
self.cle_btn.setEnabled(False)
self.ci_list.setEnabled(False)
self.ksbutton.setEnabled(False)
self.llo.setEnabled(False)
self.nlo.setEnabled(False)
self.sbutton.setEnabled(False)
self.fbutton.setEnabled(False)
else:
if self.P.isRunning():
self.P.stop()
self.lbtn.setEnabled(False)
self.cle_btn.setEnabled(True)
self.ci_list.setEnabled(True)
self.ksbutton.setEnabled(True)
self.llo.setEnabled(True)
self.nlo.setEnabled(True)
self.sbutton.setEnabled(True)
self.fbutton.setEnabled(True)
self.s_bar.clearMessage()
return True
def ds_bar(self, glo):
self.s_bar = QStatusBar()
self.s_error = "QStatusBar{color:red;font-weight:1000;}"
self.s_loop = "QStatusBar{color:black;font-weight:1000;}"
self.s_norm = "QStatusBar{color:blue;font-style:italic;"
self.s_norm += "font-weight:500;}"
self.s_bar.setStyleSheet(self.s_norm)
glo.addWidget(self.s_bar)
def eout(self):
msgg = "<center>"
msgg += " Opps, a critical error has occurred, we will be "
msgg += " grateful if you can help fixing it, by reporting to us "
msgg += " at : <br><br> "
msgg += "<b><a href='https://github.io/mrf345/chrome-cut'> "
msgg += "https://github.io/mrf345/chrome-cut </a></b> </center>"
mm = QMessageBox.critical(self, "Critical Error", msgg, QMessageBox.Ok)
exit(0)
class ProfileSelection(QDialog):
'''
classdocs
'''
removeProfile = Signal(Athlete_Model)
profileSelected = Signal(Athlete_Model)
profileUpdate_request = Signal(Athlete_Model, Athlete_Model)
lastProfileDeleted = Signal()
def __init__(self, athletesList):
'''
Constructor
'''
QDialog.__init__(self)
self.setWindowTitle("Profile Selection")
self.athletesList = athletesList
self._initGUI()
def _initGUI(self):
topHLayout = QHBoxLayout()
hLayout = QHBoxLayout()
vLayout = QVBoxLayout()
# Label
greeterText = QLabel("Welcome to <b>Pushup app</b>." + \
"<br><br> Select a profile:")
vLayout.addWidget(greeterText)
# List
self.list = QListWidget()
self.list.setMinimumWidth(150)
self.list.setSelectionMode(QAbstractItemView.SingleSelection)
# SingleSelection is the default value, but I prefer to be sure
self.list.itemSelectionChanged.connect(self._activateButtons)
for athlete in self.athletesList:
iconW = QIcon.fromTheme("user-available")
# doens't work on Mac and Windows
# http://qt-project.org/doc/qt-4.8/qicon.html#fromTheme
listW = QListWidgetItem(iconW, athlete._name)
listW.setData(Qt.UserRole, athlete)
self.list.addItem(listW)
topHLayout.addWidget(self.list)
self.profileWidget = ProfileFormWidget()
self.profileWidget.hide()
topHLayout.addWidget(self.profileWidget)
vLayout.addLayout(topHLayout)
vLayout.addLayout(hLayout)
# Buttons
self.okBtn = QPushButton("Ok")
self.okBtn.setDisabled(True)
self.okBtn.setDefault(True)
self.okBtn.clicked.connect(self._okButtonSlot)
self.list.itemDoubleClicked.connect(self._okButtonSlot)
cancelBtn = QPushButton("Cancel")
cancelBtn.clicked.connect(self._cancelButtonSlot)
self.editBtn = QPushButton("Edit")
self.editBtn.setDisabled(True)
self.editBtn.setCheckable(True)
self.editBtn.clicked.connect(self._toggleProfileEdit)
self.saveBtn = QPushButton("Save changes") # Saves the changes made on the profile
self.saveBtn.hide()
self.saveBtn.clicked.connect(self._saveButtonSlot)
self.removeProfileBtn = QPushButton("Remove Profile")
self.removeProfileBtn.setDisabled(True)
self.removeProfileBtn.clicked.connect(self._removeProfile_Dialog)
hLayout.addWidget(self.editBtn)
hLayout.addWidget(self.removeProfileBtn)
hLayout.addWidget(cancelBtn)
hLayout.addWidget(self.okBtn)
hLayout.addWidget(self.saveBtn)
self.setLayout(vLayout)
def getSelectedProfile(self):
selectedListItem = self.list.selectedItems()[0]
athleteProfile = selectedListItem.data(Qt.UserRole)
return athleteProfile
def updateList(self, athletes):
self.list.clear()
self.athletesList = athletes
for athlete in self.athletesList:
iconW = QIcon.fromTheme("user-available")
# doens't work on Mac and Windows
# http://qt-project.org/doc/qt-4.8/qicon.html#fromTheme
listW = QListWidgetItem(iconW, athlete._name)
listW.setData(Qt.UserRole, athlete)
self.list.addItem(listW)
def resetWidget(self):
""" Resets the widget to the initial laoyout.
Should be used only in specific cases
"""
self.editBtn.setChecked(False)
self._toggleProfileEdit()
def _removeProfile_Dialog(self):
"""Runs a prompt dialog.
Ask the user if he really wants to remove the selected profile.
"""
confirmationDialog = QMessageBox()
confirmationDialog.setText("Do you really want to remove the selected profile ?")
confirmationDialog.setInformativeText("Profile deletion can not be undone")
confirmationDialog.setIcon(QMessageBox.Question)
confirmationDialog.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
confirmationDialog.accepted.connect(self._emitRemoveProfile)
ret = confirmationDialog.exec_()
if ret==QMessageBox.Yes:
self._emitRemoveProfile()
def _emitRemoveProfile(self):
athlete = self.getSelectedProfile()
rowToDelete = 0
for index, element in enumerate(self.athletesList):
if element == athlete:
rowToDelete = index
self.list.takeItem(rowToDelete)
self.athletesList.remove(athlete)
self.removeProfile.emit(athlete)
def _okButtonSlot(self):
athlete = self.list.selectedItems()[0].data(Qt.UserRole)
self.accept() # is it correct ? Maybe self.close() is better ?
# Or should I redefine the accept() method ?
#athleteProfile = self.getSelectedProfile()
self.profileSelected.emit(athlete)
def _cancelButtonSlot(self):
if len(self.athletesList) == 0:
self.lastProfileDeleted.emit()
self.reject()
def _saveButtonSlot(self):
selectedProfile = self.getSelectedProfile()
updatedProfile = self.profileWidget.getProfile()
self.profileUpdate_request.emit(selectedProfile, updatedProfile)
#self._toggleProfileEdit()
def _toggleProfileEdit(self):
if self.editBtn.isChecked():
self.profileWidget.setProfile(self.getSelectedProfile())
self.profileWidget.show()
self.saveBtn.show()
self.okBtn.hide()
self.removeProfileBtn.hide()
else:
self.saveBtn.hide()
self.profileWidget.hide()
self.okBtn.show()
self.removeProfileBtn.show()
def _activateButtons(self):
selectedItems = self.list.selectedItems()
if len(selectedItems)!=0 :
self.okBtn.setDisabled(False)
self.removeProfileBtn.setDisabled(False)
self.editBtn.setDisabled(False)
else :
self.okBtn.setDisabled(True)
self.removeProfileBtn.setDisabled(True)
self.editBtn.setDisabled(True)
class UiMain(QMainWindow):
""" The main gui interface, invokes all windows and ties everything
together
"""
def __init__(self):
""" automatically called __init__ function """
super(UiMain, self).__init__()
# initialize all the variables that are going to be defined in the
# future
self.update_dialog = None
self.update_dialog_lbl = None
self.app_select_box = None
self.selector_lbl = None
self.current_playing_lbl = None
self.current_playing = None
self.misc_messages = None
self.start_btn = None
self.output_dir_lbl = None
self.select_output_dir_btn = None
self.output_cur_dir_lbl = None
self.active_items_list = None
self.inactive_items_list = None
self.switch_active_item_button_off = None
self.switch_active_item_button_on = None
self.switch_output_split_btn = None
self.switch_output_split_lbl = None
# initialize the system tray
# self.system_tray = QSystemTrayIcon(self)
# self.system_tray.setIcon(QIcon(resource_path('icon.png')))
# self.system_tray.show()
# self.system_tray.setToolTip('SMG')
# self.system_tray.activated.connect(self.on_systray_activated)
# initialize the main window
self.setObjectName('self')
self.setWindowTitle('SMG - By Azeirah')
self.resize(400, 250)
# Gives the self an icon
self.setWindowIcon(QIcon(resource_path('icon.png')))
# create the tabs
# the tab widget itself
self.tabbed_windows = QTabWidget(self)
self.tabbed_windows.resize(400, 300)
# tab 1, contains the music player selection
self.music_players = QFrame()
# tab 2, contains options
self.options = QFrame()
self.tabbed_windows.addTab(self.music_players, 'Music players')
self.tabbed_windows.addTab(self.options, 'Options')
# initializes the two tabs, with all the code down below
self.tab_music_players()
self.tab_options()
# shows the main window
self.show()
# self.update()
CheckUpdateThread = Thread(target=self.update)
CheckUpdateThread.setName('CheckUpdateThread')
CheckUpdateThread.run()
def closeEvent(self, event):
""" an automatically called function when the program is about to
close.
"""
# Stops all Threads. These would continue to run in the background
# Even if the window was closed.
Main.running = False
# close the ZuneNowPlaying.exe process
if Constants.SUBP:
Constants.SUBP.kill()
def changeEvent(self, event):
# if event.type() == QEvent.WindowStateChange:
# if self.isMinimized():
# event.ignore()
# self.hide()
# self.system_tray.showMessage('Running', 'Running in the
# background.')
# return
super(UiMain, self).changeEvent(event)
def on_systray_activated(self, reason):
if reason == QSystemTrayIcon.DoubleClick:
self.show()
@staticmethod
def toggle_split(event):
# 0 = Qt.Unchecked The item is unchecked.
# 1 = Qt.PartiallyChecked The item is partially checked. Items in
# hierarchical models may be partially checked if some, but not all,
# of
# their children are checked.
# 2 = Qt.Checked The item is checked.
if event == 0:
Constants.OPTIONS['splitText'] = False
elif event == 2:
Constants.OPTIONS['splitText'] = True
def tab_music_players(self):
""" Everything inside the Music players tab gets created here."""
# self.music_players
# Creates the box with all the music players inside of it
self.app_select_box = QComboBox(self.music_players)
self.app_select_box.setGeometry(135, 10, 150, 25)
# Whenever you change the application, it runs the selectnewapp func
self.app_select_box.activated[str].connect(self.select_new_app)
# Creates the label for the selection combobox
self.selector_lbl = QLabel(self.music_players)
self.selector_lbl.setGeometry(10, 10, 150, 25)
self.selector_lbl.setText('Select your music player: ')
# Creates the label for the current playing song (and the current
# playing song label)
self.current_playing_lbl = QLabel(self.music_players)
self.current_playing_lbl.setGeometry(10, 45, 150, 25)
self.current_playing_lbl.setText('Current playing song: ')
self.current_playing = QLabel(self.music_players)
self.current_playing.setGeometry(117, 45, 250, 25)
self.current_playing.setText(Misc.noSongPlaying)
# Creates a label which displays any additional messages
self.misc_messages = QLabel(self.music_players)
self.misc_messages.setGeometry(10, 80, 390, 24)
self.misc_messages.setText(Misc.misc_message())
self.misc_messages.setOpenExternalLinks(True)
# adds all the music players into the combobox
self.app_select_box.addItem(None)
for item in Constants.ACTIVEITEMS:
if item == '__name__' or item == 'active':
continue
self.app_select_box.addItem(item)
# creates the start button
self.start_btn = QPushButton(self.music_players)
self.start_btn.setGeometry(75, 120, 250, 35)
self.start_btn.setText('Start')
# links the start button to the self.start function
QObject.connect(self.start_btn, SIGNAL("clicked()"),
lambda: Thread(target=self.start, name='startbutton').start())
def tab_options(self):
""" Everything inside the Options tab gets created here. """
# self.options
# This section is for selecting output dir
# Creates the output dir label
self.output_dir_lbl = QLabel(self.options)
self.output_dir_lbl.setGeometry(10, 10, 125, 15)
self.output_dir_lbl.setText('Change Output Directory: ')
# Creates the output dir button
self.select_output_dir_btn = QPushButton(self.options)
self.select_output_dir_btn.setGeometry(137, 8, 30, 20)
self.select_output_dir_btn.setText('...')
# Creates the output dir currentdir Lineedit
self.output_cur_dir_lbl = QLineEdit(self.options)
self.output_cur_dir_lbl.setGeometry(170, 6, 210, 25)
self.output_cur_dir_lbl.setReadOnly(True)
self.output_cur_dir_lbl.setText(Constants.CONFIG.
get('directories', 'current_song'))
# when the '...' button is clicked, show a dialog (fire func
# disp_dialog)
QObject.connect(self.select_output_dir_btn, SIGNAL("clicked()"),
self.disp_dialog)
# This section is for selecting what players you use
# The box with all the active players
self.active_items_list = QListWidget(self.options)
self.active_items_list.setGeometry(10, 40, 150, 100)
# The box with all the inactive players
self.inactive_items_list = QListWidget(self.options)
self.inactive_items_list.setGeometry(230, 40, 150, 100)
# Populate the two boxes with active and inactive items
for item in Constants.ACTIVEITEMS:
if item == '__name__' or item == 'active':
continue
self.active_items_list.addItem(item)
for item in Constants.INACTIVEITEMS:
if item == '__name__' or item == 'active':
continue
self.inactive_items_list.addItem(item)
# The buttons responsible for switching
# off button
self.switch_active_item_button_off = QPushButton(self.options)
self.switch_active_item_button_off.setText('->'.decode('utf-8'))
# Makes the -> readable and clear
self.switch_active_item_button_off.setFont(QFont('SansSerif', 17))
self.switch_active_item_button_off.setGeometry(175, 55, 40, 30)
# on button
self.switch_active_item_button_on = QPushButton(self.options)
self.switch_active_item_button_on.setText('<-'.decode('utf-8'))
# makes <- readable and clear
self.switch_active_item_button_on.setFont(QFont('SansSerif', 17))
self.switch_active_item_button_on.setGeometry(175, 90, 40, 30)
QObject.connect(self.switch_active_item_button_on, SIGNAL
("clicked()"), self.switch_item_on)
QObject.connect(self.switch_active_item_button_off, SIGNAL
("clicked()"), self.switch_item_off)
# A button to toggle the split output in half option. It's a temporary
# fix for the Foobar double output problem.
self.switch_output_split_btn = QCheckBox(self.options)
self.switch_output_split_btn.setCheckState(Qt.CheckState.Unchecked)
self.switch_output_split_btn.setGeometry(10, 140, 40, 30)
self.switch_output_split_btn.stateChanged.connect(self.toggle_split)
# The label for the split toggle
self.switch_output_split_lbl = QLabel(self.options)
self.switch_output_split_lbl.setText(
"Split the output text in half (don't use this if you don't need it)")
self.switch_output_split_lbl.setGeometry(30, 140, 300, 30)
def switch_item_on(self):
""" Switches items (musicapps) on """
try:
# If an item from the active box is selected
# Remove it and place it inside the inactive box
item_taken = self.inactive_items_list.takeItem(
self.inactive_items_list.currentRow())
self.active_items_list.addItem(item_taken)
active_items = {}
inactive_items = {}
for i in range(self.active_items_list.count()):
active_items[self.active_items_list.item(i).text()] =\
ITEMS[self.active_items_list.item(i).text()
.encode('utf-8')]
for i in range(self.inactive_items_list.count()):
inactive_items[self.inactive_items_list.item(i).text()] =\
ITEMS[self.inactive_items_list.item(i).text()
.encode('utf-8')]
Constants.ACTIVE_ITEMS = active_items
Constants.INACTIVE_ITEMS = inactive_items
# clear the selection combobox
self.app_select_box.clear()
# Repopulate the combobox
self.app_select_box.addItem(None)
for item in active_items:
self.app_select_box.addItem(item)
Constants.CONFIG.set('active', item_taken.text(),
ITEMS[item_taken.text()])
Constants.CONFIG.remove_option('inactive', item_taken.text())
# Updates the config file to be up to date with activeItems
Constants.CONFIG.update()
except:
raise
def switch_item_off(self):
""" Switches items (musicapps) off """
try:
# If an item from the inactive box is selected.
# Remove it and place it inside the active box
item_taken = self.active_items_list.takeItem(
self.active_items_list.currentRow())
self.inactive_items_list.addItem(item_taken)
# update activeItems
active_items = {}
inactive_items = {}
for i in range(self.active_items_list.count()):
active_items[self.active_items_list.item(i).text()] =\
ITEMS[self.active_items_list.item(i).text()
.encode('utf-8')]
for i in range(self.inactive_items_list.count()):
inactive_items[self.inactive_items_list.item(i).text()] =\
ITEMS[self.inactive_items_list.item(i).text()
.encode('utf-8')]
Constants.ACTIVE_ITEMS = active_items
Constants.INACTIVE_ITEMS = inactive_items
# clear the selection combobox
self.app_select_box.clear()
# Repopulate the combobox
self.app_select_box.addItem(None)
for item in active_items:
self.app_select_box.addItem(item)
# Updates the active items Constants property
Constants.CONFIG.set('inactive', item_taken.text(),
ITEMS[item_taken.text()])
Constants.CONFIG.remove_option('active', item_taken.text())
# Updates the config file to be up to date with activeItems
Constants.CONFIG.update()
except:
raise
def disp_dialog(self):
""" displays the dialog which select a directory for output. """
fname = QFileDialog.getExistingDirectory()
Constants.CONFIG.set('directories', 'current_song', fname)
self.output_cur_dir_lbl.setText(Constants.CONFIG.
get('directories', 'current_song'))
def select_new_app(self, text):
""" Sets the new application to check for """
try:
Main.selectedProgram = ITEMS[text]
except KeyError:
# catches the empty option, it's obviously not in the dict
pass
# custom message for zune
if Main.selectedProgram == 'zune':
self.misc_messages.setText(Misc.ZuneNotification)
# custom message for webplayers which require the groovemarklet
elif text.find('*'):
self.misc_messages.setText(Misc.GetGroovemarklet)
def start(self):
""" When the start button is pressed, start the main program loop """
if Main.selectedProgram:
if not Main.running:
self.start_btn.setText('Stop')
Main.running = True
try:
pythoncom.CoInitializeEx(pythoncom.COINIT_MULTITHREADED)
except pythoncom.com_error:
# already initialized.
pass
thread = Thread(
target=Main.enumWindows, name='enumWindows')
thread.run()
else:
self.start_btn.setText('Start')
Main.running = False
self.set_playing(Misc.noSongPlaying)
Wr.write('')
def set_playing(self, title=''):
""" Sets the text of the label of what song is playing """
# print 'setting title: ', title
self.current_playing.setText(title)
