rp 0.1.1268__tar.gz → 0.1.1270__tar.gz

{rp-0.1.1268 → rp-0.1.1270}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: rp
-Version: 0.1.1268
+Version: 0.1.1270
 Summary: Ryan's Python
 Home-page: https://github.com/RyannDaGreat/Quick-Python
 Author: Ryan Burgert

{rp-0.1.1268 → rp-0.1.1270}/rp/r.py

@@ -2440,6 +2440,101 @@ def overlay_images(*images,mode='normal'):
         output=blend_images(output,image,mode=mode)
     return output
     
+def laplacian_blend(bot, top, alpha, levels=None):
+    """
+    Uses laplacian pyramid blending on two images with a given alpha mask.
+    Note: Right now only RGB images of equal dimensions are supported (non-rgb will be converted to RGB)
+    
+    Arguments:
+        - levels (int, None): The number of laplacian pyramid levels. Defaults to None, where the max number of levels is selected automatically from the image sizes. No errors will be thrown if you input a level number too high - it will effectively just use the max level.
+    
+    EXAMPLE:
+        >>> top_url = "https://fastly.picsum.photos/id/9/5000/3269.jpg?hmac=cZKbaLeduq7rNB8X-bigYO8bvPIWtT-mh8GRXtU3vPc"
+        ... bot_url = "https://fastly.picsum.photos/id/21/3008/2008.jpg?hmac=T8DSVNvP-QldCew7WD4jj_S3mWwxZPqdF0CNPksSko4"
+        ... 
+        ... S = 512
+        ... alpha = get_checkerboard_image(S, S, tile_size=64,first_color=0,second_color=1)
+        ... 
+        ... bot, top = resize_images_to_fit(crop_images_to_square(load_images(download_urls_to_cache(bot_url, top_url), use_cache=True)), height=S, width=S)
+        ... 
+        ... bot, top, alpha = as_float_images(as_rgb_images([bot, top, alpha]))
+        ... 
+        ... ic("FOR DEBUGGING", alpha.shape, alpha.dtype, bot.shape, bot.dtype, top.shape, top.dtype)
+        ... 
+        ... regular_blended_image = blend_images(bot, top, alpha)
+        ... laplacian_blended_image = laplacian_blend(bot, top, alpha)
+        ... 
+        ... laplacian_levels_anim = [laplacian_blend(bot, top, alpha,levels) for levels in range(10)]
+        ... 
+        ... display_image(grid_concatenated_images([[bot, top, alpha], [regular_blended_image, laplacian_blended_image]]), block=True)
+        ... display_image_slideshow(video_with_progress_bar(laplacian_levels_anim))
+    """
+    pip_import('cv2')
+    pip_import('numpy')
+
+    import numpy as np
+    import cv2
+
+    #Make all inputs RGB float images
+    bot, top, alpha = as_rgb_images(as_float_images([bot, top, alpha], copy=False), copy=False)
+
+    #Input validation: Make sure all images have same shape
+    if not (bot.shape == top.shape == alpha.shape):
+        raise ValueError(
+            "laplacian_blend: All input images must have same dimensions, but got bot.shape==%s and top.shape==%s and alpha.shape==%s" % 
+            (bot.shape, top.shape, alpha.shape)
+        )
+
+    # Calculate max possible levels if None is provided
+    if levels is None:
+        min_dim = min(bot.shape[:2] + top.shape[:2] + alpha.shape[:2])
+        levels = int(np.ceil(np.log2(min_dim)))
+
+    # Generate Gaussian pyramids
+    gaussian_pyr1 = [bot]
+    gaussian_pyr2 = [top]
+    gaussian_alpha = [alpha]
+
+    for _ in range(levels):
+        gaussian_pyr1.append(cv2.pyrDown(gaussian_pyr1[-1]))
+        gaussian_pyr2.append(cv2.pyrDown(gaussian_pyr2[-1]))
+        gaussian_alpha.append(cv2.pyrDown(gaussian_alpha[-1]))
+
+    # Generate Laplacian pyramids directly
+    laplacian_pyr1 = []
+    laplacian_pyr2 = []
+
+    for i in range(levels):
+        src_img1 = gaussian_pyr1[i + 1]
+        src_img2 = gaussian_pyr2[i + 1]
+
+        dst_size1 = (gaussian_pyr1[i].shape[1], gaussian_pyr1[i].shape[0])
+        dst_size2 = (gaussian_pyr2[i].shape[1], gaussian_pyr2[i].shape[0])
+        
+        level_expanded1 = cv2.pyrUp(src_img1, dstsize=dst_size1)
+        level_expanded2 = cv2.pyrUp(src_img2, dstsize=dst_size2)
+
+        laplacian_pyr1.append(cv2.subtract(gaussian_pyr1[i], level_expanded1))
+        laplacian_pyr2.append(cv2.subtract(gaussian_pyr2[i], level_expanded2))
+
+    # Add the smallest level directly (Gaussian level)
+    laplacian_pyr1.append(gaussian_pyr1[-1])
+    laplacian_pyr2.append(gaussian_pyr2[-1])
+
+    # Blend pyramids
+    blended_pyramid = []
+    for lap1, lap2, alpha_mask in zip(laplacian_pyr1, laplacian_pyr2, gaussian_alpha):
+        blended = lap1 * alpha_mask + lap2 * (1 - alpha_mask)
+        blended_pyramid.append(blended)
+
+    # Reconstruct the blended image
+    blended_image = blended_pyramid[-1]
+    for level in reversed(blended_pyramid[:-1]):
+        blended_image = cv2.pyrUp(blended_image, dstsize=(level.shape[1], level.shape[0]))
+        blended_image = cv2.add(blended_image, level)
+
+    return blended_image
+
 def get_checkerboard_image(height=64,
                            width=64,
                            *,
@@ -9662,7 +9757,7 @@ def globalize_locals(func):
     Useful for making reusable cells in a Jupyter notebook.
     
     When a function decorated with @globalize_locals is called, all local variables created
-    within the function become available in the global namespace after execution.
+    within the function become available in the global namespace during execution
     This includes both function parameters and local variables.
     
     The global namespace modification is the primary feature of this decorator,
@@ -9751,55 +9846,26 @@ def globalize_locals(func):
         caller_frame = sys._getframe(1)
         caller_globals = caller_frame.f_globals
         
-        # Create a dictionary to store function's locals
-        func_locals = {}
-        
         # First, bind the arguments to parameter names to ensure parameters are captured
         sig = inspect.signature(func)
         bound_args = sig.bind(*args, **kwargs)
         bound_args.apply_defaults()
-        func_locals.update(bound_args.arguments)
-        
-        # Set up trace function to capture variables on each line
-        def trace_func(frame, event, arg):
-            # We're only interested in the function we're decorating
-            if frame.f_code == func.__code__:
-                # Copy all variables from the frame's locals to our dictionary
-                for var_name, value in frame.f_locals.items():
-                    func_locals[var_name] = value
-                # Keep tracing this function
-                return trace_func
-            # Don't trace other functions
-            return None
+        
+        def trace_func(frame, event, arg):            # Set up trace function to capture variables on each line
+            if frame.f_code == func.__code__:         # We're only interested in the function we're decorating
+                caller_globals.update(frame.f_locals) # Push all local variables to globals
+                return trace_func                     # Keep tracing this function
+            return None                               # Don't trace other functions
             
         # Set tracing and execute the function
         original_trace = sys.gettrace()
         sys.settrace(trace_func)
         
-        try:
-            # Call the function and get its result
-            result = func(*args, **kwargs)
-            return result
-        except Exception:
-            # Capture the exception info
-            exc_info = sys.exc_info()
-            # Will re-raise after transferring variables
-            raise_later = True
-        finally:
-            # Restore the original trace function
-            sys.settrace(original_trace)
+        try:     return func(*args, **kwargs) # Call the function and get its result
+        finally: sys.settrace(original_trace) # Restore the original trace function
             
-            # Copy the captured locals to the caller's globals
-            for name, value in func_locals.items():
-                caller_globals[name] = value
-                
-            # If there was an exception, re-raise it
-            if 'raise_later' in locals() and raise_later:
-                raise exc_info[1].with_traceback(exc_info[2])
-    
     return wrapper
 
-
 def _filter_dict_via_fzf(input_dict,*,preview=None):
     """Uses fzf to select a subset of a dict and returns that dict."""
     #Refactored using GPT4 from a mess: https://chat.openai.com/share/66251028-75eb-4c55-960c-1e7477e34060
@@ -19221,7 +19287,7 @@ def _convert_powerpoint_file(path,message=None):
 
 
 def _write_default_gitignore():
-    types_to_ignore='pyc bak swo swp swn swm swh swi swj swk swl swm swn swo swp un~ gstmp ipynb_checkpoints DS_Store'.split()
+    types_to_ignore='pyc bak swh swi swj swk swl swm swn swo swp un~ gstmp ipynb_checkpoints DS_Store'.split()
     types_to_ignore=['*.'+x for x in types_to_ignore]
 
     new_lines = (
@@ -20470,9 +20536,9 @@ def pseudo_terminal(
 
         IASM $import_all_submodules(ans,verbose=True);
 
-        SUH $sublime('.')
+        SUH $sublime($get_current_directory())
         SUA $sublime(ans)
-        COH $vscode('.')
+        COH !code .
         COA $vscode(ans)
 
         SG $save_gist(ans)
@@ -37900,6 +37966,42 @@ def _sort_imports_via_isort(code):
     return isort.code(code)
 sort_imports_via_isort = _sort_imports_via_isort
 
+def clean_imports_via_unimport(code: str) -> str:
+    """
+    Removes dead imports
+
+    EXAMPLE:
+        >>> code = "import numpy\nprint(123)"
+        >>> print(code)
+        import numpy
+        print(123)
+        >>> print(clean_imports_via_unimport(code))
+        print(123)
+    """
+    pip_import('unimport')
+    
+    import unimport
+    from unimport.analyzers.main import MainAnalyzer
+    from unimport.statement import Import
+    import unimport.refactor
+
+    analyzer = MainAnalyzer(source=code)
+    
+    # Analyze the code to find unused imports
+    analyzer.traverse()
+    
+    # Get unused imports
+    unused_imports = list(Import.get_unused_imports())
+    
+    # Remove unused imports from the code
+    result = unimport.refactor.refactor_string(source=code, unused_imports=unused_imports)
+    
+    # Clean up
+    analyzer.clear()
+    
+    return result
+
+
 _ryan_tmux_conf=r'''
 #Ryan Burgert's Tmux config
 #Main changes:
@@ -40137,7 +40239,7 @@ def inverted_color(color):
     else:
         raise TypeError('Unknown color format: '+repr(color))
 
-def inverted_image(image,invert_alpha=False):
+def inverted_image(image, invert_alpha=False):
     """ Inverts the colors of an image. By default, it doesn't touch the alpha channel (if one exists) """
     assert is_image(image)
     image=image.copy()
@@ -40156,6 +40258,13 @@ def inverted_image(image,invert_alpha=False):
         elif is_binary_image(image):
             image=~image
     return image
+invert_image = inverted_image
+
+def inverted_images(images, invert_alpha=False):
+    if is_numpy_array(images):
+        return as_numpy_array(gather_args_call(inverted_images, list(images)))
+    return [gather_args_call(inverted_image, image) for image in images]
+invert_images = inverted_images
 
 def make_zip_file_from_folder(src_folder:str=None, dst_zip_file:str=None)->str:
     """

{rp-0.1.1268 → rp-0.1.1270}/rp/rp_ptpython/key_bindings.py

@@ -2818,6 +2818,7 @@ def load_python_bindings(python_input):
                                              }
                         header_commands={
                                          '\\sim':'sort_imports',
+                                         '\\cim':'clean_imports',
                                          '\\rms':'remove_star',
                                          '\\bla':'black',
                                          '\\gpt':'gpt',
@@ -3254,10 +3255,17 @@ def load_python_bindings(python_input):
                                     try:
                                         text=buffer.document.text
                                         text=rp.r._sort_imports_via_isort(text)
-                                        # buffer.document=Document((text),min(len(text),buffer.document.cursor_position),buffer.document.selection)
                                         replace_buffer_text(buffer, text)
                                     except BaseException as e:
                                         buffer.insert_text('#sort_imports: Error: '+str(e).replace('\n',' ; '))
+                                if header=='clean_imports':
+                                    import rp
+                                    try:
+                                        text=buffer.document.text
+                                        text=rp.r.clean_imports_via_unimport(text)
+                                        replace_buffer_text(buffer, text)
+                                    except BaseException as e:
+                                        buffer.insert_text('#clean_imports: Error: '+str(e).replace('\n',' ; '))
                                 if header=='remove_star':
                                     try:
                                         from rp.r import _removestar

rp-0.1.1270/rp/version.py

@@ -0,0 +1 @@
+1270

{rp-0.1.1268 → rp-0.1.1270}/rp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: rp
-Version: 0.1.1268
+Version: 0.1.1270
 Summary: Ryan's Python
 Home-page: https://github.com/RyannDaGreat/Quick-Python
 Author: Ryan Burgert

rp-0.1.1270/version.py

@@ -0,0 +1 @@
+1270

rp-0.1.1268/rp/version.py

@@ -1 +0,0 @@
-1268

rp-0.1.1268/version.py

@@ -1 +0,0 @@
-1268