def start_SYGMA(): frame = framework.framework() frame.set_default_display_style(padding="0.25em", background_color="white", border_color="LightGrey", border_radius="0.5em") frame.set_default_io_style(padding="0.25em", margin="0.25em", border_color="LightGrey", border_radius="0.5em") tablist = ["sim_page", "plot_page", "custom_imf_page", "get_table_page"] yield_list = { "Analytic perscription": { "Delay": "yield_tables/isotope_yield_table_MESA_only_fryer12_delay.txt", "Rapid": "yield_tables/isotope_yield_table_MESA_only_fryer12_rapid.txt", "Exclude neutron-alpha rich freeze-out": "yield_tables/isotope_yield_table_MESA_only_fryer12_exclnalpha.txt" }, "Ye=0.4982": { "Fallback at Ye": "yield_tables/isotope_yield_table_MESA_only_ye.txt", "Fallback motivated by GCE": "yield_tables/isotope_yield_table_MESA_only_ye_fallback.txt" } } group_style = {"border_style": "none", "border_radius": "0em"} text_box_style = {"width": "10em"} button_style = {"font_size": "1.25em", "font_weight": "bold"} first_tab_style = {"border_radius": "0em 0.5em 0.5em 0.5em"} custom_imf_dir = os.environ["SYGMADIR"] + "/SYGMA_widget_imfs/" default_custom_imf_text = "\n#File to define a custom IMF\n#Define your IMF in custom_imf\n#so that the return value represents\n#the chosen IMF value for the input mass\n\ndef custom_imf(mass):\n #Salpeter IMF\n return mass**-2.35\n" states_plot = [ "plot_totmasses", "plot_mass", "plot_spectro", "plot_mass_range" ] states_sim_plot = ["run_sim"] + states_plot states_cimf = ["custom_imf", "load_custom_imf"] states = states_sim_plot + states_cimf frame.add_state(states) isotopes_all = [ 'H-1', 'H-2', 'He-3', 'He-4', 'Li-7', 'B-11', 'C-12', 'C-13', 'N-14', 'N-15', 'O-16', 'O-17', 'O-18', 'F-19', 'Ne-20', 'Ne-21', 'Ne-22', 'Na-23', 'Mg-24', 'Mg-25', 'Mg-26', 'Al-27', 'Si-28', 'Si-29', 'Si-30', 'P-31', 'S-32', 'S-33', 'S-34', 'S-36', 'Cl-35', 'Cl-37', 'Ar-36', 'Ar-38', 'Ar-40', 'K-39', 'K-40', 'K-41', 'Ca-40', 'Ca-42', 'Ca-43', 'Ca-44', 'Ca-46', 'Ca-48', 'Sc-45', 'Ti-46', 'Ti-47', 'Ti-48', 'Ti-49', 'Ti-50', 'V-50', 'V-51', 'Cr-50', 'Cr-52', 'Cr-53', 'Cr-54', 'Mn-55', 'Fe-54', 'Fe-56', 'Fe-57', 'Fe-58', 'Co-59', 'Ni-58', 'Ni-60', 'Ni-61', 'Ni-62', 'Ni-64' ] elements_all = [ 'H', 'He', 'Li', 'B', 'C', 'N', 'O', 'F', 'Ne', 'Na', 'Mg', 'Al', 'Si', 'P', 'S', 'Cl', 'Ar', 'K', 'Ca', 'Sc', 'Ti', 'V', 'Cr', 'Mn', 'Fe', 'Co', 'Ni' ] isotopes_sn1a = [ 'C-12', 'C-13', 'N-14', 'N-15', 'O-16', 'O-17', 'O-18', 'F-19', 'Ne-20', 'Ne-21', 'Ne-22', 'Na-23', 'Mg-24', 'Mg-25', 'Mg-26', 'Al-27', 'Si-28', 'Si-29', 'Si-30', 'P-31', 'S-32', 'S-33', 'S-34', 'S-36', 'Cl-35', 'Cl-37', 'Ar-36', 'Ar-38', 'Ar-40', 'K-39', 'K-40', 'K-41', 'Ca-40', 'Ca-42', 'Ca-43', 'Ca-44', 'Ca-46', 'Ca-48', 'Sc-45', 'Ti-46', 'Ti-47', 'Ti-48', 'Ti-49', 'Ti-50', 'V-50', 'V-51', 'Cr-50', 'Cr-52', 'Cr-53', 'Cr-54', 'Mn-55', 'Fe-54', 'Fe-56', 'Fe-57', 'Fe-58', 'Co-59', 'Ni-58', 'Ni-60', 'Ni-61', 'Ni-62', 'Ni-64' ] elements_sn1a = [ 'C', 'N', 'O', 'F', 'Ne', 'Na', 'Mg', 'Al', 'Si', 'P', 'S', 'Cl', 'Ar', 'K', 'Ca', 'Sc', 'Ti', 'V', 'Cr', 'Mn', 'Fe', 'Co', 'Ni' ] isotopes_sel_mult = ["All"] + isotopes_all elements_sel_mult = ["All"] + elements_all line_styles = ["-", "--", ":", "-."] # line_colors=["k", "r", "g", "b", "c", "m", "y"] line_colors = ["k", "r", "b", "c", "m", "y"] #remove "g" line_markers = ["o", "s", "x", "D", "v", "^", "<", ">", "p", "*", "+"] styles = auto_styles() styles.set_line_styles(line_styles) styles.set_line_colors(line_colors) styles.set_line_markers(line_markers) frame.set_state_data("elements", elements_all) frame.set_state_data("isotopes", isotopes_all) frame.set_state_data("over_plotting_data", []) frame.set_state_data("styles", styles) frame.set_state_data("old_state", None) frame.set_state_data("runs", []) frame.set_state_data("run_count", 0) def add_run(data, name, Z): run_count = frame.get_state_data("run_count") runs_data = frame.get_state_data("runs") widget_name = "runs_widget_#" + str(run_count) frame.add_io_object(widget_name) frame.set_state_attribute(widget_name, visible=True, description=name) frame.set_object(widget_name, widgets.Checkbox()) runs_data.append((data, name, Z, widget_name)) frame.set_state_children("runs", [widget_name]) frame.set_state_data("runs", runs_data) run_count += 1 frame.set_state_data("run_count", run_count) def remove_runs(): data = frame.get_state_data("runs") children = ["runs_title"] tmp_data = [] for i in xrange(len(data)): widget_name = data[i][3] if frame.get_attribute(widget_name, "value"): frame.remove_object(widget_name) else: children.append(widget_name) tmp_data.append(data[i]) data = tmp_data frame.set_state_children("runs", children, append=False) frame.set_state_data("runs", data) def list_custom_imf(): if os.path.isdir(custom_imf_dir): files_dirs = os.listdir(custom_imf_dir) imfs = [] for file in files_dirs: file_full = custom_imf_dir + file if os.path.isfile(file_full): file = file.rsplit(".", 1) if file[1] == "py": imfs.append(file[0]) return imfs else: return [] frame.add_display_object("window") frame.add_io_object("title") frame.add_display_object("widget_runs_group") frame.add_display_object("widget") frame.add_display_object("runs") frame.add_io_object("runs_title") ###Sim page### frame.add_display_object("sim_page") frame.add_display_object("mass_Z_group") frame.add_io_object("init_Z") frame.add_io_object("mass_gas") frame.add_display_object("time_group") frame.add_io_object("t_end") frame.add_io_object("dt") frame.add_display_object("imf_type_group") frame.add_io_object("imf_type") frame.add_io_object("imf_alpha") frame.add_display_object("imf_mass_group") frame.add_io_object("imf_mass_min") frame.add_io_object("imf_mass_max") frame.add_display_object("sn1a_group") frame.add_io_object("use_sn1a") frame.add_io_object("sn1a_rates") frame.add_display_object("yield_table_group") frame.add_io_object("yield_table_selection") frame.add_io_object("yield_table_list") frame.add_display_object("run_sim_remove_run_group") frame.add_io_object("run_sim") frame.add_io_object("remove_run") frame.add_io_object("run_name") frame.add_io_object("plot_type") frame.add_io_object("sim_responce") frame.set_state_children("window", ["title", "widget_runs_group"]) frame.set_state_children("widget_runs_group", ["widget", "runs"]) frame.set_state_children("widget", ["sim_page"], titles=["Simulation"]) frame.set_state_children("sim_page", [ "mass_Z_group", "time_group", "imf_type_group", "imf_mass_group", "sn1a_group", "yield_table_group", "run_sim_remove_run_group", "sim_responce" ]) frame.set_state_children("mass_Z_group", ["mass_gas", "init_Z"]) frame.set_state_children("time_group", ["t_end", "dt"]) frame.set_state_children("imf_type_group", ["imf_type", "imf_alpha"]) frame.set_state_children("imf_mass_group", ["imf_mass_min", "imf_mass_max"]) frame.set_state_children("sn1a_group", ["use_sn1a", "sn1a_rates"]) frame.set_state_children("yield_table_group", ["yield_table_selection", "yield_table_list"]) frame.set_state_children("run_sim_remove_run_group", ["run_sim", "remove_run", "run_name"]) frame.set_state_children("runs", ["runs_title"]) ###plotting page### frame.add_display_object("plot_page") frame.add_io_object("warning_msg") frame.add_io_object("plot_name") frame.add_display_object("source_over_plotting_group") frame.add_io_object("source") frame.add_io_object("over_plotting") frame.add_io_object("clear_plot") frame.add_display_object("species_group") frame.add_io_object("iso_or_elem") frame.add_io_object("species") frame.add_io_object("elem_numer") frame.add_io_object("elem_denom") frame.add_io_object("plot") frame.set_state_children("widget", ["plot_page"], titles=["Plotting"]) frame.set_state_children("plot_page", [ "warning_msg", "plot_type", "plot_name", "source_over_plotting_group", "species_group", "elem_numer", "elem_denom", "plot" ]) frame.set_state_children("source_over_plotting_group", ["source", "over_plotting", "clear_plot"]) frame.set_state_children("species_group", ["iso_or_elem", "species"]) ###Custom IMF page### frame.add_display_object("custom_imf_page") frame.add_display_object("load_save_imf_group") frame.add_io_object("load_imf") frame.add_io_object("list_imfs") frame.add_io_object("name_imf") frame.add_io_object("save_imf") frame.add_io_object("delete_imf") frame.add_io_object("text_imf") frame.add_io_object("test_imf") frame.set_state_children("widget", ["custom_imf_page"], titles=["Custom IMF"]) frame.set_state_children("custom_imf_page", ["load_save_imf_group", "text_imf", "test_imf"]) frame.set_state_children( "load_save_imf_group", ["load_imf", "list_imfs", "name_imf", "save_imf", "delete_imf"]) ###plotting page### frame.add_display_object("get_table_page") frame.add_display_object("species_mult_group") frame.add_io_object("species_mult") frame.add_io_object("get_table") frame.add_io_object("table_links") frame.set_state_children("widget", ["get_table_page"], titles=["Download Tables"]) frame.set_state_children( "get_table_page", ["warning_msg", "species_mult_group", "get_table", "table_links"]) frame.set_state_children("species_mult_group", ["iso_or_elem", "species_mult"]) frame.set_state_attribute('window', visible=True, **group_style) frame.set_state_attribute('title', visible=True, value="<center><h1>SYGMA</h1></center>") frame.set_state_attribute("widget_runs_group", visible=True, **group_style) frame.set_state_attribute('widget', visible=True, **group_style) frame.set_state_attribute("runs", visible=True, margin="3.15em 0em 0em 0em") frame.set_state_attribute("runs", states_cimf, visible=False) frame.set_state_attribute("runs_title", visible=True, value="<center><h2>Runs</h2></center>", **group_style) frame.set_state_attribute('sim_page', visible=True, **first_tab_style) frame.set_state_attribute("mass_Z_group", visible=True, **group_style) frame.set_state_attribute("mass_gas", visible=True, description="Total stellar mass [$M_{\odot}$]:", value="1.0", **text_box_style) frame.set_state_attribute( 'init_Z', visible=True, description="Initial metallicity: ", options=["0.02", "0.01", "0.006", "0.001", "0.0001", "0.0"]) frame.set_state_attribute('time_group', visible=True, **group_style) frame.set_state_attribute('t_end', visible=True, description="Final time [yrs]: ", value="1.0e10", **text_box_style) frame.set_state_attribute('dt', visible=True, description="Time step [yrs]: ", value="1.0e7", **text_box_style) frame.set_state_attribute('imf_type_group', visible=True, **group_style) frame.set_state_attribute( 'imf_type', visible=True, description="IMF type: ", options=['salpeter', 'chabrier', 'kroupa', 'alphaimf'] + list_custom_imf()) frame.set_state_attribute('imf_alpha', description="Set alpha: ", min=0, max=5) frame.set_state_attribute("imf_mass_group", visible=True, **group_style) frame.set_state_attribute('imf_mass_min', visible=True, description="IMF lower limit [$M_{\odot}$]: ", value="1.0", **text_box_style) frame.set_state_attribute('imf_mass_max', visible=True, description="IMF upper limit [$M_{\odot}$]: ", value="30.0", **text_box_style) frame.set_state_attribute('sn1a_group', visible=True, **group_style) frame.set_state_attribute('use_sn1a', visible=True, description="Include SNe Ia: ", value=True) frame.set_state_attribute("yield_table_group", visible=True, **group_style) frame.set_state_attribute("yield_table_selection", visible=True, description="CCSN remnant prescription:", options=["Analytic perscription", "Ye=0.4982"], value="Ye=0.4982") frame.set_state_attribute('yield_table_list', visible=True, options=yield_list["Ye=0.4982"], selected_label="Fallback at Ye") frame.set_state_links("sn1a_link", [("use_sn1a", "value"), ("sn1a_rates", "visible")], directional=True) frame.set_state_attribute( 'sn1a_rates', description="SNe Ia rates: ", options=['Power law', 'Exponential', 'Gaussian', 'Maoz12']) frame.set_state_attribute("run_sim_remove_run_group", visible=True, **group_style) frame.set_state_attribute('run_sim', visible=True, description="Run simulation", **button_style) frame.set_state_attribute("remove_run", visible=True, description="Remove selected", **button_style) frame.set_state_attribute("run_name", visible=True, description="Run name: ", placeholder="Enter name", value="", **text_box_style) frame.set_state_attribute("sim_responce", value="<p>Simulation data loaded.</p>", **group_style) frame.set_state_attribute("sim_responce", states_sim_plot, visible=True) frame.set_state_attribute('plot_type', states_sim_plot, visible=True, description="Plot type: ", options=[ "Total mass", "Species mass", "Species spectroscopic", "Mass range contributions" ]) def mass_gas_handler(name, value): frame.set_attributes("mass_gas", value=float_text(value)) def t_end_handler(name, value): frame.set_attributes("t_end", value=float_text(value)) def dt_handler(name, value): frame.set_attributes("dt", value=float_text(value)) def imf_mass_min_handler(name, value): frame.set_attributes("imf_mass_min", value=float_text(value)) def imf_mass_max_handler(name, value): frame.set_attributes("imf_mass_max", value=float_text(value)) def sel_imf_type(attribute, value): if value == "alphaimf": frame.set_state_attribute("imf_alpha", visible=True) frame.set_attributes("imf_alpha", visible=True, value=2.35) else: frame.set_state_attribute("imf_alpha", visible=False) frame.set_attributes("imf_alpha", visible=False) def yield_table_selection_handler(name, value): frame.set_attributes("yield_table_list", options=yield_list[value]) if value == "Analytic perscription": frame.set_attributes("yield_table_list", selected_label="Delay") elif value == "Ye=0.4982": frame.set_attributes("yield_table_list", selected_label="Fallback at Ye") def run_simulation(widget): frame.set_attributes("sim_responce", visible=False) clear_output() pyplot.close("all") sn1a_map = { "Power law": "power_law", "Gaussian": "gauss", "Exponential": "exp", "Maoz12": "maoz" } mgal = float(frame.get_attribute("mass_gas", "value")) iniZ = float(frame.get_attribute("init_Z", "value")) imf_type = frame.get_attribute("imf_type", "value") alphaimf = frame.get_attribute("imf_alpha", "value") mass_min = float(frame.get_attribute("imf_mass_min", "value")) mass_max = float(frame.get_attribute("imf_mass_max", "value")) imf_bdys = [mass_min, mass_max] sn1a_on = frame.get_attribute("use_sn1a", "value") sn1a_rate = sn1a_map[frame.get_attribute("sn1a_rates", "value")] dt = float(frame.get_attribute("dt", "value")) tend = float(frame.get_attribute("t_end", "value")) yield_table = frame.get_attribute("yield_table_list", "value") if not (imf_type in ['salpeter', 'chabrier', 'kroupa', 'alphaimf']): destination = os.environ["SYGMADIR"] + "/imf_input.py" shutil.copy(custom_imf_dir + imf_type + ".py", destination) imf_type = "input" run_count = frame.get_state_data("run_count") name = frame.get_attribute("run_name", "value") if name == "": name = "Run - " + "%03d" % (run_count + 1, ) if iniZ == 0.0: data = s.sygma(mgal=mgal, iniZ=iniZ, imf_type=imf_type, alphaimf=alphaimf, imf_bdys=[10.1, 100.0], imf_bdys_pop3=imf_bdys, sn1a_on=sn1a_on, sn1a_rate=sn1a_rate, dt=dt, tend=tend, table=yield_table) else: data = s.sygma(mgal=mgal, iniZ=iniZ, imf_type=imf_type, alphaimf=alphaimf, imf_bdys=imf_bdys, sn1a_on=sn1a_on, sn1a_rate=sn1a_rate, dt=dt, tend=tend, table=yield_table) frame.set_state("run_sim") ##force reset plottype frame.set_attributes("plot_type", selected_label="Species mass", value="Species mass") frame.set_attributes("plot_type", selected_label="Total mass", value="Total mass") add_run(data, name, iniZ) frame.update() def remove_simulation(widget): remove_runs() frame.update() def sel_tab(name, value): open_tab = tablist[value] if open_tab == "custom_imf_page": frame.set_state_data("old_state", frame.get_state()) frame.set_state("custom_imf") if frame.get_attribute("name_imf", "value") == "": frame.set_attributes("text_imf", value=default_custom_imf_text) else: if frame.get_state() in states_cimf: old_state = frame.get_state_data("old_state") if old_state != None: frame.set_state(old_state) else: frame.set_state("default") frame.set_state_data("old_state", None) frame.set_state_callbacks("mass_gas", mass_gas_handler) frame.set_state_callbacks("t_end", t_end_handler) frame.set_state_callbacks("dt", dt_handler) frame.set_state_callbacks("imf_mass_min", imf_mass_min_handler) frame.set_state_callbacks("imf_mass_max", imf_mass_max_handler) frame.set_state_callbacks("imf_type", sel_imf_type) frame.set_state_callbacks("yield_table_selection", yield_table_selection_handler) frame.set_state_callbacks("run_sim", run_simulation, attribute=None, type="on_click") frame.set_state_callbacks("remove_run", remove_simulation, attribute=None, type="on_click") frame.set_state_callbacks("widget", sel_tab, "selected_index") frame.set_object("window", widgets.Box()) frame.set_object("title", widgets.HTML()) frame.set_object("widget_runs_group", widgets.HBox()) frame.set_object("widget", widgets.Tab()) frame.set_object("runs", widgets.VBox()) frame.set_object("runs_title", widgets.HTML()) frame.set_object("sim_page", widgets.VBox()) frame.set_object("mass_Z_group", widgets.HBox()) frame.set_object("mass_gas", widgets.Text()) frame.set_object("init_Z", widgets.Dropdown()) frame.set_object("time_group", widgets.HBox()) frame.set_object("t_end", widgets.Text()) frame.set_object("dt", widgets.Text()) frame.set_object("imf_type_group", widgets.HBox()) frame.set_object("imf_type", widgets.Dropdown()) frame.set_object("imf_alpha", widgets.FloatSlider()) frame.set_object("imf_mass_group", widgets.HBox()) frame.set_object("imf_mass_min", widgets.Text()) frame.set_object("imf_mass_max", widgets.Text()) frame.set_object("sn1a_group", widgets.HBox()) frame.set_object("use_sn1a", widgets.Checkbox()) frame.set_object("sn1a_rates", widgets.Dropdown()) frame.set_object("yield_table_group", widgets.HBox()) frame.set_object("yield_table_selection", widgets.ToggleButtons()) frame.set_object("yield_table_list", widgets.Dropdown()) frame.set_object("run_sim_remove_run_group", widgets.HBox()) frame.set_object("run_sim", widgets.Button()) frame.set_object("remove_run", widgets.Button()) frame.set_object("run_name", widgets.Text()) frame.set_object("sim_responce", widgets.HTML()) frame.set_object("plot_type", widgets.Dropdown()) frame.set_state_attribute("plot_page", visible=True) frame.set_state_attribute("warning_msg", visible=True, value="<h3>Error: No simulation data!</h3>", **group_style) frame.set_state_attribute("warning_msg", states_plot, visible=False) frame.set_state_attribute("plot_name", **group_style) frame.set_state_attribute("plot_name", "plot_totmasses", visible=True, value="<h2>Plot: Total mass evolution</h2>") frame.set_state_attribute("plot_name", "plot_mass", visible=True, value="<h2>Plot: Species mass evolution</h2>") frame.set_state_attribute( "plot_name", "plot_spectro", visible=True, value="<h2>Plot: Spectroscopic Mass evolution</h2>") frame.set_state_attribute( "plot_name", "plot_mass_range", visible=True, value= "<h2>Plot: Mass range contributions</h2><p>Only ejecta from AGB and massive stars are considered.</p>" ) frame.set_state_attribute("source_over_plotting_group", states_plot, visible=True, **group_style) frame.set_state_attribute("source", ["plot_totmasses", "plot_mass", "plot_spectro"], visible=True, description="Yield source: ", options=["All", "AGB", "SNe Ia", "Massive"], selected_label="All") frame.set_state_attribute("over_plotting", visible=True, description="Over plotting", value=False, **button_style) frame.set_state_attribute("clear_plot", description="Clear plot", **button_style) frame.set_state_links("clear_plot_link", [("over_plotting", "value"), ("clear_plot", "visible")], directional=True) frame.set_state_attribute("species_group", ["plot_mass", "plot_mass_range"], visible=True, **group_style) frame.set_state_attribute("iso_or_elem", visible=True, description="species type: ", options=["Elements", "Isotopes"], selected_label="Elements") frame.set_state_attribute("species", visible=True, description="Element: ", options=elements_all, **text_box_style) frame.set_state_attribute("elem_numer", "plot_spectro", visible=True, description="Y-axis [X/Y], choose X: ", options=elements_all, **text_box_style) frame.set_state_attribute("elem_denom", "plot_spectro", visible=True, description="Y-axis [X/Y], choose Y: ", options=elements_all, **text_box_style) frame.set_state_attribute("plot", states_plot, visible=True, description="Generate Plot", **button_style) def clear_plot_handler(widget): clear_output() pyplot.close("all") frame.set_state_data("over_plotting_data", []) def sel_plot_type(attribute, value): if value == "Total mass": frame.set_state("plot_totmasses") elif value == "Species mass": frame.set_state("plot_mass") elif value == "Species spectroscopic": frame.set_state("plot_spectro") elif value == "Mass range contributions": frame.set_state("plot_mass_range") # iniZ = float(frame.get_attribute("init_Z", "value")) # if iniZ==0.0: # frame.set_attributes("source", options=["All", "AGB", "Massive"]) # else: frame.set_attributes("source", options=["All", "AGB", "SNe Ia", "Massive"]) frame.set_state_data("over_plotting_data", []) def sel_source(attribute, value): if value == "SNe Ia": frame.set_state_data("elements", elements_sn1a) frame.set_state_data("isotopes", isotopes_sn1a) else: frame.set_state_data("elements", elements_all) frame.set_state_data("isotopes", isotopes_all) frame.set_attributes("elem_numer", options=[]) frame.set_attributes("elem_denom", options=[]) frame.set_attributes("species", options=[]) elements = frame.get_state_data("elements") isotopes = frame.get_state_data("isotopes") frame.set_attributes("elem_numer", options=elements) frame.set_attributes("elem_denom", options=elements) if frame.get_attribute("iso_or_elem", "value") == "Isotopes": frame.set_attributes("species", description="Isotope: ", options=isotopes) elif frame.get_attribute("iso_or_elem", "value") == "Elements": frame.set_attributes("species", description="Element: ", options=elements) def sel_iso_or_elem(attribute, value): elements = frame.get_state_data("elements") isotopes = frame.get_state_data("isotopes") if value == "Isotopes": frame.set_attributes("species", description="Isotope: ", options=isotopes) frame.set_attributes("species_mult", description="Isotope: ", options=isotopes_sel_mult) elif value == "Elements": frame.set_attributes("species", description="Element: ", options=elements) frame.set_attributes("species_mult", description="Element: ", options=elements_sel_mult) def run(widget): styles = frame.get_state_data("styles") styles.reset_line_count() clear_output() pyplot.close("all") over_plotting = frame.get_attribute("over_plotting", "value") source_map = { "All": "all", "AGB": "agb", "SNe Ia": "sn1a", "Massive": "massive" } label_map = { "All": "", "AGB": ", AGB", "SNe Ia": ", SNIa", "Massive": ", Massive" } tot_mass_labels = { "all": "All", "agb": "AGB", "sn1a": "SNIa", "massive": "Massive" } state = frame.get_state() runs = frame.get_state_data("runs") source = source_map[frame.get_attribute("source", "value")] label_source = label_map[frame.get_attribute("source", "value")] species = frame.get_attribute("species", "value") no_runs = True if state == "plot_totmasses": plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("source", source)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + tot_mass_labels[kwargs["source"]] kwargs.update({"label": label}) kwargs.update(styles.get_style()) data.plot_totmasses(**kwargs) elif state == "plot_mass": plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("specie", species), ("source", source)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + kwargs["specie"] if not kwargs["source"] == "all": label = label + ", " + tot_mass_labels[ kwargs["source"]] kwargs.update({"label": label}) kwargs.update(styles.get_style()) data.plot_mass(**kwargs) elif state == "plot_spectro": X = frame.get_attribute("elem_numer", "value") Y = frame.get_attribute("elem_denom", "value") yaxis = "[" + X + "/" + Y + "]" plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("yaxis", yaxis), ("source", source)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + kwargs[ "yaxis"] + ", " + tot_mass_labels[kwargs["source"]] kwargs.update({"label": label}) kwargs.update(styles.get_style()) data.plot_spectro(**kwargs) elif state == "plot_mass_range": plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("specie", species)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + kwargs["specie"] kwargs.update({"label": label}) kwargs.update(styles.get_style()) data.plot_mass_range_contributions(**kwargs) if no_runs: print "No runs selected." frame.set_state_callbacks("clear_plot", clear_plot_handler, attribute=None, type="on_click") frame.set_state_callbacks("plot_type", sel_plot_type) frame.set_state_callbacks("source", sel_source, state=["plot_spectro", "plot_mass"]) frame.set_state_callbacks("iso_or_elem", sel_iso_or_elem) frame.set_state_callbacks("plot", run, attribute=None, type="on_click") frame.set_object("plot_page", widgets.VBox()) frame.set_object("warning_msg", widgets.HTML()) frame.set_object("plot_name", widgets.HTML()) frame.set_object("source_over_plotting_group", widgets.HBox()) frame.set_object("source", widgets.Dropdown()) frame.set_object("over_plotting", widgets.ToggleButton()) frame.set_object("clear_plot", widgets.Button()) frame.set_object("species_group", widgets.VBox()) frame.set_object("iso_or_elem", widgets.RadioButtons()) frame.set_object("species", widgets.Select()) frame.set_object("elem_numer", widgets.Select()) frame.set_object("elem_denom", widgets.Select()) frame.set_object("plot", widgets.Button()) frame.set_state_attribute("custom_imf_page", visible=True) frame.set_state_attribute("load_save_imf_group", visible=True, **group_style) frame.set_state_attribute("load_imf", "custom_imf", visible=True, description="Load custom IMF", **button_style) frame.set_state_attribute("list_imfs", "load_custom_imf", visible=True, description="Select IMF") frame.set_state_attribute("name_imf", "custom_imf", visible=True, description="IMF name", **text_box_style) frame.set_state_attribute("save_imf", "custom_imf", visible=True, description="Save IMF", **button_style) frame.set_state_attribute("delete_imf", "custom_imf", visible=True, description="Delete IMF", **button_style) frame.set_state_attribute("text_imf", visible=True) frame.set_state_attribute("test_imf", visible=True, description="Test selected IMF file", **button_style) def load_imf_handler(widget): frame.set_state("load_custom_imf") options = ["", "Preset: default"] + list_custom_imf() frame.set_attributes("list_imfs", options=options, value="", selected_label="") def sel_custom_imf(name, value): if value != "": frame.set_state("custom_imf") if value == "Preset: default": frame.set_attributes("name_imf", value="") frame.set_attributes("text_imf", value=default_custom_imf_text) else: text = "" with open(custom_imf_dir + value + ".py", "r") as fin: text = fin.read() frame.set_attributes("name_imf", value=value) frame.set_attributes("text_imf", value=text) def save_imf_handler(widget): clear_output() pyplot.close("all") imf_name = frame.get_attribute("name_imf", "value") if imf_name == "": print("Error: not IMF name given") else: if not os.path.isdir(custom_imf_dir): os.mkdir(custom_imf_dir) with open(custom_imf_dir + imf_name + ".py", "w") as fout: fout.write(frame.get_attribute("text_imf", "value")) print("Custom IMF saved to " + imf_name) frame.set_state_attribute( 'imf_type', options=['salpeter', 'chabrier', 'kroupa', 'alphaimf'] + list_custom_imf()) def delete_imf_handler(widget): clear_output() pyplot.close("all") imf_name = frame.get_attribute("name_imf", "value") if imf_name == "": print("Error: not IMF name given") else: if os.path.isdir(custom_imf_dir): if os.path.isfile(custom_imf_dir + imf_name + ".py"): os.remove(custom_imf_dir + imf_name + ".py") frame.set_attributes("name_imf", value="") frame.set_attributes("text_imf", value=default_custom_imf_text) print("Custom IMF " + imf_name + " deleted.") else: print("Custom IMF, " + imf_name + ", not found!") frame.set_state_attribute( 'imf_type', options=['salpeter', 'chabrier', 'kroupa', 'alphaimf'] + list_custom_imf()) def test_imf_handler(widget): clear_output() pyplot.close("all") imf_name = frame.get_attribute("name_imf", "value") if imf_name == "": print("No IMF name given!") return try: ci = load_source("custom_imf", custom_imf_dir + imf_name + ".py") except IOError: print("Failed to load " + imf_name + ", no IMF file found.") return mass_min = 0.0 mass_max = 30.0 xaxis = numpy.linspace(mass_min, mass_max, 1000)[1:] yaxis = [0 for x in xaxis] for i, x in enumerate(xaxis): yaxis[i] = ci.custom_imf(x) pyplot.plot(xaxis, numpy.log10(yaxis)) pyplot.title("IMF file test: " + imf_name) pyplot.xlabel("mass [$M_{\odot}$]") pyplot.ylabel("IMF") pyplot.show() frame.set_state_callbacks("load_imf", load_imf_handler, attribute=None, type="on_click") frame.set_state_callbacks("list_imfs", sel_custom_imf) frame.set_state_callbacks("save_imf", save_imf_handler, attribute=None, type="on_click") frame.set_state_callbacks("delete_imf", delete_imf_handler, attribute=None, type="on_click") frame.set_state_callbacks("test_imf", test_imf_handler, attribute=None, type="on_click") frame.set_object("custom_imf_page", widgets.VBox()) frame.set_object("load_save_imf_group", widgets.HBox()) frame.set_object("load_imf", widgets.Button()) frame.set_object("list_imfs", widgets.Dropdown()) frame.set_object("name_imf", widgets.Text()) frame.set_object("save_imf", widgets.Button()) frame.set_object("delete_imf", widgets.Button()) frame.set_object("text_imf", widgets.Textarea()) frame.set_object("test_imf", widgets.Button()) frame.set_state_attribute("get_table_page", visible=True) frame.set_state_attribute("warning_msg", visible=True, value="<h3>Error: No simulation data!</h3>", **group_style) frame.set_state_attribute("warning_msg", states_plot, visible=False) frame.set_state_attribute("species_mult_group", states_sim_plot, visible=True, **group_style) frame.set_state_attribute("iso_or_elem", visible=True, description="species type: ", options=["Elements", "Isotopes"], selected_label="Elements") frame.set_state_attribute("species_mult", visible=True, description="Element: ", options=elements_sel_mult, **text_box_style) frame.set_state_attribute("get_table", states_sim_plot, visible=True, description="Get table links", **button_style) frame.set_state_attribute("table_links", states_sim_plot, visible=True, value="", **group_style) def species_mult_handler(name, value): if "All" in value: iso_or_elem = frame.get_attribute("iso_or_elem", "value") if iso_or_elem == "Elements": value = tuple(elements_all) elif iso_or_elem == "Isotopes": value = tuple(isotopes_all) frame.set_attributes("species_mult", value=value, selected_labels=value) frame.set_attributes("table_links", value="") def get_table_handler(widget): iso_or_elem = frame.get_attribute("iso_or_elem", "value") species = list(frame.get_attribute("species_mult", "value")) runs = frame.get_state_data("runs") title = "<h3>Data table links:</h3>" html = title if not os.path.isdir("./evol_tables"): os.mkdir("./evol_tables") for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): file = "evol_tables/" + widget_name.replace("#", "") + "file" if iso_or_elem == "Elements": data.write_evol_table(species, [], file, "./") elif iso_or_elem == "Isotopes": data.write_evol_table([], species, file, "./") html = html + "<p><a href=\"" + file + "\" target=\"_blank\" download>" + name + "</a></p>\n" if html == title: html = "" print("No runs selected.") frame.set_attributes("table_links", value=html) clear_output() pyplot.close("all") frame.set_state_callbacks("species_mult", species_mult_handler) frame.set_state_callbacks("get_table", get_table_handler, attribute=None, type="on_click") frame.set_object("species_mult_group", widgets.VBox()) frame.set_object("species_mult", widgets.SelectMultiple()) frame.set_object("get_table_page", widgets.VBox()) frame.set_object("get_table", widgets.Button()) frame.set_object("table_links", widgets.HTML()) ##start widget## frame.display_object("window")
def start_OMEGA(): tablist = ["sculptor", "carina", "fornax"] yield_table = 'yield_tables/isotope_yield_table_MESA_only_ye.txt' mgal_factor = {"sculptor":1.51e9, "carina":3.4e6, "fornax":7.08e8} elements_sculpt = ['Mg', 'Si', 'Ca', 'Ti'] elements_carina = ['Na','Mg', 'Si', 'Sc', 'Ca', 'Ti', 'V', 'Cr', 'Mn', 'Ni', 'Zn'] elements_fornax = ['Mg', 'Si', 'Ca', 'Ti'] line_styles=['-', '--', '-.', ':'] line_colors=['b', 'g', 'r', 'c', 'm', 'y', 'k'] styles = auto_styles() styles.set_line_styles(line_styles) styles.set_line_colors(line_colors) styles.set_line_markers([]) color_convert = colors.ColorConverter() frame = framework.framework() frame.set_default_display_style(padding="0.25em",background_color="white", border_color="LightGrey", border_radius="0.5em") frame.set_default_io_style(padding="0.25em", margin="0.25em", border_color="LightGrey", border_radius="0.5em") group_style = {"border_style":"none", "border_radius":"0em"} text_box_style = {"width":"10em"} button_style = {"font_size":"1.25em", "font_weight":"bold"} first_tab_style = {"border_radius":"0em 0.5em 0.5em 0.5em"} states = ["sculptor", "carina", "fornax"] frame.add_state(states) frame.set_state_data("runs",[],"sculptor") frame.set_state_data("runs",[],"carina") frame.set_state_data("runs",[],"fornax") frame.set_state_data("styles", styles) frame.set_state_data("run_count", 0) def add_run(state, data, name): run_count = frame.get_state_data("run_count") state_data = frame.get_state_data("runs",state) styles = frame.get_state_data("styles") style = styles.get_style() line_color = style["color"] line_style = style["shape"] widget_name = "runs_widget_#"+str(run_count) frame.add_io_object(widget_name) frame.set_state_attribute(widget_name, state, visible=True, description=name) frame.set_object(widget_name, widgets.Checkbox()) state_data.append((data, name, line_style, line_color, widget_name)) frame.set_state_children("runs", [widget_name], state=state) frame.set_state_data("runs", state_data, state) def remove_runs(state): state_data = frame.get_state_data("runs",state) children = ["runs_title"] tmp_data=[] for i in xrange(len(state_data)): widget_name = state_data[i][4] if frame.get_attribute(widget_name, "value"): frame.remove_object(widget_name) else: children.append(widget_name) tmp_data.append(state_data[i]) state_data=tmp_data frame.set_state_children("runs", children, state=state, append=False) frame.set_state_data("runs", state_data, state) frame.add_display_object("window") frame.add_io_object("title") frame.add_display_object("widget") frame.add_display_object("simulation") frame.add_display_object("sculptor") frame.add_display_object("carina") frame.add_display_object("fornax") frame.add_display_object("baryon_name_group") frame.add_io_object("f_baryon") frame.add_io_object("run_name") frame.add_io_object("loading_mass") frame.add_io_object("sn1a_pmil") frame.add_display_object("run_add_rm_group") frame.add_io_object("run_sim") frame.add_io_object("rm_sim") frame.add_display_object("plotting_runs_group") frame.add_display_object("plotting") frame.add_io_object("plotting_title") frame.add_display_object("runs") frame.add_io_object("runs_title") frame.add_io_object("select_elem") frame.add_io_object("plot") frame.add_io_object("warning_msg") frame.set_state_children("window", ["title", "widget"]) frame.set_state_children("widget", ["simulation", "plotting_runs_group"]) frame.set_state_children("simulation", tablist, titles=["Sculptor Galaxy", "Carina Galaxy", "Fornax Galaxy"]) frame.set_state_children("sculptor", ["baryon_name_group", "loading_mass", "sn1a_pmil", "run_add_rm_group"]) frame.set_state_children("carina", ["baryon_name_group", "loading_mass", "sn1a_pmil", "run_add_rm_group"]) frame.set_state_children("fornax", ["baryon_name_group", "loading_mass", "sn1a_pmil", "run_add_rm_group"]) frame.set_state_children("baryon_name_group", ["f_baryon", "run_name"]) frame.set_state_children("run_add_rm_group", ["run_sim", "rm_sim"]) frame.set_state_children("plotting_runs_group", ["plotting", "runs"]) frame.set_state_children("plotting", ["plotting_title", "select_elem", "plot", "warning_msg"]) frame.set_state_children("runs", ["runs_title"], state=states) frame.set_state_attribute('window', visible=True, **group_style) frame.set_state_attribute('title', visible=True, value="<center><h1>OMEGA</h1></center>") frame.set_state_attribute('widget', visible=True, **group_style) frame.set_state_attribute('simulation', visible=True, **group_style) frame.set_state_attribute('sculptor', visible=True, **first_tab_style) frame.set_state_attribute('carina', visible=True) frame.set_state_attribute('fornax', visible=True) frame.set_state_attribute("baryon_name_group", visible=True, **group_style) frame.set_state_attribute("f_baryon", visible=True, description="Baryon fraction: ", min=0.00005, max=0.1, step=0.00005) frame.set_state_attribute("f_baryon", "sculptor", value=0.001) frame.set_state_attribute("f_baryon", "carina", value=0.05) frame.set_state_attribute("f_baryon", "fornax", value=0.001) frame.set_state_attribute("run_name", visible=True, description="Run name: ", placeholder="Enter name", **text_box_style) frame.set_state_attribute("loading_mass", visible=True, description="Mass loading factor: ", **text_box_style) frame.set_state_attribute("loading_mass", ["sculptor", "carina"], value="10.0") frame.set_state_attribute("loading_mass", "fornax", value="4.0") frame.set_state_attribute("sn1a_pmil", visible=True, description="SNe Ia per stellar mass formed: ", **text_box_style) frame.set_state_attribute("sn1a_pmil", ["sculptor", "carina"], value="2.0e-3") frame.set_state_attribute("sn1a_pmil", "fornax", value="4.0e-3") frame.set_state_attribute("run_add_rm_group", visible=True, **group_style) frame.set_state_attribute("run_sim", visible=True, description="Run simulation", **button_style) frame.set_state_attribute("rm_sim", visible=True, description="Remove selected run", **button_style) frame.set_state_attribute("plotting_runs_group", visible=True, **group_style) frame.set_state_attribute("plotting", visible=True) frame.set_state_attribute("plotting_title", visible=True, value="<h2>Plotting options</h2>", **group_style) frame.set_state_attribute("runs", visible=True) frame.set_state_attribute("runs_title", visible=True, value="<h2>Runs</h2>", **group_style) frame.set_state_attribute("select_elem", visible=True, description="Select Element: ", **text_box_style) frame.set_state_attribute("select_elem", "sculptor", options=elements_sculpt) frame.set_state_attribute("select_elem", "carina", options=elements_carina) frame.set_state_attribute("select_elem", "fornax", options=elements_fornax) frame.set_state_attribute("plot", visible=True, description="Generate Plot", **button_style) frame.set_state_attribute("warning_msg", visible=False, value="<h3>Error no runs selected!</h3>", **group_style) def loading_mass_handler(name, value): if (value.strip())[0] == "-": value = "0.0" frame.set_attributes("loading_mass", value=float_text(value)) def sn1a_pmil_handler(name, value): frame.set_attributes("sn1a_pmil", value=float_text(value)) def simulation_run(widget): run_count = frame.get_state_data("run_count") run_count += 1 state = frame.get_state() data = None name = frame.get_attribute("run_name", "value") if name == "": name="Run: "+"%03d" % (run_count,) f_baryon = frame.get_attribute("f_baryon", "value") loading_mass = float(frame.get_attribute("loading_mass", "value")) sn1a_pmil = float(frame.get_attribute("sn1a_pmil", "value")) mgal = f_baryon * mgal_factor[state] data = omega.omega(galaxy=state, in_out_control=True, mgal=mgal, mass_loading=loading_mass, nb_1a_per_m=sn1a_pmil, table=yield_table, Z_trans=-1, in_out_ratio=2.0) add_run(state, data, name) frame.update() frame.set_attributes("run_name", value="") frame.set_state_data("run_count", run_count) def remove_simulation(widget): state = frame.get_state() remove_runs(state) frame.update() def sel_tab(name, value): clear_output() pyplot.close("all") open_tab = tablist[value] frame.set_state(open_tab) frame.set_attributes("run_name", value="") frame.update() def generate_plot(widget): clear_output() pyplot.close("all") state = frame.get_state() data = frame.get_state_data("runs", state) plotted_data = False comp_data = True if len(data) != 0: st = stellab.stellab() element = frame.get_attribute("select_elem", "value") yaxis = '['+element+'/Fe]' st.plot_spectro(xaxis='[Fe/H]', yaxis=yaxis,norm='Grevesse_Sauval_1998',galaxy=state,show_err=True) for i in xrange(len(data)): instance, name, line_style, line_color, widget_name = data[i] label = name+", "+yaxis selected = frame.get_attribute(widget_name, "value") if selected: plotted_data=True xy = instance.plot_spectro(xaxis='[Fe/H]', yaxis=yaxis, return_x_y=True) pyplot.plot(xy[0], xy[1], color=color_convert.to_rgba("w", 0.8), linestyle="-", marker=" ", linewidth=4) if comp_data: comp_data=False for i in xrange(len(data)): instance, name, line_style, line_color, widget_name = data[i] label = name+", "+yaxis selected = frame.get_attribute(widget_name, "value") if selected: plotted_data=True xy = instance.plot_spectro(xaxis='[Fe/H]', yaxis=yaxis, return_x_y=True) pyplot.plot(xy[0], xy[1], color=line_color, linestyle=line_style, marker=" ", linewidth=2, label=label) if comp_data: comp_data=False #pyplot.legend(loc='center left', bbox_to_anchor=(1.01, 0.5), markerscale=0.8, fontsize=12) if plotted_data: frame.set_attributes("warning_msg", visible=False) if state=="sculptor": pyplot.ylim(-1.0,1.6) pyplot.xlim(-4.0, -0.5) elif state=="carina": pyplot.ylim(-1.0,1.6) pyplot.xlim(-4.0, -0.5) if state=="fornax": pyplot.ylim(-1.0,1.4) pyplot.xlim(-3.5, 0) matplotlib.rcParams.update({'font.size': 14}) pyplot.subplots_adjust(right=0.6) pyplot.subplots_adjust(bottom=0.15) pyplot.legend(loc='center left', bbox_to_anchor=(1.01, 0.5), prop={'size':12}) pyplot.show() else: frame.set_attributes("warning_msg", visible=True, value="<h3>Error no runs selected!</h3>") else: frame.set_attributes("warning_msg", visible=True, value="<h3>Error no run data!</h3>") frame.set_state_callbacks("loading_mass", loading_mass_handler) frame.set_state_callbacks("sn1a_pmil", sn1a_pmil_handler) frame.set_state_callbacks("run_sim", simulation_run, attribute=None, type="on_click") frame.set_state_callbacks("rm_sim", remove_simulation, attribute=None, type="on_click") frame.set_state_callbacks("simulation", sel_tab, "selected_index") frame.set_state_callbacks("plot", generate_plot, attribute=None, type="on_click") frame.set_object("window", widgets.Box()) frame.set_object("title", widgets.HTML()) frame.set_object("widget", widgets.VBox()) frame.set_object("simulation", widgets.Tab()) frame.set_object("sculptor", widgets.Box()) frame.set_object("carina", widgets.Box()) frame.set_object("fornax", widgets.Box()) frame.set_object("baryon_name_group", widgets.HBox()) frame.set_object("f_baryon", widgets.FloatSlider()) frame.set_object("run_name", widgets.Text()) frame.set_object("loading_mass", widgets.Text()) frame.set_object("sn1a_pmil", widgets.Text()) frame.set_object("run_add_rm_group", widgets.HBox()) frame.set_object("run_sim", widgets.Button()) frame.set_object("rm_sim", widgets.Button()) frame.set_object("plotting_runs_group", widgets.HBox()) frame.set_object("plotting", widgets.VBox()) frame.set_object("plotting_title", widgets.HTML()) frame.set_object("runs", widgets.VBox()) frame.set_object("runs_title", widgets.HTML()) frame.set_object("select_elem", widgets.Dropdown()) frame.set_object("plot", widgets.Button()) frame.set_object("warning_msg", widgets.HTML()) ##start widget## frame.display_object("window") tab_index = frame.get_attribute("simulation", "selected_index") sel_tab("selected_index", tab_index)
def start_test_with_alpha_elements(): yield_table_names = ["NuGrid raw", "NuGrid 25Mo Nomoto 2006", "NuGrid 30Mo Nomoto 2006", "NuGrid 40Mo Nomoto 2006"] yield_tables = {"NuGrid raw":'yield_tables/isotope_yield_table.txt', "NuGrid 25Mo Nomoto 2006":'yield_tables/isotope_yield_table_N06_25Mo_full_IMF.txt', "NuGrid 30Mo Nomoto 2006":'yield_tables/isotope_yield_table_N06_30Mo_full_IMF.txt', "NuGrid 40Mo Nomoto 2006":'yield_tables/isotope_yield_table_N06_40Mo_full_IMF.txt'} elements = ['O', 'Mg', 'Si', 'S', 'Ca'] line_styles=['-', '--', '-.', ':'] line_colors=['b', 'g', 'r', 'c', 'm', 'y', 'k'] styles = auto_styles() styles.set_line_styles(line_styles) styles.set_line_colors(line_colors) styles.set_line_markers([]) color_convert = colors.ColorConverter() frame = framework.framework() frame.set_default_display_style(padding="0.25em",background_color="white", border_color="LightGrey", border_radius="0.5em") frame.set_default_io_style(padding="0.25em", margin="0.25em", border_color="LightGrey", border_radius="0.5em") group_style = {"border_style":"none", "border_radius":"0em"} text_box_style = {"width":"10em"} button_style = {"font_size":"1.25em", "font_weight":"bold"} first_tab_style = {"border_radius":"0em 0.5em 0.5em 0.5em"} frame.set_state_data("runs",[]) frame.set_state_data("styles", styles) frame.set_state_data("run_count", 0) def add_run(data, name): run_count = frame.get_state_data("run_count") state_data = frame.get_state_data("runs") styles = frame.get_state_data("styles") style = styles.get_style() line_color = style["color"] line_style = style["shape"] widget_name = "runs_widget_#"+str(run_count) frame.add_io_object(widget_name) frame.set_state_attribute(widget_name, visible=True, description=name) frame.set_object(widget_name, widgets.Checkbox()) state_data.append((data, name, line_style, line_color, widget_name)) frame.set_state_children("runs", [widget_name]) frame.set_state_data("runs", state_data) def remove_runs(): state_data = frame.get_state_data("runs") children = ["runs_title"] tmp_data=[] for i in xrange(len(state_data)): widget_name = state_data[i][4] if frame.get_attribute(widget_name, "value"): frame.remove_object(widget_name) else: children.append(widget_name) tmp_data.append(state_data[i]) state_data=tmp_data frame.set_state_children("runs", children, append=False) frame.set_state_data("runs", state_data) frame.add_display_object("window") frame.add_io_object("title") frame.add_display_object("widget") frame.add_display_object("simulation") frame.add_display_object("table_name_group") frame.add_io_object("select_table") frame.add_io_object("run_name") frame.add_display_object("run_add_rm_group") frame.add_io_object("run_sim") frame.add_io_object("rm_sim") frame.add_display_object("plotting_runs_group") frame.add_display_object("plotting") frame.add_io_object("plotting_title") frame.add_display_object("runs") frame.add_io_object("runs_title") frame.add_io_object("select_elem") frame.add_io_object("plot") frame.add_io_object("warning_msg") frame.set_state_children("window", ["title", "widget"]) frame.set_state_children("widget", ["simulation", "plotting_runs_group"]) frame.set_state_children("simulation", ["table_name_group", "run_add_rm_group"]) frame.set_state_children("table_name_group", ["select_table", "run_name"]) frame.set_state_children("run_add_rm_group", ["run_sim", "rm_sim"]) frame.set_state_children("plotting_runs_group", ["plotting", "runs"]) frame.set_state_children("plotting", ["plotting_title", "select_elem", "plot", "warning_msg"]) frame.set_state_children("runs", ["runs_title"]) frame.set_state_attribute('window', visible=True, **group_style) frame.set_state_attribute('title', visible=True, value="<center><h1>Test with Alpha elements</h1><center>") frame.set_state_attribute('widget', visible=True, **group_style) frame.set_state_attribute('simulation', visible=True) frame.set_state_attribute("table_name_group", visible=True, **group_style) frame.set_state_attribute("select_table", visible=True, description="Yield table: ", options=yield_table_names) frame.set_state_attribute("run_name", visible=True, description="Run name: ", placeholder="Enter name", **text_box_style) frame.set_state_attribute("run_add_rm_group", visible=True, **group_style) frame.set_state_attribute("run_sim", visible=True, description="Run simulation", **button_style) frame.set_state_attribute("rm_sim", visible=True, description="Remove selected run", **button_style) frame.set_state_attribute("plotting_runs_group", visible=True, **group_style) frame.set_state_attribute("plotting", visible=True) frame.set_state_attribute("plotting_title", visible=True, value="<h2>Plotting options</h2>", **group_style) frame.set_state_attribute("runs", visible=True) frame.set_state_attribute("runs_title", visible=True, value="<h2>Runs</h2>", **group_style) frame.set_state_attribute("select_elem", visible=True, description="Select Element: ", **text_box_style) frame.set_state_attribute("select_elem", options=elements) frame.set_state_attribute("plot", visible=True, description="Generate Plot", **button_style) frame.set_state_attribute("warning_msg", visible=False, value="<h3>Error no runs selected!</h3>", **group_style) def loading_mass_handler(name, value): if (value.strip())[0] == "-": value = "0.0" frame.set_attributes("loading_mass", value=float_text(value)) def sn1a_pmil_handler(name, value): frame.set_attributes("sn1a_pmil", value=float_text(value)) def simulation_run(widget): run_count = frame.get_state_data("run_count") run_count += 1 data = None name = frame.get_attribute("run_name", "value") if name == "": name="Run: "+"%03d" % (run_count,) select_table = yield_tables[frame.get_attribute("select_table", "value")] data = omega.omega(galaxy='milky_way', table=select_table) add_run(data, name) frame.update() frame.set_attributes("run_name", value="") frame.set_state_data("run_count", run_count) def remove_simulation(widget): remove_runs() frame.update() def generate_plot(widget): clear_output() pyplot.close("all") data = frame.get_state_data("runs") plotted_data = False comp_data = True if len(data) != 0: element = frame.get_attribute("select_elem", "value") for i in xrange(len(data)): instance, name, line_style, line_color, widget_name = data[i] yaxis = '['+element+'/Fe]' label = name+", "+yaxis selected = frame.get_attribute(widget_name, "value") if selected: plotted_data=True instance.plot_spectro(xaxis="[Fe/H]", yaxis=yaxis, show_data=comp_data, color=color_convert.to_rgba("w", 0.8), shape="-", marker=" ", linewidth=4, fsize=[10, 4.5], show_legend=False) if comp_data: comp_data=False for i in xrange(len(data)): instance, name, line_style, line_color, widget_name = data[i] yaxis = '['+element+'/Fe]' label = name+", "+yaxis selected = frame.get_attribute(widget_name, "value") if selected: plotted_data=True instance.plot_spectro(xaxis="[Fe/H]", yaxis=yaxis, label=label, show_data=comp_data, color=line_color, shape=line_style, marker=" ", linewidth=2, fsize=[10, 4.5]) if comp_data: comp_data=False if plotted_data: frame.set_attributes("warning_msg", visible=False) pyplot.ylim(-0.5,1.4) matplotlib.rcParams.update({'font.size': 14}) pyplot.subplots_adjust(right=0.6) pyplot.subplots_adjust(bottom=0.15) pyplot.legend(loc='center left', bbox_to_anchor=(1.01, 0.5), prop={'size':12}) pyplot.show() else: frame.set_attributes("warning_msg", visible=True, value="<h3>Error no runs selected!</h3>") else: frame.set_attributes("warning_msg", visible=True, value="<h3>Error no run data!</h3>") frame.set_state_callbacks("run_sim", simulation_run, attribute=None, type="on_click") frame.set_state_callbacks("rm_sim", remove_simulation, attribute=None, type="on_click") frame.set_state_callbacks("plot", generate_plot, attribute=None, type="on_click") frame.set_object("window", widgets.Box()) frame.set_object("title", widgets.HTML()) frame.set_object("widget", widgets.VBox()) frame.set_object("simulation", widgets.VBox()) frame.set_object("table_name_group", widgets.HBox()) frame.set_object("select_table", widgets.Dropdown()) frame.set_object("run_name", widgets.Text()) frame.set_object("run_add_rm_group", widgets.HBox()) frame.set_object("run_sim", widgets.Button()) frame.set_object("rm_sim", widgets.Button()) frame.set_object("plotting_runs_group", widgets.HBox()) frame.set_object("plotting", widgets.VBox()) frame.set_object("plotting_title", widgets.HTML()) frame.set_object("runs", widgets.VBox()) frame.set_object("runs_title", widgets.HTML()) frame.set_object("select_elem", widgets.Dropdown()) frame.set_object("plot", widgets.Button()) frame.set_object("warning_msg", widgets.HTML()) ##start widget## frame.display_object("window")
def start_SYGMA(): frame = framework.framework() frame.set_default_display_style(padding="0.25em",background_color="white", border_color="LightGrey", border_radius="0.5em") frame.set_default_io_style(padding="0.25em", margin="0.25em", border_color="LightGrey", border_radius="0.5em") tablist = ["sim_page", "plot_page", "custom_imf_page", "get_table_page"] yield_list = {"Analytic perscription":{"Delay":"yield_tables/isotope_yield_table_MESA_only_fryer12_delay.txt", "Rapid":"yield_tables/isotope_yield_table_MESA_only_fryer12_rapid.txt", "Exclude neutron-alpha rich freeze-out":"yield_tables/isotope_yield_table_MESA_only_fryer12_exclnalpha.txt"}, "Ye=0.4982":{"Fallback at Ye":"yield_tables/isotope_yield_table_MESA_only_ye.txt", "Fallback motivated by GCE":"yield_tables/isotope_yield_table_MESA_only_ye_fallback.txt"}} group_style = {"border_style":"none", "border_radius":"0em"} text_box_style = {"width":"10em"} button_style = {"font_size":"1.25em", "font_weight":"bold"} first_tab_style = {"border_radius":"0em 0.5em 0.5em 0.5em"} custom_imf_dir = os.environ["SYGMADIR"] + "/SYGMA_widget_imfs/" default_custom_imf_text = "\n#File to define a custom IMF\n#Define your IMF in custom_imf\n#so that the return value represents\n#the chosen IMF value for the input mass\n\ndef custom_imf(mass):\n #Salpeter IMF\n return mass**-2.35\n" states_plot = ["plot_totmasses", "plot_mass", "plot_spectro", "plot_mass_range"] states_sim_plot = ["run_sim"] + states_plot states_cimf = ["custom_imf", "load_custom_imf"] states = states_sim_plot + states_cimf frame.add_state(states) isotopes_all=['H-1','H-2','He-3','He-4','Li-7','B-11','C-12','C-13','N-14','N-15','O-16','O-17','O-18','F-19','Ne-20','Ne-21','Ne-22','Na-23','Mg-24','Mg-25','Mg-26','Al-27','Si-28','Si-29','Si-30','P-31','S-32','S-33','S-34','S-36','Cl-35','Cl-37','Ar-36','Ar-38','Ar-40','K-39','K-40','K-41','Ca-40','Ca-42','Ca-43','Ca-44','Ca-46','Ca-48','Sc-45','Ti-46','Ti-47','Ti-48','Ti-49','Ti-50','V-50','V-51','Cr-50','Cr-52','Cr-53','Cr-54','Mn-55','Fe-54','Fe-56','Fe-57','Fe-58','Co-59','Ni-58','Ni-60','Ni-61','Ni-62','Ni-64'] elements_all=['H','He','Li','B','C','N','O','F','Ne','Na','Mg','Al','Si','P','S','Cl','Ar','K','Ca','Sc','Ti','V','Cr','Mn','Fe','Co','Ni'] isotopes_sn1a=['C-12','C-13','N-14','N-15','O-16','O-17','O-18','F-19','Ne-20','Ne-21','Ne-22','Na-23','Mg-24','Mg-25','Mg-26','Al-27','Si-28','Si-29','Si-30','P-31','S-32','S-33','S-34','S-36','Cl-35','Cl-37','Ar-36','Ar-38','Ar-40','K-39','K-40','K-41','Ca-40','Ca-42','Ca-43','Ca-44','Ca-46','Ca-48','Sc-45','Ti-46','Ti-47','Ti-48','Ti-49','Ti-50','V-50','V-51','Cr-50','Cr-52','Cr-53','Cr-54','Mn-55','Fe-54','Fe-56','Fe-57','Fe-58','Co-59','Ni-58','Ni-60','Ni-61','Ni-62','Ni-64'] elements_sn1a=['C','N','O','F','Ne','Na','Mg','Al','Si','P','S','Cl','Ar','K','Ca','Sc','Ti','V','Cr','Mn','Fe','Co','Ni'] isotopes_sel_mult = ["All"] + isotopes_all elements_sel_mult = ["All"] + elements_all line_styles=["-", "--", ":", "-."] # line_colors=["k", "r", "g", "b", "c", "m", "y"] line_colors=["k", "r", "b", "c", "m", "y"]#remove "g" line_markers = ["o", "s", "x", "D", "v", "^", "<", ">", "p", "*", "+"] styles = auto_styles() styles.set_line_styles(line_styles) styles.set_line_colors(line_colors) styles.set_line_markers(line_markers) frame.set_state_data("elements", elements_all) frame.set_state_data("isotopes", isotopes_all) frame.set_state_data("over_plotting_data", []) frame.set_state_data("styles", styles) frame.set_state_data("old_state", None) frame.set_state_data("runs", []) frame.set_state_data("run_count", 0) def add_run(data, name, Z): run_count = frame.get_state_data("run_count") runs_data = frame.get_state_data("runs") widget_name = "runs_widget_#"+str(run_count) frame.add_io_object(widget_name) frame.set_state_attribute(widget_name, visible=True, description=name) frame.set_object(widget_name, widgets.Checkbox()) runs_data.append((data, name, Z, widget_name)) frame.set_state_children("runs", [widget_name]) frame.set_state_data("runs", runs_data) run_count += 1 frame.set_state_data("run_count", run_count) def remove_runs(): data = frame.get_state_data("runs") children = ["runs_title"] tmp_data = [] for i in xrange(len(data)): widget_name = data[i][3] if frame.get_attribute(widget_name, "value"): frame.remove_object(widget_name) else: children.append(widget_name) tmp_data.append(data[i]) data = tmp_data frame.set_state_children("runs", children, append=False) frame.set_state_data("runs", data) def list_custom_imf(): if os.path.isdir(custom_imf_dir): files_dirs = os.listdir(custom_imf_dir) imfs = [] for file in files_dirs: file_full = custom_imf_dir + file if os.path.isfile(file_full): file = file.rsplit(".", 1) if file[1] == "py": imfs.append(file[0]) return imfs else: return [] frame.add_display_object("window") frame.add_io_object("title") frame.add_display_object("widget_runs_group") frame.add_display_object("widget") frame.add_display_object("runs") frame.add_io_object("runs_title") ###Sim page### frame.add_display_object("sim_page") frame.add_display_object("mass_Z_group") frame.add_io_object("init_Z") frame.add_io_object("mass_gas") frame.add_display_object("time_group") frame.add_io_object("t_end") frame.add_io_object("dt") frame.add_display_object("imf_type_group") frame.add_io_object("imf_type") frame.add_io_object("imf_alpha") frame.add_display_object("imf_mass_group") frame.add_io_object("imf_mass_min") frame.add_io_object("imf_mass_max") frame.add_display_object("sn1a_group") frame.add_io_object("use_sn1a") frame.add_io_object("sn1a_rates") frame.add_display_object("yield_table_group") frame.add_io_object("yield_table_selection") frame.add_io_object("yield_table_list") frame.add_display_object("run_sim_remove_run_group") frame.add_io_object("run_sim") frame.add_io_object("remove_run") frame.add_io_object("run_name") frame.add_io_object("plot_type") frame.add_io_object("sim_responce") frame.set_state_children("window", ["title", "widget_runs_group"]) frame.set_state_children("widget_runs_group", ["widget", "runs"]) frame.set_state_children("widget", ["sim_page"], titles=["Simulation"]) frame.set_state_children("sim_page", ["mass_Z_group", "time_group", "imf_type_group", "imf_mass_group", "sn1a_group", "yield_table_group", "run_sim_remove_run_group", "sim_responce"]) frame.set_state_children("mass_Z_group", ["mass_gas", "init_Z"]) frame.set_state_children("time_group", ["t_end", "dt"]) frame.set_state_children("imf_type_group", ["imf_type", "imf_alpha"]) frame.set_state_children("imf_mass_group", ["imf_mass_min", "imf_mass_max"]) frame.set_state_children("sn1a_group", ["use_sn1a", "sn1a_rates"]) frame.set_state_children("yield_table_group", ["yield_table_selection", "yield_table_list"]) frame.set_state_children("run_sim_remove_run_group", ["run_sim", "remove_run", "run_name"]) frame.set_state_children("runs", ["runs_title"]) ###plotting page### frame.add_display_object("plot_page") frame.add_io_object("warning_msg") frame.add_io_object("plot_name") frame.add_display_object("source_over_plotting_group") frame.add_io_object("source") frame.add_io_object("over_plotting") frame.add_io_object("clear_plot") frame.add_display_object("species_group") frame.add_io_object("iso_or_elem") frame.add_io_object("species") frame.add_io_object("elem_numer") frame.add_io_object("elem_denom") frame.add_io_object("plot") frame.set_state_children("widget", ["plot_page"], titles=["Plotting"]) frame.set_state_children("plot_page", ["warning_msg", "plot_type", "plot_name", "source_over_plotting_group", "species_group", "elem_numer", "elem_denom", "plot"]) frame.set_state_children("source_over_plotting_group", ["source", "over_plotting", "clear_plot"]) frame.set_state_children("species_group", ["iso_or_elem", "species"]) ###Custom IMF page### frame.add_display_object("custom_imf_page") frame.add_display_object("load_save_imf_group") frame.add_io_object("load_imf") frame.add_io_object("list_imfs") frame.add_io_object("name_imf") frame.add_io_object("save_imf") frame.add_io_object("delete_imf") frame.add_io_object("text_imf") frame.add_io_object("test_imf") frame.set_state_children("widget", ["custom_imf_page"], titles=["Custom IMF"]) frame.set_state_children("custom_imf_page", ["load_save_imf_group", "text_imf", "test_imf"]) frame.set_state_children("load_save_imf_group", ["load_imf", "list_imfs", "name_imf", "save_imf", "delete_imf"]) ###plotting page### frame.add_display_object("get_table_page") frame.add_display_object("species_mult_group") frame.add_io_object("species_mult") frame.add_io_object("get_table") frame.add_io_object("table_links") frame.set_state_children("widget", ["get_table_page"], titles=["Download Tables"]) frame.set_state_children("get_table_page", ["warning_msg", "species_mult_group", "get_table", "table_links"]) frame.set_state_children("species_mult_group", ["iso_or_elem", "species_mult"]) frame.set_state_attribute('window', visible=True, **group_style) frame.set_state_attribute('title', visible=True, value="<center><h1>SYGMA</h1></center>") frame.set_state_attribute("widget_runs_group", visible=True, **group_style) frame.set_state_attribute('widget', visible=True, **group_style) frame.set_state_attribute("runs", visible=True, margin="3.15em 0em 0em 0em") frame.set_state_attribute("runs", states_cimf, visible=False) frame.set_state_attribute("runs_title", visible=True, value="<center><h2>Runs</h2></center>", **group_style) frame.set_state_attribute('sim_page', visible=True, **first_tab_style) frame.set_state_attribute("mass_Z_group", visible=True, **group_style) frame.set_state_attribute("mass_gas", visible=True, description="Total stellar mass [$M_{\odot}$]:", value="1.0", **text_box_style) frame.set_state_attribute('init_Z', visible=True, description="Initial metallicity: ", options=["0.02", "0.01", "0.006", "0.001", "0.0001", "0.0"]) frame.set_state_attribute('time_group', visible=True, **group_style) frame.set_state_attribute('t_end', visible=True, description="Final time [yrs]: ", value="1.0e10", **text_box_style) frame.set_state_attribute('dt', visible=True, description="Time step [yrs]: ", value="1.0e7", **text_box_style) frame.set_state_attribute('imf_type_group', visible=True, **group_style) frame.set_state_attribute('imf_type', visible=True, description="IMF type: ", options=['salpeter', 'chabrier', 'kroupa', 'alphaimf'] + list_custom_imf()) frame.set_state_attribute('imf_alpha', description="Set alpha: ", min=0, max=5) frame.set_state_attribute("imf_mass_group", visible=True, **group_style) frame.set_state_attribute('imf_mass_min', visible=True, description="IMF lower limit [$M_{\odot}$]: ", value="1.0", **text_box_style) frame.set_state_attribute('imf_mass_max', visible=True, description="IMF upper limit [$M_{\odot}$]: ", value="30.0", **text_box_style) frame.set_state_attribute('sn1a_group', visible=True, **group_style) frame.set_state_attribute('use_sn1a', visible=True, description="Include SNe Ia: ", value=True) frame.set_state_attribute("yield_table_group", visible=True, **group_style) frame.set_state_attribute("yield_table_selection", visible=True, description="CCSN remnant prescription:", options=["Analytic perscription", "Ye=0.4982"], value="Ye=0.4982") frame.set_state_attribute('yield_table_list', visible=True, options=yield_list["Ye=0.4982"], selected_label="Fallback at Ye") frame.set_state_links("sn1a_link", [("use_sn1a", "value"), ("sn1a_rates", "visible")], directional=True) frame.set_state_attribute('sn1a_rates', description="SNe Ia rates: ", options=['Power law', 'Exponential', 'Gaussian','Maoz12']) frame.set_state_attribute("run_sim_remove_run_group", visible=True, **group_style) frame.set_state_attribute('run_sim', visible=True, description="Run simulation", **button_style) frame.set_state_attribute("remove_run", visible=True, description="Remove selected", **button_style) frame.set_state_attribute("run_name", visible=True, description="Run name: ", placeholder="Enter name", value="", **text_box_style) frame.set_state_attribute("sim_responce", value="<p>Simulation data loaded.</p>", **group_style) frame.set_state_attribute("sim_responce", states_sim_plot, visible=True) frame.set_state_attribute('plot_type', states_sim_plot, visible=True, description="Plot type: ", options=["Total mass", "Species mass", "Species spectroscopic", "Mass range contributions"]) def mass_gas_handler(name, value): frame.set_attributes("mass_gas", value=float_text(value)) def t_end_handler(name, value): frame.set_attributes("t_end", value=float_text(value)) def dt_handler(name, value): frame.set_attributes("dt", value=float_text(value)) def imf_mass_min_handler(name, value): frame.set_attributes("imf_mass_min", value=float_text(value)) def imf_mass_max_handler(name, value): frame.set_attributes("imf_mass_max", value=float_text(value)) def sel_imf_type(attribute, value): if value=="alphaimf": frame.set_state_attribute("imf_alpha", visible=True) frame.set_attributes("imf_alpha", visible=True, value=2.35) else: frame.set_state_attribute("imf_alpha", visible=False) frame.set_attributes("imf_alpha", visible=False) def yield_table_selection_handler(name, value): frame.set_attributes("yield_table_list", options=yield_list[value]) if value == "Analytic perscription": frame.set_attributes("yield_table_list", selected_label="Delay") elif value == "Ye=0.4982": frame.set_attributes("yield_table_list", selected_label="Fallback at Ye") def run_simulation(widget): frame.set_attributes("sim_responce", visible=False) clear_output() pyplot.close("all") sn1a_map = {"Power law":"power_law", "Gaussian":"gauss","Exponential":"exp","Maoz12":"maoz"} mgal = float(frame.get_attribute("mass_gas", "value")) iniZ = float(frame.get_attribute("init_Z", "value")) imf_type = frame.get_attribute("imf_type", "value") alphaimf = frame.get_attribute("imf_alpha", "value") mass_min = float(frame.get_attribute("imf_mass_min", "value")) mass_max = float(frame.get_attribute("imf_mass_max", "value")) imf_bdys = [mass_min, mass_max] sn1a_on = frame.get_attribute("use_sn1a", "value") sn1a_rate = sn1a_map[frame.get_attribute("sn1a_rates", "value")] dt = float(frame.get_attribute("dt", "value")) tend = float(frame.get_attribute("t_end", "value")) yield_table = frame.get_attribute("yield_table_list", "value") if not (imf_type in ['salpeter', 'chabrier', 'kroupa', 'alphaimf']): destination = os.environ["SYGMADIR"] + "/imf_input.py" shutil.copy(custom_imf_dir + imf_type + ".py", destination) imf_type = "input" run_count = frame.get_state_data("run_count") name = frame.get_attribute("run_name", "value") if name == "": name = "Run - "+"%03d" % (run_count + 1, ) if iniZ==0.0: data=s.sygma(mgal=mgal, iniZ=iniZ, imf_type=imf_type, alphaimf=alphaimf, imf_bdys=[10.1, 100.0], imf_bdys_pop3=imf_bdys, sn1a_on=sn1a_on, sn1a_rate=sn1a_rate, dt=dt,tend=tend, table=yield_table) else: data=s.sygma(mgal=mgal, iniZ=iniZ, imf_type=imf_type, alphaimf=alphaimf, imf_bdys=imf_bdys, sn1a_on=sn1a_on, sn1a_rate=sn1a_rate, dt=dt,tend=tend, table=yield_table) frame.set_state("run_sim") ##force reset plottype frame.set_attributes("plot_type", selected_label="Species mass", value="Species mass") frame.set_attributes("plot_type", selected_label="Total mass", value="Total mass") add_run(data, name, iniZ) frame.update() def remove_simulation(widget): remove_runs() frame.update() def sel_tab(name, value): open_tab = tablist[value] if open_tab == "custom_imf_page": frame.set_state_data("old_state", frame.get_state()) frame.set_state("custom_imf") if frame.get_attribute("name_imf", "value") == "": frame.set_attributes("text_imf", value=default_custom_imf_text) else: if frame.get_state() in states_cimf: old_state = frame.get_state_data("old_state") if old_state != None: frame.set_state(old_state) else: frame.set_state("default") frame.set_state_data("old_state", None) frame.set_state_callbacks("mass_gas", mass_gas_handler) frame.set_state_callbacks("t_end", t_end_handler) frame.set_state_callbacks("dt", dt_handler) frame.set_state_callbacks("imf_mass_min", imf_mass_min_handler) frame.set_state_callbacks("imf_mass_max", imf_mass_max_handler) frame.set_state_callbacks("imf_type", sel_imf_type) frame.set_state_callbacks("yield_table_selection", yield_table_selection_handler) frame.set_state_callbacks("run_sim", run_simulation, attribute=None, type="on_click") frame.set_state_callbacks("remove_run", remove_simulation, attribute=None, type="on_click") frame.set_state_callbacks("widget", sel_tab, "selected_index") frame.set_object("window", widgets.Box()) frame.set_object("title", widgets.HTML()) frame.set_object("widget_runs_group", widgets.HBox()) frame.set_object("widget", widgets.Tab()) frame.set_object("runs", widgets.VBox()) frame.set_object("runs_title", widgets.HTML()) frame.set_object("sim_page", widgets.VBox()) frame.set_object("mass_Z_group", widgets.HBox()) frame.set_object("mass_gas", widgets.Text()) frame.set_object("init_Z", widgets.Dropdown()) frame.set_object("time_group", widgets.HBox()) frame.set_object("t_end", widgets.Text()) frame.set_object("dt", widgets.Text()) frame.set_object("imf_type_group", widgets.HBox()) frame.set_object("imf_type", widgets.Dropdown()) frame.set_object("imf_alpha", widgets.FloatSlider()) frame.set_object("imf_mass_group", widgets.HBox()) frame.set_object("imf_mass_min", widgets.Text()) frame.set_object("imf_mass_max", widgets.Text()) frame.set_object("sn1a_group", widgets.HBox()) frame.set_object("use_sn1a", widgets.Checkbox()) frame.set_object("sn1a_rates", widgets.Dropdown()) frame.set_object("yield_table_group", widgets.HBox()) frame.set_object("yield_table_selection", widgets.ToggleButtons()) frame.set_object("yield_table_list", widgets.Dropdown()) frame.set_object("run_sim_remove_run_group", widgets.HBox()) frame.set_object("run_sim", widgets.Button()) frame.set_object("remove_run", widgets.Button()) frame.set_object("run_name", widgets.Text()) frame.set_object("sim_responce", widgets.HTML()) frame.set_object("plot_type", widgets.Dropdown()) frame.set_state_attribute("plot_page", visible=True) frame.set_state_attribute("warning_msg", visible=True, value="<h3>Error: No simulation data!</h3>", **group_style) frame.set_state_attribute("warning_msg", states_plot, visible=False) frame.set_state_attribute("plot_name", **group_style) frame.set_state_attribute("plot_name", "plot_totmasses", visible=True, value="<h2>Plot: Total mass evolution</h2>") frame.set_state_attribute("plot_name", "plot_mass", visible=True, value="<h2>Plot: Species mass evolution</h2>") frame.set_state_attribute("plot_name", "plot_spectro", visible=True, value="<h2>Plot: Spectroscopic Mass evolution</h2>") frame.set_state_attribute("plot_name", "plot_mass_range", visible=True, value="<h2>Plot: Mass range contributions</h2><p>Only ejecta from AGB and massive stars are considered.</p>") frame.set_state_attribute("source_over_plotting_group", states_plot, visible=True, **group_style) frame.set_state_attribute("source", ["plot_totmasses", "plot_mass", "plot_spectro"], visible=True, description="Yield source: ", options=["All", "AGB", "SNe Ia", "Massive"], selected_label="All") frame.set_state_attribute("over_plotting", visible=True, description="Over plotting", value=False, **button_style) frame.set_state_attribute("clear_plot", description="Clear plot", **button_style) frame.set_state_links("clear_plot_link", [("over_plotting", "value"), ("clear_plot", "visible")], directional=True) frame.set_state_attribute("species_group", ["plot_mass", "plot_mass_range"], visible=True, **group_style) frame.set_state_attribute("iso_or_elem", visible=True, description="species type: ", options=["Elements", "Isotopes"], selected_label="Elements") frame.set_state_attribute("species", visible=True, description="Element: ", options=elements_all, **text_box_style) frame.set_state_attribute("elem_numer", "plot_spectro", visible=True, description="Y-axis [X/Y], choose X: ", options=elements_all, **text_box_style) frame.set_state_attribute("elem_denom", "plot_spectro", visible=True, description="Y-axis [X/Y], choose Y: ", options=elements_all, **text_box_style) frame.set_state_attribute("plot", states_plot, visible=True, description="Generate Plot", **button_style) def clear_plot_handler(widget): clear_output() pyplot.close("all") frame.set_state_data("over_plotting_data", []) def sel_plot_type(attribute, value): if value=="Total mass": frame.set_state("plot_totmasses") elif value=="Species mass": frame.set_state("plot_mass") elif value=="Species spectroscopic": frame.set_state("plot_spectro") elif value=="Mass range contributions": frame.set_state("plot_mass_range") # iniZ = float(frame.get_attribute("init_Z", "value")) # if iniZ==0.0: # frame.set_attributes("source", options=["All", "AGB", "Massive"]) # else: frame.set_attributes("source", options=["All", "AGB", "SNe Ia", "Massive"]) frame.set_state_data("over_plotting_data", []) def sel_source(attribute, value): if value=="SNe Ia": frame.set_state_data("elements", elements_sn1a) frame.set_state_data("isotopes", isotopes_sn1a) else: frame.set_state_data("elements", elements_all) frame.set_state_data("isotopes", isotopes_all) frame.set_attributes("elem_numer", options=[]) frame.set_attributes("elem_denom", options=[]) frame.set_attributes("species", options=[]) elements = frame.get_state_data("elements") isotopes = frame.get_state_data("isotopes") frame.set_attributes("elem_numer", options=elements) frame.set_attributes("elem_denom", options=elements) if frame.get_attribute("iso_or_elem", "value")=="Isotopes": frame.set_attributes("species", description="Isotope: ", options=isotopes) elif frame.get_attribute("iso_or_elem", "value")=="Elements": frame.set_attributes("species", description="Element: ", options=elements) def sel_iso_or_elem(attribute, value): elements = frame.get_state_data("elements") isotopes = frame.get_state_data("isotopes") if value=="Isotopes": frame.set_attributes("species", description="Isotope: ", options=isotopes) frame.set_attributes("species_mult", description="Isotope: ", options=isotopes_sel_mult) elif value=="Elements": frame.set_attributes("species", description="Element: ", options=elements) frame.set_attributes("species_mult", description="Element: ", options=elements_sel_mult) def run(widget): styles = frame.get_state_data("styles") styles.reset_line_count() clear_output() pyplot.close("all") over_plotting = frame.get_attribute("over_plotting", "value") source_map = {"All":"all", "AGB":"agb", "SNe Ia":"sn1a", "Massive":"massive"} label_map = {"All":"", "AGB":", AGB", "SNe Ia":", SNIa", "Massive":", Massive"} tot_mass_labels = {"all":"All", "agb":"AGB", "sn1a":"SNIa", "massive":"Massive"} state = frame.get_state() runs = frame.get_state_data("runs") source = source_map[frame.get_attribute("source", "value")] label_source = label_map[frame.get_attribute("source", "value")] species = frame.get_attribute("species", "value") no_runs = True if state=="plot_totmasses": plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("source", source)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + tot_mass_labels[kwargs["source"]] kwargs.update({"label":label}) kwargs.update(styles.get_style()) data.plot_totmasses(**kwargs) elif state=="plot_mass": plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("specie", species), ("source", source)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + kwargs["specie"] if not kwargs["source"] == "all": label = label + ", " + tot_mass_labels[kwargs["source"]] kwargs.update({"label":label}) kwargs.update(styles.get_style()) data.plot_mass(**kwargs) elif state=="plot_spectro": X = frame.get_attribute("elem_numer", "value") Y = frame.get_attribute("elem_denom", "value") yaxis = "["+X+"/"+Y+"]" plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("yaxis", yaxis), ("source", source)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + kwargs["yaxis"] + ", " + tot_mass_labels[kwargs["source"]] kwargs.update({"label":label}) kwargs.update(styles.get_style()) data.plot_spectro(**kwargs) elif state=="plot_mass_range": plot_data = frame.get_state_data("over_plotting_data") if not over_plotting: plot_data = [] over_plot = [("specie", species)] if not over_plot in plot_data: plot_data.append(over_plot) frame.set_state_data("over_plotting_data", plot_data) for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): no_runs = False for item in plot_data: kwargs = dict(item) label = name + ": " + kwargs["specie"] kwargs.update({"label":label}) kwargs.update(styles.get_style()) data.plot_mass_range_contributions(**kwargs) if no_runs: print "No runs selected." frame.set_state_callbacks("clear_plot", clear_plot_handler, attribute=None, type="on_click") frame.set_state_callbacks("plot_type", sel_plot_type) frame.set_state_callbacks("source", sel_source, state=["plot_spectro", "plot_mass"]) frame.set_state_callbacks("iso_or_elem", sel_iso_or_elem) frame.set_state_callbacks("plot", run, attribute=None, type="on_click") frame.set_object("plot_page", widgets.VBox()) frame.set_object("warning_msg", widgets.HTML()) frame.set_object("plot_name", widgets.HTML()) frame.set_object("source_over_plotting_group", widgets.HBox()) frame.set_object("source", widgets.Dropdown()) frame.set_object("over_plotting", widgets.ToggleButton()) frame.set_object("clear_plot", widgets.Button()) frame.set_object("species_group", widgets.VBox()) frame.set_object("iso_or_elem", widgets.RadioButtons()) frame.set_object("species", widgets.Select()) frame.set_object("elem_numer", widgets.Select()) frame.set_object("elem_denom", widgets.Select()) frame.set_object("plot", widgets.Button()) frame.set_state_attribute("custom_imf_page", visible=True) frame.set_state_attribute("load_save_imf_group", visible=True, **group_style) frame.set_state_attribute("load_imf", "custom_imf", visible=True, description="Load custom IMF", **button_style) frame.set_state_attribute("list_imfs", "load_custom_imf", visible=True, description="Select IMF") frame.set_state_attribute("name_imf", "custom_imf", visible=True, description="IMF name", **text_box_style) frame.set_state_attribute("save_imf", "custom_imf", visible=True, description="Save IMF", **button_style) frame.set_state_attribute("delete_imf", "custom_imf", visible=True, description="Delete IMF", **button_style) frame.set_state_attribute("text_imf", visible=True) frame.set_state_attribute("test_imf", visible=True, description="Test selected IMF file", **button_style) def load_imf_handler(widget): frame.set_state("load_custom_imf") options = ["", "Preset: default"] + list_custom_imf() frame.set_attributes("list_imfs", options=options, value="", selected_label="") def sel_custom_imf(name, value): if value != "": frame.set_state("custom_imf") if value == "Preset: default": frame.set_attributes("name_imf", value="") frame.set_attributes("text_imf", value=default_custom_imf_text) else: text = "" with open(custom_imf_dir + value + ".py", "r") as fin: text = fin.read() frame.set_attributes("name_imf", value=value) frame.set_attributes("text_imf", value=text) def save_imf_handler(widget): clear_output() pyplot.close("all") imf_name = frame.get_attribute("name_imf", "value") if imf_name == "": print("Error: not IMF name given") else: if not os.path.isdir(custom_imf_dir): os.mkdir(custom_imf_dir) with open(custom_imf_dir + imf_name + ".py", "w") as fout: fout.write(frame.get_attribute("text_imf", "value")) print("Custom IMF saved to " + imf_name) frame.set_state_attribute('imf_type', options=['salpeter', 'chabrier', 'kroupa', 'alphaimf'] + list_custom_imf()) def delete_imf_handler(widget): clear_output() pyplot.close("all") imf_name = frame.get_attribute("name_imf", "value") if imf_name == "": print("Error: not IMF name given") else: if os.path.isdir(custom_imf_dir): if os.path.isfile(custom_imf_dir + imf_name + ".py"): os.remove(custom_imf_dir + imf_name + ".py") frame.set_attributes("name_imf", value="") frame.set_attributes("text_imf", value=default_custom_imf_text) print("Custom IMF " + imf_name + " deleted.") else: print("Custom IMF, " + imf_name + ", not found!") frame.set_state_attribute('imf_type', options=['salpeter', 'chabrier', 'kroupa', 'alphaimf'] + list_custom_imf()) def test_imf_handler(widget): clear_output() pyplot.close("all") imf_name = frame.get_attribute("name_imf", "value") if imf_name == "": print("No IMF name given!") return try: ci = load_source("custom_imf", custom_imf_dir + imf_name + ".py") except IOError: print("Failed to load " + imf_name + ", no IMF file found.") return mass_min = 0.0 mass_max = 30.0 xaxis = numpy.linspace(mass_min, mass_max, 1000)[1:] yaxis = [0 for x in xaxis] for i, x in enumerate(xaxis): yaxis[i] = ci.custom_imf(x) pyplot.plot(xaxis, numpy.log10(yaxis)) pyplot.title("IMF file test: " + imf_name) pyplot.xlabel("mass [$M_{\odot}$]") pyplot.ylabel("IMF") pyplot.show() frame.set_state_callbacks("load_imf", load_imf_handler, attribute=None, type="on_click") frame.set_state_callbacks("list_imfs", sel_custom_imf) frame.set_state_callbacks("save_imf", save_imf_handler, attribute=None, type="on_click") frame.set_state_callbacks("delete_imf", delete_imf_handler, attribute=None, type="on_click") frame.set_state_callbacks("test_imf", test_imf_handler, attribute=None, type="on_click") frame.set_object("custom_imf_page", widgets.VBox()) frame.set_object("load_save_imf_group", widgets.HBox()) frame.set_object("load_imf", widgets.Button()) frame.set_object("list_imfs", widgets.Dropdown()) frame.set_object("name_imf", widgets.Text()) frame.set_object("save_imf", widgets.Button()) frame.set_object("delete_imf", widgets.Button()) frame.set_object("text_imf", widgets.Textarea()) frame.set_object("test_imf", widgets.Button()) frame.set_state_attribute("get_table_page", visible=True) frame.set_state_attribute("warning_msg", visible=True, value="<h3>Error: No simulation data!</h3>", **group_style) frame.set_state_attribute("warning_msg", states_plot, visible=False) frame.set_state_attribute("species_mult_group", states_sim_plot, visible=True, **group_style) frame.set_state_attribute("iso_or_elem", visible=True, description="species type: ", options=["Elements", "Isotopes"], selected_label="Elements") frame.set_state_attribute("species_mult", visible=True, description="Element: ", options=elements_sel_mult, **text_box_style) frame.set_state_attribute("get_table", states_sim_plot, visible=True, description="Get table links", **button_style) frame.set_state_attribute("table_links", states_sim_plot, visible=True, value="", **group_style) def species_mult_handler(name, value): if "All" in value: iso_or_elem = frame.get_attribute("iso_or_elem", "value") if iso_or_elem == "Elements": value = tuple(elements_all) elif iso_or_elem == "Isotopes": value = tuple(isotopes_all) frame.set_attributes("species_mult", value=value, selected_labels=value) frame.set_attributes("table_links", value="") def get_table_handler(widget): iso_or_elem = frame.get_attribute("iso_or_elem", "value") species = list(frame.get_attribute("species_mult", "value")) runs = frame.get_state_data("runs") title = "<h3>Data table links:</h3>" html = title if not os.path.isdir("./evol_tables"): os.mkdir("./evol_tables") for data, name, Z, widget_name in runs: if frame.get_attribute(widget_name, "value"): file = "evol_tables/" + widget_name.replace("#", "") + "file" if iso_or_elem == "Elements": data.write_evol_table(species, [], file, "./") elif iso_or_elem == "Isotopes": data.write_evol_table([], species, file, "./") html = html + "<p><a href=\"" + file + "\" target=\"_blank\" download>" + name + "</a></p>\n" if html == title: html = "" print("No runs selected.") frame.set_attributes("table_links", value=html) clear_output() pyplot.close("all") frame.set_state_callbacks("species_mult", species_mult_handler) frame.set_state_callbacks("get_table", get_table_handler, attribute=None, type="on_click") frame.set_object("species_mult_group", widgets.VBox()) frame.set_object("species_mult", widgets.SelectMultiple()) frame.set_object("get_table_page", widgets.VBox()) frame.set_object("get_table", widgets.Button()) frame.set_object("table_links", widgets.HTML()) ##start widget## frame.display_object("window")