polynomials.h

// Copyright (c) 2021 Matthew Scroggs
// FEniCS Project
// SPDX-License-Identifier:    MIT
 
#pragma once
 
#include "cell.h"
#include "mdspan.hpp"
#include "types.h"
#include <array>
#include <concepts>
#include <utility>
#include <vector>

namespace basix::polynomials
{
enum class type
{
  legendre = 0,
  lagrange = 1,
  bernstein = 2,
};

template <std::floating_point T>
std::pair<std::vector<T>, std::array<std::size_t, 2>>
tabulate(polynomials::type polytype, cell::type celltype, int d,
         md::mdspan<const T, md::dextents<std::size_t, 2>> x);

int dim(polynomials::type polytype, cell::type cell, int d);
 
} // namespace basix::polynomials