1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// CXSparse/Source/cs_convert: convert between real and complex
// CXSparse, Copyright (c) 2006-2022, Timothy A. Davis. All Rights Reserved.
// SPDX-License-Identifier: LGPL-2.1+
#include "cs.h"
/* convert from complex to real (int version) */
/* C = real(A) if real is true, imag(A) otherwise */
cs_di *cs_i_real (cs_ci *A, int real)
{
cs_di *C ;
int n, triplet, nn, p, nz, *Ap, *Ai, *Cp, *Ci ;
cs_complex_t *Ax ;
double *Cx ;
if (!A || !A->x) return (NULL) ; /* return if A NULL or pattern-only */
n = A->n ; Ap = A->p ; Ai = A->i ; Ax = A->x ;
triplet = (A->nz >= 0) ; /* true if A is a triplet matrix */
nz = triplet ? A->nz : Ap [n] ;
C = cs_di_spalloc (A->m, n, A->nzmax, 1, triplet) ;
if (!C) return (NULL) ;
Cp = C->p ; Ci = C->i ; Cx = C->x ;
nn = triplet ? nz : (n+1) ;
for (p = 0 ; p < nz ; p++) Ci [p] = Ai [p] ;
for (p = 0 ; p < nn ; p++) Cp [p] = Ap [p] ;
for (p = 0 ; p < nz ; p++) Cx [p] = real ? creal (Ax [p]) : cimag (Ax [p]) ;
if (triplet) C->nz = nz ;
return (C) ;
}
/* convert from real to complex (int version) */
/* C = A if real is true, or C = i*A otherwise */
cs_ci *cs_i_complex (cs_di *A, int real)
{
cs_ci *C ;
int n, triplet, nn, p, nz, *Ap, *Ai, *Cp, *Ci ;
double *Ax ;
cs_complex_t *Cx ;
if (!A || !A->x) return (NULL) ; /* return if A NULL or pattern-only */
n = A->n ; Ap = A->p ; Ai = A->i ; Ax = A->x ;
triplet = (A->nz >= 0) ; /* true if A is a triplet matrix */
nz = triplet ? A->nz : Ap [n] ;
C = cs_ci_spalloc (A->m, n, A->nzmax, 1, triplet) ;
if (!C) return (NULL) ;
Cp = C->p ; Ci = C->i ; Cx = C->x ;
nn = triplet ? nz : (n+1) ;
for (p = 0 ; p < nz ; p++) Ci [p] = Ai [p] ;
for (p = 0 ; p < nn ; p++) Cp [p] = Ap [p] ;
for (p = 0 ; p < nz ; p++) Cx [p] = real ? Ax [p] : (I * Ax [p]) ;
if (triplet) C->nz = nz ;
return (C) ;
}
/* convert from complex to real (int64_t version) */
/* C = real(A) if real is true, imag(A) otherwise */
cs_dl *cs_l_real (cs_cl *A, int64_t real)
{
cs_dl *C ;
int64_t n, triplet, nn, p, nz, *Ap, *Ai, *Cp, *Ci ;
cs_complex_t *Ax ;
double *Cx ;
if (!A || !A->x) return (NULL) ; /* return if A NULL or pattern-only */
n = A->n ; Ap = A->p ; Ai = A->i ; Ax = A->x ;
triplet = (A->nz >= 0) ; /* true if A is a triplet matrix */
nz = triplet ? A->nz : Ap [n] ;
C = cs_dl_spalloc (A->m, n, A->nzmax, 1, triplet) ;
if (!C) return (NULL) ;
Cp = C->p ; Ci = C->i ; Cx = C->x ;
nn = triplet ? nz : (n+1) ;
for (p = 0 ; p < nz ; p++) Ci [p] = Ai [p] ;
for (p = 0 ; p < nn ; p++) Cp [p] = Ap [p] ;
for (p = 0 ; p < nz ; p++) Cx [p] = real ? creal (Ax [p]) : cimag (Ax [p]) ;
if (triplet) C->nz = nz ;
return (C) ;
}
/* convert from real to complex (int64_t version) */
/* C = A if real is true, or C = i*A otherwise */
cs_cl *cs_l_complex (cs_dl *A, int64_t real)
{
cs_cl *C ;
int64_t n, triplet, nn, p, nz, *Ap, *Ai, *Cp, *Ci ;
double *Ax ;
cs_complex_t *Cx ;
if (!A || !A->x) return (NULL) ; /* return if A NULL or pattern-only */
n = A->n ; Ap = A->p ; Ai = A->i ; Ax = A->x ;
triplet = (A->nz >= 0) ; /* true if A is a triplet matrix */
nz = triplet ? A->nz : Ap [n] ;
C = cs_cl_spalloc (A->m, n, A->nzmax, 1, triplet) ;
if (!C) return (NULL) ;
Cp = C->p ; Ci = C->i ; Cx = C->x ;
nn = triplet ? nz : (n+1) ;
for (p = 0 ; p < nz ; p++) Ci [p] = Ai [p] ;
for (p = 0 ; p < nn ; p++) Cp [p] = Ap [p] ;
for (p = 0 ; p < nz ; p++) Cx [p] = real ? Ax [p] : (I * Ax [p]) ;
if (triplet) C->nz = nz ;
return (C) ;
}
|
