timeshifter/tacview-splitter
1
0
Fork 0
mirror of https://github.com/132nd-vWing/tacview-splitter.git synced 2026-08-26 06:48:36 +02:00
Code Issues Projects Releases Wiki Activity

turned line type constants into LineType enum

Browse Source
This commit is contained in:
132nd-Professor
2021-07-17 19:14:12 +02:00
parent 67c5e61d21
commit 467274816f
1 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main.rs
+16 -13
View File
@@ -10,10 +10,13 @@ const MINUS: char = '-';
const EXTENSION_ZIP: &str = ".zip.acmi"; const EXTENSION_ZIP: &str = ".zip.acmi";
const EXTENSION_TXT: &str = ".txt.acmi"; const EXTENSION_TXT: &str = ".txt.acmi";
const UNKNOWN: u8 = 0; #[derive(PartialEq)]
const TIMESTAMP: u8 = 1; enum LineType {
const DESTRUCTION: u8 = 2; Unknown,
const TELEMETRY: u8 = 3; Timestamp,
Destruction,
Telemetry
}
fn main() { fn main() {
let (input_filename, is_zip) = find_input_file(); let (input_filename, is_zip) = find_input_file();
@@ -46,14 +49,14 @@ fn split_into_header_and_body(lines: Vec<String>) -> (Vec<String>, Vec<String>)
fn divide_body_by_coalition(body: &Vec<String>) -> lib::BodiesByCoalition { fn divide_body_by_coalition(body: &Vec<String>) -> lib::BodiesByCoalition {
let mut bbc = lib::BodiesByCoalition{blue: Vec::new(), red: Vec::new(), violet: Vec::new()}; let mut bbc = lib::BodiesByCoalition{blue: Vec::new(), red: Vec::new(), violet: Vec::new()};
let mut continued = false; let mut continued = false;
let mut line_type: u8 = UNKNOWN; let mut line_type = LineType::Unknown;
let mut ids = lib::IDs{blue: Vec::new(), red: Vec::new(), violet: Vec::new(), unknown: Vec::new()}; let mut ids = lib::IDs{blue: Vec::new(), red: Vec::new(), violet: Vec::new(), unknown: Vec::new()};
for line in body { for line in body {
let result = process_line(continued, &mut ids, line, line_type); let result = process_line(continued, &mut ids, line, line_type);
line_type = result.0; line_type = result.0;
continued = result.1; continued = result.1;
let id = result.2; let id = result.2;
if line_type == TIMESTAMP { if line_type == LineType::Timestamp {
bbc.blue.push(line); bbc.blue.push(line);
bbc.red.push(line); bbc.red.push(line);
bbc.violet.push(line); bbc.violet.push(line);
@@ -70,14 +73,14 @@ fn divide_body_by_coalition(body: &Vec<String>) -> lib::BodiesByCoalition {
bbc bbc
} }
fn process_line<'a>(continued: bool, ids: &mut lib::IDs<'a>, line: &'a String, last_line_type: u8) -> (u8, bool, &'a str) { fn process_line<'a>(continued: bool, ids: &mut lib::IDs<'a>, line: &'a String, last_line_type: LineType) -> (LineType, bool, &'a str) {
let mut id = "both"; let mut id = "both";
let line_type; let line_type;
let will_continue: bool; let will_continue: bool;
if !continued { if !continued {
line_type = find_line_type(line); line_type = find_line_type(line);
if line_type == TELEMETRY { if line_type == LineType::Telemetry {
id = line id = line
.split_once(',') // TODO catch the None .split_once(',') // TODO catch the None
.unwrap() .unwrap()
@@ -106,16 +109,16 @@ fn process_line<'a>(continued: bool, ids: &mut lib::IDs<'a>, line: &'a String, l
(line_type, will_continue, id) (line_type, will_continue, id)
} }
fn find_line_type(line: &String) -> u8 { fn find_line_type(line: &String) -> LineType {
let line_type: u8; let line_type: LineType;
let first_char = line.chars().nth(0).expect("malformed line"); let first_char = line.chars().nth(0).expect("malformed line");
if first_char == COMMENT { if first_char == COMMENT {
line_type = TIMESTAMP; line_type = LineType::Timestamp;
} else if first_char == MINUS { } else if first_char == MINUS {
line_type = DESTRUCTION; line_type = LineType::Destruction;
} else { } else {
line_type = TELEMETRY; line_type = LineType::Telemetry;
} }
return line_type return line_type
} }