Parallel

Minimal example of parallel bubble solving

/home/docs/checkouts/readthedocs.org/user_builds/pttools/checkouts/main/pttools/bubble/alpha.py:40: NumbaWarning: The TBB threading layer requires TBB version 2021 update 6 or later i.e., TBB_INTERFACE_VERSION >= 12060. Found TBB_INTERFACE_VERSION = 12050. The TBB threading layer is disabled.
  return alpha_n_max_deflagration_bag(v_wall, n_xi)

import matplotlib.pyplot as plt
import numpy as np

from pttools.analysis import BubbleGridVWAlpha, VwAlphaPlot
from pttools.bubble import get_kappa_omega
from pttools.models import BagModel


def main():
    # Create the arrays of v_wall and alpha_n points that will be used for the grid
    v_walls = np.linspace(0.05, 0.95, 20)
    alpha_ns = np.linspace(0.05, 0.3, 20)
    # Create the equation of state
    model = BagModel(a_s=1.1, a_b=1, V_s=1)

    # Parallel computation
    grid = BubbleGridVWAlpha(model, v_walls, alpha_ns, get_kappa_omega)
    # bubbles = grid.bubbles
    kappas = grid.data[0]
    # omegas = grid.data[1]

    # Plotting
    plot = VwAlphaPlot(grid)
    plot.contourf_plusminus(kappas, label=r"$\kappa$")
    plot.chapman_jouguet()


if __name__ == "__main__":
    main()
    plt.show()

Estimated memory usage: 429 MB