aboutsummaryrefslogtreecommitdiff
path: root/src/libutil/istringstream_nocopy.hh
blob: f7beac578e39c4d5a5b6f22d88e012698ebfa273 (plain)
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/* This file provides a variant of std::istringstream that doesn't
   copy its string argument. This is useful for large strings. The
   caller must ensure that the string object is not destroyed while
   it's referenced by this object. */

#pragma once

#include <string>
#include <iostream>

template <class CharT, class Traits = std::char_traits<CharT>, class Allocator = std::allocator<CharT>>
class basic_istringbuf_nocopy : public std::basic_streambuf<CharT, Traits>
{
public:
    typedef std::basic_string<CharT, Traits, Allocator> string_type;

    typedef typename std::basic_streambuf<CharT, Traits>::off_type off_type;

    typedef typename std::basic_streambuf<CharT, Traits>::pos_type pos_type;

    typedef typename std::basic_streambuf<CharT, Traits>::int_type int_type;

    typedef typename std::basic_streambuf<CharT, Traits>::traits_type traits_type;

private:
    const string_type & s;

    off_type off;

public:
    basic_istringbuf_nocopy(const string_type & s) : s{s}, off{0}
    {
    }

private:
    pos_type seekoff(off_type off, std::ios_base::seekdir dir, std::ios_base::openmode which)
    {
        if (which & std::ios_base::in) {
            this->off = dir == std::ios_base::beg
                ? off
                : (dir == std::ios_base::end
                    ? s.size() + off
                    : this->off + off);
        }
        return pos_type(this->off);
    }

    pos_type seekpos(pos_type pos, std::ios_base::openmode which)
    {
        return seekoff(pos, std::ios_base::beg, which);
    }

    std::streamsize showmanyc()
    {
        return s.size() - off;
    }

    int_type underflow()
    {
        if (typename string_type::size_type(off) == s.size())
            return traits_type::eof();
        return traits_type::to_int_type(s[off]);
    }

    int_type uflow()
    {
        if (typename string_type::size_type(off) == s.size())
            return traits_type::eof();
        return traits_type::to_int_type(s[off++]);
    }

    int_type pbackfail(int_type ch)
    {
        if (off == 0 || (ch != traits_type::eof() && ch != s[off - 1]))
            return traits_type::eof();

        return traits_type::to_int_type(s[--off]);
    }

};

template <class CharT, class Traits = std::char_traits<CharT>, class Allocator = std::allocator<CharT>>
class basic_istringstream_nocopy : public std::basic_iostream<CharT, Traits>
{
    typedef basic_istringbuf_nocopy<CharT, Traits, Allocator> buf_type;
    buf_type buf;
public:
    basic_istringstream_nocopy(const typename buf_type::string_type & s) :
        std::basic_iostream<CharT, Traits>(&buf), buf(s) {};
};

typedef basic_istringstream_nocopy<char> istringstream_nocopy;