RandomPartitioner.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* 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 "RandomPartitioner.h"

#include "MooseApp.h"
#include "MooseMesh.h"
#include "MooseRandom.h"

#include "libmesh/elem.h"

registerMooseObject("MooseApp", RandomPartitioner);

InputParameters
RandomPartitioner::validParams()
{
  InputParameters params = MoosePartitioner::validParams();

  params.addParam<unsigned int>("seed", 0, "Seed for the random generator");

  params.addClassDescription("Assigns element processor ids randomly with a given seed.");

  return params;
}

RandomPartitioner::RandomPartitioner(const InputParameters & params)
  : MoosePartitioner(params), _num_procs(_app.getCommunicator()->size())
{
  MooseRandom::seed(getParam<unsigned int>("seed"));
}

RandomPartitioner::~RandomPartitioner() {}

std::unique_ptr<Partitioner>
RandomPartitioner::clone() const
{
  return _app.getFactory().clone(*this);
}

void
RandomPartitioner::_do_partition(MeshBase & mesh, const unsigned int /*n*/)
{
  // Random number is on [0, 1]: scale to number of procs and round down
  for (auto & elem_ptr : mesh.active_element_ptr_range())
    elem_ptr->processor_id() = std::floor(MooseRandom::rand() * _num_procs);
}