about summary refs log tree commit diff
path: root/lang/cpptour/mycomplex.h
diff options
context:
space:
mode:
Diffstat (limited to 'lang/cpptour/mycomplex.h')
-rw-r--r--lang/cpptour/mycomplex.h15
1 files changed, 15 insertions, 0 deletions
diff --git a/lang/cpptour/mycomplex.h b/lang/cpptour/mycomplex.h
new file mode 100644
index 0000000..b7440ee
--- /dev/null
+++ b/lang/cpptour/mycomplex.h
@@ -0,0 +1,15 @@
+class complex {
+  double re, im;
+public:
+  complex (double r, double i) : re {r}, im {i} {}
+  complex (double r) : re {r}, im {0.0} {}
+  complex () : re {0.0}, im {0.0} {}
+
+  double real () const { return re; }
+  void real (double r) { re = r; }
+  double imag () const { return im; }
+  void imag (double i) { im = i; }
+
+  complex& operator+= (complex z) { re += z.re, im += z.im; return *this; }
+  complex& operator-= (complex z) { re -= z.re, im -= z.im; return *this; }
+};