org.dmfs.rfc5545.recur.ByWeekNoYearlyExpander Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2018 Marten Gajda 
 *
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.dmfs.rfc5545.recur;

import org.dmfs.rfc5545.Instance;
import org.dmfs.rfc5545.calendarmetrics.CalendarMetrics;
import org.dmfs.rfc5545.recur.RecurrenceRule.Part;


/**
 * A filter expands recurrence rules by week of year. This is allowed for yearly rules only.  If a BYMONTH part is present and any BY*DAY rules follows this
 * filter also expands weeks that overlap the expanded month. That means two subsequent interval sets can include the same week. The BY*DAY filters will take
 * care of filtering those. 
 *
 * @author Marten Gajda
 */
final class ByWeekNoYearlyExpander extends ByExpander
{
    /**
     * The week numbers to expand.
     */
    private final int[] mByWeekNo;

    private final int mOriginalWeekDay;


    public ByWeekNoYearlyExpander(RecurrenceRule rule, RuleIterator previous, CalendarMetrics calendarTools, long start)
    {
        super(previous, calendarTools, start);

        mByWeekNo = StaticUtils.ListToSortedArray(rule.getByPart(Part.BYWEEKNO));

        // BYWEEKNO expansion preserves the original day of week
        mOriginalWeekDay = calendarTools.getDayOfWeek(Instance.year(start), Instance.month(start), Instance.dayOfMonth(start));
    }


    @Override
    void expand(long instance, long notBefore)
    {
        int year = Instance.year(instance);

        // get the number of weeks in that year
        int yearWeeks = mCalendarMetrics.getWeeksPerYear(year);

        for (int weekOfYear : mByWeekNo)
        {
            int actualWeek = weekOfYear;
            if (weekOfYear < 0)
            {
                actualWeek = yearWeeks + weekOfYear + 1;
            }

            if (actualWeek <= 0 || actualWeek > yearWeeks)
            {
                if (actualWeek <= 0)
                {
                    addInstance(Instance.setMonthAndDayOfMonth(instance, 0, 0));
                }
                else
                {
                    int monthAndDay = mCalendarMetrics.getMonthAndDayOfYearDay(year, mCalendarMetrics.getDaysPerYear(year));
                    addInstance(
                            Instance.setMonthAndDayOfMonth(instance, CalendarMetrics.packedMonth(monthAndDay), CalendarMetrics.dayOfMonth(monthAndDay) + 1));
                }
                continue;
            }

            int yearDay = mCalendarMetrics.getYearDayOfIsoYear(year, actualWeek, mOriginalWeekDay);

            int yearOffset = 0;
            if (yearDay < 1)
            {
                yearOffset = -1;
                yearDay += mCalendarMetrics.getDaysPerYear(year);
            }
            else if (yearDay > mCalendarMetrics.getDaysPerYear(year))
            {
                yearDay -= mCalendarMetrics.getDaysPerYear(year);
                yearOffset = 1;
            }

            final int monthAndDay = mCalendarMetrics.getMonthAndDayOfYearDay(year, yearDay);
            addInstance(Instance.setYear(Instance.setMonthAndDayOfMonth(instance, CalendarMetrics.packedMonth(monthAndDay),
                    CalendarMetrics.dayOfMonth(monthAndDay)), year + yearOffset));
        }
    }
}