pyTEMlib 0.2024.1.1__tar.gz → 0.2024.2.0__tar.gz

{pyTEMlib-0.2024.1.1/pyTEMlib.egg-info → pyTEMlib-0.2024.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: pyTEMlib
-Version: 0.2024.1.1
+Version: 0.2024.2.0
 Summary: pyTEM: TEM Data Quantification library through a model-based approach
 Home-page: https://pycroscopy.github.io/pyTEMlib/about.html
 Author: Gerd Duscher

{pyTEMlib-0.2024.1.1 → pyTEMlib-0.2024.2.0}/pyTEMlib/eds_tools.py

@@ -424,7 +424,7 @@ def fit_model(spectrum, elements, use_detector_efficiency=False):
     y = np.array(spectrum)  # .compute()
     [p, _] = leastsq(residuals, pin, args=(y))
 
-    print(pin[-6:], p[-6:])
+    # print(pin[-6:], p[-6:])
 
     update_fit_values(out_tags, p)
 

{pyTEMlib-0.2024.1.1 → pyTEMlib-0.2024.2.0}/pyTEMlib/image_tools.py

@@ -23,6 +23,7 @@ from tqdm.auto import trange, tqdm
 from itertools import product
 
 from scipy import fftpack
+import scipy
 # from scipy import signal
 from scipy.interpolate import interp1d  # , interp2d
 import scipy.optimize as optimization
@@ -730,6 +731,76 @@ class LineSelector(matplotlib.widgets.PolygonSelector):
             self.line_verts[moved_point] = self.new_point
             self.set_linewidth()
 
+def get_profile(dataset, line):
+    xv, yv = get_line_selection_points(line)
+    
+    
+    if dataset.data_type.name == 'IMAGE':
+        dataset.get_image_dims()
+        xv /= (dataset.x[1] - dataset.x[0])
+        yv /= (dataset.y[1] - dataset.y[0])
+        profile = scipy.ndimage.map_coordinates(np.array(dataset), [xv,yv])
+        
+        profile_dataset = sidpy.Dataset.from_array(profile.sum(axis=0))
+        profile_dataset.data_type='spectrum'
+        profile_dataset.units = dataset.units
+        profile_dataset.quantity = dataset.quantity
+        profile_dataset.set_dimension(0, sidpy.Dimension(np.linspace(xv[0,0], xv[-1,-1], profile_dataset.shape[0]), 
+                                                  name='x', units=dataset.x.units, quantity=dataset.x.quantity,
+                                                  dimension_type='spatial'))
+
+        profile_dataset
+
+    if dataset.data_type.name == 'SPECTRAL_IMAGE':
+        spectral_axis = dataset.get_spectral_dims(return_axis=True)[0]
+        profile = np.zeros([xv.shape[1], 2, len(spectral_axis)])
+        data =np.array(dataset)
+
+        for index_x in range(xv.shape[1]):
+            for  index_y  in range(xv.shape[0]):
+                x = xv[index_y, index_x]
+                y = yv[index_y, index_x]
+                profile[index_x, 0] +=data[int(x),int(y)]
+        profile_dataset = sidpy.Dataset.from_array(profile)
+        profile_dataset.data_type='spectral_image'
+        profile_dataset.units = dataset.units
+        profile_dataset.quantity = dataset.quantity
+        profile_dataset.set_dimension(0, sidpy.Dimension(np.linspace(xv[0,0], xv[-1,-1], profile_dataset.shape[0]), 
+                                                  name='x', units=dataset.x.units, quantity=dataset.x.quantity,
+                                                  dimension_type='spatial'))
+        profile_dataset.set_dimension(1, sidpy.Dimension([0, 1], 
+                                                  name='y', units=dataset.x.units, quantity=dataset.x.quantity,
+                                                  dimension_type='spatial'))
+        
+        profile_dataset.set_dimension(2, spectral_axis)
+    return profile_dataset
+
+
+def get_line_selection_points(line):
+    
+    start_point = line.line_verts[3]
+    right_point = line.line_verts[0]
+    low_point = line.line_verts[2]
+
+    if start_point[0] > right_point[0]:
+        start_point = line.line_verts[0]
+        right_point = line.line_verts[3]
+        low_point = line.line_verts[1]
+    m = (right_point[1] - start_point[1]) / (right_point[0] - start_point[0])
+    length_x = int(abs(start_point[0]-right_point[0]))
+    length_v = int(np.linalg.norm(start_point-right_point))
+    
+    linewidth = int(abs(start_point[1]-low_point[1]))
+    x = np.linspace(0,length_x, length_v)
+    y = np.linspace(0,linewidth, line.line_width)
+    xv, yv = np.meshgrid(x, y)
+    
+    yy = yv +x*m+start_point[1] 
+    xx = (xv.T -y*m ).T + start_point[0]
+    
+    return xx, yy
+
+
 
 def histogram_plot(image_tags):
     """interactive histogram"""

pyTEMlib-0.2024.2.0/pyTEMlib/version.py

@@ -0,0 +1,6 @@
+"""
+version
+"""
+_version = '0.2024.02.0'
+__version__ = _version
+_time = '2024-02-08 19:58:26'

{pyTEMlib-0.2024.1.1 → pyTEMlib-0.2024.2.0/pyTEMlib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: pyTEMlib
-Version: 0.2024.1.1
+Version: 0.2024.2.0
 Summary: pyTEM: TEM Data Quantification library through a model-based approach
 Home-page: https://pycroscopy.github.io/pyTEMlib/about.html
 Author: Gerd Duscher

pyTEMlib-0.2024.1.1/pyTEMlib/version.py

@@ -1,6 +0,0 @@
-"""
-version
-"""
-_version = '0.2024.01.1'
-__version__ = _version
-_time = '2024-01-08 19:58:26'