From ce35f3a25174ae45cb308c8e1d5eb6b15e8c28f4 Mon Sep 17 00:00:00 2001
From: Ansgar Wehrhahn <31626864+AWehrhahn@users.noreply.github.com>
Date: Tue, 4 May 2021 16:05:43 +0200
Subject: [PATCH] fix getLineOpacities sizes

---
 src/pysme/sme_synth.py | 36 +++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 15 deletions(-)

diff --git a/src/pysme/sme_synth.py b/src/pysme/sme_synth.py
index 985991ef..9dd07e64 100644
--- a/src/pysme/sme_synth.py
+++ b/src/pysme/sme_synth.py
@@ -377,24 +377,27 @@ def Opacity(self, getData=False, motype=1):
 
         Returns
         -------
-        copblu : array of size (nmu,)
+        copblu : array of size (nrhox,)
             only if getData is True
-        copred : array of size (nmu,)
+        copred : array of size (nrhox,)
             only if getData is True
-        copstd : array of size (nmu,)
+        copstd : array of size (nrhox,)
             only if getData is True and motype is 0
         """
         args = []
         type = ""
         if getData:
-            nmu = self.nmu
-            copblu = np.zeros(nmu)
-            copred = np.zeros(nmu)
-            args = [nmu, copblu, copred]
+            assert (
+                self.atmo is not None
+            ), "Need to set the atmosphere model before retrievieng Opacities"
+            nrhox = len(self.atmo.rhox)
+            copblu = np.zeros(nrhox)
+            copred = np.zeros(nrhox)
+            args = [nrhox, copblu, copred]
             type = ["s", "d", "d"]
 
             if motype == 0:
-                copstd = np.zeros(nmu)
+                copstd = np.zeros(nrhox)
                 args += [copstd]
                 type += ["d"]
 
@@ -661,14 +664,17 @@ def GetLineOpacity(self, wave):
         csf : array
             Continuum source function
         """
-        nmu = self.nmu
-        lop = np.zeros(nmu)
-        cop = np.zeros(nmu)
-        scr = np.zeros(nmu)
-        tsf = np.zeros(nmu)
-        csf = np.zeros(nmu)
+        assert (
+            self.atmo is not None
+        ), "Need to set the atmosphere model before retrievieng Opacities"
+        nrhox = len(self.atmo.rhox)
+        lop = np.zeros(nrhox)
+        cop = np.zeros(nrhox)
+        scr = np.zeros(nrhox)
+        tsf = np.zeros(nrhox)
+        csf = np.zeros(nrhox)
         type = "dsddddd"
-        self.lib.GetLineOpacity(wave, nmu, lop, cop, scr, tsf, csf, type=type)
+        self.lib.GetLineOpacity(wave, nrhox, lop, cop, scr, tsf, csf, type=type)
         return lop, cop, scr, tsf, csf
 
     def GetLineRange(self):