passagemath-repl 10.4.62__py3-none-any.whl

sage/repl/display/formatter.py

@@ -0,0 +1,318 @@
+# sage_setup: distribution = sagemath-repl
+r"""
+IPython Displayhook Formatters
+
+The classes in this module can be used as IPython displayhook
+formatters. It has two main features, by default the displayhook
+contains a new facility for displaying lists of matrices in an easier
+to read format::
+
+    sage: [identity_matrix(i) for i in range(2, 5)]                                     # needs sage.modules
+    [
+                    [1 0 0 0]
+           [1 0 0]  [0 1 0 0]
+    [1 0]  [0 1 0]  [0 0 1 0]
+    [0 1], [0 0 1], [0 0 0 1]
+    ]
+
+This facility uses :meth:`_repr_` (and a simple string) to try do a nice read
+format (see :meth:`sage.structure.parent.Parent._repr_option` for details).
+
+With this displayhook there exists an other way for displaying object and more
+generally, all sage expression as an ASCII art object::
+
+    sage: from sage.repl.interpreter import get_test_shell
+    sage: shell = get_test_shell()
+    sage: shell.run_cell('%display ascii_art')
+    sage: shell.run_cell('integral(x^2/pi^x, x)')                                       # needs sage.symbolic
+       -x / 2    2                      \
+    -pi  *\x *log (pi) + 2*x*log(pi) + 2/
+    --------------------------------------
+                     3
+                   log (pi)
+    sage: shell.run_cell("i = var('i')")                                                # needs sage.symbolic
+    sage: shell.run_cell('sum(i*x^i, i, 0, 10)')                                        # needs sage.symbolic
+        10      9      8      7      6      5      4      3      2
+    10*x   + 9*x  + 8*x  + 7*x  + 6*x  + 5*x  + 4*x  + 3*x  + 2*x  + x
+    sage: shell.run_cell('StandardTableaux(4).list()')                                  # needs sage.combinat
+    [
+    [                                                                  1  4    1  3
+    [                 1  3  4    1  2  4    1  2  3    1  3    1  2    2       2
+    [   1  2  3  4,   2      ,   3      ,   4      ,   2  4,   3  4,   3   ,   4   ,
+    <BLANKLINE>
+               1 ]
+       1  2    2 ]
+       3       3 ]
+       4   ,   4 ]
+    sage: shell.run_cell('%display default')
+    sage: shell.quit()
+
+This other facility uses a simple
+:class:`~sage.typeset.ascii_art.AsciiArt` object (see and
+:meth:`sage.structure.sage_object.SageObject._ascii_art_`).  """
+
+# ****************************************************************************
+#       Copyright (C) 2014 Volker Braun <vbraun.name@gmail.com>
+#
+#  Distributed under the terms of the GNU General Public License (GPL)
+#  as published by the Free Software Foundation; either version 2 of
+#  the License, or (at your option) any later version.
+#                  https://www.gnu.org/licenses/
+# ****************************************************************************
+
+from io import StringIO
+
+from IPython.core.formatters import DisplayFormatter, PlainTextFormatter
+from IPython.utils.py3compat import unicode_to_str
+from IPython.core.display import DisplayObject
+
+from ipywidgets import Widget
+
+from sage.repl.display.pretty_print import SagePrettyPrinter
+from sage.misc.lazy_import import lazy_import
+
+IPYTHON_NATIVE_TYPES = (DisplayObject, Widget)
+
+PLAIN_TEXT = 'text/plain'
+TEXT_LATEX = 'text/latex'
+TEXT_HTML = 'text/html'
+
+lazy_import('matplotlib.figure', 'Figure')
+
+
+class SageDisplayFormatter(DisplayFormatter):
+
+    def __init__(self, *args, **kwds):
+        """
+        This is where the Sage rich objects are translated to IPython.
+
+        INPUT/OUTPUT: see the IPython documentation
+
+        EXAMPLES:
+
+        This is part of how Sage works with the IPython output
+        system. It cannot be used in doctests::
+
+            sage: from sage.repl.display.formatter import SageDisplayFormatter
+            sage: fmt = SageDisplayFormatter()
+            Traceback (most recent call last):
+            ...
+            RuntimeError: check failed: current backend is invalid
+        """
+        super().__init__(*args, **kwds)
+        from sage.repl.rich_output.display_manager import get_display_manager
+        self.dm = get_display_manager()
+        from sage.repl.rich_output.backend_ipython import BackendIPython
+        self.dm.check_backend_class(BackendIPython)
+
+        pt_formatter = self.formatters[PLAIN_TEXT]
+        pt_formatter.observe(self._ipython_float_precision_changed,
+                             names=['float_precision'])
+
+    def format(self, obj, include=None, exclude=None):
+        r"""
+        Use the Sage rich output instead of IPython.
+
+        INPUT/OUTPUT: see the IPython documentation
+
+        EXAMPLES::
+
+            sage: [identity_matrix(i) for i in range(3,7)]                              # needs sage.modules
+            [
+                                             [1 0 0 0 0 0]
+                                [1 0 0 0 0]  [0 1 0 0 0 0]
+                     [1 0 0 0]  [0 1 0 0 0]  [0 0 1 0 0 0]
+            [1 0 0]  [0 1 0 0]  [0 0 1 0 0]  [0 0 0 1 0 0]
+            [0 1 0]  [0 0 1 0]  [0 0 0 1 0]  [0 0 0 0 1 0]
+            [0 0 1], [0 0 0 1], [0 0 0 0 1], [0 0 0 0 0 1]
+            ]
+            sage: from sage.repl.interpreter import get_test_shell
+            sage: shell = get_test_shell()
+            sage: shell.run_cell('%display ascii_art')   # indirect doctest
+            sage: shell.run_cell("i = var('i')")                                        # needs sage.symbolic
+            sage: shell.run_cell('sum(i*x^i, i, 0, 10)')                                # needs sage.symbolic
+                10      9      8      7      6      5      4      3      2
+            10*x   + 9*x  + 8*x  + 7*x  + 6*x  + 5*x  + 4*x  + 3*x  + 2*x  + x
+            sage: shell.run_cell('%display default')
+            sage: shell.quit()
+
+        TESTS::
+
+            sage: import os
+            sage: import importlib.resources
+            sage: import warnings
+            sage: warnings.filterwarnings('ignore', category=DeprecationWarning,
+            ....:     message=r'path is deprecated\. Use files\(\) instead\.')
+            sage: from sage.repl.rich_output.backend_ipython import BackendIPython
+            sage: backend = BackendIPython()
+            sage: shell = get_test_shell()
+            sage: backend.install(shell=shell)
+            sage: shell.run_cell('get_ipython().display_formatter')
+            <sage.repl.display.formatter.SageDisplayFormatter object at 0x...>
+            sage: shell.run_cell('from IPython.display import Image')
+            sage: with importlib.resources.path(sage.repl.rich_output, 'example.png') as example_png:
+            ....:     shell.run_cell('ipython_image = Image("{0}")'.format(example_png))
+            sage: shell.run_cell('ipython_image')
+            <IPython.core.display.Image object>
+            sage: shell.run_cell('get_ipython().display_formatter.format(ipython_image)')
+            ({'image/png': ...,
+              'text/plain': '<IPython.core.display.Image object>'},
+            {})
+
+        Test that IPython images and widgets still work even in latex output mode::
+
+            sage: shell.run_cell('%display latex')   # indirect doctest
+            sage: shell.run_cell('set(get_ipython().display_formatter.format(ipython_image)[0].keys())'
+            ....:                ' == set(["text/plain", "image/png"])')
+            True
+
+            sage: shell.run_cell('import ipywidgets')
+            sage: shell.run_cell('slider = ipywidgets.IntSlider()')
+            sage: shell.run_cell('get_ipython().display_formatter.format(slider)')
+            ...IntSlider(value=0)..., {})
+
+            sage: shell.run_cell('%display default')
+            sage: shell.quit()
+
+        Test that ``__repr__`` is only called once when generating text output::
+
+            sage: class Repper():
+            ....:    def __repr__(self):
+            ....:        print('__repr__ called')
+            ....:        return 'I am repper'
+            sage: Repper()
+            __repr__ called
+            I am repper
+        """
+        sage_format, sage_metadata = self.dm.displayhook(obj)
+        assert PLAIN_TEXT in sage_format, 'plain text is always present'
+
+        # use Sage rich output for any except those native to IPython, but only
+        # if it is not plain and dull
+        if (not isinstance(obj, IPYTHON_NATIVE_TYPES) and
+            not set(sage_format.keys()).issubset([PLAIN_TEXT]) and
+                not isinstance(obj, Figure)):
+            return sage_format, sage_metadata
+
+        if self.ipython_display_formatter(obj):
+            # object handled itself, don't proceed
+            return {}, {}
+
+        # try IPython display formatter
+        if exclude is not None:
+            exclude = list(exclude) + [PLAIN_TEXT]
+        else:
+            exclude = [PLAIN_TEXT]
+        ipy_format, ipy_metadata = super().format(obj, include=include, exclude=exclude)
+        if not ipy_format:
+            return sage_format, sage_metadata
+        ipy_format[PLAIN_TEXT] = sage_format[PLAIN_TEXT]
+        if PLAIN_TEXT in sage_metadata:
+            ipy_metadata[PLAIN_TEXT] = sage_metadata[PLAIN_TEXT]
+        return ipy_format, ipy_metadata
+
+    @staticmethod
+    def _ipython_float_precision_changed(change):
+        """
+        Update the current float precision for the display of matrices in Sage.
+
+        This function is called when the IPython ``%precision`` magic is
+        invoked.
+
+        TESTS::
+
+            sage: from sage.repl.interpreter import get_test_shell
+            sage: shell = get_test_shell()
+            sage: shell.run_cell('%precision 4')
+            '%.4f'
+            sage: shell.run_cell('matrix.options.precision')  # indirect doctest        # needs sage.modules
+            4
+            sage: shell.run_cell('%precision')
+            '%r'
+            sage: shell.run_cell('matrix.options.precision')  # indirect doctest        # needs sage.modules
+            None
+        """
+        try:
+            from sage.matrix.constructor import options
+        except ImportError:
+            pass
+        else:
+            s = change.new
+            if not s:
+                # unset the precision
+                options.precision = None
+            else:
+                try:
+                    prec = int(s)
+                    if prec >= 0:
+                        options.precision = prec
+                    # otherwise ignore the change
+                except ValueError:
+                    pass
+
+
+class SagePlainTextFormatter(PlainTextFormatter):
+
+    def __init__(self, *args, **kwds):
+        r"""
+        Improved plain text IPython formatter.
+
+        In particular, it correctly print lists of matrices or other
+        objects (see
+        :meth:`sage.structure.parent.Parent._repr_option`).
+
+        .. warning::
+
+            This IPython formatter is NOT used. You could use it to
+            enable Sage formatting in IPython, but Sage uses its own
+            rich output system that is more flexible and supports
+            different backends.
+
+        INPUT/OUTPUT: see the IPython documentation
+
+        EXAMPLES::
+
+            sage: from sage.repl.interpreter import get_test_shell
+            sage: shell = get_test_shell()
+            sage: shell.display_formatter.formatters['text/plain']
+            <IPython.core.formatters.PlainTextFormatter object at 0x...>
+            sage: shell.quit()
+        """
+        super().__init__(*args, **kwds)
+
+    def __call__(self, obj):
+        r"""
+        Compute the pretty representation of the object.
+
+        Adapted from ``IPython.core.formatters.PlainTextPrettyPrint``.
+
+        INPUT:
+
+        - ``obj`` -- anything
+
+        OUTPUT: string; the plain text representation
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.formatter import SagePlainTextFormatter
+            sage: fmt = SagePlainTextFormatter()
+            sage: fmt
+            <sage.repl.display.formatter.SagePlainTextFormatter object at 0x...>
+            sage: fmt(2)
+            ---- calling ipython formatter ----
+            '2'
+            sage: a = identity_matrix(ZZ, 2)                                            # needs sage.modules
+            sage: fmt([a, a])                                                           # needs sage.modules
+            ---- calling ipython formatter ----
+            '[\n[1 0]  [1 0]\n[0 1], [0 1]\n]'
+        """
+        from sage.doctest import DOCTEST_MODE
+        if DOCTEST_MODE:
+            # Just to show that this is never executed in any other doctests in the Sage library
+            print('---- calling ipython formatter ----')
+        stream = StringIO()
+        printer = SagePrettyPrinter(
+            stream, self.max_width, unicode_to_str(self.newline))
+        printer.pretty(obj)
+        printer.flush()
+        return stream.getvalue()

