bufferutils.h

Go to the documentation of this file.

 
#pragma once
 
struct SplitPush : Sink<Buffer<uint8_t>> {
    Sink<Buffer<uint8_t>> &left, &right;
    bool full() override {
        return left.full() || right.full();
    }
    void push(Buffer<uint8_t> &&in) override {
        left.push(in);
        right.push(in);
    }
    SplitPush(Sink<Buffer<uint8_t>> &left, Sink<Buffer<uint8_t>> &right) : left(left), right(right) {}
};
 
struct SplitPull {
    struct splitter : Source<Buffer<uint8_t>>, Sink<Buffer<uint8_t>> {
        Source<Buffer<uint8_t>> &from;
        Queue<Buffer<uint8_t>> q{20};
        Sink<Buffer<uint8_t>> *other;
        bool empty() override {
            while (!q.full() && !from.empty()) {
                auto b = from.pop();
                other->push(b);
                q.push(std::move(b));
            }
            return q.empty();
        }
        Buffer<uint8_t> pop() override {
            return q.pop();
        }
        bool full() override {
            return q.full();
        }
        void push(Buffer<uint8_t> &&t) override {
            q.push(std::move(t));
        }
        splitter(Source<Buffer<uint8_t>> &from) : from(from) { }
    };
    splitter a;
    splitter b;
    SplitPull(Source<Buffer<uint8_t>> &f): a(f), b(f) {
        a.other = &b;
        b.other = &a;
    }
};
 
struct Tee : Source<Buffer<uint8_t>> {
    Source<Buffer<uint8_t>> &from;
    Sink<Buffer<uint8_t>> &to;
    Queue<Buffer<uint8_t>> q{20};
    bool empty() override {
        while (!q.full() && !from.empty()) {
            auto b = from.pop();
            q.push(b);
            to.trypush(std::move(b));
        }
        return q.empty();
    }
    Buffer<uint8_t> pop() override {
        return q.pop();
    }
    Tee(Source<Buffer<uint8_t>> &from, Sink<Buffer<uint8_t>> &to)
        : from(from), to(to) { }
};
 
struct Hexify : Sink<Buffer<uint8_t>> {
    static constexpr uint8_t map[]="0123456789abcdef";
    static constexpr size_t blen = 512*3;
    Sink<Buffer<uint8_t>> &down;
    Buffer<uint8_t> work = blen;
    bool full() override {
        return down.full();
    }
    void push(Buffer<uint8_t> &&in) override {
        for (auto b: in) {
            work.append({'\\', map[b >> 4], map[b & 0xf]});
        }
        down.push(std::move(work));
        work = blen;
    }
    Hexify(Sink<Buffer<uint8_t>> &down) : down(down) {}
};