Line data    Source code

       1             : // The libMesh Finite Element Library.
       2             : // Copyright (C) 2002-2025 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner
       3             : 
       4             : // This library is free software; you can redistribute it and/or
       5             : // modify it under the terms of the GNU Lesser General Public
       6             : // License as published by the Free Software Foundation; either
       7             : // version 2.1 of the License, or (at your option) any later version.
       8             : 
       9             : // This library is distributed in the hope that it will be useful,
      10             : // but WITHOUT ANY WARRANTY; without even the implied warranty of
      11             : // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      12             : // Lesser General Public License for more details.
      13             : 
      14             : // You should have received a copy of the GNU Lesser General Public
      15             : // License along with this library; if not, write to the Free Software
      16             : // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
      17             : 
      18             : 
      19             : 
      20             : #include "libmesh/meshfunction_solution_transfer.h"
      21             : 
      22             : #include "libmesh/system.h"
      23             : #include "libmesh/numeric_vector.h"
      24             : #include "libmesh/mesh_function.h"
      25             : #include "libmesh/node.h"
      26             : #include "libmesh/variable.h"
      27             : 
      28             : namespace libMesh
      29             : {
      30             : 
      31           0 : MeshFunctionSolutionTransfer::MeshFunctionSolutionTransfer(const libMesh::Parallel::Communicator & comm_in) :
      32           0 :   SolutionTransfer(comm_in)
      33           0 : {}
      34             : 
      35           0 : MeshFunctionSolutionTransfer::~MeshFunctionSolutionTransfer() = default;
      36             : 
      37             : void
      38           0 : MeshFunctionSolutionTransfer::transfer(const Variable & from_var,
      39             :                                        const Variable & to_var)
      40             : {
      41             :   // This only works when transferring to a Lagrange variable
      42           0 :   libmesh_assert(to_var.type().family == LAGRANGE);
      43             : 
      44           0 :   unsigned int to_var_num = to_var.number();
      45             : 
      46           0 :   System * from_sys = from_var.system();
      47           0 :   System * to_sys = to_var.system();
      48             : 
      49             :   // Only works with a serialized mesh to transfer from!
      50           0 :   libmesh_assert(from_sys->get_mesh().is_serial());
      51             : 
      52           0 :   unsigned int to_sys_num = to_sys->number();
      53             : 
      54           0 :   EquationSystems & from_es = from_sys->get_equation_systems();
      55             : 
      56             :   //Create a serialized version of the solution vector
      57           0 :   std::unique_ptr<NumericVector<Number>> serialized_solution = NumericVector<Number>::build(from_sys->get_mesh().comm());
      58           0 :   serialized_solution->init(from_sys->n_dofs(), false, SERIAL);
      59             : 
      60             :   // Need to pull down a full copy of this vector on every processor
      61             :   // so we can get values in parallel
      62           0 :   from_sys->solution->localize(*serialized_solution);
      63             : 
      64           0 :   MeshFunction from_func(from_es, *serialized_solution, from_sys->get_dof_map(), to_var_num);
      65           0 :   from_func.init();
      66             : 
      67             :   // Now loop over the nodes of the 'To' mesh setting values for each variable.
      68           0 :   for (const auto & node : to_sys->get_mesh().local_node_ptr_range())
      69           0 :     to_sys->solution->set(node->dof_number(to_sys_num, to_var_num, 0), from_func(*node)); // 0 is for the value component
      70             : 
      71           0 :   to_sys->solution->close();
      72           0 :   to_sys->update();
      73           0 : }
      74             : 
      75             : } // namespace libMesh