#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 #####
