Cmpx.cu

#include <cmath>
#include <assert.h>
#include "Cmpx.hpp"

// constructor
HOSTDEVICE
Cmpx::Cmpx() {
    re = 0;
    im = 0;
}

HOSTDEVICE
Cmpx::Cmpx(const fptype x, const fptype y) {
    re = x;
    im = y;
}

// Cmpx& Cmpx::set_re(const fptype x) {
    // re = x;
    // return *this;
// }

// Cmpx& Cmpx::set_im(const fptype x) {
    // im = x;
    // return *this;
// }

// Cmpx Cmpx::conjugate() const {
    // return Cmpx(re, -im);
// }

HOSTDEVICE
fptype Cmpx::get_mag() const {
    return sqrtf(re*re + im*im);
}

HOSTDEVICE
fptype Cmpx::get_ang() const {
    return atan2f(im, re);
}

// Cmpx& Cmpx::operator+=(const Cmpx &z) {
HOSTDEVICE
void Cmpx::operator+=(const Cmpx &z) {
    re += z.re;
    im += z.im;
    // return *this;
}

// Cmpx& Cmpx::operator-=(const Cmpx &z) {
    // re -= z.re;
    // im -= z.im;
    // return *this;
// }

// Cmpx& Cmpx::operator*=(const Cmpx &z) {
HOSTDEVICE
void Cmpx::operator*=(const Cmpx &z) {
    fptype x = re*z.re - im*z.im;
    fptype y = re*z.im + im*z.re;
    re = x;
    im = y;
    // return *this;
}

// Cmpx& Cmpx::operator*=(const fptype k) {
HOSTDEVICE
void Cmpx::operator*=(const fptype k) {
    re *= k;
    im *= k;
    // return *this;
}

HOST
char* Cmpx::polar() const {
    char *ch = new char[100];
    sprintf(ch, "%g<%gr", get_mag(), get_ang()); return ch;
}

HOST
char* Cmpx::cartesian() const {
    char *ch = new char[100];
    sprintf(ch, "%g+%gj", re, im); return ch;
}