Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-21 09:05:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

calculate resource usage/work - add poll messages

Browse Source
This commit is contained in:
Andrej730
2024-11-14 14:12:28 +05:00
parent 6a20ecb2a0
commit a8d8600087
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/bonsai/bonsai/bim/module/resource/operator.py
+15 -9
View File
@@ -189,9 +189,13 @@ class CalculateResourceWork(bpy.types.Operator, tool.Ifc.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
active_resource = tool.Resource.get_highlighted_resource() active_resource = tool.Resource.get_highlighted_resource()
if active_resource: if not active_resource:
if tool.Resource.get_productivity(active_resource, should_inherit=True): cls.poll_message_set("No resource is active.")
return True return False
if not tool.Resource.get_productivity(active_resource, should_inherit=True):
cls.poll_message_set("No productivity data for active resource.")
return False
return True
def _execute(self, context): def _execute(self, context):
core.calculate_resource_work(tool.Ifc, tool.Resource, resource=tool.Ifc.get().by_id(self.resource)) core.calculate_resource_work(tool.Ifc, tool.Resource, resource=tool.Ifc.get().by_id(self.resource))
@@ -415,12 +419,14 @@ class CalculateResourceUsage(bpy.types.Operator, tool.Ifc.Operator):
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
active_resource = tool.Resource.get_highlighted_resource() active_resource = tool.Resource.get_highlighted_resource()
if active_resource: if not active_resource:
if active_resource.Usage and active_resource.Usage.ScheduleWork: cls.poll_message_set("No resource is active.")
task = tool.Resource.get_task_assignments(active_resource) return False
if task and tool.Sequence.has_duration(task): if active_resource.Usage and active_resource.Usage.ScheduleWork:
return True task = tool.Resource.get_task_assignments(active_resource)
if task and tool.Sequence.has_duration(task):
return True
cls.poll_message_set("No usage data for active resource.")
return False return False
def _execute(self, context): def _execute(self, context):