Update bitaxe_hashrate_benchmark.py

Couple of changes:
- color not working in windows terminal
- suffix in filename in case of voltage/frequency argument
- limited (1 successive) retries in case of overheat (previously any chip overheat reached ended the benchmark)

Author: Co1nB3e

bitaxe_hashrate_benchmark.py

@@ -5,6 +5,9 @@
 import sys
 import argparse
 
+import os
+os.system("")  # enables ansi escape characters in terminal - colors were not working in Windows terminal
+
 # ANSI Color Codes
 GREEN = "\033[92m"
 YELLOW = "\033[93m"
@@ -72,6 +75,13 @@ def parse_arguments():
 if benchmark_time / sample_interval < 7:
     raise ValueError(RED + f"Error: Benchmark time is too short. Please increase the benchmark time or decrease the sample interval. At least 7 samples are required." + RESET)
 
+# Add suffix to filename in case of manual initial voltage/frequency
+file_suffix = ""
+if initial_voltage != 1150:
+    file_suffix = file_suffix + " v" + str(initial_voltage)
+if initial_frequency != 500:
+    file_suffix = file_suffix + " f" + str(initial_frequency)
+
 # Results storage
 results = []
 
@@ -307,7 +317,7 @@ def save_results():
     try:
         # Extract IP from bitaxe_ip global variable and remove 'http://'
         ip_address = bitaxe_ip.replace('http://', '')
-        filename = f"bitaxe_benchmark_results_{ip_address}.json"
+        filename = f"bitaxe_benchmark_results_{ip_address}{file_suffix}.json"
         with open(filename, "w") as f:
             json.dump(results, f, indent=4)
         print(GREEN + f"Results saved to {filename}" + RESET)
@@ -346,12 +356,14 @@ def reset_to_best_setting():
 
     current_voltage = initial_voltage
     current_frequency = initial_frequency
+    retry_upon_overheat = 0
 
     while current_voltage <= max_allowed_voltage and current_frequency <= max_allowed_frequency:
         set_system_settings(current_voltage, current_frequency)
         avg_hashrate, avg_temp, efficiency_jth, hashrate_ok, avg_vr_temp, error_reason = benchmark_iteration(current_voltage, current_frequency)
 
         if avg_hashrate is not None and avg_temp is not None and efficiency_jth is not None:
+            retry_upon_overheat = 0
             result = {
                 "coreVoltage": current_voltage,
                 "frequency": current_frequency,
@@ -370,20 +382,37 @@ def reset_to_best_setting():
                 # If hashrate is good, try increasing frequency
                 if current_frequency + frequency_increment <= max_allowed_frequency:
                     current_frequency += frequency_increment
+                    print(GREEN + "Hashrate is good. Increasing frequency for next try." + RESET)
                 else:
+                    print(GREEN + "Reached max frequency with good results. Stopping further testing." + RESET)
                     break  # We've reached max frequency with good results
             else:
                 # If hashrate is not good, go back one frequency step and increase voltage
                 if current_voltage + voltage_increment <= max_allowed_voltage:
                     current_voltage += voltage_increment
                     current_frequency -= frequency_increment  # Go back to one frequency step and retry
-                    print(YELLOW + f"Hashrate to low compared to expected. Decreasing frequency to {current_frequency}MHz and increasing voltage to {current_voltage}mV" + RESET)
+                    print(YELLOW + f"Hashrate too low compared to expected. Decreasing frequency to {current_frequency}MHz and increasing voltage to {current_voltage}mV" + RESET)
                 else:
+                    print(YELLOW + "Reached max voltage without good results. Stopping further testing." + RESET)
                     break  # We've reached max voltage without good results
         else:
             # If we hit thermal limits or other issues, we've found the highest safe settings
-            print(GREEN + "Reached thermal or stability limits. Stopping further testing." + RESET)
-            break  # Stop testing higher values
+            # In case of max Chip Temperature reached, continue loop to next voltage with decreased frequency
+            # Condition added to avoid successive overheat tries and reset to high initial frequency
+            if avg_hashrate is None and avg_temp is None and efficiency_jth is None and hashrate_ok is False and avg_vr_temp is None and error_reason == 'CHIP_TEMP_EXCEEDED' and initial_frequency <= current_frequency + frequency_increment and retry_upon_overheat < 1:
+                # If overheat, return to initial frequency while increasing voltage (considering max_allowed_voltage)
+                #and current_frequency + frequency_increment <= max_allowed_frequency and current_voltage + voltage_increment <= max_allowed_voltage
+                retry_upon_overheat += 1
+                if current_voltage + voltage_increment <= max_allowed_voltage:
+                    current_frequency = initial_frequency
+                    current_voltage += voltage_increment
+                    print(GREEN + "Reached thermal limit for the current voltage/frequency. Switching to next voltage increment." + RESET)
+                else:
+                    print(GREEN + "Reached thermal limit for the current voltage/frequency. Next voltage increment out of voltage limit. Stopping further testing." + RESET)
+                    break  # We've reached max voltage, can't increase voltage anymore
+            else:
+                print(GREEN + "Reached thermal or stability limits. Stopping further testing." + RESET)
+                break  # Stop testing higher values
 
         save_results()
 
@@ -445,7 +474,7 @@ def reset_to_best_setting():
 
         # Save the final data to JSON
         ip_address = bitaxe_ip.replace('http://', '')
-        filename = f"bitaxe_benchmark_results_{ip_address}.json"
+        filename = f"bitaxe_benchmark_results_{ip_address}{file_suffix}.json"
         with open(filename, "w") as f:
             json.dump(final_data, f, indent=4)
 
@@ -495,3 +524,4 @@ def cleanup_and_exit(reason=None):
             print(RED + f"Benchmarking stopped: {reason}" + RESET)
         print(GREEN + "Benchmarking completed." + RESET)
         sys.exit(0)
+