fix for darcy2d

Author: HydrogenSulfate

examples/darcy/darcy2d_v2.py

@@ -150,18 +150,9 @@ def poisson_ref_compute_func(_in):
     validator = {residual_validator.name: residual_validator}
 
     # set visualizer(optional)
-    NPOINT_BC = NPOINT_TOP + NPOINT_BOTTOM + NPOINT_LEFT + NPOINT_RIGHT
-    vis_interior_points = geom["rect"].sample_interior(NPOINT_PDE, evenly=True)
-    vis_boundary_points = geom["rect"].sample_boundary(NPOINT_BC, evenly=True)
-
     # manually collate input data for visualization,
-    # interior+boundary
-    vis_points = {}
-    for key in vis_interior_points:
-        vis_points[key] = np.concatenate(
-            (vis_interior_points[key], vis_boundary_points[key])
-        )
-
+    NPOINT_BC = NPOINT_TOP + NPOINT_BOTTOM + NPOINT_LEFT + NPOINT_RIGHT
+    vis_points = geom["rect"].sample_interior(NPOINT_PDE + NPOINT_BC, evenly=True)
     visualizer = {
         "visulzie_u_v": ppsci.visualize.VisualizerVtu(
             vis_points,
@@ -194,6 +185,7 @@ def poisson_ref_compute_func(_in):
     solver.visualize()
 
     # directly evaluate pretrained model(optional)
+    logger.init_logger("ppsci", f"{output_dir}/eval.log", "info")
     solver = ppsci.solver.Solver(
         model,
         constraint,

ppsci/equation/__init__.py

@@ -30,8 +30,8 @@
     "Laplace",
     "NavierStokes",
     "NormalDotVec",
-    "Vibration",
     "Poisson",
+    "Vibration",
     "build_equation",
 ]
 

ppsci/equation/pde/__init__.py

@@ -26,6 +26,6 @@
     "Laplace",
     "NavierStokes",
     "NormalDotVec",
-    "Vibration",
     "Poisson",
+    "Vibration",
 ]

ppsci/equation/pde/poisson.py

@@ -0,0 +1,41 @@
+# Copyright (c) 2023 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.
+
+from ppsci.autodiff import hessian
+from ppsci.equation.pde import base
+
+
+class Poisson(base.PDE):
+    """Poisson
+
+    Args:
+        dim (int): Dimension of equation.
+
+    Examples:
+        >>> import ppsci
+        >>> pde = ppsci.equation.Poisson(2)
+    """
+
+    def __init__(self, dim: int):
+        super().__init__()
+        self.dim = dim
+
+        def poisson_compute_func(out):
+            invars = ("x", "y", "z")[: self.dim]
+            poisson = 0
+            for invar in invars:
+                poisson += hessian(out["p"], out[invar])
+            return poisson
+
+        self.add_equation("poisson", poisson_compute_func)