Function MakeCoefficient

fem/python_fem.cpp:35–77

  NGS_DLL_HEADER void ExportStdMathFunctions(py::module &m);

  shared_ptr<CoefficientFunction> MakeCoefficient (py::object val)
  {
    py::extract<shared_ptr<CoefficientFunction>> ecf(val);
    if (ecf.check()) return ecf();

    // a numpy.complex converts itself to a real, and prints a warning
    // thus we check for it first
    // if (string(py::str(val.get_type())) == "<class 'numpy.complex128'>")
    if (string(py::str(py::type::of(val))) == "<class 'numpy.complex128'>")
      return MakeConstantCoefficientFunction(val.cast<Complex>());

    try
      {
        double value = val.cast<double>();
        if (value)
          return make_shared<ConstantCoefficientFunction> (value);
        else
          return ZeroCF(Array<int>());
      }
    catch(const py::cast_error&) {}
    try { return MakeConstantCoefficientFunction(val.cast<Complex>()); }
    catch(const py::cast_error&) {}

    if (py::isinstance<py::list>(val))
      {
        auto el = py::cast<py::list>(val);
        Array<shared_ptr<CoefficientFunction>> cflist(el.size());
        for (int i : Range(cflist.Size()))
          cflist[i] = MakeCoefficient(el[i]);
        return MakeDomainWiseCoefficientFunction(std::move(cflist));
      }

    if (py::isinstance<py::tuple>(val))
      {
        auto et = py::cast<py::tuple>(val);
        Array<shared_ptr<CoefficientFunction>> cflist(et.size());
        for (int i : Range(cflist.Size()))
          cflist[i] = MakeCoefficient(et[i]);
        return MakeVectorialCoefficientFunction(std::move(cflist));
      }
    // throw std::invalid_argument(string("Cannot make CoefficientFunction from ") + string(py::str(val)) + " of type " + string(py::str(val.get_type())));
    throw std::invalid_argument(string("Cannot make CoefficientFunction from ") + string(py::str(val)) + " of type " + string(py::str(py::type::of(val))));
  }

  Array<shared_ptr<CoefficientFunction>> MakeCoefficients (py::object py_coef)
  {