policies.hpp

Go to the documentation of this file.

// Copyright (c) 2019--present, The Simons Foundation
// This file is part of TRIQS/nda and is licensed under the Apache License, Version 2.0.
// SPDX-License-Identifier: Apache-2.0
// See LICENSE in the root of this distribution for details.

 
#pragma once
 
#include "./idx_map.hpp"
#include "../traits.hpp"
 
#include <cstdint>
#include <type_traits>
 
namespace nda {


  // Forward declarations.
  struct C_stride_layout;
  struct F_stride_layout;

  struct C_layout {
    template <int Rank>
    using mapping = idx_map<Rank, 0, C_stride_order<Rank>, layout_prop_e::contiguous>;

    using with_lowest_guarantee_t = C_stride_layout;

    using contiguous_t = C_layout;
  };

  struct F_layout {
    template <int Rank>
    using mapping = idx_map<Rank, 0, Fortran_stride_order<Rank>, layout_prop_e::contiguous>;

    using with_lowest_guarantee_t = F_stride_layout;

    using contiguous_t = F_layout;
  };

  struct C_stride_layout {
    template <int Rank>
    using mapping = idx_map<Rank, 0, C_stride_order<Rank>, layout_prop_e::none>;

    using with_lowest_guarantee_t = C_stride_layout;

    using contiguous_t = C_layout;
  };

  struct F_stride_layout {
    template <int Rank>
    using mapping = idx_map<Rank, 0, Fortran_stride_order<Rank>, layout_prop_e::none>;

    using with_lowest_guarantee_t = F_stride_layout;

    using contiguous_t = F_layout;
  };

  template <uint64_t StaticExtents, uint64_t StrideOrder, layout_prop_e LayoutProp>
  struct basic_layout {
    template <int Rank>
    using mapping = idx_map<Rank, StaticExtents, StrideOrder, LayoutProp>;

    using with_lowest_guarantee_t = basic_layout<StaticExtents, StrideOrder, layout_prop_e::none>;

    using contiguous_t = basic_layout<StaticExtents, StrideOrder, layout_prop_e::contiguous>;
  };

  template <uint64_t StrideOrder>
  using contiguous_layout_with_stride_order = basic_layout<0, StrideOrder, layout_prop_e::contiguous>;

  template <int Rank, uint64_t StrideOrder>
  using get_contiguous_layout_policy =
     std::conditional_t<StrideOrder == C_stride_order<Rank>, C_layout,
                        std::conditional_t<StrideOrder == Fortran_stride_order<Rank>, F_layout, contiguous_layout_with_stride_order<StrideOrder>>>;
 
  namespace detail {

    // Forward declarations.
    template <typename L>
    struct layout_to_policy;
 
    // Get the correct layout policy given a general nda::idx_map.
    template <int Rank, uint64_t StaticExtents, uint64_t StrideOrder, layout_prop_e LayoutProp>
    struct layout_to_policy<idx_map<Rank, StaticExtents, StrideOrder, LayoutProp>> {
      using type = basic_layout<StaticExtents, StrideOrder, LayoutProp>;
    };
 
    // Get the correct layout policy given a contiguous nda::idx_map with C-order.
    template <int Rank>
    struct layout_to_policy<idx_map<Rank, 0, C_stride_order<Rank>, layout_prop_e::contiguous>> {
      using type = C_layout;
    };
 
    // Get the correct layout policy given a strided nda::idx_map with C-order.
    template <int Rank>
    struct layout_to_policy<idx_map<Rank, 0, C_stride_order<Rank>, layout_prop_e::none>> {
      using type = C_stride_layout;
    };
 
    // Get the correct layout policy given a contiguous nda::idx_map with Fortran-order.
    template <int Rank>
      requires(Rank > 1)
    struct layout_to_policy<idx_map<Rank, 0, Fortran_stride_order<Rank>, layout_prop_e::contiguous>> {
      using type = F_layout;
    };
 
    // Get the correct layout policy given a strided nda::idx_map with Fortran-order.
    template <int Rank>
      requires(Rank > 1)
    struct layout_to_policy<idx_map<Rank, 0, Fortran_stride_order<Rank>, layout_prop_e::none>> {
      using type = F_stride_layout;
    };
 
  } // namespace detail

 
} // namespace nda