#frame_range = np.array(range(start, max_frame + 1, rate)) rossby_number_over_time = mp_array("d", len(frame_range)) rossby_number_over_time_98 = mp_array("d", len(frame_range)) rossby_number_over_time_95 = mp_array("d", len(frame_range)) #maxima_over_time = mp_array("d", len(frame_range)) #minima_over_time = mp_array("d", len(frame_range)) #contrasts_over_time = mp_array("d", len(frame_range)) #for i, frame in enumerate(frame_range): # get_contrasts((i, frame)) pool_args = [(i, frame) for i, frame in enumerate(frame_range)] p = Pool(num_cores) p.map(get_contrasts, pool_args) p.terminate() ##### Helper Functions ##### def make_patch_spines_invisible(ax): ax.set_frame_on(True) ax.patch.set_visible(False) for sp in ax.spines.values(): sp.set_visible(False) ############################################################################### ##### PLOTTING #####
if version is None: save_fn = "%s/azimuthalVelocityProfile_%04d.png" % (save_directory, frame) else: save_fn = "%s/v%04d_azimuthalVelocityProfile_%04d.png" % ( save_directory, version, frame) plot.savefig(save_fn, bbox_inches='tight', dpi=dpi) if show: plot.show() plot.close(fig) # Close Figure (to avoid too many figures) ##### Make Plots! ##### # Iterate through frames if len(frame_range) == 1: make_plot(frame_range[0], show=show) else: if num_cores > 1: p = Pool( num_cores ) # default number of processes is multiprocessing.cpu_count() p.map(make_plot, frame_range) p.terminate() else: for frame in frame_range: make_plot(frame)
if version is None: save_fn = "%s/verticalShearMap_%04d-z%04d.png" % (save_directory, frame, z_level) else: save_fn = "%s/v%04d_verticalShearMap_%04d-z%04d.png" % ( save_directory, version, frame, z_level) plot.savefig(save_fn, bbox_inches='tight', dpi=dpi) if show: plot.show() plot.close(fig) # Close Figure (to avoid too many figures) ##### Make Plots! ##### # Iterate through frames if len(zs_range) == 1: make_plot(zs_range[0], show=show) else: if num_cores > 1: p = Pool( num_cores ) # default number of processes is multiprocessing.cpu_count() p.map(make_plot, zs_range) p.terminate() else: for z_level in zs_range: make_plot(z_level)
#mass_over_time = np.zeros(len(frame_range)) #peak_over_time = np.zeros(len(frame_range)) mass_over_time = mp_array("d", len(frame_range)) peak_over_time = mp_array("d", len(frame_range)) mass_over_time_compare = mp_array("d", len(frame_range)) peak_over_time_compare = mp_array("d", len(frame_range)) #for i, frame in enumerate(frame_range): # get_excess_mass((i, frame)) pool_args = [(i, frame) for i, frame in enumerate(frame_range)] p = Pool(num_cores) p.map(get_excess_mass, pool_args) p.terminate() max_mass = np.max(mass_over_time) max_peak = np.max(peak_over_time) if args.compare: max_mass_compare = np.max(mass_over_time_compare) max_peak_compare = np.max(peak_over_time_compare) ## Pickle to combine later ## pickle.dump(np.array(frame_range), open("dust_variance_frames.p", "wb")) pickle.dump(np.array(mass_over_time), open("dust_variance_values.p", "wb")) ##### PLOTTING #####
#else: # density, new_sizes = distribute_density(density) density = convert_units(density) output_density_txt(density, frame) output_density_pickle(density, frame) if frame == frame_range[0]: generate_secondary_files(new_rad, new_theta, sizes) save_id_parameters() ############################################################################### ### Generate Synthetic Input ### # Iterate through frames if len(frame_range) == 1: full_procedure(frame_range[0]) else: if num_cores > 1: p = Pool( num_cores ) # default number of processes is multiprocessing.cpu_count() p.map(full_procedure, frame_range) p.terminate() else: for frame in frame_range: full_procedure(frame)
peak_rad_over_time = mp_array("d", len(frame_range)) rossby_rad_difference_over_time = mp_array("d", len(frame_range)) inner_peak_difference_over_time = mp_array("d", len(frame_range)) outer_peak_difference_over_time = mp_array("d", len(frame_range)) inner_rossby_value_over_time = mp_array("d", len(frame_range)) outer_rossby_value_over_time = mp_array("d", len(frame_range)) #for i, frame in enumerate(frame_range): # get_rossby_criteria((i, frame)) pool_args = [(i, frame) for i, frame in enumerate(frame_range)] p = Pool(num_cores) p.map(get_rossby_criteria, pool_args) p.terminate() ##### Helper Functions ##### def make_patch_spines_invisible(ax): ax.set_frame_on(True) ax.patch.set_visible(False) for sp in ax.spines.values(): sp.set_visible(False) ############################################################################### ##### PLOTTING #####
max_frame = 100 #util.find_max_frame() #frame_range = np.array(range(start, max_frame + 1, rate)) azimuthal_extent_over_time = mp_array("d", len(frame_range)) radial_extent_over_time = mp_array("d", len(frame_range)) radial_peak_over_time = mp_array("d", len(frame_range)) radial_peak_over_time_a = mp_array("d", len(frame_range)) #contrasts_over_time = mp_array("d", len(frame_range)) #for i, frame in enumerate(frame_range): # get_extents((i, frame)) pool_args = [(i, frame) for i, frame in enumerate(frame_range)] p = Pool(num_cores) p.map(get_extents, pool_args) p.terminate() ##### Helper Functions ##### def make_patch_spines_invisible(ax): ax.set_frame_on(True) ax.patch.set_visible(False) for sp in ax.spines.values(): sp.set_visible(False) ############################################################################### ##### PLOTTING #####