On Mon, May 5, 2025 at 2:55 PM Tomasz Kamiński <tkami...@redhat.com> wrote:
> This patch provides an equality operator between _Sink_iter and > default_sentinel, > that returns true, if any further writes to the _Sink_iter and underlying > _Sink, > will be discared, and thus can be omitted. This operator is implemented in > terms > new of _M_discarding virtual function in _Sink. > > Currently only the _Padding_sink reports discarding mode of if width of > sequence > characters is greater than _M_maxwidth (precision), or underlying _Sink is > discarding characters. The _M_discarding override, is separate function > from > _M_ignoring, that remain annotated with [[__gnu__::__always_inline__]]. > > Despite having notion of maximum characters to be written (_M_max), > _Iter_sink > nevers discard characters, as the total number of characters that would be > written > needs to be returned by format_to_n. This is documented in-source by > providing an > _Iter_sink::_M_discarding override, that always returns false. > > The function is currently queried only by the _Padding_sinks, that may be > stacked > for example a range is formatted, with padding with being specified both > for range > itself and it's elements. The state of underlying sink is checked during > construction > and after each write (_M_sync_discarding). > > libstdc++-v3/ChangeLog: > > * include/std/format (__Sink_iter<_CharT>::operator==) > (__Sink<_CharT>::_M_discarding, _Iter_sink<_CharT, > _OutIter>::_M_discarding) > (_Padding_sinl<_CharT, _Out>::_M_padwidth) > (_Padding_sink<_CharT, _Out>::_M_maxwidth): Remove const. > (_Padding_sink<_CharT, _Out>::_M_sync_discarding) > (_Padding_sink<_CharT, _Out>::_M_discarding): Define. > (_Padding_sink<_CharT, _Out>::_Padding_sink(_Out, size_t, size_tt)) > (_Padding_sink<_CharT, _Out>::_M_force_update): > (_Padding_sink<_CharT, _Out>::_M_flush): Call _M_sync_discarding. > (_Padding_sink<_CharT, _Out>::_Padding_sink(_Out, size_t)): > Delegate. > --- > Tested on x86_64-linux. OK for trunk? > libstdc++-v3/include/std/format | 66 +++++++++++++++++++++++++++------ > 1 file changed, 54 insertions(+), 12 deletions(-) > > diff --git a/libstdc++-v3/include/std/format > b/libstdc++-v3/include/std/format > index 054ce350440..08a28fc92da 100644 > --- a/libstdc++-v3/include/std/format > +++ b/libstdc++-v3/include/std/format > @@ -3144,6 +3144,10 @@ namespace __format > auto > _M_reserve(size_t __n) const > { return _M_sink->_M_reserve(__n); } > + > + friend bool > + operator==(_Sink_iter __it, default_sentinel_t) > + { return __it._M_sink->_M_discarding(); } > }; > > // Abstract base class for type-erased character sinks. > @@ -3263,6 +3267,11 @@ namespace __format > _M_bump(size_t __n) > { _M_next += __n; } > > + // Returns true if the _Sink is discarding incoming characters. > + virtual bool > + _M_discarding() const > + { return false; } > + > public: > _Sink(const _Sink&) = delete; > _Sink& operator=(const _Sink&) = delete; > @@ -3488,6 +3497,15 @@ namespace __format > _M_count += __s.size(); > } > > + bool > + _M_discarding() const override > + { > + // format_to_n return total number of characters, that would be > written, > + // thus we need to see and all characters. > + // see https://eel.is/c%2B%2Bdraft/format#functions-22 > + return false; > + } > + > public: > [[__gnu__::__always_inline__]] > explicit > @@ -3550,6 +3568,15 @@ namespace __format > } > } > > + bool > + _M_discarding() const override > + { > + // format_to_n return total number of characters, that would be > written, > + // thus we need to see and all characters. > + // see https://eel.is/c%2B%2Bdraft/format#functions-22 > + return false; > + } > + > typename _Sink<_CharT>::_Reservation > _M_reserve(size_t __n) final > { > @@ -3636,17 +3663,15 @@ namespace __format > template<typename _Out, typename _CharT> > class _Padding_sink : public _Str_sink<_CharT> > { > - const size_t _M_padwidth; > - const size_t _M_maxwidth; > + size_t _M_padwidth; > + size_t _M_maxwidth; > _Out _M_out; > size_t _M_printwidth; > > [[__gnu__::__always_inline__]] > bool > _M_ignoring() const > - { > - return _M_printwidth >= _M_maxwidth; > - } > + { return _M_printwidth >= _M_maxwidth; } > > [[__gnu__::__always_inline__]] > bool > @@ -3659,12 +3684,21 @@ namespace __format > return false; > } > > + void > + _M_sync_discarding() > + { > + if constexpr (is_same_v<_Out, _Sink_iter<_CharT>>) > + if (_M_out == default_sentinel) > + _M_maxwidth = _M_printwidth; > + } > + > void > _M_flush() > { > span<_CharT> __new = this->_M_used(); > basic_string_view<_CharT> __str(__new.data(), __new.size()); > _M_out = __format::__write(std::move(_M_out), __str); > + _M_sync_discarding(); > this->_M_rewind(); > } > > @@ -3682,7 +3716,10 @@ namespace __format > // We have more characters than padidng, no padding is needed, > // write direclty to _M_out. > if (_M_printwidth >= _M_padwidth) > - _M_out = __format::__write(std::move(_M_out), __str); > + { > + _M_out = __format::__write(std::move(_M_out), __str); > + _M_sync_discarding(); > + } > // We reached _M_maxwidth that is smaller than _M_padwidth. > // Store the prefix sequence in _M_seq, and free _M_buf. > else > @@ -3718,6 +3755,10 @@ namespace __format > _Str_sink<_CharT>::_M_overflow(); > } > > + bool > + _M_discarding() const override > + { return _M_ignoring(); } > + > typename _Sink<_CharT>::_Reservation > _M_reserve(size_t __n) override > { > @@ -3752,15 +3793,16 @@ namespace __format > > public: > [[__gnu__::__always_inline__]] > - explicit _Padding_sink(_Out __out, size_t __padwidth) > - : _M_padwidth(__padwidth), _M_maxwidth(-1), > + explicit > + _Padding_sink(_Out __out, size_t __padwidth, size_t __maxwidth) > + : _M_padwidth(__padwidth), _M_maxwidth(__maxwidth), > _M_out(std::move(__out)), _M_printwidth(0) > - { } > + { _M_sync_discarding(); } > > [[__gnu__::__always_inline__]] > - explicit _Padding_sink(_Out __out, size_t __padwidth, size_t > __maxwidth) > - : _M_padwidth(__padwidth), _M_maxwidth(__maxwidth), > - _M_out(std::move(__out)), _M_printwidth(0) > + explicit > + _Padding_sink(_Out __out, size_t __padwidth) > + : _Padding_sink(std::move(__out), __padwidth, (size_t)-1) > { } > > _Out > -- > 2.49.0 > >