From 4e2f157604599f2f53ef5d085fb2e86e7b679332 Mon Sep 17 00:00:00 2001
From: conrad <conrad@localhost.localdomain>
Date: Fri, 24 Jan 2025 10:56:07 +1000
Subject: [PATCH] reinterpret pow(x,2) as square(x)

---
 include/armadillo_bits/fn_accu.hpp | 39 +++++++++++++++++++++++++++++-
 1 file changed, 38 insertions(+), 1 deletion(-)

diff --git a/include/armadillo_bits/fn_accu.hpp b/include/armadillo_bits/fn_accu.hpp
index 347e9372..0c1af010 100644
--- a/include/armadillo_bits/fn_accu.hpp
+++ b/include/armadillo_bits/fn_accu.hpp
@@ -891,7 +891,44 @@ accu(const BaseCube<typename T1::elem_type,T1>& X)
 
 
 
-// TODO: accu(const eOpCube<T1,eop_pow>& expr)
+template<typename T1>
+arma_warn_unused
+inline
+typename T1::elem_type
+accu(const eOpCube<T1,eop_pow>& expr)
+  {
+  arma_debug_sigprint();
+  
+  typedef eOpCube<T1,eop_pow> expr_type;
+  
+  typedef typename T1::elem_type eT;
+  
+  if(expr.aux == eT(2))
+    {
+    typedef typename ProxyCube<T1>::stored_type expr_P_stored_type;
+    
+    if(is_Cube<expr_P_stored_type>::value)
+      {
+      const unwrap_cube<expr_P_stored_type> U(expr.P.Q);
+      
+      const eT* X_mem = U.M.memptr();
+      
+      return op_dot::direct_dot(U.M.n_elem, X_mem, X_mem);
+      }
+    else
+      {
+      typedef eOpCube<T1,eop_square> modified_expr_type;
+      
+      const ProxyCube<modified_expr_type> P( reinterpret_cast< const modified_expr_type& >(expr) );
+      
+      return (ProxyCube<modified_expr_type>::use_at) ? accu_cube_proxy_at(P) : accu_cube_proxy_linear(P);
+      }
+    }
+  
+  const ProxyCube<expr_type> P(expr);
+  
+  return (ProxyCube<expr_type>::use_at) ? accu_cube_proxy_at(P) : accu_cube_proxy_linear(P);
+  }