foamlib 0.6.10__py3-none-any.whl → 0.6.12__py3-none-any.whl

foamlib/_files/_parsing.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import array
 import sys
 from typing import Tuple, Union, cast
@@ -80,19 +82,19 @@ def _keyword_entry_of(
 
 def _unpack_binary_field(
     tks: ParseResults,
-) -> Sequence[Union[Sequence[float], Sequence[Sequence[float]]]]:
+) -> Sequence[Sequence[float] | Sequence[Sequence[float]]]:
     elsize = len(tks[0]) // 8
 
     arr = array.array("d", "".join(tks).encode("latin-1"))
 
-    all: Union[Sequence[float], Sequence[Sequence[float]]]
+    values: Sequence[float] | Sequence[Sequence[float]]
 
     if elsize != 1:
-        all = [arr[i : i + elsize].tolist() for i in range(0, len(arr), elsize)]
+        values = [arr[i : i + elsize].tolist() for i in range(0, len(arr), elsize)]
     else:
-        all = arr.tolist()
+        values = arr.tolist()
 
-    return [all]
+    return [values]
 
 
 _IDENTCHARS = identchars + "$"
@@ -207,28 +209,29 @@ _FILE = (
 )
 
 
-class Parsed(Mapping[Tuple[str, ...], Union[FoamFileBase.Data, EllipsisType]]):
+class Parsed(Mapping[Tuple[str, ...], Union[FoamFileBase._DataEntry, EllipsisType]]):
     def __init__(self, contents: bytes) -> None:
         self._parsed: MutableMapping[
-            Tuple[str, ...],
-            Tuple[int, Union[FoamFileBase.Data, EllipsisType], int],
+            tuple[str, ...],
+            tuple[int, FoamFileBase._DataEntry | EllipsisType, int],
         ] = {}
-        self._end = len(contents)
-
         for parse_result in _FILE.parse_string(
             contents.decode("latin-1"), parse_all=True
         ):
             self._parsed.update(self._flatten_result(parse_result))
 
+        self.contents = contents
+        self.modified = False
+
     @staticmethod
     def _flatten_result(
-        parse_result: ParseResults, *, _keywords: Tuple[str, ...] = ()
+        parse_result: ParseResults, *, _keywords: tuple[str, ...] = ()
     ) -> Mapping[
-        Tuple[str, ...], Tuple[int, Union[FoamFileBase.Data, EllipsisType], int]
+        tuple[str, ...], tuple[int, FoamFileBase._DataEntry | EllipsisType, int]
     ]:
         ret: MutableMapping[
-            Tuple[str, ...],
-            Tuple[int, Union[FoamFileBase.Data, EllipsisType], int],
+            tuple[str, ...],
+            tuple[int, FoamFileBase._DataEntry | EllipsisType, int],
         ] = {}
         start = parse_result.locn_start
         assert isinstance(start, int)
@@ -255,35 +258,75 @@ class Parsed(Mapping[Tuple[str, ...], Union[FoamFileBase.Data, EllipsisType]]):
         return ret
 
     def __getitem__(
-        self, keywords: Union[str, Tuple[str, ...]]
-    ) -> Union[FoamFileBase.Data, EllipsisType]:
-        if isinstance(keywords, str):
-            keywords = (keywords,)
-
+        self, keywords: tuple[str, ...]
+    ) -> FoamFileBase._DataEntry | EllipsisType:
         _, data, _ = self._parsed[keywords]
         return data
 
