Source code for pyro.viscous_burgers.problems.converge

import sys

import numpy

from pyro.mesh import patch
from pyro.util import msg




def init_data(my_data, rp):
    """ initialize the smooth burgers convergence problem """
    del rp  # this problem doesn't use runtime params

    msg.bold("initializing the smooth burgers convergence problem...")

    # make sure that we are passed a valid patch object
    if not isinstance(my_data, patch.CellCenterData2d):
        print("ERROR: patch invalid in converge.py")
        print(my_data.__class__)
        sys.exit()

    u = my_data.get_var("x-velocity")
    v = my_data.get_var("y-velocity")

    xmin = my_data.grid.xmin
    xmax = my_data.grid.xmax

    ymin = my_data.grid.ymin
    ymax = my_data.grid.ymax

    xctr = 0.5*(xmin + xmax)
    yctr = 0.5*(ymin + ymax)

    # A represents some magnitude that defines the initial u and v.

    A = 0.05

    u[:, :] = A + A * numpy.exp(-50.0*((my_data.grid.x2d-xctr)**2 +
                                        (my_data.grid.y2d-yctr)**2))
    v[:, :] = A + A * numpy.exp(-50.0*((my_data.grid.x2d-xctr)**2 +
                                        (my_data.grid.y2d-yctr)**2))





def finalize():
    """ print out any information to the user at the end of the run """