/***************************************************************************
Program:	A2Q2.cpp


Purpose: Program to overload operators using a Date class. 
         A menu is added to test the functions.
****************************************************************************/

 


#include <iostream>
#include <math.h>
using namespace std;


bool testdate(int yeara, int montha, int daya);

class date
{
public:
date (int newday, int newmonth);
date (int newday, int newmonth, int newyear);

// int nextdate ();

int getday() const;
int getmonth() const;
int getyear() const;
void setday(int newday1);
void setmonth(int newmonth1);
void setyear(int newyear1);
//bool testdate(int datetest);
private: 
	int day;
	int month;
	int year;
};

date::date (int newday, int newmonth)  // constructor without year
		:day(newday), month(newmonth),year(2004)
{
	if ((testdate(2004,newmonth,newday))==0){
		cout<<" ERROR invalid date"<<endl;
		cout<<" please re-initlise date"<<endl;
	};

};

date::date (int newday, int newmonth, int newyear) // constructor with year
		:day(newday), month(newmonth), year(newyear)
{
	if ((testdate(newyear,newmonth,newday))==0){
		cout<<" ERROR invalid date"<<endl;
		cout<<" please re-initlise date"<<endl;
	};
};


bool operator ==(const date& date1, const date& date2)
{
	return ((date1.getday()==date2.getday())
		&&(date1.getmonth()==date2.getmonth())
		&&(date1.getyear()==date2.getyear()));
};

bool operator !=(const date& date1, const date& date2)
{
	return ((date1.getday()!=date2.getday())
		||(date1.getmonth()!=date2.getmonth())
		||(date1.getyear()!=date2.getyear()));
};

bool operator <(const date& date1, const date& date2)
{
	if (date1.getyear() < date2.getyear()){
		return 1;
	};
	
	if (date1.getyear()==date2.getyear()){
		if(date1.getmonth() < date2.getmonth()){
			return 1;
		};
			if(date1.getday()< date2.getday()){
				return 1;
			};
	
	};
	return 0;
};

bool operator <=(const date& date1, const date& date2)
{
	return ((date1<date2) || (date1 == date2));

};


bool operator >(const date& date1, const date& date2)
{
	if (date1.getyear() > date2.getyear()){
		return 1;
	};
	
	if (date1.getyear()==date2.getyear()){
		if(date1.getmonth() > date2.getmonth()){
			return 1;
		};
			if(date1.getday()> date2.getday()){
				return 1;
			};
	};

return 0;
		
};


bool operator >=(const date& date1, const date& date2)
{
	return ((date1>date2) || (date1 == date2));

};
 
ostream& operator <<(ostream& outputstream, const date& date1)
{
	
	cout<<date1.getday();

	switch(date1.getmonth())
	{
case 1:
	cout<<"-Jan-";
	break;
case 2:
	cout<<"-Feb-";
	break;
case 3:
	cout<<"-Mar-";
	break;
case 4:
	cout<<"-Apr-";
	break;
case 5:
	cout<<"-May-";
	break;
case 6:
	cout<<"-Jun-";
	break;
case 7:
	cout<<"-Jul-";
	break;
case 8:
	cout<<"-Aug-";
	break;
case 9:
	cout<<"-Sep-";
	break;
case 10:
	cout<<"-Oct-";
	break;
case 11:
	cout<<"-Nov-";
	break;
case 12:
	cout<<"-Dec-";
	break;

	};
	
	cout<<date1.getyear();

	return outputstream;
};

date addate(int addnumb, int daysinmonth, date date2)// add days to date
{

	date2.setday(date2.getday()+addnumb);

	if (date2.getday()>daysinmonth){
		date2.setday(date2.getday()-daysinmonth);
		date2.setmonth(date2.getmonth()+1);
		if (date2.getmonth()> 12) {
			date2.setmonth(1);
			date2.setyear(date2.getyear()+1);
		};
	};

	return date2;
};
	

istream& operator >>(istream& inputstream, date& date1)
{
	int correct=0;
while (correct==0){
	correct=1;
	int tempyear=1;
	int temp;
	cout<<endl;
	cout << "Enter 4 character year(ex 2004): ";
	cin >> temp;
	date1.setyear(temp);
	cout << "Enter month : ";
	cin >> temp;
	date1.setmonth(temp);
	cout << "Enter day : ";
	cin >> temp;
	date1.setday(temp);

	if ((testdate(date1.getyear(),date1.getmonth(),date1.getday()))==0){
		cout<<endl;
		cout<< "Invalid date entry, please re-enter"<<endl;
		correct=0;
	};
};
	return inputstream;
};

date operator +(date date1, int a)// add integer to date
{
	
int leapset=28;
double leapcheck=0;

while (a>364)
{
	date1.setyear(date1.getyear()+1);
	a=a-365;
};

int flag=1;
	
while (flag==1)

{
	if ((date1.getmonth()==1) || (date1.getmonth()==3) || (date1.getmonth()==5)||//check 31 day months
		(date1.getmonth()==7) || (date1.getmonth()==8)||(date1.getmonth()==10)||
		(date1.getmonth()==12)){
		date1=addate(a,31,date1);// add 'a' to 31 day months
		a=0;
		if (date1.getday()<31){
			flag=0;
		};
	};



	if ((date1.getmonth()==4) || (date1.getmonth()==6) || (date1.getmonth()==9)||//check 30 day months
	(date1.getmonth()==11)){
		date1=addate(a,30,date1);// add 'a' to 30 day months
		a=0;
		if (date1.getday()<30){
			flag=0;
		};
	};

	
	if (date1.getmonth()==2){            // february
		leapcheck=(date1.getyear());
		if (((leapcheck/4)-(ceil(leapcheck/4)))==0){   // check for leap year
			leapset=29;
		};
		date1=addate(a,leapset,date1);// add 'a' to february month
		leapset=28;// reset back to 28 days
		a=0;
		if (date1.getday()<leapset){
			flag=0;
		};
	
	};

};
	return date1;
};




