from __future__ import annotations
import logging
from .core import stk_available, IAgStkObjectRoot, IAgScenario
from .utils import timed_operation, safe_stk_command
logger = logging.getLogger(__name__)
@timed_operation
def setup_scenario_internal(
stk_root: IAgStkObjectRoot,
scenario_name: str,
start_time: str,
duration_hours: float
) -> tuple[bool, str, IAgScenario | None]:
"""
Internal logic to create/configure an STK Scenario.
Returns:
tuple: (success_flag, status_message, scenario_object_or_None)
"""
if not stk_available or not stk_root:
return False, "STK Root object not available.", None
try:
# Close existing scenario if open
if stk_root.Children.Count > 0:
current_scen_name = stk_root.CurrentScenario.InstanceName
logger.info(" Closing existing scenario: %s", current_scen_name)
stk_root.CloseScenario()
# Create new scenario
logger.info(" Creating new scenario: %s", scenario_name)
stk_root.NewScenario(scenario_name)
scenario = stk_root.CurrentScenario
if scenario is None:
raise Exception("Failed to create or get the new scenario object.")
# Set time period
duration_str = f"+{duration_hours} hours"
logger.info(" Setting scenario time: Start='%s', Duration='%s'", start_time, duration_str)
scenario.SetTimePeriod(start_time, duration_str)
# Reset animation time
stk_root.Rewind()
# Optional: Maximize windows
try:
logger.info(" Maximizing STK windows...")
safe_stk_command(stk_root, 'Application / Raise')
safe_stk_command(stk_root, 'Application / Maximize')
# Consider checking for 3D window existence if needed
# stk_root.ExecuteCommand('Window3D * Maximize')
except Exception as cmd_e:
logger.warning(" Could not execute maximize commands: %s", cmd_e)
return True, f"Successfully created and configured scenario: '{scenario_name}'", scenario
except Exception as e:
error_msg = f"Error setting up scenario '{scenario_name}': {e}"
logger.error(" %s", error_msg)
# import traceback
# traceback.print_exc()
return False, error_msg, None
