blob: 5305898b3a69b685e038d27c900876db19ce935c [file] [log] [blame]
// (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com)
// (C) Copyright 2005-2007 Jonathan Turkanis
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
// See http://www.boost.org/libs/iostreams for documentation.
#ifndef NDNBOOST_IOSTREAMS_DETAIL_COUNTED_ARRAY_HPP_INCLUDED
#define NDNBOOST_IOSTREAMS_DETAIL_COUNTED_ARRAY_HPP_INCLUDED
#include <algorithm> // min.
#include <ndnboost/iostreams/categories.hpp>
#include <ndnboost/iostreams/detail/char_traits.hpp>
#include <ndnboost/iostreams/detail/ios.hpp> // streamsize.
namespace ndnboost { namespace iostreams { namespace detail {
template<typename Ch>
class counted_array_source {
public:
typedef Ch char_type;
typedef source_tag category;
counted_array_source(const Ch* buf, std::streamsize size)
: buf_(buf), ptr_(0), end_(size)
{ }
std::streamsize read(Ch* s, std::streamsize n)
{
std::streamsize result = (std::min)(n, end_ - ptr_);
NDNBOOST_IOSTREAMS_CHAR_TRAITS(char_type)::copy
(s, buf_ + ptr_, result);
ptr_ += result;
return result;
}
std::streamsize count() const { return ptr_; }
private:
const Ch* buf_;
std::streamsize ptr_, end_;
};
template<typename Ch>
struct counted_array_sink {
public:
typedef Ch char_type;
typedef sink_tag category;
counted_array_sink(Ch* buf, std::streamsize size)
: buf_(buf), ptr_(0), end_(size)
{ }
std::streamsize write(const Ch* s, std::streamsize n)
{
std::streamsize result = (std::min)(n, end_ - ptr_);
NDNBOOST_IOSTREAMS_CHAR_TRAITS(char_type)::copy
(buf_ + ptr_, s, result);
ptr_ += result;
return result;
}
std::streamsize count() const { return ptr_; }
private:
Ch* buf_;
std::streamsize ptr_, end_;
};
} } } // End namespaces iostreams, boost.
#endif // #ifndef NDNBOOST_IOSTREAMS_DETAIL_COUNTED_ARRAY_HPP_INCLUDED