From 11b23127eefad95050491a54a63a9e6ae4d6daa1 Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Thu, 21 Jul 2022 16:07:26 +0200
Subject: [PATCH] keep nan in plot for bad pixel code fig

---
 notebooks/LPD/LPDChar_Darks_NBC.ipynb | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/notebooks/LPD/LPDChar_Darks_NBC.ipynb b/notebooks/LPD/LPDChar_Darks_NBC.ipynb
index 1122d39ea..277d5231c 100644
--- a/notebooks/LPD/LPDChar_Darks_NBC.ipynb
+++ b/notebooks/LPD/LPDChar_Darks_NBC.ipynb
@@ -722,6 +722,8 @@
     "                title = '{} value'.format(const)\n",
     "                if const == 'BadPixelsDark':\n",
     "                    vmax = 4\n",
+    "                    bpix_code = data.astype(np.float32)\n",
+    "                    bpix_code[bpix_code == 0] = np.nan\n",
     "                    title = 'Bad pixel code'\n",
     "                    label = title\n",
     "\n",
@@ -730,12 +732,12 @@
     "                                 '3 {}'.format(BadPixels.OFFSET_OUT_OF_THRESHOLD.name),\n",
     "                                 '4 {}'.format('MIXED')]\n",
     "\n",
-    "                    heatmapPlot(data, add_panels=False, cmap='viridis',\n",
+    "                    heatmapPlot(bpix_code, add_panels=False, cmap='viridis',\n",
     "                                y_label='Rows', x_label='Columns',\n",
     "                                lut_label='', vmax=vmax,\n",
     "                                use_axis=ax, cb_ticklabels=cb_labels, cb_ticks = np.arange(4)+1,\n",
     "                                title='{}'.format(title))\n",
-    "\n",
+    "                    del bpix_code\n",
     "                else:\n",
     "\n",
     "                    heatmapPlot(data, add_panels=False, cmap='viridis',\n",
@@ -1157,7 +1159,7 @@
     "                l_data.append(np.count_nonzero(badpix_g[cap][qm][:,:,:,gain] & bit.value))\n",
     "            \n",
     "            if old_const[cap][qm]['BadPixelsDark'] is not None:\n",
-    "                old_const[cap][qm]['BadPixelsDark'] = old_const[cap][qm]['BadPixelsDark'].astype(np.float32)\n",
+    "                old_const[cap][qm]['BadPixelsDark'] = old_const[cap][qm]['BadPixelsDark'].astype(np.uint32)\n",
     "                dataold = np.copy(old_const[cap][qm]['BadPixelsDark'][:, :, :, gain])\n",
     "                l_data_old.append(len(dataold[dataold>0].flatten()))\n",
     "                for bit in bits:\n",
-- 
GitLab