Skip to content

Commit

Permalink
Sphere SDDR atlas repaired, needs testing for materials outputs
Browse files Browse the repository at this point in the history
  • Loading branch information
Wheeler, Dylan committed Feb 21, 2024
1 parent 6925765 commit f375a44
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions jade/sphereoutput.py
Original file line number Diff line number Diff line change
Expand Up @@ -1489,18 +1489,17 @@ def _get_organized_output(self):
"""
zaids = []
for code, library_outputs in self.outputs.items():
print(library_outputs)
for (zaidnum, mt, lib), outputslib in library_outputs.items():
zaids.append((zaidnum, mt))

zaids = list(set(zaids))
libs = [] # Not used
outputs = [] # Not used

return zaids, libs, outputs
return libs, zaids, outputs

def _generate_single_plots(self, outpath):
allzaids, libs, outputs = self._get_organized_output()
libs, allzaids, outputs = self._get_organized_output()
globalname = self.lib
self._generate_plots(libs, allzaids, outputs, globalname, outpath)

Expand Down Expand Up @@ -1543,6 +1542,7 @@ def _generate_plots(self, libraries, allzaids, outputs, globalname, outpath):
fluxquantity = "Photon Flux"
fluxunit = r"$p/(cm^2\cdot\#_S)$"
allzaids.sort()
print(allzaids)
# --- Binned plots of the gamma flux ---
for zaidnum, mt in tqdm(allzaids, desc=" Binned flux plots"):
# Get everything for the title of the zaid
Expand All @@ -1559,6 +1559,8 @@ def _generate_plots(self, libraries, allzaids, outputs, globalname, outpath):
times = self.times
atlas.doc.add_heading(title, level=2)

print(times)

for time in times:
atlas.doc.add_heading("Cooldown time = {}".format(time), level=3)
title = "Gamma Leakage flux after a {} cooldown".format(time)
Expand Down Expand Up @@ -1849,8 +1851,7 @@ def _compute_compare_result(self, reflib, tarlib):
error_dfs.append(error_df)
lib_dics.append(outputs)
for dic in lib_dics:
code_outputs.update(dic)
print(code_outputs)
code_outputs.update(dic)
self.outputs["d1s"] = code_outputs
# Consider only common zaids
idx1 = comp_dfs[0].index
Expand Down

0 comments on commit f375a44

Please sign in to comment.