Revert "Update load_1D.py"
This reverts commit cc3f1d28a5670dad6fdaec83a19edbee0b1a09ce.
This commit is contained in:
parent
cc3f1d28a5
commit
ab002be606
@ -2,22 +2,7 @@ import re
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
from decimal import Decimal
|
from decimal import Decimal
|
||||||
|
|
||||||
|
META_VARS_STR = (
|
||||||
def load_1D(filepath):
|
|
||||||
"""
|
|
||||||
Loads *.ccl or *.dat file (Distinguishes them based on last 3 chars in string of filepath
|
|
||||||
to add more variables to read, extend the elif list
|
|
||||||
the file must include '#data' and number of points in right place to work properly
|
|
||||||
|
|
||||||
:arg filepath
|
|
||||||
:returns det_variables
|
|
||||||
- dictionary of all detector/scan variables and dictinionary for every measurement.
|
|
||||||
Names of these dictionaries are M + measurement number. They include HKL indeces, angles, monitors,
|
|
||||||
stepsize and array of counts
|
|
||||||
"""
|
|
||||||
|
|
||||||
det_variables = {"file_type": str(filepath)[-3:], "meta": {}}
|
|
||||||
meta_vars_str = (
|
|
||||||
"instrument",
|
"instrument",
|
||||||
"title",
|
"title",
|
||||||
"sample",
|
"sample",
|
||||||
@ -31,7 +16,7 @@ def load_1D(filepath):
|
|||||||
"proposal_title",
|
"proposal_title",
|
||||||
"proposal_email",
|
"proposal_email",
|
||||||
)
|
)
|
||||||
meta_vars_float = (
|
META_VARS_FLOAT = (
|
||||||
"mf",
|
"mf",
|
||||||
"2-theta",
|
"2-theta",
|
||||||
"chi",
|
"chi",
|
||||||
@ -66,28 +51,35 @@ def load_1D(filepath):
|
|||||||
"s2hr",
|
"s2hr",
|
||||||
"s2hl",
|
"s2hl",
|
||||||
)
|
)
|
||||||
meta_ub_matrix = ("ub1j", "ub2j", "ub3j")
|
META_UB_MATRIX = ("ub1j", "ub2j", "ub3j")
|
||||||
|
|
||||||
|
|
||||||
|
def load_1D(filepath):
|
||||||
|
"""
|
||||||
|
Loads *.ccl or *.dat file (Distinguishes them based on last 3 chars in string of filepath
|
||||||
|
to add more variables to read, extend the elif list
|
||||||
|
the file must include '#data' and number of points in right place to work properly
|
||||||
|
|
||||||
|
:arg filepath
|
||||||
|
:returns det_variables
|
||||||
|
- dictionary of all detector/scan variables and dictinionary for every measurement.
|
||||||
|
Names of these dictionaries are M + measurement number. They include HKL indeces, angles,
|
||||||
|
monitors, stepsize and array of counts
|
||||||
|
"""
|
||||||
|
det_variables = {"file_type": str(filepath)[-3:], "meta": {}}
|
||||||
with open(filepath, "r") as infile:
|
with open(filepath, "r") as infile:
|
||||||
for line in infile:
|
for line in infile:
|
||||||
det_variables["Measurements"] = {}
|
det_variables["Measurements"] = {}
|
||||||
if "=" in line:
|
if "=" in line:
|
||||||
variable, value = line.split("=")
|
variable, value = line.split("=")
|
||||||
variable = variable.strip()
|
variable = variable.strip()
|
||||||
try:
|
if variable in META_VARS_FLOAT:
|
||||||
if variable in meta_vars_float:
|
|
||||||
det_variables["meta"][variable] = float(value)
|
det_variables["meta"][variable] = float(value)
|
||||||
elif variable in meta_vars_str:
|
elif variable in META_VARS_STR:
|
||||||
det_variables["meta"][variable] = str(value)[:-1].strip()
|
det_variables["meta"][variable] = str(value)[:-1].strip()
|
||||||
elif variable in meta_ub_matrix:
|
elif variable in META_UB_MATRIX:
|
||||||
det_variables["meta"][variable] = re.findall(
|
det_variables["meta"][variable] = re.findall(r"[-+]?\d*\.\d+|\d+", str(value))
|
||||||
r"[-+]?\d*\.\d+|\d+", str(value)
|
|
||||||
)
|
|
||||||
except ValueError as error:
|
|
||||||
print(
|
|
||||||
"Some values are not in expected format (str or float), error:",
|
|
||||||
str(error),
|
|
||||||
)
|
|
||||||
elif "#data" in line:
|
elif "#data" in line:
|
||||||
if det_variables["file_type"] == "ccl":
|
if det_variables["file_type"] == "ccl":
|
||||||
decimal = list()
|
decimal = list()
|
||||||
@ -119,7 +111,6 @@ def load_1D(filepath):
|
|||||||
d["gamma_angle"] = float(lines.split()[4]) # gamma
|
d["gamma_angle"] = float(lines.split()[4]) # gamma
|
||||||
d["omega_angle"] = float(lines.split()[5]) # omega
|
d["omega_angle"] = float(lines.split()[5]) # omega
|
||||||
d["nu_angle"] = float(lines.split()[6]) # nu
|
d["nu_angle"] = float(lines.split()[6]) # nu
|
||||||
d["unkwn_angle"] = float(lines.split()[7])
|
|
||||||
|
|
||||||
next_line = data[position + 1]
|
next_line = data[position + 1]
|
||||||
d["number_of_measurements"] = int(next_line.split()[0])
|
d["number_of_measurements"] = int(next_line.split()[0])
|
||||||
@ -146,12 +137,6 @@ def load_1D(filepath):
|
|||||||
)
|
)
|
||||||
d["counts"] = counts
|
d["counts"] = counts
|
||||||
det_variables["Measurements"][str("M" + str(measurement_number))] = d
|
det_variables["Measurements"][str("M" + str(measurement_number))] = d
|
||||||
if all(decimal):
|
|
||||||
det_variables["meta"]["indices"] = "hkl"
|
|
||||||
else:
|
|
||||||
det_variables["meta"]["indices"] = "real"
|
|
||||||
print("indeces:", det_variables["meta"]["indices"])
|
|
||||||
|
|
||||||
|
|
||||||
elif det_variables["file_type"] == "dat":
|
elif det_variables["file_type"] == "dat":
|
||||||
data = infile.readlines()
|
data = infile.readlines()
|
||||||
@ -177,6 +162,9 @@ def load_1D(filepath):
|
|||||||
det_variables["Measurements"]["time"] = time
|
det_variables["Measurements"]["time"] = time
|
||||||
else:
|
else:
|
||||||
print("Unknown file extention")
|
print("Unknown file extention")
|
||||||
|
if all(decimal):
|
||||||
|
det_variables["meta"]["indices"] = "hkl"
|
||||||
|
else:
|
||||||
|
det_variables["meta"]["indices"] = "real"
|
||||||
|
|
||||||
return det_variables
|
return det_variables
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user