Added conjugate function conj().

This commit is contained in:
Jim Monte 2019-12-15 14:51:37 -05:00 committed by Holger Vogt
parent 7681f34664
commit 76cf00853c
4 changed files with 75 additions and 31 deletions

View File

@ -161,6 +161,7 @@ struct func ft_funcs[] = {
{ "re", cx_real },
{ "imag", cx_imag },
{ "im", cx_imag },
{ "conj", cx_conj },
{ "db", cx_db },
{ "log", cx_log },
{ "log10", cx_log10 },

View File

@ -60,6 +60,7 @@ extern void *cx_unwrap(void *, short int , int , int *, short int *);
extern void *cx_j(void *, short int , int , int *, short int *);
extern void *cx_real(void *, short int , int , int *, short int *);
extern void *cx_imag(void *, short int , int , int *, short int *);
extern void *cx_conj(void *, short int , int , int *, short int *);
extern void *cx_pos(void *, short int , int , int *, short int *);
extern void *cx_db(void *, short int , int , int *, short int *);
extern void *cx_log10(void *, short int , int , int *, short int *);

View File

@ -117,68 +117,109 @@ cx_unwrap(void *data, short int type, int length, int *newlength, short int *new
}
/* If this is pure imaginary we might get real, but never mind... */
void *
cx_j(void *data, short int type, int length, int *newlength, short int *newtype)
void *cx_j(void *data, short int type, int length, int *newlength,
short int *newtype)
{
ngcomplex_t *c = alloc_c(length);
ngcomplex_t *cc = (ngcomplex_t *) data;
double *dd = (double *) data;
int i;
*newlength = length;
*newtype = VF_COMPLEX;
if (type == VF_COMPLEX)
if (type == VF_COMPLEX) {
ngcomplex_t *cc = (ngcomplex_t *) data;
int i;
for (i = 0; i < length; i++) {
realpart(c[i]) = - imagpart(cc[i]);
realpart(c[i]) = -imagpart(cc[i]);
imagpart(c[i]) = realpart(cc[i]);
}
else
}
else {
double *dd = (double *) data;
int i;
for (i = 0; i < length; i++) {
imagpart(c[i]) = dd[i];
/* Real part is already 0. */
}
return ((void *) c);
}
return (void *) c;
}
void *
cx_real(void *data, short int type, int length, int *newlength, short int *newtype)
void *cx_real(void *data, short int type, int length, int *newlength,
short int *newtype)
{
double *d = alloc_d(length);
double *dd = (double *) data;
ngcomplex_t *cc = (ngcomplex_t *) data;
int i;
*newlength = length;
*newtype = VF_REAL;
if (type == VF_COMPLEX)
for (i = 0; i < length; i++)
if (type == VF_COMPLEX) {
ngcomplex_t *cc = (ngcomplex_t *) data;
int i;
for (i = 0; i < length; i++) {
d[i] = realpart(cc[i]);
else
for (i = 0; i < length; i++)
}
}
else {
double *dd = (double *) data;
int i;
for (i = 0; i < length; i++) {
d[i] = dd[i];
return ((void *) d);
}
}
return (void *) d;
}
void *
cx_imag(void *data, short int type, int length, int *newlength, short int *newtype)
void *cx_imag(void *data, short int type, int length, int *newlength,
short int *newtype)
{
double *d = alloc_d(length);
double *dd = (double *) data;
ngcomplex_t *cc = (ngcomplex_t *) data;
int i;
*newlength = length;
*newtype = VF_REAL;
if (type == VF_COMPLEX)
for (i = 0; i < length; i++)
if (type == VF_COMPLEX) {
ngcomplex_t *cc = (ngcomplex_t *) data;
int i;
for (i = 0; i < length; i++) {
d[i] = imagpart(cc[i]);
else
for (i = 0; i < length; i++)
}
}
else {
double *dd = (double *) data;
int i;
for (i = 0; i < length; i++) {
d[i] = dd[i];
return ((void *) d);
}
}
return (void *) d;
}
/* Create complex conjugate of data */
void *cx_conj(void *data, short int type, int length,
int *p_newlength, short int *p_newtype)
{
/* Length and type do not change */
*p_newlength = length;
*p_newtype = type;
/* For complex, copy with conjugation */
if (type == VF_COMPLEX) {
ngcomplex_t * const c_dst = alloc_c(length);
ngcomplex_t *c_dst_cur = c_dst;
ngcomplex_t *c_src_cur = (ngcomplex_t *) data;
ngcomplex_t * const c_src_end = c_src_cur + length;
for ( ; c_src_cur < c_src_end; c_src_cur++, c_dst_cur++) {
c_dst_cur->cx_real = c_src_cur->cx_real;
c_dst_cur->cx_imag = -c_src_cur->cx_imag;
}
return (void *) c_dst;
}
/* Else real, so just copy */
return memcpy(alloc_d(length), data, length * sizeof(double));
} /* end of function cx_conj */
/* This is obsolete... */
void *

View File

@ -14,6 +14,7 @@ void * cx_unwrap(void *data, short int type, int length, int *newlength, short i
void * cx_j(void *data, short int type, int length, int *newlength, short int *newtype);
void * cx_real(void *data, short int type, int length, int *newlength, short int *newtype);
void * cx_imag(void *data, short int type, int length, int *newlength, short int *newtype);
void * cx_conj(void *data, short int type, int length, int *newlength, short int *newtype);
void * cx_pos(void *data, short int type, int length, int *newlength, short int *newtype);
void * cx_db(void *data, short int type, int length, int *newlength, short int *newtype);
void * cx_log10(void *data, short int type, int length, int *newlength, short int *newtype);