AllLocalDofIndicesThread Class Reference

Grab all the (possibly semi)local dof indices for the variables passed in, in the system passed in. More...

#include <AllLocalDofIndicesThread.h>

Inheritance diagram for AllLocalDofIndicesThread:

Public Member Functions
	AllLocalDofIndicesThread (SubProblem &problem, std::vector< std::string > vars, bool include_semilocal=false)
	AllLocalDofIndicesThread (AllLocalDofIndicesThread &x, libMesh::Threads::split split)
void	operator() (const libMesh::ConstElemRange &range)
void	join (const AllLocalDofIndicesThread &y)
const std::set< dof_id_type > &	getDofIndices () const
void	dofIndicesSetUnion ()
const Parallel::Communicator &	comm () const
processor_id_type	n_processors () const
processor_id_type	processor_id () const

Protected Attributes
SubProblem &	_problem
libMesh::System *	_sys
std::vector< unsigned int >	_var_numbers
const bool	_include_semilocal
	Whether to include semilocal dof indices. More...
THREAD_ID	_tid
std::set< dof_id_type >	_all_dof_indices
const Parallel::Communicator &	_communicator

Detailed Description

Grab all the (possibly semi)local dof indices for the variables passed in, in the system passed in.

Definition at line 30 of file AllLocalDofIndicesThread.h.

Constructor & Destructor Documentation

AllLocalDofIndicesThread() [1/2]

LIBMESH_DEFINE_HASH_POINTERS AllLocalDofIndicesThread::AllLocalDofIndicesThread	(	SubProblem &	problem,
		std::vector< std::string >	vars,
		bool	include_semilocal = false
	)

Definition at line 26 of file AllLocalDofIndicesThread.C.

  : 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]));
  }
}

AllLocalDofIndicesThread() [2/2]

AllLocalDofIndicesThread::AllLocalDofIndicesThread	(	AllLocalDofIndicesThread &	x,
		libMesh::Threads::split	split
	)

Definition at line 57 of file AllLocalDofIndicesThread.C.

  : ParallelObject(x._problem.comm()),
    _problem(x._problem),
    _sys(x._sys),
    _var_numbers(x._var_numbers),
    _include_semilocal(x._include_semilocal)
{
}

Member Function Documentation

dofIndicesSetUnion()

void AllLocalDofIndicesThread::dofIndicesSetUnion

(

)

Definition at line 105 of file AllLocalDofIndicesThread.C.

{
  _communicator.set_union(_all_dof_indices);
}

getDofIndices()

const std::set<dof_id_type>& AllLocalDofIndicesThread::getDofIndices ( ) const

inline

Definition at line 43 of file AllLocalDofIndicesThread.h.

Referenced by FixedPointSolve::solve(), TransientMultiApp::solveStep(), and SystemBase::zeroVariables().

{ return _all_dof_indices; }

join()

void AllLocalDofIndicesThread::join

(

const AllLocalDofIndicesThread &

y

)

Definition at line 99 of file AllLocalDofIndicesThread.C.

{
  _all_dof_indices.insert(y._all_dof_indices.begin(), y._all_dof_indices.end());
}

operator()()

void AllLocalDofIndicesThread::operator()

(

const libMesh::ConstElemRange &

range

)

Definition at line 68 of file AllLocalDofIndicesThread.C.

{
  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);
      }
    }
  }
}

Member Data Documentation

_all_dof_indices

std::set<dof_id_type> AllLocalDofIndicesThread::_all_dof_indices

protected

Definition at line 57 of file AllLocalDofIndicesThread.h.

Referenced by dofIndicesSetUnion(), getDofIndices(), join(), and operator()().

_include_semilocal

const bool AllLocalDofIndicesThread::_include_semilocal

protected

Whether to include semilocal dof indices.

Definition at line 53 of file AllLocalDofIndicesThread.h.

Referenced by operator()().

_problem

SubProblem& AllLocalDofIndicesThread::_problem

protected

Definition at line 48 of file AllLocalDofIndicesThread.h.

Referenced by AllLocalDofIndicesThread().

_sys

libMesh::System* AllLocalDofIndicesThread::_sys

protected

Definition at line 49 of file AllLocalDofIndicesThread.h.

Referenced by AllLocalDofIndicesThread(), and operator()().

_tid

THREAD_ID AllLocalDofIndicesThread::_tid

protected

Definition at line 55 of file AllLocalDofIndicesThread.h.

Referenced by operator()().

_var_numbers

std::vector<unsigned int> AllLocalDofIndicesThread::_var_numbers

protected

Definition at line 50 of file AllLocalDofIndicesThread.h.

Referenced by AllLocalDofIndicesThread(), and operator()().

---

The documentation for this class was generated from the following files:

• include/loops/AllLocalDofIndicesThread.h
• src/loops/AllLocalDofIndicesThread.C