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(__sysd.time_since_epoch().count())) {}
45 _LIBCPP_HIDE_FROM_ABI inline explicit constexpr weekday(const local_days& __locd) noexcept
46 : __wd_(__weekday_from_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 _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned c_encoding() const noexcept { return __wd_; }
69 _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned iso_encoding() const noexcept { return __wd_ == 0u ? 7 : __wd_; }
70 _LIBCPP_HIDE_FROM_ABI inline constexpr bool ok() const noexcept { return __wd_ <= 6; }
71 _LIBCPP_HIDE_FROM_ABI constexpr weekday_indexed operator[](unsigned __index) const noexcept;
72 _LIBCPP_HIDE_FROM_ABI constexpr weekday_last operator[](last_spec) const noexcept;
73};
74
75// https://howardhinnant.github.io/date_algorithms.html#weekday_from_days
76_LIBCPP_HIDE_FROM_ABI inline constexpr unsigned char weekday::__weekday_from_days(int __days) noexcept {
77 return static_cast<unsigned char>(static_cast<unsigned>(__days >= -4 ? (__days + 4) % 7 : (__days + 5) % 7 + 6));
78}
79
80_LIBCPP_HIDE_FROM_ABI inline constexpr bool operator==(const weekday& __lhs, const weekday& __rhs) noexcept {
81 return __lhs.c_encoding() == __rhs.c_encoding();
82}
83
84_LIBCPP_HIDE_FROM_ABI inline constexpr weekday operator+(const weekday& __lhs, const days& __rhs) noexcept {
85 auto const __mu = static_cast<long long>(__lhs.c_encoding()) + __rhs.count();
86 auto const __yr = (__mu >= 0 ? __mu : __mu - 6) / 7;
87 return weekday{static_cast<unsigned>(__mu - __yr * 7)};
88}
89
90_LIBCPP_HIDE_FROM_ABI inline constexpr weekday operator+(const days& __lhs, const weekday& __rhs) noexcept {
91 return __rhs + __lhs;
92}
93
94_LIBCPP_HIDE_FROM_ABI inline constexpr weekday operator-(const weekday& __lhs, const days& __rhs) noexcept {
95 return __lhs + -__rhs;
96}
97
98_LIBCPP_HIDE_FROM_ABI inline constexpr days operator-(const weekday& __lhs, const weekday& __rhs) noexcept {
99 const int __wdu = __lhs.c_encoding() - __rhs.c_encoding();
100 const int __wk = (__wdu >= 0 ? __wdu : __wdu - 6) / 7;
101 return days{__wdu - __wk * 7};
102}
103
104_LIBCPP_HIDE_FROM_ABI inline constexpr weekday& weekday::operator+=(const days& __dd) noexcept {
105 *this = *this + __dd;
106 return *this;
107}
108
109_LIBCPP_HIDE_FROM_ABI inline constexpr weekday& weekday::operator-=(const days& __dd) noexcept {
110 *this = *this - __dd;
111 return *this;
112}
113
114class weekday_indexed {
115private:
116 chrono::weekday __wd_;
117 unsigned char __idx_;
118
119public:
120 weekday_indexed() = default;
121 _LIBCPP_HIDE_FROM_ABI inline constexpr weekday_indexed(const chrono::weekday& __wdval, unsigned __idxval) noexcept
122 : __wd_{__wdval}, __idx_(__idxval) {}
123 _LIBCPP_HIDE_FROM_ABI inline constexpr chrono::weekday weekday() const noexcept { return __wd_; }
124 _LIBCPP_HIDE_FROM_ABI inline constexpr unsigned index() const noexcept { return __idx_; }
125 _LIBCPP_HIDE_FROM_ABI inline constexpr bool ok() const noexcept { return __wd_.ok() && __idx_ >= 1 && __idx_ <= 5; }
126};
127
128_LIBCPP_HIDE_FROM_ABI inline constexpr bool
129operator==(const weekday_indexed& __lhs, const weekday_indexed& __rhs) noexcept {
130 return __lhs.weekday() == __rhs.weekday() && __lhs.index() == __rhs.index();
131}
132
133class weekday_last {
134private:
135 chrono::weekday __wd_;
136
137public:
138 _LIBCPP_HIDE_FROM_ABI explicit constexpr weekday_last(const chrono::weekday& __val) noexcept : __wd_{__val} {}
139 _LIBCPP_HIDE_FROM_ABI constexpr chrono::weekday weekday() const noexcept { return __wd_; }
140 _LIBCPP_HIDE_FROM_ABI constexpr bool ok() const noexcept { return __wd_.ok(); }
141};
142
143_LIBCPP_HIDE_FROM_ABI inline constexpr bool operator==(const weekday_last& __lhs, const weekday_last& __rhs) noexcept {
144 return __lhs.weekday() == __rhs.weekday();
145}
146
147_LIBCPP_HIDE_FROM_ABI inline constexpr weekday_indexed weekday::operator[](unsigned __index) const noexcept {
148 return weekday_indexed{*this, __index};
149}
150
151_LIBCPP_HIDE_FROM_ABI inline constexpr weekday_last weekday::operator[](last_spec) const noexcept {
152 return weekday_last{*this};
153}
154
155inline constexpr weekday Sunday{0};
156inline constexpr weekday Monday{1};
157inline constexpr weekday Tuesday{2};
158inline constexpr weekday Wednesday{3};
159inline constexpr weekday Thursday{4};
160inline constexpr weekday Friday{5};
161inline constexpr weekday Saturday{6};
162
163} // namespace chrono
164
165# if _LIBCPP_STD_VER >= 26
166
167template <>
168struct hash<chrono::weekday> {
169 _LIBCPP_HIDE_FROM_ABI static size_t operator()(const chrono::weekday& __w) noexcept { return __w.c_encoding(); }
170};
171
172template <>
173struct hash<chrono::weekday_indexed> {
174 _LIBCPP_HIDE_FROM_ABI static size_t operator()(const chrono::weekday_indexed& __wi) noexcept {
175 return std::__hash_combine(hash<chrono::weekday>{}(__wi.weekday()), __wi.index());
176 }
177};
178
179template <>
180struct hash<chrono::weekday_last> {
181 _LIBCPP_HIDE_FROM_ABI static size_t operator()(const chrono::weekday_last& __wl) noexcept {
182 return hash<chrono::weekday>{}(__wl.weekday());
183 }
184};
185
186# endif // _LIBCPP_STD_VER >= 26
187
188_LIBCPP_END_NAMESPACE_STD
189
190#endif // _LIBCPP_STD_VER >= 20
191
192#endif // _LIBCPP___CHRONO_WEEKDAY_H