Add a enable_early_exit flag under fi (#885)

Its value is set as true by default.

Author: ycho

src/encoder.rs

@@ -569,6 +569,7 @@ pub struct FrameInvariants {
     pub me_range_scale: u8,
     pub use_tx_domain_distortion: bool,
     pub inter_cfg: Option<InterPropsConfig>,
+    pub enable_early_exit: bool,
 }
 
 impl FrameInvariants {
@@ -642,6 +643,7 @@ impl FrameInvariants {
             me_range_scale: 1,
             use_tx_domain_distortion: use_tx_domain_distortion,
             inter_cfg: None,
+            enable_early_exit: true,
         }
     }
 
@@ -2422,7 +2424,7 @@ fn encode_partition_bottomup(
 
                 if cost != std::f64::MAX {
                     rd_cost += cost;
-                    if rd_cost >= best_rd || rd_cost >= ref_rd_cost {
+                    if fi.enable_early_exit && (rd_cost >= best_rd || rd_cost >= ref_rd_cost) {
                         assert!(cost != std::f64::MAX);
                         early_exit = true;
                         if partition == PartitionType::PARTITION_SPLIT { break; }

src/rdo.rs

@@ -970,7 +970,6 @@ pub fn rdo_partition_decision(
   let mut best_partition = cached_block.part_type;
   let mut best_rd = cached_block.rd_cost;
   let mut best_pred_modes = cached_block.part_modes.clone();
-  let mut early_exit;
 
   let cw_checkpoint = cw.checkpoint();
   let w_pre_checkpoint = w_pre_cdef.checkpoint();
@@ -981,9 +980,9 @@ pub fn rdo_partition_decision(
     if partition == cached_block.part_type {
       continue;
     }
-    early_exit = false;
     let mut cost: f64 = 0.0;
     let mut child_modes = std::vec::Vec::new();
+    let mut early_exit = false;
 
     match partition {
       PartitionType::PARTITION_NONE => {
@@ -1050,7 +1049,7 @@ pub fn rdo_partition_decision(
 
           rd_cost_sum += mode_decision.rd_cost;
 
-          if rd_cost_sum > best_rd {
+          if fi.enable_early_exit && rd_cost_sum > best_rd {
             early_exit = true;
             break;
           }