zlib_wrapper.cpp

/* Copyright (C) 2019 Meng Zhou and Andrew D. Smith
 *
 * Author: Meng Zhou and Andrew D. Smith
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This 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 General Public License for more details.
 */

#include "zlib_wrapper.hpp"

#include <algorithm>

using std::string;

char igzfstream::peek() {
  const char c = gzgetc(fileobj);
  if (!gzeof(fileobj))
    gzungetc(c, fileobj);
  return c;
}

igzfstream &getline(igzfstream &in, string &line) {
  if (gzgets(in.fileobj, &in.buf[0], in.chunk_size)) {
    auto eol = std::find(begin(in.buf), end(in.buf), '\n');
    line.clear();
    // in case line already has a reserve
    copy(begin(in.buf), eol, std::back_inserter(line));
  }
  return in;
}

igzfstream &operator>>(igzfstream &in, string &line) {
  return getline(in, line);
}

ogzfstream &operator<<(ogzfstream &out, const string &line) {
  gzputs(out.fileobj, line.c_str());
  return out;
}

ogzfstream &operator<<(ogzfstream &out, const char c) {
  gzputc(out.fileobj, c);
  return out;
}