From 840a94a7b95c2a45c63884b1c2b0d0c8c8bcb03d Mon Sep 17 00:00:00 2001 From: Camille Woicekowska Date: Thu, 11 Jan 2024 16:49:34 -0800 Subject: [PATCH 1/3] Added function to get --- flopy/modflow/mfdis.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/flopy/modflow/mfdis.py b/flopy/modflow/mfdis.py index a146eeb8d0..16f11713d1 100644 --- a/flopy/modflow/mfdis.py +++ b/flopy/modflow/mfdis.py @@ -343,6 +343,31 @@ def get_final_totim(self): """ return self.get_totim()[-1] + + def get_stp_per_perlen(self): + """ + Get lists for stress periods, time steps, and period lengths in the model. + + Returns + ------- + per: list of ints + Stress periods in the model + stp: list of ints + Timesteps in each stress period + perlen: list of ints + Length of each timestep + """ + + nstp = self.nstp.array + nperlen = self.perlen.array + per = [] + stp = [] + perlen = [] + for iper in range(self.nper): + per.append(iper+1) # Use 1-based indexing + stp.append(nstp[iper]) + perlen.append(nperlen[iper]) + return per, stp, perlen def get_kstp_kper_toffset(self, t=0.0, use_cached_totim=False): """ From 7455ccbd46c1b149be2aea7797e4fc66502db262 Mon Sep 17 00:00:00 2001 From: Camille Woicekowska Date: Thu, 11 Jan 2024 16:50:55 -0800 Subject: [PATCH 2/3] Added function to get per, stp, perlen --- flopy/modflow/mfdis.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flopy/modflow/mfdis.py b/flopy/modflow/mfdis.py index 16f11713d1..c77dff033d 100644 --- a/flopy/modflow/mfdis.py +++ b/flopy/modflow/mfdis.py @@ -344,7 +344,7 @@ def get_final_totim(self): """ return self.get_totim()[-1] - def get_stp_per_perlen(self): + def get_per_stp_perlen(self): """ Get lists for stress periods, time steps, and period lengths in the model. @@ -357,7 +357,7 @@ def get_stp_per_perlen(self): perlen: list of ints Length of each timestep """ - + nstp = self.nstp.array nperlen = self.perlen.array per = [] From 3c60197d87fc41d8dc89438780c472dfe78600c9 Mon Sep 17 00:00:00 2001 From: Camille Woicekowska Date: Fri, 12 Jan 2024 09:25:46 -0800 Subject: [PATCH 3/3] Remformatted with black --- flopy/modflow/mfdis.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/flopy/modflow/mfdis.py b/flopy/modflow/mfdis.py index c77dff033d..69b933f052 100644 --- a/flopy/modflow/mfdis.py +++ b/flopy/modflow/mfdis.py @@ -343,17 +343,17 @@ def get_final_totim(self): """ return self.get_totim()[-1] - + def get_per_stp_perlen(self): """ - Get lists for stress periods, time steps, and period lengths in the model. + Get lists for stress periods, time steps, and period lengths in the model. Returns ------- per: list of ints Stress periods in the model stp: list of ints - Timesteps in each stress period + Timesteps in each stress period perlen: list of ints Length of each timestep """ @@ -364,10 +364,10 @@ def get_per_stp_perlen(self): stp = [] perlen = [] for iper in range(self.nper): - per.append(iper+1) # Use 1-based indexing + per.append(iper + 1) # Use 1-based indexing stp.append(nstp[iper]) perlen.append(nperlen[iper]) - return per, stp, perlen + return per, stp, perlen def get_kstp_kper_toffset(self, t=0.0, use_cached_totim=False): """