LAROMANCE6DInterpolation.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* 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

#ifdef NEML2_ENABLED

#include "neml2/models/Model.h"
#include "nlohmann/json.h"

namespace neml2
{
class LAROMANCE6DInterpolation : public Model
{
public:
  static OptionSet expected_options();

  LAROMANCE6DInterpolation(const OptionSet & options);

  void request_AD() override;

  enum class TransformEnum
  {
    COMPRESS,
    DECOMPRESS,
    LOG10BOUNDED,
    EXP10BOUNDED,
    MINMAX
  };

protected:
  void set_value(bool, bool, bool) override;

private:
  Scalar _stress_grid;
  Scalar _temperature_grid;
  Scalar _plastic_strain_grid;
  Scalar _cell_grid;
  Scalar _wall_grid;
  Scalar _env_grid;

  Scalar _grid_values;

  // @{
  const Variable<Scalar> & _vm_stress;
  const Variable<Scalar> & _temperature;
  const Variable<Scalar> & _ep_strain;
  const Variable<Scalar> & _cell_dd;
  const Variable<Scalar> & _wall_dd;
  const Variable<Scalar> & _env_fac;
  // @}

  // @{
  Variable<Scalar> & _output_rate;
  // @}

  nlohmann::json _json;

  std::pair<Scalar, Scalar> findLeftIndexAndFraction(const Scalar & grid,
                                                     const Scalar & interp_points) const;

  Scalar interpolate_and_transform() const;

  Scalar transform_data(const Scalar & data,
                        const std::vector<double> & param,
                        TransformEnum transform_type) const;

  std::string json_to_string(const std::string & key) const;

  std::vector<double> json_to_vector(const std::string & key) const;

  Scalar json_6Dvector_to_torch(const std::string & key) const;

  Scalar json_vector_to_torch(const std::string & key) const;

  Scalar compute_interpolation(const std::vector<std::pair<Scalar, Scalar>> index_and_fraction,
                               const Scalar grid_values) const;

  TransformEnum get_transform_enum(const std::string & name) const;

  TransformEnum _stress_transform_enum;
  TransformEnum _temperature_transform_enum;
  TransformEnum _plastic_strain_transform_enum;
  TransformEnum _cell_transform_enum;
  TransformEnum _wall_transform_enum;
  TransformEnum _env_transform_enum;

  std::vector<double> _stress_transform_values;
  std::vector<double> _temperature_transform_values;
  std::vector<double> _plastic_strain_transform_values;
  std::vector<double> _cell_transform_values;
  std::vector<double> _wall_transform_values;
  std::vector<double> _env_transform_values;

  std::string _output_rate_name;
  std::vector<double> _output_transform_values;
  TransformEnum _output_transform_enum;

  // @{
  Scalar transform_compress(const Scalar & data, const std::vector<double> & params) const;
  Scalar transform_decompress(const Scalar & data, const std::vector<double> & params) const;
  Scalar transform_log10_bounded(const Scalar & data, const std::vector<double> & params) const;
  Scalar transform_exp10_bounded(const Scalar & data, const std::vector<double> & params) const;
  Scalar transform_min_max(const Scalar & data, const std::vector<double> & params) const;
  // @}
};
} // namespace neml2

#endif