sage/repl/display/jsmol_iframe.py

@@ -0,0 +1,290 @@
+# sage_setup: distribution = sagemath-repl
+"""
+HTML Generator for JSmol
+
+This is all an evil iframe hack to get JSmol to display 3-d graphics
+while separating JSmol's j2s machinery from your actual web page.
+
+There are some caveats for how to load JSmol, in particular it cannot
+just load its code from a ``file://`` uri. To use a html file
+generated by this module, you need
+
+* A web server,
+
+* The JSmol directory tree must be served by your web server,
+
+* The output of :meth:`JSMolHtml.inner_html` or
+  :meth:`JSMolHtml.outer_html` must be served by the same web server.
+
+See https://github.com/phetsims/molecule-polarity/issues/6 for a
+discussion of loading JSMol.
+"""
+
+import io
+import os
+import zipfile
+
+from sage.cpython.string import bytes_to_str
+from sage.structure.sage_object import SageObject
+from sage.misc.cachefunc import cached_method
+
+
+INNER_HTML_TEMPLATE = """
+<html>
+<head>
+  <style>
+    * {{
+      margin: 0;
+      padding: 0;
+      overflow: hidden;
+    }}
+    body, html {{
+      height: 100%;
+      width: 100%;
+    }}
+  </style>
+  <script type="text/javascript" src="{path_to_jsmol}/JSmol.min.js"></script>
+</head>
+<body>
+  <script type="text/javascript">
+    delete Jmol._tracker; // Prevent JSmol from phoning home.
+    var script = {script};
+    var Info = {{
+      width: '{width}',
+      height: '{height}',
+      debug: false,
+      disableInitialConsole: true,   // very slow when used with inline mesh
+      color: '#3131ff',
+      addSelectionOptions: false,
+      use: 'HTML5',
+      j2sPath: '{path_to_jsmol}/j2s',
+      script: script,
+    }};
+    var jmolApplet0 = Jmol.getApplet('jmolApplet0', Info);
+  </script>
+</body>
+</html>
+"""
+
+
+IFRAME_TEMPLATE = """
+<iframe srcdoc="{escaped_inner_html}"
+        width="{width}"
+        height="{height}"
+        style="border: 0;">
+</iframe>
+"""
+
+
+OUTER_HTML_TEMPLATE = """
+<html>
+<head>
+  <title>JSmol 3D Scene</title>
+</head>
+</body>
+  {iframe}
+</body>
+</html>
+""".format(iframe=IFRAME_TEMPLATE)
+
+
+class JSMolHtml(SageObject):
+
+    def __init__(self, jmol, path_to_jsmol=None, width='100%', height='100%'):
+        """
+        INPUT:
+
+        - ``jmol`` -- 3-d graphics or
+          :class:`sage.repl.rich_output.output_graphics3d.OutputSceneJmol`
+          instance. The 3-d scene to show.
+
+        - ``path_to_jsmol`` -- string (default:
+          ``'/nbextensions/jupyter-jsmol/jsmol'``). The path (relative or absolute)
+          where ``JSmol.min.js`` is served on the web server.
+
+        - ``width`` -- integer or string (default:
+          ``'100%'``). The width of the JSmol applet using CSS
+          dimensions.
+
+        - ``height`` -- integer or string (default:
+          ``'100%'``). The height of the JSmol applet using CSS
+          dimensions.
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: JSMolHtml(sphere(), width=500, height=300)                            # needs sage.plot
+            JSmol Window 500x300
+        """
+        from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+        if not isinstance(jmol, OutputSceneJmol):
+            jmol = jmol._rich_repr_jmol()
+        self._jmol = jmol
+        self._zip = zipfile.ZipFile(io.BytesIO(self._jmol.scene_zip.get()))
+        if path_to_jsmol is None:
+            self._path = os.path.join('/', 'nbextensions', 'jupyter-jsmol', 'jsmol')
+        else:
+            self._path = path_to_jsmol
+        self._width = width
+        self._height = height
+
+    @cached_method
+    def script(self):
+        r"""
+        Return the JMol script file.
+
+        This method extracts the Jmol script from the Jmol spt file (a
+        zip archive) and inlines meshes.
+
+        OUTPUT: string
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+            sage: jsmol = JSMolHtml(OutputSceneJmol.example(), width=500, height=300)
+            sage: jsmol.script()
+            'data "model list"\n10\nempt...aliasdisplay on;\n'
+        """
+        script = []
+        with self._zip.open('SCRIPT') as SCRIPT:
+            for line in SCRIPT:
+                if line.startswith(b'pmesh'):
+                    command, obj, meshfile = line.split(b' ', 3)
+                    assert command == b'pmesh'
+                    if meshfile not in [b'dots\n', b'mesh\n']:
+                        assert (meshfile.startswith(b'"') and
+                                meshfile.endswith(b'"\n'))
+                        meshfile = bytes_to_str(meshfile[1:-2])  # strip quotes
+                        script += [
+                            'pmesh {0} inline "'.format(bytes_to_str(obj)),
+                            bytes_to_str(self._zip.open(meshfile).read()),
+                            '"\n'
+                        ]
+                        continue
+                script += [bytes_to_str(line)]
+        return ''.join(script)
+
+    def js_script(self):
+        r"""
+        The :meth:`script` as Javascript string.
+
+        Since the many shortcomings of Javascript include multi-line
+        strings, this actually returns Javascript code to reassemble
+        the script from a list of strings.
+
+        OUTPUT:
+
+        String. Javascript code that evaluates to :meth:`script` as
+        Javascript string.
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+            sage: jsmol = JSMolHtml(OutputSceneJmol.example(), width=500, height=300)
+            sage: print(jsmol.js_script())
+            [
+              'data "model list"',
+              ...
+              'isosurface fullylit; pmesh o* fullylit; set antialiasdisplay on;',
+            ].join('\n');
+        """
+        script = [r"["]
+        for line in self.script().splitlines():
+            script += [r"  '{0}',".format(line)]
+        script += [r"].join('\n');"]
+        return '\n'.join(script)
+
+    def _repr_(self):
+        """
+        Return as string representation.
+
+        OUTPUT: string
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+            sage: JSMolHtml(OutputSceneJmol.example(), width=500, height=300)._repr_()
+            'JSmol Window 500x300'
+        """
+        return 'JSmol Window {0}x{1}'.format(self._width, self._height)
+
+    def inner_html(self):
+        """
+        Return a HTML document containing a JSmol applet.
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+            sage: jmol = JSMolHtml(OutputSceneJmol.example(), width=500, height=300)
+            sage: print(jmol.inner_html())
+            <html>
+            <head>
+              <style>
+                * {
+                  margin: 0;
+                  padding: 0;
+                    ...
+            </html>
+        """
+        return INNER_HTML_TEMPLATE.format(
+            script=self.js_script(),
+            width=self._width,
+            height=self._height,
+            path_to_jsmol=self._path,
+        )
+
+    def iframe(self):
+        """
+        Return HTML iframe.
+
+        OUTPUT: string
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+            sage: jmol = JSMolHtml(OutputSceneJmol.example())
+            sage: print(jmol.iframe())
+            <iframe srcdoc="
+            ...
+            </iframe>
+        """
+        escaped_inner_html = self.inner_html().replace('"', '&quot;')
+        return IFRAME_TEMPLATE.format(width=self._width,
+                                      height=self._height,
+                                      escaped_inner_html=escaped_inner_html)
+
+    def outer_html(self):
+        """
+        Return a HTML document containing an iframe with a JSmol applet.
+
+        OUTPUT: string
+
+        EXAMPLES::
+
+            sage: from sage.repl.display.jsmol_iframe import JSMolHtml
+            sage: from sage.repl.rich_output.output_graphics3d import OutputSceneJmol
+            sage: jmol = JSMolHtml(OutputSceneJmol.example(), width=500, height=300)
+            sage: print(jmol.outer_html())
+            <html>
+            <head>
+              <title>JSmol 3D Scene</title>
+            </head>
+            </body>
+            <BLANKLINE>
+            <iframe srcdoc="
+                    ...
+            </html>
+        """
+        escaped_inner_html = self.inner_html().replace('"', '&quot;')
+        outer = OUTER_HTML_TEMPLATE.format(
+            script=self.js_script(),
+            width=self._width,
+            height=self._height,
+            escaped_inner_html=escaped_inner_html,
+        )
+        return outer