library/cpp/yson_pull/detail/stream_counter.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

#pragma once

#include <library/cpp/yson_pull/position_info.h>

#include <cstddef>

namespace NYsonPull { 
    namespace NDetail { 
        template <bool EnableLinePositionInfo> 
        class stream_counter; 

        template <> 
        class stream_counter<true> { 
        private: 
            size_t offset_ = 0; 
            size_t line_ = 1; 
            size_t column_ = 1; 

        public: 
            TPositionInfo info() const { 
                return {offset_, line_, column_}; 
            } 

            void update(const ui8* begin, const ui8* end) { 
                offset_ += end - begin; 
                for (auto current = begin; current != end; ++current) { 
                    ++column_; 
                    if (*current == '\n') { //TODO: memchr 
                        ++line_; 
                        column_ = 1; 
                    } 
                } 
            }
        }; 

        template <> 
        class stream_counter<false> { 
        private: 
            size_t offset_ = 0; 

        public: 
            TPositionInfo info() const { 
                return {offset_, {}, {}}; 
            } 

            void update(const ui8* begin, const ui8* end) { 
                offset_ += end - begin; 
            } 
        }; 
    }
}