diff --git a/test/drs_import.py b/test/drs_import.py index 694d9b3b489e931f4958d9e2b8edf3413711c1a7..336aa712d6bf54049c73a3c3ed26db8a7b7b379a 100644 --- a/test/drs_import.py +++ b/test/drs_import.py @@ -8,12 +8,25 @@ parser = argparse.ArgumentParser(description="Test",) parser.add_argument("--root_dir", help="Root directory containing MS and PAN folders", required=True) params = parser.parse_args() +# Find all DIMAPS files +def _find_dimaps(pth): + return [y for x in os.walk(pth) for y in glob(os.path.join(x[0], "DIM*.XML"))] + # List files print("List files...") -files = [y for x in os.walk(params.root_dir) for y in glob(os.path.join(x[0], "DIM*.XML"))] -print(len(files)) +dimap_xs_files = _find_dimaps(params.root_dir + "/MS") +print("Found {} DIMAP files in MS folder".format(len(dimap_xs_files))) -for i in range(0,100): - print(files[i]) +# find pairs +print("Find DIMAP pairs...") +for dimap_xs_file in dimap_xs_files: + pan_path = xs_pth[:dimap_xs_file.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") -# Test spot class +# create scenes \ No newline at end of file