vtkDGOperatorEntry.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 vtkDGOperatorEntry_h
#define vtkDGOperatorEntry_h
 
#include "vtkFiltersCellGridModule.h" // For export macro.
#include "vtkStringToken.h"           // For ivar.
 
#include <array>
#include <functional>
#include <vector>
 
VTK_ABI_NAMESPACE_BEGIN

class VTKFILTERSCELLGRID_EXPORT vtkDGOperatorEntry
{
public:
  using OperatorFunction = std::function<void(
    const std::array<double, 3>&, std::vector<double>&, const std::vector<int>&)>;

  using FunctionCounter = std::function<int(const std::vector<int>&)>;
 
  vtkDGOperatorEntry() = default;
  vtkDGOperatorEntry(const vtkDGOperatorEntry&) = default;
  vtkDGOperatorEntry& operator=(const vtkDGOperatorEntry&) = default;

  vtkDGOperatorEntry(int numFunc, int opSize, OperatorFunction op, const std::string& shader)
    : NumberOfFunctions(numFunc)
    , OperatorSize(opSize)
    , Op(op)
    , ShaderOp(shader)
  {
  }

  vtkDGOperatorEntry(
    FunctionCounter counter, int opSize, OperatorFunction op, const std::string& shader)
    : OperatorSize(opSize)
    , Op(op)
    , ShaderOp(shader)
    , FunctionCount(counter)
  {
  }

  operator bool() const { return !!this->Op && this->NumberOfFunctions > 0; }

  bool IsOrderDependent() const { return !!this->FunctionCount; }

  bool SetOrder(const std::vector<int>& order);

  void Evaluate(const std::array<double, 3>& rst, std::vector<double>& values) const
  {
    this->Op(rst, values, this->Order);
  }


  static int TensorProductFunctionCount(const std::vector<int>& order);

  static int SimplexFunctionCount(const std::vector<int>& order);

  static int WedgeFunctionCount(const std::vector<int>& order);

  std::string GetShaderString(
    const std::string& functionName, const std::string& parameterName, int order) const;

  int NumberOfFunctions{ 0 };

  int OperatorSize{ 1 };

  OperatorFunction Op;

  std::string ShaderOp;

  std::vector<int> Order;

  FunctionCounter FunctionCount;
};
 
VTK_ABI_NAMESPACE_END
#endif // vtkDGOperatorEntry_h