class HashmalMain(QMainWindow):
# Signals
# Emitted when the list of user's layouts changes.
layoutsChanged = QtCore.pyqtSignal()
def __init__(self, app):
super(HashmalMain, self).__init__()
self.app = app
self.app.setStyleSheet(hashmal_style())
self.changes_saved = True
# {Qt.DockWidgetArea: [dock0, dock1, ...], ...}
self.dock_orders = defaultdict(list)
self.setCorner(QtCore.Qt.BottomRightCorner, QtCore.Qt.RightDockWidgetArea)
self.config = Config()
self.init_logger()
self.config.optionChanged.connect(self.on_option_changed)
QtCore.QCoreApplication.setOrganizationName('mazaclub')
QtCore.QCoreApplication.setApplicationName('hashmal')
self.qt_settings = QtCore.QSettings()
active_params = self.config.get_option('chainparams', 'Bitcoin')
# True if chainparams needs to be set after plugins load.
needs_params_change = False
# An exception is thrown if the last-active chainparams preset
# only exists due to a plugin that defines it.
try:
chainparams.set_to_preset(active_params)
except KeyError:
chainparams.set_to_preset('Bitcoin')
needs_params_change = True
self.download_controller = DownloadController()
self.setDockNestingEnabled(True)
# Plugin Handler loads plugins and handles their dock widgets.
self.plugin_handler = PluginHandler(self)
self.plugin_handler.load_plugins()
self.plugin_handler.do_default_layout()
# Attempt to load chainparams preset again if necessary.
if needs_params_change:
try:
chainparams.set_to_preset(active_params)
except KeyError:
self.log_message('Core', 'Chainparams preset "%s" does not exist. Setting chainparams to Bitcoin.', logging.ERROR)
self.config.set_option('chainparams', 'Bitcoin')
# Filename of script being edited.
self.filename = ''
# The last text that we saved.
self.last_saved = ''
self.create_script_editor()
# Set up script editor font.
script_font = self.qt_settings.value('editor/font', defaultValue=QtCore.QVariant('default')).toString()
if script_font == 'default':
font = monospace_font
else:
font = QFont()
font.fromString(script_font)
self.script_editor.setFont(font)
self.settings_dialog = SettingsDialog(self)
self.create_menubar()
self.create_toolbar()
self.create_actions()
self.new_script()
self.statusBar().setVisible(True)
self.statusBar().messageChanged.connect(self.change_status_bar)
self.restoreState(self.qt_settings.value('toolLayout/default/state').toByteArray())
self.restoreGeometry(self.qt_settings.value('toolLayout/default/geometry').toByteArray())
self.script_editor.setFocus()
if self.qt_settings.value('quickTipsOnStart', defaultValue=QtCore.QVariant(True)).toBool():
QtCore.QTimer.singleShot(500, self.do_quick_tips)
def sizeHint(self):
return QtCore.QSize(800, 500)
def init_logger(self):
"""Initialize logger."""
handler = logging.StreamHandler()
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s', '%Y-%m-%d %H:%M:%S')
handler.setFormatter(formatter)
logger = logging.getLogger()
logger.addHandler(handler)
self.change_log_level(self.config.get_option('log_level', 'INFO'))
def change_log_level(self, level_str):
level_str = level_str.upper()
if level_str not in ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']:
level_str = 'INFO'
level = getattr(logging, level_str)
logging.getLogger().setLevel(level)
def create_menubar(self):
menubar = QMenuBar()
file_menu = menubar.addMenu('&File')
file_menu.addAction('&New', self.new_script).setShortcut(QKeySequence.New)
file_menu.addAction('Save As...', self.save_script_as).setShortcut(QKeySequence.SaveAs)
file_menu.addAction('&Open', self.open_script).setShortcut(QKeySequence.Open)
file_menu.addAction('&Save', self.save_script).setShortcut(QKeySequence.Save)
file_menu.addAction('&Quit', self.close)
# Script actions
script_menu = menubar.addMenu('&Script')
script_menu.addAction('&Evaluate', self.plugin_handler.evaluate_current_script)
script_menu.addAction('&Copy Hex', self.script_editor.copy_hex)
# Settings and tool toggling
tools_menu = menubar.addMenu('&Tools')
tools_menu.addAction('&Settings', self.show_settings_dialog)
tools_menu.addAction('&Plugin Manager', lambda: PluginManager(self).exec_())
tools_menu.addSeparator()
self.plugin_handler.create_menu(tools_menu)
help_menu = menubar.addMenu('&Help')
help_menu.addAction('&About', self.do_about)
help_menu.addAction('&Quick Tips', self.do_quick_tips)
self.setMenuBar(menubar)
def show_settings_dialog(self):
self.settings_dialog.show()
def show_status_message(self, msg, error=False):
self.statusBar().showMessage(msg, 3000)
if error:
self.statusBar().setProperty('hasError', True)
else:
self.statusBar().setProperty('hasError', False)
self.style().polish(self.statusBar())
def log_message(self, plugin_name, msg, level):
message = '[%s] -> %s' % (plugin_name, msg)
logging.log(level, message)
self.show_status_message(message, True if level == logging.ERROR else False)
log_plugin = self.plugin_handler.get_plugin('Log')
if log_plugin:
log_plugin.ui.add_log_message(time.time(), level, plugin_name, msg)
def change_status_bar(self, new_msg):
# Unset hasError if an error is removed.
if not new_msg and self.statusBar().property('hasError'):
self.statusBar().setProperty('hasError', False)
self.style().polish(self.statusBar())
def on_text_changed(self):
s = str(self.script_editor.toPlainText())
saved = False
if s == self.last_saved and self.filename:
saved = True
title = ''.join(['Hashmal - ', self.filename])
if not saved:
title = ''.join([title, ' *'])
self.setWindowTitle(title)
self.changes_saved = saved
def closeEvent(self, event):
# Save layout if configured to.
if self.qt_settings.value('saveLayoutOnExit', defaultValue=QtCore.QVariant(False)).toBool():
self.qt_settings.setValue('toolLayout/default', self.saveState())
if self.close_script():
logging.shutdown()
event.accept()
else:
event.ignore()
def close_script(self):
# Confirm discarding changes if an unsaved file is open.
if str(self.script_editor.toPlainText()) and not self.changes_saved:
msgbox = QMessageBox(self)
msgbox.setWindowTitle('Hashmal - Save Changes')
text = 'Do you want to save this script before closing?'
if self.filename:
text = 'Do you want to save your changes to ' + self.filename + ' before closing?'
msgbox.setText(text)
msgbox.setStandardButtons(QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel)
msgbox.setDefaultButton(QMessageBox.Save)
msgbox.setIcon(QMessageBox.Question)
result = msgbox.exec_()
if result == QMessageBox.Save:
self.save_script()
elif result == QMessageBox.Cancel:
return False
self.filename = ''
self.changes_saved = True
self.script_editor.clear()
return True
def new_script(self, filename=''):
if not self.close_script():
return
if not filename:
base_name = ''.join(['Untitled-', str(time.time()), '.coinscript'])
filename = os.path.expanduser(base_name)
self.load_script(filename)
def save_script(self):
filename = self.filename
if not filename:
filename = str(QFileDialog.getSaveFileName(self, 'Save script', filter=script_file_filter))
if not filename: return
if not filename.endswith('.coinscript'):
filename += '.coinscript'
self.filename = filename
with open(self.filename, 'w') as file:
file.write(str(self.script_editor.toPlainText()))
self.last_saved = str(self.script_editor.toPlainText())
self.on_text_changed()
def save_script_as(self):
filename = str(QFileDialog.getSaveFileName(self, 'Save script as', filter=script_file_filter))
if not filename: return
if not filename.endswith('.coinscript'):
filename += '.coinscript'
self.filename = filename
self.save_script()
def open_script(self):
filename = str(QFileDialog.getOpenFileName(self, 'Open script', '.', filter=script_file_filter))
if not filename:
return
if self.close_script():
self.load_script(filename)
def load_script(self, filename):
if os.path.exists(filename):
self.filename = filename
with open(self.filename,'r') as file:
self.script_editor.setPlainText(file.read())
else:
self.script_editor.clear()
self.last_saved = str(self.script_editor.toPlainText())
self.on_text_changed()
def create_script_editor(self):
vbox = QVBoxLayout()
self.format_combo = QComboBox()
self.format_combo.setWhatsThis('Use this to change the format that script editor displays and writes scripts in.')
self.format_combo.addItems(known_script_formats)
self.script_editor = ScriptEditor(self)
self.script_editor.textChanged.connect(self.on_text_changed)
self.script_editor.setWhatsThis('The script editor lets you write transaction scripts in a human-readable format. You can also write and edit scripts in their raw, hex-encoded format if you prefer.')
self.format_combo.currentIndexChanged.connect(lambda index: self.script_editor.set_format(known_script_formats[index]))
hbox = QHBoxLayout()
hbox.addWidget(QLabel('Format: '))
hbox.addWidget(self.format_combo)
hbox.addStretch(1)
vbox.addLayout(hbox)
vbox.addWidget(self.script_editor)
w = QWidget()
w.setLayout(vbox)
self.setCentralWidget(w)
def create_toolbar(self):
toolbar = ToolBar(self, 'Toolbar')
self.addToolBar(toolbar)
def create_actions(self):
hide_dock = QAction('Hide Dock', self)
hide_dock.setShortcut(QKeySequence(QKeySequence.Close))
hide_dock.triggered.connect(self.hide_current_dock)
self.addAction(hide_dock)
move_left_dock = QAction('Move Left', self)
move_left_dock.setShortcut(QKeySequence(QKeySequence.Back))
move_left_dock.triggered.connect(lambda: self.move_one_dock(reverse=True))
self.addAction(move_left_dock)
move_right_dock = QAction('Move Right', self)
move_right_dock.setShortcut(QKeySequence(QKeySequence.Forward))
move_right_dock.triggered.connect(self.move_one_dock)
self.addAction(move_right_dock)
def move_one_dock(self, reverse=False):
"""Move focus to the next or previous dock."""
w = get_active_dock()
if not w: return
docks = filter(lambda dock: dock.isVisible(), self.dock_orders[self.dockWidgetArea(w)])
index = docks.index(w)
if reverse:
if index == 0:
index = len(docks)
docks[index - 1].needsFocus.emit()
else:
if index >= len(docks) - 1:
index = -1
docks[index + 1].needsFocus.emit()
def tabifyDockWidget(self, bottom, top):
"""Overloaded method for purposes of remembering dock positions."""
docks = self.dock_orders[self.dockWidgetArea(bottom)]
area = self.dockWidgetArea(bottom)
if len(docks) == 0:
docks.append(bottom)
super(HashmalMain, self).tabifyDockWidget(bottom, top)
idx = docks.index(bottom)
docks.insert(idx + 1, top)
def createPopupMenu(self):
menu = QMenu(self)
plugins_menu = menu.addMenu('All Plugins')
for p in sorted(self.plugin_handler.loaded_plugins, key = lambda x: x.name):
if not p.has_gui:
continue
plugins_menu.addAction(p.ui.toggleViewAction())
return menu
def do_about(self):
d = QDialog(self)
vbox = QVBoxLayout()
about_label = QLabel()
about_label.setWordWrap(True)
txt = []
txt.append(' '.join([
'Hashmal is an IDE for Bitcoin transaction scripts and a general cryptocurrency development toolbox.',
'Its primary purpose is to make it easier to write, evaluate, and learn about transaction scripts.',
]))
txt.append('Hashmal is intended for cryptocurrency developers and power users.')
txt = '\n\n'.join(txt)
about_label.setText(txt)
close_button = QPushButton('Close')
close_button.clicked.connect(d.close)
btn_box = floated_buttons([close_button])
vbox.addWidget(about_label)
vbox.addLayout(btn_box)
d.setLayout(vbox)
d.setWindowTitle('About Hashmal')
d.exec_()
def do_quick_tips(self):
QuickTips(self).exec_()
def hide_current_dock(self):
w = get_active_dock()
if not w: return
docks = filter(lambda dock: dock.isVisible(), self.tabifiedDockWidgets(w))
w.toggleViewAction().trigger()
if docks:
docks[0].needsFocus.emit()
def on_option_changed(self, key):
if key == 'log_level':
self.change_log_level(self.config.get_option('log_level', 'INFO'))
class HashmalMain(QMainWindow):
# Signals
# Emitted when the list of user's layouts changes.
layoutsChanged = QtCore.pyqtSignal()
def __init__(self, app):
super(HashmalMain, self).__init__()
self.app = app
self.app.setStyleSheet(hashmal_style())
self.changes_saved = True
# {Qt.DockWidgetArea: [dock0, dock1, ...], ...}
self.dock_orders = defaultdict(list)
self.setCorner(QtCore.Qt.BottomRightCorner,
QtCore.Qt.RightDockWidgetArea)
self.config = Config()
self.init_logger()
self.config.optionChanged.connect(self.on_option_changed)
QtCore.QCoreApplication.setOrganizationName('mazaclub')
QtCore.QCoreApplication.setApplicationName('hashmal')
self.qt_settings = QtCore.QSettings()
active_params = self.config.get_option('chainparams', 'Bitcoin')
# True if chainparams needs to be set after plugins load.
needs_params_change = False
# An exception is thrown if the last-active chainparams preset
# only exists due to a plugin that defines it.
try:
chainparams.set_to_preset(active_params)
except KeyError:
chainparams.set_to_preset('Bitcoin')
needs_params_change = True
self.download_controller = DownloadController()
self.setDockNestingEnabled(True)
# Plugin Handler loads plugins and handles their dock widgets.
self.plugin_handler = PluginHandler(self)
self.plugin_handler.load_plugins()
self.plugin_handler.do_default_layout()
# Attempt to load chainparams preset again if necessary.
if needs_params_change:
try:
chainparams.set_to_preset(active_params)
except KeyError:
self.log_message(
'Core',
'Chainparams preset "%s" does not exist. Setting chainparams to Bitcoin.',
logging.ERROR)
self.config.set_option('chainparams', 'Bitcoin')
# Filename of script being edited.
self.filename = ''
# The last text that we saved.
self.last_saved = ''
self.create_script_editor()
# Set up script editor font.
script_font = self.qt_settings.value(
'editor/font', defaultValue=QtCore.QVariant('default')).toString()
if script_font == 'default':
font = monospace_font
else:
font = QFont()
font.fromString(script_font)
self.script_editor.setFont(font)
self.settings_dialog = SettingsDialog(self)
self.create_menubar()
self.create_toolbar()
self.create_actions()
self.new_script()
self.statusBar().setVisible(True)
self.statusBar().messageChanged.connect(self.change_status_bar)
self.restoreState(
self.qt_settings.value('toolLayout/default/state').toByteArray())
self.restoreGeometry(
self.qt_settings.value(
'toolLayout/default/geometry').toByteArray())
self.script_editor.setFocus()
if self.qt_settings.value('quickTipsOnStart',
defaultValue=QtCore.QVariant(True)).toBool():
QtCore.QTimer.singleShot(500, self.do_quick_tips)
def sizeHint(self):
return QtCore.QSize(800, 500)
def init_logger(self):
"""Initialize logger."""
handler = logging.StreamHandler()
formatter = logging.Formatter(
'%(asctime)s - %(levelname)s - %(message)s', '%Y-%m-%d %H:%M:%S')
handler.setFormatter(formatter)
logger = logging.getLogger()
logger.addHandler(handler)
self.change_log_level(self.config.get_option('log_level', 'INFO'))
def change_log_level(self, level_str):
level_str = level_str.upper()
if level_str not in ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']:
level_str = 'INFO'
level = getattr(logging, level_str)
logging.getLogger().setLevel(level)
def create_menubar(self):
menubar = QMenuBar()
file_menu = menubar.addMenu('&File')
file_menu.addAction('&New',
self.new_script).setShortcut(QKeySequence.New)
file_menu.addAction('Save As...', self.save_script_as).setShortcut(
QKeySequence.SaveAs)
file_menu.addAction('&Open',
self.open_script).setShortcut(QKeySequence.Open)
file_menu.addAction('&Save',
self.save_script).setShortcut(QKeySequence.Save)
file_menu.addAction('&Quit', self.close)
# Script actions
script_menu = menubar.addMenu('&Script')
script_menu.addAction('&Evaluate',
self.plugin_handler.evaluate_current_script)
script_menu.addAction('&Copy Hex', self.script_editor.copy_hex)
# Settings and tool toggling
tools_menu = menubar.addMenu('&Tools')
tools_menu.addAction('&Settings', self.show_settings_dialog)
tools_menu.addAction('&Plugin Manager',
lambda: PluginManager(self).exec_())
tools_menu.addSeparator()
self.plugin_handler.create_menu(tools_menu)
help_menu = menubar.addMenu('&Help')
help_menu.addAction('&About', self.do_about)
help_menu.addAction('&Quick Tips', self.do_quick_tips)
self.setMenuBar(menubar)
def show_settings_dialog(self):
self.settings_dialog.show()
def show_status_message(self, msg, error=False):
self.statusBar().showMessage(msg, 3000)
if error:
self.statusBar().setProperty('hasError', True)
else:
self.statusBar().setProperty('hasError', False)
self.style().polish(self.statusBar())
def log_message(self, plugin_name, msg, level):
message = '[%s] -> %s' % (plugin_name, msg)
logging.log(level, message)
self.show_status_message(message,
True if level == logging.ERROR else False)
log_plugin = self.plugin_handler.get_plugin('Log')
if log_plugin:
log_plugin.ui.add_log_message(time.time(), level, plugin_name, msg)
def change_status_bar(self, new_msg):
# Unset hasError if an error is removed.
if not new_msg and self.statusBar().property('hasError'):
self.statusBar().setProperty('hasError', False)
self.style().polish(self.statusBar())
def on_text_changed(self):
s = str(self.script_editor.toPlainText())
saved = False
if s == self.last_saved and self.filename:
saved = True
title = ''.join(['Hashmal - ', self.filename])
if not saved:
title = ''.join([title, ' *'])
self.setWindowTitle(title)
self.changes_saved = saved
def closeEvent(self, event):
# Save layout if configured to.
if self.qt_settings.value(
'saveLayoutOnExit',
defaultValue=QtCore.QVariant(False)).toBool():
self.qt_settings.setValue('toolLayout/default', self.saveState())
if self.close_script():
logging.shutdown()
event.accept()
else:
event.ignore()
def close_script(self):
# Confirm discarding changes if an unsaved file is open.
if str(self.script_editor.toPlainText()) and not self.changes_saved:
msgbox = QMessageBox(self)
msgbox.setWindowTitle('Hashmal - Save Changes')
text = 'Do you want to save this script before closing?'
if self.filename:
text = 'Do you want to save your changes to ' + self.filename + ' before closing?'
msgbox.setText(text)
msgbox.setStandardButtons(QMessageBox.Save | QMessageBox.Discard
| QMessageBox.Cancel)
msgbox.setDefaultButton(QMessageBox.Save)
msgbox.setIcon(QMessageBox.Question)
result = msgbox.exec_()
if result == QMessageBox.Save:
self.save_script()
elif result == QMessageBox.Cancel:
return False
self.filename = ''
self.changes_saved = True
self.script_editor.clear()
return True
def new_script(self, filename=''):
if not self.close_script():
return
if not filename:
base_name = ''.join(['Untitled-', str(time.time()), '.coinscript'])
filename = os.path.expanduser(base_name)
self.load_script(filename)
def save_script(self):
filename = self.filename
if not filename:
filename = str(
QFileDialog.getSaveFileName(self,
'Save script',
filter=script_file_filter))
if not filename: return
if not filename.endswith('.coinscript'):
filename += '.coinscript'
self.filename = filename
with open(self.filename, 'w') as file:
file.write(str(self.script_editor.toPlainText()))
self.last_saved = str(self.script_editor.toPlainText())
self.on_text_changed()
def save_script_as(self):
filename = str(
QFileDialog.getSaveFileName(self,
'Save script as',
filter=script_file_filter))
if not filename: return
if not filename.endswith('.coinscript'):
filename += '.coinscript'
self.filename = filename
self.save_script()
def open_script(self):
filename = str(
QFileDialog.getOpenFileName(self,
'Open script',
'.',
filter=script_file_filter))
if not filename:
return
if self.close_script():
self.load_script(filename)
def load_script(self, filename):
if os.path.exists(filename):
self.filename = filename
with open(self.filename, 'r') as file:
self.script_editor.setPlainText(file.read())
else:
self.script_editor.clear()
self.last_saved = str(self.script_editor.toPlainText())
self.on_text_changed()
def create_script_editor(self):
vbox = QVBoxLayout()
self.format_combo = QComboBox()
self.format_combo.setWhatsThis(
'Use this to change the format that script editor displays and writes scripts in.'
)
self.format_combo.addItems(known_script_formats)
self.script_editor = ScriptEditor(self)
self.script_editor.textChanged.connect(self.on_text_changed)
self.script_editor.setWhatsThis(
'The script editor lets you write transaction scripts in a human-readable format. You can also write and edit scripts in their raw, hex-encoded format if you prefer.'
)
self.format_combo.currentIndexChanged.connect(
lambda index: self.script_editor.set_format(known_script_formats[
index]))
hbox = QHBoxLayout()
hbox.addWidget(QLabel('Format: '))
hbox.addWidget(self.format_combo)
hbox.addStretch(1)
vbox.addLayout(hbox)
vbox.addWidget(self.script_editor)
w = QWidget()
w.setLayout(vbox)
self.setCentralWidget(w)
def create_toolbar(self):
toolbar = ToolBar(self, 'Toolbar')
self.addToolBar(toolbar)
def create_actions(self):
hide_dock = QAction('Hide Dock', self)
hide_dock.setShortcut(QKeySequence(QKeySequence.Close))
hide_dock.triggered.connect(self.hide_current_dock)
self.addAction(hide_dock)
move_left_dock = QAction('Move Left', self)
move_left_dock.setShortcut(QKeySequence(QKeySequence.Back))
move_left_dock.triggered.connect(
lambda: self.move_one_dock(reverse=True))
self.addAction(move_left_dock)
move_right_dock = QAction('Move Right', self)
move_right_dock.setShortcut(QKeySequence(QKeySequence.Forward))
move_right_dock.triggered.connect(self.move_one_dock)
self.addAction(move_right_dock)
def move_one_dock(self, reverse=False):
"""Move focus to the next or previous dock."""
w = get_active_dock()
if not w: return
docks = filter(lambda dock: dock.isVisible(),
self.dock_orders[self.dockWidgetArea(w)])
index = docks.index(w)
if reverse:
if index == 0:
index = len(docks)
docks[index - 1].needsFocus.emit()
else:
if index >= len(docks) - 1:
index = -1
docks[index + 1].needsFocus.emit()
def tabifyDockWidget(self, bottom, top):
"""Overloaded method for purposes of remembering dock positions."""
docks = self.dock_orders[self.dockWidgetArea(bottom)]
area = self.dockWidgetArea(bottom)
if len(docks) == 0:
docks.append(bottom)
super(HashmalMain, self).tabifyDockWidget(bottom, top)
idx = docks.index(bottom)
docks.insert(idx + 1, top)
def createPopupMenu(self):
menu = QMenu(self)
plugins_menu = menu.addMenu('All Plugins')
for p in sorted(self.plugin_handler.loaded_plugins,
key=lambda x: x.name):
if not p.has_gui:
continue
plugins_menu.addAction(p.ui.toggleViewAction())
return menu
def do_about(self):
d = QDialog(self)
vbox = QVBoxLayout()
about_label = QLabel()
about_label.setWordWrap(True)
txt = []
txt.append(' '.join([
'Hashmal is an IDE for Bitcoin transaction scripts and a general cryptocurrency development toolbox.',
'Its primary purpose is to make it easier to write, evaluate, and learn about transaction scripts.',
]))
txt.append(
'Hashmal is intended for cryptocurrency developers and power users.'
)
txt = '\n\n'.join(txt)
about_label.setText(txt)
close_button = QPushButton('Close')
close_button.clicked.connect(d.close)
btn_box = floated_buttons([close_button])
vbox.addWidget(about_label)
vbox.addLayout(btn_box)
d.setLayout(vbox)
d.setWindowTitle('About Hashmal')
d.exec_()
def do_quick_tips(self):
QuickTips(self).exec_()
def hide_current_dock(self):
w = get_active_dock()
if not w: return
docks = filter(lambda dock: dock.isVisible(),
self.tabifiedDockWidgets(w))
w.toggleViewAction().trigger()
if docks:
docks[0].needsFocus.emit()
def on_option_changed(self, key):
if key == 'log_level':
self.change_log_level(self.config.get_option('log_level', 'INFO'))
class AdaplinControl:
def __init__(self, iface):
# Save reference to the QGIS interface
self.iface = iface
self.canvas = self.iface.mapCanvas()
# Create dialog form
self.dlg = AdaplinDialog()
# Create settings dialog
self.settingsDialog = SettingsDialog()
def initGui(self):
mc = self.canvas
layer = mc.currentLayer()
# Create action for the plugin icon and the plugin menu
self.action = QAction(QIcon(":/plugins/AdaplinTool/icon.png"),
"Adaplin", self.iface.mainWindow())
# Button starts disabled and unchecked
self.action.setEnabled(False)
self.action.setCheckable(True)
self.action.setChecked(False)
# Add Settings to the [Vector] Menu
self.settingsAction = QAction("Settings", self.iface.mainWindow())
#self.iface.addPluginToVectorMenu("&Adaplin Settings", self.settingsAction)
self.iface.addPluginToMenu("&Adaplin", self.settingsAction)
self.settingsAction.triggered.connect(self.openSettings)
# Add the plugin to Plugin Menu and the Plugin Icon to the Toolbar
self.iface.addPluginToMenu("&Adaplin", self.action)
self.iface.addToolBarIcon(self.action)
# Connect signals for button behaviour (map layer change, run the tool and change of QGIS map tool set)
self.iface.currentLayerChanged['QgsMapLayer*'].connect(self.toggle)
self.action.triggered.connect(self.run)
QObject.connect(mc, SIGNAL("mapToolSet(QgsMapTool*)"), self.deactivate)
# Connect the change of the Raster Layer in the ComboBox to function trata_combo
self.dlg.comboBox.currentIndexChanged.connect(self.trata_combo)
def toggle(self):
# Get current layer
mc = self.canvas
layer = mc.currentLayer()
#print "Adaplin layer = ", layer.name()
# In case the current layer is None we do nothing
if layer is None:
return
#print 'raster type = ', layer.type()
#print 'e vector = ', layer.type() == layer.VectorLayer
# This is to decide when the plugin button is enabled or disabled
# The layer must be a Vector Layer
if layer.type() == layer.VectorLayer:
# We only care about the Line and Polygon layers
if layer.geometryType() == QGis.Line or layer.geometryType(
) == QGis.Polygon:
# First we disconnect all possible previous signals associated with this layer
# If layer is editable, SIGNAL "editingStopped()" is connected to toggle
# If it is not editable, SIGNAL "editingStarted()" is connected to toggle
try:
layer.editingStarted.disconnect(self.toggle)
except:
pass
try:
layer.editingStopped.disconnect(self.toggle)
except:
pass
# If current layer is editable, the button is enabled
if layer.isEditable():
self.action.setEnabled(True)
# If we stop editing, we run toggle function to disable the button
layer.editingStopped.connect(self.toggle)
# Layer is not editable
else:
self.action.setEnabled(False)
# In case we start editing, we run toggle function to enable the button
layer.editingStarted.connect(self.toggle)
else:
self.action.setEnabled(False)
else:
self.action.setEnabled(False)
def deactivate(self):
self.action.setChecked(False)
def unload(self):
# Removes item from Plugin Menu, Vector Menu and removes the toolbar icon
self.iface.removePluginMenu("&Adaplin", self.action)
self.iface.removePluginMenu("&Adaplin", self.settingsAction)
self.iface.removeToolBarIcon(self.action)
def trata_combo(self):
# ComboBox Selected Raster Layer
indiceCamada = self.dlg.comboBox.currentIndex()
camadaSelecionada = self.camadas_raster[indiceCamada]
# Clear ComboBoxs of Bands
self.dlg.comboBox_2.clear()
self.dlg.comboBox_3.clear()
self.dlg.comboBox_4.clear()
# Get number of raster bands
numBandas = camadaSelecionada.bandCount()
# List image bands by numbers and add them to Bands ComboBoxs
lista_bandas = [str(b) for b in range(1, numBandas + 1)]
self.dlg.comboBox_2.addItems(lista_bandas)
self.dlg.comboBox_3.addItems(lista_bandas)
self.dlg.comboBox_4.addItems(lista_bandas)
def run(self):
# Unpress the button if it is pressed
if not self.action.isChecked():
print self.action.isChecked()
self.action.setChecked(False)
return
# On-the-fly SRS must be Projected
mapCanvasSrs = self.iface.mapCanvas().mapRenderer().destinationCrs()
if mapCanvasSrs.geographicFlag():
QMessageBox.information(
self.iface.mainWindow(), 'Error',
'<h2> Please choose an On-the-Fly Projected Coordinate System</h2>'
)
return
# Clear ComboBox of Raster Selection.
# This command generate a bug sometimes (for example, when using the plugin with a raster, removing this raster and adding it again)
# when trying to clear the comboBox when it is triggered with trata_combo
# Disconnect and connect the signal is possible but not elegant
self.dlg.comboBox.clear()
# List rasters of Legend Interface
self.camadas_raster = []
camadas = self.iface.legendInterface().layers()
for camada in camadas:
if camada.type() == QgsMapLayer.RasterLayer:
self.camadas_raster.append(camada)
self.dlg.comboBox.addItem(camada.name())
# Error in case there are no raster layers in the Legend Interface
if not self.camadas_raster:
QMessageBox.information(
self.iface.mainWindow(), 'Error',
'<h2>There are no raster layers in the Legend Interace</h2>')
return
# Finish the dialog box and run the dialog event loop
self.dlg.show()
result = self.dlg.exec_()
# See if OK was pressed
if result:
# Get Raster Selected from ComboBox
indiceCamada = self.dlg.comboBox.currentIndex()
camadaSelecionada = self.camadas_raster[indiceCamada]
# Get Bands selected for this Raster
numBandas = camadaSelecionada.bandCount()
lista_bandas = [str(b) for b in range(1, numBandas + 1)]
indiceCombo2 = self.dlg.comboBox_2.currentIndex()
indiceCombo3 = self.dlg.comboBox_3.currentIndex()
indiceCombo4 = self.dlg.comboBox_4.currentIndex()
bandas_selecao = (lista_bandas[indiceCombo2],
lista_bandas[indiceCombo3],
lista_bandas[indiceCombo4])
# Activate our tool if OK is pressed
self.adaplin = Adaplin(self.iface, camadaSelecionada,
bandas_selecao)
mc = self.canvas
layer = mc.currentLayer()
mc.setMapTool(self.adaplin)
self.action.setChecked(True)
else:
self.action.setChecked(False)
def openSettings(self):
# Default settings to reload
def valoresPadrao():
self.settingsDialog.doubleSpinBox.setValue(DEFAULT_ESPACAMENTO)
self.settingsDialog.spinBox.setValue(DEFAULT_QPONTOS)
# Connect button to function that reload default values
self.settingsDialog.pushButton.clicked.connect(valoresPadrao)
self.settingsDialog.show()
result = self.settingsDialog.exec_()
# If OK is pressed we update the settings
if result:
QSettings().setValue(SETTINGS_NAME + "/qpontos",
self.settingsDialog.spinBox.value())
QSettings().setValue(SETTINGS_NAME + "/espacamento",
self.settingsDialog.doubleSpinBox.value())
# Disconnect signal connected previously
self.settingsDialog.pushButton.clicked.disconnect(valoresPadrao)
def action_settings(self):
sett_dialog = SettingsDialog()
sett_dialog.show()
sett_dialog.exec_()
self.reinit_tree()
def settings(self):
sett_dialog = SettingsDialog()
sett_dialog.show()
result = sett_dialog.exec_()
def do_change_settings(self):
dlg_settings = SettingsDialog()
dlg_settings.show()
dlg_settings.projectsDatabaseHasChanged.connect(
self.do_load_projects_database)
result = dlg_settings.exec_()
logo = QPixmap('./ui/png/tasklogo.png')
app.setWindowIcon(QIcon(logo))
style = qdarkstyle.load_stylesheet()
app.setStyleSheet(style)
try:
lbl = _show_splash_screen()
istgutgelaufen = True
app.loadConfig()
app.connectServer()
except Exception as ex:
istgutgelaufen = False
QMessageBox.critical(None, "ma", str(ex))
finally:
lbl.hide()
if istgutgelaufen:
app.mainWindow = MainWindow()
app.mainWindow.show()
app.exec_()
app.closeConnection()
else:
# wenns nicht gut geht, dialog auftun, dass man connection aendern kann
dlg = SettingsDialog(app.settings)
dlg.show()
app.exec_()
if dlg.result() == SettingsDialog.Accepted:
app.settings = dlg.getData()
app.saveConfig()
def action_settings(self):
sett_dialog = SettingsDialog()
sett_dialog.show()
sett_dialog.exec_()
self.reinit_tree()