"""
Change timestep options
=======================

"""

# %%
# This example shows how to interact with analysis timesteps.
# We first import PyGeoStudio and open the study
import numpy as np
import PyGeoStudio as pgs
src_file = "../GeoStudio_files/Rapid drawdown.gsz" #specify the geostudio analyses file
geofile = pgs.GeoStudioFile(src_file) #open it with PyGeoStudio

# %%
# Timesteps are defined in analysis, and can retrieven with
geofile.showAnalysisTree()
analysis = geofile.getAnalysisByID(4)
timestepping = analysis["TimeIncrements"]

# %%
# Print timestep details:
timestepping.showTimeSteps()

# %%
# Set new timesteps:
end = timestepping["Duration"]
new_times = list(np.logspace(2,np.log10(end),50)) #exponential timesteps
saved = [False for x in new_times]
new_times += [1000, 1e4, 1e6, 1e6] #add user defined timesteps
saved += [True] * 4
timestepping.setTimeSteps(new_times, saved)

# %%
# Modification can be saved under a new file
print("----------------------")
out_file = "./rapid_drawdown_changed_timestep.gsz"
geofile.saveAs(out_file)