Ejemplo n.º 1
0
Archivo: plot.py Proyecto: BSAM/BSAM
import matplotlib.pyplot as plt
import numpy
import bsam_io
import bsam_plot

# Create plots
fig = plt.figure()

ax  = fig.add_subplot(211)
data = bsam_io.parse_adaptive_data_file("out/m00000.dat")
root_patch = data['levels'][0][0]
bsam_plot.plot_grid(ax, data, 5)
bsam_plot.plot_interface(ax, root_patch)
bsam_plot.plot_contourf(ax, root_patch, 0)

ax  = fig.add_subplot(212)
data = bsam_io.parse_adaptive_data_file("out/m00010.dat")
root_patch = data['levels'][0][0]
#bsam_plot.plot_grid(ax, data, 5)
#bsam_plot.plot_interface(ax, root_patch)
bsam_plot.plot_contourf(ax, root_patch, 0)

plt.show()
Ejemplo n.º 2
0
import matplotlib.pyplot as plt
import numpy
import bsam_io
import bsam_plot

# Parse datafile
data = bsam_io.parse_adaptive_data_file("out/m00001.dat")
root_patch = data['levels'][0][0]

# Create plots
fig = plt.figure(figsize=(18,14))

ax  = fig.add_subplot(221)
levels = bsam_plot.create_levels(-0.1,0.7,50)
bsam_plot.plot_contourf(ax, root_patch, 0, levels)

ax  = fig.add_subplot(222)
cs1 = bsam_plot.plot_contourf(ax, root_patch, 3, levels)
fig.colorbar(cs1)
#bsam_plot.plot_bounding_boxes(ax, data)

ax  = fig.add_subplot(223)
bsam_plot.plot_xprofile(ax, root_patch,
                        [(0, 'k'),
                         (3, 'k--')],
                        [(-0.05,0.3),
                         (-1.0,1.0)])

ax  = fig.add_subplot(224)
bsam_plot.plot_xprofile(ax, root_patch,
                        [(0, 'k'),