Use inplace_[neg/abs/conj] if applicable in [arb/acb/fmpq]_mat

Author: albinahlback

src/acb_mat/conjugate.c

@@ -16,13 +16,18 @@ acb_mat_conjugate(acb_mat_t B, const acb_mat_t A)
 {
     slong i, j;
 
-    if ((acb_mat_nrows(B) != acb_mat_nrows(A)) ||
-        (acb_mat_ncols(B) != acb_mat_ncols(A)))
+    if (A == B)
+        for (i = 0; i < acb_mat_nrows(A); i++)
+            for (j = 0; j < acb_mat_ncols(A); j++)
+                acb_inplace_conj(acb_mat_entry(A, i, j));
+    else
     {
-        flint_throw(FLINT_ERROR, "acb_mat_conjugate: incompatible dimensions.\n");
-    }
+        if ((acb_mat_nrows(B) != acb_mat_nrows(A)) ||
+                (acb_mat_ncols(B) != acb_mat_ncols(A)))
+            flint_throw(FLINT_ERROR, "acb_mat_conjugate: incompatible dimensions.\n");
 
-    for (i = 0; i < acb_mat_nrows(A); i++)
-        for (j = 0; j < acb_mat_ncols(A); j++)
-            acb_conj(acb_mat_entry(B, i, j), acb_mat_entry(A, i, j));
+        for (i = 0; i < acb_mat_nrows(A); i++)
+            for (j = 0; j < acb_mat_ncols(A); j++)
+                acb_conj(acb_mat_entry(B, i, j), acb_mat_entry(A, i, j));
+    }
 }

src/acb_mat/neg.c

@@ -16,8 +16,13 @@ acb_mat_neg(acb_mat_t dest, const acb_mat_t src)
 {
     slong i, j;
 
-    for (i = 0; i < acb_mat_nrows(src); i++)
-        for (j = 0; j < acb_mat_ncols(src); j++)
-            acb_neg(acb_mat_entry(dest, i, j),
-                acb_mat_entry(src, i, j));
+    if (dest == src)
+        for (i = 0; i < acb_mat_nrows(src); i++)
+            for (j = 0; j < acb_mat_ncols(src); j++)
+                acb_inplace_neg(acb_mat_entry(dest, i, j));
+    else
+        for (i = 0; i < acb_mat_nrows(src); i++)
+            for (j = 0; j < acb_mat_ncols(src); j++)
+                acb_neg(acb_mat_entry(dest, i, j),
+                        acb_mat_entry(src, i, j));
 }

src/arb_mat/neg.c

@@ -16,8 +16,13 @@ arb_mat_neg(arb_mat_t dest, const arb_mat_t src)
 {
     slong i, j;
 
-    for (i = 0; i < arb_mat_nrows(src); i++)
-        for (j = 0; j < arb_mat_ncols(src); j++)
-            arb_neg(arb_mat_entry(dest, i, j),
-                arb_mat_entry(src, i, j));
+    if (dest == src)
+        for (i = 0; i < arb_mat_nrows(src); i++)
+            for (j = 0; j < arb_mat_ncols(src); j++)
+                arb_inplace_neg(arb_mat_entry(dest, i, j));
+    else
+        for (i = 0; i < arb_mat_nrows(src); i++)
+            for (j = 0; j < arb_mat_ncols(src); j++)
+                arb_neg(arb_mat_entry(dest, i, j),
+                        arb_mat_entry(src, i, j));
 }

src/fmpq_mat/neg.c

@@ -16,8 +16,13 @@ void fmpq_mat_neg(fmpq_mat_t rop, const fmpq_mat_t op)
 {
     slong i, j;
 
-    for (i = 0; i < op->r; i++)
-        for (j = 0; j < op->c; j++)
-            fmpq_neg(fmpq_mat_entry(rop, i, j),
-                     fmpq_mat_entry(op, i, j));
+    if (rop == op)
+        for (i = 0; i < op->r; i++)
+            for (j = 0; j < op->c; j++)
+                fmpq_inplace_neg(fmpq_mat_entry(rop, i, j));
+    else
+        for (i = 0; i < op->r; i++)
+            for (j = 0; j < op->c; j++)
+                fmpq_neg(fmpq_mat_entry(rop, i, j),
+                        fmpq_mat_entry(op, i, j));
 }