PrintBuffer.h

Go to the documentation of this file.

/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/SmingHub/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * PrintBuffer.h
 *
 ****/
 
#pragma once
 
#include <Data/Stream/ReadWriteStream.h>
#include <memory>
 
class BasePrintBuffer : public Print
{
public:
    BasePrintBuffer(Print& output, uint8_t buffer[], size_t bufferSize)
        : output(output), buffer(buffer), bufferSize(bufferSize)
    {
    }
 
    ~BasePrintBuffer()
    {
        flush();
    }
 
    size_t write(uint8_t c) override;
 
    size_t write(const uint8_t* data, size_t size) override;
 
    void flush();
 
private:
    Print& output;
    uint8_t* buffer;
    size_t bufferSize;
    size_t writeOffset{};
};
 
template <size_t size> class StaticPrintBuffer : public BasePrintBuffer
{
public:
    StaticPrintBuffer(Print& output) : BasePrintBuffer(output, buffer, size)
    {
    }
 
private:
    uint8_t buffer[size];
};
 
class HeapPrintBuffer : public BasePrintBuffer
{
public:
    HeapPrintBuffer(Print& output, size_t size) : HeapPrintBuffer(output, new uint8_t[size], size)
    {
    }
 
private:
    HeapPrintBuffer(Print& output, uint8_t* buffer, size_t size) : BasePrintBuffer(output, buffer, size), buffer(buffer)
    {
    }
 
    std::unique_ptr<uint8_t[]> buffer;
};
 
class DynamicPrintBuffer : public BasePrintBuffer
{
public:
    DynamicPrintBuffer(std::unique_ptr<Print>&& output, size_t size)
        : DynamicPrintBuffer(output.release(), new uint8_t[size], size)
    {
    }
 
    ~DynamicPrintBuffer()
    {
        flush();
    }
 
private:
    DynamicPrintBuffer(Print* output, uint8_t* buffer, size_t size)
        : BasePrintBuffer(*output, buffer, size), output(output), buffer(buffer)
    {
    }
 
    std::unique_ptr<Print> output;
    std::unique_ptr<uint8_t[]> buffer;
};