deleted zebra.py
This commit is contained in:
parent
8aafbb395c
commit
b649fbef3a
@ -1,86 +0,0 @@
|
|||||||
import h5py
|
|
||||||
import numpy as np
|
|
||||||
|
|
||||||
def read_h5meta(filepath):
|
|
||||||
"""Read and parse content of a h5meta file.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
filepath (str): File path of a h5meta file.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
dict: A dictionary with section names and their content.
|
|
||||||
"""
|
|
||||||
h5meta_content = dict()
|
|
||||||
with open(filepath, "r") as h5meta_file:
|
|
||||||
line = h5meta_file.readline()
|
|
||||||
while line:
|
|
||||||
if line.startswith("#begin"):
|
|
||||||
# read section
|
|
||||||
section = line[7:-1] # len("#begin ") = 7
|
|
||||||
h5meta_content[section] = []
|
|
||||||
line = h5meta_file.readline()
|
|
||||||
while not line.startswith("#end"):
|
|
||||||
h5meta_content[section].append(line[:-1])
|
|
||||||
line = h5meta_file.readline()
|
|
||||||
|
|
||||||
# read next line after section's end
|
|
||||||
line = h5meta_file.readline()
|
|
||||||
|
|
||||||
return h5meta_content
|
|
||||||
|
|
||||||
|
|
||||||
def read_detector_data(filepath):
|
|
||||||
"""Read detector data from an h5 file.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
filepath (str): File path of an h5 file.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
ndarray: A 3D array of data.
|
|
||||||
"""
|
|
||||||
with h5py.File(filepath, "r") as h5f:
|
|
||||||
detector_data = h5f["/entry1/area_detector2/data"][:]
|
|
||||||
|
|
||||||
# reshape data to a correct shape (2006 issue)
|
|
||||||
n, cols, rows = detector_data.shape
|
|
||||||
detector_data = detector_data.reshape(n, rows, cols)
|
|
||||||
|
|
||||||
return detector_data
|
|
||||||
|
|
||||||
|
|
||||||
def open_h5meta(filepath):
|
|
||||||
"""Open h5meta file
|
|
||||||
|
|
||||||
Args:
|
|
||||||
filepath (str): File path of a h5meta file.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
dict: A dictionary with h5 names and their detector data.
|
|
||||||
"""
|
|
||||||
data = dict()
|
|
||||||
h5meta_content = read_h5meta(filepath)
|
|
||||||
for file in h5meta_content["filelist"]:
|
|
||||||
data[file] = read_detector_data(file)
|
|
||||||
|
|
||||||
return data
|
|
||||||
|
|
||||||
def z4frgn(wave,ga,nu):
|
|
||||||
"""CALCULATES DIFFRACTION VECTOR IN LAB SYSTEM FROM GA AND NU
|
|
||||||
|
|
||||||
Args:
|
|
||||||
WAVE,GA,NU
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
Z4
|
|
||||||
"""
|
|
||||||
sin = np.sin
|
|
||||||
cos = np.cos
|
|
||||||
pir = 180/np.pi
|
|
||||||
gar = ga/pir
|
|
||||||
nur = nu/pir
|
|
||||||
z4 = [0., 0., 0.]
|
|
||||||
z4[0]=( sin(gar)*cos(nur) )/wave
|
|
||||||
z4[1]=( cos(gar)*cos(nur)-1. )/wave
|
|
||||||
z4[2]=( sin(nur) )/wave
|
|
||||||
|
|
||||||
return z4
|
|
Loading…
x
Reference in New Issue
Block a user