aboutsummaryrefslogtreecommitdiff
path: root/src/raw
diff options
context:
space:
mode:
Diffstat (limited to 'src/raw')
-rw-r--r--src/raw/mod.rs2
-rw-r--r--src/raw/parser.rs42
2 files changed, 22 insertions, 22 deletions
diff --git a/src/raw/mod.rs b/src/raw/mod.rs
index 88b58ac..77a8571 100644
--- a/src/raw/mod.rs
+++ b/src/raw/mod.rs
@@ -21,7 +21,7 @@ use std::ffi::CStr;
use std::io::{BufReader, Read, Seek};
/// Parse a complete log that was compressed as a zip file.
-pub fn parse_zip<T: Read + Seek>(input: &mut T) -> ParseResult<Evtc> {
+pub fn parse_zip<R: Read + Seek>(input: R) -> ParseResult<Evtc> {
let mut archive = ZipArchive::new(input)?;
let mut file = BufReader::new(archive.by_index(0)?);
parse_file(&mut file)
diff --git a/src/raw/parser.rs b/src/raw/parser.rs
index 5544fd2..a9d8e8a 100644
--- a/src/raw/parser.rs
+++ b/src/raw/parser.rs
@@ -132,7 +132,7 @@ pub type ParseResult<T> = Result<T, ParseError>;
/// It is expected that the file cursor is at the very first byte of the file.
///
/// * `input` - Input stream.
-pub fn parse_header<T: Read>(input: &mut T) -> ParseResult<Header> {
+pub fn parse_header<R: Read>(mut input: R) -> ParseResult<Header> {
// Make sure the magic number matches
let mut magic_number = [0; 4];
input.read_exact(&mut magic_number)?;
@@ -178,10 +178,10 @@ pub fn parse_header<T: Read>(input: &mut T) -> ParseResult<Header> {
///
/// * `input` - Input stream.
/// * `count` - Number of agents (found in the header).
-pub fn parse_agents<T: Read>(input: &mut T, count: u32) -> ParseResult<Vec<Agent>> {
+pub fn parse_agents<R: Read>(mut input: R, count: u32) -> ParseResult<Vec<Agent>> {
let mut result = Vec::with_capacity(count as usize);
for _ in 0..count {
- result.push(parse_agent(input)?);
+ result.push(parse_agent(&mut input)?);
}
Ok(result)
}
@@ -189,7 +189,7 @@ pub fn parse_agents<T: Read>(input: &mut T, count: u32) -> ParseResult<Vec<Agent
/// Parse a single agent.
///
/// * `input` - Input stream.
-pub fn parse_agent<T: Read>(input: &mut T) -> ParseResult<Agent> {
+pub fn parse_agent<R: Read>(mut input: R) -> ParseResult<Agent> {
let addr = input.read_u64::<LittleEndian>()?;
let prof = input.read_u32::<LittleEndian>()?;
let is_elite = input.read_u32::<LittleEndian>()?;
@@ -226,10 +226,10 @@ pub fn parse_agent<T: Read>(input: &mut T) -> ParseResult<Agent> {
///
/// * `input` - Input stream.
/// * `count` - Number of skills to parse.
-pub fn parse_skills<T: Read>(input: &mut T, count: u32) -> ParseResult<Vec<Skill>> {
+pub fn parse_skills<R: Read>(mut input: R, count: u32) -> ParseResult<Vec<Skill>> {
let mut result = Vec::with_capacity(count as usize);
for _ in 0..count {
- result.push(parse_skill(input)?);
+ result.push(parse_skill(&mut input)?);
}
Ok(result)
}
@@ -237,7 +237,7 @@ pub fn parse_skills<T: Read>(input: &mut T, count: u32) -> ParseResult<Vec<Skill
/// Parse a single skill.
///
/// * `input` - Input stream.
-pub fn parse_skill<T: Read>(input: &mut T) -> ParseResult<Skill> {
+pub fn parse_skill<R: Read>(mut input: R) -> ParseResult<Skill> {
let id = input.read_i32::<LittleEndian>()?;
let mut name = [0; 64];
input.read_exact(&mut name)?;
@@ -248,13 +248,13 @@ pub fn parse_skill<T: Read>(input: &mut T) -> ParseResult<Skill> {
///
/// * `input` - Input stream.
/// * `parser` - The parse function to use.
-pub fn parse_events<T: Read>(
- input: &mut T,
- parser: fn(&mut T) -> ParseResult<CbtEvent>,
+pub fn parse_events<R: Read>(
+ mut input: R,
+ parser: fn(&mut R) -> ParseResult<CbtEvent>,
) -> ParseResult<Vec<CbtEvent>> {
let mut result = Vec::new();
loop {
- let event = parser(input);
+ let event = parser(&mut input);
match event {
Ok(x) => result.push(x),
Err(ParseError::Io(ref e)) if e.kind() == ErrorKind::UnexpectedEof => {
@@ -270,7 +270,7 @@ pub fn parse_events<T: Read>(
/// This works for old combat events, i.e. files with revision == 0.
///
/// * `input` - Input stream.
-pub fn parse_event_rev0<T: Read>(input: &mut T) -> ParseResult<CbtEvent> {
+pub fn parse_event_rev0<R: Read>(mut input: R) -> ParseResult<CbtEvent> {
let time = input.read_u64::<LittleEndian>()?;
let src_agent = input.read_u64::<LE>()?;
let dst_agent = input.read_u64::<LE>()?;
@@ -334,7 +334,7 @@ pub fn parse_event_rev0<T: Read>(input: &mut T) -> ParseResult<CbtEvent> {
/// This works for new combat events, i.e. files with revision == 1.
///
/// * `input` - Input stream.
-pub fn parse_event_rev1<T: Read>(input: &mut T) -> ParseResult<CbtEvent> {
+pub fn parse_event_rev1<R: Read>(mut input: R) -> ParseResult<CbtEvent> {
let time = input.read_u64::<LittleEndian>()?;
let src_agent = input.read_u64::<LE>()?;
let dst_agent = input.read_u64::<LE>()?;
@@ -394,9 +394,9 @@ pub fn parse_event_rev1<T: Read>(input: &mut T) -> ParseResult<CbtEvent> {
/// Parse a partial EVTC file.
///
/// * `input` - Input stream.
-pub fn parse_partial_file<T: Read>(input: &mut T) -> ParseResult<PartialEvtc> {
- let header = parse_header(input)?;
- let agents = parse_agents(input, header.agent_count)?;
+pub fn parse_partial_file<R: Read>(mut input: R) -> ParseResult<PartialEvtc> {
+ let header = parse_header(&mut input)?;
+ let agents = parse_agents(&mut input, header.agent_count)?;
let skill_count = input.read_u32::<LittleEndian>()?;
let skills = parse_skills(input, skill_count)?;
@@ -412,10 +412,10 @@ pub fn parse_partial_file<T: Read>(input: &mut T) -> ParseResult<PartialEvtc> {
///
/// * `partial` - The partial EVTC.
/// * `input` - The input stream.
-pub fn finish_parsing<T: Read>(partial: PartialEvtc, input: &mut T) -> ParseResult<Evtc> {
+pub fn finish_parsing<R: Read>(partial: PartialEvtc, input: R) -> ParseResult<Evtc> {
let events = match partial.header.revision {
- 0 => parse_events(input, parse_event_rev0)?,
- 1 => parse_events(input, parse_event_rev1)?,
+ 0 => parse_events(input, |r| parse_event_rev0(r))?,
+ 1 => parse_events(input, |r| parse_event_rev1(r))?,
x => return Err(ParseError::UnknownRevision(x)),
};
@@ -431,7 +431,7 @@ pub fn finish_parsing<T: Read>(partial: PartialEvtc, input: &mut T) -> ParseResu
/// Parse a complete EVTC file.
///
/// * `input` - Input stream.
-pub fn parse_file<T: Read>(input: &mut T) -> ParseResult<Evtc> {
- let partial = parse_partial_file(input)?;
+pub fn parse_file<R: Read>(mut input: R) -> ParseResult<Evtc> {
+ let partial = parse_partial_file(&mut input)?;
finish_parsing(partial, input)
}