Changeset e935ddb in sasview
 Timestamp:
 Apr 4, 2017 8:15:03 AM (9 months ago)
 Branches:
 master, ESS_GUI, ESS_GUI_CategroyManager, ESS_GUI_Pr, ESS_GUI_cns, ESS_GUI_corf, ESS_GUI_py2_OLD, SVCC1, SasView664, costrafo411, docs, ticket873, ticket976, ticket993, unittestsaveload
 Children:
 ecc8d1a8
 Parents:
 7cbbacd
 gitauthor:
 Adam Washington <adam.washington@…> (04/04/17 08:11:46)
 gitcommitter:
 Adam Washington <adam.washington@…> (04/04/17 08:15:03)
 File:

 1 edited
Legend:
 Unmodified
 Added
 Removed

src/sas/sascalc/dataloader/readers/sesans_reader.py
r7caf3e5 re935ddb 6 6 Jurrian Bakker 7 7 """ 8 import numpy 8 import numpy as np 9 9 import os 10 10 from sas.sascalc.dataloader.data_info import Data1D … … 52 52 _, extension = os.path.splitext(basename) 53 53 if self.allow_all or extension.lower() in self.ext: 54 try:54 with open(path, 'r') as input_f: 55 55 # Read in binary mode since GRASP frequently has noascii 56 56 # characters that brakes the open operation 57 input_f = open(path,'rb') 58 except: 59 raise RuntimeError, "sesans_reader: cannot open %s" % path 60 buff = input_f.read() 61 lines = buff.splitlines() 62 x = numpy.zeros(0) 63 y = numpy.zeros(0) 64 dy = numpy.zeros(0) 65 lam = numpy.zeros(0) 66 dlam = numpy.zeros(0) 67 dx = numpy.zeros(0) 68 69 #temp. space to sort data 70 tx = numpy.zeros(0) 71 ty = numpy.zeros(0) 72 tdy = numpy.zeros(0) 73 tlam = numpy.zeros(0) 74 tdlam = numpy.zeros(0) 75 tdx = numpy.zeros(0) 76 output = Data1D(x=x, y=y, lam=lam, dy=dy, dx=dx, dlam=dlam, isSesans=True) 77 self.filename = output.filename = basename 57 line = input_f.readline() 58 params = {} 59 while line.strip() != "": 60 if line.strip() == "": 61 break 62 terms = line.strip().split("\t") 63 params[terms[0].strip()] = " ".join(terms[1:]).strip() 64 line = input_f.readline() 65 headers_temp = input_f.readline().strip().split("\t") 66 headers = {} 67 for h in headers_temp: 68 temp = h.strip().split() 69 headers[h[:1].strip()] = temp[1][1:1] 70 data = np.loadtxt(input_f) 71 x = data[:, 0] 72 dx = data[:, 3] 73 lam = data[:, 4] 74 dlam = data[:, 5] 75 y = data[:, 1] 76 dy = data[:, 2] 78 77 79 paramnames=[] 80 paramvals=[] 81 zvals=[] 82 dzvals=[] 83 lamvals=[] 84 dlamvals=[] 85 Pvals=[] 86 dPvals=[] 78 lam_unit = _header_fetch(headers, "wavelength") 79 if lam_unit == "AA": 80 lam_unit = "A" 81 p_unit = headers["polarisation"] 87 82 88 for line in lines: 89 # Initial try for CSV (split on ,) 90 line=line.strip() 91 toks = line.split('\t') 92 if len(toks)==2: 93 paramnames.append(toks[0]) 94 paramvals.append(toks[1]) 95 if len(toks)>5: 96 zvals.append(toks[0]) 97 dzvals.append(toks[3]) 98 lamvals.append(toks[4]) 99 dlamvals.append(toks[5]) 100 Pvals.append(toks[1]) 101 dPvals.append(toks[2]) 102 else: 103 continue 83 x, x_unit = self._unit_conversion(x, lam_unit, 84 _fetch_unit(headers, 85 "spin echo length")) 86 dx, dx_unit = self._unit_conversion(dx, lam_unit, 87 _fetch_unit(headers, 88 "error SEL")) 89 dlam, dlam_unit = self._unit_conversion(dlam, lam_unit, 90 _fetch_unit(headers, 91 "error wavelength")) 92 y_unit = r'\AA^{2} cm^{1}' 104 93 105 x=[] 106 y=[] 107 lam=[] 108 dx=[] 109 dy=[] 110 dlam=[] 111 lam_header = lamvals[0].split() 112 data_conv_z = None 113 default_z_unit = "A" 114 data_conv_P = None 115 default_p_unit = " " # Adjust unit for axis (L^3) 116 lam_unit = lam_header[1].replace("[","").replace("]","") 117 if lam_unit == 'AA': 118 lam_unit = 'A' 119 varheader=[zvals[0],dzvals[0],lamvals[0],dlamvals[0],Pvals[0],dPvals[0]] 120 valrange=range(1, len(zvals)) 121 for i in valrange: 122 x.append(float(zvals[i])) 123 y.append(float(Pvals[i])) 124 lam.append(float(lamvals[i])) 125 dy.append(float(dPvals[i])) 126 dx.append(float(dzvals[i])) 127 dlam.append(float(dlamvals[i])) 94 output = Data1D(x=x, y=y, lam=lam, dy = dy, dx=dx, dlam=dlam, isSesans=True) 95 self.filename = output.filename = basename 96 output.xaxis(r"\rm{z}", x_unit) 97 output.yaxis(r"\rm{ln(P)/(t \lambda^2)}", y_unit) # Adjust label to ln P/(lam^2 t), remove lam column refs 98 # Store loading process information 99 output.meta_data['loader'] = self.type_name 100 #output.sample.thickness = float(paramvals[6]) 101 output.sample.name = params["Sample"] 102 output.sample.ID = params["DataFileTitle"] 128 103 129 x,y,lam,dy,dx,dlam = [ 130 numpy.asarray(v, 'double') 131 for v in (x,y,lam,dy,dx,dlam) 132 ] 104 output.sample.zacceptance = (float(_header_fetch(params, "Q_zmax")), 105 _fetch_unit(params, "Q_zmax")) 133 106 134 input_f.close() 135 136 output.x, output.x_unit = self._unit_conversion(x, lam_unit, default_z_unit) 137 output.y = y 138 output.y_unit = r'\AA^{2} cm^{1}' # output y_unit added 139 output.dx, output.dx_unit = self._unit_conversion(dx, lam_unit, default_z_unit) 140 output.dy = dy 141 output.lam, output.lam_unit = self._unit_conversion(lam, lam_unit, default_z_unit) 142 output.dlam, output.dlam_unit = self._unit_conversion(dlam, lam_unit, default_z_unit) 143 144 output.xaxis(r"\rm{z}", output.x_unit) 145 output.yaxis(r"\rm{ln(P)/(t \lambda^2)}", output.y_unit) # Adjust label to ln P/(lam^2 t), remove lam column refs 146 147 # Store loading process information 148 output.meta_data['loader'] = self.type_name 149 #output.sample.thickness = float(paramvals[6]) 150 output.sample.name = paramvals[1] 151 output.sample.ID = paramvals[0] 152 zaccept_unit_split = paramnames[7].split("[") 153 zaccept_unit = zaccept_unit_split[1].replace("]","") 154 if zaccept_unit.strip() == r'\AA^1' or zaccept_unit.strip() == r'\A^1': 155 zaccept_unit = "1/A" 156 output.sample.zacceptance=(float(paramvals[7]),zaccept_unit) 157 output.vars = varheader 107 output.sample.yacceptance = (float(_header_fetch(params, "Q_ymax")), 108 _fetch_unit(params, "Q_ymax")) 158 109 159 110 if len(output.x) < 1: … … 173 124 new_unit = value_unit 174 125 return value, new_unit 126 127 128 def _header_fetch(headers, key): 129 index = [k for k in headers.keys() 130 if k.startswith(key)][0] 131 return headers[index] 132 133 def _fetch_unit(params, key): 134 index = [k for k in params.keys() 135 if k.startswith(key)][0] 136 unit = index.strip().split()[1][1:1] 137 if unit.startswith(r"\A"): 138 unit = "1/A" 139 return unit
Note: See TracChangeset
for help on using the changeset viewer.