WIP implement document assign/unassign. See #1222.

src/ifcblenderexport/blenderbim/bim/module/document/ui.py

@@ -23,7 +23,7 @@ class BIM_PT_documents(Panel):
 
         if not self.props.is_editing or self.props.is_editing == "information":
             row = self.layout.row(align=True)
-            row.label(text="{} Documents Found".format(len(Data.information)))
+            row.label(text="{} Documents Found".format(len(Data.information)), icon="FILE")
             if self.props.is_editing == "information":
                 row.operator("bim.disable_document_editing_ui", text="", icon="CHECKMARK")
                 row.operator("bim.add_information", text="", icon="ADD")
@@ -32,7 +32,7 @@ class BIM_PT_documents(Panel):
 
         if not self.props.is_editing or self.props.is_editing == "reference":
             row = self.layout.row(align=True)
-            row.label(text="{} References Found".format(len(Data.references)))
+            row.label(text="{} References Found".format(len(Data.references)), icon="FILE_HIDDEN")
             if self.props.is_editing == "reference":
                 row.operator("bim.disable_document_editing_ui", text="", icon="CHECKMARK")
                 row.operator("bim.add_document_reference", text="", icon="ADD")
@@ -64,6 +64,77 @@ class BIM_PT_documents(Panel):
                 row.prop(attribute, "is_null", icon="RADIOBUT_OFF" if attribute.is_null else "RADIOBUT_ON", text="")
 
 
+class BIM_PT_object_documents(Panel):
+    bl_label = "IFC Documents"
+    bl_idname = "BIM_PT_object_documents"
+    bl_options = {"DEFAULT_CLOSED"}
+    bl_space_type = "PROPERTIES"
+    bl_region_type = "WINDOW"
+    bl_context = "object"
+
+    @classmethod
+    def poll(cls, context):
+        return bool(context.active_object.BIMObjectProperties.ifc_definition_id)
+
+    def draw(self, context):
+        obj = context.active_object
+        self.oprops = obj.BIMObjectProperties
+        self.sprops = context.scene.BIMDocumentProperties
+        self.props = obj.BIMObjectDocumentProperties
+        self.file = IfcStore.get_file()
+        if not Data.is_loaded:
+            Data.load()
+        if self.oprops.ifc_definition_id not in Data.products:
+            Data.load(self.oprops.ifc_definition_id)
+
+        self.draw_add_ui()
+
+        document_ids = Data.products[self.oprops.ifc_definition_id]
+
+        if not document_ids:
+            row = self.layout.row(align=True)
+            row.label(text="No Documents", icon="FILE")
+
+        for document_id in document_ids:
+            try:
+                document = Data.information[document_id]
+                document_type = "IfcDocumentInformation"
+                icon = "FILE"
+            except:
+                document = Data.references[document_id]
+                document_type = "IfcDocumentReference"
+                icon = "FILE_HIDDEN"
+            row = self.layout.row(align=True)
+            if self.file.schema == "IFC2X3":
+                if document_type == "IfcDocumentInformation":
+                    row.label(text=document.get("DocumentId") or "*", icon=icon)
+                elif document_type == "IfcDocumentReference":
+                    row.label(text=document.get("ItemReference") or "*", icon=icon)
+            else:
+                row.label(text=document.get("Identification") or "*", icon=icon)
+            row.label(text=document.get("Name") or "Unnamed")
+            row.operator("bim.unassign_document", text="", icon="X").document = document_id
+
+    def draw_add_ui(self):
+        if self.props.is_adding:
+            row = self.layout.row(align=True)
+            icon = "FILE" if self.props.is_adding == "IfcDocumentInformation" else "FILE_HIDDEN"
+            row.label(text="Adding {}".format(self.props.is_adding), icon=icon)
+            row.operator("bim.disable_assigning_document", text="", icon="X")
+            self.layout.template_list(
+                "BIM_UL_object_documents",
+                "",
+                self.sprops,
+                "documents",
+                self.sprops,
+                "active_document_index",
+            )
+        else:
+            row = self.layout.row(align=True)
+            row.prop(self.props, "available_document_types", text="")
+            row.operator("bim.enable_assigning_document", text="", icon="ADD")
+
+
 class BIM_UL_documents(UIList):
     def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
         if item:
@@ -82,3 +153,12 @@ class BIM_UL_documents(UIList):
                 op = row.operator("bim.enable_editing_document", text="", icon="GREASEPENCIL")
                 op.document = item.ifc_definition_id
                 row.operator("bim.remove_document", text="", icon="X").document = item.ifc_definition_id
+
+
+class BIM_UL_object_documents(UIList):
+    def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
+        if item:
+            row = layout.row(align=True)
+            row.label(text=item.identification)
+            row.label(text=item.name)
+            row.operator("bim.assign_document", text="", icon="ADD").document = item.ifc_definition_id