Refactor constructField

alpaka-group · Jan 21, 2024 · 3c73b92 · 3c73b92
1 parent c7d9806
commit 3c73b92
Showing 1 changed file with 13 additions and 18 deletions.
diff --git a/include/llama/View.hpp b/include/llama/View.hpp
@@ -98,34 +98,29 @@ namespace llama
         typename Mapping::ArrayExtents::Index ai,
         RecordCoord<RCs...> rc)
     {
-        using FieldType = GetType<typename Mapping::RecordDim, decltype(rc)>;
-
-        // this handles physical and computed mappings
-        if constexpr(sizeof...(RCs) == 0)
+        auto init = [](auto&& ref) LLAMA_LAMBDA_INLINE
         {
-            using RefType = decltype(view(ai));
-            if constexpr(isProxyReference<RefType>)
+            using FieldType = GetType<typename Mapping::RecordDim, RecordCoord<RCs...>>;
+            using RefType = decltype(ref);
+            if constexpr(isProxyReference<std::remove_reference_t<RefType>>)
             {
-                view(ai) = FieldType{};
+                ref = FieldType{};
             }
             else if constexpr(
                 std::is_lvalue_reference_v<RefType> && !std::is_const_v<std::remove_reference_t<RefType>>)
             {
-                new(&view(ai)) FieldType{};
+                new(&ref) FieldType{};
             }
+        };
+
+        // this handles physical and computed mappings
+        if constexpr(sizeof...(RCs) == 0)
+        {
+            init(view(ai));
         }
         else
         {
-            using RefType = decltype(view(ai)(rc));
-            if constexpr(isProxyReference<RefType>)
-            {
-                view(ai)(rc) = FieldType{};
-            }
-            else if constexpr(
-                std::is_lvalue_reference_v<RefType> && !std::is_const_v<std::remove_reference_t<RefType>>)
-            {
-                new(&view(ai)(rc)) FieldType{};
-            }
+            init(view(ai)(rc));
         }
     }