diff --git a/test/drs_import.py b/test/drs_import.py index 6951ad7517badf24e32839b3337055f7ade09025..f01a8d24a896a4ebc0d15f7612f8c93f409ee70d 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