pyimagecuda 0.0.8__cp313-cp313-win_amd64.whl → 0.0.9__cp313-cp313-win_amd64.whl

pyimagecuda/__init__.py

@@ -1,7 +1,7 @@
 import ctypes
 import sys
 
-__version__ = "0.0.7"
+__version__ = "0.0.9"
 
 def _check_nvidia_driver():
     try:
@@ -37,7 +37,7 @@ except ImportError as e:
 if _INTERNAL_LOADED:
     try:
         from .image import Image, ImageU8
-        from .io import upload, download, copy, save, load, convert_float_to_u8, convert_u8_to_float, from_numpy, to_numpy
+        from .io import upload, download, copy, save, load, convert_float_to_u8, convert_u8_to_float, from_numpy, to_numpy, save_u8
         from .fill import Fill
         from .resize import Resize
         from .blend import Blend
@@ -46,6 +46,7 @@ if _INTERNAL_LOADED:
         from .adjust import Adjust
         from .transform import Transform
         from .text import Text
+        from .gl_interop import GLResource
         from .pyimagecuda_internal import cuda_sync # type: ignore
     except ImportError as e:
         print(f"Warning: Error importing Python wrappers: {e}")

pyimagecuda/gl_interop.py

@@ -0,0 +1,83 @@
+from .pyimagecuda_internal import (  # type: ignore
+    register_gl_pbo,
+    unregister_gl_resource,
+    copy_to_gl_pbo
+)
+from .image import ImageU8
+
+
+class GLResource:
+    """
+    CUDA-OpenGL interop resource for direct GPU-to-GPU transfers.
+    
+    Represents a registered OpenGL PBO that can receive ImageU8 data
+    directly without CPU roundtrips.
+    
+    Docs & Examples: https://offerrall.github.io/pyimagecuda/opengl/
+    """
+    
+    def __init__(self, pbo_id: int):
+        """
+        Parameters:
+            pbo_id: Valid OpenGL PBO ID. The PBO must exist and remain 
+                    valid for the lifetime of this GLResource.
+        
+        Raises:
+            ValueError: If pbo_id is invalid
+            RuntimeError: If CUDA registration fails
+        
+        Docs & Examples: https://offerrall.github.io/pyimagecuda/opengl/
+        """
+        if not isinstance(pbo_id, int) or pbo_id <= 0:
+            raise ValueError(f"Invalid PBO ID: {pbo_id}")
+        
+        self._handle = None
+        self.pbo_id = pbo_id
+        self._handle = register_gl_pbo(pbo_id)
+    
+    def copy_from(self, image: ImageU8) -> None:
+        """
+        Copies ImageU8 data directly to the registered PBO (GPU→GPU).
+        
+        Docs & Examples: https://offerrall.github.io/pyimagecuda/opengl/
+        """
+        if not isinstance(image, ImageU8):
+            raise TypeError(f"Expected ImageU8, got {type(image).__name__}")
+        
+        if self._handle is None:
+            raise RuntimeError("GLResource has been freed")
+        
+        copy_to_gl_pbo(
+            image._buffer._handle,
+            self._handle,
+            image.width,
+            image.height
+        )
+    
+    def free(self) -> None:
+        """
+        Unregisters the OpenGL resource from CUDA.
+        
+        Docs & Examples: https://offerrall.github.io/pyimagecuda/opengl/
+        """
+        if self._handle is not None:
+            unregister_gl_resource(self._handle)
+            self._handle = None
+    
+    def __enter__(self):
+        return self
+    
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.free()
+        return False
+    
+    def __del__(self):
+        if hasattr(self, '_handle') and self._handle is not None:
+            try:
+                self.free()
+            except Exception:
+                pass
+    
+    def __repr__(self) -> str:
+        status = "active" if getattr(self, '_handle', None) is not None else "freed"
+        return f"GLResource(pbo_id={self.pbo_id}, status={status})"

pyimagecuda/io.py

