polyset.h

// Copyright (c) 2020 Chris Richardson & Matthew Scroggs
// FEniCS Project
// SPDX-License-Identifier:    MIT
 
#pragma once
 
#include "cell.h"
#include "mdspan.hpp"
#include <array>
#include <concepts>
#include <utility>
#include <vector>
 
namespace basix::polyset
{
 
enum class type
{
  standard = 0,
  macroedge = 1,
};
 
template <std::floating_point T>
std::pair<std::vector<T>, std::array<std::size_t, 3>>
tabulate(cell::type celltype, polyset::type ptype, int d, int n,
         MDSPAN_IMPL_STANDARD_NAMESPACE::mdspan<
             const T, MDSPAN_IMPL_STANDARD_NAMESPACE::dextents<std::size_t, 2>>
             x);
 
template <std::floating_point T>
void tabulate(
    MDSPAN_IMPL_STANDARD_NAMESPACE::mdspan<
        T, MDSPAN_IMPL_STANDARD_NAMESPACE::dextents<std::size_t, 3>>
        P,
    cell::type celltype, polyset::type ptype, int d, int n,
    MDSPAN_IMPL_STANDARD_NAMESPACE::mdspan<
        const T, MDSPAN_IMPL_STANDARD_NAMESPACE::dextents<std::size_t, 2>>
        x);
 
int dim(cell::type cell, polyset::type ptype, int d);
 
int nderivs(cell::type cell, int d);
 
polyset::type superset(cell::type cell, polyset::type type1,
                       polyset::type type2);
 
polyset::type restriction(polyset::type ptype, cell::type cell,
                          cell::type restriction_cell);
 
} // namespace basix::polyset