#ifndef DATE_DURATION_TYPES_HPP___

#define DATE_DURATION_TYPES_HPP___

/* Copyright (c) 2004 CrystalClear Software, Inc.

 * 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: 2008-02-27 15:00:24 -0500 (Wed, 27 Feb 2008) $

 */

#include <boost/date_time/int_adapter.hpp>

#include <boost/date_time/special_defs.hpp>

#include <boost/date_time/date_duration.hpp>

namespace boost {

namespace date_time {

  //! Additional duration type that represents a number of n*7 days

  template <class duration_config>

  class weeks_duration : public date_duration<duration_config> {

  public:

    weeks_duration(typename duration_config::impl_type w) 

      : date_duration<duration_config>(w * 7) {}

    weeks_duration(special_values sv) 

      : date_duration<duration_config>(sv) {}

  };

  // predeclare

  template<class t>

  class years_duration;

  //! additional duration type that represents a logical month

  /*! A logical month enables things like: "date(2002,Mar,2) + months(2) -> 

   * 2002-May2". If the date is a last day-of-the-month, the result will 

   * also be a last-day-of-the-month.

   */

  template<class base_config>

  class months_duration 

  {

    private:

      typedef typename base_config::int_rep int_rep;

      typedef typename int_rep::int_type int_type;

      typedef typename base_config::date_type date_type;

      typedef typename date_type::duration_type duration_type;

      typedef typename base_config::month_adjustor_type month_adjustor_type;

      typedef months_duration<base_config> months_type;

      typedef years_duration<base_config> years_type;

    public:

      months_duration(int_rep num) : _m(num) {}

      months_duration(special_values sv) : _m(sv) 

      {

        _m = int_rep::from_special(sv);

      }

      int_rep number_of_months() const { return _m; }

      //! returns a negative duration

      duration_type get_neg_offset(const date_type& d) const

      {

        month_adjustor_type m_adj(_m.as_number());

        return duration_type(m_adj.get_neg_offset(d));

      }

      duration_type get_offset(const date_type& d) const

      {

        month_adjustor_type m_adj(_m.as_number());

        return duration_type(m_adj.get_offset(d));

      }

      bool operator==(const months_type& rhs) const

      {

        return(_m == rhs._m);

      }

      bool operator!=(const months_type& rhs) const

      {

        return(_m != rhs._m);

      }

      months_type operator+(const months_type& rhs)const

      {

        return months_type(_m + rhs._m);

      }

      months_type& operator+=(const months_type& rhs)

      {

        _m = _m + rhs._m;

        return *this;

      }

      months_type operator-(const months_type& rhs)const

      {

        return months_type(_m - rhs._m);

      }

      months_type& operator-=(const months_type& rhs)

      {

        _m = _m - rhs._m;

        return *this;

      }

      months_type operator*(const int_type rhs)const

      {

        return months_type(_m * rhs);

      }

      months_type& operator*=(const int_type rhs)

      {

        _m = _m * rhs;

        return *this;

      }

      months_type operator/(const int_type rhs)const

      {

        return months_type(_m / rhs);

      }

      months_type& operator/=(const int_type rhs)

      {

        _m = _m / rhs;

        return *this;

      }

      months_type operator+(const years_type& y)const

      {

        return months_type(y.number_of_years() * 12 + _m);

      }

      months_type& operator+=(const years_type& y)

      {

        _m = y.number_of_years() * 12 + _m;

        return *this;

      }

      months_type operator-(const years_type& y) const

      {

        return months_type(_m - y.number_of_years() * 12);

      }

      months_type& operator-=(const years_type& y)

      {

        _m = _m - y.number_of_years() * 12;

        return *this;

      }

      //

      friend date_type operator+(const date_type& d, const months_type& m)

      {

        return d + m.get_offset(d);

      }

      friend date_type operator+=(date_type& d, const months_type& m)

      {

        return d += m.get_offset(d);

      }

      friend date_type operator-(const date_type& d, const months_type& m)

      {

        // get_neg_offset returns a negative duration, so we add

        return d + m.get_neg_offset(d);

      }

      friend date_type operator-=(date_type& d, const months_type& m)

      {

        // get_neg_offset returns a negative duration, so we add

        return d += m.get_neg_offset(d);

      }

    private:

      int_rep _m;

  };

  //! additional duration type that represents a logical year

  /*! A logical year enables things like: "date(2002,Mar,2) + years(2) -> 

   * 2004-Mar-2". If the date is a last day-of-the-month, the result will 

   * also be a last-day-of-the-month (ie date(2001-Feb-28) + years(3) ->

   * 2004-Feb-29).

   */

  template<class base_config>

  class years_duration 

  {

    private:

      typedef typename base_config::int_rep int_rep;

      typedef typename int_rep::int_type int_type;

      typedef typename base_config::date_type date_type;

      typedef typename date_type::duration_type duration_type;

      typedef typename base_config::month_adjustor_type month_adjustor_type;

      typedef years_duration<base_config> years_type;

      typedef months_duration<base_config> months_type;

    public:

      years_duration(int_rep num) : _y(num) {}

      years_duration(special_values sv) : _y(sv) 

      {

        _y = int_rep::from_special(sv);

      }

      int_rep number_of_years() const { return _y; }

      //! returns a negative duration

      duration_type get_neg_offset(const date_type& d) const

      {

        month_adjustor_type m_adj(_y.as_number() * 12);

        return duration_type(m_adj.get_neg_offset(d));

      }

      duration_type get_offset(const date_type& d) const

      {

        month_adjustor_type m_adj(_y.as_number() * 12);

        return duration_type(m_adj.get_offset(d));

      }

      bool operator==(const years_type& rhs) const

      {

        return(_y == rhs._y);

      }

      bool operator!=(const years_type& rhs) const

      {

        return(_y != rhs._y);

      }

      years_type operator+(const years_type& rhs)const

      {

        return years_type(_y + rhs._y);

      }

      years_type& operator+=(const years_type& rhs)

      {

        _y = _y + rhs._y;

        return *this;

      }

      years_type operator-(const years_type& rhs)const

      {

        return years_type(_y - rhs._y);

      }

      years_type& operator-=(const years_type& rhs)

      {

        _y = _y - rhs._y;

        return *this;

      }

      years_type operator*(const int_type rhs)const

      {

        return years_type(_y * rhs);

      }

      years_type& operator*=(const int_type rhs)

      {

        _y = _y * rhs;

        return *this;

      }

      years_type operator/(const int_type rhs)const

      {

        return years_type(_y / rhs);

      }

      years_type& operator/=(const int_type rhs)

      {

        _y = _y / rhs;

        return *this;

      }

      months_type operator+(const months_type& m) const

      {

        return(months_type(_y * 12 + m.number_of_months()));

      }

      months_type operator-(const months_type& m) const

      {

        return(months_type(_y * 12 - m.number_of_months()));

      }

      //

      friend date_type operator+(const date_type& d, const years_type& y)

      {

        return d + y.get_offset(d);

      }

      friend date_type operator+=(date_type& d, const years_type& y)

      {

        return d += y.get_offset(d);

      }

      friend date_type operator-(const date_type& d, const years_type& y)

      {

        // get_neg_offset returns a negative duration, so we add

        return d + y.get_neg_offset(d);

      }

      friend date_type operator-=(date_type& d, const years_type& y)

      {

        // get_neg_offset returns a negative duration, so we add

        return d += y.get_neg_offset(d);

      }

    private:

      int_rep _y;

  };

}} // namespace boost::date_time

#endif // DATE_DURATION_TYPES_HPP___