StdioDevice.h

/*
    StdioDevice.h 
    Copyright (c) 2025 Phil Schatzmann. All right reserved.
 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.
 
    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.
 
    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#pragma once
 
#include <iostream>
#include <streambuf>
#include "api/Stream.h"
#include "api/Printable.h"
 
namespace arduino {
 
class StdioDevice : public Stream {
 public:
  StdioDevice(bool autoFlush = true) { auto_flush = autoFlush; }
 
  ~StdioDevice() {}
 
  operator bool() const {
    return true;
  }  // For classic while(!Serial) { ... } pattern for USB ready wait
 
  virtual void begin(int speed) {
    // nothing to be done
  }
 
  virtual size_t print(const char* str) {
    std::cout << str;
    if (auto_flush) flush();
    return strlen(str);
  }
 
  virtual size_t println(const char* str = "") {
    std::cout << str << "\n";
    if (auto_flush) flush();
    return strlen(str) + 1;
  }
 
  virtual size_t print(int val, int radix = DEC) {
    size_t result = Stream::print(val, radix);
    if (auto_flush) flush();
    return result;
  }
 
  virtual size_t println(int val, int radix = DEC) {
    size_t result = Stream::println(val, radix);
    if (auto_flush) flush();
    return result;
  }
 
  virtual size_t println(String& str) { return println(str.c_str()); }
 
  virtual size_t print(String& str) { return print(str.c_str()); }
 
  virtual size_t println(Printable& p) {
    size_t result = p.printTo(*this);
    std::cout << "\n";
    if (auto_flush) flush();
    return result + 1;
  }
 
  virtual size_t print(Printable& p) {
    auto result = p.printTo(*this);
    if (auto_flush) flush();
    return result;
  }
 
  void flush() override { std::cout.flush(); }
 
  virtual size_t write(const char* str, size_t len) {
    std::cout.write(str, len);
    if (auto_flush) flush();
    return len;
  }
 
  virtual size_t write(uint8_t* str, size_t len) {
    std::cout.write((const char*)str, len);
    if (auto_flush) flush();
    return len;
  }
  size_t write(const uint8_t* str, size_t len) override {
    std::cout.write((const char*)str, len);
    if (auto_flush) flush();
    return len;
  }
 
  virtual size_t write(int32_t value) {
    std::cout.put(value);
    if (auto_flush) flush();
    return 1;
  }
 
  size_t write(uint8_t value) override {
    std::cout.put(value);
    if (auto_flush) flush();
    return 1;
  }
 
  int available() override { return std::cin.rdbuf()->in_avail(); };
 
  int read() override { return std::cin.get(); }
 
  int peek() override { return std::cin.peek(); }
 
 protected:
  bool auto_flush = true;
};
 
static StdioDevice Serial;
#ifndef USE_RPI
static StdioDevice Serial2;
#endif
 
}  // namespace arduino