Trajectree 0.0.0__py3-none-any.whl → 0.0.1__py3-none-any.whl

trajectree/__init__.py

@@ -0,0 +1,3 @@
+import sys
+import os
+sys.path.append(os.path.abspath(__file__).replace('__init__.py', 'quimb'))

trajectree/fock_optics/devices.py

@@ -3,7 +3,7 @@ from scipy.linalg import expm
 import numpy as np
 from numpy import kron
 
-from quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from ..quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
 
 import qutip as qt
 

trajectree/fock_optics/light_sources.py

@@ -8,8 +8,8 @@ import numpy as np
 from numpy.linalg import matrix_power
 from numpy import kron, sqrt
 
-from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-from quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
+from ..quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from ..quimb.quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
 
 import qutip as qt
 from math import factorial

trajectree/fock_optics/measurement.py

@@ -9,9 +9,9 @@ import numpy as np
 from numpy.linalg import matrix_power
 from numpy import sqrt
 
-from quimb.tensor import MatrixProductOperator as mpo #type: ignore
-from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-from quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
+from ..quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from ..quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from ..quimb.quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
 
 import qutip as qt
 from math import factorial

trajectree/fock_optics/utils.py

@@ -1,12 +1,12 @@
 import numpy as np
 from numpy import sqrt
 
-from quimb.tensor import MatrixProductState as mps #type: ignore
-from quimb.tensor import MatrixProductOperator as mpo #type: ignore
-from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-from quimb.tensor.tensor_core import new_bond #type: ignore
-from quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
-from quimb.tensor.tensor_1d import TensorNetwork1DOperator #type: ignore
+from ..quimb.quimb.tensor import MatrixProductState as mps #type: ignore
+from ..quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from ..quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from ..quimb.quimb.tensor.tensor_core import new_bond #type: ignore
+from ..quimb.quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
+from ..quimb.quimb.tensor.tensor_1d import TensorNetwork1DOperator #type: ignore
 
 import qutip as qt
 import re

trajectree/quimb/docs/_pygments/_pygments_dark.py

