From e9fafdbb8a6399634ce2034ca85285e55b01d15e Mon Sep 17 00:00:00 2001
From: HappyZ <zhuyanzi@gmail.com>
Date: Wed, 5 Jun 2019 12:41:15 -0500
Subject: [PATCH] add orientation check in foldername

---
 preprocessor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/preprocessor.py b/preprocessor.py
index 97fb430..e0d84bc 100644
--- a/preprocessor.py
+++ b/preprocessor.py
@@ -119,6 +119,9 @@ def main(args):
     if not os.path.isdir(args.folder):
         print("Err: folder {} does not exist".format(args.folder))
         sys.exit(2)
+    if 'orient' in args.folder:
+        orientation = int(args.folder.rstrip('/').split('_')[-1])
+        print("orientation: {}".format(orientation))
     f_map, f_loc, f_sig, f_gt, is_csi = get_files(args.folder)
     if f_loc is None or f_sig is None:
         print("Err: desired files not exist")