Add some information to the completed message

Author: MeguminBOT

TextureAtlas to GIF and Frames.py

@@ -8,6 +8,7 @@
 import webbrowser
 import xml.etree.ElementTree as ET
 import json
+import time
 
 from PIL import Image
 import tkinter as tk
@@ -243,7 +244,13 @@ def store_input():
 
 def process_directory(input_dir, output_dir, progress_var, tk_root, create_gif, create_webp, keep_frames, set_framerate, set_loopdelay, set_threshold):
     if not (create_gif or create_webp or keep_frames):
+        messagebox.showinfo("Information", "Nothing to process.\nPlease select tasks.")
         return
+    
+    total_frames_generated = 0
+    total_gifs_generated = 0
+    total_sprites_failed = 0
+
     progress_var.set(0)
     total_files = count_xml_files(input_dir)
     progress_bar["maximum"] = total_files
@@ -253,6 +260,8 @@ def process_directory(input_dir, output_dir, progress_var, tk_root, create_gif,
     else:
         cpuThreads = os.cpu_count() // 2
 
+    start_time = time.time()
+
     with concurrent.futures.ThreadPoolExecutor(max_workers=cpuThreads) as executor:
         futures = []
 
@@ -270,23 +279,42 @@ def process_directory(input_dir, output_dir, progress_var, tk_root, create_gif,
 
         for future in concurrent.futures.as_completed(futures):
             try:
-                future.result()
+                result = future.result()
+                total_frames_generated += result['frames_generated']
+                total_gifs_generated += result['gifs_generated']
+                total_sprites_failed += result['sprites_failed']
             except ET.ParseError as e:
+                total_sprites_failed += 1
                 messagebox.showerror("Error", f"Something went wrong!!\n{e}")
                 if not messagebox.askyesno("Continue?", "Do you want to try continue processing?"):
                     sys.exit()
             except Exception as e:
+                total_sprites_failed += 1
                 messagebox.showerror("Error", f"Something went wrong!!\n{str(e)}")
                 if not messagebox.askyesno("Continue?", "Do you want to try continue processing?"):
                     sys.exit()
 
             progress_var.set(progress_var.get() + 1)
             tk_root.update_idletasks()
 
-    messagebox.showinfo("Information","Finished processing all files.")
+    end_time = time.time()
+    duration = end_time - start_time
+    minutes, seconds = divmod(duration, 60)
+
+    messagebox.showinfo(
+        "Information",
+        f"Finished processing all files.\n\n"
+        f"Frames Generated: {total_frames_generated}\n"
+        f"GIFs Generated: {total_gifs_generated}\n"
+        f"Sprites Failed: {total_sprites_failed}\n\n"
+        f"Processing Duration: {int(minutes)} minutes and {int(seconds)} seconds",
+    )
 
 ## Extraction logic
 def extract_sprites(atlas_path, metadata_path, output_dir, create_gif, create_webp, keep_frames, set_framerate, set_loopdelay, set_threshold):
+    frames_generated = 0
+    gifs_generated = 0
+    sprites_failed = 0
     try:
         atlas = Image.open(atlas_path)
         if metadata_path.endswith('.xml'):
@@ -340,6 +368,8 @@ def extract_sprites(atlas_path, metadata_path, output_dir, create_gif, create_we
             os.makedirs(sprite_folder, exist_ok=True)
 
             frame_image.save(os.path.join(sprite_folder, f"{name}.png"))
+            if keep_frames:
+                frames_generated += 1
 
             if create_gif or create_webp:
                 animations.setdefault(folder_name, []).append((name, frame_image))
@@ -395,20 +425,32 @@ def extract_sprites(atlas_path, metadata_path, output_dir, create_gif, create_we
                 durations = [round(1000/fps)] * len(cropped_images)
                 durations[-1] += delay
                 cropped_images[0].save(os.path.join(output_dir, os.path.splitext(spritesheet_name)[0] + f" {animation_name}.gif"), save_all=True, append_images=cropped_images[1:], disposal=2, optimize=False, duration=durations, loop=0, comment=f'GIF generated by: TextureAtlas to GIF and Frames v{current_version}')
+                gifs_generated += 1
+
             if not keep_frames:
                 frames_folder = os.path.join(output_dir, animation_name)
                 for i in os.listdir(frames_folder):
                     os.remove(os.path.join(frames_folder, i))
                 os.rmdir(frames_folder)
 
+        return {
+            'frames_generated': frames_generated,
+            'gifs_generated': gifs_generated,
+            'sprites_failed': sprites_failed
+        }
+
     except ET.ParseError:
+        sprites_failed += 1
         raise ET.ParseError(f"Badly formatted XML file:\n\n{metadata_path}")
     except Exception as e:
         if "Coordinate '" in str(e) and "' is less than '" in str(e):
+            sprites_failed += 1
             raise Exception(f"XML or TXT frame dimension data doesn't match the spritesheet dimensions.\n\nError: {str(e)}\n\nFile: {metadata_path}")
         elif "'NoneType' object is not subscriptable" in str(e):
+            sprites_failed += 1
             raise Exception(f"XML or TXT frame dimension data doesn't match the spritesheet dimensions.\n\nError: {str(e)}\n\nFile: {metadata_path}")
         else:
+            sprites_failed += 1
             raise Exception(f"An error occurred: {str(e)}.\n\nFile:{metadata_path}")
 
 ## FNF specific stuff