WIP: drs import

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