Issue #64: correct output files splitting

ec-jrc · Mar 15, 2024 · b7debaf · b7debaf
1 parent a2b9c0d
commit b7debaf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/lisvap/utils/writers.py b/src/lisvap/utils/writers.py
@@ -78,6 +78,9 @@ def get_output_parameters_monthly(start_date, timestep, time_frequency, timestep
  output_index = 0
  elif current_date == last_date_last_month:
  filename_suffix = day_inside_last_month.strftime('%Y%m')
+ first_date_last_month = day_inside_last_month.replace(day=1) + datetime.timedelta(seconds=(2 * timestep_stride))
+ num_steps_done_in_last_month = int((last_date_last_month - first_date_last_month).total_seconds() / timestep_stride) + 1
+ output_index = num_steps_done_in_last_month
  elif current_output_index >= num_steps_done_in_current_month:
  output_index = num_steps_done_in_current_month - 1
  return filename_suffix, output_index
@@ -98,6 +101,9 @@ def get_output_parameters_yearly(start_date, timestep, time_frequency, timestep_
  output_index = 0
  elif current_date == last_date_last_year:
  filename_suffix = day_inside_last_year.strftime('%Y')
+ first_date_last_year = first_date_current_year.replace(year=current_date.year - 1)
+ num_steps_done_in_last_year = int((last_date_last_year - first_date_last_year).total_seconds() / timestep_stride) + 1
+ output_index = num_steps_done_in_last_year - 1
  elif current_output_index >= num_steps_done_in_current_year:
  output_index = num_steps_done_in_current_year - 1
  return filename_suffix, output_index