

def dovis(self):
    ' \n        Do runtime visualization\n        '
    pylab.clf()
    pylab.rc('font', size=10)
    u = self.cc_data.get_var('x-velocity')
    v = self.cc_data.get_var('y-velocity')
    myg = self.cc_data.grid
    vort = myg.scratch_array()
    divU = myg.scratch_array()
    vort[myg.ilo:(myg.ihi + 1), myg.jlo:(myg.jhi + 1)] = (((0.5 * (v[(myg.ilo + 1):(myg.ihi + 2), myg.jlo:(myg.jhi + 1)] - v[(myg.ilo - 1):myg.ihi, myg.jlo:(myg.jhi + 1)])) / myg.dx) - ((0.5 * (u[myg.ilo:(myg.ihi + 1), (myg.jlo + 1):(myg.jhi + 2)] - u[myg.ilo:(myg.ihi + 1), (myg.jlo - 1):myg.jhi])) / myg.dy))
    divU[myg.ilo:(myg.ihi + 1), myg.jlo:(myg.jhi + 1)] = (((0.5 * (u[(myg.ilo + 1):(myg.ihi + 2), myg.jlo:(myg.jhi + 1)] - u[(myg.ilo - 1):myg.ihi, myg.jlo:(myg.jhi + 1)])) / myg.dx) + ((0.5 * (v[myg.ilo:(myg.ihi + 1), (myg.jlo + 1):(myg.jhi + 2)] - v[myg.ilo:(myg.ihi + 1), (myg.jlo - 1):myg.jhi])) / myg.dy))
    (fig, axes) = pylab.subplots(nrows=2, ncols=2, num=1)
    pylab.subplots_adjust(hspace=0.25)
    fields = [u, v, vort, divU]
    field_names = ['u', 'v', '$\\nabla \\times U$', '$\\nabla \\cdot U$']
    for n in range(4):
        ax = axes.flat[n]
        f = fields[n]
        img = ax.imshow(numpy.transpose(f[myg.ilo:(myg.ihi + 1), myg.jlo:(myg.jhi + 1)]), interpolation='nearest', origin='lower', extent=[myg.xmin, myg.xmax, myg.ymin, myg.ymax])
        ax.set_xlabel('x')
        ax.set_ylabel('y')
        ax.set_title(field_names[n])
        pylab.colorbar(img, ax=ax)
    pylab.figtext(0.05, 0.0125, ('t = %10.5f' % self.cc_data.t))
    pylab.draw()

dataset/ETHPy150Open zingale/pyro2/lm_combustion/simulation.py Simulation.dovis/original