replace powers of complex unit matrix by single matrix

FlexibleSUSY · Dec 5, 2015 · e748570 · e748570
1 parent ad1df1c
commit e748570
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/meta/BetaFunction.m b/meta/BetaFunction.m
@@ -134,7 +134,10 @@
               splitExpr = SplitExpression[expr, FlexibleSUSY`FSMaximumExpressionSize];
               result = MapIndexed[
                   ConvertSingleExprToC[
-                      #1 * CConversion`CreateUnitMatrix[type] /. CConversion`UNITMATRIX[r_]^_ :> CConversion`UNITMATRIX[r],
+                      #1 * CConversion`CreateUnitMatrix[type] /. {
+                          CConversion`UNITMATRIX[r_]^_        :> CConversion`UNITMATRIX[r],
+                          CConversion`UNITMATRIXCOMPLEX[r_]^_ :> CConversion`UNITMATRIXCOMPLEX[r]
+                      },
                       type, target <> "_" <> ToString[#2[[1]]]
                   ]&,
                   splitExpr