ostream.h

Go to the documentation of this file.

 
#ifndef ostream_h
#define ostream_h
#include "ios.h"
//==============================================================================
class ostream : public virtual ios
{
  public:
    ostream()
    {
    }
 
    ostream &operator<<(ostream &(*pf)(ostream &str))
    {
        return pf(*this);
    }
    ostream &operator<<(ios_base &(*pf)(ios_base &str))
    {
        pf(*this);
        return *this;
    }
    ostream &operator<<(bool arg)
    {
        putBool(arg);
        return *this;
    }
    ostream &operator<<(const char *arg)
    {
        putStr(arg);
        return *this;
    }
    ostream &operator<<(const signed char *arg)
    {
        putStr((const char *)arg);
        return *this;
    }
    ostream &operator<<(const unsigned char *arg)
    {
        putStr((const char *)arg);
        return *this;
    }
#if ENABLE_ARDUINO_STRING
    ostream &operator<<(const String &arg)
    {
        putStr(arg.c_str());
        return *this;
    }
#endif // ENABLE_ARDUINO_STRING
    ostream &operator<<(char arg)
    {
        putChar(arg);
        return *this;
    }
    ostream &operator<<(signed char arg)
    {
        putChar(static_cast<char>(arg));
        return *this;
    }
    ostream &operator<<(unsigned char arg)
    {
        putChar(static_cast<char>(arg));
        return *this;
    }
    ostream &operator<<(double arg)
    {
        putDouble(arg);
        return *this;
    }
    ostream &operator<<(float arg)
    {
        putDouble(arg);
        return *this;
    }
    ostream &operator<<(short arg)
    { // NOLINT
        putNum((int32_t)arg);
        return *this;
    }
    ostream &operator<<(unsigned short arg)
    { // NOLINT
        putNum((uint32_t)arg);
        return *this;
    }
    ostream &operator<<(int arg)
    {
        putNum((int32_t)arg);
        return *this;
    }
    ostream &operator<<(unsigned int arg)
    {
        putNum((uint32_t)arg);
        return *this;
    }
    ostream &operator<<(long arg)
    { // NOLINT
        putNum((int32_t)arg);
        return *this;
    }
    ostream &operator<<(unsigned long arg)
    { // NOLINT
        putNum((uint32_t)arg);
        return *this;
    }
    ostream &operator<<(long long arg)
    { // NOLINT
        putNum((int64_t)arg);
        return *this;
    }
    ostream &operator<<(unsigned long long arg)
    { // NOLINT
        putNum((uint64_t)arg);
        return *this;
    }
    ostream &operator<<(const void *arg)
    {
        putNum(reinterpret_cast<uint32_t>(arg));
        return *this;
    }
    ostream &operator<<(const __FlashStringHelper *arg)
    {
        putPgm(reinterpret_cast<const char *>(arg));
        return *this;
    }
    ostream &put(char ch)
    {
        putch(ch);
        return *this;
    }
    //  ostream& write(char *str, streamsize count);
    ostream &flush()
    {
        if (!sync())
        {
            setstate(badbit);
        }
        return *this;
    }
    pos_type tellp()
    {
        return tellpos();
    }
    ostream &seekp(pos_type pos)
    {
        if (!seekpos(pos))
        {
            setstate(failbit);
        }
        return *this;
    }
    ostream &seekp(off_type off, seekdir way)
    {
        if (!seekoff(off, way))
        {
            setstate(failbit);
        }
        return *this;
    }
 
  protected:
 
    virtual void putch(char ch) = 0;
    virtual void putstr(const char *str) = 0;
    virtual bool seekoff(off_type pos, seekdir way) = 0;
    virtual bool seekpos(pos_type pos) = 0;
    virtual bool sync() = 0;
    virtual pos_type tellpos() = 0;
  private:
    void do_fill(unsigned len);
    void fill_not_left(unsigned len);
    void putBool(bool b);
    void putChar(char c);
    void putDouble(double n);
    void putNum(int32_t n);
    void putNum(int64_t n);
    void putNum(uint32_t n)
    {
        putNum(n, false);
    }
    void putNum(uint64_t n)
    {
        putNum(n, false);
    }
    void putPgm(const char *str);
    void putStr(const char *str);
 
    template <typename T> char *fmtNum(T n, char *ptr, uint8_t base)
    {
        char a = flags() & uppercase ? 'A' - 10 : 'a' - 10;
        do
        {
            T m = n;
            n /= base;
            char c = m - base * n;
            *--ptr = c < 10 ? c + '0' : c + a;
        } while (n);
        return ptr;
    }
 
    template <typename T> void putNum(T n, bool neg)
    {
        char buf[(8 * sizeof(T) + 2) / 3 + 2];
        char *ptr = buf + sizeof(buf) - 1;
        char *num;
        char *str;
        uint8_t base = flagsToBase();
        *ptr = '\0';
        str = num = fmtNum(n, ptr, base);
        if (base == 10)
        {
            if (neg)
            {
                *--str = '-';
            }
            else if (flags() & showpos)
            {
                *--str = '+';
            }
        }
        else if (flags() & showbase)
        {
            if (flags() & hex)
            {
                *--str = flags() & uppercase ? 'X' : 'x';
            }
            *--str = '0';
        }
        uint8_t len = ptr - str;
        fmtflags adj = flags() & adjustfield;
        if (adj == internal)
        {
            while (str < num)
            {
                putch(*str++);
            }
            do_fill(len);
        }
        else
        {
            // do fill for right
            fill_not_left(len);
        }
        putstr(str);
        do_fill(len);
    }
};
#endif // ostream_h