if show:
plot.show()
plot.close(fig) # Close Figure (to avoid too many figures)
i = frame
choose_axis(i, "normal")
choose_axis(i, "zoom")
##### Plot One File or All Files #####
if len(sys.argv) > 1:
frame_number = int(sys.argv[1])
if frame_number == -1:
# Plot Sample
max_frame = util.find_max_frame()
sample = np.linspace(10, max_frame, 10) # 10 evenly spaced frames
for i in sample:
make_plot(i)
else:
# Plot Single
make_plot(frame_number, show=True)
else:
# Search for maximum frame
density_files = glob.glob("gasdens*.dat")
max_frame = find_max_frame()
num_frames = max_frame + 1
#for i in range(num_frames):
# make_plot(i)
peak_rad, peak_rad_index, peak_density = find_radial_peak(averagedDensity)
# Find Peaks in Azimuthal Profiles (and center around that peak)
density_peak_theta, density_theta_index, max_density = find_azimuthal_extrema(density[peak_rad_index], maximum = True) # Max
vortensity_peak_theta, vortensity_peak_theta_index, min_vortensity = find_azimuthal_extrema(vortensity[peak_rad_index], maximum = False) # Min
print "%d, %.2f, %.3f" % (frame, max_density, min_vortensity)
#return max_density, min_vortensity
maximum_densities[i] = max_density
minimum_vortensities[i] = min_vortensity
## Use These Frames ##
rate = 5 # 5 works better, but is very slow
start_of_vortex = 0
max_frame = util.find_max_frame()
if max_frame > 6000:
max_frame = np.max([3000, 1.5 * int(mass_taper)])
frame_range = range(start_of_vortex, max_frame, rate)
maximum_densities = mp_array("d", len(frame_range))
minimum_vortensities = mp_array("d", len(frame_range))
#for i, frame in enumerate(frame_range):
# find_extrema(i, frame)
pool_args = [(i, frame) for i, frame in enumerate(frame_range)]
p = Pool(10)
p.map(find_extrema, pool_args)
p.terminate()
