Complex_Number_H

#ifndef Complex_Number_H
#define Complex_Number_H
#include <iostream>
#include <string>
#include <sstream>
using namespace std;

class ComplexNumber
{
private:
	double real,imag;
public:
	ComplexNumber();
	ComplexNumber(double x,double y);
	void display(string name);
	void display();
	void sets(double x,double y);
	void gets(double& x,double& y);
	void operator=(ComplexNumber other);
	void operator=(string str);
	ComplexNumber operator+(ComplexNumber& other);
	ComplexNumber operator-(ComplexNumber& other);
	ComplexNumber operator*(ComplexNumber& other);
	bool operator==(ComplexNumber& c1);
	friend ostream &operator<<(ostream& output,const ComplexNumber& C)
	{
		if(C.real!=0&&C.imag>0)
		{
			output << C.real << \" + \";
			if(C.imag!=1)output << C.imag;
			output << \"i\" << endl;
		}
		else if(C.real==0&&C.imag!=0)
		{
			if(C.imag!=1&&C.imag!=-1)output << C.imag;
			else if(C.imag==-1)output<<\"-\";
			output << \"i\" << endl;
		}
		else if(C.real==0&&C.imag==0)
		{
			output << \"0\" << endl;
		}
		else if(C.real!=0&&C.imag==0)
		{
			output << C.real << endl;
		} 
		else
		{
			output << C.real << \" - \";
			if(C.imag!=-1)output << -1*C.imag;
			output << \"i\" << endl;
		}
	}
};

ComplexNumber::ComplexNumber()
{
	real = 0;
	imag = 0;
}

ComplexNumber::ComplexNumber(double x,double y)
{
	real = x;
	imag = y;
}

void ComplexNumber::gets(double& x,double& y)
{
	x = real;
	y = imag;
}

void ComplexNumber::sets(double x,double y)
{
	real = x;
	imag = y;
}

ComplexNumber ComplexNumber::operator+(ComplexNumber& other)
{
	double otherreal,otherimag;
	other.gets(otherreal,otherimag);
	ComplexNumber ret(real + otherreal,imag + otherimag);
	return ret;
}

ComplexNumber ComplexNumber::operator-(ComplexNumber& other)
{
	double otherreal,otherimag;
	other.gets(otherreal,otherimag);
	ComplexNumber ret(real - otherreal,imag - otherimag);
	return ret;
}

ComplexNumber ComplexNumber::operator*(ComplexNumber& other)
{
	double otherreal,otherimag;
	other.gets(otherreal,otherimag);
	ComplexNumber ret(real * otherreal - imag * otherimag,real * otherimag + imag * otherreal);
	return ret;
}

void ComplexNumber::operator=(ComplexNumber other)
{
	double otherreal,otherimag;
	other.gets(otherreal,otherimag);
	real = otherreal;
	imag = otherimag;
}

void ComplexNumber::operator=(string str)
{
	string refined;
	for(int i = 0;i < str.size();i++)
	{
		if(str[i] != \' \')refined+=str[i]; 
	}

	stringstream ss(refined);
	if(refined.find(\"i\") == -1)
	{
		ss >> real;
		imag = 0;
	}
	else if(refined.find(\"+\") == -1 && refined.find(\"-\") == -1)
	{
		if(!(ss >> imag))imag = 1;
		real = 0;
	}
	else if(refined.find(\"-\") == -1)
	{
		char buf;
		ss >> real >> buf;
		if(!(ss >> imag))imag = 1;
	}
	else if(refined.find(\"+\") == -1)
	{
		if(refined.find(\"-\")==0)
		{
			if(!(ss >> imag))imag = -1;
			real = 0;
		}
		else
		{
			char buf;
			double temp;
			ss >> real >> buf;
			if(!(ss >> temp))imag = -1;
			else imag = -1 * temp;
		}
	}
}

bool ComplexNumber::operator==(ComplexNumber& c1)
{
	double x1,y1;
	c1.gets(x1,y1);
	if(x1 == real && y1 == imag)return true;
	return false;
}

void ComplexNumber::display()
{
	if(real!=0&&imag>0)
	{
		cout << real << \" + \";
		if(imag!=1)cout << imag;
		cout << \"i\" << endl;
	}
	else if(real==0&&imag!=0)
	{
		if(imag!=1&&imag!=-1)cout << imag;
		else if(imag==-1)cout<<\"-\";
		cout << \"i\" << endl;
	}
	else if(real==0&&imag==0)
	{
		cout << \"0\" << endl;
	}
	else if(real!=0&&imag==0)
	{
		cout << real << endl;
	} 
	else
	{
		cout << real << \" - \";
		if(imag!=-1)cout << -1*imag;
		cout << \"i\" << endl;
	}
}

void ComplexNumber::display(string name)
{
	if(real!=0&&imag>0)
	{
		cout << name << \" is \" << real << \" + \";
		if(imag!=1)cout << imag;
		cout << \"i\" << endl;
	}
	else if(real==0&&imag!=0)
	{
		cout << name << \" is \";

		if(imag!=1&&imag!=-1)cout << imag;
		else if(imag==-1)cout<<\"-\";
		cout << \"i\" << endl;
	}
	else if(real==0&&imag==0)
	{
		cout << name << \" is \" << \"0\" << endl;
	}
	else if(real!=0&&imag==0)
	{
		cout << name << \" is \" << real << endl;
	} 
	else
	{
		cout << name << \" is \" << real << \" - \";
		if(imag!=-1)cout << -1*imag;
		cout << \"i\" << endl;
	}
}



#endif