diffinytrace 2.1__py3-none-any.whl

diffinytrace/utils/irradiance_importer.py

@@ -0,0 +1,134 @@
+# Copyright (c) 2025 Martin Pflaum
+# This file is part of the diffinytrace project, licensed under the MIT License.
+
+__all__ = [
+    "create_irradiance_from_image_square",
+    "pil_center_crop",
+    "load_image"
+]
+
+import torch
+from PIL import Image
+import numpy as np
+from ..target_grid import GridSquare
+import gc
+
+"""
+def from_image_square(file_name,padding_ratio,grey_ratio,aperture_radius):
+    #TODO maybe generalize to rectangle  - change apreture_radius_detector to target_grid-nöö
+    image = load_image(file_name,padding_ratio=padding_ratio,grey_ratio=grey_ratio)
+    image = torch.tensor(image).to(torch.get_default_dtype())
+    image = image.T
+    image_flat = image.reshape(-1)
+    grid = dit.target_grid.GridSquare(aperture_radius,grid_size=image.shape[0])
+    area = grid.get_area()
+    def desired_irradiance_func(x):    
+        k = grid.get_k(x)
+        tmp = image_flat[k]    
+        return tmp/area
+    return desired_irradiance_func
+
+
+"""
+def create_irradiance_from_image_square(file_name,padding_ratio,grey_ratio,aperture_radius,
+                                        dtype=torch.get_default_dtype(),shape=None):
+    """
+    Create a function that returns the desired irradiance from an image file.
+    
+    Args:
+        file_name (str): The path to the image file.
+        padding_ratio (float): The ratio of padding to be applied to the image.
+        grey_ratio (float): The ratio of grey value to be applied to the image.
+        aperture_radius (float): The radius of the aperture.
+        dtype (torch.dtype): The data type for the output tensor.
+        shape (tuple, optional): The shape to resize the image to. If None, no resizing is done.
+    Returns:
+        function: A function that takes a tensor as input and returns the desired irradiance.
+    """
+    
+    #TODO maybe generalize to rectangle  - change apreture_radius_detector to target_grid-nöö
+    image = load_image(file_name,padding_ratio=padding_ratio,grey_ratio=grey_ratio,shape=shape)
+    #image = image.T
+    image = np.array(image[::-1])
+    image = np.array(image)
+    image = torch.tensor(image).to(dtype=dtype)
+    shape = image.shape
+    
+    image = image.reshape(-1)
+    grid = GridSquare(aperture_radius,grid_size=shape[0])
+    area = grid.get_area()
+    def desired_irradiance_func(x):
+        device = x.device
+        dtype = x.dtype
+        
+        x = torch.clamp(x, min=-aperture_radius, max=aperture_radius)
+        k = grid.get_k(x,round_to_bounds=True)
+        k = k.cpu()
+        
+        tmp = image[k] 
+        out = tmp/area
+        out = out.to(device=device,dtype=dtype)
+        return out
+    
+    return desired_irradiance_func
+
+
+def pil_center_crop(image):
+    """
+    Crop the image to a square by centering it.
+    
+    Args:
+        image (PIL.Image): The image to be cropped.
+    
+    Returns:
+        PIL.Image: The cropped image.
+    """
+
+    width, height = image.size
+    crop_box = None
+    if width < height:
+        crop_start = (height-width)//2
+        crop_box = (0,crop_start , width, crop_start+width)  # (left, upper, right, lower)
+    else:
+        crop_start = (width-height)//2
+        crop_box = (crop_start,0 ,crop_start+height,height)  # (left, upper, right, lower)
+
+    cropped_image = image.crop(crop_box)
+    return cropped_image
+
+
+def load_image(name,padding_ratio,grey_ratio,shape=None):
+    """
+    Load an image, convert it to grayscale, and apply padding.
+    
+    Args:
+        name (str): The path to the image file.
+        padding_ratio (float): The ratio of padding to be applied to the image.
+        grey_ratio (float): The ratio of grey value to be applied to the image.
+        shape (tuple, optional): The shape to resize the image to. If None, no resizing is done.
+    
+    Returns:
+        np.ndarray: The processed image as a numpy array.
+    """
+    # Open and process the image
+    image = Image.open(name)
+    image = pil_center_crop(image)  # Assuming this function crops the image to its center
+    image = image.convert("L")  # Convert the image to grayscale
+    if shape is not None:
+        image = image.resize(shape)
+
+    # Convert the image to a numpy array
+    image_array = np.array(image)/255.0
+    image_array = np.ones_like(image_array)*grey_ratio+(1.-grey_ratio)*image_array
+    
+    # Calculate the padding size as 20% of the original dimensions
+    pad_height = int(image_array.shape[0] * padding_ratio*0.5)  # 20% of the height
+    pad_width = int(image_array.shape[1] * padding_ratio*0.5)   # 20% of the width
+
+    # Pad the image with constant value 0 (or any value you prefer)
+    padded_image = np.pad(image_array, 
+                        pad_width=((pad_height, pad_height), (pad_width, pad_width)), 
+                        mode='constant', 
+                        constant_values=0)
+    padded_image = padded_image / np.sum(padded_image)
+    return padded_image

