mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-09 17:31:45 +00:00
You can now duplicate tasks
This commit is contained in:
Sigma Dimensions
@@ -0,0 +1,142 @@
|
||||
# IfcOpenShell - IFC toolkit and geometry engine
|
||||
# Copyright (C) 2021-2022 Dion Moult <dion@thinkmoult.com>, Yassine Oualid <yassine@sigmadimensions.com>
|
||||
#
|
||||
# This file is part of IfcOpenShell.
|
||||
#
|
||||
# IfcOpenShell is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Lesser General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# IfcOpenShell is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Lesser General Public License
|
||||
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import ifcopenshell
|
||||
import ifcopenshell.util.system
|
||||
import ifcopenshell.util.element
|
||||
|
||||
|
||||
class Usecase:
|
||||
def __init__(self, file, task=None):
|
||||
"""Duplicates a task in the project
|
||||
|
||||
The following relationships are also duplicated:
|
||||
|
||||
* The copy will have the same attributes and property sets as the original task
|
||||
* The copy will be assigned to the parent task or work schedule
|
||||
* The copy will have duplicated nested tasks
|
||||
|
||||
:param task: The task to be duplicated
|
||||
:type task: ifcopenshell.entity_instance.entity_instance
|
||||
:return: The duplicated task
|
||||
:rtype: ifcopenshell.entity_instance.entity_instance
|
||||
|
||||
Example:
|
||||
.. code:: python
|
||||
|
||||
# We have a task
|
||||
original_task = Task(name="Design new feature", deadline="2023-03-01")
|
||||
|
||||
# And now we have two
|
||||
duplicated_task = project.duplicate_task(original_task)
|
||||
"""
|
||||
self.file = file
|
||||
self.settings = {"task": task}
|
||||
|
||||
def execute(self):
|
||||
result = ifcopenshell.util.element.copy(self.file, self.settings["task"])
|
||||
self.copy_indirect_attributes(self.settings["task"], result)
|
||||
return result
|
||||
|
||||
def copy_sequence_relationship(self, original_tasks, duplicated_tasks):
|
||||
for i, original_task in enumerate(original_tasks):
|
||||
for inverse in self.file.get_inverse(original_task):
|
||||
if inverse.is_a("IfcRelSequence") and (
|
||||
inverse.RelatingProcess == original_task
|
||||
or inverse.RelatedProcess == original_task
|
||||
):
|
||||
original_task_index = original_tasks.index(original_task)
|
||||
duplicated_task = duplicated_tasks[original_task_index]
|
||||
relating_process, related_process = None, None
|
||||
if inverse.RelatingProcess == original_task:
|
||||
relating_process = duplicated_task
|
||||
else:
|
||||
related_process = duplicated_task
|
||||
if inverse.RelatedProcess in original_tasks:
|
||||
related_process_index = original_tasks.index(
|
||||
inverse.RelatedProcess
|
||||
)
|
||||
related_process = duplicated_tasks[related_process_index]
|
||||
if inverse.RelatingProcess in original_tasks:
|
||||
relating_process_index = original_tasks.index(
|
||||
inverse.RelatingProcess
|
||||
)
|
||||
relating_process = duplicated_tasks[relating_process_index]
|
||||
if relating_process and related_process:
|
||||
rel = ifcopenshell.api.run(
|
||||
"sequence.assign_sequence",
|
||||
self.file,
|
||||
relating_process=relating_process,
|
||||
related_process=related_process,
|
||||
)
|
||||
if inverse.TimeLag:
|
||||
ifcopenshell.api.run(
|
||||
"sequence.assign_lag_time",
|
||||
self.file,
|
||||
sequence=rel,
|
||||
lag_time=inverse.TimeLag.LagValue,
|
||||
duration_type=inverse.TimeLag.DurationType,
|
||||
)
|
||||
|
||||
def copy_indirect_attributes(self, from_element, to_element):
|
||||
for inverse in self.file.get_inverse(from_element):
|
||||
if inverse.is_a("IfcRelDefinesByProperties"):
|
||||
inverse = ifcopenshell.util.element.copy(self.file, inverse)
|
||||
inverse.RelatedObjects = [to_element]
|
||||
pset = ifcopenshell.util.element.copy_deep(
|
||||
self.file, inverse.RelatingPropertyDefinition
|
||||
)
|
||||
inverse.RelatingPropertyDefinition = pset
|
||||
elif inverse.is_a("IfcRelNests") and inverse.RelatingObject == from_element:
|
||||
nested_tasks = [e for e in inverse.RelatedObjects if e.is_a("IfcTask")]
|
||||
if nested_tasks:
|
||||
new_tasks = [
|
||||
ifcopenshell.api.run(
|
||||
"sequence.duplicate_task", self.file, task=t
|
||||
)
|
||||
for t in nested_tasks
|
||||
]
|
||||
inverse = ifcopenshell.util.element.copy(self.file, inverse)
|
||||
inverse.RelatingObject = to_element
|
||||
inverse.RelatedObjects = new_tasks
|
||||
for task in new_tasks:
|
||||
ifcopenshell.api.run(
|
||||
"nest.unassign_object", self.file, related_object=task
|
||||
)
|
||||
rel = ifcopenshell.api.run(
|
||||
"nest.assign_object",
|
||||
self.file,
|
||||
related_object=task,
|
||||
relating_object=to_element,
|
||||
)
|
||||
self.copy_sequence_relationship(nested_tasks, new_tasks)
|
||||
|
||||
elif inverse.is_a("IfcRelSequence") and (
|
||||
inverse.RelatingProcess == from_element
|
||||
or inverse.RelatedProcess == from_element
|
||||
):
|
||||
continue
|
||||
else:
|
||||
for i, value in enumerate(inverse):
|
||||
if value == from_element:
|
||||
new_inverse = ifcopenshell.util.element.copy(self.file, inverse)
|
||||
new_inverse[i] = to_element
|
||||
elif isinstance(value, (tuple, list)) and from_element in value:
|
||||
new_value = list(value)
|
||||
new_value.append(to_element)
|
||||
inverse[i] = new_value
|
||||
Reference in New Issue
Block a user