dof-transformations.h

// Copyright (c) 2020 Chris Richardson & Matthew Scroggs
// FEniCS Project
// SPDX-License-Identifier:    MIT
 
#pragma once
 
#include "cell.h"
#include "maps.h"
#include "mdspan.hpp"
#include "polyset.h"
#include <array>
#include <concepts>
#include <map>
#include <utility>
#include <vector>

namespace basix::doftransforms
{
template <std::floating_point T>
std::map<cell::type, std::pair<std::vector<T>, std::array<std::size_t, 3>>>
compute_entity_transformations(
    cell::type cell_type,
    std::array<std::vector<md::mdspan<const T, md::dextents<std::size_t, 2>>>,
               4>
        x,
    std::array<std::vector<md::mdspan<const T, md::dextents<std::size_t, 4>>>,
               4>
        M,
    md::mdspan<const T, md::dextents<std::size_t, 2>> coeffs, int degree,
    std::size_t vs, maps::type map_type, polyset::type ptype);
 
} // namespace basix::doftransforms