Add option to profile the importer

.gitignore

@@ -7,3 +7,4 @@ images/template*
 env/
 .venv/
 release/
+*.prof

pcb2blender_importer/importer.py

@@ -1,4 +1,4 @@
-import io, random, re, shutil, struct, tempfile
+import io, random, re, shutil, struct, sys, tempfile
 from dataclasses import dataclass
 from enum import Enum
 from math import inf, radians
@@ -21,6 +21,13 @@
 from .blender_addon_utils import ErrorHelper
 from .materials import *
 
+ENABLE_PROFILER = False
+if ENABLE_PROFILER:
+    from cProfile import Profile
+
+    def has_debugger_attached():
+        return sys.gettrace() is not None
+
 PCB = "pcb.wrl"
 COMPONENTS = "components"
 LAYERS = "layers"
@@ -200,6 +207,10 @@ def __init__(self):
     def execute(self, context):
         filepath = Path(self.filepath)
 
+        if ENABLE_PROFILER and has_debugger_attached():
+            profiler = Profile()
+            profiler.enable()
+
         # import boards
 
         if (pcb := self.import_pcb3d(context, filepath)) == {"CANCELLED"}:
@@ -276,6 +287,12 @@ def execute(self, context):
         if self.enhance_materials:
             enhance_materials(self.new_materials)
 
+        if ENABLE_PROFILER and has_debugger_attached():
+            profiler.disable()
+            profiler.dump_stats(
+                Path(__file__).parent.resolve() / Path(__file__).with_suffix(".prof").name
+            )
+
         return {"FINISHED"}
 
     def import_pcb3d(self, context, filepath):

requirements-dev.txt

@@ -3,6 +3,7 @@ gitpython
 pytest
 pytest-blender
 requests
+snakeviz
 
 fake-bpy-module
 numpy