applied fix

youknowone · arihant2math · commit 4ef2a506f981 · 2025-01-21T23:36:25.000-08:00
Signed-off-by: Ashwin Naren &lt;arihant2math@gmail.com&gt;
diff --git a/compiler/codegen/src/compile.rs b/compiler/codegen/src/compile.rs
@@ -1840,7 +1840,7 @@ impl Compiler {
     fn jump_to_fail_pop(&mut self, pattern_context: &mut PatternContext) -> CompileResult<()> {
         let pops = pattern_context.on_top + pattern_context.stores.len();
         self.ensure_fail_pop(pattern_context, pops)?;
-        self.switch_to_block(pattern_context.fail_pop[pops]);
+        // self.switch_to_block(pattern_context.fail_pop[pops]);
         Ok(())
     }
     // static int
@@ -2054,7 +2054,11 @@ impl Compiler {
     ) -> CompileResult<()> {
         self.compile_expression(subject)?;
         // Block at the end of the switch statement that we jump to after finishing a branch
-        let end_block = self.new_block();
+        let pattern_blocks = std::iter::repeat_with(|| self.new_block())
+            .take(cases.len() + 1)
+            .collect::<Vec<_>>();
+        eprintln!("created pattern_blocks: {:?} - {:?}(end block)", pattern_blocks.first().unwrap(), pattern_blocks.last().unwrap());
+        let end_block = *pattern_blocks.last().unwrap();
 
         let match_case_type = cases.last().expect("cases is not empty");
         let has_default = match_case_type.pattern.is_match_star() && 1 < cases.len();
@@ -2088,7 +2092,7 @@ impl Compiler {
             // }
             if let Some(guard) = &m.guard {
                 self.ensure_fail_pop(pattern_context, 0)?;
-                self.compile_jump_if(guard, false, pattern_context.fail_pop[0])?;
+                self.compile_jump_if(guard, false, pattern_blocks[i + 1])?;
             }
             if i != cases.len() - (has_default as usize) - 1 {
                 // ADDOP(c, LOC(m->pattern), POP_TOP);
@@ -2103,6 +2107,7 @@ impl Compiler {
             // A trailing "case _" is common, and lets us save a bit of redundant
             // pushing and popping in the loop above:
             let m = &cases.last().unwrap();
+            self.switch_to_block(*pattern_blocks.last().unwrap());
             if cases.len() == 1 {
                 // No matches. Done with the subject:
                 // ADDOP(c, LOC(m->pattern), POP_TOP);
@@ -3419,6 +3424,7 @@ impl Compiler {
             u32::MAX,
             "switching {prev:?} -> {block:?} from block that's already got a next"
         );
+        eprintln!("switch_to_block {prev:?} -> {block:?}");
         prev_block.next = block;
         code.current_block = block;
     }
