cfstream

fbuf

Line Code
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
// original code by the Minima Authors under the MIT license. See AUTHORS for the list of authors.

#include "popenCPP.h"
#include "FWCore/Utilities/interface/Exception.h"
#include <cstdio>
#include <string>
#include <istream>

namespace {
  class fbuf final : public std::streambuf {
    FILE *file;
    char *buf;
    constexpr static size_t bufsz = 4096;

  public:
    fbuf(FILE *f);
    ~fbuf() override;

  protected:
    std::streambuf *setbuf(char_type *s, std::streamsize n) override;
    int sync() override;
    int_type underflow() override;
  };

  class cfstream final : public std::istream {
    fbuf buf;

  public:
    cfstream(FILE *f);
  };
}  // namespace

std::unique_ptr<std::istream> reco::exprEvalDetails::popenCPP(const std::string &cmdline) {
  FILE *f = popen(cmdline.c_str(), "r");
  if (!f)
    throw cms::Exception("PopenCPP", "(\"" + cmdline + "\") failed");
  return std::unique_ptr<std::istream>(new cfstream(f));
}

fbuf::fbuf(FILE *f) : file(f), buf(new char[bufsz]) { this->setg(this->buf, this->buf + bufsz, this->buf + bufsz); }

fbuf::~fbuf() {
  delete[] this->buf;
  pclose(this->file);
}

std::streambuf *fbuf::setbuf(char_type *s, std::streamsize n) { return nullptr; }

int fbuf::sync() {
  if (fflush(this->file) != 0)
    return -1;
  return 0;
}

fbuf::int_type fbuf::underflow() {
  if (this->gptr() < this->egptr()) {
    char c = *this->gptr();
    this->setg(this->eback(), this->gptr() + 1, this->egptr());
    return traits_type::to_int_type(c);
  }
  size_t n = fread(this->buf, 1, sizeof(this->buf), this->file);
  if (n == 0)
    return traits_type::eof();
  this->setg(this->buf, this->buf, this->buf + n);
  return traits_type::to_int_type(*this->gptr());
}

cfstream::cfstream(FILE *f) : std::istream(&buf), buf(f) {}