diffinytrace-2.1.dist-info/METADATA

@@ -0,0 +1,26 @@
+Metadata-Version: 2.4
+Name: diffinytrace
+Version: 2.1
+Author: Martin Pflaum
+Author-email: contact@martinpflaum.com
+Requires-Python: ==3.12
+License-File: LICENSE
+Requires-Dist: colour-science==0.4.7
+Requires-Dist: matplotlib==3.10.8
+Requires-Dist: numpy==2.3.5
+Requires-Dist: pandas==3.0.1
+Requires-Dist: Pillow==12.0.0
+Requires-Dist: plotly==6.6.0
+Requires-Dist: pvlib==0.15.0
+Requires-Dist: scikit_learn==1.8.0
+Requires-Dist: scipy==1.17.1
+Requires-Dist: tqdm==4.67.3
+Requires-Dist: nbformat==5.1.3
+Requires-Dist: ipykernel==7.2.0
+Requires-Dist: cadquery==2.7.0
+Requires-Dist: torchmetrics==1.9.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python

diffinytrace-2.1.dist-info/RECORD

@@ -0,0 +1,38 @@
+diffinytrace/__init__.py,sha256=jEr3e-31VQCVGYB1_etH2DVOT7riR-7uo9NrYZVuvlo,3367
+diffinytrace/config.py,sha256=QH6hTvGnXrFsM722-OoCLgIL1SuoehTinVqty8rBZ4M,4345
+diffinytrace/constraints.py,sha256=qEhoMQzezuCQBvJmNvNzvJrfrZERKw8RKcl5icEnbLg,1388
+diffinytrace/element.py,sha256=OrN-IWsh5IoZBFTJxAGm-A2e3PBcpoLqOTekY1_wuuA,67713
+diffinytrace/gaussian_smoother.py,sha256=VAaesEUO4oqqC5JCSJJ7ZMNZoj8uYdglMtb23Ddrf4M,26395
+diffinytrace/hat_smoother.py,sha256=k2pLqafJLr_N-488Dhhs-N35QbMg30DbNL2N4nEH48s,1307
+diffinytrace/integrators.py,sha256=NnSkmwzIFeObH1yQtJ_aH_Q7GvAsUFbbxMw79zFdjjU,19030
+diffinytrace/intersection.py,sha256=An2lPg6KfBZuMAOG2Pa18SgsImYNaGLFpFxMsxztriQ,10537
+diffinytrace/optimize.py,sha256=VVUg8MZ79NsYzHxYPD0d4Felrm3m6h9W-jTo7Wk3VHM,33358
+diffinytrace/physical_object.py,sha256=ic88YjeCtvWMfJkrkUpMUKC0FnZ_vAYuttkGjwIepGY,4788
+diffinytrace/refractive_index.py,sha256=8vZhE5exZiCM6vtH6E2u4bUpTdaRX5_Hvw4cJ_D2mX0,5555
+diffinytrace/render.py,sha256=_j4ROqbjHrmSz9HJFxtERMtNbnTuAc26qdryoTutiJ4,3598
+diffinytrace/source.py,sha256=s0ugQ89qvJCETfGI9r4l3DQ-gVTZoflBn8C0xbMi2qA,25911
+diffinytrace/spectrum.py,sha256=0hNU0K8aMWvvTUetsM9nvI3zMUhKhubBOt_ZVjOPSEA,2516
+diffinytrace/surface.py,sha256=RSjaIlFBeNFiPGKamyKgoCJIFuAuROnEUSvCO7GsHkg,18950
+diffinytrace/target_grid.py,sha256=QTF1hvKLMrCW6l1q8BvWYpMvT4khEkh80GjzeZR0xio,15613
+diffinytrace/transforms.py,sha256=6CHyZXtSrgKRiy_XRODFZBURA8ocESOgDHED-5IbpDA,16195
+diffinytrace/basis_functions/__init__.py,sha256=h2Bahd7ZBm4Z-t2e_aNPIUoa0YO_LNF_qYMHxnH5Eoc,297
+diffinytrace/basis_functions/bspline.py,sha256=tig1xgUxcTnxK9wWvAhACWKhRzpWRVliP6iMh1dC_LY,20404
+diffinytrace/basis_functions/chebyshev.py,sha256=cguPQOs2ibEDlsL99hjQ2Wo4cr8Hcea2NVzTsCbF7lw,120
+diffinytrace/basis_functions/legendre.py,sha256=8CdpC_Z-IwEh0I-4HoYXqm1T1c4bID446VjvKyn-QqQ,2483
+diffinytrace/basis_functions/zernike.py,sha256=bNIv89mY0ox09-irW_uFMYkeF64YqCZLjNKdanU_9lA,8294
+diffinytrace/export/__init__.py,sha256=RsjN9PavLbNtwchNHE3Y6nWl12mYCv5VND6uAGtMf1k,166
+diffinytrace/export/cad.py,sha256=kYVB_wUIXnYPlkNRs48-iyJCzwYr-M9UI914q7CX2nE,9103
+diffinytrace/plotting/__init__.py,sha256=7v_jp_Utpi7LLTEqB3YsB0GFdt6oOuyTHnA0jT94jI8,354
+diffinytrace/plotting/core.py,sha256=ggY3ArGyvYod7PcsuZEkzM_AZXVOjkika1hHTMNNmrs,3335
+diffinytrace/plotting/quantity2D.py,sha256=mXOrjx0EgDPb9DGJwHlGVjKZn4GDz3NcPqPu_fYy9Yk,7920
+diffinytrace/plotting/system2D.py,sha256=vrI0i4N0obiYKO3Sj1cqoJuAjt4_EEUV31UpOPpdx_4,7628
+diffinytrace/plotting/system3D.py,sha256=kn3r3arb7s4T6FvDQqXc-VAJKbIYCEoCiWpdxaOg2H0,11292
+diffinytrace/plotting/wavelength.py,sha256=ZmZMw1ay2wc7NyAo4Z_9jNVIuwaGmfHw2jCHhJHaHnQ,7384
+diffinytrace/utils/__init__.py,sha256=BevGtOt6JSKgBExvpqaKLIk39ebFdzFqAJgL_DBUXxo,211
+diffinytrace/utils/autograd.py,sha256=VDnbXJfe-uIY5Mbc_bIljZFWcKIZfk6jXamT-8CZKWM,4435
+diffinytrace/utils/irradiance_importer.py,sha256=FvaTK0uRi2KnEZc6CcD0Tdh03I8Wum7XjdbMGeIZQlQ,5000
+diffinytrace-2.1.dist-info/licenses/LICENSE,sha256=zcS2z4c_EPqi2EDV5ZpOv4FC4flqC7kYw4qOKtWHKjY,1091
+diffinytrace-2.1.dist-info/METADATA,sha256=LyMc5UfTxbjqw0s8w1K5GAMWxSk5p4bgiYnAsYqomWo,732
+diffinytrace-2.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+diffinytrace-2.1.dist-info/top_level.txt,sha256=fK-f6Fho578N7Hz-tgLor5IEiUVOqrmLI07lp6s-iyg,13
+diffinytrace-2.1.dist-info/RECORD,,

diffinytrace-2.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

diffinytrace-2.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Martin Pflaum
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

diffinytrace-2.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+diffinytrace