Integrate LGM & Update 3DGS to Mesh Conversion Algorithm

shared_utils/common_utils.py

@@ -1,4 +1,6 @@
 import os
+from os import listdir
+from os.path import isfile, join
 import sys
 from datetime import datetime
 
@@ -83,9 +85,9 @@ cstr.color.add_code("error", f"{cstr.color.RED}[Comfy3D] [ERROR] {cstr.color.END
 
 def get_persistent_directory(folder_name):
     if sys.platform == "win32":
-        folder = os.path.join(os.path.expanduser("~"), "AppData", "Local", folder_name)
+        folder = join(os.path.expanduser("~"), "AppData", "Local", folder_name)
     else:
-        folder = os.path.join(os.path.expanduser("~"), "." + folder_name)
+        folder = join(os.path.expanduser("~"), "." + folder_name)
     
     os.makedirs(folder, exist_ok=True)
     return folder
@@ -96,7 +98,7 @@ def parse_save_filename(save_path, output_directory, supported_extensions, class
     filename, file_extension = os.path.splitext(filename)
     if file_extension.lower() in supported_extensions:
         if not os.path.isabs(save_path):
-            folder_path = os.path.join(output_directory, folder_path)
+            folder_path = join(output_directory, folder_path)
         
         os.makedirs(folder_path, exist_ok=True)
         
@@ -107,10 +109,14 @@ def parse_save_filename(save_path, output_directory, supported_extensions, class
             if date_format in filename:
                 filename = filename.replace(date_format, now.strftime(date_format))
                 
-        save_path = os.path.join(folder_path, filename) + file_extension
+        save_path = join(folder_path, filename) + file_extension
         cstr(f"[{class_name}] Saving model to {save_path}").msg.print()
         return save_path
     else:
         cstr(f"[{class_name}] File name {filename} does not end with supported file extensions: {supported_extensions}").error.print()
     
-    return None
+    return None
+
+# Get all files in a folder, if extension_filter is provided then only reture the files with extensions in extension_filter
+def get_list_filenames(directory, extension_filter=None):
+    return [f for f in listdir(directory) if isfile(join(directory, f)) and (extension_filter is None or f.lower().endswith(extension_filter))]