Changes in src/AST/Pass.proto.hpp [dff6452:0b8bf27]

File:

• src/AST/Pass.proto.hpp (modified) (2 diffs)

src/AST/Pass.proto.hpp

@@ -115 +115 @@
                 static constexpr bool value = std::is_void< ret_t >::value ||
                         std::is_base_of<const node_t, typename std::remove_pointer<ret_t>::type >::value;
+        };
+
+        template<bool is_void>
+        struct __assign;
+
+        template<>
+        struct __assign<true> {
+                template<typename pass_t, typename node_t>
+                static inline void result( pass_t & pass, const node_t * & node ) {
+                        pass.previsit( node );
+                }
+        };
+
+        template<>
+        struct __assign<false> {
+                template<typename pass_t, typename node_t>
+                static inline void result( pass_t & pass, const node_t * & node ) {
+                        node = pass.previsit( node );
+                        assertf(node, "Previsit must not return NULL");
+                }
         };
 
@@ -138 +158 @@
                         "Previsit may not change the type of the node. It must return its paremeter or void."
                 );
-                if(std::is_void< decltype( pass.previsit(node) ) >::value) {
-                        pass.previsit( node );
-                } else {
-                        node = pass.previsit( node );
-                        assert(node);
-                }
+
+                __assign<
+                        std::is_void<
+                                decltype( pass.previsit( node ) )
+                        >::value
+                >::result( pass, node );
         }