# ---------------------------------------------
# Your Name
# Joy and Beauty of Data, Complex Number Assignment
# Last Updated: Month Day, Year
# ---------------------------------------------
# A brief description of the assignment goes here.
# ---------------------------------------------


# -------- Complex Number Class ---------------
# 
# Your solution goes here
#
#--------- End of Complex Number Class --------


# ---------------------------------------------
# Print a menu of choices
# ---------------------------------------------

def printMenu():
    print()
    print("1. Enter Complex Number 1.")
    print("2. Enter Complex Number 2.")
    print("3. Add Complex Numbers 1 and 2.")
    print("4. Multiply Complex Numbers 1 and 2.")
    print("5. Divide Complex Number 1 by Complex Number 2.")
    print("6. Divide Complex Number 2 by Complex Number 1.")
    print("7. Quit.")
    print()

# ---------------------------------------------
# Enable the user to create a Complex Number
# ---------------------------------------------

def createComplexNumber(complex_number):
    real = float(input("Enter the complex number's real part: "))
    complex_number.setReal(real)
    imaginary = float(input("Enter the complex number's imaginary part: ")) 
    complex_number.setImaginary(imaginary)

# ---------------------------------------------
# main program
# ---------------------------------------------

def main():

    print("Complex Number Assignment")
    print("-------------------------")

    complex_number_1 = ComplexNumber()  # defaults to 0 + 0i
    complex_number_2 = ComplexNumber()  # defaults to 0 + 0i

    choice = 0

    while choice != 7:                  # 7 is the quit option
        printMenu()
        choice = int(input("Enter choice: "))
        if choice == 1:
            createComplexNumber(complex_number_1)
            print("Complex Number 1: ", complex_number_1)
        elif choice == 2:
            createComplexNumber(complex_number_2)
            print("Complex Number 2: ", complex_number_2)
        elif choice == 3:
            print("Result of Addition: ", complex_number_1 + complex_number_2)
        elif choice == 4:
            print("Result of Multiplication: ", complex_number_1 * complex_number_2)
        elif choice == 5:
            print("Result of Division: ", complex_number_1 / complex_number_2)
        elif choice == 6:
            print("Result of Division: ", complex_number_2 / complex_number_1)
        elif choice == 7:
            print("Goodbye!")
        else:
            print("Invalid choice.  Please try again.")

# ---------------------------------------------

main()