179 lines
5.9 KiB
Plaintext
179 lines
5.9 KiB
Plaintext
#ifndef _DATE_TIME_ADJUST_FUNCTORS_HPP___
|
|
#define _DATE_TIME_ADJUST_FUNCTORS_HPP___
|
|
|
|
/* Copyright (c) 2002,2003 CrystalClear Software, Inc.
|
|
* Use, modification and distribution is subject to the
|
|
* Boost Software License, Version 1.0. (See accompanying
|
|
* file LICENSE_1_0.txt or http://www.boost.org/LICENSE_1_0.txt)
|
|
* Author: Jeff Garland, Bart Garst
|
|
* $Date$
|
|
*/
|
|
|
|
#include "boost/date_time/date.hpp"
|
|
#include "boost/date_time/wrapping_int.hpp"
|
|
|
|
namespace boost {
|
|
namespace date_time {
|
|
|
|
|
|
//! Functor to iterate a fixed number of days
|
|
template<class date_type>
|
|
class day_functor
|
|
{
|
|
public:
|
|
typedef typename date_type::duration_type duration_type;
|
|
day_functor(int f) : f_(f) {}
|
|
duration_type get_offset(const date_type& d) const
|
|
{
|
|
// why is 'd' a parameter???
|
|
// fix compiler warnings
|
|
d.year();
|
|
return duration_type(f_);
|
|
}
|
|
duration_type get_neg_offset(const date_type& d) const
|
|
{
|
|
// fix compiler warnings
|
|
d.year();
|
|
return duration_type(-f_);
|
|
}
|
|
private:
|
|
int f_;
|
|
};
|
|
|
|
|
|
//! Provides calculation to find next nth month given a date
|
|
/*! This adjustment function provides the logic for 'month-based'
|
|
* advancement on a ymd based calendar. The policy it uses
|
|
* to handle the non existant end of month days is to back
|
|
* up to the last day of the month. Also, if the starting
|
|
* date is the last day of a month, this functor will attempt
|
|
* to adjust to the end of the month.
|
|
|
|
*/
|
|
template<class date_type>
|
|
class month_functor
|
|
{
|
|
public:
|
|
typedef typename date_type::duration_type duration_type;
|
|
typedef typename date_type::calendar_type cal_type;
|
|
typedef typename cal_type::ymd_type ymd_type;
|
|
typedef typename cal_type::day_type day_type;
|
|
|
|
month_functor(int f) : f_(f), origDayOfMonth_(0) {}
|
|
duration_type get_offset(const date_type& d) const
|
|
{
|
|
ymd_type ymd(d.year_month_day());
|
|
if (origDayOfMonth_ == 0) {
|
|
origDayOfMonth_ = ymd.day;
|
|
day_type endOfMonthDay(cal_type::end_of_month_day(ymd.year,ymd.month));
|
|
if (endOfMonthDay == ymd.day) {
|
|
origDayOfMonth_ = -1; //force the value to the end of month
|
|
}
|
|
}
|
|
typedef date_time::wrapping_int2<short,1,12> wrap_int2;
|
|
typedef typename wrap_int2::int_type int_type;
|
|
wrap_int2 wi(ymd.month);
|
|
//calc the year wrap around, add() returns 0 or 1 if wrapped
|
|
int_type year = wi.add(static_cast<int_type>(f_));
|
|
year = static_cast<int_type>(year + ymd.year); //calculate resulting year
|
|
// std::cout << "trace wi: " << wi.as_int() << std::endl;
|
|
// std::cout << "trace year: " << year << std::endl;
|
|
//find the last day for the new month
|
|
day_type resultingEndOfMonthDay(cal_type::end_of_month_day(year, wi.as_int()));
|
|
//original was the end of month -- force to last day of month
|
|
if (origDayOfMonth_ == -1) {
|
|
return date_type(year, wi.as_int(), resultingEndOfMonthDay) - d;
|
|
}
|
|
day_type dayOfMonth = origDayOfMonth_;
|
|
if (dayOfMonth > resultingEndOfMonthDay) {
|
|
dayOfMonth = resultingEndOfMonthDay;
|
|
}
|
|
return date_type(year, wi.as_int(), dayOfMonth) - d;
|
|
}
|
|
//! Returns a negative duration_type
|
|
duration_type get_neg_offset(const date_type& d) const
|
|
{
|
|
ymd_type ymd(d.year_month_day());
|
|
if (origDayOfMonth_ == 0) {
|
|
origDayOfMonth_ = ymd.day;
|
|
day_type endOfMonthDay(cal_type::end_of_month_day(ymd.year,ymd.month));
|
|
if (endOfMonthDay == ymd.day) {
|
|
origDayOfMonth_ = -1; //force the value to the end of month
|
|
}
|
|
}
|
|
typedef date_time::wrapping_int2<short,1,12> wrap_int2;
|
|
typedef typename wrap_int2::int_type int_type;
|
|
wrap_int2 wi(ymd.month);
|
|
//calc the year wrap around, add() returns 0 or 1 if wrapped
|
|
int_type year = wi.subtract(static_cast<int_type>(f_));
|
|
year = static_cast<int_type>(year + ymd.year); //calculate resulting year
|
|
//find the last day for the new month
|
|
day_type resultingEndOfMonthDay(cal_type::end_of_month_day(year, wi.as_int()));
|
|
//original was the end of month -- force to last day of month
|
|
if (origDayOfMonth_ == -1) {
|
|
return date_type(year, wi.as_int(), resultingEndOfMonthDay) - d;
|
|
}
|
|
day_type dayOfMonth = origDayOfMonth_;
|
|
if (dayOfMonth > resultingEndOfMonthDay) {
|
|
dayOfMonth = resultingEndOfMonthDay;
|
|
}
|
|
return date_type(year, wi.as_int(), dayOfMonth) - d;
|
|
}
|
|
private:
|
|
int f_;
|
|
mutable short origDayOfMonth_;
|
|
};
|
|
|
|
|
|
//! Functor to iterate a over weeks
|
|
template<class date_type>
|
|
class week_functor
|
|
{
|
|
public:
|
|
typedef typename date_type::duration_type duration_type;
|
|
typedef typename date_type::calendar_type calendar_type;
|
|
week_functor(int f) : f_(f) {}
|
|
duration_type get_offset(const date_type& d) const
|
|
{
|
|
// why is 'd' a parameter???
|
|
// fix compiler warnings
|
|
d.year();
|
|
return duration_type(f_*calendar_type::days_in_week());
|
|
}
|
|
duration_type get_neg_offset(const date_type& d) const
|
|
{
|
|
// fix compiler warnings
|
|
d.year();
|
|
return duration_type(-f_*calendar_type::days_in_week());
|
|
}
|
|
private:
|
|
int f_;
|
|
};
|
|
|
|
//! Functor to iterate by a year adjusting for leap years
|
|
template<class date_type>
|
|
class year_functor
|
|
{
|
|
public:
|
|
//typedef typename date_type::year_type year_type;
|
|
typedef typename date_type::duration_type duration_type;
|
|
year_functor(int f) : _mf(f * 12) {}
|
|
duration_type get_offset(const date_type& d) const
|
|
{
|
|
return _mf.get_offset(d);
|
|
}
|
|
duration_type get_neg_offset(const date_type& d) const
|
|
{
|
|
return _mf.get_neg_offset(d);
|
|
}
|
|
private:
|
|
month_functor<date_type> _mf;
|
|
};
|
|
|
|
|
|
} }//namespace date_time
|
|
|
|
|
|
#endif
|
|
|