Program Listing for File ffinterpolation.h

↰ Return to documentation for file (framework/field_functions/interpolation/ffinterpolation.h)

// SPDX-FileCopyrightText: 2024 The OpenSn Authors <https://open-sn.github.io/opensn/>
// SPDX-License-Identifier: MIT

#pragma once

#include <memory>
#include <stdexcept>
#include <vector>
#include <string>

namespace opensn
{
class FieldFunction;
class FieldFunctionGridBased;

enum class FieldFunctionInterpolationType : int
{
  SLICE = 1,
  LINE = 2,
  VOLUME = 3,
  POINT = 4
};

enum class FieldFunctionInterpolationOperation : int
{
  OP_SUM = 10,
  OP_AVG = 11,
  OP_MAX = 12,
  OP_SUM_FUNC = 13,
  OP_AVG_FUNC = 14,
  OP_MAX_FUNC = 15,
};

enum class FieldFunctionInterpolationProperty : int
{
  PROBEPOINT = 0,
  SLICEPOINT = 1,
  SLICENORMAL = 2,
  SLICETANGENT = 3,
  SLICEBINORM = 4,
  OPERATION = 5,
  LOGICAL_VOLUME = 8,
  ADD_FIELD_FUNCTION = 9,
  SET_FIELD_FUNCTIONS = 10,
  FIRSTPOINT = 11,
  SECONDPOINT = 12,
  NUMBEROFPOINTS = 13,
  CUSTOM_ARRAY = 14,
};

/// Base class for field-function interpolation objects.
class FieldFunctionInterpolation
{
protected:
  FieldFunctionInterpolationType type_;
  unsigned int ref_component_;
  std::vector<std::shared_ptr<FieldFunctionGridBased>> field_functions_;

public:
  explicit FieldFunctionInterpolation(FieldFunctionInterpolationType type)
    : type_(type), ref_component_(0)
  {
  }

  virtual ~FieldFunctionInterpolation() = default;

  std::vector<std::shared_ptr<FieldFunctionGridBased>>& GetFieldFunctions()
  {
    return field_functions_;
  }

  void AddFieldFunction(std::shared_ptr<FieldFunction> ff)
  {
    auto gbff = std::dynamic_pointer_cast<FieldFunctionGridBased>(ff);
    if (gbff)
      field_functions_.push_back(gbff);
    else
      throw std::runtime_error("Expected FieldFunctionGridBased field function");
  }

  FieldFunctionInterpolationType Type() const { return type_; }

  virtual void Initialize() {};

  virtual void Execute() {};

  virtual void ExportToCSV(std::string base_name) const {};

  virtual void ExportToPython(std::string base_name) {};
};

} // namespace opensn