+    def put(
+        self,
+        keywords: tuple[str, ...],
+        data: FoamFileBase._DataEntry | EllipsisType,
+        content: bytes,
+    ) -> None:
+        start, end = self.entry_location(keywords, missing_ok=True)
+
+        diff = len(content) - (end - start)
+        for k, (s, d, e) in self._parsed.items():
+            if s > end:
+                self._parsed[k] = (s + diff, d, e + diff)
+            elif e > start:
+                self._parsed[k] = (s, d, e + diff)
+
+        self._parsed[keywords] = (start, data, end + diff)
+
+        self.contents = self.contents[:start] + content + self.contents[end:]
+        self.modified = True
+
+        for k in list(self._parsed):
+            if keywords != k and keywords == k[: len(keywords)]:
+                del self._parsed[k]
+
+    def __delitem__(self, keywords: tuple[str, ...]) -> None:
+        start, end = self.entry_location(keywords)
+        del self._parsed[keywords]
+
+        for k in list(self._parsed):
+            if keywords == k[: len(keywords)]:
+                del self._parsed[k]
+
+        diff = end - start
+        for k, (s, d, e) in self._parsed.items():
+            if s > end:
+                self._parsed[k] = (s - diff, d, e - diff)
+            elif e > start:
+                self._parsed[k] = (s, d, e - diff)
+
+        self.contents = self.contents[:start] + self.contents[end:]
+        self.modified = True
+
     def __contains__(self, keywords: object) -> bool:
         return keywords in self._parsed
 
-    def __iter__(self) -> Iterator[Tuple[str, ...]]:
+    def __iter__(self) -> Iterator[tuple[str, ...]]:
         return iter(self._parsed)
 
     def __len__(self) -> int:
         return len(self._parsed)
 
     def entry_location(
-        self, keywords: Tuple[str, ...], *, missing_ok: bool = False
-    ) -> Tuple[int, int]:
+        self, keywords: tuple[str, ...], *, missing_ok: bool = False
+    ) -> tuple[int, int]:
         try:
             start, _, end = self._parsed[keywords]
         except KeyError:
             if missing_ok:
                 if len(keywords) > 1:
-                    _, _, end = self._parsed[keywords[:-1]]
-                    end -= 1
+                    assert self[keywords[:-1]] is ...
+                    start, end = self.entry_location(keywords[:-1])
+                    end = self.contents.rindex(b"}", start, end)
                 else:
-                    end = self._end
+                    end = len(self.contents)
 
                 start = end
             else:

foamlib/_files/_serialization.py

@@ -1,12 +1,15 @@
+from __future__ import annotations
+
 import array
 import itertools
 import sys
 from enum import Enum, auto
+from typing import cast
 
 if sys.version_info >= (3, 9):
-    from collections.abc import Mapping
+    from collections.abc import Mapping, Sequence
 else:
-    from typing import Mapping
+    from typing import Mapping, Sequence
 
 from ._base import FoamFileBase
 from ._util import is_sequence
@@ -28,7 +31,7 @@ class Kind(Enum):
 
 
 def dumps(
-    data: FoamFileBase._SetData,
+    data: FoamFileBase.Data,
     *,
     kind: Kind = Kind.DEFAULT,
 ) -> bytes:
@@ -53,7 +56,7 @@ def dumps(
     ):
         return b"[" + b" ".join(dumps(v) for v in data) + b"]"
 
