check range before plotting

vis.py

@@ -202,6 +202,8 @@ def draw_figures(out_folder, job_name, fig_range_dict, counts_by_element, symbol
     # ------------------------------
     # TODO: limit drawing to ranges
     profile_mean_range = fig_range_dict['profile_mean_range']
+    if profile_mean_range[1] == math.inf:
+        profile_mean_range[1] = collect_range[1]
     if profile_mean_range:
         # make a new figure in slot 1
         figure_mean = plt.figure(1, figsize=(7, 7))
@@ -272,6 +274,8 @@ def draw_figures(out_folder, job_name, fig_range_dict, counts_by_element, symbol
                                       repeat=False, interval=100)
 
     energy_range = fig_range_dict['energy_range']
+    if energy_range[1] == math.inf:
+        energy_range[1] = collect_range[1]
     if energy_range:
         # make a new figure in slot 3
         figure_energy = plt.figure(3, figsize=(7, 7))