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_IOMANIP
11#define _LIBCPP_IOMANIP
12
13/*
14    iomanip synopsis
15
16namespace std {
17
18// types T1, T2, ... are unspecified implementation types
19T1 resetiosflags(ios_base::fmtflags mask);
20T2 setiosflags (ios_base::fmtflags mask);
21T3 setbase(int base);
22template<charT> T4 setfill(charT c);
23T5 setprecision(int n);
24T6 setw(int n);
25template <class moneyT> T7 get_money(moneyT& mon, bool intl = false);
26template <class charT, class moneyT> T8 put_money(const moneyT& mon, bool intl = false);
27template <class charT> T9 get_time(struct tm* tmb, const charT* fmt);
28template <class charT> T10 put_time(const struct tm* tmb, const charT* fmt);
29
30template <class charT>
31  T11 quoted(const charT* s, charT delim=charT('"'), charT escape=charT('\\')); // C++14
32
33template <class charT, class traits, class Allocator>
34  T12 quoted(const basic_string<charT, traits, Allocator>& s,
35             charT delim=charT('"'), charT escape=charT('\\')); // C++14
36
37template <class charT, class traits, class Allocator>
38  T13 quoted(basic_string<charT, traits, Allocator>& s,
39             charT delim=charT('"'), charT escape=charT('\\')); // C++14
40
41}  // std
42
43*/
44
45#include <__assert> // all public C++ headers provide the assertion handler
46#include <__config>
47#include <istream>
48#include <version>
49
50#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
51#  pragma GCC system_header
52#endif
53
54_LIBCPP_BEGIN_NAMESPACE_STD
55
56// resetiosflags
57
58class __iom_t1
59{
60    ios_base::fmtflags __mask_;
61public:
62    _LIBCPP_INLINE_VISIBILITY
63    explicit __iom_t1(ios_base::fmtflags __m) : __mask_(__m) {}
64
65    template <class _CharT, class _Traits>
66    friend
67    _LIBCPP_INLINE_VISIBILITY
68    basic_istream<_CharT, _Traits>&
69    operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t1& __x)
70    {
71        __is.unsetf(__x.__mask_);
72        return __is;
73    }
74
75    template <class _CharT, class _Traits>
76    friend
77    _LIBCPP_INLINE_VISIBILITY
78    basic_ostream<_CharT, _Traits>&
79    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t1& __x)
80    {
81        __os.unsetf(__x.__mask_);
82        return __os;
83    }
84};
85
86inline _LIBCPP_INLINE_VISIBILITY
87__iom_t1
88resetiosflags(ios_base::fmtflags __mask)
89{
90    return __iom_t1(__mask);
91}
92
93// setiosflags
94
95class __iom_t2
96{
97    ios_base::fmtflags __mask_;
98public:
99    _LIBCPP_INLINE_VISIBILITY
100    explicit __iom_t2(ios_base::fmtflags __m) : __mask_(__m) {}
101
102    template <class _CharT, class _Traits>
103    friend
104    _LIBCPP_INLINE_VISIBILITY
105    basic_istream<_CharT, _Traits>&
106    operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t2& __x)
107    {
108        __is.setf(__x.__mask_);
109        return __is;
110    }
111
112    template <class _CharT, class _Traits>
113    friend
114    _LIBCPP_INLINE_VISIBILITY
115    basic_ostream<_CharT, _Traits>&
116    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t2& __x)
117    {
118        __os.setf(__x.__mask_);
119        return __os;
120    }
121};
122
123inline _LIBCPP_INLINE_VISIBILITY
124__iom_t2
125setiosflags(ios_base::fmtflags __mask)
126{
127    return __iom_t2(__mask);
128}
129
130// setbase
131
132class __iom_t3
133{
134    int __base_;
135public:
136    _LIBCPP_INLINE_VISIBILITY
137    explicit __iom_t3(int __b) : __base_(__b) {}
138
139    template <class _CharT, class _Traits>
140    friend
141    _LIBCPP_INLINE_VISIBILITY
142    basic_istream<_CharT, _Traits>&
143    operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t3& __x)
144    {
145        __is.setf(__x.__base_ == 8  ? ios_base::oct :
146                  __x.__base_ == 10 ? ios_base::dec :
147                  __x.__base_ == 16 ? ios_base::hex :
148                  ios_base::fmtflags(0), ios_base::basefield);
149        return __is;
150    }
151
152    template <class _CharT, class _Traits>
153    friend
154    _LIBCPP_INLINE_VISIBILITY
155    basic_ostream<_CharT, _Traits>&
156    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t3& __x)
157    {
158        __os.setf(__x.__base_ == 8  ? ios_base::oct :
159                  __x.__base_ == 10 ? ios_base::dec :
160                  __x.__base_ == 16 ? ios_base::hex :
161                  ios_base::fmtflags(0), ios_base::basefield);
162        return __os;
163    }
164};
165
166inline _LIBCPP_INLINE_VISIBILITY
167__iom_t3
168setbase(int __base)
169{
170    return __iom_t3(__base);
171}
172
173// setfill
174
175template<class _CharT>
176class __iom_t4
177{
178    _CharT __fill_;
179public:
180    _LIBCPP_INLINE_VISIBILITY
181    explicit __iom_t4(_CharT __c) : __fill_(__c) {}
182
183    template <class _Traits>
184    friend
185    _LIBCPP_INLINE_VISIBILITY
186    basic_ostream<_CharT, _Traits>&
187    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t4& __x)
188    {
189        __os.fill(__x.__fill_);
190        return __os;
191    }
192};
193
194template<class _CharT>
195inline _LIBCPP_INLINE_VISIBILITY
196__iom_t4<_CharT>
197setfill(_CharT __c)
198{
199    return __iom_t4<_CharT>(__c);
200}
201
202// setprecision
203
204class __iom_t5
205{
206    int __n_;
207public:
208    _LIBCPP_INLINE_VISIBILITY
209    explicit __iom_t5(int __n) : __n_(__n) {}
210
211    template <class _CharT, class _Traits>
212    friend
213    _LIBCPP_INLINE_VISIBILITY
214    basic_istream<_CharT, _Traits>&
215    operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t5& __x)
216    {
217        __is.precision(__x.__n_);
218        return __is;
219    }
220
221    template <class _CharT, class _Traits>
222    friend
223    _LIBCPP_INLINE_VISIBILITY
224    basic_ostream<_CharT, _Traits>&
225    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t5& __x)
226    {
227        __os.precision(__x.__n_);
228        return __os;
229    }
230};
231
232inline _LIBCPP_INLINE_VISIBILITY
233__iom_t5
234setprecision(int __n)
235{
236    return __iom_t5(__n);
237}
238
239// setw
240
241class __iom_t6
242{
243    int __n_;
244public:
245    _LIBCPP_INLINE_VISIBILITY
246    explicit __iom_t6(int __n) : __n_(__n) {}
247
248    template <class _CharT, class _Traits>
249    friend
250    _LIBCPP_INLINE_VISIBILITY
251    basic_istream<_CharT, _Traits>&
252    operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t6& __x)
253    {
254        __is.width(__x.__n_);
255        return __is;
256    }
257
258    template <class _CharT, class _Traits>
259    friend
260    _LIBCPP_INLINE_VISIBILITY
261    basic_ostream<_CharT, _Traits>&
262    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t6& __x)
263    {
264        __os.width(__x.__n_);
265        return __os;
266    }
267};
268
269inline _LIBCPP_INLINE_VISIBILITY
270__iom_t6
271setw(int __n)
272{
273    return __iom_t6(__n);
274}
275
276// get_money
277
278template <class _MoneyT> class __iom_t7;
279
280template <class _CharT, class _Traits, class _MoneyT>
281basic_istream<_CharT, _Traits>&
282operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t7<_MoneyT>& __x);
283
284template <class _MoneyT>
285class __iom_t7
286{
287    _MoneyT& __mon_;
288    bool __intl_;
289public:
290    _LIBCPP_INLINE_VISIBILITY
291    __iom_t7(_MoneyT& __mon, bool __intl)
292        : __mon_(__mon), __intl_(__intl) {}
293
294    template <class _CharT, class _Traits, class _Mp>
295    friend
296    basic_istream<_CharT, _Traits>&
297    operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t7<_Mp>& __x);
298};
299
300template <class _CharT, class _Traits, class _MoneyT>
301basic_istream<_CharT, _Traits>&
302operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t7<_MoneyT>& __x)
303{
304#ifndef _LIBCPP_NO_EXCEPTIONS
305    try
306    {
307#endif // _LIBCPP_NO_EXCEPTIONS
308        typename basic_istream<_CharT, _Traits>::sentry __s(__is);
309        if (__s)
310        {
311            typedef istreambuf_iterator<_CharT, _Traits> _Ip;
312            typedef money_get<_CharT, _Ip> _Fp;
313            ios_base::iostate __err = ios_base::goodbit;
314            const _Fp& __mf = use_facet<_Fp>(__is.getloc());
315            __mf.get(_Ip(__is), _Ip(), __x.__intl_, __is, __err, __x.__mon_);
316            __is.setstate(__err);
317        }
318#ifndef _LIBCPP_NO_EXCEPTIONS
319    }
320    catch (...)
321    {
322        __is.__set_badbit_and_consider_rethrow();
323    }
324#endif // _LIBCPP_NO_EXCEPTIONS
325    return __is;
326}
327
328template <class _MoneyT>
329inline _LIBCPP_INLINE_VISIBILITY
330__iom_t7<_MoneyT>
331get_money(_MoneyT& __mon, bool __intl = false)
332{
333    return __iom_t7<_MoneyT>(__mon, __intl);
334}
335
336// put_money
337
338template <class _MoneyT> class __iom_t8;
339
340template <class _CharT, class _Traits, class _MoneyT>
341basic_ostream<_CharT, _Traits>&
342operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t8<_MoneyT>& __x);
343
344template <class _MoneyT>
345class __iom_t8
346{
347    const _MoneyT& __mon_;
348    bool __intl_;
349public:
350    _LIBCPP_INLINE_VISIBILITY
351    __iom_t8(const _MoneyT& __mon, bool __intl)
352        : __mon_(__mon), __intl_(__intl) {}
353
354    template <class _CharT, class _Traits, class _Mp>
355    friend
356    basic_ostream<_CharT, _Traits>&
357    operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t8<_Mp>& __x);
358};
359
360template <class _CharT, class _Traits, class _MoneyT>
361basic_ostream<_CharT, _Traits>&
362operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t8<_MoneyT>& __x)
363{
364#ifndef _LIBCPP_NO_EXCEPTIONS
365    try
366    {
367#endif // _LIBCPP_NO_EXCEPTIONS
368        typename basic_ostream<_CharT, _Traits>::sentry __s(__os);
369        if (__s)
370        {
371            typedef ostreambuf_iterator<_CharT, _Traits> _Op;
372            typedef money_put<_CharT, _Op> _Fp;
373            const _Fp& __mf = use_facet<_Fp>(__os.getloc());
374            if (__mf.put(_Op(__os), __x.__intl_, __os, __os.fill(), __x.__mon_).failed())
375                __os.setstate(ios_base::badbit);
376        }
377#ifndef _LIBCPP_NO_EXCEPTIONS
378    }
379    catch (...)
380    {
381        __os.__set_badbit_and_consider_rethrow();
382    }
383#endif // _LIBCPP_NO_EXCEPTIONS
384    return __os;
385}
386
387template <class _MoneyT>
388inline _LIBCPP_INLINE_VISIBILITY
389__iom_t8<_MoneyT>
390put_money(const _MoneyT& __mon, bool __intl = false)
391{
392    return __iom_t8<_MoneyT>(__mon, __intl);
393}
394
395// get_time
396
397template <class _CharT> class __iom_t9;
398
399template <class _CharT, class _Traits>
400basic_istream<_CharT, _Traits>&
401operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t9<_CharT>& __x);
402
403template <class _CharT>
404class __iom_t9
405{
406    tm* __tm_;
407    const _CharT* __fmt_;
408public:
409    _LIBCPP_INLINE_VISIBILITY
410    __iom_t9(tm* __tm, const _CharT* __fmt)
411        : __tm_(__tm), __fmt_(__fmt) {}
412
413    template <class _Cp, class _Traits>
414    friend
415    basic_istream<_Cp, _Traits>&
416    operator>>(basic_istream<_Cp, _Traits>& __is, const __iom_t9<_Cp>& __x);
417};
418
419template <class _CharT, class _Traits>
420basic_istream<_CharT, _Traits>&
421operator>>(basic_istream<_CharT, _Traits>& __is, const __iom_t9<_CharT>& __x)
422{
423#ifndef _LIBCPP_NO_EXCEPTIONS
424    try
425    {
426#endif // _LIBCPP_NO_EXCEPTIONS
427        typename basic_istream<_CharT, _Traits>::sentry __s(__is);
428        if (__s)
429        {
430            typedef istreambuf_iterator<_CharT, _Traits> _Ip;
431            typedef time_get<_CharT, _Ip> _Fp;
432            ios_base::iostate __err = ios_base::goodbit;
433            const _Fp& __tf = use_facet<_Fp>(__is.getloc());
434            __tf.get(_Ip(__is), _Ip(), __is, __err, __x.__tm_,
435                     __x.__fmt_, __x.__fmt_ + _Traits::length(__x.__fmt_));
436            __is.setstate(__err);
437        }
438#ifndef _LIBCPP_NO_EXCEPTIONS
439    }
440    catch (...)
441    {
442        __is.__set_badbit_and_consider_rethrow();
443    }
444#endif // _LIBCPP_NO_EXCEPTIONS
445    return __is;
446}
447
448template <class _CharT>
449inline _LIBCPP_INLINE_VISIBILITY
450__iom_t9<_CharT>
451get_time(tm* __tm, const _CharT* __fmt)
452{
453    return __iom_t9<_CharT>(__tm, __fmt);
454}
455
456// put_time
457
458template <class _CharT> class __iom_t10;
459
460template <class _CharT, class _Traits>
461basic_ostream<_CharT, _Traits>&
462operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t10<_CharT>& __x);
463
464template <class _CharT>
465class __iom_t10
466{
467    const tm* __tm_;
468    const _CharT* __fmt_;
469public:
470    _LIBCPP_INLINE_VISIBILITY
471    __iom_t10(const tm* __tm, const _CharT* __fmt)
472        : __tm_(__tm), __fmt_(__fmt) {}
473
474    template <class _Cp, class _Traits>
475    friend
476    basic_ostream<_Cp, _Traits>&
477    operator<<(basic_ostream<_Cp, _Traits>& __os, const __iom_t10<_Cp>& __x);
478};
479
480template <class _CharT, class _Traits>
481basic_ostream<_CharT, _Traits>&
482operator<<(basic_ostream<_CharT, _Traits>& __os, const __iom_t10<_CharT>& __x)
483{
484#ifndef _LIBCPP_NO_EXCEPTIONS
485    try
486    {
487#endif // _LIBCPP_NO_EXCEPTIONS
488        typename basic_ostream<_CharT, _Traits>::sentry __s(__os);
489        if (__s)
490        {
491            typedef ostreambuf_iterator<_CharT, _Traits> _Op;
492            typedef time_put<_CharT, _Op> _Fp;
493            const _Fp& __tf = use_facet<_Fp>(__os.getloc());
494            if (__tf.put(_Op(__os), __os, __os.fill(), __x.__tm_,
495                         __x.__fmt_, __x.__fmt_ + _Traits::length(__x.__fmt_)).failed())
496                __os.setstate(ios_base::badbit);
497        }
498#ifndef _LIBCPP_NO_EXCEPTIONS
499    }
500    catch (...)
501    {
502        __os.__set_badbit_and_consider_rethrow();
503    }
504#endif // _LIBCPP_NO_EXCEPTIONS
505    return __os;
506}
507
508template <class _CharT>
509inline _LIBCPP_INLINE_VISIBILITY
510__iom_t10<_CharT>
511put_time(const tm* __tm, const _CharT* __fmt)
512{
513    return __iom_t10<_CharT>(__tm, __fmt);
514}
515
516#if _LIBCPP_STD_VER >= 11
517
518template <class _CharT, class _Traits>
519_LIBCPP_HIDE_FROM_ABI basic_ostream<_CharT, _Traits>&
520__quoted_output(basic_ostream<_CharT, _Traits>& __os,
521                const _CharT *__first, const _CharT *__last, _CharT __delim, _CharT __escape)
522{
523    basic_string<_CharT, _Traits> __str;
524    __str.push_back(__delim);
525    for (; __first != __last; ++__first) {
526        if (_Traits::eq(*__first, __escape) || _Traits::eq(*__first, __delim))
527            __str.push_back(__escape);
528        __str.push_back(*__first);
529    }
530    __str.push_back(__delim);
531    return _VSTD::__put_character_sequence(__os, __str.data(), __str.size());
532}
533
534template <class _CharT, class _Traits, class _String>
535_LIBCPP_HIDE_FROM_ABI basic_istream<_CharT, _Traits>&
536__quoted_input(basic_istream<_CharT, _Traits>& __is, _String& __string, _CharT __delim, _CharT __escape)
537{
538    __string.clear();
539    _CharT __c;
540    __is >> __c;
541    if (__is.fail())
542        return __is;
543
544    if (!_Traits::eq(__c, __delim)) {
545        // no delimiter, read the whole string
546        __is.unget();
547        __is >> __string;
548        return __is;
549    }
550
551    __save_flags<_CharT, _Traits> __sf(__is);
552    std::noskipws(__is);
553    while (true) {
554        __is >> __c;
555        if (__is.fail())
556            break;
557        if (_Traits::eq(__c, __escape)) {
558            __is >> __c;
559            if (__is.fail())
560                break;
561        } else if (_Traits::eq(__c, __delim))
562            break;
563        __string.push_back(__c);
564    }
565    return __is;
566}
567
568template <class _CharT, class _Traits>
569struct _LIBCPP_HIDDEN __quoted_output_proxy
570{
571    const _CharT *__first_;
572    const _CharT *__last_;
573    _CharT __delim_;
574    _CharT __escape_;
575
576    _LIBCPP_HIDE_FROM_ABI
577    explicit __quoted_output_proxy(const _CharT *__f, const _CharT *__l, _CharT __d, _CharT __e)
578        : __first_(__f), __last_(__l), __delim_(__d), __escape_(__e) {}
579
580    template<class _T2, __enable_if_t<_IsSame<_Traits, void>::value || _IsSame<_Traits, _T2>::value>* = nullptr>
581    friend _LIBCPP_HIDE_FROM_ABI basic_ostream<_CharT, _T2>&
582    operator<<(basic_ostream<_CharT, _T2>& __os, const __quoted_output_proxy& __p) {
583        return std::__quoted_output(__os, __p.__first_, __p.__last_, __p.__delim_, __p.__escape_);
584    }
585};
586
587template <class _CharT, class _Traits, class _Allocator>
588struct _LIBCPP_HIDDEN __quoted_proxy
589{
590    basic_string<_CharT, _Traits, _Allocator>& __string_;
591    _CharT __delim_;
592    _CharT __escape_;
593
594    _LIBCPP_HIDE_FROM_ABI
595    explicit __quoted_proxy(basic_string<_CharT, _Traits, _Allocator>& __s, _CharT __d, _CharT __e)
596        : __string_(__s), __delim_(__d), __escape_(__e) {}
597
598    friend _LIBCPP_HIDE_FROM_ABI basic_ostream<_CharT, _Traits>&
599    operator<<(basic_ostream<_CharT, _Traits>& __os, const __quoted_proxy& __p) {
600        return std::__quoted_output(__os, __p.__string_.data(), __p.__string_.data() + __p.__string_.size(), __p.__delim_, __p.__escape_);
601    }
602
603    friend _LIBCPP_HIDE_FROM_ABI basic_istream<_CharT, _Traits>&
604    operator>>(basic_istream<_CharT, _Traits>& __is, const __quoted_proxy& __p) {
605        return std::__quoted_input(__is, __p.__string_, __p.__delim_, __p.__escape_);
606    }
607};
608
609template <class _CharT, class _Traits, class _Allocator>
610_LIBCPP_HIDE_FROM_ABI
611__quoted_output_proxy<_CharT, _Traits>
612__quoted(const basic_string<_CharT, _Traits, _Allocator>& __s, _CharT __delim = _CharT('"'), _CharT __escape = _CharT('\\'))
613{
614    return __quoted_output_proxy<_CharT, _Traits>(__s.data(), __s.data() + __s.size(), __delim, __escape);
615}
616
617template <class _CharT, class _Traits, class _Allocator>
618_LIBCPP_HIDE_FROM_ABI
619__quoted_proxy<_CharT, _Traits, _Allocator>
620__quoted(basic_string<_CharT, _Traits, _Allocator>& __s, _CharT __delim = _CharT('"'), _CharT __escape = _CharT('\\'))
621{
622    return __quoted_proxy<_CharT, _Traits, _Allocator>(__s, __delim, __escape);
623}
624
625#endif // _LIBCPP_STD_VER >= 11
626
627#if _LIBCPP_STD_VER > 11
628
629template <class _CharT>
630_LIBCPP_HIDE_FROM_ABI
631auto quoted(const _CharT *__s, _CharT __delim = _CharT('"'), _CharT __escape = _CharT('\\'))
632{
633    const _CharT *__end = __s;
634    while (*__end) ++__end;
635    return __quoted_output_proxy<_CharT, void>(__s, __end, __delim, __escape);
636}
637
638template <class _CharT, class _Traits, class _Allocator>
639_LIBCPP_HIDE_FROM_ABI
640auto quoted(const basic_string<_CharT, _Traits, _Allocator>& __s, _CharT __delim = _CharT('"'), _CharT __escape = _CharT('\\'))
641{
642    return __quoted_output_proxy<_CharT, _Traits>(__s.data(), __s.data() + __s.size(), __delim, __escape);
643}
644
645template <class _CharT, class _Traits, class _Allocator>
646_LIBCPP_HIDE_FROM_ABI
647auto quoted(basic_string<_CharT, _Traits, _Allocator>& __s, _CharT __delim = _CharT('"'), _CharT __escape = _CharT('\\'))
648{
649    return __quoted_proxy<_CharT, _Traits, _Allocator>(__s, __delim, __escape);
650}
651
652template <class _CharT, class _Traits>
653_LIBCPP_HIDE_FROM_ABI
654auto quoted(basic_string_view<_CharT, _Traits> __sv, _CharT __delim = _CharT('"'), _CharT __escape = _CharT('\\'))
655{
656    return __quoted_output_proxy<_CharT, _Traits>(__sv.data(), __sv.data() + __sv.size(), __delim, __escape);
657}
658
659#endif // _LIBCPP_STD_VER > 11
660
661_LIBCPP_END_NAMESPACE_STD
662
663#endif // _LIBCPP_IOMANIP
664