Update plotspectra.py

artis-mcrt · Mar 7, 2025 · 80da12d · 80da12d
1 parent ed45526
commit 80da12d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/artistools/spectra/plotspectra.py b/artistools/spectra/plotspectra.py
@@ -426,7 +426,9 @@ def plot_artis_spectrum(
             if plotpacketcount:
                 dfspectrum = dfspectrum.with_columns(y=pl.col("packetcount"))
 
-            dfspectrum = dfspectrum.filter(pl.col("x").is_between(supxmin * 0.9, supxmax * 1.1))
+            dfspectrum = dfspectrum.filter(pl.col("x").is_between(supxmin * 0.9, supxmax * 1.1)).sort(
+                "x", maintain_order=True
+            )
 
             atspectra.print_integrated_flux(dfspectrum["y"], dfspectrum["x"])
 
@@ -454,8 +456,6 @@ def plot_artis_spectrum(
 
                 dfspectrum = pl.DataFrame({"x": new_lambda_angstroms, "y": binned_flux})
 
-            dfspectrum = dfspectrum.sort("x", maintain_order=True)
-
             axis.plot(
                 dfspectrum["x"], dfspectrum["y"], label=linelabel_withdirbin if axindex == 0 else None, **plotkwargs
             )

diff --git a/artistools/spectra/spectra.py b/artistools/spectra/spectra.py
@@ -102,10 +102,10 @@ def get_exspec_bins(
 
 
 def convert_angstroms_to_units(value_angstroms: float, new_units: str) -> float:
-    h = 4.1356677e-15  # Planck's constant [eV s]
     c = 2.99792458e18  # speed of light [angstroms/s]
-    hc_ev_angstroms = h * c  # [eV angstroms]
     if new_units.lower() == "kev":
+        h = 4.1356677e-15  # Planck's constant [eV s]
+        hc_ev_angstroms = h * c  # [eV angstroms]
         return hc_ev_angstroms / value_angstroms / 1e3  # [keV]
     if new_units.lower() == "hz":
         return c / value_angstroms