From 0d07a9c92673219a9ed8e04ed1d4494b57bd5a33 Mon Sep 17 00:00:00 2001
From: Remi Cresson <remi.cresson@irstea.fr>
Date: Fri, 4 Mar 2022 22:44:46 +0100
Subject: [PATCH] ADD: app to serialize collections of S2 level 2A or 3A
---
test/drs_import.py | 4 ++--
test/s2_import.py | 6 ++++--
2 files changed, 6 insertions(+), 4 deletions(-)
diff --git a/test/drs_import.py b/test/drs_import.py
index 2be81e5..a0c561f 100644
--- a/test/drs_import.py
+++ b/test/drs_import.py
@@ -3,13 +3,13 @@ from scenes import drs
# Arguments
parser = argparse.ArgumentParser(description="Test",)
-parser.add_argument("--root_dir", nargs='+', help="Root directories containing MS and PAN folders", required=True)
+parser.add_argument("--root_dirs", nargs='+', help="Root directories containing MS and PAN folders", required=True)
parser.add_argument("--out_pickle", help="Output pickle file", required=True)
params = parser.parse_args()
# Get all scenes in the root_dir
scenes = []
-for root_dir in params.root_dir:
+for root_dir in params.root_dirs:
scenes += drs.get_spot67_scenes(root_dir)
# Save scenes in a pickle file
diff --git a/test/s2_import.py b/test/s2_import.py
index 9c01fd3..ca30e5e 100644
--- a/test/s2_import.py
+++ b/test/s2_import.py
@@ -3,14 +3,16 @@ from scenes import utils, sentinel, drs
# Arguments
parser = argparse.ArgumentParser(description="Test",)
-parser.add_argument("--root_dir", nargs='+', help="Root directories containing S2 archives (.zip)", required=True)
+parser.add_argument("--root_dirs", nargs='+', help="Root directories containing S2 archives (.zip)", required=True)
parser.add_argument("--out_pickle", help="Output pickle file", required=True)
parser.add_argument('--level', default='2A', const='2A', nargs='?', choices=['2A', '3A'], help='Product level')
params = parser.parse_args()
# Get all scenes in the root_dir
product_type = sentinel.Sentinel23AScene if params.level == '2A' else sentinel.Sentinel23AScene
-archives = utils.find_files_in_all_subdirs(pth=params.root_dir, pattern="*.zip", case_sensitive=False)
+archives = []
+for root_dir in params.root_dirs:
+ archives += utils.find_files_in_all_subdirs(pth=root_dir, pattern="*.zip", case_sensitive=False)
scenes = [product_type(archive=archive) for archive in archives]
# Save scenes in a pickle file
--
GitLab