c_cpp complex.h

Posted 2021-05-14

#ifndef COMPLEX_H
#define COMPLEX_H
#include <iostream>
#include <cmath>

class Complex {
protected:
  float real;
  float imaginary;
public:
  Complex(float a, float b): real(a), imaginary(b) {}
  Complex(){
    real = 0;
    imaginary=0;
  }
  void setReal(float a);
  void setImaginary(float b);
  float getReal();
  float getImaginary();
  Complex Conjugate();

  friend std::ostream& operator<<(std::ostream& os, Complex i){
    if (std::abs(i.getImaginary())==i.getImaginary()) {
      os << i.getReal() << "+" << i.getImaginary() << "i"  << std::endl;
    } else {
      os << i.getReal() << i.getImaginary() << "i" << std::endl;
    }
    //wtf?!
      //os << i.getReal() << (std::abs(i.getImaginary())==i.getImaginary())?"+":"" << i.getImaginary() << "i" << std::endl;
    return os;
  };
  friend Complex operator+(Complex a, Complex b){
    Complex temp;
    temp.setReal(a.getReal() + b.getReal());
    temp.setImaginary(a.getImaginary() + b.getImaginary());
    return temp;
  };
  friend Complex operator-(Complex a, Complex b){
    Complex temp;
    temp.setReal(a.getReal() - b.getReal());
    temp.setImaginary(a.getImaginary() - b.getImaginary());
    return temp;
  };
  friend Complex operator*(Complex a, Complex b){
    Complex temp;
    temp.setReal(a.getReal() * b.getReal() - a.getImaginary() * b.getImaginary());
    temp.setImaginary(a.getReal() * b.getImaginary() + a.getImaginary() * b.getReal());
    return temp;
  };
  friend Complex operator/(Complex a, float i){
    Complex temp;
    temp.setReal(a.getReal() / i);
    temp.setImaginary(a.getImaginary() / i);
    return temp;
  };
  friend Complex operator/(Complex a, Complex i){
    Complex temp;
    temp = a * i.Conjugate();
    temp = temp / (pow(i.getReal(),2) + pow(i.getImaginary(),2));
    return temp;
  };

};



#endif //COMPLEX_H