foamlib 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

foamlib/__init__.py

@@ -1,6 +1,6 @@
 """A Python interface for interacting with OpenFOAM."""
 
-__version__ = "0.5.0"
+__version__ = "0.5.1"
 
 from ._cases import (
     AsyncFoamCase,

foamlib/_cases/_async.py

@@ -5,16 +5,18 @@ import sys
 import tempfile
 from contextlib import asynccontextmanager
 from pathlib import Path
-from typing import (
-    Callable,
-    Optional,
-    Union,
-)
+from typing import Callable, Optional, TypeVar, Union
 
 if sys.version_info >= (3, 9):
-    from collections.abc import AsyncGenerator, Collection, Sequence
+    from collections.abc import (
+        AsyncGenerator,
+        Awaitable,
+        Collection,
+        Iterable,
+        Sequence,
+    )
 else:
-    from typing import AsyncGenerator, Collection, Sequence
+    from typing import AsyncGenerator, Awaitable, Collection, Iterable, Sequence
 
 if sys.version_info >= (3, 11):
     from typing import Self
@@ -27,6 +29,9 @@ from ._recipes import _FoamCaseRecipes
 from ._subprocess import run_async
 from ._util import awaitableasynccontextmanager
 
+X = TypeVar("X")
+Y = TypeVar("Y")
+
 
 class AsyncFoamCase(_FoamCaseRecipes):
     """
@@ -231,3 +236,10 @@ class AsyncFoamCase(_FoamCaseRecipes):
             await self._rmtree(dst.parent)
         else:
             await self._rmtree(dst)
+
+    @staticmethod
+    def map(coro: Callable[[X], Awaitable[Y]], iterable: Iterable[X]) -> Iterable[Y]:
+        """Run an async function on each element of an iterable concurrently."""
+        return asyncio.get_event_loop().run_until_complete(
+            asyncio.gather(*(coro(arg) for arg in iterable))
+        )

foamlib/_cases/_sync.py

@@ -36,6 +36,10 @@ class FoamCase(_FoamCaseRecipes):
     :param path: The path to the case directory.
     """
 
+    def __init__(self, path: Union["os.PathLike[str]", str] = Path()):
+        super().__init__(path)
+        self._tmp: Optional[bool] = None
+
     @staticmethod
     def _rmtree(
         path: Union["os.PathLike[str]", str], *, ignore_errors: bool = False
@@ -55,6 +59,10 @@ class FoamCase(_FoamCaseRecipes):
         shutil.copytree(src, dest, symlinks=symlinks, ignore=ignore)
 
     def __enter__(self) -> "FoamCase":
+        if self._tmp is None:
+            raise RuntimeError(
+                "Cannot use a non-copied/cloned case as a context manager"
+            )
         return self
 
     def __exit__(
@@ -63,7 +71,15 @@ class FoamCase(_FoamCaseRecipes):
         exc_val: Optional[BaseException],
         exc_tb: Optional[TracebackType],
     ) -> None:
-        self._rmtree(self.path)
+        if self._tmp is not None:
+            if self._tmp:
+                self._rmtree(self.path.parent)
+            else:
+                self._rmtree(self.path)
+        else:
+            raise RuntimeError(
+                "Cannot use a non-copied/cloned case as a context manager"
+            )
 
     def clean(
         self,
@@ -160,11 +176,17 @@ class FoamCase(_FoamCaseRecipes):
         """
         if dst is None:
             dst = Path(tempfile.mkdtemp(), self.name)
+            tmp = True
+        else:
+            tmp = False
 
         for name, args, kwargs in self._copy_cmds(dst):
             getattr(self, name)(*args, **kwargs)
 
-        return type(self)(dst)
+        ret = type(self)(dst)
+        ret._tmp = tmp
+
+        return ret
 
     def clone(self, dst: Optional[Union["os.PathLike[str]", str]] = None) -> "Self":
         """
@@ -176,8 +198,14 @@ class FoamCase(_FoamCaseRecipes):
         """
         if dst is None:
             dst = Path(tempfile.mkdtemp(), self.name)
+            tmp = True
+        else:
+            tmp = False
 
         for name, args, kwargs in self._clone_cmds(dst):
             getattr(self, name)(*args, **kwargs)
 
-        return type(self)(dst)
+        ret = type(self)(dst)
+        ret._tmp = tmp
+
+        return ret

foamlib/_cases/_util.py

@@ -1,15 +1,33 @@
+import functools
+import sys
 from types import TracebackType
-from typing import Any, AsyncContextManager, Callable, Optional, Type
+from typing import (
+    Any,
+    AsyncContextManager,
+    Callable,
+    Generic,
+    Optional,
+    Type,
+    TypeVar,
+)
 
+if sys.version_info >= (3, 9):
+    from collections.abc import Generator
+else:
+    from typing import Generator
 
-class _AwaitableAsyncContextManager:
-    def __init__(self, cm: "AsyncContextManager[Any]"):
+
+R = TypeVar("R")
+
+
+class _AwaitableAsyncContextManager(Generic[R]):
+    def __init__(self, cm: "AsyncContextManager[R]"):
         self._cm = cm
 
-    def __await__(self) -> Any:
+    def __await__(self) -> Generator[Any, Any, R]:
         return self._cm.__aenter__().__await__()
 
-    async def __aenter__(self) -> Any:
+    async def __aenter__(self) -> R:
         return await self._cm.__aenter__()
 
     async def __aexit__(
@@ -17,11 +35,15 @@ class _AwaitableAsyncContextManager:
         exc_type: Optional[Type[BaseException]],
         exc_val: Optional[BaseException],
         exc_tb: Optional[TracebackType],
-    ) -> Any:
+    ) -> Optional[bool]:
         return await self._cm.__aexit__(exc_type, exc_val, exc_tb)
 
 
 def awaitableasynccontextmanager(
-    cm: Callable[..., "AsyncContextManager[Any]"],
-) -> Callable[..., _AwaitableAsyncContextManager]:
-    return lambda *args, **kwargs: _AwaitableAsyncContextManager(cm(*args, **kwargs))
+    cm: Callable[..., "AsyncContextManager[R]"],
+) -> Callable[..., _AwaitableAsyncContextManager[R]]:
+    @functools.wraps(cm)
+    def f(*args: Any, **kwargs: Any) -> _AwaitableAsyncContextManager[R]:
+        return _AwaitableAsyncContextManager(cm(*args, **kwargs))
+
+    return f

{foamlib-0.5.0.dist-info → foamlib-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foamlib
-Version: 0.5.0
+Version: 0.5.1
 Summary: A Python interface for interacting with OpenFOAM
 Author-email: "Gabriel S. Gerlero" <ggerlero@cimec.unl.edu.ar>
 Project-URL: Homepage, https://github.com/gerlero/foamlib
@@ -155,6 +155,25 @@ U = FoamFieldFile(Path(my_pitz) / "0/U")
 print(U.internal_field)
 ```
 
+### Run an optimization loop in parallel
+
+```python
+import os
+from pathlib import Path
+from foamlib import AsyncFoamCase
+from scipy.optimize import differential_evolution
+
+base = AsyncFoamCase(Path(os.environ["FOAM_TUTORIALS"]) / "incompressible/simpleFoam/pitzDaily")
+
+async def cost(x):
+    async with base.clone() as clone:
+        clone[0]["U"].boundary_field["inlet"].value = [x[0], 0, 0]
+        await clone.run()
+        return abs(clone[-1]["U"].internal_field[0][0])
+
+result = differential_evolution(cost, bounds=[(-1, 1)], workers=AsyncFoamCase.map, polish=False)
+```
+
 ## Documentation
 
 For more information, check out the [documentation](https://foamlib.readthedocs.io/).

{foamlib-0.5.0.dist-info → foamlib-0.5.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
-foamlib/__init__.py,sha256=VR4c4hlAQ8I180EqNsDls6vyYKbnOScayXFOr2fKAEk,392
+foamlib/__init__.py,sha256=bRzkPtbKy1RniCL9eYinI6Yc8g8utDoXrPV6hFimMZY,392
 foamlib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 foamlib/_cases/__init__.py,sha256=C0mpRu7c-X-4uVMKmVrZhwIyhBNyvUoCv0o-BQ72RC0,236
-foamlib/_cases/_async.py,sha256=QVZaYJCOaUgt0_xI_cYsZAJcdfpdZoYnh5dojTjMX08,7816
+foamlib/_cases/_async.py,sha256=um1nrAjGcT8Vp1hz8XvC_87xVBjfeU3MrpnuzAHN5GY,8251
 foamlib/_cases/_base.py,sha256=1CUkkK4afBxDgP79dmho97WJdj-GLgYhnrCSf_52Eao,6604
 foamlib/_cases/_recipes.py,sha256=UTFnVuTvEf-9wn-Fr30a9wOOmyOxvWeDhbqBhdtbhzA,9692
 foamlib/_cases/_subprocess.py,sha256=CfUy_LrqLnMLR9FHINqInCd3soN6eYvonwZ30epiLu8,2234
-foamlib/_cases/_sync.py,sha256=JQAkKXwgdhR_drGy7gadun1XPGjNddKR7Y07HK-TpoA,5964
-foamlib/_cases/_util.py,sha256=4jlv8pe2Ro1-ZtpY5DtR2B1Vpbj84SlqH_HzwGYPoH4,861
+foamlib/_cases/_sync.py,sha256=R32Ea4BTOPq-x9OIndObcYZACF8e0RnzB4TywNvDxbA,6733
+foamlib/_cases/_util.py,sha256=BpPW_91bFLzV3b4r50u-2pGMR3tfQQfY2xqToOHCUBk,1204
 foamlib/_files/__init__.py,sha256=-UqB9YTH6mrJfXCX00kPTAAY20XG64u1MGPw_1ewLVs,148
 foamlib/_files/_base.py,sha256=zaFDjLE6jB7WtGWk8hfKusjLtlGu6CZV16AHJpRUibs,1929
 foamlib/_files/_files.py,sha256=6Fdrc0lksFK99i1a6wEsBi-BtYgSQnUc5s10h2LQ9ew,16091
@@ -14,8 +14,8 @@ foamlib/_files/_io.py,sha256=f_tYI7AqaFsQ8mtK__fEoIUqpYb3YmrI8X5D8updmNM,2084
 foamlib/_files/_parsing.py,sha256=8V2CKZ45mKE3f9fP8lAfexIdhPGrq7elIZkpBkkGB6Q,8773
 foamlib/_files/_serialization.py,sha256=pb8_cIVgRhGS_ZV2p3x8p5_lK1SS6xzQHscAYYuOgFY,3407
 foamlib/_files/_util.py,sha256=UMzXmTFgvbp46w6k3oEZJoYC98pFgEK6LN5uLOwrlCg,397
-foamlib-0.5.0.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
-foamlib-0.5.0.dist-info/METADATA,sha256=pecBt-ku69uh-HjSN02M40tCqhT22L-sZ9Fcf6AezwM,5759
-foamlib-0.5.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-foamlib-0.5.0.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
-foamlib-0.5.0.dist-info/RECORD,,
+foamlib-0.5.1.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
+foamlib-0.5.1.dist-info/METADATA,sha256=4Fq_okfGu-zY0IGrcUCD5IuuJxAESwNtr0EgXD0NzF8,6338
+foamlib-0.5.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+foamlib-0.5.1.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
+foamlib-0.5.1.dist-info/RECORD,,