polyset.h

// Copyright (c) 2020 Chris Richardson & 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::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,
         md::mdspan<const T, md::dextents<std::size_t, 2>> x);

template <std::floating_point T>
void tabulate(md::mdspan<T, md::dextents<std::size_t, 3>> P,
              cell::type celltype, polyset::type ptype, int d, int n,
              md::mdspan<const T, md::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