void date::setday(int newday)
{
	day=newday;
};

void date::setmonth(int newmonth)
{
	month=newmonth;
};

void date::setyear(int newyear)
{
	year=newyear;
};


int date::getday() const
{
	return day;
};

int date::getmonth() const
{
	return month;
};

int date::getyear() const
{
	return year;
};

bool testdate(int testyear, int testmonth, int testday)
{
	double leapcheck=testyear;
	int leapset =28;

	if (testyear<=0){
		return 0;
	};
	if ((testmonth<=0) || (testmonth >12)){
		return 0;
	}
	 if ((testmonth==1) || (testmonth==3) || (testmonth==5)||// 31 day months
		(testmonth==7) || (testmonth==8)||(testmonth==10)||
		(testmonth==12)){
		 if ((testday<=0)||(testday>31)){
			 return 0;
		 };
		 return 1;
	 };

	if ((testmonth==4) || (testmonth==6)||// 30 day months
		(testmonth==9) || (testmonth==11)){
		 if ((testday<=0)||(testday>30)){
			 return 0;
		 };
		 return 1;
	 };
	if ((testmonth==2)){
		if  (((leapcheck/4)-(ceil(leapcheck/4)))==0){   // check for leap year
			leapset=29;
		};
		if ((testday<=0)|| (testday>leapset)){
			return 0;
		};


	return 1;
	};
	return 1;
};

int main ()
{


// Input two dates, Date1 and Date 2///

	date somedate1(1,1);// test initialisation of only day and month, year defaults to 2004
	date somedate2(1,1,1);
	date olddate(1,1,1);
	int dayadd=100;
	cout<< "This program performs several operator functions on two dates of the year"<<endl;
	cout<<endl;

	cout<< "Please enter Date 1:"<<endl;
		cin>>somedate1;// test overload function >>
		cout<<endl;

	cout<< "Please enter Date 2:"<<endl;
		cin>>somedate2;
		cout<<endl;

		cout<< "Date 1 =: "<< somedate1<<endl; // test overload function <<
		cout<< "Date 2 =: "<< somedate2<<endl;



// Make menu for function selection 
		
int select=1;

while ((select > 0) && (select <8))
{
	cout<<endl;
	cout<< " Enter (1) to check if Date 1 < Date 2"<<endl;
	cout<< " Enter (2) to check if Date 1 > Date 2"<<endl;
	cout<< " Enter (3) to check if Date 1 <= Date 2"<<endl;
	cout<< " Enter (4) to check if Date 1 >= Date 2"<<endl;
	cout<< " Enter (5) to check if Date 1 == Date 2"<<endl;
	cout<< " Enter (6) to check if Date 1 != Date 2"<<endl;
	cout<< " Enter (7) to add days to Date 1 and store in Date2"<<endl;
	cout<< " Enter (0) to EXIT"<<endl;
	cin>> select;

switch (select)
{

	case 1:// Check if Date 1 < Date 2

		cout<<"The satement "<<somedate1<<" < "<<somedate2;
		if (somedate1 < somedate2){
			cout<<" is TRUE"<<endl;		
		}
		else{
			cout<< "is FALSE"<<endl;
		};

	break;

	case 2:// Check if Date 1 > Date 2

		cout<<"The satement "<<somedate1<<" > "<<somedate2;
		if (somedate1 > somedate2){
			cout<<" is TRUE"<<endl;		
		}
		else{
			cout<< "is FALSE"<<endl;
		};

	break;
	case 3:// Check if Date 1 <= Date 2

		cout<<"The satement "<<somedate1<<" <= "<<somedate2;
		if (somedate1 <= somedate2){
			cout<<" is TRUE"<<endl;		
		}
		else{
			cout<< "is FALSE"<<endl;
		};

	break;
	case 4:// Check if Date 1 >= Date 2

		cout<<"The satement "<<somedate1<<" >= "<<somedate2;
		if (somedate1 >= somedate2){
			cout<<" is TRUE"<<endl;		
		}
		else{
			cout<< "is FALSE"<<endl;
		};

	break;
	case 5:// Check if Date 1 == Date 2

		cout<<"The satement "<<somedate1<<" == "<<somedate2;
		if (somedate1 == somedate2){
			cout<<" is TRUE"<<endl;		
		}
		else{
			cout<< "is FALSE"<<endl;
		};

	break;
	case 6:// Check if Date 1 != Date 2

		cout<<"The satement "<<somedate1<<" != "<<somedate2;
		if (somedate1 != somedate2){
			cout<<" is TRUE"<<endl;		
		}
		else{
			cout<< "is FALSE"<<endl;
		};

	break;
	case 7:// Add an integer to Date1 and store in Date 2

		cout<<endl;
		cout<<"Enter number of days to add to Date 1: ";
		cin>>dayadd;

	olddate=somedate1;
	somedate2=somedate1+dayadd;
	somedate1=olddate;
	cout<<"Date1 =: "<<somedate1<<endl;
	cout<<"Date2 =: "<<somedate2<<endl;

	break;

};

};

	return 0;
};