第31课 完善的复数类

#ifndef COMPLEX_H
#define COMPLEX_H

class Complex
{
 private:
    int a;
    int b;

 public:
    Complex(double a=0, double b=0);
    double GetA();
    double GetB();
    double GetModulus();

    Complex operator + (const Complex& p);
    Complex operator - (const Complex& p);
    Complex operator * (const Complex& p);
    Complex operator / (const Complex& p);

    bool operator == (const Complex& p);
    bool operator != (const Complex& p);

    Complex& operator = (const Complex& p);


};
#endif // COMPLEX_H

#include "complex.h"
#include "math.h"

Complex::Complex(double a, double b)
{
    this->a = a;
    this->b = b;
}

double Complex::GetA()
{
    return a;
}

double Complex::GetB()
{
    return b;
}

double Complex::GetModulus()
{
    return sqrt(a * a + b * b);
}


Complex Complex::operator +(const Complex& p)
{
   double na = a + p.a;
   double nb = b + p.b;
   Complex ret(na,nb);

   return ret;

}

Complex Complex::operator -(const Complex& p)
{
    double na = a - p.a;
    double nb = b - p.b;
    Complex ret(na,nb);

    return ret;

}

Complex Complex::operator *(const Complex& p)
{
   // c1 = a + bi
   // c2 = c + di
   // c1 * c2 = (a*c -b*d) +(b*c+ a*d)

    double na = (a * p.a) - (b * p.b);
    double nb = (b * p.a) + (a * p.b);
    Complex ret(na,nb);

    return ret;
}

Complex Complex::operator /(const Complex& p)
{

    double cm = p.a * p.a + p.b * p.b;
    double na = (a * p.a + b * p.b) / cm;
    double nb = (b * p.a - a * p.b) / cm;
    Complex ret(na,nb);

    return ret;
}

bool Complex::operator ==(const Complex& p)
{
    return (a == p.a) && (b == p.b);
}

bool Complex::operator !=(const Complex& p)
{
    return !(*this == p);
}

Complex& Complex::operator =(const Complex& p)
{
    if(this != &p)
    {
        a = p.a;
        b = p.b;
    }

    return *this;
}

#include <stdio.h>
#include "complex.h"

int main()
{
    Complex c1(1,2);
    Complex c2(3,6);

    Complex c3 = c2 - c1;
    Complex c4 = c1 * c3;
    Complex c5 = c2 / c1;

    printf("c3.a = %f,c3.b = %f\n",c3.GetA(),c3.GetB());
    printf("c4.a = %f,c4.b = %f\n",c4.GetA(),c4.GetB());
    printf("c5.a = %f,c5.b = %f\n",c5.GetA(),c5.GetB());

    Complex c6(2,4);
    printf("c3 == c6: %d\n",c3 == c6);
    printf("c3 != c4: %d\n",c3 != c4);

    (c3 = c2) = c1;
    printf("c3.a = %f,c3.b = %f\n",c3.GetA(),c3.GetB());
    printf("c2.a = %f,c2.b = %f\n",c2.GetA(),c2.GetB());
    printf("c1.a = %f,c1.b = %f\n",c1.GetA(),c1.GetB());


    return 0;
}