'source_strengths': strength_list,
'epsilon': 0.01,
'trap_radius': trap_radius,
'source_radius': radius_sources
}
traps = trap_models.TrapModel(trap_param)
#Import wind and odor fields
wind_file = '/home/annie/work/programming/pompy_duplicate/' + sys.argv[1]
plume_file = '/home/annie/work/programming/pompy_duplicate/' + sys.argv[2]
try:
conc_file = '/home/annie/work/programming/pompy_duplicate/' + sys.argv[
3]
importedConc = data_importers.ImportedConc(conc_file, release_delay)
plotting = True
except (IndexError):
plotting = False
plotting = True
importedWind = data_importers.ImportedWind(wind_file, release_delay)
wind_params = importedWind.run_param
array_z = 0.01
array_dim_x = 1000
array_dim_y = array_dim_x
puff_mol_amount = 1.
importedPlumes = data_importers.ImportedPlumes(plume_file,
import odor_tracking_sim.swarm_models as swarm_models import odor_tracking_sim.trap_models as trap_models import odor_tracking_sim.utility as utility import odor_tracking_sim.simulation_running_tools as srt import data_importers file_name = 'imported_plumes_test' fig = plt.figure(1) fig.set_size_inches(7.5, 9, True) ax = fig.add_subplot(111) plt.pause(0.001) t = 0.0 conc_file = '/home/annie/work/programming/pompy_duplicate/' + sys.argv[1] importedConc = data_importers.ImportedConc(conc_file) image = importedConc.plot(0) xmin, xmax, ymin, ymax = image.get_extent() ax.set_xlim(xmin, xmax) ax.set_ylim(ymin, ymax) # wind_file = '/home/annie/work/programming/pompy_duplicate/'+sys.argv[2] # importedWind = data_importers.ImportedWind(wind_file) # x_coords,y_coords = importedWind.get_plotting_points() # u,v = importedWind.quiver_at_time(0) # vector_field = ax.quiver(x_coords,y_coords,u,v) dt = 0.25 frame_rate = 20 times_real_time = 5 # seconds of simulation / sec in video capture_interval = times_real_time * int((1. / frame_rate) / dt)