File indexing completed on 2024-05-12 05:21:22

 /*
   This file is part of KOrganizer.
   SPDX-FileCopyrightText: 2002 Cornelius Schumacher <schumacher@kde.org>
   SPDX-FileCopyrightText: 2003-2004 Reinhold Kainhofer <reinhold@kainhofer.com>
 
   SPDX-FileCopyrightText: 2010 Klarälvdalens Datakonsult AB, a KDAB Group company <info@kdab.net>
   SPDX-FileContributor: Sergio Martins <sergio@kdab.com>
 
   SPDX-License-Identifier: GPL-2.0-or-later WITH Qt-Commercial-exception-1.0
 */
 
 #include "datenavigator.h"
 #include "koglobals.h"
 #include "korganizer_debug.h"
 
 DateNavigator::DateNavigator(QObject *parent)
     : QObject(parent)
 {
     mSelectedDates.append(QDate::currentDate());
 }
 
 DateNavigator::~DateNavigator() = default;
 
 KCalendarCore::DateList DateNavigator::selectedDates() const
 {
     return mSelectedDates;
 }
 
 int DateNavigator::datesCount() const
 {
     return mSelectedDates.count();
 }
 
 void DateNavigator::selectDates(const KCalendarCore::DateList &dateList, QDate preferredMonth)
 {
     if (!dateList.isEmpty()) {
         mSelectedDates = dateList;
         emitSelected(preferredMonth);
     }
 }
 
 void DateNavigator::selectDate(QDate date)
 {
     if (!date.isValid()) {
         qCDebug(KORGANIZER_LOG) << "an invalid date was passed as a parameter!";
         date = QDate::currentDate();
     }
     mSelectedDates.clear();
     mSelectedDates.append(date);
 
     emitSelected();
 }
 
 void DateNavigator::selectDates(int count)
 {
     count = qMin(count, static_cast<int>(MAX_SELECTABLE_DAYS));
     selectDates(mSelectedDates.first(), count);
 }
 
 void DateNavigator::selectDates(const QDate &d, int count, const QDate &preferredMonth)
 {
     KCalendarCore::DateList dates;
     dates.reserve(count);
 
     for (int i = 0; i < count; ++i) {
         dates.append(d.addDays(i));
     }
 
     mSelectedDates = dates;
     emitSelected(preferredMonth);
 }
 
 void DateNavigator::selectWeekByDay(int weekDay, QDate d, QDate preferredMonth)
 {
     const int dateCount = mSelectedDates.count();
     const bool weekStart = (weekDay == KOGlobals::self()->firstDayOfWeek());
     if (weekStart && dateCount == 7) {
         selectWeek(d, preferredMonth);
     } else {
         selectDates(d, dateCount, preferredMonth);
     }
 }
 
 void DateNavigator::selectWeek()
 {
     selectWeek(mSelectedDates.first());
 }
 
 void DateNavigator::selectWeek(const QDate &d, const QDate &preferredMonth)
 {
     const int dayOfWeek = d.dayOfWeek();
     const int weekStart = KOGlobals::self()->firstDayOfWeek();
 
     QDate firstDate = d.addDays(weekStart - dayOfWeek);
 
     if (weekStart != 1 && dayOfWeek < weekStart) {
         firstDate = firstDate.addDays(-7);
     }
 
     selectDates(firstDate, 7, preferredMonth);
 }
 
 void DateNavigator::selectWorkWeek()
 {
     selectWorkWeek(mSelectedDates.first());
 }
 
 void DateNavigator::selectWorkWeek(QDate d)
 {
     const int weekStart = KOGlobals::self()->firstDayOfWeek();
     const int dayOfWeek = d.dayOfWeek();
     QDate currentDate = d.addDays(weekStart - dayOfWeek);
 
     if (weekStart != 1 && dayOfWeek < weekStart) {
         currentDate = currentDate.addDays(-7);
     }
 
     mSelectedDates.clear();
     const int mask = KOGlobals::self()->getWorkWeekMask();
 
     for (int i = 0; i < 7; ++i) {
         if ((1 << ((i + weekStart + 6) % 7)) & (mask)) {
             mSelectedDates.append(currentDate.addDays(i));
         }
     }
 
     emitSelected(/* preferredMonth= */ d);
 }
 
 void DateNavigator::selectToday()
 {
     const QDate d = QDate::currentDate();
 
     const int dateCount = mSelectedDates.count();
 
     if (dateCount == 7) {
         selectWeek(d, /* preferredMonth= */ d);
     } else if (dateCount == 5) {
         selectWorkWeek(d);
     } else {
         selectDates(d, dateCount);
     }
 }
 
 void DateNavigator::selectPreviousYear()
 {
     QDate firstSelected = mSelectedDates.first();
     const int weekDay = firstSelected.dayOfWeek();
     firstSelected = firstSelected.addYears(-1);
 
     selectWeekByDay(weekDay, firstSelected);
 }
 
 void DateNavigator::selectPreviousMonth(const QDate &currentMonth, const QDate &selectionLowerLimit, const QDate &selectionUpperLimit)
 {
     shiftMonth(currentMonth, selectionLowerLimit, selectionUpperLimit, -1);
 }
 
 void DateNavigator::selectPreviousWeek()
 {
     QDate firstSelected = mSelectedDates.first();
     const int weekDay = firstSelected.dayOfWeek();
     firstSelected = firstSelected.addDays(-7);
 
     selectWeekByDay(weekDay, firstSelected);
 }
 
 void DateNavigator::selectNextWeek()
 {
     QDate firstSelected = mSelectedDates.first();
     const int weekDay = firstSelected.dayOfWeek();
 
     firstSelected = firstSelected.addDays(7);
 
     selectWeekByDay(weekDay, firstSelected);
 }
 
 void DateNavigator::shiftMonth(const QDate &currentMonth, const QDate &selectionLowerLimit, const QDate &selectionUpperLimit, int offset)
 {
     QDate firstSelected = mSelectedDates.first();
     const int weekDay = firstSelected.dayOfWeek();
     firstSelected = firstSelected.addMonths(offset);
 
     /* Don't trust firstSelected to calculate the nextMonth. firstSelected
        can belong to a month other than currentMonth because KDateNavigator
        displays 7*6 days. firstSelected should only be used for selection
        purposes */
     const QDate nextMonth = currentMonth.isValid() ? currentMonth.addMonths(offset) : firstSelected;
 
     /* When firstSelected doesn't belong to currentMonth it can happen
        that the new selection won't be visible on our KDateNavigators
        so we must adjust it */
     if (selectionLowerLimit.isValid() && firstSelected < selectionLowerLimit) {
         firstSelected = selectionLowerLimit;
     } else if (selectionUpperLimit.isValid() && firstSelected > selectionUpperLimit) {
         firstSelected = selectionUpperLimit.addDays(-6);
     }
 
     selectWeekByDay(weekDay, firstSelected, nextMonth);
 }
 
 void DateNavigator::selectNextMonth(const QDate &currentMonth, const QDate &selectionLowerLimit, const QDate &selectionUpperLimit)
 {
     shiftMonth(currentMonth, selectionLowerLimit, selectionUpperLimit, 1);
 }
 
 void DateNavigator::selectNextYear()
 {
     QDate firstSelected = mSelectedDates.first();
     int weekDay = firstSelected.dayOfWeek();
     firstSelected = firstSelected.addYears(1);
 
     selectWeekByDay(weekDay, firstSelected);
 }
 
 void DateNavigator::selectPrevious()
 {
     int offset = -7;
     if (datesCount() == 1) {
         offset = -1;
     }
 
     selectDates(mSelectedDates.first().addDays(offset), datesCount());
 }
 
 void DateNavigator::selectNext()
 {
     int offset = 7;
     if (datesCount() == 1) {
         offset = 1;
     }
 
     selectDates(mSelectedDates.first().addDays(offset), datesCount());
 }
 
 void DateNavigator::selectMonth(int month)
 {
     // always display starting at the first week of the specified month
     QDate firstSelected = QDate(mSelectedDates.first().year(), month, 1);
 
     int day = firstSelected.day();
     firstSelected.setDate(firstSelected.year(), month, 1);
     int days = firstSelected.daysInMonth();
     // As day we use either the selected date, or if the month has less days
     // than that, we use the max day of that month
     if (day > days) {
         day = days;
     }
     QDate requestedMonth;
     firstSelected.setDate(firstSelected.year(), month, day);
     requestedMonth.setDate(firstSelected.year(), month, 1);
 
     selectWeekByDay(1, firstSelected, requestedMonth);
 }
 
 void DateNavigator::selectYear(int year)
 {
     QDate firstSelected = mSelectedDates.first();
     const int deltaYear = year - firstSelected.year();
     firstSelected = firstSelected.addYears(deltaYear);
 
     const int weekDay = firstSelected.dayOfWeek();
     selectWeekByDay(weekDay, firstSelected);
 }
 
 void DateNavigator::emitSelected(const QDate &preferredMonth)
 {
     Q_EMIT datesSelected(mSelectedDates, preferredMonth);
 }
 
 #include "moc_datenavigator.cpp"