1 files changed, 214 insertions, 0 deletions
diff --git a/silx/gui/console.py b/silx/gui/console.py
new file mode 100644
index 0000000..13760b4
--- /dev/null
+++ b/silx/gui/console.py
@@ -0,0 +1,214 @@
+# coding: utf-8
+# /*##########################################################################
+#
+# Copyright (c) 2004-2017 European Synchrotron Radiation Facility
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+# ###########################################################################*/
+"""This module provides an IPython console widget.
+
+You can push variables - any python object - to the
+console's interactive namespace. This provides users with an advanced way
+of interacting with your program. For instance, if your program has a
+:class:`PlotWidget` or a :class:`PlotWindow`, you can push a reference to
+these widgets to allow your users to add curves, save data to files… by using
+the widgets' methods from the console.
+
+.. note::
+
+    This module has a dependency on
+    `IPython <https://pypi.python.org/pypi/ipython>`_ and
+    `qtconsole <https://pypi.python.org/pypi/qtconsole>`_ (or *ipython.qt* for
+    older versions of *IPython*). An ``ImportError`` will be raised if it is
+    imported while the dependencies are not satisfied.
+
+Basic usage example::
+
+    from silx.gui import qt
+    from silx.gui.console import IPythonWidget
+
+    app = qt.QApplication([])
+
+    hello_button = qt.QPushButton("Hello World!", None)
+    hello_button.show()
+
+    console = IPythonWidget()
+    console.show()
+    console.pushVariables({"the_button": hello_button})
+
+    app.exec_()
+
+This program will display a console widget and a push button in two separate
+windows. You will be able to interact with the button from the console,
+for example change its text::
+
+    >>> the_button.setText("Spam spam")
+
+An IPython interactive console is a powerful tool that enables you to work
+with data and plot it.
+See `this tutorial <https://plot.ly/python/ipython-notebook-tutorial/>`_
+for more information on some of the rich features of IPython.
+"""
+__authors__ = ["Tim Rae", "V.A. Sole", "P. Knobel"]
+__license__ = "MIT"
+__date__ = "24/05/2016"
+
+import logging
+
+from . import qt
+
+_logger = logging.getLogger(__name__)
+
+try:
+    import IPython
+except ImportError as e:
+    raise ImportError("Failed to import IPython, required by " + __name__)
+    
+# This widget cannot be used inside an interactive IPython shell.
+# It would raise MultipleInstanceError("Multiple incompatible subclass
+# instances of InProcessInteractiveShell are being created").
+try:
+    __IPYTHON__
+except NameError:
+    pass  # Not in IPython
+else:
+    msg = "Module " + __name__ + " cannot be used within an IPython shell"
+    raise ImportError(msg)
+
+# qtconsole is a separate module in recent versions of IPython/Jupyter
+# http://blog.jupyter.org/2015/04/15/the-big-split/
+if IPython.__version__.startswith("2"):
+    qtconsole = None
+else:
+    try:
+        import qtconsole
+    except ImportError:
+        qtconsole = None
+
+if qtconsole is not None:
+    try:
+        from qtconsole.rich_ipython_widget import RichJupyterWidget as \
+            RichIPythonWidget
+    except ImportError:
+        try:
+            from qtconsole.rich_ipython_widget import RichIPythonWidget
+        except ImportError as e:
+            qtconsole = None
+        else:
+            from qtconsole.inprocess import QtInProcessKernelManager
+    else:
+        from qtconsole.inprocess import QtInProcessKernelManager
+
+
+if qtconsole is None:
+    # Import the console machinery from ipython
+
+    # The `has_binding` test of IPython does not find the Qt bindings
+    # in case silx is used in a frozen binary
+    import IPython.external.qt_loaders
+
+    def has_binding(*var, **kw):
+        return True
+
+    IPython.external.qt_loaders.has_binding = has_binding
+
+    from IPython.qt.console.rich_ipython_widget import RichIPythonWidget
+    from IPython.qt.inprocess import QtInProcessKernelManager
+
+
+class IPythonWidget(RichIPythonWidget):
+    """Live IPython console widget.
+
+    :param custom_banner: Custom welcome message to be printed at the top of
+       the console.
+    """
+
+    def __init__(self, parent=None, custom_banner=None, *args, **kwargs):
+        if parent is not None:
+            kwargs["parent"] = parent
+        super(IPythonWidget, self).__init__(*args, **kwargs)
+        if custom_banner is not None:
+            self.banner = custom_banner
+        self.setWindowTitle(self.banner)
+        self.kernel_manager = kernel_manager = QtInProcessKernelManager()
+        kernel_manager.start_kernel()
+        self.kernel_client = kernel_client = self._kernel_manager.client()
+        kernel_client.start_channels()
+
+        def stop():
+            kernel_client.stop_channels()
+            kernel_manager.shutdown_kernel()
+        self.exit_requested.connect(stop)
+
+    def sizeHint(self):
+        """Return a reasonable default size for usage in :class:`PlotWindow`"""
+        return qt.QSize(500, 300)
+
+    def pushVariables(self, variable_dict):
+        """ Given a dictionary containing name / value pairs, push those
+        variables to the IPython console widget.
+
+        :param variable_dict: Dictionary of variables to be pushed to the
+            console's interactive namespace (```{variable_name: object, …}```)
+        """
+        self.kernel_manager.kernel.shell.push(variable_dict)
+
+
+class IPythonDockWidget(qt.QDockWidget):
+    """Dock Widget including a :class:`IPythonWidget` inside
+    a vertical layout.
+
+    :param available_vars: Dictionary of variables to be pushed to the
+        console's interactive namespace: ``{"variable_name": object, …}``
+    :param custom_banner: Custom welcome message to be printed at the top of
+        the console
+    :param title: Dock widget title
+    :param parent: Parent :class:`qt.QMainWindow` containing this
+        :class:`qt.QDockWidget`
+    """
+    def __init__(self, parent=None, available_vars=None, custom_banner=None,
+                 title="Console"):
+        super(IPythonDockWidget, self).__init__(title, parent)
+
+        self.ipyconsole = IPythonWidget(custom_banner=custom_banner)
+
+        self.layout().setContentsMargins(0, 0, 0, 0)
+        self.setWidget(self.ipyconsole)
+
+        if available_vars is not None:
+            self.ipyconsole.pushVariables(available_vars)
+
+    def showEvent(self, event):
+        """Make sure this widget is raised when it is shown
+        (when it is first created as a tab in PlotWindow or when it is shown
+        again after hiding).
+        """
+        self.raise_()
+
+
+def main():
+    """Run a Qt app with an IPython console"""
+    app = qt.QApplication([])
+    widget = IPythonDockWidget()
+    widget.show()
+    app.exec_()
+
+if __name__ == '__main__':
+    main()