[857] | 1 | #ifndef _DATE_TIME_ADJUST_FUNCTORS_HPP___
|
---|
| 2 | #define _DATE_TIME_ADJUST_FUNCTORS_HPP___
|
---|
| 3 |
|
---|
| 4 | /* Copyright (c) 2002,2003 CrystalClear Software, Inc.
|
---|
| 5 | * Use, modification and distribution is subject to the
|
---|
| 6 | * Boost Software License, Version 1.0. (See accompanying
|
---|
| 7 | * file LICENSE-1.0 or http://www.boost.org/LICENSE-1.0)
|
---|
| 8 | * Author: Jeff Garland, Bart Garst
|
---|
| 9 | * $Date: 2004/07/04 18:40:23 $
|
---|
| 10 | */
|
---|
| 11 |
|
---|
| 12 | #include "boost/date_time/date.hpp"
|
---|
| 13 | #include "boost/date_time/wrapping_int.hpp"
|
---|
| 14 |
|
---|
| 15 | namespace boost {
|
---|
| 16 | namespace date_time {
|
---|
| 17 |
|
---|
| 18 |
|
---|
| 19 | //! Functor to iterate a fixed number of days
|
---|
| 20 | template<class date_type>
|
---|
| 21 | class day_functor
|
---|
| 22 | {
|
---|
| 23 | public:
|
---|
| 24 | typedef typename date_type::duration_type duration_type;
|
---|
| 25 | day_functor(int f) : f_(f) {}
|
---|
| 26 | duration_type get_offset(const date_type& d) const
|
---|
| 27 | {
|
---|
| 28 | // why is 'd' a parameter???
|
---|
| 29 | // fix compiler warnings
|
---|
| 30 | d.year();
|
---|
| 31 | return duration_type(f_);
|
---|
| 32 | }
|
---|
| 33 | duration_type get_neg_offset(const date_type& d) const
|
---|
| 34 | {
|
---|
| 35 | // fix compiler warnings
|
---|
| 36 | d.year();
|
---|
| 37 | return duration_type(-f_);
|
---|
| 38 | }
|
---|
| 39 | private:
|
---|
| 40 | int f_;
|
---|
| 41 | };
|
---|
| 42 |
|
---|
| 43 |
|
---|
| 44 | //! Provides calculation to find next nth month given a date
|
---|
| 45 | /*! This adjustment function provides the logic for 'month-based'
|
---|
| 46 | * advancement on a ymd based calendar. The policy it uses
|
---|
| 47 | * to handle the non existant end of month days is to back
|
---|
| 48 | * up to the last day of the month. Also, if the starting
|
---|
| 49 | * date is the last day of a month, this functor will attempt
|
---|
| 50 | * to adjust to the end of the month.
|
---|
| 51 |
|
---|
| 52 | */
|
---|
| 53 | template<class date_type>
|
---|
| 54 | class month_functor
|
---|
| 55 | {
|
---|
| 56 | public:
|
---|
| 57 | typedef typename date_type::duration_type duration_type;
|
---|
| 58 | typedef typename date_type::calendar_type cal_type;
|
---|
| 59 | typedef typename cal_type::ymd_type ymd_type;
|
---|
| 60 | typedef typename cal_type::day_type day_type;
|
---|
| 61 |
|
---|
| 62 | month_functor(int f) : f_(f), origDayOfMonth_(0) {}
|
---|
| 63 | duration_type get_offset(const date_type& d) const
|
---|
| 64 | {
|
---|
| 65 | ymd_type ymd(d.year_month_day());
|
---|
| 66 | if (origDayOfMonth_ == 0) {
|
---|
| 67 | origDayOfMonth_ = ymd.day;
|
---|
| 68 | day_type endOfMonthDay(cal_type::end_of_month_day(ymd.year,ymd.month));
|
---|
| 69 | if (endOfMonthDay == ymd.day) {
|
---|
| 70 | origDayOfMonth_ = -1; //force the value to the end of month
|
---|
| 71 | }
|
---|
| 72 | }
|
---|
| 73 | typedef date_time::wrapping_int2<short,1,12> wrap_int2;
|
---|
| 74 | typedef typename wrap_int2::int_type int_type;
|
---|
| 75 | wrap_int2 wi(ymd.month);
|
---|
| 76 | //calc the year wrap around, add() returns 0 or 1 if wrapped
|
---|
| 77 | int_type year = wi.add(static_cast<int_type>(f_));
|
---|
| 78 | year = static_cast<int_type>(year + ymd.year); //calculate resulting year
|
---|
| 79 | // std::cout << "trace wi: " << wi.as_int() << std::endl;
|
---|
| 80 | // std::cout << "trace year: " << year << std::endl;
|
---|
| 81 | //find the last day for the new month
|
---|
| 82 | day_type resultingEndOfMonthDay(cal_type::end_of_month_day(year, wi.as_int()));
|
---|
| 83 | //original was the end of month -- force to last day of month
|
---|
| 84 | if (origDayOfMonth_ == -1) {
|
---|
| 85 | return date_type(year, wi.as_int(), resultingEndOfMonthDay) - d;
|
---|
| 86 | }
|
---|
| 87 | day_type dayOfMonth = origDayOfMonth_;
|
---|
| 88 | if (dayOfMonth > resultingEndOfMonthDay) {
|
---|
| 89 | dayOfMonth = resultingEndOfMonthDay;
|
---|
| 90 | }
|
---|
| 91 | return date_type(year, wi.as_int(), dayOfMonth) - d;
|
---|
| 92 | }
|
---|
| 93 | //! Returns a negative duration_type
|
---|
| 94 | duration_type get_neg_offset(const date_type& d) const
|
---|
| 95 | {
|
---|
| 96 | ymd_type ymd(d.year_month_day());
|
---|
| 97 | if (origDayOfMonth_ == 0) {
|
---|
| 98 | origDayOfMonth_ = ymd.day;
|
---|
| 99 | day_type endOfMonthDay(cal_type::end_of_month_day(ymd.year,ymd.month));
|
---|
| 100 | if (endOfMonthDay == ymd.day) {
|
---|
| 101 | origDayOfMonth_ = -1; //force the value to the end of month
|
---|
| 102 | }
|
---|
| 103 | }
|
---|
| 104 | typedef date_time::wrapping_int2<short,1,12> wrap_int2;
|
---|
| 105 | typedef typename wrap_int2::int_type int_type;
|
---|
| 106 | wrap_int2 wi(ymd.month);
|
---|
| 107 | //calc the year wrap around, add() returns 0 or 1 if wrapped
|
---|
| 108 | int_type year = wi.subtract(static_cast<int_type>(f_));
|
---|
| 109 | year = static_cast<int_type>(year + ymd.year); //calculate resulting year
|
---|
| 110 | //find the last day for the new month
|
---|
| 111 | day_type resultingEndOfMonthDay(cal_type::end_of_month_day(year, wi.as_int()));
|
---|
| 112 | //original was the end of month -- force to last day of month
|
---|
| 113 | if (origDayOfMonth_ == -1) {
|
---|
| 114 | return date_type(year, wi.as_int(), resultingEndOfMonthDay) - d;
|
---|
| 115 | }
|
---|
| 116 | day_type dayOfMonth = origDayOfMonth_;
|
---|
| 117 | if (dayOfMonth > resultingEndOfMonthDay) {
|
---|
| 118 | dayOfMonth = resultingEndOfMonthDay;
|
---|
| 119 | }
|
---|
| 120 | return date_type(year, wi.as_int(), dayOfMonth) - d;
|
---|
| 121 | }
|
---|
| 122 | private:
|
---|
| 123 | int f_;
|
---|
| 124 | mutable short origDayOfMonth_;
|
---|
| 125 | };
|
---|
| 126 |
|
---|
| 127 |
|
---|
| 128 | //! Functor to iterate a over weeks
|
---|
| 129 | template<class date_type>
|
---|
| 130 | class week_functor
|
---|
| 131 | {
|
---|
| 132 | public:
|
---|
| 133 | typedef typename date_type::duration_type duration_type;
|
---|
| 134 | typedef typename date_type::calendar_type calendar_type;
|
---|
| 135 | week_functor(int f) : f_(f) {}
|
---|
| 136 | duration_type get_offset(const date_type& d) const
|
---|
| 137 | {
|
---|
| 138 | // why is 'd' a parameter???
|
---|
| 139 | // fix compiler warnings
|
---|
| 140 | d.year();
|
---|
| 141 | return duration_type(f_*calendar_type::days_in_week());
|
---|
| 142 | }
|
---|
| 143 | duration_type get_neg_offset(const date_type& d) const
|
---|
| 144 | {
|
---|
| 145 | // fix compiler warnings
|
---|
| 146 | d.year();
|
---|
| 147 | return duration_type(-f_*calendar_type::days_in_week());
|
---|
| 148 | }
|
---|
| 149 | private:
|
---|
| 150 | int f_;
|
---|
| 151 | };
|
---|
| 152 |
|
---|
| 153 | //! Functor to iterate by a year adjusting for leap years
|
---|
| 154 | template<class date_type>
|
---|
| 155 | class year_functor
|
---|
| 156 | {
|
---|
| 157 | public:
|
---|
| 158 | //typedef typename date_type::year_type year_type;
|
---|
| 159 | typedef typename date_type::duration_type duration_type;
|
---|
| 160 | year_functor(int f) : _mf(f * 12) {}
|
---|
| 161 | duration_type get_offset(const date_type& d) const
|
---|
| 162 | {
|
---|
| 163 | return _mf.get_offset(d);
|
---|
| 164 | }
|
---|
| 165 | duration_type get_neg_offset(const date_type& d) const
|
---|
| 166 | {
|
---|
| 167 | return _mf.get_neg_offset(d);
|
---|
| 168 | }
|
---|
| 169 | private:
|
---|
| 170 | month_functor<date_type> _mf;
|
---|
| 171 | };
|
---|
| 172 |
|
---|
| 173 |
|
---|
| 174 | } }//namespace date_time
|
---|
| 175 |
|
---|
| 176 |
|
---|
| 177 | #endif
|
---|
| 178 |
|
---|