blockifier: change input to felt_counts fn count_blake_opcodes

Author: avivg-starkware

crates/blockifier/src/execution/execution_utils.rs

@@ -448,9 +448,9 @@ fn estimate_steps_of_encode_felt252_data_and_calc_blake_hash(
 
 /// Returns the number of BLAKE opcodes needed to hash the given felts.
 /// Each BLAKE opcode processes 16 u32s (partial messages are padded).
-fn count_blake_opcode(n_big_felts: usize, n_small_felts: usize) -> usize {
+fn count_blake_opcode(felt_size_groups: &FeltSizeCount) -> usize {
     // Count the total number of u32s to be hashed.
-    let total_u32s = total_u32s_from_felts(n_big_felts, n_small_felts);
+    let total_u32s = total_u32s_from_felts(felt_size_groups.large, felt_size_groups.small);
     total_u32s.div_ceil(FeltSizeCount::U32_WORDS_PER_MESSAGE)
 }
 
@@ -482,7 +482,7 @@ pub fn encode_and_blake_hash_resources(
 
     EstimatedExecutionResources::V2Hash {
         resources,
-        blake_count: count_blake_opcode(felt_size_groups.large, felt_size_groups.small),
+        blake_count: count_blake_opcode(felt_size_groups),
     }
 }
 

crates/blockifier/src/execution/execution_utils_test.rs

@@ -30,14 +30,12 @@ fn test_u32_constants() {
 #[test]
 fn test_zero_inputs() {
     // logic was written.
-    let steps = estimate_steps_of_encode_felt252_data_and_calc_blake_hash(&FeltSizeCount {
-        large: 0,
-        small: 0,
-    });
+    let steps =
+        estimate_steps_of_encode_felt252_data_and_calc_blake_hash(&FeltSizeCount::default());
     assert_eq!(steps, STEPS_EMPTY_INPUT, "Unexpected base step cost for zero inputs");
 
     // No opcodes should be emitted.
-    let opcodes = count_blake_opcode(0, 0);
+    let opcodes = count_blake_opcode(&FeltSizeCount::default());
     assert_eq!(opcodes, 0, "Expected zero BLAKE opcodes for zero inputs");
 
     // Should result in base cost only (no opcode cost).