#include "mfem.hpp"
#include <cmath>

#include "polyCoeff.h"

namespace polycoeff{
  double xi_coeff_func(const mfem::Vector &x)
  {
    return std::pow(x(0), 2);
  }

  void vec_xi_coeff_func(const mfem::Vector &x, mfem::Vector &v)
  {
    v.SetSize(1);
    v[0] = -std::pow(x(0), 2);
  }

  double theta_initial_guess(const mfem::Vector &x, double root)
  {
    double xi = x[0];
    return 1 - std::pow(xi / root, 2);
  }
}