use precalculated dow offsets

This commit is contained in:
gorhill 2013-09-11 11:18:45 -04:00
parent 0e96f12377
commit 09aefb5f32
1 changed files with 24 additions and 10 deletions

View File

@ -19,6 +19,18 @@ import (
/******************************************************************************/ /******************************************************************************/
var dowNormalizedOffsets = [][]int{
{1, 8, 15, 22, 29},
{2, 9, 16, 23, 30},
{3, 10, 17, 24, 31},
{4, 11, 18, 25},
{5, 12, 19, 26},
{6, 13, 20, 27},
{7, 14, 21, 28},
}
/******************************************************************************/
func (expr *Expression) nextYear(t time.Time) time.Time { func (expr *Expression) nextYear(t time.Time) time.Time {
// Find index at which item in list is greater or equal to // Find index at which item in list is greater or equal to
// candidate year // candidate year
@ -223,19 +235,21 @@ func (expr *Expression) calculateActualDaysOfMonth(year, month int) []int {
// days of week // days of week
// offset : (7 - day_of_week_of_1st_day_of_month) // offset : (7 - day_of_week_of_1st_day_of_month)
// target : 1 + (7 * week_of_month) + (offset + day_of_week) % 7 // target : 1 + (7 * week_of_month) + (offset + day_of_week) % 7
for w := 0; w <= 4; w += 1 { for v := range expr.daysOfWeek {
for v := range expr.daysOfWeek { w := dowNormalizedOffsets[offset%7+v]
v := 1 + w*7 + (offset+v)%7 actualDaysOfMonthMap[w[0]] = true
if v <= lastDayOfMonth.Day() { actualDaysOfMonthMap[w[1]] = true
actualDaysOfMonthMap[v] = true actualDaysOfMonthMap[w[2]] = true
} actualDaysOfMonthMap[w[3]] = true
if len(w) > 4 && w[4] <= lastDayOfMonth.Day() {
actualDaysOfMonthMap[w[4]] = true
} }
} }
// days of week of specific week in the month // days of week of specific week in the month
// offset : (7 - day_of_week_of_1st_day_of_month) // offset : (7 - day_of_week_of_1st_day_of_month)
// target : 1 + (7 * week_of_month) + (offset + day_of_week) % 7 // target : 1 + (7 * week_of_month) + (offset + day_of_week) % 7
for v := range expr.specificWeekDaysOfWeek { for v := range expr.specificWeekDaysOfWeek {
v := 1 + 7*(v/7) + (offset+v)%7 v = 1 + 7*(v/7) + (offset+v)%7
if v <= lastDayOfMonth.Day() { if v <= lastDayOfMonth.Day() {
actualDaysOfMonthMap[v] = true actualDaysOfMonthMap[v] = true
} }
@ -244,7 +258,7 @@ func (expr *Expression) calculateActualDaysOfMonth(year, month int) []int {
lastWeekOrigin := firstDayOfMonth.AddDate(0, 1, -7) lastWeekOrigin := firstDayOfMonth.AddDate(0, 1, -7)
offset = 7 - int(lastWeekOrigin.Weekday()) offset = 7 - int(lastWeekOrigin.Weekday())
for v := range expr.lastWeekDaysOfWeek { for v := range expr.lastWeekDaysOfWeek {
v := lastWeekOrigin.Day() + (offset+v)%7 v = lastWeekOrigin.Day() + (offset+v)%7
if v <= lastDayOfMonth.Day() { if v <= lastDayOfMonth.Day() {
actualDaysOfMonthMap[v] = true actualDaysOfMonthMap[v] = true
} }
@ -255,16 +269,16 @@ func (expr *Expression) calculateActualDaysOfMonth(year, month int) []int {
} }
func workdayOfMonth(targetDom, lastDom time.Time) int { func workdayOfMonth(targetDom, lastDom time.Time) int {
// If saturday, then friday
// If sunday, then monday
dom := targetDom.Day() dom := targetDom.Day()
dow := targetDom.Weekday() dow := targetDom.Weekday()
// If saturday, then friday
if dow == time.Saturday { if dow == time.Saturday {
if dom > 1 { if dom > 1 {
dom -= 1 dom -= 1
} else { } else {
dom += 2 dom += 2
} }
// If sunday, then monday
} else if dow == time.Sunday { } else if dow == time.Sunday {
if dom < lastDom.Day() { if dom < lastDom.Day() {
dom += 1 dom += 1