@@ -0,0 +1,118 @@
+# -*- coding: utf-8 -*-
+"""
+Pygments version of the sublime Mariana theme.
+
+Pygments template by Jan T. Sott (https://github.com/idleberg)
+"""
+
+from pygments.style import Style
+from pygments.token import Keyword, Name, Comment, String, Error, Text, \
+     Number, Operator, Generic, Whitespace, Punctuation, Other, Literal
+
+
+BACKGROUND = "#1a1c1e"
+CURRENT_LINE = "#4e5a65"
+SELECTION = "#343d46"
+FOREGROUND = "#d8dee9"
+COMMENT = "#5a6272"
+RED = "#ec5f66"
+ORANGE = "#f9ae58"
+YELLOW = "#fac761"
+GREEN = "#99c794"
+AQUA = "#70c2bb"
+BLUE = "#6699cc"
+PURPLE = "#c695c6"
+
+
+class MarianaDark(Style):
+
+    default_style = ''
+
+    background_color = BACKGROUND
+    highlight_color = SELECTION
+
+    background_color = BACKGROUND
+    highlight_color = SELECTION
+
+    styles = {
+        # No corresponding class for the following:
+        Text:                      FOREGROUND,  # class:  ''
+        Whitespace:                "",          # class: 'w'
+        Error:                     RED,         # class: 'err'
+        Other:                     "",          # class 'x'
+
+        Comment:                   COMMENT,   # class: 'c'
+        Comment.Multiline:         "",        # class: 'cm'
+        Comment.Preproc:           "",        # class: 'cp'
+        Comment.Single:            "",        # class: 'c1'
+        Comment.Special:           "",        # class: 'cs'
+
+        Keyword:                   PURPLE,    # class: 'k'
+        Keyword.Constant:          RED,       # class: 'kc'
+        Keyword.Declaration:       "",        # class: 'kd'
+        Keyword.Namespace:         PURPLE,    # class: 'kn'
+        Keyword.Pseudo:            RED,       # class: 'kp'
+        Keyword.Reserved:          "",        # class: 'kr'
+        Keyword.Type:              YELLOW,    # class: 'kt'
+
+        Operator:                  RED,       # class: 'o'
+        Operator.Word:             "",        # class: 'ow' - like keywords
+
+        Punctuation:               AQUA,       # class: 'p'
+
+        Name:                      FOREGROUND,  # class: 'n'
+        Name.Attribute:            BLUE,        # class: 'na' - to be revised
+        Name.Builtin:              BLUE,          # class: 'nb'
+        Name.Builtin.Pseudo:       RED,         # class: 'bp'
+        Name.Class:                ORANGE,      # class: 'nc' - to be revised
+        Name.Constant:             RED,         # class: 'no' - to be revised
+        Name.Decorator:            AQUA,        # class: 'nd' - to be revised
+        Name.Entity:               BLUE,          # class: 'ni'
+        Name.Exception:            RED,         # class: 'ne'
+        Name.Function:             AQUA,        # class: 'nf'
+        Name.Function.Magic:       BLUE,        # class: 'nf'
+        Name.Property:             BLUE,        # class: 'py'
+        Name.Label:                BLUE,          # class: 'nl'
+        Name.Namespace:            BLUE,        # class: 'nn' - to be revised
+        Name.Other:                BLUE,        # class: 'nx'
+        Name.Tag:                  AQUA,        # class: 'nt' - like a keyword
+        Name.Variable:             RED,         # class: 'nv' - to be revised
+        Name.Variable.Class:       "",          # class: 'vc' - to be revised
+        Name.Variable.Global:      "",          # class: 'vg' - to be revised
+        Name.Variable.Instance:    "",          # class: 'vi' - to be revised
+
+        Number:                    ORANGE,    # class: 'm'
+        Number.Float:              "",        # class: 'mf'
+        Number.Hex:                "",        # class: 'mh'
+        Number.Integer:            "",        # class: 'mi'
+        Number.Integer.Long:       "",        # class: 'il'
+        Number.Oct:                "",        # class: 'mo'
+
+        Literal:                   ORANGE,    # class: 'l'
+        Literal.Date:              GREEN,     # class: 'ld'
+
+        String:                    GREEN,       # class: 's'
+        String.Backtick:           "",          # class: 'sb'
+        String.Char:               FOREGROUND,  # class: 'sc'
+        String.Doc:                COMMENT,     # class: 'sd' - like a comment
+        String.Double:             "",          # class: 's2'
+        String.Escape:             ORANGE,      # class: 'se'
+        String.Heredoc:            "",          # class: 'sh'
+        String.Interpol:           ORANGE,      # class: 'si'
+        String.Other:              "",          # class: 'sx'
+        String.Regex:              "",          # class: 'sr'
+        String.Single:             "",          # class: 's1'
+        String.Symbol:             "",          # class: 'ss'
+
+        Generic:                   "",                    # class: 'g'
+        Generic.Deleted:           RED,                   # class: 'gd',
+        Generic.Emph:              "italic",              # class: 'ge'
+        Generic.Error:             "",                    # class: 'gr'
+        Generic.Heading:           "bold " + FOREGROUND,  # class: 'gh'
+        Generic.Inserted:          GREEN,                 # class: 'gi'
+        Generic.Output:            "",                    # class: 'go'
+        Generic.Prompt:            "bold " + COMMENT,     # class: 'gp'
+        Generic.Strong:            "bold",                # class: 'gs'
+        Generic.Subheading:        "bold " + AQUA,        # class: 'gu'
+        Generic.Traceback:         "",                    # class: 'gt'
+    }

trajectree/quimb/docs/_pygments/_pygments_light.py

