complex.hpp

Go to the documentation of this file.

#ifndef COMPLEX_HPP
#define COMPLEX_HPP

#include "space_settings.hpp"

struct Complex {
    double real;
    double imag;

    // Constructor to initialize the complex numbers
    // Initializes to zero
    // Initializes to real if provided one argument
    KOKKOS_INLINE_FUNCTION Complex(double r = 0.0, double i = 0.0) : real(r), imag(i) {}

    // Overload the + operator
    KOKKOS_INLINE_FUNCTION Complex operator+(const Complex& other) const {
        return Complex(real + other.real, imag + other.imag);
    }

    // Overload the * operator
    KOKKOS_INLINE_FUNCTION Complex operator*(const Complex& other) const {
        return Complex(real * other.real - imag * other.imag, 
                       real * other.imag + imag * other.real);
    }

    // Overload the * operator
    KOKKOS_INLINE_FUNCTION Complex operator*(double scalar) const {
        return Complex(real * scalar, imag * scalar); 
    }

    // Multiply two complex numbers, but keep only the real component
    KOKKOS_INLINE_FUNCTION static double r_mult(const Complex& a, const Complex& b) {
        return a.real * b.real - a.imag * b.imag;
    }

    KOKKOS_INLINE_FUNCTION static Complex expc(const Complex& c) {
        double exp_real = exp(c.real);
        return Complex(exp_real * cos(c.imag), exp_real * sin(c.imag));
    }

    // exp of Complex(0.0, imag_in)
    KOKKOS_INLINE_FUNCTION static Complex exp_i(double imag_in) {
        return Complex(cos(imag_in), sin(imag_in));
    }
};

#endif