batoolset 0.1.0__py3-none-any.whl

batoolset/partials/tools.py

@@ -0,0 +1,83 @@
+from functools import partial
+
+
+def dummy():
+    print('test')
+
+def dummy2():
+    print('test2')
+
+def get_method_name(partial_func):
+    """
+    Returns the name of the wrapped function in a partial object.
+
+    Args:
+        partial_func (functools.partial): The partial object.
+
+    Returns:
+        str: The name of the wrapped function.
+
+    Examples:
+        >>> tst = partial(dummy)
+        >>> print(get_method_name(tst))
+        'dummy'
+    """
+    return partial_func.func.__name__
+
+def is_partial(func):
+    """
+    Checks if an object is a partial object.
+
+    Args:
+        func: The object to check.
+
+    Returns:
+        bool: True if the object is a partial object, False otherwise.
+
+    Examples:
+        >>> tst = partial(dummy)
+        >>> print(is_partial(tst))
+        True
+    """
+    return isinstance(func, partial)
+
+def are_function_names_matching(func1, func2):
+    """
+    Compares the names of two function objects or partial objects and returns True if they match, False otherwise.
+
+    Args:
+        func1: The first function object or partial object.
+        func2: The second function object or partial object.
+
+    Returns:
+        bool: True if the names match, False otherwise.
+
+    Examples:
+        >>> print(are_function_names_matching(dummy, dummy2))
+        False
+        >>> print(are_function_names_matching(dummy, dummy))
+        True
+        >>> tst = partial(dummy)
+        >>> print(are_function_names_matching(tst, dummy))
+        True
+    """
+    if is_partial(func1):
+        func1 = get_method_name(func1)
+    else:
+        func1 = func1.__name__
+    if is_partial(func2):
+        func2 = get_method_name(func2)
+    else:
+        func2 = func2.__name__
+    return func1 == func2
+
+if __name__ == '__main__':
+
+    tst = partial(dummy)
+
+
+    print('is partial:', is_partial(tst))
+    print('get method name:', get_method_name(tst))
+    print('are function names matching:', are_function_names_matching(dummy, dummy2))
+    print('are function names matching:', are_function_names_matching(dummy, dummy))
+    print('are function names matching:', are_function_names_matching(tst, dummy))

batoolset/plot_tools/tools.py

@@ -0,0 +1,70 @@
+import random
+import matplotlib.pyplot as plt
+import numpy as np
+
+def plot_image_with_pixels(image, cmap='gray'):
+    '''
+    Plots an image with white lines surrounding each pixel. NB: This only make sense for small images
+
+    Args:
+        image (numpy.ndarray): The input image array.
+        cmap (str, optional): The colormap to use for displaying the image. Defaults to 'gray'.
+
+    Returns:
+        matplotlib.figure.Figure: The figure object containing the plotted image.
+
+    # Examples:
+    #     >>> image = np.random.rand(10, 10)
+    #     >>> fig = plot_image_with_pixels(image)
+    #     >>> plt.show()
+    '''
+    # Create a new figure and axes
+    fig, ax = plt.subplots()
+
+    # Show the image using imshow()
+    ax.imshow(image, cmap=cmap, interpolation=None)
+
+    # Set the tick locations to center the ticks at the pixels
+    ax.set_xticks(np.arange(image.shape[1]), minor=False)
+    ax.set_yticks(np.arange(image.shape[0]), minor=False)
+
+    # Shift the grid by 0.5 to surround the pixels
+    ax.set_xticks(np.arange(image.shape[1]+1)-0.5, minor=True)
+    ax.set_yticks(np.arange(image.shape[0]+1)-0.5, minor=True)
+
+    # If the image is too large, hide the tick labels
+    if image.shape[0] > 50:
+        ax.set_yticklabels([])
+
+    if image.shape[1] > 50:
+        ax.set_xticklabels([])
+
+    # Add the lines around each pixel using grid()
+    ax.grid(which='minor', color='white', linestyle='-', linewidth=1)
+
+    # Return the figure object
+    return fig
+
+
+if __name__ == '__main__':
+
+    if True:
+        # Generate a random image of a random type
+        image = random.choice([np.random.rand(10, 10), np.random.rand(100, 10, 3)])
+
+        # Plot the image with white lines surrounding each pixel
+        fig = plot_image_with_pixels(image)
+
+        # Show the plot
+        plt.show()
+
+
+    if True:
+        from batoolset.img import pop, Img
+        img = Img('/E/Sample_images/sample_images_FIJI/AuPbSn40.jpg')
+        pop(img)
+        # sys.exit(0)
+
+
+
+