From 2acbdefc62c106534565d7288aa5f3a227bb351d Mon Sep 17 00:00:00 2001
From: Hugo Buddelmeijer <hugo@buddelmeijer.nl>
Date: Fri, 13 Sep 2019 12:26:09 +0000
Subject: [PATCH] Add some initial support for CreateVarDqLayers_make
---
src/micado/main/DataExtensionImage.py | 12 +++++
src/micado/recipes/CreateVarDqLayers_make.py | 57 ++++++++++++++++++++
2 files changed, 69 insertions(+)
create mode 100644 src/micado/recipes/CreateVarDqLayers_make.py
diff --git a/src/micado/main/DataExtensionImage.py b/src/micado/main/DataExtensionImage.py
index 7dfc6508e..3263ffb25 100644
--- a/src/micado/main/DataExtensionImage.py
+++ b/src/micado/main/DataExtensionImage.py
@@ -37,3 +37,15 @@ class DataExtensionImage(DataExtension):
self.NAXIS2 = 2 * int(numpy.round(detector.yhw / detector.pixsize))
if self.EXTNAME == '':
self.EXTNAME = detector.id
+
+
+class DataExtensionVariance(DataExtensionImage):
+ """
+ Variance layer of the image.
+ """
+
+
+class DataExtensionQuality(DataExtensionImage):
+ """
+ Quality layer of the image.
+ """
diff --git a/src/micado/recipes/CreateVarDqLayers_make.py b/src/micado/recipes/CreateVarDqLayers_make.py
new file mode 100644
index 000000000..9bc82c111
--- /dev/null
+++ b/src/micado/recipes/CreateVarDqLayers_make.py
@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+"""
+CreateVarDqLayers
+"""
+
+import numpy
+
+from micado.main.Img import Img
+from micado.main.DataExtensionImage import (
+ DataExtensionImage, DataExtensionVariance, DataExtensionQuality
+)
+
+
+def make(raw):
+ """
+ CreateVarDqLayers
+ """
+ extensions = [
+ DataExtensionImage(
+ NAXIS=rawextension.NAXIS,
+ NAXIS1=rawextension.NAXIS1,
+ NAXIS2=rawextension.NAXIS2,
+ BITPIX=rawextension.BITPIX,
+ data=rawextension.data,
+ )
+ for rawextension in raw.extensions
+ ]
+ variances = [
+ DataExtensionVariance(
+ NAXIS=rawextension.NAXIS,
+ NAXIS1=rawextension.NAXIS1,
+ NAXIS2=rawextension.NAXIS2,
+ BITPIX=rawextension.BITPIX,
+ data=rawextension.data,
+ )
+ for rawextension in raw.extensions
+ ]
+ qualities = [
+ DataExtensionQuality(
+ NAXIS=rawextension.NAXIS,
+ NAXIS1=rawextension.NAXIS1,
+ NAXIS2=rawextension.NAXIS2,
+ BITPIX=rawextension.BITPIX,
+ data=numpy.zeros(
+ (rawextension.NAXIS1, rawextension.NAXIS2),
+ dtype="int32",
+ ),
+ )
+ for rawextension in raw.extensions
+ ]
+
+ img = Img()
+ img.extensionn = extensions
+ img.variances = variances
+ img.qualities = qualities
+
+ return img
--
GitLab