TimeManager.h

Go to the documentation of this file.

#pragma once

 
#include "chess.hpp" // For chess::Color
#include <algorithm>
#include <chrono>
#include <cmath>
#include <cstdint>
 
struct SearchParams;

class TimeManager {
public:
  enum class TimeControl {
    BULLET,        
    BLITZ,         
    RAPID,         
    CLASSICAL,     
    CORRESPONDENCE 
  };

  struct TimeAllocation {
    int64_t softLimit;   
    int64_t hardLimit;   
    int64_t optimalTime; 
    int64_t panicTime;   
    TimeControl control; 
  };

  struct Config {
    // Phase multipliers (0.0 = opening, 1.0 = endgame)
    float openingMultiplier = 0.75f;    
    float middlegameMultiplier = 1.25f; 
    float endgameMultiplier = 0.70f;    
 
    // Time control multipliers
    float bulletMultiplier = 0.55f;     
    float blitzMultiplier = 0.75f;      
    float rapidMultiplier = 1.0f;       
    float classicalMultiplier = 1.15f;  
 
    // Eval-based multipliers
    float criticalMultiplier = 1.4f;    
    float edgeMultiplier = 1.0f;        
    float winningMultiplier = 0.65f;    
    float losingMultiplier = 0.85f;     
 
    // Instability extension
    float maxInstabilityExtension = 2.0f; 
    int instabilityThreshold = 3;         
 
    // Increment handling
    float incrementUsage = 0.7f;   
    float incrementReserve = 0.3f; 
    int minMovesForIncrement = 5;  
 
    // Safety margins
    int moveOverheadMs = 50;     
    int minTimeMs = 10;          
    int panicThresholdMs = 3000; 
    int ultraPanicMs = 1000;     
  };
 
  TimeManager() = default;
  explicit TimeManager(const Config &cfg) : config_(cfg) {}

  TimeAllocation calculate(const SearchParams &params, chess::Color side,
                           float phase, float evalCp, int moveNumber) const;

  bool shouldStop(int64_t elapsed, int depth, int bestMoveChanges,
                  int scoreDrops, const TimeAllocation &allocation) const;

  float getExtensionFactor(int bestMoveChanges, int scoreDrops) const;
 
  const Config &config() const { return config_; }
  void setConfig(const Config &cfg) { config_ = cfg; }
 
private:
  Config config_;
 
  TimeControl detectTimeControl(int baseTimeMs, int incrementMs) const;
  float getPhaseMultiplier(float phase) const;
  float getEvalMultiplier(float evalCp) const;
  float getControlMultiplier(TimeControl control) const;
  int estimateMovesRemaining(int moveNumber, float phase, int movestogo) const;
};