vtkPartitionBalancer.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkPartitionBalancer_h
#define vtkPartitionBalancer_h
 
#include "vtkFiltersParallelModule.h" // for export macros
#include "vtkPartitionedDataSetAlgorithm.h"
 
VTK_ABI_NAMESPACE_BEGIN
class vtkMultiProcessController;
 
class VTKFILTERSPARALLEL_EXPORT vtkPartitionBalancer : public vtkPartitionedDataSetAlgorithm
{
public:
  static vtkPartitionBalancer* New();
  vtkTypeMacro(vtkPartitionBalancer, vtkPartitionedDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
 
  void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);
 
  enum Mode
  {
    Expand,
    Squash
  };
 
 
  vtkSetClampMacro(Mode, int, vtkPartitionBalancer::Expand, vtkPartitionBalancer::Squash);
  vtkGetMacro(Mode, int);
 
  void SetModeToExpand() { this->SetMode(vtkPartitionBalancer::Expand); }
 
  void SetModeToSquash() { this->SetMode(vtkPartitionBalancer::Squash); }
 
protected:
  vtkPartitionBalancer();
  ~vtkPartitionBalancer() override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  vtkMultiProcessController* Controller;
 
  int Mode;
 
private:
  vtkPartitionBalancer(const vtkPartitionBalancer&) = delete;
  void operator=(const vtkPartitionBalancer&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif