tidy up 08 and make plans

python/energy_flow_proper/08_dynamic_one_bath/coupling_modulation.org

@@ -13,8 +13,6 @@ Here we try the same thing as in [[file:coupling_and_system_cycle.org][here]] bu
   import scipy
   import utilities as ut
   import matplotlib
-  matplotlib.rcParams.update(fs.MPL_RC_POSTER)
-
   import itertools
 #+end_src
 
@@ -170,8 +168,6 @@ Init ray and silence stocproc.
       strobe_t, strobe_indices = ut.strobe_times(model.t, Δ)
 
       return model, strobe_t, strobe_indices
-
-
 #+end_src
 
 #+RESULTS:
@@ -2138,7 +2134,7 @@ effect persists. Ok, it was the exponents in the bcf expansion :P.
 [[file:./.ob-jupyter/c9f9872a8dfae5d029676840920eeaaff011fc40.svg]]
 :END:
 
-* TODO Model: Testing the Inital Slip
+* Testing the Inital Slip
 #+begin_src jupyter-python
   from hops.util.dynamic_matrix import SmoothStep, Periodic, Harmonic, ConstantMatrix, Piecewise
 

python/energy_flow_proper/08_dynamic_one_bath/energy_shovel.py

@@ -0,0 +1,66 @@
+import figsaver as fs
+from hiro_models.one_qubit_model import QubitModel, StocProcTolerances
+import hiro_models.model_auxiliary as aux
+from hops.util.utilities import relative_entropy, relative_entropy_single, entropy, trace_distance
+from hopsflow.util import EnsembleValue
+import numpy as np
+import qutip as qt
+import scipy
+import utilities as ut
+import matplotlib
+import itertools
+
+from hops.util.dynamic_matrix import (
+    SmoothStep,
+    Periodic,
+    Harmonic,
+    ConstantMatrix,
+    ScaleTime,
+)
+
+def energy_shovel(
+    Δ=2,
+    γ=1 / 10,
+    periods=5,
+    L_op=(1 / 2 * qt.sigmax()).full(),
+    H_op=1 / 2 * (qt.sigmaz() + 1).full(),
+    H_time_op=qt.sigmay().full() / 2,
+    dt=.01,
+    detune=0,
+    ω_c=1,
+    modulate_system=True,
+    k_max=7,
+):
+    t_max = 2*np.pi/Δ * periods
+    L = ConstantMatrix(L_op) - Harmonic(
+        L_op, Δ, np.pi / 2
+    )
+    H = ConstantMatrix(H_op)
+    if modulate_system:
+        H = H + γ * Δ * Harmonic(H_time_op, Δ)
+
+    model = QubitModel(
+        δ=1,
+        ω_c=ω_c,
+        ω_s=1 + Δ - ω_c + detune,
+        t=ut.linspace_with_strobe(0, t_max, int(t_max/dt + 1), Δ),
+        ψ_0=(0.5 * 0 * qt.basis([2], [0]) + qt.basis([2], [1])),
+        description=f"Testing the time dependent coupling with smooth step.",
+        k_max=k_max,
+        bcf_terms=7,
+        truncation_scheme="simplex",
+        driving_process_tolerance=StocProcTolerances(1e-5, 1e-5),
+        thermal_process_tolerance=StocProcTolerances(1e-5, 1e-5),
+        T=5,
+        L=L,
+        H=H,
+        therm_method="fft",
+    )
+
+    strobe_t, strobe_indices = ut.strobe_times(model.t, Δ)
+
+    return model, strobe_t, strobe_indices
+
+def ergo(T, ω=1):
+    """The ergotropy of a qubit coupled to a bath."""
+      return (T * np.log(1 + np.exp(-ω / T)))

python/energy_flow_proper/08_dynamic_one_bath/flake.lock

