From a03fbab4d50004d10aea9faeed79d6b46c387b55 Mon Sep 17 00:00:00 2001
From: Andrej730 <azhilenkov@gmail.com>
Date: Fri, 31 Mar 2023 11:05:39 +0500
Subject: [PATCH] Added description to some annotation operators

---
 src/blenderbim/blenderbim/bim/module/drawing/operator.py | 5 +++++
 src/blenderbim/blenderbim/bim/module/drawing/ui.py       | 2 ++
 src/blenderbim/blenderbim/bim/module/pset/operator.py    | 1 +
 3 files changed, 8 insertions(+)

diff --git a/src/blenderbim/blenderbim/bim/module/drawing/operator.py b/src/blenderbim/blenderbim/bim/module/drawing/operator.py
index 01f1bf0532..a0311a9325 100644
--- a/src/blenderbim/blenderbim/bim/module/drawing/operator.py
+++ b/src/blenderbim/blenderbim/bim/module/drawing/operator.py
@@ -668,11 +668,16 @@ class AddAnnotation(bpy.types.Operator, Operator):
     bl_options = {"REGISTER", "UNDO"}
     object_type: bpy.props.StringProperty()
     data_type: bpy.props.StringProperty()
+    description: bpy.props.StringProperty()
 
     @classmethod
     def poll(cls, context):
         return IfcStore.get_file() and context.scene.camera
 
+    @classmethod
+    def description(cls, context, operator):
+        return operator.description or ""
+
     def _execute(self, context):
         drawing = tool.Ifc.get_entity(context.scene.camera)
         if not drawing:
diff --git a/src/blenderbim/blenderbim/bim/module/drawing/ui.py b/src/blenderbim/blenderbim/bim/module/drawing/ui.py
index 7b48f4f2f2..1520ca27ad 100644
--- a/src/blenderbim/blenderbim/bim/module/drawing/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/drawing/ui.py
@@ -446,6 +446,7 @@ class BIM_PT_annotation_utilities(Panel):
         op = row.operator("bim.add_annotation", text="Stair Arrow", icon="SCREEN_BACK")
         op.object_type = "STAIR_ARROW"
         op.data_type = "curve"
+        op.description = "Add stair arrow annotation.\nIf you have IfcStairFlight object selected, it will be used as a reference for the annotation"
         op = row.operator("bim.add_annotation", text="Hidden", icon="CON_TRACKTO")
         op.object_type = "HIDDEN_LINE"
         op.data_type = "mesh"
@@ -470,6 +471,7 @@ class BIM_PT_annotation_utilities(Panel):
         op = row.operator("bim.add_annotation", text="Batting", icon="FORCE_FORCE")
         op.object_type = "BATTING"
         op.data_type = "mesh"
+        op.description = "Add batting annotation.\nThickness could be changed through Thickness property of BBIM_Batting property set"
         op = row.operator("bim.add_annotation", text="Fill Area", icon="NODE_TEXTURE")
         op.object_type = "FILL_AREA"
 
diff --git a/src/blenderbim/blenderbim/bim/module/pset/operator.py b/src/blenderbim/blenderbim/bim/module/pset/operator.py
index bb60f9273c..73be56bc98 100644
--- a/src/blenderbim/blenderbim/bim/module/pset/operator.py
+++ b/src/blenderbim/blenderbim/bim/module/pset/operator.py
@@ -298,6 +298,7 @@ class EditPset(bpy.types.Operator, Operator):
             bpy.ops.bim.load_cost_item_quantities()
 
         bpy.ops.bim.disable_pset_editing(obj=self.obj, obj_type=self.obj_type)
+        tool.Blender.update_viewport()
 
 
 class RemovePset(bpy.types.Operator, Operator):