Add properties for uv name and enable unwrap (#15)

Author: automas-dev

README.md

@@ -21,13 +21,15 @@ Download this repo as a zip file and install through the Blender addons.
 
 ### Inputs
 
-| Field       | Description                                            |
-| ----------- | ------------------------------------------------------ |
-| Name        | Name used for baked images and material                |
-| Mode        | What to do with images after bake                      |
-| Size        | Bake image size in pixels                              |
-| Save Images | Save images to external directory                      |
-| Layers      | Enable baking of different layers, one image per layer |
+| Field         | Description                                            |
+| ------------- | ------------------------------------------------------ |
+| Name          | Name used for baked images and material                |
+| Mode          | What to do with images after bake                      |
+| Size          | Bake image size in pixels                              |
+| Bake UV       | UV layer name used for baking                          |
+| Unwrap Object | Unwrap the object using smart project before baking    |
+| Save Images   | Save images to external directory                      |
+| Layers        | Enable baking of different layers, one image per layer |
 
 Mode has the following options
 

docs/panel.png

@@ -214,16 +214,17 @@ def execute(self, context: bpy.types.Context):
         return {"FINISHED"}
 
     def unwrap_object(self, mesh: bpy.types.Mesh) -> bpy.types.MeshUVLoopLayer:
-        uv_name = "bake_uv"
-        _log.debug("Unwrapping mesh %s with uv layer %s", mesh.name, uv_name)
+        _log.debug("Unwrapping mesh %s with uv layer %s", mesh.name, self.props.uv_name)
 
         # Use existing or create new uv layer for baking
-        bake_uv = mesh.uv_layers.get(uv_name)
+        bake_uv = mesh.uv_layers.get(self.props.uv_name)
         if bake_uv is None:
-            _log.info("Creating new uv layer %s", uv_name)
-            bake_uv = mesh.uv_layers.new(name=uv_name)
+            _log.info("Creating new uv layer %s", self.props.uv_name)
+            bake_uv = mesh.uv_layers.new(name=self.props.uv_name)
         else:
-            _log.debug("Reusing existing uv layer %s", uv_name)
+            _log.debug("Reusing existing uv layer %s", self.props.uv_name)
+            if not self.props.unwrap_object:
+                return bake_uv
 
         # Store currently active layer
         active_layer = None

quickbake/op.py

@@ -25,8 +25,10 @@ def draw(self, context):
         props = scene.QuickBakeToolPropertyGroup  # type: ignore
 
         layout.prop(props, "bake_name")
-        layout.prop(props, "mat_mode")
         layout.prop(props, "bake_size")
+        layout.prop(props, "uv_name")
+        layout.prop(props, "unwrap_object")
+        layout.prop(props, "mat_mode")
         layout.prop(props, "save_img")
 
         row = layout.row()

quickbake/panel.py

@@ -28,6 +28,21 @@ class QuickBakeToolPropertyGroup(bpy.types.PropertyGroup):
         step=1024,  # not yet implemented
     )
 
+    uv_name: bpy.props.StringProperty(
+        name="Bake UV",
+        description="UV layer name used for bake texture",
+        default="bake_uv",
+    )
+
+    unwrap_object: bpy.props.BoolProperty(
+        name="Unwrap Object",
+        description=(
+            "Unwrap object using smart project before baking. "
+            "If the uv layer does not exist, the object will be unwrapped regardless of this option."
+        ),
+        default=True,
+    )
+
     mat_mode: bpy.props.EnumProperty(
         name="Mode",
         description="What to do with images after baking",