Source code for pyro.viscous_burgers.problems.tophat

import sys

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




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

    msg.bold("initializing the tophat burgers problem...")

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

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

    xmin = myd.grid.xmin
    xmax = myd.grid.xmax

    ymin = myd.grid.ymin
    ymax = myd.grid.ymax

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

    u[:, :] = 0.0
    v[:, :] = 0.0

    R = 0.1

    inside = (myd.grid.x2d - xctr)**2 + (myd.grid.y2d - yctr)**2 < R**2

    u[inside] = 1.0
    v[inside] = 1.0





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