adds object as source

build/.version

@@ -1 +1 @@
-1.0.0+build.18
+1.1.0+build.1

src/operators/bake_to_id_map.py

@@ -50,4 +50,5 @@ class BakeToIDMapOperator(bpy.types.Operator):
 
         target = get_target(props.target)
         target.paint_targets(props, targets, colors)
-        source.after_painting(context, props)
+        if 'after_painting' in dir(source):
+            source.after_painting(context, props)

src/types/__init__.py

@@ -1,9 +1,11 @@
 from . sources import material_index as source_mat_index
+from . sources import object as source_object
 
 from . targets import vertex_colors as target_vertex_colors
 
 _sources = [
-    source_mat_index
+    source_mat_index,
+    source_object
 ]
 
 _targets = [

src/types/sources/object.py

@@ -0,0 +1,16 @@
+source_id = 'OBJECT'
+name = 'Object ID'
+description = 'Uses the object id as basis for the ID mask.'
+
+connected_properties = []
+
+def get_targets(context):
+    targets = []
+    for obj in context.selected_objects:
+        mesh = obj.data
+        targets.append((mesh, mesh.polygons))
+
+    return targets
+
+def estimate_ids(context, props):
+    return len(context.selected_objects)