moments.h

// Copyright (c) 2020 Chris Richardson & Matthew Scroggs
// FEniCS Project
// SPDX-License-Identifier:    MIT
 
#pragma once
 
#include "cell.h"
#include "polyset.h"
#include <array>
#include <concepts>
#include <tuple>
#include <utility>
#include <vector>
 
namespace basix
{
template <std::floating_point T>
class FiniteElement;
 
namespace moments
{
 
template <std::floating_point T>
std::tuple<std::vector<std::vector<T>>, std::array<std::size_t, 2>,
           std::vector<std::vector<T>>, std::array<std::size_t, 4>>
make_integral_moments(const FiniteElement<T>& moment_space, cell::type celltype,
                      polyset::type ptype, std::size_t value_size, int q_deg);
 
template <std::floating_point T>
std::tuple<std::vector<std::vector<T>>, std::array<std::size_t, 2>,
           std::vector<std::vector<T>>, std::array<std::size_t, 4>>
make_dot_integral_moments(const FiniteElement<T>& V, cell::type celltype,
                          polyset::type ptype, std::size_t value_size,
                          int q_deg);
 
template <std::floating_point T>
std::tuple<std::vector<std::vector<T>>, std::array<std::size_t, 2>,
           std::vector<std::vector<T>>, std::array<std::size_t, 4>>
make_tangent_integral_moments(const FiniteElement<T>& V, cell::type celltype,
                              polyset::type ptype, std::size_t value_size,
                              int q_deg);
 
template <std::floating_point T>
std::tuple<std::vector<std::vector<T>>, std::array<std::size_t, 2>,
           std::vector<std::vector<T>>, std::array<std::size_t, 4>>
make_normal_integral_moments(const FiniteElement<T>& V, cell::type celltype,
                             polyset::type ptype, std::size_t value_size,
                             int q_deg);
 
} // namespace moments
} // namespace basix