WorkBalance.h

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

#pragma once

#include "GeneralVectorPostprocessor.h"

class WorkBalance : public GeneralVectorPostprocessor
{
public:
  static InputParameters validParams();

  WorkBalance(const InputParameters & parameters);

  enum SystemEnum
  {
    // Ordered this way because NL is always system 0 and Aux is 1
    ALL = -1,
    NL,
    AUX,
  };

  virtual void initialize() override;
  virtual void execute() override;
  virtual void finalize() override;
  void gather(int balance_id, VectorPostprocessorValue & vppv);

protected:
  int _system;

  const RankMap & _rank_map;

  unsigned int _my_hardware_id;

  bool _sync_to_all_procs;

  dof_id_type _local_num_elems;
  dof_id_type _local_num_nodes;
  dof_id_type _local_num_dofs;
  dof_id_type _local_num_partition_sides;
  Real _local_partition_surface_area;

  // These are measuring the size of inter-nodal (compute nodes) communication
  dof_id_type _local_num_partition_hardware_id_sides;
  Real _local_partition_hardware_id_surface_area;

  VectorPostprocessorValue & _pid;

  MultiMooseEnum _balances;

  std::map<std::string, VectorPostprocessorValue *> _balance_vectors;
};