vtkYieldCriteria.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 vtkYieldCriteria_h
#define vtkYieldCriteria_h
 
#include "vtkDataSetAlgorithm.h"
#include "vtkFiltersTensorModule.h" // For export macro
#include "vtkNew.h"                 // For vtkNew
 
#include <string> // for std::string
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDataArray;
class vtkDataArraySelection;
class vtkDataSet;
class vtkTensorPrincipalInvariants;
 
class VTKFILTERSTENSOR_EXPORT vtkYieldCriteria : public vtkDataSetAlgorithm
{
public:
  static vtkYieldCriteria* New();
  vtkTypeMacro(vtkYieldCriteria, vtkDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  enum class Criterion
  {
    PrincipalStress = 0,
    Tresca,
    VonMises
  };


  vtkGetMacro(ScaleVectors, bool);
  vtkSetMacro(ScaleVectors, bool);

  vtkGetObjectMacro(PointDataArraySelection, vtkDataArraySelection);

  vtkGetObjectMacro(CellDataArraySelection, vtkDataArraySelection);

  vtkGetObjectMacro(CriteriaSelection, vtkDataArraySelection);
 
protected:
  vtkYieldCriteria();
  ~vtkYieldCriteria() override = default;

  bool ComputeYieldCriteria(vtkDataSet* output, vtkDataArray* array, const std::string& arrayName,
    vtkIdType nbTuples, bool isPointData) const;
 
  int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  vtkNew<vtkTensorPrincipalInvariants> InvariantsFilter;
  vtkNew<vtkDataArraySelection> PointDataArraySelection;
  vtkNew<vtkDataArraySelection> CellDataArraySelection;
  vtkNew<vtkDataArraySelection> CriteriaSelection;
  bool ScaleVectors = false;
 
private:
  vtkYieldCriteria(const vtkYieldCriteria&) = delete;
  void operator=(const vtkYieldCriteria&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif