[959eb01] | 1 | """ |
---|
| 2 | """ |
---|
| 3 | ##################################################################### |
---|
| 4 | #This software was developed by the University of Tennessee as part of the |
---|
| 5 | #Distributed Data Analysis of Neutron Scattering Experiments (DANSE) |
---|
| 6 | #project funded by the US National Science Foundation. |
---|
| 7 | #See the license text in license.txt |
---|
| 8 | #copyright 2008, University of Tennessee |
---|
| 9 | ###################################################################### |
---|
| 10 | |
---|
| 11 | import numpy as np |
---|
| 12 | import os |
---|
| 13 | from sas.sascalc.dataloader.data_info import Data1D |
---|
| 14 | from sas.sascalc.dataloader.data_info import Detector |
---|
| 15 | |
---|
| 16 | has_converter = True |
---|
| 17 | try: |
---|
| 18 | from sas.sascalc.data_util.nxsunit import Converter |
---|
| 19 | except: |
---|
| 20 | has_converter = False |
---|
| 21 | |
---|
| 22 | |
---|
| 23 | class Reader: |
---|
| 24 | """ |
---|
| 25 | Class to load IGOR reduced .ABS files |
---|
| 26 | """ |
---|
| 27 | ## File type |
---|
| 28 | type_name = "IGOR 1D" |
---|
| 29 | ## Wildcards |
---|
| 30 | type = ["IGOR 1D files (*.abs)|*.abs"] |
---|
| 31 | ## List of allowed extensions |
---|
| 32 | ext = ['.abs', '.ABS'] |
---|
| 33 | |
---|
| 34 | def read(self, path): |
---|
| 35 | """ |
---|
| 36 | Load data file. |
---|
| 37 | |
---|
| 38 | :param path: file path |
---|
| 39 | |
---|
| 40 | :return: Data1D object, or None |
---|
| 41 | |
---|
| 42 | :raise RuntimeError: when the file can't be opened |
---|
| 43 | :raise ValueError: when the length of the data vectors are inconsistent |
---|
| 44 | """ |
---|
| 45 | if os.path.isfile(path): |
---|
| 46 | basename = os.path.basename(path) |
---|
| 47 | root, extension = os.path.splitext(basename) |
---|
| 48 | if extension.lower() in self.ext: |
---|
| 49 | try: |
---|
| 50 | input_f = open(path,'r') |
---|
| 51 | except: |
---|
| 52 | raise RuntimeError, "abs_reader: cannot open %s" % path |
---|
| 53 | buff = input_f.read() |
---|
| 54 | lines = buff.split('\n') |
---|
| 55 | x = np.zeros(0) |
---|
| 56 | y = np.zeros(0) |
---|
| 57 | dy = np.zeros(0) |
---|
| 58 | dx = np.zeros(0) |
---|
| 59 | output = Data1D(x, y, dy=dy, dx=dx) |
---|
| 60 | detector = Detector() |
---|
| 61 | output.detector.append(detector) |
---|
| 62 | output.filename = basename |
---|
| 63 | |
---|
| 64 | is_info = False |
---|
| 65 | is_center = False |
---|
| 66 | is_data_started = False |
---|
| 67 | |
---|
| 68 | data_conv_q = None |
---|
| 69 | data_conv_i = None |
---|
| 70 | |
---|
| 71 | if has_converter == True and output.x_unit != '1/A': |
---|
| 72 | data_conv_q = Converter('1/A') |
---|
| 73 | # Test it |
---|
| 74 | data_conv_q(1.0, output.x_unit) |
---|
| 75 | |
---|
| 76 | if has_converter == True and output.y_unit != '1/cm': |
---|
| 77 | data_conv_i = Converter('1/cm') |
---|
| 78 | # Test it |
---|
| 79 | data_conv_i(1.0, output.y_unit) |
---|
| 80 | |
---|
| 81 | for line in lines: |
---|
| 82 | |
---|
| 83 | # Information line 1 |
---|
| 84 | if is_info == True: |
---|
| 85 | is_info = False |
---|
| 86 | line_toks = line.split() |
---|
| 87 | |
---|
| 88 | # Wavelength in Angstrom |
---|
| 89 | try: |
---|
| 90 | value = float(line_toks[1]) |
---|
| 91 | if has_converter == True and \ |
---|
| 92 | output.source.wavelength_unit != 'A': |
---|
| 93 | conv = Converter('A') |
---|
| 94 | output.source.wavelength = conv(value, |
---|
| 95 | units=output.source.wavelength_unit) |
---|
| 96 | else: |
---|
| 97 | output.source.wavelength = value |
---|
| 98 | except: |
---|
| 99 | #goes to ASC reader |
---|
| 100 | msg = "abs_reader: cannot open %s" % path |
---|
| 101 | raise RuntimeError, msg |
---|
| 102 | |
---|
| 103 | # Distance in meters |
---|
| 104 | try: |
---|
| 105 | value = float(line_toks[3]) |
---|
| 106 | if has_converter == True and \ |
---|
| 107 | detector.distance_unit != 'm': |
---|
| 108 | conv = Converter('m') |
---|
| 109 | detector.distance = conv(value, |
---|
| 110 | units=detector.distance_unit) |
---|
| 111 | else: |
---|
| 112 | detector.distance = value |
---|
| 113 | except: |
---|
| 114 | #goes to ASC reader |
---|
| 115 | msg = "abs_reader: cannot open %s" % path |
---|
| 116 | raise RuntimeError, msg |
---|
| 117 | # Transmission |
---|
| 118 | try: |
---|
| 119 | output.sample.transmission = float(line_toks[4]) |
---|
| 120 | except: |
---|
| 121 | # Transmission is not a mandatory entry |
---|
| 122 | pass |
---|
| 123 | |
---|
| 124 | # Thickness in mm |
---|
| 125 | try: |
---|
| 126 | value = float(line_toks[5]) |
---|
| 127 | if has_converter == True and \ |
---|
| 128 | output.sample.thickness_unit != 'cm': |
---|
| 129 | conv = Converter('cm') |
---|
| 130 | output.sample.thickness = conv(value, |
---|
| 131 | units=output.sample.thickness_unit) |
---|
| 132 | else: |
---|
| 133 | output.sample.thickness = value |
---|
| 134 | except: |
---|
| 135 | # Thickness is not a mandatory entry |
---|
| 136 | pass |
---|
| 137 | |
---|
| 138 | #MON CNT LAMBDA DET ANG DET DIST TRANS THICK |
---|
| 139 | # AVE STEP |
---|
| 140 | if line.count("LAMBDA") > 0: |
---|
| 141 | is_info = True |
---|
| 142 | |
---|
| 143 | # Find center info line |
---|
| 144 | if is_center == True: |
---|
| 145 | is_center = False |
---|
| 146 | line_toks = line.split() |
---|
| 147 | # Center in bin number |
---|
| 148 | center_x = float(line_toks[0]) |
---|
| 149 | center_y = float(line_toks[1]) |
---|
| 150 | |
---|
| 151 | # Bin size |
---|
| 152 | if has_converter == True and \ |
---|
| 153 | detector.pixel_size_unit != 'mm': |
---|
| 154 | conv = Converter('mm') |
---|
| 155 | detector.pixel_size.x = conv(5.0, |
---|
| 156 | units=detector.pixel_size_unit) |
---|
| 157 | detector.pixel_size.y = conv(5.0, |
---|
| 158 | units=detector.pixel_size_unit) |
---|
| 159 | else: |
---|
| 160 | detector.pixel_size.x = 5.0 |
---|
| 161 | detector.pixel_size.y = 5.0 |
---|
| 162 | |
---|
| 163 | # Store beam center in distance units |
---|
| 164 | # Det 640 x 640 mm |
---|
| 165 | if has_converter == True and \ |
---|
| 166 | detector.beam_center_unit != 'mm': |
---|
| 167 | conv = Converter('mm') |
---|
| 168 | detector.beam_center.x = conv(center_x * 5.0, |
---|
| 169 | units=detector.beam_center_unit) |
---|
| 170 | detector.beam_center.y = conv(center_y * 5.0, |
---|
| 171 | units=detector.beam_center_unit) |
---|
| 172 | else: |
---|
| 173 | detector.beam_center.x = center_x * 5.0 |
---|
| 174 | detector.beam_center.y = center_y * 5.0 |
---|
| 175 | |
---|
| 176 | # Detector type |
---|
| 177 | try: |
---|
| 178 | detector.name = line_toks[7] |
---|
| 179 | except: |
---|
| 180 | # Detector name is not a mandatory entry |
---|
| 181 | pass |
---|
| 182 | |
---|
| 183 | #BCENT(X,Y) A1(mm) A2(mm) A1A2DIST(m) DL/L |
---|
| 184 | # BSTOP(mm) DET_TYP |
---|
| 185 | if line.count("BCENT") > 0: |
---|
| 186 | is_center = True |
---|
| 187 | |
---|
| 188 | # Parse the data |
---|
| 189 | if is_data_started == True: |
---|
| 190 | toks = line.split() |
---|
| 191 | |
---|
| 192 | try: |
---|
| 193 | _x = float(toks[0]) |
---|
| 194 | _y = float(toks[1]) |
---|
| 195 | _dy = float(toks[2]) |
---|
| 196 | _dx = float(toks[3]) |
---|
| 197 | |
---|
| 198 | if data_conv_q is not None: |
---|
| 199 | _x = data_conv_q(_x, units=output.x_unit) |
---|
| 200 | _dx = data_conv_i(_dx, units=output.x_unit) |
---|
| 201 | |
---|
| 202 | if data_conv_i is not None: |
---|
| 203 | _y = data_conv_i(_y, units=output.y_unit) |
---|
| 204 | _dy = data_conv_i(_dy, units=output.y_unit) |
---|
| 205 | |
---|
| 206 | x = np.append(x, _x) |
---|
| 207 | y = np.append(y, _y) |
---|
| 208 | dy = np.append(dy, _dy) |
---|
| 209 | dx = np.append(dx, _dx) |
---|
| 210 | |
---|
| 211 | except: |
---|
| 212 | # Could not read this data line. If we are here |
---|
| 213 | # it is because we are in the data section. Just |
---|
| 214 | # skip it. |
---|
| 215 | pass |
---|
| 216 | |
---|
| 217 | #The 6 columns are | Q (1/A) | I(Q) (1/cm) | std. dev. |
---|
| 218 | # I(Q) (1/cm) | sigmaQ | meanQ | ShadowFactor| |
---|
| 219 | if line.count("The 6 columns") > 0: |
---|
| 220 | is_data_started = True |
---|
| 221 | |
---|
| 222 | # Sanity check |
---|
| 223 | if not len(y) == len(dy): |
---|
| 224 | msg = "abs_reader: y and dy have different length" |
---|
| 225 | raise ValueError, msg |
---|
| 226 | # If the data length is zero, consider this as |
---|
| 227 | # though we were not able to read the file. |
---|
| 228 | if len(x) == 0: |
---|
| 229 | raise ValueError, "ascii_reader: could not load file" |
---|
| 230 | |
---|
| 231 | output.x = x[x != 0] |
---|
| 232 | output.y = y[x != 0] |
---|
| 233 | output.dy = dy[x != 0] |
---|
| 234 | output.dx = dx[x != 0] |
---|
| 235 | if data_conv_q is not None: |
---|
| 236 | output.xaxis("\\rm{Q}", output.x_unit) |
---|
| 237 | else: |
---|
| 238 | output.xaxis("\\rm{Q}", 'A^{-1}') |
---|
| 239 | if data_conv_i is not None: |
---|
| 240 | output.yaxis("\\rm{Intensity}", output.y_unit) |
---|
| 241 | else: |
---|
| 242 | output.yaxis("\\rm{Intensity}", "cm^{-1}") |
---|
| 243 | |
---|
| 244 | # Store loading process information |
---|
| 245 | output.meta_data['loader'] = self.type_name |
---|
| 246 | return output |
---|
| 247 | else: |
---|
| 248 | raise RuntimeError, "%s is not a file" % path |
---|
| 249 | return None |
---|