First commit

First operations implemented

Author: omdxp

cmpx.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Tue Sep 10 17:36:25 2019
+
+@author: Omar Belghaouti
+"""
+from math import sqrt
+# Complex class for complex number manipulations
+class Complex():
+    # Constructor
+    def __init__(self, re=0, im=0):
+        self.re = re
+        self.im = im
+    # Operator overloading 1 : +
+    def __add__(self, other):
+        return Complex(self.re + other.re, self.im + other.im)
+    # Operator overloading 2 : -
+    def __sub__(self, other):
+        return Complex(self.re - other.re, self.im - other.im)
+    # Operator overloading 3 : *
+    def __mul__(self, other):
+        return Complex(self.re * other.re - self.im * other.im, self.re * other.im + self.im * other.re)
+    # Operator overloading 4 : /
+    def __truediv__(self, other):
+        try:
+            den = other * other.con()
+            num = self * other.con()
+            return Complex(num.re / den.re, num.im / den.re)
+        except ZeroDivisionError as err:
+            print('Error: {}'.format(err))
+    # Operator overloading 5 : //
+    def __floordiv__(self, other):
+        try:
+            den = other * other.con()
+            num = self * other.con()
+            return Complex(num.re // den.re, num.im // den.re)
+        except ZeroDivisionError as err:
+            print('Error: {}'.format(err))
+    def __gt__(self, other):
+        return self.mod() > other.mod()
+    # Operator overloading 7 : >=
+    def __ge__(self, other):
+        return self.mod() >= other.mod()
+    # Operator overloading 8: <
+    def __lt__(self, other):
+        return not self >= other
+    # Operator overloading 9: <=
+    def __le__(self, other):
+        return not self > other
+    # Operator overloading 10: ==
+    def __eq__(self, other):
+        return (self.re == other.re) and (self.im == other.im)
+    # Operator overloading 11: !=
+    def __ne__(self, other):
+        return not self == other
+    # Operator overloading 12: +=
+    def __iadd__(self, other):
+        self.re += other.re
+        self.im += other.im
+    # Operator overloading 13: -=
+    def __isub__(self, other):
+        self.re -= other.re
+        self.im -= other.im
+    # Operator overloading 14: *=
+    def __imul__(self, other):
+        self.re = self.re * other.re - self.im * other.im
+        self.im = self.re * other.im + self.im * other.re
+    # Operator overloading 15: /=
+    def __idiv__(self, other):
+        try:
+            den = other * Complex(other.re, - other.im)
+            num = self * Complex(other.re, - other.im)
+            self.re = num.re / den.re
+            self.im = num.im / den.re
+        except ZeroDivisionError as err:
+            print('Error: {}'.format(err))
+    # Operator overloading 16: //=
+    def __ifloordiv__(self, other):
+        try:
+            den = other * Complex(other.re, - other.im)
+            num = self * Complex(other.re, - other.im)
+            self.re = num.re // den.re
+            self.im = num.im // den.re
+        except ZeroDivisionError as err:
+            print('Error: {}'.format(err))
+    ## Helper functions
+    # Module function to calculate the modulus of a complex number
+    def mod(self):
+        return sqrt(self.re**2 + self.im**2)
+    # Conjugated of a complex number
+    def con(self):
+        return Complex(self.re, - self.im)
+    # String function
+    def __str__(self):
+        output = str(self.re) + ' + ' + str(self.im) + 'j' if(self.im != 0) else str(self.re)
+        output = str(self.re) + ' - ' + str(-self.im) + 'j' if (self.im < 0) else output
+        return output
+    # Callable module
+    def __call__(self):
+        return self

main.py

@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Tue Sep 10 17:37:40 2019
+
+@author: Omar Belghaouti
+"""
+from cmpx import Complex
+
+def main():
+    # Instanciation 1
+    number = Complex() # --> Real = 0, Imaginary = 0
+    print(number)
+    # Instanctiation 2
+    number = Complex(42) # --> Real = 42, Imaginary = 0
+    print(number)
+    # Instanciation 3
+    number = Complex(12, -3.2) # --> Real = 12, Imaginary = -1
+    print(number)
+    # Instanciation 4
+    number = Complex(im=13.2, re=5)
+    print(number)
+    
+    # Basic operations
+    num1 = Complex(re=3, im=-2.5)
+    num2 = Complex(re=4.2, im=13.2)
+    # Summation
+    print('({}) + ({}) = {}'.format(num1, num2, num1 + num2))
+    # Substraction
+    print('({}) - ({}) = {}'.format(num1, num2, num1 - num2))
+    # Multiplication
+    print('({}) * ({}) = {}'.format(num1, num2, num1 * num2))
+    # Division
+    print('({}) / ({}) = {}'.format(num1, num2, num1 / num2))
+    # Floor division
+    print('({}) // ({}) = {}'.format(num1, num2, num1 // num2))
+    # Congugated of a complex number
+    print('con({}) = {}'.format(num1, num1.con()))
+    # Module of a complex number
+    print('mod({}) = {}'.format(num1, num1.mod()))
+    
+    # Basic comparisons
+    num1 = Complex(im=3.1, re=-5)
+    num2 = Complex(im=-1.5, re=0.6)
+    # Greater than
+    if(num1 > num2): print('({}) > ({})'.format(num1, num2))
+    # Greater or equal
+    if(num1 >= num2): print('({}) >= ({})'.format(num1, num2))
+    # Less than
+    if(num1 < num2): print('({}) < ({})'.format(num1, num2))
+    # Less or equal
+    if(num1 <= num2): print('({}) <= ({})'.format(num1, num2))
+    # Equal
+    if(num1 == num2): print('({}) == ({})'.format(num1, num2))
+    # Not equal
+    if(num1 != num2): print('({}) != ({})'.format(num1, num2))
+    
+if __name__ == '__main__': main()