@@ -0,0 +1,118 @@
+# -*- coding: utf-8 -*-
+"""
+Pygments (light) version of the sublime Mariana theme.
+
+Pygments template by Jan T. Sott (https://github.com/idleberg)
+"""
+
+from pygments.style import Style
+from pygments.token import Keyword, Name, Comment, String, Error, Text, \
+     Number, Operator, Generic, Whitespace, Punctuation, Other, Literal
+
+
+BACKGROUND = "#f8f9fb"
+CURRENT_LINE = "#b7c0c8"
+SELECTION = "#d0d6dc"
+FOREGROUND = "#161c27"
+COMMENT = "#8d95a5"
+RED = "#e7323b"
+ORANGE = "#f79626"
+YELLOW = "#f9b52f"
+GREEN = "#70b069"
+AQUA = "#3d8f88"
+BLUE = "#407fbf"
+PURPLE = "#b474b4"
+
+
+class MarianaLight(Style):
+
+    default_style = ''
+
+    background_color = BACKGROUND
+    highlight_color = SELECTION
+
+    background_color = BACKGROUND
+    highlight_color = SELECTION
+
+    styles = {
+        # No corresponding class for the following:
+        Text:                      FOREGROUND,  # class:  ''
+        Whitespace:                "",          # class: 'w'
+        Error:                     RED,         # class: 'err'
+        Other:                     "",          # class 'x'
+
+        Comment:                   COMMENT,   # class: 'c'
+        Comment.Multiline:         "",        # class: 'cm'
+        Comment.Preproc:           "",        # class: 'cp'
+        Comment.Single:            "",        # class: 'c1'
+        Comment.Special:           "",        # class: 'cs'
+
+        Keyword:                   PURPLE,    # class: 'k'
+        Keyword.Constant:          RED,       # class: 'kc'
+        Keyword.Declaration:       "",        # class: 'kd'
+        Keyword.Namespace:         PURPLE,    # class: 'kn'
+        Keyword.Pseudo:            RED,       # class: 'kp'
+        Keyword.Reserved:          "",        # class: 'kr'
+        Keyword.Type:              YELLOW,    # class: 'kt'
+
+        Operator:                  RED,       # class: 'o'
+        Operator.Word:             "",        # class: 'ow' - like keywords
+
+        Punctuation:               AQUA,       # class: 'p'
+
+        Name:                      FOREGROUND,  # class: 'n'
+        Name.Attribute:            BLUE,        # class: 'na' - to be revised
+        Name.Builtin:              BLUE,          # class: 'nb'
+        Name.Builtin.Pseudo:       RED,         # class: 'bp'
+        Name.Class:                ORANGE,      # class: 'nc' - to be revised
+        Name.Constant:             RED,         # class: 'no' - to be revised
+        Name.Decorator:            AQUA,        # class: 'nd' - to be revised
+        Name.Entity:               BLUE,          # class: 'ni'
+        Name.Exception:            RED,         # class: 'ne'
+        Name.Function:             AQUA,        # class: 'nf'
+        Name.Function.Magic:       BLUE,        # class: 'nf'
+        Name.Property:             BLUE,        # class: 'py'
+        Name.Label:                BLUE,          # class: 'nl'
+        Name.Namespace:            BLUE,        # class: 'nn' - to be revised
+        Name.Other:                BLUE,        # class: 'nx'
+        Name.Tag:                  AQUA,        # class: 'nt' - like a keyword
+        Name.Variable:             RED,         # class: 'nv' - to be revised
+        Name.Variable.Class:       "",          # class: 'vc' - to be revised
+        Name.Variable.Global:      "",          # class: 'vg' - to be revised
+        Name.Variable.Instance:    "",          # class: 'vi' - to be revised
+
+        Number:                    ORANGE,    # class: 'm'
+        Number.Float:              "",        # class: 'mf'
+        Number.Hex:                "",        # class: 'mh'
+        Number.Integer:            "",        # class: 'mi'
+        Number.Integer.Long:       "",        # class: 'il'
+        Number.Oct:                "",        # class: 'mo'
+
+        Literal:                   ORANGE,    # class: 'l'
+        Literal.Date:              GREEN,     # class: 'ld'
+
+        String:                    GREEN,       # class: 's'
+        String.Backtick:           "",          # class: 'sb'
+        String.Char:               FOREGROUND,  # class: 'sc'
+        String.Doc:                COMMENT,     # class: 'sd' - like a comment
+        String.Double:             "",          # class: 's2'
+        String.Escape:             ORANGE,      # class: 'se'
+        String.Heredoc:            "",          # class: 'sh'
+        String.Interpol:           ORANGE,      # class: 'si'
+        String.Other:              "",          # class: 'sx'
+        String.Regex:              "",          # class: 'sr'
+        String.Single:             "",          # class: 's1'
+        String.Symbol:             "",          # class: 'ss'
+
+        Generic:                   "",                    # class: 'g'
+        Generic.Deleted:           RED,                   # class: 'gd',
+        Generic.Emph:              "italic",              # class: 'ge'
+        Generic.Error:             "",                    # class: 'gr'
+        Generic.Heading:           "bold " + FOREGROUND,  # class: 'gh'
+        Generic.Inserted:          GREEN,                 # class: 'gi'
+        Generic.Output:            "",                    # class: 'go'
+        Generic.Prompt:            "bold " + COMMENT,     # class: 'gp'
+        Generic.Strong:            "bold",                # class: 'gs'
+        Generic.Subheading:        "bold " + AQUA,        # class: 'gu'
+        Generic.Traceback:         "",                    # class: 'gt'
+    }

