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_WEEKDAY_H
11#define _LIBCPP___CHRONO_WEEKDAY_H
12
13#include <__chrono/calendar.h>
14#include <__chrono/duration.h>
15#include <__chrono/system_clock.h>
16#include <__chrono/time_point.h>
17#include <__config>
18#include <__cstddef/size_t.h>
19#include <__functional/hash.h>
20
21#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
22# pragma GCC system_header
23#endif
24
25#if _LIBCPP_STD_VER >= 20
26
27_LIBCPP_BEGIN_NAMESPACE_STD
28
29namespace chrono {
30
31class weekday_indexed;
32class weekday_last;
33
34class weekday {
35private:
36 unsigned char __wd_;
37 _LIBCPP_HIDE_FROM_ABI static constexpr unsigned char __weekday_from_days(int __days) noexcept;
38
39public:
40 weekday() = default;
41 _LIBCPP_HIDE_FROM_ABI inline explicit constexpr weekday(unsigned __val) noexcept
42 : __wd_(static_cast<unsigned char>(__val == 7 ? 0 : __val)) {}
43 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday(const sys_days& __sysd) noexcept
44 : __wd_(__weekday_from_days(days: __sysd.time_since_epoch().count())) {}
45 _LIBCPP_HIDE_FROM_ABI inline explicit constexpr weekday(const local_days& __locd) noexcept
46 : __wd_(__weekday_from_days(days: __locd.time_since_epoch().count())) {}
47
48 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday& operator++() noexcept {
49 __wd_ = (__wd_ == 6 ? 0 : __wd_ + 1);
50 return *this;
51 }
52 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday operator++(int) noexcept {
53 weekday __tmp = *this;
54 ++(*this);
55 return __tmp;
56 }
57 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday& operator--() noexcept {
58 __wd_ = (__wd_ == 0 ? 6 : __wd_ - 1);
59 return *this;
60 }
61 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday operator--(int) noexcept {
62 weekday __tmp = *this;
63 --(*this);
64 return __tmp;
65 }
66 _LIBCPP_HIDE_FROM_ABI constexpr weekday& operator+=(const days& __dd) noexcept;
67 _LIBCPP_HIDE_FROM_ABI constexpr weekday& operator-=(const days& __dd) noexcept;
68 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned c_encoding() const noexcept { return __wd_; }
69 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned iso_encoding() const noexcept {
70 return __wd_ == 0u ? 7 : __wd_;
71 }
72 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr bool ok() const noexcept { return __wd_ <= 6; }
73 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr weekday_indexed operator[](unsigned __index) const noexcept;
74 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr weekday_last operator[](last_spec) const noexcept;
75};
76
77// https://howardhinnant.github.io/date_algorithms.html#weekday_from_days
78_LIBCPP_HIDE_FROM_ABI inline constexpr unsigned char weekday::__weekday_from_days(int __days) noexcept {
79 return static_cast<unsigned char>(static_cast<unsigned>(__days >= -4 ? (__days + 4) % 7 : (__days + 5) % 7 + 6));
80}
81
82_LIBCPP_HIDE_FROM_ABI inline constexpr bool operator==(const weekday& __lhs, const weekday& __rhs) noexcept {
83 return __lhs.c_encoding() == __rhs.c_encoding();
84}
85
86[[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr weekday
87operator+(const weekday& __lhs, const days& __rhs) noexcept {
88 auto const __mu = static_cast<long long>(__lhs.c_encoding()) + __rhs.count();
89 auto const __yr = (__mu >= 0 ? __mu : __mu - 6) / 7;
90 return weekday{static_cast<unsigned>(__mu - __yr * 7)};
91}
92
93[[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr weekday
94operator+(const days& __lhs, const weekday& __rhs) noexcept {
95 return __rhs + __lhs;
96}
97
98[[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr weekday
99operator-(const weekday& __lhs, const days& __rhs) noexcept {
100 return __lhs + -__rhs;
101}
102
103[[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr days
104operator-(const weekday& __lhs, const weekday& __rhs) noexcept {
105 const int __wdu = __lhs.c_encoding() - __rhs.c_encoding();
106 const int __wk = (__wdu >= 0 ? __wdu : __wdu - 6) / 7;
107 return days{__wdu - __wk * 7};
108}
109
110_LIBCPP_HIDE_FROM_ABI inline constexpr weekday& weekday::operator+=(const days& __dd) noexcept {
111 *this = *this + __dd;
112 return *this;
113}
114
115_LIBCPP_HIDE_FROM_ABI inline constexpr weekday& weekday::operator-=(const days& __dd) noexcept {
116 *this = *this - __dd;
117 return *this;
118}
119
120class weekday_indexed {
121private:
122 chrono::weekday __wd_;
123 unsigned char __idx_;
124
125public:
126 weekday_indexed() = default;
127 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday_indexed(const chrono::weekday& __wdval, unsigned __idxval) noexcept
128 : __wd_{__wdval}, __idx_(__idxval) {}
129 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::weekday weekday() const noexcept { return __wd_; }
130 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned index() const noexcept { return __idx_; }
131 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI inline constexpr bool ok() const noexcept {
132 return __wd_.ok() && __idx_ >= 1 && __idx_ <= 5;
133 }
134};
135
136_LIBCPP_HIDE_FROM_ABI inline constexpr bool
137operator==(const weekday_indexed& __lhs, const weekday_indexed& __rhs) noexcept {
138 return __lhs.weekday() == __rhs.weekday() && __lhs.index() == __rhs.index();
139}
140
141class weekday_last {
142private:
143 chrono::weekday __wd_;
144
145public:
146 _LIBCPP_HIDE_FROM_ABI explicit constexpr weekday_last(const chrono::weekday& __val) noexcept : __wd_{__val} {}
147 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr chrono::weekday weekday() const noexcept { return __wd_; }
148 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI constexpr bool ok() const noexcept { return __wd_.ok(); }
149};
150
151_LIBCPP_HIDE_FROM_ABI inline constexpr bool operator==(const weekday_last& __lhs, const weekday_last& __rhs) noexcept {
152 return __lhs.weekday() == __rhs.weekday();
153}
154
155_LIBCPP_HIDE_FROM_ABI inline constexpr weekday_indexed weekday::operator[](unsigned __index) const noexcept {
156 return weekday_indexed{*this, __index};
157}
158
159_LIBCPP_HIDE_FROM_ABI inline constexpr weekday_last weekday::operator[](last_spec) const noexcept {
160 return weekday_last{*this};
161}
162
163inline constexpr weekday Sunday{0};
164inline constexpr weekday Monday{1};
165inline constexpr weekday Tuesday{2};
166inline constexpr weekday Wednesday{3};
167inline constexpr weekday Thursday{4};
168inline constexpr weekday Friday{5};
169inline constexpr weekday Saturday{6};
170
171} // namespace chrono
172
173# if _LIBCPP_STD_VER >= 26
174
175template <>
176struct hash<chrono::weekday> {
177 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI static size_t operator()(const chrono::weekday& __w) noexcept {
178 return __w.c_encoding();
179 }
180};
181
182template <>
183struct hash<chrono::weekday_indexed> {
184 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI static size_t operator()(const chrono::weekday_indexed& __wi) noexcept {
185 return std::__hash_combine(hash<chrono::weekday>{}(__wi.weekday()), __wi.index());
186 }
187};
188
189template <>
190struct hash<chrono::weekday_last> {
191 [[nodiscard]] _LIBCPP_HIDE_FROM_ABI static size_t operator()(const chrono::weekday_last& __wl) noexcept {
192 return hash<chrono::weekday>{}(__wl.weekday());
193 }
194};
195
196# endif // _LIBCPP_STD_VER >= 26
197
198_LIBCPP_END_NAMESPACE_STD
199
200#endif // _LIBCPP_STD_VER >= 20
201
202#endif // _LIBCPP___CHRONO_WEEKDAY_H
203