From 4cd0a8307ab0fd55692a67ee0ebebfd02df58bef Mon Sep 17 00:00:00 2001
From: Niv Drory <drory@astro.as.utexas.edu>
Date: Wed, 2 Oct 2024 09:22:19 -0500
Subject: [PATCH] correct gaia resolution, add example notebook

---
 examples/lco_com/gaia_conv.ipynb         | 89 ++++++++++++++++++++++++
 python/lvmdrp/functions/fluxCalMethod.py |  2 +-
 2 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 examples/lco_com/gaia_conv.ipynb

diff --git a/examples/lco_com/gaia_conv.ipynb b/examples/lco_com/gaia_conv.ipynb
new file mode 100644
index 00000000..6db6a527
--- /dev/null
+++ b/examples/lco_com/gaia_conv.ipynb
@@ -0,0 +1,89 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import numpy as np\n",
+    "import matplotlib.pyplot as plt\n",
+    "import lvmdrp.core.fluxcal as fluxcal\n",
+    "from lvmdrp.functions.fluxCalMethod import get_bprp_convolution_matrix\n",
+    "from astropy.io import fits\n",
+    "from astropy.table import Table\n",
+    "\n",
+    "with fits.open('/Users/droryn/work/LVM/data/sas/sdsswork/lvm/spectro/redux/1.1.0dev/1039XX/1039949/60356/lvmCFrame-00012618.fits') as hdu:\n",
+    "    d = hdu['FLUX'].data\n",
+    "    w = hdu['WAVE'].data\n",
+    "    slitmap = Table(hdu['SLITMAP'].data)\n",
+    "\n",
+    "gw, gf = fluxcal.retrive_gaia_star('5611678276101933056', GAIA_CACHE_DIR='/Users/droryn/work/LVM/data/sas/sdsswork/lvm/sandbox/calib/gaia_cache')\n",
+    "stdflux = np.interp(w, gw, gf)  # interpolate to our wavelength grid\n",
+    "f = d[1297,:]\n",
+    "plt.plot(w,f, color='blue', linewidth=1)\n",
+    "plt.plot(gw,gf, color='red', linewidth=1)\n",
+    "\n",
+    "bprp_conv = get_bprp_convolution_matrix(w)\n",
+    "spec_c = bprp_conv@f\n",
+    "\n",
+    "plt.plot(w,spec_c, color='green')\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "slitmap[slitmap['orig_ifulabel']=='P1-2']"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "s = '/Users/droryn/prog/lvm/lvmcore/etc/Gaia_BPRP_resolution.txt'\n",
+    "txt = np.genfromtxt(s)\n",
+    "l_bprp, r_bprp = txt[:, 0], txt[:, 1]\n",
+    "sigmas = l_bprp/r_bprp/2.35\n",
+    "plt.plot(l_bprp, sigmas)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "lvmdrp",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.12.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}
diff --git a/python/lvmdrp/functions/fluxCalMethod.py b/python/lvmdrp/functions/fluxCalMethod.py
index 0b59ed05..986c41db 100644
--- a/python/lvmdrp/functions/fluxCalMethod.py
+++ b/python/lvmdrp/functions/fluxCalMethod.py
@@ -646,7 +646,7 @@ def get_bprp_convolution_matrix(w):
 
     Input: w numpy.ndarray of wavelengths for target spectrum
     '''
-    s = os.getenv('LVMCORE_DIR')+'/etc/Gaia1'
+    s = os.getenv('LVMCORE_DIR')+'/etc/Gaia_BPRP_resolution.txt'
     log.info(f"Reading GAIA PB-RP resolution {s}")
     txt = np.genfromtxt(s)
     l_bprp, r_bprp = txt[:, 0], txt[:, 1]