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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#![no_std]
extern crate containers;
extern crate io;
extern crate loca;
use containers::collections::{RawVec, Vec};
use core::cmp;
use loca::Alloc;
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct Read<T, R, A: Alloc> {
r: R,
k: usize,
buf: Vec<T, A>,
}
impl<T, R, A: Alloc + Default> Read<T, R, A> {
#[inline]
pub fn with_capacity(r: R, cap: usize) -> Option<Self> {
Vec::with_capacity(cap).map(|buf| Read { r, buf, k: 0 })
}
}
impl<T, R, A: Alloc> Read<T, R, A> {
#[inline]
pub fn with_capacity_in(a: A, r: R, cap: usize) -> Option<Self> {
Vec::with_capacity_in(a, cap).map(|buf| Read { r, buf, k: 0 })
}
#[inline]
pub fn from_raw(r: R, raw: RawVec<T, A>) -> Self {
Read { r, buf: Vec::from_raw(raw), k: 0 }
}
#[inline]
pub fn as_ref(&self) -> &R { &self.r }
#[inline]
pub fn as_mut(&mut self) -> &mut R { &mut self.r }
}
impl<T: Copy, R: ::io::Read<T>, A: Alloc> ::io::Read<T> for Read<T, R, A> {
type Err = R::Err;
#[inline]
fn read(&mut self, buf: &mut [T]) -> Result<usize, Self::Err> {
if self.k == self.buf.len() {
self.buf.truncate(0);
self.k = 0;
{
let (m, n_opt) = self.r.size_hint();
self.buf.reserve(n_opt.unwrap_or(m));
}
if buf.len() >= self.buf.capacity() { return self.r.read(buf); }
match self.r.read_onto_vec(&mut self.buf) {
Ok(n) if n > 0 => (),
x => return x,
}
}
let l = cmp::min(self.buf.len() - self.k, buf.len());
buf[0..l].copy_from_slice(&self.buf[0..l]);
self.k += l;
Ok(l)
}
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
let l = self.buf.len();
(l, self.r.size_hint().1.and_then(|n| usize::checked_add(l, n)))
}
}
impl<T: Copy, R: ::io::Write<T>, A: Alloc> ::io::Write<T> for Read<T, R, A> {
type Err = R::Err;
#[inline]
fn write(&mut self, buf: &[T]) -> Result<usize, Self::Err> { self.r.write(buf) }
#[inline]
fn writev(&mut self, buf: &[&[T]]) -> Result<usize, Self::Err> { self.r.writev(buf) }
#[inline]
fn flush(&mut self) -> Result<(), Self::Err> { self.r.flush() }
}