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()