common: some helper funcs

src/common.hpp

@@ -11,4 +11,14 @@
 typedef CppAD::AD<double> ADDouble;
 typedef std::complex<ADDouble> ADComplex;
 
+ADComplex adcomplex(const ADComplex& cpx)
+{
+    return {cpx};
+}
+
+ADComplex adcomplex(const std::complex<double>& cpx)
+{
+    return {cpx};
+}
+
 #endif

src/main.cpp

@@ -2,6 +2,7 @@
 #include "array.hpp"
 #include "action.hpp"
 
+using namespace std::complex_literals;
 
 template <typename Scalar>
 void dump(Array<Scalar,3> Q)
@@ -34,7 +35,9 @@ int main()
     //auto mat = Q.to_matrix();
     auto mat = Q.to_fixed_matrix<2,2>();
 
-    mat = mat * mat * mat;
+    mat = mat * mat * mat * adcomplex(1j);
+
+    mat(0,0) = mat(1,1) = mat.trace();
 
     dump(Q0);
     std::cout << mat << std::endl;