Added code and first unit tests for the GenerateMonthGrid function

[xestiacalendar/.git] / source / common / monthviewgen.cpp

diff --git a/source/common/monthviewgen.cpp b/source/common/monthviewgen.cpp
index a80adb4..f1bd702 100644 (file)
--- a/source/common/monthviewgen.cpp
+++ b/source/common/monthviewgen.cpp
@@ -70,3 +70,212 @@ int PreviousMonthNumberofDays(int Month, int Year){
        
 }
 
+XCCalendarMonthViewGrid GenerateMonthGrid(int Month, int Year){
+       
+       XCCalendarMonthViewGrid MonthViewOutput;
+       
+       // Work out which day of the week the first day of 
+       // the month is.
+               
+       time_t EmptyTime;
+       struct tm * FirstDay;
+       int PreviousMonthNumDays = PreviousMonthNumberofDays(Month, Year);
+       int PreviousMonthDays = 0;
+       int PreviousMonth = 0;
+       int PreviousYear = Year;
+       int DayWeekCount = 0;
+       bool ProcessGrid = true;
+       bool FirstWeekProcessing = true;
+       
+       time(&EmptyTime);
+       
+       FirstDay = localtime(&EmptyTime);
+       FirstDay->tm_year = Year - 1900;
+       FirstDay->tm_mon = Month - 1;
+       FirstDay->tm_mday = 1;
+       
+       mktime(FirstDay);
+       
+       if (FirstDay->tm_wday == 0){
+               PreviousMonthDays = 6;
+       } else {
+               PreviousMonthDays = FirstDay->tm_wday - 1;
+       }
+       
+       if (Month == 1){
+               
+               PreviousMonth = 12;
+               PreviousYear--;
+               
+       } else {
+
+               PreviousMonth = Month - 1;
+               
+       }
+       
+       // Workout the days of the previous month for the
+       // first week to be displayed.
+       
+       XCCalendarMonthViewGridDayWeek FirstWeek;
+       
+       for (int PreviousMonthDaysProcessing = 0; 
+               PreviousMonthDaysProcessing < PreviousMonthDays; 
+               PreviousMonthDaysProcessing++){
+               
+               XCCalendarMonthViewGridDayData DayItem;
+                       
+               DayItem.Day = PreviousMonthNumDays - PreviousMonthDaysProcessing;
+               DayItem.Month = PreviousMonth;
+               DayItem.Year = PreviousYear;
+               DayItem.IsInMonth = false;
+                       
+               FirstWeek.DayList.insert(FirstWeek.DayList.begin(), DayItem);
+                       
+               DayWeekCount++;
+                       
+       }
+       
+       // Process the actual days of the month.
+
+       XCCalendarMonthViewGridDayWeek MonthWeek;
+       int ProcessDay = 1;
+       
+       while (ProcessGrid == true){
+               
+               if (DayWeekCount == 7){
+                       
+                       // Add to the month view grid.
+                       
+                       if (FirstWeekProcessing == true){
+                               
+                               FirstWeekProcessing = false;
+                               MonthViewOutput.WeekList.push_back(FirstWeek);
+                               
+                       } else {
+                       
+                               MonthViewOutput.WeekList.push_back(MonthWeek);
+                               MonthWeek.DayList.clear();
+                               
+                       }
+                       
+                       DayWeekCount = 0;
+                       continue;
+                       
+               }
+
+               XCCalendarMonthViewGridDayData DayItem;
+               
+               DayItem.Day = ProcessDay;
+               DayItem.Month = Month;
+               DayItem.Year = Year;
+               DayItem.IsInMonth = true;
+               
+               if (FirstWeekProcessing == true){
+                       
+                       FirstWeek.DayList.push_back(DayItem);
+                       
+               } else {
+                       
+                       MonthWeek.DayList.push_back(DayItem);
+                       
+               }
+
+               DayWeekCount++;
+               
+               if ((Month == 1 || Month == 3 || Month == 5 || 
+                       Month == 7 || Month == 8 || Month == 10 || Month == 12) && 
+                       ProcessDay == 31){
+       
+                       break;
+                       
+               }
+       
+               if ((Month == 4 || Month == 6 || Month == 9 || Month == 11) &&
+                       ProcessDay == 30){
+               
+                       break;
+               
+               }
+               
+               if (Year % 4 == 0){
+               
+                       if (Year % 100 == 0){
+                       
+                               if (Year % 400 == 0){
+                       
+                                       if (Month == 2 && ProcessDay == 29){
+                                       
+                                               break;
+                                               
+                                       }
+                               
+                               } else {
+                               
+                                       if (Month == 2 && ProcessDay == 28){
+                                               
+                                               break;
+                                               
+                                       }
+                               
+                               }
+                       
+                       }
+               
+                       if (Month == 2 && ProcessDay == 28){
+                       
+                               break;
+                               
+                       }
+               
+               } else if (Month == 2 && ProcessDay == 28){
+               
+                       break;
+               
+               }
+               
+               ProcessDay++;
+               
+               
+       }
+       
+       // At the end of the month add any remaining days
+       // (if required).
+       
+       int NextMonthYear = Year;
+       int NextMonth = 0;
+       int NextMonthDay = 1;
+       
+       if (Month == 12){
+               
+               NextMonth = 1;
+               NextMonthYear++;
+               
+       } else {
+               
+               NextMonth = Month + 1;
+               
+       }
+       
+       for (; DayWeekCount < 7; DayWeekCount++){
+       
+               XCCalendarMonthViewGridDayData DayItem;
+               
+               DayItem.Day = NextMonthDay;
+               DayItem.Month = NextMonth;
+               DayItem.Year = NextMonthYear;
+               DayItem.IsInMonth = false;
+
+               MonthWeek.DayList.push_back(DayItem);
+               NextMonthDay++;
+               
+       }
+       
+       if (DayWeekCount == 7){
+
+               MonthViewOutput.WeekList.push_back(MonthWeek);
+               
+       }
+       
+       return MonthViewOutput;
+       
+}
\ No newline at end of file