def do_rr(self, rest):
#self.plotdata.refresh_frames=True
outdir = os.path.abspath(self.plotdata.outdir)
key = (self.frameno, outdir)
xxx = self.plotdata.framesoln_dict.pop(key,None)
if xxx is None:
print 'No frame data to clear for frame ',self.frameno
else:
print 'Cleared data for frame ',self.frameno
print 'Reading data from outdir = ',self.plotdata.outdir
self.current_data = frametools.plotframe(self.frameno, self.plotdata)
self.plotdata.refresh_frames=False
def preloop(self):
print('\nInteractive plotting for Clawpack output... ')
print(('\nPlotting data from outdir = ', self.plotdata.outdir))
print('Type ? at PLOTCLAW prompt for list of commands')
startframeno = eval(input('\n Start at which frame [default=%i] ? '\
% self.prevframeno))
makeplot = True
if startframeno == '':
self.frameno = self.prevframeno
self.plotdata = self.prevplotdata
if self.restart:
replot = eval(input(' Replot data for frame %s [no] ? ' \
% self.frameno))
if replot not in ['y', 'yes', 'Y']:
makeplot = False
if makeplot:
reload = eval(input(' Reload data for frame %s [no] ? ' \
% self.frameno))
if reload in ['y', 'yes', 'Y']:
self.plotdata.refresh_frames = True
else:
self.plotdata.refresh_frames = False
else:
self.plotdata = self.prevplotdata
try:
self.frameno = int(startframeno)
except:
print(
'\n *** Error: frameno must be an integer, resetting to 0'
)
self.frameno = 0
if makeplot:
self.current_data = frametools.plotframe(self.frameno,
self.plotdata)
self.plotdata.refresh_frames = False
#pylab.draw()
self.lastcmd = 'n'
self.restart = True
def do_j(self, rest):
try:
newframeno = int(rest)
except:
newframeno = raw_input('\n Jump to which frame? ')
if newframeno == 'n':
self.do_n(rest)
self.lastcmd = 'n'
elif newframeno == 'p':
self.do_p(rest)
self.lastcmd = 'p'
else:
try:
newframeno = int(newframeno)
self.frameno = newframeno
self.current_data = frametools.plotframe(self.frameno, self.plotdata)
except:
print '\n *** Error: frameno must be an integer, n, or p'
def preloop(self):
print '\nInteractive plotting for Clawpack output... '
print '\nPlotting data from outdir = ', self.plotdata.outdir
print 'Type ? at PLOTCLAW prompt for list of commands'
startframeno = raw_input('\n Start at which frame [default=%i] ? '\
% self.prevframeno)
makeplot = True
if startframeno == '':
self.frameno = self.prevframeno
self.plotdata = self.prevplotdata
if self.restart:
replot = raw_input(' Replot data for frame %s [no] ? ' \
% self.frameno)
if replot not in ['y','yes','Y']:
makeplot = False
if makeplot:
reload = raw_input(' Reload data for frame %s [no] ? ' \
% self.frameno)
if reload in ['y','yes','Y']:
self.plotdata.refresh_frames = True
else:
self.plotdata.refresh_frames = False
else:
self.plotdata = self.prevplotdata
try:
self.frameno = int(startframeno)
except:
print '\n *** Error: frameno must be an integer, resetting to 0'
self.frameno = 0
if makeplot:
self.current_data = frametools.plotframe(self.frameno, self.plotdata)
self.plotdata.refresh_frames=False
#pylab.draw()
self.lastcmd = 'n'
self.restart = True
def do_r(self, rest):
self.current_data = frametools.plotframe(self.frameno, self.plotdata)
def do_p(self, rest):
#print ' frameno = ',self.frameno
self.frameno = max(self.frameno-1, 0)
self.current_data = frametools.plotframe(self.frameno, self.plotdata)
def do_n(self, rest):
from pyclaw.plotters import frametools, data
#print ' frameno = ',self.frameno
self.frameno = self.frameno+1
self.current_data = frametools.plotframe(self.frameno, self.plotdata)
pylab.draw()
