feat: update IntG to expr_dataclass

pytential/linalg/direct_solver_symbolic.py

@@ -102,10 +102,12 @@ def map_int_g(self, expr):
             if name not in source_args
         }
 
-        return expr.copy(target_kernel=target_kernel,
-                         source_kernels=source_kernels,
-                         densities=self.rec(expr.densities),
-                         kernel_arguments=kernel_arguments)
+        from dataclasses import replace
+        return replace(expr,
+                       target_kernel=target_kernel,
+                       source_kernels=source_kernels,
+                       densities=self.rec(expr.densities),
+                       kernel_arguments=kernel_arguments)
 
 # }}}
 

pytential/symbolic/mappers.py

@@ -20,6 +20,7 @@
 THE SOFTWARE.
 """
 
+from dataclasses import replace
 from functools import reduce
 
 from pymbolic.mapper.stringifier import (
@@ -129,9 +130,7 @@ def map_int_g(self, expr):
         if not changed:
             return expr
 
-        return expr.copy(
-                densities=densities,
-                kernel_arguments=kernel_arguments)
+        return replace(expr, densities=densities, kernel_arguments=kernel_arguments)
 
     def map_interpolation(self, expr):
         operand = self.rec(expr.operand)
@@ -261,10 +260,7 @@ def map_int_g(self, expr):
         if not changed:
             return expr
 
-        return expr.copy(
-                densities=densities,
-                kernel_arguments=kernel_arguments,
-                )
+        return replace(expr, densities=densities, kernel_arguments=kernel_arguments)
 
     def map_common_subexpression(self, expr):
         child = self.rec(expr.child)
@@ -522,9 +518,9 @@ def map_product(self, expr):
 
     def map_int_g(self, expr):
         from sumpy.kernel import AxisTargetDerivative
-        return expr.copy(
-                target_kernel=AxisTargetDerivative(
-                    self.ambient_axis, expr.target_kernel))
+
+        target_kernel = AxisTargetDerivative(self.ambient_axis, expr.target_kernel)
+        return replace(expr, target_kernel=target_kernel)
 
 
 class DerivativeSourceAndNablaComponentCollector(
@@ -570,15 +566,15 @@ def map_int_g(self, expr):
             raise ValueError(
                     "Unregularized evaluation does not support one-sided limits")
 
-        expr = expr.copy(
-                qbx_forced_limit=None,
-                densities=self.rec(expr.densities),
-                kernel_arguments={
-                    name: self.rec(arg_expr)
-                    for name, arg_expr in expr.kernel_arguments.items()
-                    })
-
-        return expr
+        return replace(
+            expr,
+            qbx_forced_limit=None,
+            densities=self.rec(expr.densities),
+            kernel_arguments={
+                name: self.rec(arg_expr)
+                for name, arg_expr in expr.kernel_arguments.items()
+            }
+        )
 
 # }}}
 
@@ -626,7 +622,7 @@ def map_num_reference_derivative(self, expr):
 
     def map_int_g(self, expr):
         if expr.target.discr_stage is None:
-            expr = expr.copy(target=expr.target.to_stage1())
+            expr = replace(expr, target=expr.target.to_stage1())
 
         if expr.source.discr_stage is not None:
             return expr
@@ -638,16 +634,18 @@ def map_int_g(self, expr):
 
         from_dd = expr.source.to_stage1()
         to_dd = from_dd.to_quad_stage2()
-        densities = [prim.interp(from_dd, to_dd, self.rec(density)) for
-            density in expr.densities]
+        densities = tuple(
+            prim.interp(from_dd, to_dd, self.rec(density)) for
+            density in expr.densities)
 
         from_dd = from_dd.copy(discr_stage=self.from_discr_stage)
         kernel_arguments = {
                 name: prim.interp(from_dd, to_dd,
                     self.rec(self.tagger(arg_expr)))
                 for name, arg_expr in expr.kernel_arguments.items()}
 
-        return expr.copy(
+        return replace(
+                expr,
                 densities=densities,
                 kernel_arguments=kernel_arguments,
                 source=to_dd)
@@ -678,7 +676,8 @@ def map_int_g(self, expr):
 
         is_self = source_discr is target_discr
 
-        expr = expr.copy(
+        expr = replace(
+                expr,
                 densities=self.rec(expr.densities),
                 kernel_arguments={
                     name: self.rec(arg_expr)
@@ -707,8 +706,8 @@ def map_int_g(self, expr):
 
         if expr.qbx_forced_limit == "avg":
             return 0.5*(
-                    expr.copy(qbx_forced_limit=+1)
-                    + expr.copy(qbx_forced_limit=-1))
+                    replace(expr, qbx_forced_limit=+1)
+                    + replace(expr, qbx_forced_limit=-1))
         else:
             return expr