Python FEniCS/ufl@b9af69f

Author: FEniCS GitHub Actions

ufl/main/_modules/ufl/action.html

@@ -380,6 +380,9 @@ <h1>Source code for ufl.action</h1><div class="highlight"><pre>
 
     <span class="k">def</span><span class="w"> </span><span class="fm">__new__</span><span class="p">(</span><span class="bp">cls</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kw</span><span class="p">):</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Create a new Action.&quot;&quot;&quot;</span>
+        <span class="kn">from</span><span class="w"> </span><span class="nn">ufl.algorithms.analysis</span><span class="w"> </span><span class="kn">import</span> <span class="n">extract_arguments</span>
+        <span class="kn">from</span><span class="w"> </span><span class="nn">ufl.algorithms.replace</span><span class="w"> </span><span class="kn">import</span> <span class="n">replace</span>
+
         <span class="n">left</span><span class="p">,</span> <span class="n">right</span> <span class="o">=</span> <span class="n">args</span>
 
         <span class="c1"># Check trivial case</span>
@@ -427,6 +430,12 @@ <h1>Source code for ufl.action</h1><div class="highlight"><pre>
             <span class="ow">and</span> <span class="nb">len</span><span class="p">(</span><span class="n">left</span><span class="o">.</span><span class="n">arguments</span><span class="p">())</span> <span class="o">==</span> <span class="mi">1</span>
         <span class="p">):</span>
             <span class="n">v</span><span class="p">,</span> <span class="n">operand</span> <span class="o">=</span> <span class="n">right</span><span class="o">.</span><span class="n">argument_slots</span><span class="p">()</span>
+            <span class="c1"># If the operand has an argument, replace it with number 0</span>
+            <span class="n">operand_args</span> <span class="o">=</span> <span class="n">extract_arguments</span><span class="p">(</span><span class="n">operand</span><span class="p">)</span>
+            <span class="k">if</span> <span class="n">operand_args</span><span class="p">:</span>
+                <span class="p">(</span><span class="n">old_arg</span><span class="p">,)</span> <span class="o">=</span> <span class="n">operand_args</span>
+                <span class="n">new_arg</span> <span class="o">=</span> <span class="nb">type</span><span class="p">(</span><span class="n">old_arg</span><span class="p">)(</span><span class="n">old_arg</span><span class="o">.</span><span class="n">ufl_function_space</span><span class="p">(),</span> <span class="mi">0</span><span class="p">,</span> <span class="n">old_arg</span><span class="o">.</span><span class="n">part</span><span class="p">())</span>
+                <span class="n">operand</span> <span class="o">=</span> <span class="n">replace</span><span class="p">(</span><span class="n">operand</span><span class="p">,</span> <span class="p">{</span><span class="n">old_arg</span><span class="p">:</span> <span class="n">new_arg</span><span class="p">})</span>
             <span class="k">if</span> <span class="n">v</span> <span class="o">==</span> <span class="n">right</span><span class="o">.</span><span class="n">arguments</span><span class="p">()[</span><span class="mi">0</span><span class="p">]:</span>
                 <span class="k">return</span> <span class="n">right</span><span class="o">.</span><span class="n">_ufl_expr_reconstruct_</span><span class="p">(</span><span class="n">operand</span><span class="p">,</span> <span class="n">v</span><span class="o">=</span><span class="n">left</span><span class="p">)</span>