Buffer.h

Go to the documentation of this file.

/*
 * Copyright (c) 2016-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under both the BSD-style license (found in the
 * LICENSE file in the root directory of this source tree) and the GPLv2 (found
 * in the COPYING file in the root directory of this source tree).
 */
#pragma once
 
#include "utils/Range.h"
 
#include <array>
#include <cstddef>
#include <memory>
 
namespace pzstd {
 
class Buffer {
  std::shared_ptr<unsigned char> buffer_;
  MutableByteRange range_;
 
  static void delete_buffer(unsigned char* buffer) {
    delete[] buffer;
  }
 
 public:
  explicit Buffer() {}
 
  explicit Buffer(std::size_t size)
      : buffer_(new unsigned char[size], delete_buffer),
        range_(buffer_.get(), buffer_.get() + size) {}
 
  explicit Buffer(std::shared_ptr<unsigned char> buffer, MutableByteRange data)
      : buffer_(buffer), range_(data) {}
 
  Buffer(Buffer&&) = default;
  Buffer& operator=(Buffer&&) = default;
 
  Buffer splitAt(std::size_t n) {
    auto firstPiece = range_.subpiece(0, n);
    range_.advance(n);
    return Buffer(buffer_, firstPiece);
  }
 
  void advance(std::size_t n) {
    range_.advance(n);
  }
 
  void subtract(std::size_t n) {
    range_.subtract(n);
  }
 
  ByteRange range() const {
    return range_;
  }
  MutableByteRange range() {
    return range_;
  }
 
  const unsigned char* data() const {
    return range_.data();
  }
 
  unsigned char* data() {
    return range_.data();
  }
 
  std::size_t size() const {
    return range_.size();
  }
 
  bool empty() const {
    return range_.empty();
  }
};
}