trajectree/quimb/docs/conf.py

@@ -0,0 +1,158 @@
+# Configuration file for the Sphinx documentation builder.
+#
+# For the full list of built-in configuration values, see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+import os
+import sys
+sys.path.append(os.path.abspath("./_pygments"))
+
+# -- Project information -----------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
+
+project = 'quimb'
+copyright = '2015-2024, Johnnie Gray'
+author = 'Johnnie Gray'
+
+# The full version, including alpha/beta/rc tags
+try:
+    from quimb import __version__
+    release = __version__
+except ImportError:
+    try:
+        from importlib.metadata import version as _version
+        release = _version('quimb')
+    except ImportError:
+        release = '0.0.0+unknown'
+
+version = '.'.join(release.split('.')[:2])
+
+# -- General configuration ---------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
+
+extensions = [
+    'sphinx.ext.intersphinx',
+    'sphinx.ext.extlinks',
+    'sphinx.ext.napoleon',
+    'sphinx.ext.linkcode',
+    'myst_nb',
+    "sphinx_design",
+    'sphinx_copybutton',
+    'autoapi.extension',
+]
+
+# msyt_nb configuration
+nb_execution_mode = "off"
+myst_heading_anchors = 4
+myst_enable_extensions = [
+    "amsmath",
+    "colon_fence",
+    "deflist",
+    "dollarmath",
+    "html_image",
+]
+
+
+# sphinx-autoapi
+autoapi_dirs = ['../quimb']
+
+extlinks = {
+    'issue': ('https://github.com/jcmgray/quimb/issues/%s', 'GH %s'),
+    'pull': ('https://github.com/jcmgray/quimb/pull/%s', 'PR %s'),
+}
+intersphinx_mapping = {
+    'python': ('https://docs.python.org/3/', None),
+    'numpy': ('https://numpy.org/doc/stable/', None),
+    'scipy': ('https://docs.scipy.org/doc/scipy/', None),
+    'cotengra': ('https://cotengra.readthedocs.io/en/latest/', None),
+    'autoray': ('https://autoray.readthedocs.io/en/latest/', None),
+}
+
+# Add any paths that contain templates here, relative to this directory.
+templates_path = ['_templates']
+
+# List of patterns, relative to source directory, that match files and
+# directories to ignore when looking for source files.
+# This pattern also affects html_static_path and html_extra_path.
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store', '**.ipynb_checkpoints']
+
+# -- Options for HTML output -------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
+
+html_theme = 'furo'
+html_theme_options = {
+    "sidebar_hide_name": True,
+    "light_css_variables": {
+        # "color-brand-primary": "hsl(45, 80%, 45%)",
+        "color-brand-primary": "hsl(210, 50%, 50%)",
+        "color-brand-content": "hsl(210, 50%, 50%)",
+    },
+    "dark_css_variables": {
+        "color-brand-primary": "hsl(210, 50%, 60%)",
+        "color-brand-content": "hsl(210, 50%, 60%)",
+    },
+    "light_logo": "quimb_logo_title.png",
+    "dark_logo": "quimb_logo_title.png",
+}
+
+pygments_style = '_pygments_light.MarianaLight'
+pygments_dark_style = "_pygments_dark.MarianaDark"
+
+html_static_path = ['_static']
+html_css_files = ["my-styles.css"]
+html_favicon = "_static/quimb.ico"
+
+
+def linkcode_resolve(domain, info):
+    """
+    Determine the URL corresponding to Python object
+    """
+    import quimb
+    import inspect
+
+    if domain != "py":
+        return None
+
+    modname = info["module"]
+    fullname = info["fullname"]
+
+    submod = sys.modules.get(modname)
+    if submod is None:
+        return None
+
+    obj = submod
+    for part in fullname.split("."):
+        try:
+            obj = getattr(obj, part)
+        except AttributeError:
+            return None
+
+    try:
+        fn = inspect.getsourcefile(inspect.unwrap(obj))
+    except TypeError:
+        fn = None
+    if not fn:
+        return None
+
+    try:
+        source, lineno = inspect.getsourcelines(obj)
+    except OSError:
+        lineno = None
+
+    if lineno:
+        linespec = f"#L{lineno}-L{lineno + len(source) - 1}"
+    else:
+        linespec = ""
+
+    fn = os.path.relpath(fn, start=os.path.dirname(quimb.__file__))
+
+    if "+" in quimb.__version__:
+        return (
+            f"https://github.com/jcmgray/quimb/blob/"
+            f"main/quimb/{fn}{linespec}"
+        )
+    else:
+        return (
+            f"https://github.com/jcmgray/quimb/blob/"
+            f"v{quimb.__version__}/quimb/{fn}{linespec}"
+        )

