From f81920a4a50e2ee50fb6c6c2a3d91cb1ab625bd6 Mon Sep 17 00:00:00 2001 From: Remi Cresson <remi.cresson@irstea.fr> Date: Tue, 15 Feb 2022 16:10:45 +0100 Subject: [PATCH] WIP: drs import --- test/drs_import.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/test/drs_import.py b/test/drs_import.py index 6951ad7..f01a8d2 100644 --- a/test/drs_import.py +++ b/test/drs_import.py @@ -1,7 +1,8 @@ -import spot import argparse import os from glob import glob +import tqdm +import spot # Arguments parser = argparse.ArgumentParser(description="Test",) @@ -18,15 +19,14 @@ dimap_xs_files = _find_dimaps(params.root_dir + "/MS") print("Found {} DIMAP files in MS folder".format(len(dimap_xs_files))) # find pairs -print("Find DIMAP pairs...") -for dimap_xs_file in dimap_xs_files: - pan_path = dimap_xs_file[:dimap_xs_file.find("/PROD_SPOT")] +print("Creating scenes...") +scene_list = [] +for dimap_file_xs in tqdm(dimap_xs_files): + pan_path = dimap_file_xs[:dimap_file_xs.find("/PROD_SPOT")] pan_path = pan_path.replace("/MS/", "/PAN/") pan_path = pan_path.replace("_MS_", "_PAN_") dimap_pan_files = _find_dimaps(pan_path) if len(dimap_pan_files) != 1: raise Exception("{} DIMAPS candidates found in {} ".format(pan_path)) - dimap_pan_file = dimap_pan_files[0] -print("ok") - -# create scenes \ No newline at end of file + dimap_file_pan = dimap_pan_files[0] + scene_list.append(spot.Spot67Scene(dimap_file_pan=dimap_file_pan, dimap_file_xs=dimap_file_xs)) \ No newline at end of file -- GitLab