blob: affd2da14e8391547f7113264e09f1f347356c71 (
plain) (
blame)
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
|
use std::{fs::File, io::{BufReader, Read}};
pub struct FileReader<const BUFF_SIZE: usize = 1024>
{
buf_reader: BufReader<File>
}
impl<const BUFF_SIZE: usize> FileReader<BUFF_SIZE> {
pub fn new(buf_reader: BufReader<File>) -> Self
{
Self
{
buf_reader
}
}
pub fn skip(&mut self, c: usize)
{
self.buf_reader.seek_relative(c as i64).expect("Cannot seek!");
}
pub fn read_array<const N: usize, T: FromBytes<N> + Sized, const C: usize>(&mut self) -> [T; C]
{
std::array::from_fn(|_| self.read())
}
pub fn read<const N: usize, T: FromBytes<N> + Sized>(&mut self) -> T
{
let mut data = [0u8; N];
self.buf_reader.read_exact(&mut data).expect("Bad read");
T::from_le_bytes(&data)
}
}
pub trait FromBytes<const N: usize>: Sized
{
fn from_le_bytes(buffer: &[u8; N]) -> Self;
}
impl FromBytes<4> for u32
{
fn from_le_bytes(buffer: &[u8; 4]) -> Self {
u32::from_le_bytes(*buffer)
}
}
impl FromBytes<2> for u16
{
fn from_le_bytes(buffer: &[u8; 2]) -> Self {
u16::from_le_bytes(*buffer)
}
}
impl FromBytes<1> for u8
{
fn from_le_bytes(buffer: &[u8; 1]) -> Self {
u8::from_le_bytes(*buffer)
}
}
|
