import sys
import numpy as np
from pyro.mesh import patch
from pyro.util import msg
[docs]
def init_data(my_data, rp):
""" initialize the quadrant problem """
msg.bold("initializing the quadrant problem...")
# make sure that we are passed a valid patch object
if not isinstance(my_data, patch.CellCenterData2d):
print("ERROR: patch invalid in quad.py")
print(my_data.__class__)
sys.exit()
# get the height, momenta, and energy as separate variables
h = my_data.get_var("height")
xmom = my_data.get_var("x-momentum")
ymom = my_data.get_var("y-momentum")
X = my_data.get_var("fuel")
# initialize the components, remember, that ener here is
# h*eint + 0.5*h*v**2, where eint is the specific
# internal energy (erg/g)
r1 = rp.get_param("quadrant.h1")
u1 = rp.get_param("quadrant.u1")
v1 = rp.get_param("quadrant.v1")
r2 = rp.get_param("quadrant.h2")
u2 = rp.get_param("quadrant.u2")
v2 = rp.get_param("quadrant.v2")
r3 = rp.get_param("quadrant.h3")
u3 = rp.get_param("quadrant.u3")
v3 = rp.get_param("quadrant.v3")
r4 = rp.get_param("quadrant.h4")
u4 = rp.get_param("quadrant.u4")
v4 = rp.get_param("quadrant.v4")
cx = rp.get_param("quadrant.cx")
cy = rp.get_param("quadrant.cy")
# there is probably an easier way to do this, but for now, we
# will just do an explicit loop. Also, we really want to set
# the pressure and get the internal energy from that, and then
# compute the total energy (which is what we store). For now
# we will just fake this
myg = my_data.grid
iq1 = np.logical_and(myg.x2d >= cx, myg.y2d >= cy)
iq2 = np.logical_and(myg.x2d < cx, myg.y2d >= cy)
iq3 = np.logical_and(myg.x2d < cx, myg.y2d < cy)
iq4 = np.logical_and(myg.x2d >= cx, myg.y2d < cy)
# quadrant 1
h[iq1] = r1
xmom[iq1] = r1*u1
ymom[iq1] = r1*v1
X[iq1] = 1.0
# quadrant 2
h[iq2] = r2
xmom[iq2] = r2*u2
ymom[iq2] = r2*v2
X[iq2] = 0.0
# quadrant 3
h[iq3] = r3
xmom[iq3] = r3*u3
ymom[iq3] = r3*v3
X[iq3] = 1.0
# quadrant 4
h[iq4] = r4
xmom[iq4] = r4*u4
ymom[iq4] = r4*v4
X[iq4] = 0.0
X *= h
[docs]
def finalize():
""" print out any information to the user at the end of the run """