trajectree/quimb/docs/examples/ex_mpi_expm_evo.py

@@ -0,0 +1,62 @@
+"""This script illustrates how quimb's various MPI modes work.
+
+It can be run as:
+
+python ex_mpi_modes_expm_evo.py
+quimb-mpi-python ex_mpi_modes_expm_evo.py
+quimb-mpi-python --syncro ex_mpi_modes_expm_evo.py
+
+And will display slightly different output for each that explains the
+three modes.
+"""
+import quimb as qu
+from mpi4py import MPI
+
+# Get some MPI information
+comm = MPI.COMM_WORLD
+rank, size = comm.Get_rank(), comm.Get_size()
+print(f"I am worker {rank} of total {size} running main script...")
+
+
+# setup a verbose version of the ham_heis constructor, and make it Lazy
+n = 18
+shape = (2**n, 2**n)
+
+# this makes the function print some information when called
+#     - in order to be pickled is has to be located in the main package
+ham_heis_verbose = qu.utils.Verbosify(qu.ham_heis,
+                                      highlight='ownership', mpi=True)
+
+H = qu.Lazy(ham_heis_verbose, n=n, sparse=True, shape=shape)
+
+# random initial state
+#     - must make sure all processes have the same seed to be pure
+psi0 = qu.rand_ket(2**n, seed=42)
+
+# evolve the system, processes split 'hard' work (slepc computations)
+#     - should see each worker gets given a different ownership rows
+#     - but all end up with the results.
+evo = qu.Evolution(psi0, H, method='expm', expm_backend='slepc')
+evo.update_to(5)
+
+
+print(f"{rank}: I have final state norm {qu.expec(evo.pt, evo.pt)}")
+
+# Now lets demonstrate using the MPI pool construct
+pool = qu.get_mpi_pool()
+
+dims = [2] * n
+bsz = 5
+logneg_subsys_verbose = qu.utils.Verbosify(qu.logneg_subsys,
+                                           highlight='sysb', mpi=True)
+
+# each process only computes its own fraction of these
+#     - should see each process calls logneg with different ``sysb``.
+fs = [pool.submit(logneg_subsys_verbose, evo.pt, dims=dims,
+                  sysa=range(0, bsz), sysb=range(i, i + bsz))
+      for i in range(bsz, n - bsz)]
+
+# but then the results are comminucated to everyone
+rs = [f.result() for f in fs]
+
+print(f"{rank}: I have logneg results: {rs}")