fv_diffusion_utils.cc

Go to the documentation of this file.

#include "console/chi_console.h"
#include "chi_lua.h"
#include "fv_diffusion_solver.h"
 
#include "physics/FieldFunction/fieldfunction_gridbased.h"
 
//###################################################################
/**Calls a lua function with xyz coordinates.
 * \param L The lua state.
 * \param lua_func_name The name used to define this lua function in the lua
 *                      state.
 * \param imat The material ID of the cell
 * \param xyz The xyz coordinates of the point where the function is called.
 *
 * \return The function evaluation.*/
double fv_diffusion::Solver::CallLua_iXYZFunction(
  lua_State* L,
  const std::string& lua_func_name,
  const int imat,
  const chi_mesh::Vector3& xyz)
{
  //============= Load lua function
  lua_getglobal(L, lua_func_name.c_str());
 
  //============= Error check lua function
  if (not lua_isfunction(L, -1))
    throw std::logic_error("CallLua_iXYZFunction attempted to access lua-function, " +
                           lua_func_name + ", but it seems the function"
                                           " could not be retrieved.");
 
  //============= Push arguments
  lua_pushinteger(L, imat);
  lua_pushnumber(L, xyz.x);
  lua_pushnumber(L, xyz.y);
  lua_pushnumber(L, xyz.z);
 
  //============= Call lua function
  //4 arguments, 1 result (double), 0=original error object
  double lua_return;
  if (lua_pcall(L,4,1,0) == 0)
  {
    LuaCheckNumberValue("CallLua_iXYZFunction", L, -1);
    lua_return = lua_tonumber(L,-1);
  }
  else
    throw std::logic_error("CallLua_iXYZFunction attempted to call lua-function, " +
                           lua_func_name + ", but the call failed." +
                           xyz.PrintStr());
 
  lua_pop(L,1); //pop the double, or error code
 
  return lua_return;
}
 
 
//###################################################################
/**Updates the field functions with the latest data.*/
void fv_diffusion::Solver::UpdateFieldFunctions()
{
  auto& ff = *field_functions_.front();
  ff.UpdateFieldVector(x_);
}