File: for_each.hpp

package info (click to toggle)
gridtools 2.0.0-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye
  • size: 21,728 kB
  • sloc: cpp: 45,263; python: 9,383; javascript: 8,445; ansic: 2,564; sh: 509; f90: 370; makefile: 216
file content (57 lines) | stat: -rw-r--r-- 2,027 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
/*
 * GridTools
 *
 * Copyright (c) 2014-2019, ETH Zurich
 * All rights reserved.
 *
 * Please, refer to the LICENSE file in the root directory.
 * SPDX-License-Identifier: BSD-3-Clause
 */

#pragma once

namespace cpp_bindgen {

    namespace for_each_detail {
        template <class List>
        struct for_each_impl;

        template <template <class...> class L, class... Ts>
        struct for_each_impl<L<Ts...>> {
            template <class Fun>
            static void exec(Fun const &fun) {
                (void)(int[]){((void)fun(Ts{}), 0)...};
            }
        };

        template <class List>
        struct for_each_type_impl;

        template <template <class...> class L, class... Ts>
        struct for_each_type_impl<L<Ts...>> {
            template <class Fun>
            static void exec(Fun const &fun) {
                (void)(int[]){((void)fun.template operator()<Ts>(), 0)...};
            }
        };
    } // namespace for_each_detail

    /// Calls fun(T{}) for each element of the type list List.
    template <class List, class Fun>
    void for_each(Fun const &fun) {
        for_each_detail::for_each_impl<List>::exec(fun);
    };

    ///  Calls fun.template operator<T>() for each element of the type list List.
    ///
    ///  Note the difference between for_each: T is passed only as a template parameter; the operator itself has to
    ///  be a nullary function. This ensures that the object of type T is nor created, nor passed to the function.
    ///  The disadvantage is that the functor can not be a [generic] lambda (in C++14 syntax) and also it limits the
    ///  ability to do operator(). However, if T is not a POD it makes sense to use this for_each flavour. Also
    ///  nvcc8 has problems with the code generation for the regular for_each even if all the types are empty
    ///  structs.
    template <class List, class Fun>
    void for_each_type(Fun const &fun) {
        for_each_detail::for_each_type_impl<List>::exec(fun);
    }
} // namespace cpp_bindgen