-    if (kind == Kind.FIELD or kind == Kind.BINARY_FIELD) and (
+    if kind in (Kind.FIELD, Kind.BINARY_FIELD) and (
         isinstance(data, (int, float))
         or is_sequence(data)
         and data
@@ -62,22 +65,27 @@ def dumps(
     ):
         return b"uniform " + dumps(data, kind=Kind.SINGLE_ENTRY)
 
-    if (kind == Kind.FIELD or kind == Kind.BINARY_FIELD) and is_sequence(data):
-        if isinstance(data[0], (int, float)):
+    if kind in (Kind.FIELD, Kind.BINARY_FIELD) and is_sequence(data):
+        if data and isinstance(data[0], (int, float)):
             tensor_kind = b"scalar"
-        elif len(data[0]) == 3:
-            tensor_kind = b"vector"
-        elif len(data[0]) == 6:
-            tensor_kind = b"symmTensor"
-        elif len(data[0]) == 9:
-            tensor_kind = b"tensor"
+        elif is_sequence(data[0]) and data[0] and isinstance(data[0][0], (int, float)):
+            if len(data[0]) == 3:
+                tensor_kind = b"vector"
+            elif len(data[0]) == 6:
+                tensor_kind = b"symmTensor"
+            elif len(data[0]) == 9:
+                tensor_kind = b"tensor"
+            else:
+                return dumps(data)
         else:
             return dumps(data)
 
         if kind == Kind.BINARY_FIELD:
             if tensor_kind == b"scalar":
+                data = cast(Sequence[float], data)
                 contents = b"(" + array.array("d", data).tobytes() + b")"
             else:
+                data = cast(Sequence[Sequence[float]], data)
                 contents = (
                     b"("
                     + array.array("d", itertools.chain.from_iterable(data)).tobytes()

foamlib/_files/_util.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 import sys
-from typing import Any
+from typing import TYPE_CHECKING
 
 if sys.version_info >= (3, 9):
     from collections.abc import Sequence
@@ -11,8 +13,11 @@ if sys.version_info >= (3, 10):
 else:
     from typing_extensions import TypeGuard
 
+if TYPE_CHECKING:
+    from ._base import FoamFileBase
+
 
 def is_sequence(
-    value: Any,
-) -> TypeGuard[Sequence[Any]]:
+    value: FoamFileBase.Data,
+) -> TypeGuard[Sequence[FoamFileBase.Data]]:
     return isinstance(value, Sequence) and not isinstance(value, str)

{foamlib-0.6.10.dist-info → foamlib-0.6.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foamlib
-Version: 0.6.10
+Version: 0.6.12
 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

foamlib-0.6.12.dist-info/RECORD

@@ -0,0 +1,22 @@
+foamlib/__init__.py,sha256=fzKDNF8_CX5rU1KVBoVQZHdjJ_nEiRbLv_sfj6-CHAY,487
+foamlib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+foamlib/_cases/__init__.py,sha256=wTUHcUgU1CBgpu0cUMtksQ5VKG6B8CFu9xc3dWwsQuo,358
+foamlib/_cases/_async.py,sha256=i6g4EBHqvI-1PkdrxsRto2ynW7sxsOga2bSYk1XVG1U,7795
+foamlib/_cases/_base.py,sha256=FKfZxP7HspWfSswQ6yZ5bGJRdZUlupQxj2tDqngXPmc,6785
+foamlib/_cases/_run.py,sha256=lveqKZium_qK_eTxYE8jOjwx0eiIoolCBbi56-zLw1o,14420
+foamlib/_cases/_slurm.py,sha256=kj4wqgr3foMyAoUkoHOZODRBmVqH1B9KqAIEEjM8ZBg,2328
+foamlib/_cases/_subprocess.py,sha256=6BlBRxknj2-BFcGkx7oVcuL63_utSaY1Axmsc1qV9j8,3887
+foamlib/_cases/_sync.py,sha256=2BJXB7Nzldb4OgPukqupgYqdceUGkI2mYhhtGPWEBrc,5901
+foamlib/_cases/_util.py,sha256=lhVca3ERY0zwYjDam6W2QMROt0yX5vAF-9_DS5RuMbM,1547
+foamlib/_files/__init__.py,sha256=-UqB9YTH6mrJfXCX00kPTAAY20XG64u1MGPw_1ewLVs,148
+foamlib/_files/_base.py,sha256=ZY_6Zxr3LZHVzJEex6SUTi9Pgo7Oi7i3Jo-wre9G3yE,1968
+foamlib/_files/_files.py,sha256=WF_FWiDhnkA4dtZ6zR1A0iGrv76eI3SetlQu6Zl6HY8,14949
+foamlib/_files/_io.py,sha256=IQLqoqnA1TpHf21NbUho2wsYWevyqC6MKo-wfpaObUU,2226
+foamlib/_files/_parsing.py,sha256=qLZx_VZMNEoyfmrkC_r85QBuABa26-6Y0fHnxCq74Q8,10880
+foamlib/_files/_serialization.py,sha256=HN9XCDbZ0I-b_UdpEYgDVNX6U4tPd6u-uYBKOoX3uqg,3681
+foamlib/_files/_util.py,sha256=VXUTD0B3NbnlaE_KCqWzrFcWvBz_2JfnIWpNp3snX3Y,526
+foamlib-0.6.12.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
+foamlib-0.6.12.dist-info/METADATA,sha256=fN1pEtQzMHd-yuzMkugYgTtpVwZwV-U0CPxU6rRTnzA,7742
+foamlib-0.6.12.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+foamlib-0.6.12.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
+foamlib-0.6.12.dist-info/RECORD,,

{foamlib-0.6.10.dist-info → foamlib-0.6.12.dist-info}/WHEEL

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

foamlib-0.6.10.dist-info/RECORD

@@ -1,22 +0,0 @@
-foamlib/__init__.py,sha256=7AIRMyiByWRuEUEkR_MGE1zy0GuXaEzOlzXnUkCvdtw,487
-foamlib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-foamlib/_cases/__init__.py,sha256=wTUHcUgU1CBgpu0cUMtksQ5VKG6B8CFu9xc3dWwsQuo,358
-foamlib/_cases/_async.py,sha256=oQ2XnD9sansJLJDuUl3_vHaUM4L9hq2G97IcRjQAOPY,7920
-foamlib/_cases/_base.py,sha256=HV7ifv60pAhgZdC3s5IxvdrDQ2vvlhzPwyB-80QQLe8,6733
-foamlib/_cases/_run.py,sha256=XFDBr5G1hJo0402RcZClLL_rv0d5jVgq2be25FgZcyQ,14725
-foamlib/_cases/_slurm.py,sha256=pUXcFSrwOK5URsoooiUOJg61X0WLLNHyv7bhUxIb7Bs,2346
-foamlib/_cases/_subprocess.py,sha256=tzsBGe44EnQw3GWqRW1ktEZ4sNQ1sSHdJ_89zIfF5L0,3952
-foamlib/_cases/_sync.py,sha256=ENvn8zseRh6YOjAgqXHzmKhKFYXdzGIW5NoMHS_NKPY,6049
-foamlib/_cases/_util.py,sha256=GNndpqw3Jg_S-Hxzl5vwRgD0czcTNb9NYHMhcfBoMBg,1493
-foamlib/_files/__init__.py,sha256=-UqB9YTH6mrJfXCX00kPTAAY20XG64u1MGPw_1ewLVs,148
-foamlib/_files/_base.py,sha256=mrZuk90sdIrSGiOeEtCcEHoz6RK4qUc6GFoDSuo7U3w,1920
-foamlib/_files/_files.py,sha256=6ZbUxb2VF225ceEHBg4L23oxHvD7M18QC9zqMPm0ZS4,16025
-foamlib/_files/_io.py,sha256=ArD9lgMw-Kjj-wl_5qU6eRdW4lZWyGBIEMWLCNV4a0Y,2467
-foamlib/_files/_parsing.py,sha256=SvPAbNb_xJwOBRGii4oLOUvhj9OdJ8auMIzdx77AYwI,9394
-foamlib/_files/_serialization.py,sha256=iIiPUbgoQzg6xeYeeTsRPWKWqrYFtAQPgglBbylPkms,3341
-foamlib/_files/_util.py,sha256=UMzXmTFgvbp46w6k3oEZJoYC98pFgEK6LN5uLOwrlCg,397
-foamlib-0.6.10.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
-foamlib-0.6.10.dist-info/METADATA,sha256=0IYRom9JRsqS6OE6x3rlRDdL6zYUubtpYl5hafW_2Bg,7742
-foamlib-0.6.10.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-foamlib-0.6.10.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
-foamlib-0.6.10.dist-info/RECORD,,