@@ -121,26 +121,13 @@ def load(
     return f32_buffer if should_return else None
 
 
-def save(image: Image, filepath: str, u8_buffer: ImageU8 | None = None, quality: int | None = None) -> None:
-    """
-    Saves the floating-point image to a file (using an 8-bit buffer for conversion).
-
-    Docs & Examples: https://offerrall.github.io/pyimagecuda/io/#saving-images
-    """
-    if u8_buffer is None:
-        u8_buffer = ImageU8(image.width, image.height)
-        owns_buffer = True
-    else:
-        ensure_capacity(u8_buffer, image.width, image.height)
-        owns_buffer = False
-    
-    convert_float_to_u8(u8_buffer, image)
-    pixel_data = download(u8_buffer)
+def _save_internal(u8_image: ImageU8, filepath: str, quality: int | None = None) -> None:
+    pixel_data = download(u8_image)
     
     vips_img = pyvips.Image.new_from_memory(
         pixel_data,
-        image.width,
-        image.height,
+        u8_image.width,
+        u8_image.height,
         bands=4,
         format='uchar'
     )
@@ -157,10 +144,37 @@ def save(image: Image, filepath: str, u8_buffer: ImageU8 | None = None, quality:
             save_kwargs['Q'] = quality
     
     vips_img.write_to_file(filepath, **save_kwargs)
+
+
+def save(image: Image, filepath: str, u8_buffer: ImageU8 | None = None, quality: int | None = None) -> None:
+    """
+    Saves the floating-point image to a file (using an 8-bit buffer for conversion).
+
+    Docs & Examples: https://offerrall.github.io/pyimagecuda/io/#saving-images
+    """
+    if u8_buffer is None:
+        u8_buffer = ImageU8(image.width, image.height)
+        owns_buffer = True
+    else:
+        ensure_capacity(u8_buffer, image.width, image.height)
+        owns_buffer = False
+    
+    convert_float_to_u8(u8_buffer, image)
+    _save_internal(u8_buffer, filepath, quality)
     
     if owns_buffer:
         u8_buffer.free()
 
+
+def save_u8(image: ImageU8, filepath: str, quality: int | None = None) -> None:
+    """
+    Saves an 8-bit unsigned integer image directly to a file.
+
+    Docs & Examples: https://offerrall.github.io/pyimagecuda/io/#saving-images
+    """
+    _save_internal(image, filepath, quality)
+
+
 def from_numpy(array, f32_buffer: Image | None = None, u8_buffer: ImageU8 | None = None) -> Image:
     """
     Creates a PyImageCUDA Image from a NumPy array (e.g. from OpenCV, Pillow, Matplotlib).

{pyimagecuda-0.0.8.dist-info → pyimagecuda-0.0.9.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.2
 Name: pyimagecuda
-Version: 0.0.8
+Version: 0.0.9
 Summary: GPU-accelerated image processing library for Python
 Author: Beltrán Offerrall
 Requires-Python: >=3.10
 Requires-Dist: pyvips[binary]
 Description-Content-Type: text/markdown
 
-# PyImageCUDA 0.0.8
+# PyImageCUDA 0.0.9
 
 [![PyPI version](https://img.shields.io/pypi/v/pyimagecuda.svg)](https://pypi.org/project/pyimagecuda/)
 [![Build Status](https://github.com/offerrall/pyimagecuda/actions/workflows/build.yml/badge.svg)](https://github.com/offerrall/pyimagecuda/actions)
@@ -47,6 +47,7 @@ with Image(1024, 1024) as bg:
 * ✅ **Ultra-lightweight:** library weighs **~1 MB**.
 * ✅ **Studio Quality:** 32-bit floating-point precision (float32) to prevent color banding.
 * ✅ **Advanced Memory Control:** Reuse GPU buffers across operations and resize without reallocation—critical for video processing and batch workflows.
+* ✅ **OpenGL Integration:** Direct GPU-to-GPU display for real-time preview widgets.
 * ✅ **API Simplicity:** Intuitive, Pythonic API designed for ease of use.
 
 ## Use Cases
@@ -54,6 +55,7 @@ with Image(1024, 1024) as bg:
 * **Generative Art:** Create thousands of unique variations in seconds.
 * **Motion Graphics:** Process video frames or generate effects in real-time.
 * **Image Compositing:** Complex multi-layer designs with GPU-accelerated effects.
+* **Node Editors & Real-time Tools:** Build responsive image editors with instant preview.
 * **Game Development:** Procedural UI assets, icons, and sprite generation.
 * **Marketing Automation:** Mass-produce personalized graphics from templates.
 * **Data Augmentation:** High-speed batch transformations for ML datasets.
@@ -73,6 +75,7 @@ pip install pyimagecuda
 * [Getting Started Guide](https://offerrall.github.io/pyimagecuda/)
 * [Image & Memory](https://offerrall.github.io/pyimagecuda/image/) (Buffer management)
 * [IO](https://offerrall.github.io/pyimagecuda/io/) (Loading and Saving)
+* [OpenGL Integration](https://offerrall.github.io/pyimagecuda/opengl/) (Real-time preview, zero-copy display)
 
 ### Operations
 * [Fill](https://offerrall.github.io/pyimagecuda/fill/) (Solid colors, Gradients, Checkerboard, Grid, Stripes, Dots, Circle, Ngon, Noise, Perlin)
@@ -129,4 +132,4 @@ pytest tests/tests.py
 Contributions welcome! Open issues or submit PRs
 
 ## License
-MIT License. See [LICENSE](LICENSE) for details.
+MIT License. See [LICENSE](LICENSE) for details.

{pyimagecuda-0.0.8.dist-info → pyimagecuda-0.0.9.dist-info}/RECORD

@@ -1,17 +1,18 @@
-pyimagecuda/__init__.py,sha256=4duktD0TUqTayWWg39PuCQgr6xPbwxSJ6cQyFx65QJg,2166
+pyimagecuda/__init__.py,sha256=uTnh4hmahYz_-W1587VmdEUe8PnEMmAGaXnAYJs1NSg,2219
 pyimagecuda/adjust.py,sha256=xGSgifVWaFkaXyL718PUcYqwtZJAlnvMHNLQHiUS40M,3231
 pyimagecuda/blend.py,sha256=sPupoAh9Mnz16dAsv6C3ynZvDHmV-0MR-A32KpEcE6Q,9132
 pyimagecuda/effect.py,sha256=KBmDNitsl3qxfuRPT075UAsRF5mSXI25aFTb06JZ1WQ,6231
 pyimagecuda/fill.py,sha256=7ybr-wkyr5_ziLL_6bvrLAW_Ixe1nMz36Lrj6jJAiOI,8934
 pyimagecuda/filter.py,sha256=MK6iDQ-tajbx1bgC8hvhrSFusAFfoymaQRDQEkwNWsU,6601
+pyimagecuda/gl_interop.py,sha256=UtKr1a7fTtAhLt6EoLh7LwGRKu9RzKjG9RHYLUs9e3w,2616
 pyimagecuda/image.py,sha256=VRDHUOF2VClwdj3Q5YSQsmy9lpirwQvbC1P8avE3PdI,2864
-pyimagecuda/io.py,sha256=zNUaljHJe7zGEd01qODXWQiydJ84xA9nlIRKrAoa1ys,8287
-pyimagecuda/pyimagecuda_internal.cp313-win_amd64.pyd,sha256=C9ZX9TYPWJlcFGESBCjaB9IHZgM-O_pNS5pl2P4iNoc,582656
+pyimagecuda/io.py,sha256=na_1pyg_SQ3D_-g4kThJginxVmcuBNw8ew5A7w0SoAg,8733
+pyimagecuda/pyimagecuda_internal.cp313-win_amd64.pyd,sha256=tCI74Mrcd3QOZH0LHjWEFNY0RKjjRJ1LgmdFwZPlXJQ,588800
 pyimagecuda/resize.py,sha256=txGYZr3KFmw9pYAbZ7vTnI6NDO-OycoZP6IeMirObcE,3291
 pyimagecuda/text.py,sha256=f3j1mi8ydk6kWiZt4gG7w85YFW--MwrezggpfuQW_tY,3558
 pyimagecuda/transform.py,sha256=cUrvPrNAkOoRDjuuOPdf-QnhZbk8VPMd01WR8IRgh_s,6450
 pyimagecuda/utils.py,sha256=8IZ35y783AVJVuoQmcStPVAyx5WK_PDhkWqLo51PbTo,633
-pyimagecuda-0.0.8.dist-info/METADATA,sha256=ILUUDcKbrj0jMtu1uF87jtRRpAyasv6TNG73p-8JeO4,6189
-pyimagecuda-0.0.8.dist-info/WHEEL,sha256=vkL3wTIkhjZa3RmEXX20hldNp6Q8qtwRjrXW6K5sw_Q,106
-pyimagecuda-0.0.8.dist-info/licenses/LICENSE,sha256=xfNh_pr4FPV8klv9rbs9GFflEfqf7vSFzcXK6BetRN0,1114
-pyimagecuda-0.0.8.dist-info/RECORD,,
+pyimagecuda-0.0.9.dist-info/METADATA,sha256=FuVdOs5g4OIaVmmtUuLKwV0hxRPliklePU3n22EL5QU,6477
+pyimagecuda-0.0.9.dist-info/WHEEL,sha256=vkL3wTIkhjZa3RmEXX20hldNp6Q8qtwRjrXW6K5sw_Q,106
+pyimagecuda-0.0.9.dist-info/licenses/LICENSE,sha256=xfNh_pr4FPV8klv9rbs9GFflEfqf7vSFzcXK6BetRN0,1114
+pyimagecuda-0.0.9.dist-info/RECORD,,