#include "StdAfx.h"
#include "LunarCalendar.h"

namespace CalendarEx
{
	//////////////////////////////////////////////////////////////////////////
	// 全局定义;
	// const static INT BEGIN_YEAR = 1901;
	// const static INT NUMBER_YEAR = 199;
	#define BEGIN_YEAR 1901
	#define NUMBER_YEAR 199

	const static INT nNormalMonth[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
	const static INT nLeapMonth[12] = { 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };

	//为了提高效率，记录每月一日是一年中的第几天;
	// 平年的情况;
	const static INT NORMAL_YDAY[12] = { 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334 };
	// 闰年的情况;
	const static INT LEAP_YDAY[12] = { 0, 31, 60, 91, 121, 152, 182, 213, 244, 274, 305, 335 };

	// 使用比特位记录每年的情况
	// 0~4 共5bit 春节日份
	// 5~6 共2bit 春节月份
	// 7~19 共13bit 13个月的大小月情况(如果无闰月，最后位无效)，大月为1,小月为0
	// 20~23 共4bit 记录闰月的月份，如果没有闰月为0
	const static DWORD LUNAR_YEARS[199] =
	{
		0x04AE53, 0x0A5748, 0x5526BD, 0x0D2650, 0x0D9544, 0x46AAB9, 0x056A4D, 0x09AD42, 0x24AEB6, 0x04AE4A, //1901-1910
		0x6A4DBE, 0x0A4D52, 0x0D2546, 0x5D52BA, 0x0B544E, 0x0D6A43, 0x296D37, 0x095B4B, 0x749BC1, 0x049754, //1911-1920
		0x0A4B48, 0x5B25BC, 0x06A550, 0x06D445, 0x4ADAB8, 0x02B64D, 0x095742, 0x2497B7, 0x04974A, 0x664B3E, //1921-1930
		0x0D4A51, 0x0EA546, 0x56D4BA, 0x05AD4E, 0x02B644, 0x393738, 0x092E4B, 0x7C96BF, 0x0C9553, 0x0D4A48, //1931-1940
		0x6DA53B, 0x0B554F, 0x056A45, 0x4AADB9, 0x025D4D, 0x092D42, 0x2C95B6, 0x0A954A, 0x7B4ABD, 0x06CA51, //1941-1950
		0x0B5546, 0x555ABB, 0x04DA4E, 0x0A5B43, 0x352BB8, 0x052B4C, 0x8A953F, 0x0E9552, 0x06AA48, 0x7AD53C, //1951-1960
		0x0AB54F, 0x04B645, 0x4A5739, 0x0A574D, 0x052642, 0x3E9335, 0x0D9549, 0x75AABE, 0x056A51, 0x096D46, //1961-1970
		0x54AEBB, 0x04AD4F, 0x0A4D43, 0x4D26B7, 0x0D254B, 0x8D52BF, 0x0B5452, 0x0B6A47, 0x696D3C, 0x095B50, //1971-1980
		0x049B45, 0x4A4BB9, 0x0A4B4D, 0xAB25C2, 0x06A554, 0x06D449, 0x6ADA3D, 0x0AB651, 0x093746, 0x5497BB, //1981-1990
		0x04974F, 0x064B44, 0x36A537, 0x0EA54A, 0x86B2BF, 0x05AC53, 0x0AB647, 0x5936BC, 0x092E50, 0x0C9645, //1991-2000
		0x4D4AB8, 0x0D4A4C, 0x0DA541, 0x25AAB6, 0x056A49, 0x7AADBD, 0x025D52, 0x092D47, 0x5C95BA, 0x0A954E, //2001-2010
		0x0B4A43, 0x4B5537, 0x0AD54A, 0x955ABF, 0x04BA53, 0x0A5B48, 0x652BBC, 0x052B50, 0x0A9345, 0x474AB9, //2011-2020
		0x06AA4C, 0x0AD541, 0x24DAB6, 0x04B64A, 0x69573D, 0x0A4E51, 0x0D2646, 0x5E933A, 0x0D534D, 0x05AA43, //2021-2030
		0x36B537, 0x096D4B, 0xB4AEBF, 0x04AD53, 0x0A4D48, 0x6D25BC, 0x0D254F, 0x0D5244, 0x5DAA38, 0x0B5A4C, //2031-2040
		0x056D41, 0x24ADB6, 0x049B4A, 0x7A4BBE, 0x0A4B51, 0x0AA546, 0x5B52BA, 0x06D24E, 0x0ADA42, 0x355B37, //2041-2050
		0x09374B, 0x8497C1, 0x049753, 0x064B48, 0x66A53C, 0x0EA54F, 0x06B244, 0x4AB638, 0x0AAE4C, 0x092E42, //2051-2060
		0x3C9735, 0x0C9649, 0x7D4ABD, 0x0D4A51, 0x0DA545, 0x55AABA, 0x056A4E, 0x0A6D43, 0x452EB7, 0x052D4B, //2061-2070
		0x8A95BF, 0x0A9553, 0x0B4A47, 0x6B553B, 0x0AD54F, 0x055A45, 0x4A5D38, 0x0A5B4C, 0x052B42, 0x3A93B6, //2071-2080
		0x069349, 0x7729BD, 0x06AA51, 0x0AD546, 0x54DABA, 0x04B64E, 0x0A5743, 0x452738, 0x0D264A, 0x8E933E, //2081-2090
		0x0D5252, 0x0DAA47, 0x66B53B, 0x056D4F, 0x04AE45, 0x4A4EB9, 0x0A4D4C, 0x0D1541, 0x2D92B5 //2091-2099
	};

	//	如：
	//	0x955ABF
	//	1001 0101010110101  01    11111
	//	闰月月份 7-19.13个月 春月  春日;

	//////////////////////////////////////////////////////////////////////////
	// 函数实现;
	BOOL IsValidLunanrDate(IN LPCTSTR lpLunarDate)
	{
		if (lpLunarDate == NULL || _tcslen(lpLunarDate) != 10 )
			return FALSE;

		CString strLunarDate = lpLunarDate;
		if (strLunarDate.IsEmpty()) return FALSE;

		static INT nLunarYear, nLunarMonth, nLunarDay;
		nLunarYear = nLunarMonth = nLunarDay = 0;

		nLunarYear = _ttoi(strLunarDate.Left(4));
		nLunarMonth = _ttoi(strLunarDate.Mid(5, 2));
		nLunarDay = _ttoi(strLunarDate.Mid(8, 2));

		//越界检查，如果越界，返回无效日期;
		if (nLunarYear <= BEGIN_YEAR || nLunarYear > BEGIN_YEAR + NUMBER_YEAR - 1)
			return FALSE;

		if (nLunarMonth < 1 || nLunarMonth > 12)
			return FALSE;

		// 1.农历年是否有闰月;
		INT nYearIndex = nLunarYear - BEGIN_YEAR;
		INT nLeapMonth = (LUNAR_YEARS[nYearIndex] >> 20) & 0xF;

		// 2.月份大小月天数;
		INT nMonthDays = 0;
		if ((LUNAR_YEARS[nYearIndex] >> (7 + 13 - nLunarMonth)) & 0x1)
			nMonthDays = 30;
		else
			nMonthDays = 29;

		if (nLunarDay < 0 || nLunarDay > nMonthDays)
			return FALSE;

		return TRUE;
	}

	BOOL IsValidLunanrDate(IN INT nLunarYear, IN INT nLunarMonth, IN INT nLunarDay)
	{
		//越界检查，如果越界，返回无效日期;
		if (nLunarYear <= BEGIN_YEAR || nLunarYear > BEGIN_YEAR + NUMBER_YEAR - 1)
			return FALSE;

		if (nLunarMonth < 1 || nLunarMonth > 12)
			return FALSE;

		// 1.农历年是否有闰月;
		INT nYearIndex = nLunarYear - BEGIN_YEAR;
		INT nLeapMonth = (LUNAR_YEARS[nYearIndex] >> 20) & 0xF;

		// 2.月份大小月天数;
		INT nMonthDays = 0;
		if ((LUNAR_YEARS[nYearIndex] >> (7 + 13 - nLunarMonth)) & 0x1)
			nMonthDays = 30;
		else
			nMonthDays = 29;

		if (nLunarDay < 0 || nLunarDay > nMonthDays)
			return FALSE;

		return TRUE;
	}

	// ------------------------------------------------------------------------------------------------------------------------
	// 对外接口函数
	//
	// 函数：IsValidSolarDate
	// 描述：判断日期是否有效;
	// 参数：
	//		strDate	格式如：2014-01-02 或 2014/01/02;
	// 返回：有效;
	//
	// ------------------------------------------------------------------------------------------------------------------------
	BOOL IsValidSolarStrDate(IN LPCTSTR lpSolarDate, IN BOOL bThisyear)
	{
		if (lpSolarDate == NULL || _tcslen(lpSolarDate) != 10) return FALSE;

		CString strSolarDate = lpSolarDate;

		// 判断日期合法性;
		static int nSolarYear = 0;
		static int nSolarMonth = 0;
		static int nSolarDay = 0;

		nSolarYear = nSolarMonth = nSolarDay = 0;

		if (bThisyear)
			nSolarYear = CTime::GetCurrentTime().GetYear();
		else
			nSolarYear = _ttoi(strSolarDate.Left(4));

		nSolarMonth = _ttoi(strSolarDate.Mid(5, 2));
		nSolarDay = _ttoi(strSolarDate.Mid(8, 2));

		if ((nSolarDay <= 0) || (nSolarDay > 31) || (nSolarMonth <= 0) || (nSolarMonth > 12)) return FALSE;

		if ((nSolarYear % 400 == 0) || (nSolarYear % 4 == 0 && nSolarYear % 100 != 0))
		{
			if (nSolarDay > nLeapMonth[nSolarMonth - 1])	return FALSE;
		}
		else
		{
			if (nSolarDay > nNormalMonth[nSolarMonth - 1]) return FALSE;
		}

		return TRUE;
	}

	// ------------------------------------------------------------------------------------------------------------------------
	// 对外接口函数
	//
	// 函数：IsValidSolarDate
	// 描述：判断日期是否有效;
	// 参数：
	//		nSolarYear		公历年;
	//		nSolarMonth		公历月;
	//		nSolarDay		公历日;
	// 返回：有效;
	//
	// ------------------------------------------------------------------------------------------------------------------------
	BOOL IsValidSolarDate(IN CONST INT &nSolarYear, IN CONST INT &nSolarMonth, IN CONST INT &nSolarDay)
	{
		if (nSolarYear < 0 ) return FALSE;

		// 判断日期合法性;
		if ((nSolarDay <= 0) || (nSolarDay > 31) || (nSolarMonth <= 0) || (nSolarMonth > 12)) return FALSE;

		if ((nSolarYear % 400 == 0) || (nSolarYear % 4 == 0 && nSolarYear % 100 != 0))
		{
			if (nSolarDay > nLeapMonth[nSolarMonth - 1])	return FALSE;
		}
		else
		{
			if (nSolarDay > nNormalMonth[nSolarMonth - 1]) return FALSE;
		}

		return TRUE;
	}


	// ------------------------------------------------------------------------------------------------------------------------
	// 对外接口函数
	//
	// 函数：DayOfSolarYear
	// 描述：计算这个公历日期是这一年中的第几天
	// 参数：
	//		nSolarYear		公历年;
	//		nSolarMonth		公历月;
	//		nSolarDay		公历日;
	// 返回：如果日期正确,返回公历日期在这一年中的第几天; 否则返回-1;
	//
	// ------------------------------------------------------------------------------------------------------------------------
	INT DayOfSolarYear(IN CONST INT &nSolarYear,IN CONST INT &nSolarMonth, IN CONST INT &nSolarDay)
	{
		if( !IsValidSolarDate(nSolarYear,nSolarMonth,nSolarDay) )
			return -1;

		const INT *nDaysYear = NORMAL_YDAY;

		//判断是否是公历闰年;
		if (nSolarYear % 4 == 0)
		{
			if (nSolarYear % 100 != 0)
				nDaysYear = LEAP_YDAY;
			if (nSolarYear % 400 == 0)
				nDaysYear = LEAP_YDAY;
		}

		return nDaysYear[nSolarMonth - 1] + nSolarDay;
	}

	// ------------------------------------------------------------------------------------------------------------------------
	// 函数：SolarToLuanrDate
	// 描述：将公历日期转为对应的农历日期
	// 参数：
	//		nSolarYear		公历年;
	//		nSolarMonth		公历月;
	//		nSolarDay		公历日;
	// 返回：返回结构体;
	//
	// ------------------------------------------------------------------------------------------------------------------------
	INT SolarToLuanrDate(IN CONST INT &nSolarYear,IN CONST INT &nSolarMonth, IN CONST INT &nSolarDay, OUT LunarDate &lnaDate )
	{
		if( !IsValidSolarDate(nSolarYear,nSolarMonth,nSolarDay) )
			return -1;

		lnaDate.nYear = nSolarYear;
		// 越界检查，如果越界，返回无效日期;
		if (nSolarYear <= BEGIN_YEAR || nSolarYear > BEGIN_YEAR + NUMBER_YEAR - 1)
			return -1;

		INT nIndexYear = nSolarYear - BEGIN_YEAR;

		//////////////////////////////////////////////////////////////////////////
		// 计算春节的公历日期;
		// 春节：Spring Festival 或者 Chinese new Year;
		// 春节的公历月份;
		INT nSpringMonth = (LUNAR_YEARS[nIndexYear] & 0x60) >> 5;
		// 春节的公历日号;
		INT nSpringDay = (LUNAR_YEARS[nIndexYear] & 0x1F);

		// 计算今天是公历年的第几天;
		INT nTodayInSolarYear = DayOfSolarYear(nSolarYear, nSolarMonth, nSolarDay);
		if( nTodayInSolarYear == -1 )
			return -1;

		// 计算春节是公历年的第几天;
		INT nSpringDayInSolarYear = DayOfSolarYear(nSolarYear, nSpringMonth, nSpringDay);
		if( nSpringDayInSolarYear == -1 )
			return -1;

		// 计算今天是农历年的第几天 = 农历今天在的公历第几天 - 今年农历春节在公历的第几天;
		INT nTodayInLunarYear = nTodayInSolarYear - nSpringDayInSolarYear + 1;

		// 如果今天在春节的前面，重新计算today_luanr_yd;
		if ( nTodayInLunarYear <= 0 )
		{
			// 农历年比当前公历年小1;
			nIndexYear--;
			lnaDate.nYear--;

			// 越界，返回无效日期;
			if (nIndexYear < 0)
				return -1;

			// 去年农历春节月;
			nSpringMonth = (LUNAR_YEARS[nIndexYear] & 0x60) >> 5;
			// 去年农历春节日;
			nSpringDay = (LUNAR_YEARS[nIndexYear] & 0x1F);
			// 去年公历年共多少天;
			int nDaysSolarYear = DayOfSolarYear(lnaDate.nYear, 12, 31);
			if( nDaysSolarYear == -1)
				return -1;

			nSpringDayInSolarYear = DayOfSolarYear(nSolarYear, nSpringMonth, nSpringDay);
			if(nSpringDayInSolarYear == -1)
				return -1;
			// 计算今天是农历年的第几天;
			nTodayInLunarYear = nTodayInSolarYear + nDaysSolarYear - nSpringDayInSolarYear + 1;

		}

		INT nLunarMonth = 1;

		// 计算月份和日期;
		for (; nLunarMonth <= 13; nLunarMonth++)
		{
			INT nMonthDays = 29;
			if ((LUNAR_YEARS[nIndexYear] >> (7 + 13 - nLunarMonth)) & 0x1)
				nMonthDays = 30;

			if (nTodayInLunarYear <= nMonthDays)
				break;
			else
				nTodayInLunarYear -= nMonthDays;
		}

		lnaDate.nDay = nTodayInLunarYear;

		// 处理闰月;
		INT nLeapMonth = (LUNAR_YEARS[nIndexYear] >> 20) & 0xf;
		if (nLeapMonth > 0 && nLeapMonth < nLunarMonth)
		{
			nLunarMonth--;
			// 如果当前月为闰月，设置闰月标志;
			if (nLunarMonth == nLeapMonth)
				lnaDate.bIsLeapMonth = true;
		}

		if ( nLeapMonth > 12 )
			return -1;

		lnaDate.nMonth = nLunarMonth;

		return 0L;
	}


	// ------------------------------------------------------------------------------------------------------------------------
	// 对外接口函数
	//
	// 函数：GetLunarDate
	// 描述：将公历日期转为对应的农历日期
	// 参数：
	//		nSolarYear		公历年;
	//		nSolarMonth		公历月;
	//		nSolarDay		公历日;
	// 返回：暂时未细致处理返回值的定义,错误的值都为负数;
	// ------------------------------------------------------------------------------------------------------------------------
	INT GetLunarDate(IN CONST int &nSolarYear, IN CONST int &nSolarMonth, IN CONST int &nSolarDay,OUT LPTSTR lpLunarDate)
	{
		if( !IsValidSolarDate(nSolarYear,nSolarMonth,nSolarDay) )
			return -1;

		LunarDate lnaDate;
		lnaDate.nYear = nSolarYear;
		lnaDate.nMonth = 0;
		lnaDate.nDay = 0;
		lnaDate.bIsLeapMonth = false;

		// 越界检查，如果越界，返回无效日期;
		if ( nSolarYear <= BEGIN_YEAR || nSolarYear > BEGIN_YEAR + NUMBER_YEAR - 1 )
			return -2;

		INT nIndexYear = nSolarYear - BEGIN_YEAR;

		// 计算春节的公历日期;
		INT nSpringMonth = (LUNAR_YEARS[nIndexYear] & 0x60) >> 5;
		INT nSpringDay = (LUNAR_YEARS[nIndexYear] & 0x1F);

		// 计算今天是公历年的第几天;
		INT nTodayInSolarYear = DayOfSolarYear(nSolarYear, nSolarMonth, nSolarDay);
		if ( nTodayInSolarYear == -1 )
			return -3;

		// 计算春节是公历年的第几天;
		INT nSpringDayInSolarYear = DayOfSolarYear(nSolarYear, nSpringMonth, nSpringDay);
		if ( nSpringDayInSolarYear == -1 )
			return -4;

		// 计算今天是农历年的第几天 = 农历今天在的公历第几天 - 今年农历春节在公历的第几天;
		INT nTodayInLunarYear = nTodayInSolarYear - nSpringDayInSolarYear + 1;

		// 如果今天在春节的前面，重新计算today_luanr_yd;
		if (nTodayInLunarYear <= 0)
		{
			// 农历年比当前公历年小1;
			nIndexYear--;
			lnaDate.nYear--;

			// 越界，返回无效日期;
			if (nIndexYear < 0)
				return -2;

			// 去年农历春节月;
			nSpringMonth = (LUNAR_YEARS[nIndexYear] & 0x60) >> 5;
			// 去年农历春节日;
			nSpringDay = (LUNAR_YEARS[nIndexYear] & 0x1F);
			// 去年公历年共多少天;
			INT nDaysSolarYear = DayOfSolarYear(lnaDate.nYear, 12, 31);
			if (nDaysSolarYear == -1)
				return -1;

			nSpringDayInSolarYear = DayOfSolarYear(nSolarYear, nSpringMonth, nSpringDay);
			if (nSpringDayInSolarYear == -1)
				return -1;

			// 计算今天是农历年的第几天;
			nTodayInLunarYear = nTodayInSolarYear + nDaysSolarYear - nSpringDayInSolarYear + 1;
		}

		INT nLunarMonth = 1;
		//计算月份和日期;
		for (; nLunarMonth <= 13; nLunarMonth++)
		{
			INT nMonthDays = 29;
			if ((LUNAR_YEARS[nIndexYear] >> (7 + 13 - nLunarMonth)) & 0x1)
				nMonthDays = 30;

			if (nTodayInLunarYear <= nMonthDays)
				break;
			else
				nTodayInLunarYear -= nMonthDays;
		}

		lnaDate.nDay = nTodayInLunarYear;

		//处理闰月;
		INT nLeapMonth = (LUNAR_YEARS[nIndexYear] >> 20) & 0xf;
		if (nLeapMonth > 0 && nLeapMonth < nLunarMonth)
		{
			nLunarMonth--;
			//如果当前月为闰月，设置闰月标志;
			if (nLunarMonth == nLeapMonth)
				lnaDate.bIsLeapMonth = true;
		}

		if ( nLeapMonth > 12)
			return -1;

		lnaDate.nMonth = nLunarMonth;
		_stprintf_s(lpLunarDate, 11, _T("%04d-%02d-%02d"),lnaDate.nYear,lnaDate.nMonth,lnaDate.nDay);

		return 0L;
	}

	// ------------------------------------------------------------------------------------------------------------------------
	// 对外接口函数
	//
	// 函数：LuanrToSolarDate
	// 描述：将农历日期转为对应的公历日期
	// 参数：
	//		nLunarYear		农历年;
	//		nLunarMonth		农历月;
	//		nLunarDay		农历日;
	//		bIsLeapMonth	TRUE,若该年有闰月且正好是输入月份,返回闰月对应的公历;
	// 返回：返回公历日期在这一年中的第几天;
	//
	// ------------------------------------------------------------------------------------------------------------------------
	BOOL LuanrToSolarDate(IN CONST int &nLunarYear, IN CONST int &nLunarMonth, IN CONST int &nLunarDay, OUT LPTSTR lpSolarDate, IN BOOL bIsLeapMonth/*=FALSE*/)
	{
		// 越界检查，如果越界，返回无效日期;
		if (nLunarYear <= BEGIN_YEAR || nLunarYear > BEGIN_YEAR + NUMBER_YEAR - 1)
			return FALSE;

		// 1.判断该农历年是否有闰月;
		INT nIndexYear = nLunarYear - BEGIN_YEAR;
		INT nLeapMonth = (LUNAR_YEARS[nIndexYear] >> 20) & 0xF;

		COleDateTime Solar;
		Solar.SetDateTime(nLunarYear, nLunarMonth, nLunarDay, 0, 0, 0);
		COleDateTimeSpan Span(1, 0, 0, 0);

		BOOL bExist = FALSE;
		LunarDate Luanr_date;
		for (INT i = 0; i < 120; i++)
		{
			if(SolarToLuanrDate( Solar.GetYear(), Solar.GetMonth(), Solar.GetDay(),Luanr_date) == -1)
				return FALSE;
			if (Luanr_date.nYear == nLunarYear && Luanr_date.nMonth == nLunarMonth && Luanr_date.nDay == nLunarDay)
			{
				if (bIsLeapMonth && nLeapMonth == nLunarMonth)
				{
					bIsLeapMonth = FALSE;
					Solar += Span;
					continue;
				}
				else
				{
					_stprintf_s(lpSolarDate,11,_T("%4d-%02d-%02d"),Solar.GetYear(), Solar.GetMonth(), Solar.GetDay());
					bExist = TRUE;
					break;
				}
			}
			Solar += Span;
		}

		return bExist;
	}


	// ------------------------------------------------------------------------------------------------------------------------
	// 对外接口函数
	//
	// 函数：
	// 描述：判断输入的农历日期的月份是否有闰月;
	// 参数：
	//		nLunarYear		农历年;
	// 返回：若有闰月返回闰月值,否则返回0;
	//
	// ------------------------------------------------------------------------------------------------------------------------
	INT IsLeapMonth(IN const int &nLunarYear)
	{
		// 越界检查，如果越界，返回无效日期;
		if (nLunarYear <= BEGIN_YEAR || nLunarYear > BEGIN_YEAR + NUMBER_YEAR - 1)
			return -1;

		// 1.判断该农历年是否有闰月;
		return (LUNAR_YEARS[nLunarYear - BEGIN_YEAR] >> 20) & 0xF;
	}
};