tunnel_primal.py

################################################################################
#                                                                              #
#       tunnel_primal.py copyright 2015 Qiqi Wang (qiqi.wang@gmail.com)        #
#                                                                              #
################################################################################

import matplotlib
matplotlib.use('agg')
matplotlib.interactive(False)

import pdb
import sys
import time
import argparse
from pylab import *

from tunnel import *

# set parameters
s = 0.1 * np.ones(2)

parser = argparse.ArgumentParser()
parser.add_argument('--restart', type=str, default='')
args = parser.parse_args()

if len(args.restart) == 0:
    w = grid.zeros([4]) + w0
    w[:] *= 1 + 0.01 * (grid.random() - 0.5)
else:
    print('restarting from ', args.restart)
    w = load(args.restart)
    assert w.shape == (Nx, Ny, 4)


fout = open('force_hist.txt','w')

figure(figsize=(28,10))
for iplot in range(200):
    for iprint in range(nPrintsPerPlot):
        for istep in range(nStepPerPrint):
            w = step(w,s)
        print('%f %f' % tuple(force(w,s)))
        
        fout.write('%f %f\n' % tuple(force(w,s)))

        sys.stdout.flush()
    w.save('w{0:06d}.npy'.format(iplot))
    r, ru, rv, p = w
    rho, u, v = r * r, ru / r, rv / r
    clf()
    u_range = linspace(-u0, u0 * 2, 100)
    subplot(2,1,1); contourf(x._data.T, y._data.T, u._data.T,
            u_range, extend='both')
    axis('scaled'); colorbar()
    v_range = linspace(-u0, u0, 100)
    subplot(2,1,2); contourf(x._data.T, y._data.T, v._data.T,
            v_range, extend='both')
    axis('scaled'); colorbar()
    savefig('fig{0:06d}.png'.format(iplot))

fout.close()

################################################################################
################################################################################
################################################################################