AllLocalDofIndicesThread.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#include "AllLocalDofIndicesThread.h"

#include "FEProblem.h"
#include "ParallelUniqueId.h"
#include "NonlinearSystemBase.h"
#include "MooseVariableFE.h"

#include "libmesh/dof_map.h"
#include "libmesh/threads.h"
#include "libmesh/system.h"

#include "timpi/communicator.h"

#include LIBMESH_INCLUDE_UNORDERED_SET
LIBMESH_DEFINE_HASH_POINTERS

AllLocalDofIndicesThread::AllLocalDofIndicesThread(SubProblem & problem,
                                                   std::vector<std::string> vars,
                                                   bool include_semilocal)
  : ParallelObject(problem.comm()),
    _problem(problem),
    _sys(nullptr),
    _include_semilocal(include_semilocal)
{
  for (unsigned int i = 0; i < vars.size(); i++)
  {
    auto & var = _problem.getVariable(0, vars[i]);
    if (_sys)
    {
      if (_sys != &var.sys().system())
        mooseError("Variables passed in AllLocalDofIndicesThread must be all in the same system.");
    }
    else
      _sys = &var.sys().system();

    if (var.isArray())
    {
      const auto & array_var = _problem.getArrayVariable(0, vars[i]);
      for (unsigned int p = 0; p < var.count(); ++p)
        _var_numbers.push_back(_sys->variable_number(array_var.componentName(p)));
    }
    else
      _var_numbers.push_back(_sys->variable_number(vars[i]));
  }
}

// Splitting Constructor
AllLocalDofIndicesThread::AllLocalDofIndicesThread(AllLocalDofIndicesThread & x,
                                                   Threads::split /*split*/)
  : ParallelObject(x._problem.comm()),
    _problem(x._problem),
    _sys(x._sys),
    _var_numbers(x._var_numbers),
    _include_semilocal(x._include_semilocal)
{
}

void
AllLocalDofIndicesThread::operator()(const ConstElemRange & range)
{
  ParallelUniqueId puid;
  _tid = puid.id;

  mooseAssert(_sys, "We should have a system, did you forget to specify any variable in vars?");
  auto & dof_map = _sys->get_dof_map();

  for (const auto & elem : range)
  {
    std::vector<dof_id_type> dof_indices;

    dof_id_type local_dof_begin = dof_map.first_dof();
    dof_id_type local_dof_end = dof_map.end_dof();

    // prepare variables
    for (unsigned int i = 0; i < _var_numbers.size(); i++)
    {
      dof_map.dof_indices(elem, dof_indices, _var_numbers[i]);
      for (unsigned int j = 0; j < dof_indices.size(); j++)
      {
        dof_id_type dof = dof_indices[j];

        if (_include_semilocal || (dof >= local_dof_begin && dof < local_dof_end))
          _all_dof_indices.insert(dof);
      }
    }
  }
}

void
AllLocalDofIndicesThread::join(const AllLocalDofIndicesThread & y)
{
  _all_dof_indices.insert(y._all_dof_indices.begin(), y._all_dof_indices.end());
}

void
AllLocalDofIndicesThread::dofIndicesSetUnion()
{
  _communicator.set_union(_all_dof_indices);
}