1 // -*- C++ -*- 2 //===----------------------------------------------------------------------===// 3 // 4 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 5 // See https://llvm.org/LICENSE.txt for license information. 6 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 7 // 8 //===----------------------------------------------------------------------===// 9 10 #ifndef _LIBCPP___CHRONO_YEAR_MONTH_WEEKDAY_H 11 #define _LIBCPP___CHRONO_YEAR_MONTH_WEEKDAY_H 12 13 #include <__chrono/calendar.h> 14 #include <__chrono/day.h> 15 #include <__chrono/duration.h> 16 #include <__chrono/month.h> 17 #include <__chrono/month_weekday.h> 18 #include <__chrono/system_clock.h> 19 #include <__chrono/time_point.h> 20 #include <__chrono/weekday.h> 21 #include <__chrono/year.h> 22 #include <__chrono/year_month.h> 23 #include <__chrono/year_month_day.h> 24 #include <__config> 25 26 #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER) 27 # pragma GCC system_header 28 #endif 29 30 #if _LIBCPP_STD_VER >= 20 31 32 _LIBCPP_BEGIN_NAMESPACE_STD 33 34 namespace chrono 35 { 36 37 class year_month_weekday { 38 chrono::year __y_; 39 chrono::month __m_; 40 chrono::weekday_indexed __wdi_; 41 public: 42 year_month_weekday() = default; 43 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday(const chrono::year& __yval, const chrono::month& __mval, 44 const chrono::weekday_indexed& __wdival) noexcept 45 : __y_{__yval}, __m_{__mval}, __wdi_{__wdival} {} 46 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday(const sys_days& __sysd) noexcept 47 : year_month_weekday(__from_days(__sysd.time_since_epoch())) {} 48 _LIBCPP_HIDE_FROM_ABI inline explicit constexpr year_month_weekday(const local_days& __locd) noexcept 49 : year_month_weekday(__from_days(__locd.time_since_epoch())) {} 50 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday& operator+=(const months&) noexcept; 51 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday& operator-=(const months&) noexcept; 52 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday& operator+=(const years&) noexcept; 53 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday& operator-=(const years&) noexcept; 54 55 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::year year() const noexcept { return __y_; } 56 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::month month() const noexcept { return __m_; } 57 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::weekday weekday() const noexcept { return __wdi_.weekday(); } 58 _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned index() const noexcept { return __wdi_.index(); } 59 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::weekday_indexed weekday_indexed() const noexcept { return __wdi_; } 60 61 _LIBCPP_HIDE_FROM_ABI inline constexpr operator sys_days() const noexcept { return sys_days{__to_days()}; } 62 _LIBCPP_HIDE_FROM_ABI inline explicit constexpr operator local_days() const noexcept { return local_days{__to_days()}; } 63 _LIBCPP_HIDE_FROM_ABI inline constexpr bool ok() const noexcept 64 { 65 if (!__y_.ok() || !__m_.ok() || !__wdi_.ok()) return false; 66 if (__wdi_.index() <= 4) return true; 67 auto __nth_weekday_day = 68 __wdi_.weekday() - 69 chrono::weekday{static_cast<sys_days>(__y_ / __m_ / 1)} + 70 days{(__wdi_.index() - 1) * 7 + 1}; 71 return static_cast<unsigned>(__nth_weekday_day.count()) <= 72 static_cast<unsigned>((__y_ / __m_ / last).day()); 73 } 74 75 _LIBCPP_HIDE_FROM_ABI static constexpr year_month_weekday __from_days(days __d) noexcept; 76 _LIBCPP_HIDE_FROM_ABI constexpr days __to_days() const noexcept; 77 }; 78 79 _LIBCPP_HIDE_FROM_ABI inline constexpr 80 year_month_weekday year_month_weekday::__from_days(days __d) noexcept 81 { 82 const sys_days __sysd{__d}; 83 const chrono::weekday __wd = chrono::weekday(__sysd); 84 const year_month_day __ymd = year_month_day(__sysd); 85 return year_month_weekday{__ymd.year(), __ymd.month(), 86 __wd[(static_cast<unsigned>(__ymd.day())-1)/7+1]}; 87 } 88 89 _LIBCPP_HIDE_FROM_ABI inline constexpr 90 days year_month_weekday::__to_days() const noexcept 91 { 92 const sys_days __sysd = sys_days(__y_/__m_/1); 93 return (__sysd + (__wdi_.weekday() - chrono::weekday(__sysd) + days{(__wdi_.index()-1)*7})) 94 .time_since_epoch(); 95 } 96 97 _LIBCPP_HIDE_FROM_ABI inline constexpr 98 bool operator==(const year_month_weekday& __lhs, const year_month_weekday& __rhs) noexcept 99 { return __lhs.year() == __rhs.year() && __lhs.month() == __rhs.month() && __lhs.weekday_indexed() == __rhs.weekday_indexed(); } 100 101 _LIBCPP_HIDE_FROM_ABI inline constexpr 102 year_month_weekday operator/(const year_month& __lhs, const weekday_indexed& __rhs) noexcept 103 { return year_month_weekday{__lhs.year(), __lhs.month(), __rhs}; } 104 105 _LIBCPP_HIDE_FROM_ABI inline constexpr 106 year_month_weekday operator/(const year& __lhs, const month_weekday& __rhs) noexcept 107 { return year_month_weekday{__lhs, __rhs.month(), __rhs.weekday_indexed()}; } 108 109 _LIBCPP_HIDE_FROM_ABI inline constexpr 110 year_month_weekday operator/(int __lhs, const month_weekday& __rhs) noexcept 111 { return year(__lhs) / __rhs; } 112 113 _LIBCPP_HIDE_FROM_ABI inline constexpr 114 year_month_weekday operator/(const month_weekday& __lhs, const year& __rhs) noexcept 115 { return __rhs / __lhs; } 116 117 _LIBCPP_HIDE_FROM_ABI inline constexpr 118 year_month_weekday operator/(const month_weekday& __lhs, int __rhs) noexcept 119 { return year(__rhs) / __lhs; } 120 121 122 _LIBCPP_HIDE_FROM_ABI inline constexpr 123 year_month_weekday operator+(const year_month_weekday& __lhs, const months& __rhs) noexcept 124 { return (__lhs.year() / __lhs.month() + __rhs) / __lhs.weekday_indexed(); } 125 126 _LIBCPP_HIDE_FROM_ABI inline constexpr 127 year_month_weekday operator+(const months& __lhs, const year_month_weekday& __rhs) noexcept 128 { return __rhs + __lhs; } 129 130 _LIBCPP_HIDE_FROM_ABI inline constexpr 131 year_month_weekday operator-(const year_month_weekday& __lhs, const months& __rhs) noexcept 132 { return __lhs + (-__rhs); } 133 134 _LIBCPP_HIDE_FROM_ABI inline constexpr 135 year_month_weekday operator+(const year_month_weekday& __lhs, const years& __rhs) noexcept 136 { return year_month_weekday{__lhs.year() + __rhs, __lhs.month(), __lhs.weekday_indexed()}; } 137 138 _LIBCPP_HIDE_FROM_ABI inline constexpr 139 year_month_weekday operator+(const years& __lhs, const year_month_weekday& __rhs) noexcept 140 { return __rhs + __lhs; } 141 142 _LIBCPP_HIDE_FROM_ABI inline constexpr 143 year_month_weekday operator-(const year_month_weekday& __lhs, const years& __rhs) noexcept 144 { return __lhs + (-__rhs); } 145 146 147 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday& year_month_weekday::operator+=(const months& __dm) noexcept { *this = *this + __dm; return *this; } 148 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday& year_month_weekday::operator-=(const months& __dm) noexcept { *this = *this - __dm; return *this; } 149 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday& year_month_weekday::operator+=(const years& __dy) noexcept { *this = *this + __dy; return *this; } 150 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday& year_month_weekday::operator-=(const years& __dy) noexcept { *this = *this - __dy; return *this; } 151 152 class year_month_weekday_last { 153 private: 154 chrono::year __y_; 155 chrono::month __m_; 156 chrono::weekday_last __wdl_; 157 public: 158 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday_last(const chrono::year& __yval, const chrono::month& __mval, 159 const chrono::weekday_last& __wdlval) noexcept 160 : __y_{__yval}, __m_{__mval}, __wdl_{__wdlval} {} 161 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday_last& operator+=(const months& __dm) noexcept; 162 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday_last& operator-=(const months& __dm) noexcept; 163 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday_last& operator+=(const years& __dy) noexcept; 164 _LIBCPP_HIDE_FROM_ABI constexpr year_month_weekday_last& operator-=(const years& __dy) noexcept; 165 166 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::year year() const noexcept { return __y_; } 167 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::month month() const noexcept { return __m_; } 168 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::weekday weekday() const noexcept { return __wdl_.weekday(); } 169 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::weekday_last weekday_last() const noexcept { return __wdl_; } 170 _LIBCPP_HIDE_FROM_ABI inline constexpr operator sys_days() const noexcept { return sys_days{__to_days()}; } 171 _LIBCPP_HIDE_FROM_ABI inline explicit constexpr operator local_days() const noexcept { return local_days{__to_days()}; } 172 _LIBCPP_HIDE_FROM_ABI inline constexpr bool ok() const noexcept { return __y_.ok() && __m_.ok() && __wdl_.ok(); } 173 174 _LIBCPP_HIDE_FROM_ABI constexpr days __to_days() const noexcept; 175 176 }; 177 178 _LIBCPP_HIDE_FROM_ABI inline constexpr 179 days year_month_weekday_last::__to_days() const noexcept 180 { 181 const sys_days __last = sys_days{__y_/__m_/last}; 182 return (__last - (chrono::weekday{__last} - __wdl_.weekday())).time_since_epoch(); 183 184 } 185 186 _LIBCPP_HIDE_FROM_ABI inline constexpr 187 bool operator==(const year_month_weekday_last& __lhs, const year_month_weekday_last& __rhs) noexcept 188 { return __lhs.year() == __rhs.year() && __lhs.month() == __rhs.month() && __lhs.weekday_last() == __rhs.weekday_last(); } 189 190 _LIBCPP_HIDE_FROM_ABI inline constexpr 191 year_month_weekday_last operator/(const year_month& __lhs, const weekday_last& __rhs) noexcept 192 { return year_month_weekday_last{__lhs.year(), __lhs.month(), __rhs}; } 193 194 _LIBCPP_HIDE_FROM_ABI inline constexpr 195 year_month_weekday_last operator/(const year& __lhs, const month_weekday_last& __rhs) noexcept 196 { return year_month_weekday_last{__lhs, __rhs.month(), __rhs.weekday_last()}; } 197 198 _LIBCPP_HIDE_FROM_ABI inline constexpr 199 year_month_weekday_last operator/(int __lhs, const month_weekday_last& __rhs) noexcept 200 { return year(__lhs) / __rhs; } 201 202 _LIBCPP_HIDE_FROM_ABI inline constexpr 203 year_month_weekday_last operator/(const month_weekday_last& __lhs, const year& __rhs) noexcept 204 { return __rhs / __lhs; } 205 206 _LIBCPP_HIDE_FROM_ABI inline constexpr 207 year_month_weekday_last operator/(const month_weekday_last& __lhs, int __rhs) noexcept 208 { return year(__rhs) / __lhs; } 209 210 211 _LIBCPP_HIDE_FROM_ABI inline constexpr 212 year_month_weekday_last operator+(const year_month_weekday_last& __lhs, const months& __rhs) noexcept 213 { return (__lhs.year() / __lhs.month() + __rhs) / __lhs.weekday_last(); } 214 215 _LIBCPP_HIDE_FROM_ABI inline constexpr 216 year_month_weekday_last operator+(const months& __lhs, const year_month_weekday_last& __rhs) noexcept 217 { return __rhs + __lhs; } 218 219 _LIBCPP_HIDE_FROM_ABI inline constexpr 220 year_month_weekday_last operator-(const year_month_weekday_last& __lhs, const months& __rhs) noexcept 221 { return __lhs + (-__rhs); } 222 223 _LIBCPP_HIDE_FROM_ABI inline constexpr 224 year_month_weekday_last operator+(const year_month_weekday_last& __lhs, const years& __rhs) noexcept 225 { return year_month_weekday_last{__lhs.year() + __rhs, __lhs.month(), __lhs.weekday_last()}; } 226 227 _LIBCPP_HIDE_FROM_ABI inline constexpr 228 year_month_weekday_last operator+(const years& __lhs, const year_month_weekday_last& __rhs) noexcept 229 { return __rhs + __lhs; } 230 231 _LIBCPP_HIDE_FROM_ABI inline constexpr 232 year_month_weekday_last operator-(const year_month_weekday_last& __lhs, const years& __rhs) noexcept 233 { return __lhs + (-__rhs); } 234 235 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday_last& year_month_weekday_last::operator+=(const months& __dm) noexcept { *this = *this + __dm; return *this; } 236 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday_last& year_month_weekday_last::operator-=(const months& __dm) noexcept { *this = *this - __dm; return *this; } 237 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday_last& year_month_weekday_last::operator+=(const years& __dy) noexcept { *this = *this + __dy; return *this; } 238 _LIBCPP_HIDE_FROM_ABI inline constexpr year_month_weekday_last& year_month_weekday_last::operator-=(const years& __dy) noexcept { *this = *this - __dy; return *this; } 239 240 } // namespace chrono 241 242 _LIBCPP_END_NAMESPACE_STD 243 244 #endif // _LIBCPP_STD_VER >= 20 245 246 #endif // _LIBCPP___CHRONO_YEAR_MONTH_WEEKDAY_H 247