//! > Test match optimizer simple scenario.

//! > test_runner_name
test_match_optimizer

//! > function
fn foo() -> Option<u16> {
   let v = get_option();
   match v {
        Option::Some(_) => Option::Some(1_u16),
        Option::None => Option::None,
   }
}

//! > function_name
foo

//! > module_code
extern fn get_option() -> Option<u16> nopanic;

//! > semantic_diagnostics

//! > lowering_diagnostics

//! > before
Parameters:
blk0 (root):
Statements:
End:
  Match(match test::get_option() {
    Option::Some(v0) => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
  (v1: core::option::Option::<core::integer::u16>) <- Option::Some(v0)
End:
  Goto(blk3, {v1 -> v4})

blk2:
Statements:
  (v2: ()) <- struct_construct()
  (v3: core::option::Option::<core::integer::u16>) <- Option::None(v2)
End:
  Goto(blk3, {v3 -> v4})

blk3:
Statements:
End:
  Match(match_enum(v4) {
    Option::Some(v5) => blk4,
    Option::None(v8) => blk5,
  })

blk4:
Statements:
  (v6: core::integer::u16) <- 1u
  (v7: core::option::Option::<core::integer::u16>) <- Option::Some(v6)
End:
  Goto(blk6, {v7 -> v11})

blk5:
Statements:
  (v9: ()) <- struct_construct()
  (v10: core::option::Option::<core::integer::u16>) <- Option::None(v9)
End:
  Goto(blk6, {v10 -> v11})

blk6:
Statements:
End:
  Return(v11)

//! > after
Parameters:
blk0 (root):
Statements:
End:
  Match(match test::get_option() {
    Option::Some(v0) => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
End:
  Goto(blk4, {})

blk2:
Statements:
  (v2: ()) <- struct_construct()
End:
  Goto(blk5, {})

blk3:
Statements:
End:
  Match(match_enum(v4) {
    Option::Some(v5) => blk7,
    Option::None(v8) => blk8,
  })

blk4:
Statements:
  (v6: core::integer::u16) <- 1u
  (v7: core::option::Option::<core::integer::u16>) <- Option::Some(v6)
End:
  Goto(blk6, {v7 -> v11})

blk5:
Statements:
  (v9: ()) <- struct_construct()
  (v10: core::option::Option::<core::integer::u16>) <- Option::None(v9)
End:
  Goto(blk6, {v10 -> v11})

blk6:
Statements:
End:
  Return(v11)

blk7:
Statements:
End:
  Goto(blk4, {})

blk8:
Statements:
End:
  Goto(blk5, {})

//! > ==========================================================================

//! > Test skipping of match optimization.

//! > test_runner_name
test_match_optimizer

//! > function
fn foo(a: felt252) -> Option<u16> {
   let v = get_option();

   // The following instruction should block the optimization.
   core::internal::revoke_ap_tracking();
   let v1 = match v {
        Option::Some(_) => Option::Some(1_u16),
        Option::None => Option::None,
   };
   match v1 {
        // v1 is used after the match.
        Option::Some(_) => v1,
        Option::None => Option::Some(2_u16),
   }
}

//! > function_name
foo

//! > module_code
extern fn get_option() -> Option<u16> nopanic;

//! > semantic_diagnostics

//! > lowering_diagnostics

//! > before
Parameters: v0: core::felt252
blk0 (root):
Statements:
End:
  Match(match test::get_option() {
    Option::Some(v1) => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
  (v2: core::option::Option::<core::integer::u16>) <- Option::Some(v1)
End:
  Goto(blk3, {v2 -> v5})

blk2:
Statements:
  (v3: ()) <- struct_construct()
  (v4: core::option::Option::<core::integer::u16>) <- Option::None(v3)
End:
  Goto(blk3, {v4 -> v5})

blk3:
Statements:
  () <- core::internal::revoke_ap_tracking()
End:
  Match(match_enum(v5) {
    Option::Some(v6) => blk4,
    Option::None(v9) => blk5,
  })

blk4:
Statements:
  (v7: core::integer::u16) <- 1u
  (v8: core::option::Option::<core::integer::u16>) <- Option::Some(v7)
End:
  Goto(blk6, {v8 -> v12})

blk5:
Statements:
  (v10: ()) <- struct_construct()
  (v11: core::option::Option::<core::integer::u16>) <- Option::None(v10)
End:
  Goto(blk6, {v11 -> v12})

blk6:
Statements:
End:
  Match(match_enum(v12) {
    Option::Some(v13) => blk7,
    Option::None(v14) => blk8,
  })

blk7:
Statements:
End:
  Goto(blk9, {v12 -> v17})

blk8:
Statements:
  (v15: core::integer::u16) <- 2u
  (v16: core::option::Option::<core::integer::u16>) <- Option::Some(v15)
End:
  Goto(blk9, {v16 -> v17})

blk9:
Statements:
End:
  Return(v17)

//! > after
Parameters: v0: core::felt252
blk0 (root):
Statements:
End:
  Match(match test::get_option() {
    Option::Some(v1) => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
  (v2: core::option::Option::<core::integer::u16>) <- Option::Some(v1)
End:
  Goto(blk3, {v2 -> v5})

blk2:
Statements:
  (v3: ()) <- struct_construct()
  (v4: core::option::Option::<core::integer::u16>) <- Option::None(v3)
End:
  Goto(blk3, {v4 -> v5})

blk3:
Statements:
  () <- core::internal::revoke_ap_tracking()
End:
  Match(match_enum(v5) {
    Option::Some(v6) => blk4,
    Option::None(v9) => blk5,
  })

blk4:
Statements:
  (v7: core::integer::u16) <- 1u
  (v8: core::option::Option::<core::integer::u16>) <- Option::Some(v7)
End:
  Goto(blk6, {v8 -> v12})

blk5:
Statements:
  (v10: ()) <- struct_construct()
  (v11: core::option::Option::<core::integer::u16>) <- Option::None(v10)
End:
  Goto(blk6, {v11 -> v12})

blk6:
Statements:
End:
  Match(match_enum(v12) {
    Option::Some(v13) => blk7,
    Option::None(v14) => blk8,
  })

blk7:
Statements:
End:
  Goto(blk9, {v12 -> v17})

blk8:
Statements:
  (v15: core::integer::u16) <- 2u
  (v16: core::option::Option::<core::integer::u16>) <- Option::Some(v15)
End:
  Goto(blk9, {v16 -> v17})

blk9:
Statements:
End:
  Return(v17)

//! > ==========================================================================

//! > Test two matches.

//! > test_runner_name
test_match_optimizer

//! > function
fn foo() -> felt252 {
   let opt = get_option();
   match opt {
      Option::Some(x) => Option::Some(x + 3),
      Option::None => Option::None
   }.unwrap()
}

//! > function_name
foo

//! > module_code
extern fn get_option() -> Option<felt252> nopanic;

//! > semantic_diagnostics

//! > lowering_diagnostics

//! > before
Parameters:
blk0 (root):
Statements:
End:
  Match(match test::get_option() {
    Option::Some(v0) => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
  (v1: core::option::Option::<core::felt252>) <- Option::Some(v0)
End:
  Goto(blk3, {v1 -> v4})

blk2:
Statements:
  (v2: ()) <- struct_construct()
  (v3: core::option::Option::<core::felt252>) <- Option::None(v2)
End:
  Goto(blk3, {v3 -> v4})

blk3:
Statements:
End:
  Match(match_enum(v4) {
    Option::Some(v5) => blk4,
    Option::None(v9) => blk6,
  })

blk4:
Statements:
End:
  Goto(blk5, {})

blk5:
Statements:
  (v6: core::felt252) <- 3u
  (v14: core::felt252) <- core::felt252_add(v5, v6)
  (v8: core::option::Option::<core::felt252>) <- Option::Some(v14)
End:
  Goto(blk7, {v8 -> v12})

blk6:
Statements:
  (v10: ()) <- struct_construct()
  (v11: core::option::Option::<core::felt252>) <- Option::None(v10)
End:
  Goto(blk7, {v11 -> v12})

blk7:
Statements:
End:
  Match(match_enum(v12) {
    Option::Some(v17) => blk8,
    Option::None(v18) => blk12,
  })

blk8:
Statements:
End:
  Goto(blk9, {})

blk9:
Statements:
End:
  Goto(blk10, {})

blk10:
Statements:
End:
  Goto(blk11, {})

blk11:
Statements:
  (v29: (core::felt252,)) <- struct_construct(v17)
  (v30: core::panics::PanicResult::<(core::felt252,)>) <- PanicResult::Ok(v29)
End:
  Return(v30)

blk12:
Statements:
  (v26: core::array::Array::<core::felt252>) <- core::array::array_new::<core::felt252>()
End:
  Goto(blk13, {})

blk13:
Statements:
  (v15: core::felt252) <- 29721761890975875353235833581453094220424382983267374u
  (v27: core::array::Array::<core::felt252>) <- core::array::array_append::<core::felt252>(v26, v15)
End:
  Goto(blk14, {})

blk14:
Statements:
  (v24: core::panics::Panic) <- struct_construct()
  (v25: (core::panics::Panic, core::array::Array::<core::felt252>)) <- struct_construct(v24, v27)
  (v31: core::panics::PanicResult::<(core::felt252,)>) <- PanicResult::Err(v25)
End:
  Return(v31)

//! > after
Parameters:
blk0 (root):
Statements:
End:
  Match(match test::get_option() {
    Option::Some(v0) => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
End:
  Goto(blk4, {v0 -> v5})

blk2:
Statements:
  (v2: ()) <- struct_construct()
End:
  Goto(blk6, {})

blk3:
Statements:
End:
  Match(match_enum(v4) {
    Option::Some(v5) => blk15,
    Option::None(v9) => blk16,
  })

blk4:
Statements:
End:
  Goto(blk5, {})

blk5:
Statements:
  (v6: core::felt252) <- 3u
  (v14: core::felt252) <- core::felt252_add(v5, v6)
End:
  Goto(blk8, {v14 -> v17})

blk6:
Statements:
  (v10: ()) <- struct_construct()
End:
  Goto(blk12, {})

blk7:
Statements:
End:
  Match(match_enum(v12) {
    Option::Some(v17) => blk17,
    Option::None(v18) => blk18,
  })

blk8:
Statements:
End:
  Goto(blk9, {})

blk9:
Statements:
End:
  Goto(blk10, {})

blk10:
Statements:
End:
  Goto(blk11, {})

blk11:
Statements:
  (v29: (core::felt252,)) <- struct_construct(v17)
  (v30: core::panics::PanicResult::<(core::felt252,)>) <- PanicResult::Ok(v29)
End:
  Return(v30)

blk12:
Statements:
  (v26: core::array::Array::<core::felt252>) <- core::array::array_new::<core::felt252>()
End:
  Goto(blk13, {})

blk13:
Statements:
  (v15: core::felt252) <- 29721761890975875353235833581453094220424382983267374u
  (v27: core::array::Array::<core::felt252>) <- core::array::array_append::<core::felt252>(v26, v15)
End:
  Goto(blk14, {})

blk14:
Statements:
  (v24: core::panics::Panic) <- struct_construct()
  (v25: (core::panics::Panic, core::array::Array::<core::felt252>)) <- struct_construct(v24, v27)
  (v31: core::panics::PanicResult::<(core::felt252,)>) <- PanicResult::Err(v25)
End:
  Return(v31)

blk15:
Statements:
End:
  Goto(blk4, {})

blk16:
Statements:
End:
  Goto(blk6, {})

blk17:
Statements:
End:
  Goto(blk8, {})

blk18:
Statements:
End:
  Goto(blk12, {})

//! > ==========================================================================

//! > withdraw_gas

//! > test_runner_name
test_match_optimizer

//! > function
fn foo() {
    gas::withdraw_gas().expect('Out of gas');
}

//! > function_name
foo

//! > module_code

//! > semantic_diagnostics

//! > lowering_diagnostics

//! > before
Parameters:
blk0 (root):
Statements:
End:
  Match(match core::gas::withdraw_gas() {
    Option::Some => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
  (v0: ()) <- struct_construct()
  (v1: core::option::Option::<()>) <- Option::Some(v0)
End:
  Goto(blk3, {v1 -> v4})

blk2:
Statements:
  (v2: ()) <- struct_construct()
  (v3: core::option::Option::<()>) <- Option::None(v2)
End:
  Goto(blk3, {v3 -> v4})

blk3:
Statements:
End:
  Match(match_enum(v4) {
    Option::Some(v8) => blk4,
    Option::None(v9) => blk7,
  })

blk4:
Statements:
End:
  Goto(blk5, {})

blk5:
Statements:
End:
  Goto(blk6, {})

blk6:
Statements:
  (v7: ()) <- struct_construct()
  (v20: ((),)) <- struct_construct(v7)
  (v21: core::panics::PanicResult::<((),)>) <- PanicResult::Ok(v20)
End:
  Return(v21)

blk7:
Statements:
  (v17: core::array::Array::<core::felt252>) <- core::array::array_new::<core::felt252>()
End:
  Goto(blk8, {})

blk8:
Statements:
  (v5: core::felt252) <- 375233589013918064796019u
  (v18: core::array::Array::<core::felt252>) <- core::array::array_append::<core::felt252>(v17, v5)
End:
  Goto(blk9, {})

blk9:
Statements:
  (v15: core::panics::Panic) <- struct_construct()
  (v16: (core::panics::Panic, core::array::Array::<core::felt252>)) <- struct_construct(v15, v18)
  (v22: core::panics::PanicResult::<((),)>) <- PanicResult::Err(v16)
End:
  Return(v22)

//! > after
Parameters:
blk0 (root):
Statements:
End:
  Match(match core::gas::withdraw_gas() {
    Option::Some => blk1,
    Option::None => blk2,
  })

blk1:
Statements:
  (v0: ()) <- struct_construct()
End:
  Goto(blk4, {})

blk2:
Statements:
  (v2: ()) <- struct_construct()
End:
  Goto(blk7, {})

blk3:
Statements:
End:
  Match(match_enum(v4) {
    Option::Some(v8) => blk10,
    Option::None(v9) => blk11,
  })

blk4:
Statements:
End:
  Goto(blk5, {})

blk5:
Statements:
End:
  Goto(blk6, {})

blk6:
Statements:
  (v7: ()) <- struct_construct()
  (v20: ((),)) <- struct_construct(v7)
  (v21: core::panics::PanicResult::<((),)>) <- PanicResult::Ok(v20)
End:
  Return(v21)

blk7:
Statements:
  (v17: core::array::Array::<core::felt252>) <- core::array::array_new::<core::felt252>()
End:
  Goto(blk8, {})

blk8:
Statements:
  (v5: core::felt252) <- 375233589013918064796019u
  (v18: core::array::Array::<core::felt252>) <- core::array::array_append::<core::felt252>(v17, v5)
End:
  Goto(blk9, {})

blk9:
Statements:
  (v15: core::panics::Panic) <- struct_construct()
  (v16: (core::panics::Panic, core::array::Array::<core::felt252>)) <- struct_construct(v15, v18)
  (v22: core::panics::PanicResult::<((),)>) <- PanicResult::Err(v16)
End:
  Return(v22)

blk10:
Statements:
End:
  Goto(blk4, {})

blk11:
Statements:
End:
  Goto(blk7, {})
