pybFoam 0.1.8__tar.gz → 0.1.10__tar.gz

{pybfoam-0.1.8 → pybfoam-0.1.10}/.github/workflows/ci.yaml

@@ -28,9 +28,15 @@ jobs:
       - name: Install build dependencies
         run: |
           python -m pip install --upgrade pip
+
       - name: Build package
         run: |
           pip install .[dev]
+
       - name: Run tests
         run: |
-          pytest
+          pytest
+
+      - name: run type checking
+        run: |
+          mypy 

{pybfoam-0.1.8 → pybfoam-0.1.10}/CMakeLists.txt

@@ -5,7 +5,7 @@ if(POLICY CMP0169)
     cmake_policy(SET CMP0169 OLD)
 endif()
 
-project(pybFoam VERSION 0.1.8 LANGUAGES CXX)
+project(pybFoam VERSION 0.1.10 LANGUAGES CXX)
 
 # Set C++ standard
 set(CMAKE_CXX_STANDARD 17)

{pybfoam-0.1.8 → pybfoam-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybFoam
-Version: 0.1.8
+Version: 0.1.10
 Summary: python bindings for OpenFOAM
 Keywords: OpenFOAM,simulation,CFD
 Author-Email: Henning Scheufler <henning.scheufler@web.de>
@@ -36,6 +36,8 @@ Requires-Dist: isort; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
 Requires-Dist: cmake>=3.18; extra == "dev"
 Requires-Dist: pybind11-stubgen>=2.5; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: types-PyYAML; extra == "dev"
 Provides-Extra: docs
 Requires-Dist: sphinx>=4.0; extra == "docs"
 Requires-Dist: sphinx-sitemap; extra == "docs"
@@ -46,6 +48,7 @@ Requires-Dist: jax; extra == "benchmark"
 Requires-Dist: seaborn; extra == "benchmark"
 Requires-Dist: matplotlib; extra == "benchmark"
 Requires-Dist: PyQt6; extra == "benchmark"
+Requires-Dist: tabulate; extra == "benchmark"
 Provides-Extra: all
 Requires-Dist: pybFoam[benchmark,dev,docs]; extra == "all"
 Description-Content-Type: text/markdown
@@ -103,9 +106,27 @@ pip install .
 
 For development:
 ```bash
-pip install -e . --no-build-isolation
+pip install -e .[all]
+```
+
+#### Generating Type Stubs (Development only)
+
+Type stubs (.pyi files) are generated post-installation using a separate script:
+
+```bash
+# Install the package first
+uv pip install -e .[all]
+
+# Generate and verify stubs
+./scripts/generate_stubs.sh
 ```
 
+This script:
+1. Generates stubs using pybind11-stubgen
+2. Cleans and formats the stubs
+3. Copies them to the source directory
+4. Verifies them with mypy
+
 ---
 
 ## Quick Start
@@ -216,6 +237,8 @@ Full documentation is available at: [https://henningscheufler.github.io/pybFoam/
 2. Install development dependencies:
    ```bash
    pip install -e ".[dev]"
+   # or build with stubs
+   pip install -e .[all] -C cmake.define.ENABLE_PYBFOAM_STUBS=ON -v
    ```
 3. Run tests before committing:
    ```bash

{pybfoam-0.1.8 → pybfoam-0.1.10}/README.md

@@ -51,9 +51,27 @@ pip install .
 
 For development:
 ```bash
-pip install -e . --no-build-isolation
+pip install -e .[all]
 ```
 
+#### Generating Type Stubs (Development only)
+
+Type stubs (.pyi files) are generated post-installation using a separate script:
+
+```bash
+# Install the package first
+uv pip install -e .[all]
+
+# Generate and verify stubs
+./scripts/generate_stubs.sh
+```
+
+This script:
+1. Generates stubs using pybind11-stubgen
+2. Cleans and formats the stubs
+3. Copies them to the source directory
+4. Verifies them with mypy
+
 ---
 
 ## Quick Start
@@ -164,6 +182,8 @@ Full documentation is available at: [https://henningscheufler.github.io/pybFoam/
 2. Install development dependencies:
    ```bash
    pip install -e ".[dev]"
+   # or build with stubs
+   pip install -e .[all] -C cmake.define.ENABLE_PYBFOAM_STUBS=ON -v
    ```
 3. Run tests before committing:
    ```bash

{pybfoam-0.1.8 → pybfoam-0.1.10}/benchmark/benchmark_expression_complex.py

@@ -86,43 +86,70 @@ def add_data(n_elements, duration, method):
 for n_elements in [10, 100, 1000, 10_000, 100_000, 1_000_000, 10_000_000]:
 
     bench = pybfoam_expression(n_elements)
-    # duration = timeit.timeit(bench, number=1)
-    t0 = time.perf_counter() # lower overhead than timeit
-    bench()
-    t1 = time.perf_counter()
-    duration = t1 - t0
-    add_data(n_elements, duration, "pybFoam")
+    n_repeat = 100
+    duration = timeit.timeit(bench, number=n_repeat)
+    # t0 = time.perf_counter() # lower overhead than timeit
+    # bench()
+    # t1 = time.perf_counter()
+    # duration = t1 - t0
+    add_data(n_elements, duration / n_repeat, "pybFoam")
 
     bench = numpy_expression(n_elements)
-    # duration = timeit.timeit(bench, number=1)
-    t0 = time.perf_counter() # lower overhead than timeit
-    bench()
-    t1 = time.perf_counter()
-    duration = t1 - t0
-    add_data(n_elements, duration, "NumPy")
+    duration = timeit.timeit(bench, number=n_repeat)
+    # t0 = time.perf_counter() # lower overhead than timeit
+    # bench()
+    # t1 = time.perf_counter()
+    # duration = t1 - t0
+    add_data(n_elements, duration / n_repeat, "NumPy")
 
     bench = jax_expression(n_elements)
     # First call to compile
     bench()
-    # duration = timeit.timeit(bench, number=1)
-    t0 = time.perf_counter() # lower overhead than timeit
-    bench()
-    t1 = time.perf_counter()
-    duration = t1 - t0
-    add_data(n_elements, duration, "JAX")
+    duration = timeit.timeit(bench, number=n_repeat)
+    # t0 = time.perf_counter() # lower overhead than timeit
+    # bench()
+    # t1 = time.perf_counter()
+    # duration = t1 - t0
+    add_data(n_elements, duration / n_repeat, "JAX")
 
 
 df_vector_add = pd.DataFrame(vector_add_data)
 df_vector_add["time_per_element [ns]"] = df_vector_add["duration"] / df_vector_add["n_elements"] * 1e9  # time per element in nanoseconds
 print("Vector Addition Benchmark:",df_vector_add)
 
+# Save results to CSV
+df_vector_add.to_csv("results/benchmark_complex.csv", index=False)
+print("\nResults saved to: results/benchmark_complex.csv")
+
+# Create pivot table for markdown
+pivot_df = df_vector_add.pivot(index="n_elements", columns="method", values="time_per_element [ns]")
+pivot_df = pivot_df.round(2)
+
+# Save markdown table
+with open("results/benchmark_complex.md", "w") as f:
+    f.write("# Complex Expression Benchmark\n\n")
+    f.write("Expression: `(x * y + b) / (a + 1.0)`\n\n")
+    f.write("where:\n")
+    f.write("- `x = a * b + c`\n")
+    f.write("- `y = d - a * c`\n\n")
+    f.write("## Results (time per element in nanoseconds)\n\n")
+    f.write(pivot_df.to_markdown())
+    f.write("\n\n## Summary\n\n")
+    for n_elem in [1_000_000, 10_000_000]:
+        if n_elem in pivot_df.index:
+            f.write(f"\n### {n_elem:,} elements:\n\n")
+            for method in pivot_df.columns:
+                f.write(f"- **{method}**: {pivot_df.loc[n_elem, method]:.2f} ns per element\n")
+
+print("Results table saved to: results/benchmark_complex.md")
+
 sns.lineplot(data=df_vector_add, x="n_elements", y="time_per_element [ns]", hue="method", marker="o")
 plt.xscale("log")
 plt.yscale("log")
 plt.xlabel("Number of Elements")
 plt.ylabel("Time per Element (nanoseconds)")
 plt.title("Expression: (x * y + b) / (a + 1.0)")
-plt.savefig("bench_complex.png")
+plt.savefig("results/bench_complex.png")
 n_elements = 10_000_000
 timings = df_vector_add[df_vector_add["n_elements"] == n_elements]
 print(f"Timings for {n_elements} elements:")

{pybfoam-0.1.8 → pybfoam-0.1.10}/benchmark/benchmark_expression_vec_add.py

@@ -97,13 +97,35 @@ df_vector_add = pd.DataFrame(vector_add_data)
 df_vector_add["time_per_element [ns]"] = df_vector_add["duration"] / df_vector_add["n_elements"] * 1e9  # time per element in nanoseconds
 print("Vector Addition Benchmark:",df_vector_add)
 
+# Save results to CSV
+df_vector_add.to_csv("results/benchmark_vec_add.csv", index=False)
+print("\nResults saved to: results/benchmark_vec_add.csv")
+
+# Create pivot table for markdown
+pivot_df = df_vector_add.pivot(index="n_elements", columns="method", values="time_per_element [ns]")
+pivot_df = pivot_df.round(2)
+
+# Save markdown table
+with open("results/benchmark_vec_add.md", "w") as f:
+    f.write("# Simple Vector Addition Benchmark\n\n")
+    f.write("Expression: `c = a + b`\n\n")
+    f.write("## Results (time per element in nanoseconds)\n\n")
+    f.write(pivot_df.to_markdown())
+    f.write("\n\n## Summary\n\n")
+    f.write(f"\n### {10_000_000:,} elements:\n\n")
+    if 10_000_000 in pivot_df.index:
+        for method in pivot_df.columns:
+            f.write(f"- **{method}**: {pivot_df.loc[10_000_000, method]:.2f} ns per element\n")
+
+print("Results table saved to: results/benchmark_vec_add.md")
+
 sns.lineplot(data=df_vector_add, x="n_elements", y="time_per_element [ns]", hue="method", marker="o")
 plt.xscale("log")
 plt.yscale("log")
 plt.xlabel("Number of Elements")
 plt.ylabel("Time per Element (nanoseconds)")
 plt.title("Expression: c = a + b")
-plt.savefig("bench_vec_add.png")
+plt.savefig("results/bench_vec_add.png")
 
 n_elements = 10_000_000
 timings = df_vector_add[df_vector_add["n_elements"] == n_elements]

{pybfoam-0.1.8 → pybfoam-0.1.10}/benchmark/benchmark_expression_vec_add_4.py

@@ -101,13 +101,35 @@ df_vector_add = pd.DataFrame(vector_add_data_4)
 df_vector_add["time_per_element [ns]"] = df_vector_add["duration"] / df_vector_add["n_elements"] * 1e9  # time per element in nanoseconds
 print("Vector Addition Benchmark:",df_vector_add)
 
+# Save results to CSV
+df_vector_add.to_csv("results/benchmark_vec_add_4.csv", index=False)
+print("\nResults saved to: results/benchmark_vec_add_4.csv")
+
+# Create pivot table for markdown
+pivot_df = df_vector_add.pivot(index="n_elements", columns="method", values="time_per_element [ns]")
+pivot_df = pivot_df.round(2)
+
+# Save markdown table
+with open("results/benchmark_vec_add_4.md", "w") as f:
+    f.write("# 4-Element Vector Addition Benchmark\n\n")
+    f.write("Expression: `c = a + b + c + a`\n\n")
+    f.write("## Results (time per element in nanoseconds)\n\n")
+    f.write(pivot_df.to_markdown())
+    f.write("\n\n## Summary\n\n")
+    f.write(f"\n### {10_000_000:,} elements:\n\n")
+    if 10_000_000 in pivot_df.index:
+        for method in pivot_df.columns:
+            f.write(f"- **{method}**: {pivot_df.loc[10_000_000, method]:.2f} ns per element\n")
+
+print("Results table saved to: results/benchmark_vec_add_4.md")
+
 sns.lineplot(data=df_vector_add, x="n_elements", y="time_per_element [ns]", hue="method", marker="o")
 plt.xscale("log")
 plt.yscale("log")
 plt.xlabel("Number of Elements")
 plt.ylabel("Time per Element (nanoseconds)")
 plt.title("Expression: c = a + b + c + a")
-plt.savefig("bench_vec_add_4.png")
+plt.savefig("results/bench_vec_add_4.png")
 n_elements = 10_000_000
 timings = df_vector_add[df_vector_add["n_elements"] == n_elements]
 print(f"Timings for {n_elements} elements:")

{pybfoam-0.1.8 → pybfoam-0.1.10}/cmake/Dependencies.cmake

@@ -42,35 +42,11 @@ function(add_pybind11)
     endif()
 endfunction()
 
-# Function to add testing dependencies
-function(add_testing_deps)
-    # Add Catch2 for C++ testing
-    CPMAddPackage(
-        NAME Catch2
-        GITHUB_REPOSITORY catchorg/Catch2
-        VERSION 3.4.0
-        OPTIONS
-            "CATCH_INSTALL_DOCS OFF"
-            "CATCH_INSTALL_EXTRAS OFF"
-    )
-    
-    if(Catch2_ADDED)
-        message(STATUS "Added Catch2 for testing")
-    endif()
-endfunction()
-
 # Main function to configure all dependencies
 function(configure_dependencies)
     message(STATUS "Configuring dependencies with CPM...")
-    
     # Essential dependencies
     add_pybind11()
-
-    option(PYBFOAM_BUILD_TESTS "Build tests" ON)
-    if(PYBFOAM_BUILD_TESTS)
-        add_testing_deps()
-    endif()
-    
     message(STATUS "Dependencies configuration complete")
 endfunction()
 

pybfoam-0.1.10/docs/api.rst

@@ -0,0 +1,6 @@
+API Reference
+=============
+
+.. automodule:: pybFoam
+   :members:
+   :undoc-members:

{pybfoam-0.1.8 → pybfoam-0.1.10}/docs/conf.py

@@ -9,7 +9,7 @@
 project = "pybFoam"
 copyright = "2025, Henning Scheufler"
 author = "Henning Scheufler"
-release = "0.1.8"
+release = "0.1.10"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

{pybfoam-0.1.8 → pybfoam-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pybFoam"
-version = "0.1.8"
+version = "0.1.10"
 authors = [
     {name = "Henning Scheufler", email = "henning.scheufler@web.de"},
 ]
@@ -42,6 +42,8 @@ dev = [
     "mypy",
     "cmake>=3.18",
     "pybind11-stubgen>=2.5",
+    "mypy",
+    "types-PyYAML",
 ]
 docs = [
     "sphinx>=4.0",
@@ -53,7 +55,8 @@ benchmark = [
     "jax",
     "seaborn",
     "matplotlib",
-    "PyQt6"
+    "PyQt6",
+    "tabulate"
 ]
 all = [
     "pybFoam[dev,docs,benchmark]",
@@ -72,7 +75,7 @@ zip-safe = false
 requires = [
     "scikit-build-core[pyproject]>=0.4.3",
     "pybind11>=2.6.0",
-    "cmake>=3.18",
+    "cmake>=3.18"
 ]
 build-backend = "scikit_build_core.build"
 
@@ -92,11 +95,19 @@ editable.rebuild = false
 [tool.scikit-build.cmake.define]
 PYBIND11_FINDPYTHON = "ON"
 CMAKE_CXX_STANDARD = "17"
+# Stub generation disabled by default
+# Developers enable with: uv sync --all-extras -C cmake.define.ENABLE_PYBFOAM_STUBS=ON
 
 # Generate stubs for better IDE support
 [tool.scikit-build.experimental]
 generate-type-stubs = true
 
 [tool.pytest.ini_options]
-addopts = "-ra -v"
+addopts = "-ra -v -s"
 testpaths = ["tests"]
+
+[tool.mypy]
+plugins = ["pydantic.mypy"]
+mypy_path = ["src"] 
+strict = true 
+files = ["src"]

pybfoam-0.1.10/scripts/clean_stubs.py

@@ -0,0 +1,166 @@
+#!/usr/bin/env python3
+"""
+Clean pybFoam_core.pyi by replacing C++ Foam:: types with Python types.
+"""
+
+import re
+from pathlib import Path
+
+# Type replacements: C++ pattern -> Python type
+# IMPORTANT: Order matters! Most complex/specific patterns FIRST, simple patterns LAST
+# This prevents partial replacements from breaking complex pattern matching
+TYPE_REPLACEMENTS = {
+    # ===================================================================
+    # MOST COMPLEX PATTERNS FIRST - tmp wrapped geometric fields
+    # ===================================================================
+    
+    # tmp wrapped geometric fields - volume
+    r'Foam::tmp<Foam::GeometricField<double,\s*Foam::fvPatchField,\s*Foam::volMesh>>': 'tmp_volScalarField',
+    r'Foam::tmp<Foam::GeometricField<Foam::Vector<double>,\s*Foam::fvPatchField,\s*Foam::volMesh>>': 'tmp_volVectorField',
+    r'Foam::tmp<Foam::GeometricField<Foam::Tensor<double>,\s*Foam::fvPatchField,\s*Foam::volMesh>>': 'tmp_volTensorField',
+    r'Foam::tmp<Foam::GeometricField<Foam::SymmTensor<double>,\s*Foam::fvPatchField,\s*Foam::volMesh>>': 'tmp_volSymmTensorField',
+    
+    # tmp wrapped geometric fields - surface
+    r'Foam::tmp<Foam::GeometricField<double,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>>': 'tmp_surfaceScalarField',
+    r'Foam::tmp<Foam::GeometricField<Foam::Vector<double>,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>>': 'tmp_surfaceVectorField',
+    r'Foam::tmp<Foam::GeometricField<Foam::Tensor<double>,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>>': 'tmp_surfaceTensorField',
+    r'Foam::tmp<Foam::GeometricField<Foam::SymmTensor<double>,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>>': 'tmp_surfaceSymmTensorField',
+    
+    # tmp wrapped fields with full type names
+    r'Foam::tmp<Foam::Field<double>>': 'tmp_scalarField',
+    r'Foam::tmp<Foam::Field<Foam::Vector<double>>>': 'tmp_vectorField',
+    r'Foam::tmp<Foam::Field<Foam::Tensor<double>>>': 'tmp_tensorField',
+    r'Foam::tmp<Foam::Field<Foam::SymmTensor<double>>>': 'tmp_symmTensorField',
+    
+    # ===================================================================
+    # GEOMETRIC FIELDS - volume and surface
+    # ===================================================================
+    
+    # Geometric fields - volume
+    r'Foam::GeometricField<double,\s*Foam::fvPatchField,\s*Foam::volMesh>': 'volScalarField',
+    r'Foam::GeometricField<Foam::Vector<double>,\s*Foam::fvPatchField,\s*Foam::volMesh>': 'volVectorField',
+    r'Foam::GeometricField<Foam::Tensor<double>,\s*Foam::fvPatchField,\s*Foam::volMesh>': 'volTensorField',
+    r'Foam::GeometricField<Foam::SymmTensor<double>,\s*Foam::fvPatchField,\s*Foam::volMesh>': 'volSymmTensorField',
+    
+    # Geometric fields - surface
+    r'Foam::GeometricField<double,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>': 'surfaceScalarField',
+    r'Foam::GeometricField<Foam::Vector<double>,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>': 'surfaceVectorField',
+    r'Foam::GeometricField<Foam::Tensor<double>,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>': 'surfaceTensorField',
+    r'Foam::GeometricField<Foam::SymmTensor<double>,\s*Foam::fvsPatchField,\s*Foam::surfaceMesh>': 'surfaceSymmTensorField',
+    
+    # ===================================================================
+    # FIELD TYPES with full type names
+    # ===================================================================
+    
+    r'Foam::Field<double>': 'scalarField',
+    r'Foam::Field<Foam::Vector<double>>': 'vectorField',
+    r'Foam::Field<Foam::Tensor<double>>': 'tensorField',
+    r'Foam::Field<Foam::SymmTensor<double>>': 'symmTensorField',
+    
+    # ===================================================================
+    # SOLVER PERFORMANCE with full type names
+    # ===================================================================
+    
+    r'Foam::SolverPerformance<double>': 'SolverScalarPerformance',
+    r'Foam::SolverPerformance<Foam::Vector<double>\s*>': 'SolverVectorPerformance',
+    r'Foam::SolverPerformance<Foam::Tensor<double>\s*>': 'SolverTensorPerformance',
+    r'Foam::SolverPerformance<Foam::SymmTensor<double>\s*>': 'SolverSymmTensorPerformance',
+    
+    # ===================================================================
+    # DIMENSIONED TYPES with full type names
+    # ===================================================================
+    
+    r'Foam::dimensioned<double>': 'dimensionedScalar',
+    r'Foam::dimensioned<Foam::Vector<double>\s*>': 'dimensionedVector',
+    r'Foam::dimensioned<Foam::Tensor<double>\s*>': 'dimensionedTensor',
+    r'Foam::dimensioned<Foam::SymmTensor<double>\s*>': 'dimensionedSymmTensor',
+    
+    # ===================================================================
+    # FV MATRIX TYPES with full type names
+    # ===================================================================
+    
+    r'Foam::fvMatrix<double>': 'fvScalarMatrix',
+    r'Foam::fvMatrix<Foam::Vector<double>\s*>': 'fvVectorMatrix',
+    r'Foam::fvMatrix<Foam::Tensor<double>\s*>': 'fvTensorMatrix',
+    r'Foam::fvMatrix<Foam::SymmTensor<double>\s*>': 'fvSymmTensorMatrix',
+    
+    # ===================================================================
+    # LISTS
+    # ===================================================================
+    
+    r'Foam::List<Foam::word>': 'wordList',
+    r'Foam::List<Foam::instant>': 'instantList',
+    r'Foam::List<int>': 'labelList',
+    r'Foam::List<bool>': 'boolList',
+    
+    # ===================================================================
+    # SIMPLE TYPES - Must come LAST!
+    # ===================================================================
+    
+    # dictionary
+    r'Foam::dictionary': 'dictionary',
+    
+    # Primitives - these come last to avoid breaking complex patterns
+    r'Foam::Vector<double>': 'vector',
+    r'Foam::Tensor<double>': 'tensor',
+    r'Foam::SymmTensor<double>': 'symmTensor',
+    
+    # Simple object types
+    r'Foam::word': 'Word',
+    r'Foam::Time': 'Time',
+    r'Foam::fvMesh': 'fvMesh',
+    r'Foam::instant': 'instant',
+}
+
+
+def main():
+    # Look for stubs directory relative to current working directory
+    cwd = Path.cwd()
+    stubs_dir = cwd / 'stubs'
+    
+    # If not found in cwd, try relative to script location
+    if not stubs_dir.exists():
+        script_dir = Path(__file__).parent.parent
+        stubs_dir = script_dir / 'stubs'
+    
+    if not stubs_dir.exists():
+        print(f"Error: stubs directory not found (tried {cwd / 'stubs'} and {script_dir / 'stubs'})")
+        return
+    
+    # Process all .pyi files in stubs directory recursively
+    stub_files = list(stubs_dir.glob('**/*.pyi'))
+    
+    if not stub_files:
+        print(f"No stub files found in {stubs_dir}")
+        return
+    
+    print(f"Cleaning {len(stub_files)} stub file(s) in {stubs_dir}...")
+    
+    for stub_file in stub_files:
+        # Show relative path for clarity
+        relative_path = stub_file.relative_to(stubs_dir)
+        print(f"  Processing {relative_path}...")
+        content = stub_file.read_text()
+        original = content
+        
+        # Normalize whitespace in C++ templates first
+        # Fix broken scope operators: "Foam: :Type" -> "Foam::Type"
+        content = re.sub(r':\s+:', '::', content)
+        # Remove spaces before closing >
+        content = re.sub(r'\s+>', '>', content)
+        
+        # Apply all replacements
+        for pattern, replacement in TYPE_REPLACEMENTS.items():
+            content = re.sub(pattern, replacement, content)
+        
+        if content != original:
+            stub_file.write_text(content)
+            print(f"    ✓ Cleaned {relative_path}")
+        else:
+            print(f"    - No changes needed for {relative_path}")
+    
+    print(f"\nStub cleaning complete!")
+
+
+if __name__ == '__main__':
+    main()

pybfoam-0.1.10/scripts/generate_stubs.sh

@@ -0,0 +1,73 @@
+#!/bin/bash
+# Generate type stubs for pybFoam after installation
+# Usage: ./scripts/generate_stubs.sh
+
+set -e
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
+STUBS_DIR="$PROJECT_ROOT/stubs"
+SRC_DIR="$PROJECT_ROOT/src/pybFoam"
+
+echo "Project Root: $PROJECT_ROOT"
+echo "Source Directory: $SRC_DIR"
+echo "Stubs Directory: $STUBS_DIR"
+
+echo "==================================="
+echo "pybFoam Stub Generator"
+echo "==================================="
+echo ""
+
+# Check if pybFoam is installed
+if ! uv run python -c "import pybFoam" 2>/dev/null; then
+    echo "Error: pybFoam is not installed!"
+    echo "Please install first: uv pip install -e .[all]"
+    exit 1
+fi
+
+echo "✓ pybFoam is installed"
+echo ""
+
+# Generate stubs
+echo "Generating stubs..."
+rm -rf "$STUBS_DIR"
+uv run pybind11-stubgen -o "$STUBS_DIR" --print-invalid-expressions-as-is pybFoam
+
+if [ ! -d "$STUBS_DIR/pybFoam" ]; then
+    echo "Error: Stub generation failed!"
+    exit 1
+fi
+
+echo "✓ Stubs generated in $STUBS_DIR"
+echo ""
+
+# Clean stubs
+echo "Cleaning stubs..."
+uv run python "$SCRIPT_DIR/clean_stubs.py"
+echo "✓ Stubs cleaned"
+echo ""
+
+# Copy stubs to source directory
+echo "Copying stubs to source directory..."
+cp -r "$STUBS_DIR/pybFoam/"* "$SRC_DIR/"
+echo "✓ Stubs copied to $SRC_DIR"
+echo ""
+
+# Verify with mypy
+echo "Verifying stubs with mypy..."
+if uv run mypy --version >/dev/null 2>&1; then
+    echo ""
+    echo "Running mypy on pybFoam..."
+    uv run mypy "$SRC_DIR" --show-error-codes || true
+    echo ""
+else
+    echo "Warning: mypy not installed, skipping verification"
+    echo "Install with: uv pip install mypy"
+fi
+
+echo "==================================="
+echo "✓ Stub generation complete!"
+echo "==================================="
+echo ""
+echo "Generated stub files:"
+find "$SRC_DIR" -name "*.pyi" -type f | sort