add early exit

Author: LaihoE

src/parser/src/parser.rs

@@ -56,6 +56,10 @@ impl Parser {
             let size = self.read_varint()?;
             self.tick = tick as i32;
 
+            if self.should_early_exit() {
+                break;
+            }
+
             // Safety check
             if self.ptr + size as usize >= self.bytes.get_len() {
                 break;
@@ -64,7 +68,7 @@ impl Parser {
             let is_compressed = (cmd & 64) == 64;
             let demo_cmd = demo_cmd_type_from_int(msg_type as i32).unwrap();
 
-            // Early exit these for performance reasons
+            // skip these for performance reasons
             if demo_cmd == DEM_Packet || demo_cmd == DEM_AnimationData {
                 self.ptr += size as usize;
                 continue;
@@ -126,6 +130,21 @@ impl Parser {
         }
         Ok(self.combine_thread_outputs(&mut ok))
     }
+    fn should_early_exit(&self) -> bool {
+        if self.only_header && !self.header.is_empty() {
+            return true;
+        }
+        if !self.wanted_ticks.is_empty() {
+            // odd ticks in beginning of demo
+            if self.tick > 1000000 {
+                return false;
+            }
+            if self.largest_wanted_tick < self.tick {
+                return true;
+            }
+        }
+        false
+    }
     // fn parse_stringtables_cmd(bytes: &[u8]) -> Result<(), DemoParserError> {}
     pub fn create_parser_thread_input(&self, offset: usize, parse_all: bool) -> ParserThreadInput {
         let cls_by_id = match &self.cls_by_id {

src/parser/src/parser_settings.rs

@@ -81,6 +81,7 @@ pub struct Parser {
     pub prop_out_id: u8,
     pub only_header: bool,
     pub prop_infos: Vec<PropInfo>,
+    pub largest_wanted_tick: i32,
 
     pub header: AHashMap<String, String>,
     pub threads_spawned: u32,
@@ -92,6 +93,7 @@ impl Parser {
         let arc_huf = inputs.huffman_lookup_table.clone();
         Parser {
             threads_spawned: 0,
+            largest_wanted_tick: *inputs.wanted_ticks.iter().max().unwrap_or(&999999999),
             stringtable_players: AHashMap::default(),
             only_header: inputs.only_header,
             ge_list_set: false,