Changeset 83c09af in sasview for src/sas/sascalc/dataloader/readers/cansas_reader.py

Timestamp:

Jan 3, 2017 11:12:00 AM (7 years ago)

Author:

krzywon

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

af08e55

Parents:

c1681ea

Message:

Fit states and projects are now able to save 2D data. Loading is still not working. #827

File:

• src/sas/sascalc/dataloader/readers/cansas_reader.py (modified) (5 diffs)

src/sas/sascalc/dataloader/readers/cansas_reader.py

@@ -20 +20 @@
 import inspect
 # For saving individual sections of data
-from sas.sascalc.dataloader.data_info import Data1D, DataInfo, plottable_1D
-from sas.sascalc.dataloader.data_info import Collimation, TransmissionSpectrum, Detector, Process, Aperture
-from sas.sascalc.dataloader.data_info import combine_data_info_with_plottable as combine_data
+from sas.sascalc.dataloader.data_info import Data1D, Data2D, DataInfo, \
+    plottable_1D
+from sas.sascalc.dataloader.data_info import Collimation, TransmissionSpectrum, \
+    Detector, Process, Aperture
+from sas.sascalc.dataloader.data_info import \
+    combine_data_info_with_plottable as combine_data
 import sas.sascalc.dataloader.readers.xml_reader as xml_reader
 from sas.sascalc.dataloader.readers.xml_reader import XMLreader
@@ -792 +795 @@
         :param datainfo: Data1D object
         """
-        if not issubclass(datainfo.__class__, Data1D):
-            raise RuntimeError, "The cansas writer expects a Data1D instance"
+        if issubclass(datainfo.__class__, Data2D):
+            is_2d = True
 
         # Get PIs and create root element
@@ -813 +816 @@
         self._write_run_names(datainfo, entry_node)
         # Add Data info to SASEntry
-        self._write_data(datainfo, entry_node)
+        if is_2d:
+            self._write_data_2d(datainfo, entry_node)
+        else:
+            self._write_data(datainfo, entry_node)
         # Transmission Spectrum Info
         self._write_trans_spectrum(datainfo, entry_node)
@@ -907 +913 @@
     def _write_data(self, datainfo, entry_node):
         """
-        Writes the I and Q data to the XML file
+        Writes 1D I and Q data to the XML file
 
         :param datainfo: The Data1D object the information is coming from
@@ -935 +941 @@
                 self.write_node(point, "dQl", datainfo.dxl[i],
                                 {'unit': datainfo.x_unit})
+
+    def _write_data_2d(self, datainfo, entry_node):
+        """
+        Writes 2D data to the XML file
+
+        :param datainfo: The Data2D object the information is coming from
+        :param entry_node: lxml node ElementTree object to be appended to
+        """
+        node = self.create_element("SASdata")
+        self.append(node, entry_node)
+
+        for i in range(len(datainfo.data)):
+            point = self.create_element("Idata")
+            node.append(point)
+            self.write_node(point, "Qx", datainfo.qx_data[i],
+                            {'unit': datainfo._xunit})
+            self.write_node(point, "Qy", datainfo.qy_data[i],
+                            {'unit': datainfo._yunit})
+            self.write_node(point, "I", datainfo.data[i],
+                            {'unit': datainfo._zunit})
+            if datainfo.err_data is not None and len(datainfo.err_data) > i:
+                self.write_node(point, "Idev", datainfo.err_data[i],
+                                {'unit': datainfo._zunit})
+            if datainfo.dqy_data is not None and len(datainfo.dqy_data) > i:
+                self.write_node(point, "Qydev", datainfo.dqy_data[i],
+                                {'unit': datainfo._yunit})
+            if datainfo.dqx_data is not None and len(datainfo.dqx_data) > i:
+                self.write_node(point, "Qxdev", datainfo.dqx_data[i],
+                                {'unit': datainfo._xunit})
+            if datainfo.mask is not None and len(datainfo.mask) > i:
+                self.write_node(point, "Mask", datainfo.err_data[i])
 
     def _write_trans_spectrum(self, datainfo, entry_node):