ece650-cpp/complex.hpp at master · eceuwaterloo/ece650-cpp · GitHub

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
#pragma once

// An example of a complex number
// See complex_main.cpp for compilation instructions

#include <cmath>
class Complex {
private:
  double re, im;

public:
  Complex(double = 0.0, double = 0.0);

  /// Accessors

  double real() const;
  double imag() const;
  double abs() const;
  Complex exp() const;

  /// Mutators
  void normalize();
};

// Constructor
inline Complex::Complex(double r, double i)
    : re(r), im(i) { /*empty constructor*/
}

// return the real component
inline double Complex::real() const { return re; }

// return the imaginary component
inline double Complex::imag() const { return im; }

// return the absolute value
inline double Complex::abs() const { return std::sqrt(re * re + im * im); }

// Return the exponential of the complex value
inline Complex Complex::exp() const {
  double exp_re = std::exp(re);

  return Complex(exp_re * std::cos(im), exp_re * std::sin(im));
}

// Normalize the complex number (giving it unit absolute value, |z| = 1)
inline void Complex::normalize() {
  if (re == 0 && im == 0) {
    return;
  }

  double absval = abs();
  re /= absval;
  im /= absval;
}