Changeset 4fdcc65 in sasview for src/sas/sascalc/dataloader/readers/cansas_reader_HDF5.py

Timestamp:

Sep 12, 2018 1:13:34 PM (6 years ago)

Author:

krzywon

Branches:

master, magnetic_scatt, release-4.2.2, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, unittest-saveload

Children:

2ca5d57b

Parents:

8d5e11c

Message:

File loader fixes and code cleanup.

File:

• src/sas/sascalc/dataloader/readers/cansas_reader_HDF5.py (modified) (8 diffs)

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

@@ -12 +12 @@
     Data1D, Data2D, DataInfo, Process, Aperture, Collimation, \
     TransmissionSpectrum, Detector
-from ..data_info import combine_data_info_with_plottable
 from ..loader_exceptions import FileContentsException, DefaultReaderException
 from ..file_reader_base_class import FileReader, decode
 
+
 def h5attr(node, key, default=None):
     return decode(node.attrs.get(key, default))
+
 
 class Reader(FileReader):
@@ -71 +72 @@
                 except Exception as e:
                     if extension not in self.ext:
-                        msg = "NXcanSAS HDF5 Reader could not load file {}".format(basename + extension)
+                        msg = "NXcanSAS Reader could not load file {}".format(
+                            basename + extension)
                         raise DefaultReaderException(msg)
                     raise FileContentsException(e.message)
@@ -85 +87 @@
                     self.raw_data.close()
 
-                for dataset in self.output:
-                    if isinstance(dataset, Data1D):
-                        if dataset.x.size < 5:
-                            self.output = []
-                            raise FileContentsException("Fewer than 5 data points found.")
+                for data_set in self.output:
+                    if isinstance(data_set, Data1D):
+                        if data_set.x.size < 5:
+                            exception = FileContentsException(
+                                "Fewer than 5 data points found.")
+                            data_set.errors.append(exception)
 
     def reset_state(self):
@@ -105 +108 @@
         self.i_name = u''
         self.i_node = u''
-        self.q_uncertainties = []
-        self.q_resolutions = []
+        self.q_uncertainties = None
+        self.q_resolutions = None
         self.i_uncertainties = u''
         self.parent_class = u''
@@ -257 +260 @@
     def process_2d_data_object(self, data_set, key, unit):
         if key == self.i_name:
-            self.current_dataset.x_bins, self.current_dataset.y_bins = \
-                data_set.shape
-            self.current_dataset.data = data_set.flatten()
+            self.current_dataset.data = data_set
             self.current_dataset.zaxis("Intensity", unit)
         elif key == self.i_uncertainties:
@@ -268 +269 @@
             if self.q_name[0] == self.q_name[1]:
                 # All q data in a single array
-                self.current_dataset.qx_data = data_set[0].flatten()
-                self.current_dataset.qy_data = data_set[1].flatten()
+                self.current_dataset.qx_data = data_set[0]
+                self.current_dataset.qy_data = data_set[1]
             elif self.q_name.index(key) == 0:
-                self.current_dataset.qx_data = data_set.flatten()
+                self.current_dataset.qx_data = data_set
             elif self.q_name.index(key) == 1:
-                self.current_dataset.qy_data = data_set.flatten()
+                self.current_dataset.qy_data = data_set
         elif key in self.q_uncertainties or key in self.q_resolutions:
             if ((self.q_uncertainties[0] == self.q_uncertainties[1]) or
@@ -537 +538 @@
             if dataset.data.ndim == 2:
                 (n_rows, n_cols) = dataset.data.shape
-                dataset.y_bins = dataset.qy_data[0::n_cols]
-                dataset.x_bins = dataset.qx_data[0::n_rows]
+                print(n_rows)
+                print(n_cols)
+                flat_qy = dataset.qy_data[0::n_cols].flatten()
+                if flat_qy[0] == flat_qy[1]:
+                    flat_qy = np.transpose(dataset.qy_data)[0::n_cols].flatten()
+                dataset.y_bins = np.unique(flat_qy)
+                flat_qx = dataset.qx_data[0::n_rows].flatten()
+                if flat_qx[0] == flat_qx[1]:
+                    flat_qx = np.transpose(dataset.qx_data)[0::n_rows].flatten()
+                dataset.x_bins = np.unique(flat_qx)
+                print(dataset.x_bins)
+                print(len(dataset.x_bins))
+                print(dataset.y_bins)
+                print(len(dataset.y_bins))
                 dataset.data = dataset.data.flatten()
+                dataset.qx_data = dataset.qx_data.flatten()
+                dataset.qy_data = dataset.qy_data.flatten()
             self.current_dataset = dataset
             self.send_to_output()
@@ -589 +604 @@
         try:
             iter(iterable)
-            if (not isinstance(iterable, np.ndarray)) or (isinstance(iterable, str)
-                    or isinstance(iterable, unicode)):
+            if (not isinstance(iterable, np.ndarray) and not isinstance(
+                    iterable, list)) or (isinstance(iterable, str) or
+                                         isinstance(iterable, unicode)):
                 raise TypeError
         except TypeError: