mavpicviewer: image review tool

MAVProxy/tools/mavpicviewer/mavpicviewer.bat

@@ -0,0 +1,4 @@
+cd ..\..\
+python setup.py build install --user
+python .\MAVProxy\tools\mavpicviewer\mavpicviewer.py
+pause

MAVProxy/tools/mavpicviewer/mavpicviewer.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python3
+
+'''
+MAV Picture Viewer
+
+Quick and efficient reviewing of images taken from a drone
+
+AP_FLAKE8_CLEAN
+'''
+
+import os
+from argparse import ArgumentParser
+from pathlib import Path
+from MAVProxy.modules.lib import multiproc
+import mavpicviewer_image
+
+prefix_str = "mavpicviewer: "
+
+
+class mavpicviewer():
+
+    # constructor
+    def __init__(self):
+        self.mavpicviewer_image = None
+
+    # display dialog to open a folder
+    def cmd_openfolder(self, folderpath):
+        file_list = self.file_list(folderpath, ['jpg', 'jpeg'])
+        if file_list is None or not file_list:
+            print("mavpicviewer: no files found")
+            return
+        self.mavpicviewer_image = mavpicviewer_image.mavpicviewer_image(file_list)
+
+    # open picture viewer to display a single file
+    def cmd_openfile(self, filepath):
+        # check file exists
+        if not Path(filepath).exists():
+            print("mavpicviewer: %s not found" % filepath)
+            return
+        filelist = []
+        filelist.append(filepath)
+        self.mavpicviewer_image = mavpicviewer_image.mavpicviewer_image(filelist)
+
+    # return an array of files for a given directory and extension
+    def file_list(self, directory, extensions):
+        '''return file list for a directory'''
+        flist = []
+        for filename in os.listdir(directory):
+            extension = filename.split('.')[-1]
+            if extension.lower() in extensions:
+                flist.append(os.path.join(directory, filename))
+        sorted_list = sorted(flist, key=str.lower)
+        return sorted_list
+
+
+# main function
+if __name__ == "__main__":
+    multiproc.freeze_support()
+    parser = ArgumentParser(description=__doc__)
+    parser.add_argument("filepath", default=".", help="filename or directory holding images")
+    args = parser.parse_args()
+
+    # check destination directory exists
+    if not os.path.exists(args.filepath):
+        exit(prefix_str + "invalid destination directory")
+
+    # check if file or directory
+    if os.path.isfile(args.filepath):
+        mavpicviewer = mavpicviewer()
+        mavpicviewer.cmd_openfile(args.filepath)
+    else:
+        mavpicviewer = mavpicviewer()
+        mavpicviewer.cmd_openfolder(args.filepath)