-
Notifications
You must be signed in to change notification settings - Fork 0
/
HSLoad.py
57 lines (48 loc) · 1.67 KB
/
HSLoad.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
# -*- coding: utf-8 -*-
"""
Created on Mon Jun 22 10:58:25 2020
@author: Mai Tai
"""
def HSLoad(filename,wavstart, wavend, wavres):
import numpy as np
import matplotlib
matplotlib.rcParams["backend"] = "Agg"
import glob
import hyperspy.api as hs
directories = [f for f in glob.glob('*_*mW')]
directories.sort()
def ParaLoad(directory):
A =[]
files = [f for f in glob.glob(str(directory)+'/parallel/*0.npy')]
files = np.sort(files)
for f in files:
a = np.load(f)
A.append(a)
return A
def PerpLoad(directory):
B =[]
files = [f for f in glob.glob(str(directory)+'/perpendicular/*0.npy')]
files = np.sort(files)
for f in files:
a = np.load(f)
B.append(a)
return B
LA = [ParaLoad(f) for f in directories]
LB = [PerpLoad(f) for f in directories]
s = hs.signals.Signal1D([LA,LB], lazy=True)
s.axes_manager[1].name = "Polarization"
s.axes_manager['Polarization'].units = 'degrees'
s = s.as_signal1D('Polarization')
s.axes_manager['Polarization'].scale = 2
s.axes_manager[0].name = "X"
s.axes_manager[1].name = "Y"
s.axes_manager[2].name = "Wavelength"
s.axes_manager['Wavelength'].offset = wavstart
s.axes_manager['Wavelength'].scale = wavres
s.axes_manager['X'].units = 'pixels'
s.axes_manager['Y'].units = 'pixels'
s.axes_manager['Wavelength'].units = 'nm'
s.axes_manager[3].name = 'Orientation'
s.axes_manager[3].units = 'Para/Perp'
s.save(filename)
return s