From 5bbe12110dc9f4767d8349906179b95b042bd574 Mon Sep 17 00:00:00 2001
From: Lionel GUEZ <guez@lmd.ens.fr>
Date: Tue, 11 Apr 2023 21:21:38 +0200
Subject: [PATCH] Import script `write_date_range.py`
This will be useful to create `n_dates_1` and `n_dates_14`. It should
be faster for this purpose than using the more general script `filter.py`.
---
Inst_eddies/Analysis/write_date_range.py | 61 ++++++++++++++++++++++++
1 file changed, 61 insertions(+)
create mode 100755 Inst_eddies/Analysis/write_date_range.py
diff --git a/Inst_eddies/Analysis/write_date_range.py b/Inst_eddies/Analysis/write_date_range.py
new file mode 100755
index 00000000..e77bdebf
--- /dev/null
+++ b/Inst_eddies/Analysis/write_date_range.py
@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+
+"""This script reads data in a date range from an SHPC and writes it
+to a new SHPC. We assume that all the dates in the date range are in
+the same slice of the input SHPC. A single slice is written in the
+output SHPC.
+
+"""
+
+import shapefile
+from os import path
+import util_eddies
+import os
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("input_SHPC")
+parser.add_argument("output_SHPC")
+parser.add_argument("date_start", type = int)
+parser.add_argument("-n", "--n_dates", type = int, default = 1)
+args = parser.parse_args()
+SHPC = util_eddies.SHPC_class(args.input_SHPC, "Anticyclones")
+slice_dir = path.join(args.output_SHPC, "Slice_0")
+date_stop = args.date_start + args.n_dates
+i_slice = SHPC.get_slice(args.date_start)
+os.makedirs(slice_dir)
+
+for orientation in ["Anticyclones", "Cyclones"]:
+ orient_dir = path.join(slice_dir, orientation)
+ os.mkdir(orient_dir)
+ extremum = SHPC.get_reader(i_slice, "extremum", orientation)
+ outermost_cont = SHPC.get_reader(i_slice, "outermost_contour", orientation)
+ max_speed_cont = SHPC.get_reader(i_slice, "max_speed_contour", orientation)
+ fname = path.join(orient_dir, "ishape_last.txt")
+
+ with shapefile.Writer(path.join(orient_dir, "extremum")) as extremum_filt, \
+ shapefile.Writer(path.join(orient_dir, "outermost_contour")) \
+ as outermost_cont_filt, \
+ shapefile.Writer(path.join(orient_dir, "max_speed_contour")) \
+ as max_speed_cont_filt, open(fname, "w") as ishape_last:
+ extremum_filt.fields = extremum.fields[1:]
+ outermost_cont_filt.fields = outermost_cont.fields[1:]
+ max_speed_cont_filt.fields = max_speed_cont.fields[1:]
+
+ for my_date in range(args.date_start, date_stop):
+ eddy_index = 0
+
+ for ishape in SHPC.ishape_range(my_date, i_slice, orientation):
+ eddy_index += 1
+
+ for sr, w in zip([extremum.shapeRecord(ishape),
+ outermost_cont.shapeRecord(ishape),
+ max_speed_cont.shapeRecord(ishape)],
+ [extremum_filt, outermost_cont_filt,
+ max_speed_cont_filt]):
+ r = sr.record
+ r["eddy_index"] = eddy_index
+ w.record(*r)
+ w.shape(sr.shape)
+
+ ishape_last.write(f"{len(extremum_filt) - 1}\n")
--
GitLab