conjugate.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/core/runtime_precision.hpp>
#include <uni20/core/scalar_concepts.hpp>
#include <uni20/mdspan/conjugate_accessor.hpp>
#include <uni20/tensor/concepts.hpp>
 
#include <cstddef>
#include <memory>
#include <new>
#include <type_traits>
#include <utility>
 
namespace uni20
{
template <TensorView Tensor> class ConstTensorView {
  public:
    using tensor_type = std::remove_cvref_t<Tensor>;
    using storage_policy = tensor_storage_policy_t<tensor_type>;
    using async_alias_tag = void;
 
    explicit constexpr ConstTensorView(tensor_type const& tensor) noexcept : tensor_(std::addressof(tensor)) {}
 
    explicit constexpr ConstTensorView(tensor_type const* tensor) noexcept : tensor_(tensor) {}
 
#if UNI20_HAS_MPREAL
    [[nodiscard]] Precision default_precision() const
      requires has_runtime_precision_v<tensor_element_t<Tensor>>
    {
      return this->base().default_precision();
    }
 
    [[nodiscard]] std::optional<Precision> default_precision_if_set() const
      requires has_runtime_precision_v<tensor_element_t<Tensor>>
    {
      return this->base().default_precision_if_set();
    }
#endif
 
    [[nodiscard]] constexpr decltype(auto) backend_selector() const
        noexcept(noexcept(std::declval<tensor_type const&>().backend_selector()))
    {
      return this->base().backend_selector();
    }
 
    [[nodiscard]] constexpr decltype(auto) mdspan() const
        noexcept(noexcept(std::declval<tensor_type const&>().mdspan()))
      requires ImmediateTensorView<tensor_type>
    {
      return this->base().mdspan();
    }
 
    [[nodiscard]] constexpr decltype(auto) mdspec() const
        noexcept(noexcept(mdspec_of(std::declval<tensor_type const&>())))
    {
      return mdspec_of(this->base());
    }
 
    [[nodiscard]] constexpr decltype(auto) extents() const
        noexcept(noexcept(std::declval<tensor_type const&>().extents()))
    {
      return this->base().extents();
    }
 
    [[nodiscard]] constexpr decltype(auto) extent(std::size_t axis) const
        noexcept(noexcept(std::declval<tensor_type const&>().extent(axis)))
    {
      return this->base().extent(axis);
    }
 
    [[nodiscard]] constexpr tensor_type const& base() const noexcept { return *std::launder(tensor_); }
 
  private:
    tensor_type const* tensor_;
};
 
template <TensorView Tensor> class ConjugatedTensorView {
  public:
    using tensor_type = std::remove_cvref_t<Tensor>;
    using storage_policy = tensor_storage_policy_t<tensor_type>;
    using async_alias_tag = void;
 
    explicit constexpr ConjugatedTensorView(tensor_type const& tensor) noexcept : tensor_(std::addressof(tensor)) {}
 
    explicit constexpr ConjugatedTensorView(tensor_type const* tensor) noexcept : tensor_(tensor) {}
 
#if UNI20_HAS_MPREAL
    [[nodiscard]] Precision default_precision() const
      requires has_runtime_precision_v<tensor_element_t<Tensor>>
    {
      return this->base().default_precision();
    }
 
    [[nodiscard]] std::optional<Precision> default_precision_if_set() const
      requires has_runtime_precision_v<tensor_element_t<Tensor>>
    {
      return this->base().default_precision_if_set();
    }
#endif
 
    [[nodiscard]] constexpr decltype(auto) backend_selector() const
        noexcept(noexcept(std::declval<tensor_type const&>().backend_selector()))
    {
      return this->base().backend_selector();
    }
 
    [[nodiscard]] constexpr auto mdspan() const
        noexcept(noexcept(uni20::conj(std::declval<tensor_type const&>().mdspan())))
      requires ImmediateTensorView<tensor_type>
    {
      return uni20::conj(this->base().mdspan());
    }
 
    [[nodiscard]] constexpr auto mdspec() const
        noexcept(noexcept(uni20::conj(mdspec_of(std::declval<tensor_type const&>()))))
    {
      return uni20::conj(mdspec_of(this->base()));
    }
 
    [[nodiscard]] constexpr decltype(auto) extents() const
        noexcept(noexcept(std::declval<tensor_type const&>().extents()))
    {
      return this->base().extents();
    }
 
    [[nodiscard]] constexpr decltype(auto) extent(std::size_t axis) const
        noexcept(noexcept(std::declval<tensor_type const&>().extent(axis)))
    {
      return this->base().extent(axis);
    }
 
    [[nodiscard]] constexpr tensor_type const& base() const noexcept { return *std::launder(tensor_); }
 
  private:
    tensor_type const* tensor_;
};
 
template <TensorView Tensor>
  requires(std::is_lvalue_reference_v<Tensor &&> && Complex<tensor_element_t<Tensor>>)
[[nodiscard]] constexpr auto conj(Tensor&& tensor)
{
  return ConjugatedTensorView<std::remove_cvref_t<Tensor>>{tensor};
}
 
template <TensorView Tensor> [[nodiscard]] constexpr auto conj(ConjugatedTensorView<Tensor>& view) noexcept
{
  return ConstTensorView<typename ConjugatedTensorView<Tensor>::tensor_type>{view.base()};
}
 
template <TensorView Tensor> [[nodiscard]] constexpr auto conj(ConjugatedTensorView<Tensor> const& view) noexcept
{
  return ConstTensorView<typename ConjugatedTensorView<Tensor>::tensor_type>{view.base()};
}
 
template <TensorView Tensor> [[nodiscard]] constexpr auto conj(ConjugatedTensorView<Tensor>&& view) noexcept
{
  return ConstTensorView<typename ConjugatedTensorView<Tensor>::tensor_type>{view.base()};
}
 
template <TensorView Tensor>
  requires Complex<tensor_element_t<Tensor>>
[[nodiscard]] constexpr auto conj(ConstTensorView<Tensor>& view) noexcept
{
  return ConjugatedTensorView<typename ConstTensorView<Tensor>::tensor_type>{view.base()};
}
 
template <TensorView Tensor>
  requires Complex<tensor_element_t<Tensor>>
[[nodiscard]] constexpr auto conj(ConstTensorView<Tensor> const& view) noexcept
{
  return ConjugatedTensorView<typename ConstTensorView<Tensor>::tensor_type>{view.base()};
}
 
template <TensorView Tensor>
  requires Complex<tensor_element_t<Tensor>>
[[nodiscard]] constexpr auto conj(ConstTensorView<Tensor>&& view) noexcept
{
  return ConjugatedTensorView<typename ConstTensorView<Tensor>::tensor_type>{view.base()};
}
 
template <TensorView Tensor>
  requires(std::is_lvalue_reference_v<Tensor &&> && !Complex<tensor_element_t<Tensor>>)
[[nodiscard]] constexpr auto conj(Tensor&& tensor) noexcept -> std::remove_reference_t<Tensor> const&
{
  return tensor;
}
 
} // namespace uni20