@@ -2,11 +2,11 @@
   "nodes": {
     "flake-utils": {
       "locked": {
-        "lastModified": 1656065134,
+        "lastModified": 1659877975,
-        "narHash": "sha256-oc6E6ByIw3oJaIyc67maaFcnjYOz1mMcOtHxbEf9NwQ=",
+        "narHash": "sha256-zllb8aq3YO3h8B/U0/J1WBgAL8EX5yWf5pMj3G0NAmc=",
         "owner": "numtide",
         "repo": "flake-utils",
-        "rev": "bee6a7250dd1b01844a2de7e02e4df7d8a0a206c",
+        "rev": "c0e246b9b83f637f4681389ecabcb2681b4f3af0",
         "type": "github"
       },
       "original": {
@@ -17,11 +17,11 @@
     },
     "flake-utils_2": {
       "locked": {
-        "lastModified": 1656065134,
+        "lastModified": 1659877975,
-        "narHash": "sha256-oc6E6ByIw3oJaIyc67maaFcnjYOz1mMcOtHxbEf9NwQ=",
+        "narHash": "sha256-zllb8aq3YO3h8B/U0/J1WBgAL8EX5yWf5pMj3G0NAmc=",
         "owner": "numtide",
         "repo": "flake-utils",
-        "rev": "bee6a7250dd1b01844a2de7e02e4df7d8a0a206c",
+        "rev": "c0e246b9b83f637f4681389ecabcb2681b4f3af0",
         "type": "github"
       },
       "original": {
@@ -32,11 +32,11 @@
     },
     "nixpkgs": {
       "locked": {
-        "lastModified": 1656753965,
+        "lastModified": 1660485612,
-        "narHash": "sha256-BCrB3l0qpJokOnIVc3g2lHiGhnjUi0MoXiw6t1o8H1E=",
+        "narHash": "sha256-sSLW1KaB1adKTJn9+Ja3h3AaS7QCZyhUKiSUStcLg80=",
         "owner": "NixOS",
         "repo": "nixpkgs",
-        "rev": "0ea7a8f1b939d74e5df8af9a8f7342097cdf69eb",
+        "rev": "6512b21eabb4d52e87ea2edcf31a288e67b2e4f8",
         "type": "github"
       },
       "original": {
@@ -47,11 +47,11 @@
     },
     "nixpkgs_2": {
       "locked": {
-        "lastModified": 1656239181,
+        "lastModified": 1660396586,
-        "narHash": "sha256-wW1xRFBn376yGloXZ4QzBE4hjipMawpV18Lshd9QSPw=",
+        "narHash": "sha256-ePuWn7z/J5p2lO7YokOG1o01M0pDDVL3VrStaPpS5Ig=",
         "owner": "NixOS",
         "repo": "nixpkgs",
-        "rev": "f2537a505d45c31fe5d9c27ea9829b6f4c4e6ac5",
+        "rev": "e105167e98817ba9fe079c6c3c544c6ef188e276",
         "type": "github"
       },
       "original": {
@@ -62,11 +62,11 @@
     },
     "nixpkgs_3": {
       "locked": {
-        "lastModified": 1656492412,
+        "lastModified": 1660546381,
-        "narHash": "sha256-CdTRqPFyE/t/nRfnfN3fxm+OOKjeJdaDbXm7E7k+Fxc=",
+        "narHash": "sha256-rEzCjeWVGhK5AyHxm1zet0lF6+AVSW3JuU5LAU2SMYU=",
         "owner": "NixOS",
         "repo": "nixpkgs",
-        "rev": "72f17cca89ec773e5845f5fa1bffcf0b7a4e2094",
+        "rev": "eb642f80f9aecc19312909e08601a3c2020b5ce2",
         "type": "github"
       },
       "original": {
@@ -81,11 +81,11 @@
         "nixpkgs": "nixpkgs_3"
       },
       "locked": {
-        "lastModified": 1656405165,
+        "lastModified": 1660542485,
-        "narHash": "sha256-p2c9QAnKsiUu0a3iHAkpHkO9jAkMWCbG1HCU7/TrYc0=",
+        "narHash": "sha256-XfklMwJMLB7bLI5ZnQTrNaK7KyBnElLGoWOL++XO3zk=",
         "owner": "nix-community",
         "repo": "poetry2nix",
-        "rev": "abc47c71a4920e654e7b2e4261e3e6399bbe2be6",
+        "rev": "8b6239cf2ded121f8f3d570d2563d69f05d4208f",
         "type": "github"
       },
       "original": {
@@ -107,11 +107,11 @@
         "poetry2nix": "poetry2nix"
       },
       "locked": {
-        "lastModified": 1656499586,
+        "lastModified": 1660547406,
-        "narHash": "sha256-tx99W8VI8xxsnPRK0zZs5V2xFo5/hfhxhrTxCQi/mQ8=",
+        "narHash": "sha256-ZRLGxqHmX5xtCBPCt1pb2E3Z70IEmMT3TA+JruosF4Q=",
         "owner": "vale981",
         "repo": "hiro-flake-utils",
-        "rev": "3c96df6681a2475d131a5f30e7cb02c223b7c567",
+        "rev": "dd1e89c5fabbca9a3dbec91cab3e3e3867da10fc",
         "type": "github"
       },
       "original": {