[API] Support sdf weight for initial constraint (#1184)

* support sdf weight for initial constraint

* add field check

Author: HydrogenSulfate

ppsci/constraint/initial_constraint.py

@@ -139,7 +139,16 @@ def __init__(
         if weight_dict is not None:
             weight = {key: np.ones_like(next(iter(label.values()))) for key in label}
             for key, value in weight_dict.items():
-                if isinstance(value, (int, float)):
+                if isinstance(value, str):
+                    if value == "sdf":
+                        if "sdf" not in input:
+                            raise ValueError(
+                                f"Missing 'sdf' field in input. Please check whether the geometry ({geom.geometry.__class__.__name__}) implements 'sdf_func'"
+                            )
+                        weight[key] = input["sdf"]
+                    else:
+                        raise NotImplementedError(f"string {value} is invalid yet.")
+                elif isinstance(value, (int, float)):
                     weight[key] = np.full_like(next(iter(label.values())), value)
                 elif isinstance(value, sympy.Basic):
                     func = sympy.lambdify(

ppsci/constraint/interior_constraint.py

@@ -138,6 +138,10 @@ def __init__(
             for key, value in weight_dict.items():
                 if isinstance(value, str):
                     if value == "sdf":
+                        if "sdf" not in input:
+                            raise ValueError(
+                                f"Missing 'sdf' field in input. Please check whether the geometry ({geom.__class__.__name__}) implements 'sdf_func'"
+                            )
                         weight[key] = input["sdf"]
                     else:
                         raise NotImplementedError(f"string {value} is invalid yet.")

test/constraint/test_initialconstraint.py

@@ -0,0 +1,46 @@
+# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+#     http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pytest
+
+from ppsci import constraint
+from ppsci import geometry
+from ppsci import loss
+
+
+def test_sdf_weight_of_initialconstraint():
+    rect = geometry.TimeXGeometry(
+        geometry.TimeDomain(0, 1),
+        geometry.Rectangle((0, 0), (1, 1)),
+    )
+    ic = constraint.InitialConstraint(
+        {"u": lambda out: out["u"]},
+        {"u": 0},
+        rect,
+        {
+            "dataset": "IterableNamedArrayDataset",
+            "iters_per_epoch": 1,
+            "batch_size": 16,
+        },
+        loss.MSELoss("mean"),
+        weight_dict={"u": "sdf"},
+        name="IC",
+    )  # doctest: +SKIP
+
+    input, _, _ = next(iter(ic.data_iter))
+    assert "sdf" in input.keys()
+
+
+if __name__ == "__main__":
+    pytest.main()