foamlib 0.3.17__py3-none-any.whl → 0.3.18__py3-none-any.whl

foamlib/__init__.py

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

foamlib/_files/_files.py

@@ -6,6 +6,7 @@ if sys.version_info >= (3, 9):
 else:
     from typing import Iterator, Mapping, MutableMapping, Sequence
 
+from .._util import is_sequence
 from ._base import FoamDict
 from ._io import FoamFileIO
 from ._serialization import Kind, dumpb
@@ -24,7 +25,7 @@ class FoamFile(
     FoamFileIO,
 ):
     """
-    An OpenFOAM dictionary file.
+    An OpenFOAM data file.
 
     Use as a mutable mapping (i.e., like a dict) to access and modify entries.
 
@@ -88,6 +89,74 @@ class FoamFile(
 
             return ret
 
+    class Header(SubDict):
+        """The header of an OpenFOAM file."""
+
+        @property
+        def version(self) -> float:
+            """Alias of `self["version"]`."""
+            ret = self["version"]
+            if not isinstance(ret, float):
+                raise TypeError("version is not a float")
+            return ret
+
+        @version.setter
+        def version(self, data: float) -> None:
+            self["version"] = data
+
+        @property
+        def format(self) -> str:
+            """Alias of `self["format"]`."""
+            ret = self["format"]
+            if not isinstance(ret, str):
+                raise TypeError("format is not a string")
+            return ret
+
+        @format.setter
+        def format(self, data: str) -> None:
+            self["format"] = data
+
+        @property
+        def class_(self) -> str:
+            """Alias of `self["class"]`."""
+            ret = self["class"]
+            if not isinstance(ret, str):
+                raise TypeError("class is not a string")
+            return ret
+
+        @class_.setter
+        def class_(self, data: str) -> None:
+            self["class"] = data
+
+        @property
+        def location(self) -> str:
+            """Alias of `self["location"]`."""
+            ret = self["location"]
+            if not isinstance(ret, str):
+                raise TypeError("location is not a string")
+            return ret
+
+        @location.setter
+        def location(self, data: str) -> None:
+            self["location"] = data
+
+        @property
+        def object(self) -> str:
+            """Alias of `self["object"]`."""
+            ret = self["object"]
+            if not isinstance(ret, str):
+                raise TypeError("object is not a string")
+            return ret
+
+    @property
+    def header(self) -> "FoamFile.Header":
+        """Alias of `self["FoamFile"]`."""
+        ret = self["FoamFile"]
+        if not isinstance(ret, FoamFile.Header):
+            assert not isinstance(ret, FoamFile.SubDict)
+            raise TypeError("FoamFile is not a dictionary")
+        return ret
+
     def __getitem__(
         self, keywords: Union[str, Tuple[str, ...]]
     ) -> Union["FoamFile.Data", "FoamFile.SubDict"]:
@@ -99,7 +168,10 @@ class FoamFile(
         value = parsed[keywords]
 
         if value is ...:
-            return FoamFile.SubDict(self, keywords)
+            if keywords == ("FoamFile",):
+                return FoamFile.Header(self, keywords)
+            else:
+                return FoamFile.SubDict(self, keywords)
         else:
             return value
 
@@ -115,23 +187,23 @@ class FoamFile(
         assume_field: bool = False,
         assume_dimensions: bool = False,
     ) -> None:
-        if not isinstance(keywords, tuple):
-            keywords = (keywords,)
-
-        kind = Kind.DEFAULT
-        if keywords == ("internalField",) or (
-            len(keywords) == 3
-            and keywords[0] == "boundaryField"
-            and keywords[2] == "value"
-        ):
-            kind = Kind.BINARY_FIELD if self._binary else Kind.FIELD
-        elif keywords == ("dimensions",):
-            kind = Kind.DIMENSIONS
-
-        contents, parsed = self._read()
-
-        if isinstance(data, Mapping):
-            with self:
+        with self:
+            if not isinstance(keywords, tuple):
+                keywords = (keywords,)
+
+            kind = Kind.DEFAULT
+            if keywords == ("internalField",) or (
+                len(keywords) == 3
+                and keywords[0] == "boundaryField"
+                and keywords[2] == "value"
+            ):
+                kind = Kind.BINARY_FIELD if self._binary else Kind.FIELD
+            elif keywords == ("dimensions",):
+                kind = Kind.DIMENSIONS
+
+            contents, parsed = self._read()
+
+            if isinstance(data, Mapping):
                 if isinstance(data, FoamDict):
                     data = data.as_dict()
 
@@ -147,16 +219,44 @@ class FoamFile(
 
                 for k, v in data.items():
                     self[(*keywords, k)] = v
-        else:
-            start, end = parsed.entry_location(keywords, missing_ok=True)
-
-            self._write(
-                contents[:start]
-                + b"\n"
-                + dumpb({keywords[-1]: data}, kind=kind)
-                + b"\n"
-                + contents[end:]
-            )
+
+            elif not self and keywords[0] != "FoamFile":
+                self["FoamFile"] = {
+                    "version": 2.0,
+                    "format": "ascii",
+                    "class": "dictionary",
+                    "location": f'"{self.path.parent.name}"',
+                    "object": self.path.name,
+                }
+                self[keywords] = data
+
+            elif (
+                kind == Kind.FIELD or kind == Kind.BINARY_FIELD
+            ) and self.header.class_ == "dictionary":
+                if not is_sequence(data):
+                    class_ = "volScalarField"
+                elif (len(data) == 3 and not is_sequence(data[0])) or len(data[0]) == 3:
+                    class_ = "volVectorField"
+                elif (len(data) == 6 and not is_sequence(data[0])) or len(data[0]) == 6:
+                    class_ = "volSymmTensorField"
+                elif (len(data) == 9 and not is_sequence(data[0])) or len(data[0]) == 9:
+                    class_ = "volTensorField"
+                else:
+                    class_ = "volScalarField"
+
+                self.header.class_ = class_
+                self[keywords] = data
+
+            else:
+                start, end = parsed.entry_location(keywords, missing_ok=True)
+
+                self._write(
+                    contents[:start]
+                    + b"\n"
+                    + dumpb({keywords[-1]: data}, kind=kind)
+                    + b"\n"
+                    + contents[end:]
+                )
 
     def __delitem__(self, keywords: Union[str, Tuple[str, ...]]) -> None:
         if not isinstance(keywords, tuple):

{foamlib-0.3.17.dist-info → foamlib-0.3.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foamlib
-Version: 0.3.17
+Version: 0.3.18
 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.3.17.dist-info → foamlib-0.3.18.dist-info}/RECORD

@@ -1,4 +1,4 @@
-foamlib/__init__.py,sha256=XJCVvMns0YE5_ztycf50szXPcPfnKIW3Upxs-ODwx04,439
+foamlib/__init__.py,sha256=0wd3tVi_qECfQD1xCAEvBlUtrcKDZesuwDO_Wz4OzEs,439
 foamlib/_util.py,sha256=UMzXmTFgvbp46w6k3oEZJoYC98pFgEK6LN5uLOwrlCg,397
 foamlib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 foamlib/_cases/__init__.py,sha256=xnQpR64EvFtCh07qnSz7kjQ1IhJXjRDwaZWwQGZhGv4,280
@@ -8,12 +8,12 @@ foamlib/_cases/_sync.py,sha256=CloQgd-93jxfSXIt7J5NxcAu3N_iF3eXMKO-NfNOgi4,4522
 foamlib/_cases/_util.py,sha256=v6sHxHCEgagsVuup0S1xJW-x9py5xj3bUye8PiFfb3o,925
 foamlib/_files/__init__.py,sha256=vDkPj8u8bX_I_m2YfeKvXBgwg8D1ufyFCfHGHKN3JPQ,140
 foamlib/_files/_base.py,sha256=YA5a-i5HZuA3JslCD6r-DwZzpSA8r42dqSXef286Ako,2050
-foamlib/_files/_files.py,sha256=4rZb3HMJADmJBGmRpChiujJNX1UIxhgHI5YPEmEOWvE,10784
+foamlib/_files/_files.py,sha256=14cgBYM-U8sgbm5ymkf2PLS7zLe5b2UBaost6LjBRuM,14262
 foamlib/_files/_io.py,sha256=hlcqQqU-1cdIbDc3YqxnMfxALo4SFAEcRIoZM2vMtnE,2083
 foamlib/_files/_parsing.py,sha256=-IgvCOFMAERNOrpw2NaHtVXubBg4Ey6xR55bzEZb2B4,7696
 foamlib/_files/_serialization.py,sha256=LCeaLWtNvkcs0dfowL7nViiByxw7U_fvgueVjFliipU,3462
-foamlib-0.3.17.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
-foamlib-0.3.17.dist-info/METADATA,sha256=aLs74nCqbOMv3PnD34XF32-OMa20FHY8IS92wDYP32o,5458
-foamlib-0.3.17.dist-info/WHEEL,sha256=nCVcAvsfA9TDtwGwhYaRrlPhTLV9m-Ga6mdyDtuwK18,91
-foamlib-0.3.17.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
-foamlib-0.3.17.dist-info/RECORD,,
+foamlib-0.3.18.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
+foamlib-0.3.18.dist-info/METADATA,sha256=UCvPiUdXGGTf2hKsl6l_mAoQkP4tOkfuwZIJfOlbi9M,5458
+foamlib-0.3.18.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
+foamlib-0.3.18.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
+foamlib-0.3.18.dist-info/RECORD,,

{foamlib-0.3.17.dist-info → foamlib-0.3.18.dist-info}/WHEEL

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