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

Support root-level activities when importing from P6 XML

Browse Source
This commit is contained in:
Dion Moult
2021-04-30 12:04:15 +10:00
parent a936a5317d
commit cdc9180111
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcp6/ifcp6/p62ifc.py
+10 -6
View File
@@ -12,6 +12,7 @@ class P62Ifc:
self.work_plan = None self.work_plan = None
self.project = {} self.project = {}
self.wbs = {} self.wbs = {}
self.root_activites = []
self.activities = {} self.activities = {}
self.relationships = {} self.relationships = {}
@@ -39,10 +40,10 @@ class P62Ifc:
for activity in project.findall("pr:Activity", ns): for activity in project.findall("pr:Activity", ns):
activity_id = activity.find("pr:ObjectId", ns).text activity_id = activity.find("pr:ObjectId", ns).text
wbs_id = activity.find("pr:WBSObjectId", ns).text wbs_id = activity.find("pr:WBSObjectId", ns).text
if not wbs_id: if wbs_id:
print("No WBS ID found for activity", activity_id) self.wbs[wbs_id]["activities"].append(activity_id)
continue else:
self.wbs[wbs_id]["activities"].append(activity_id) self.root_activites.append(activity_id)
self.activities[activity_id] = { self.activities[activity_id] = {
"Name": activity.find("pr:Name", ns).text, "Name": activity.find("pr:Name", ns).text,
"Identification": activity.find("pr:Id", ns).text, "Identification": activity.find("pr:Id", ns).text,
@@ -76,6 +77,8 @@ class P62Ifc:
def create_tasks(self, work_schedule): def create_tasks(self, work_schedule):
for wbs in self.wbs.values(): for wbs in self.wbs.values():
self.create_task_from_wbs(wbs, work_schedule) self.create_task_from_wbs(wbs, work_schedule)
for activity_id in self.root_activites:
self.create_task_from_activity(self.activities[activity_id], None, work_schedule)
def create_task_from_wbs(self, wbs, work_schedule): def create_task_from_wbs(self, wbs, work_schedule):
wbs["ifc"] = ifcopenshell.api.run( wbs["ifc"] = ifcopenshell.api.run(
@@ -91,13 +94,14 @@ class P62Ifc:
attributes={"Name": wbs["Name"], "Identification": wbs["Code"]}, attributes={"Name": wbs["Name"], "Identification": wbs["Code"]},
) )
for activity_id in wbs["activities"]: for activity_id in wbs["activities"]:
self.create_task_from_activity(self.activities[activity_id], wbs, work_schedule) self.create_task_from_activity(self.activities[activity_id], wbs, None)
def create_task_from_activity(self, activity, wbs, work_schedule): def create_task_from_activity(self, activity, wbs, work_schedule):
activity["ifc"] = ifcopenshell.api.run( activity["ifc"] = ifcopenshell.api.run(
"sequence.add_task", "sequence.add_task",
self.file, self.file,
parent_task=wbs["ifc"], work_schedule=None if wbs else work_schedule,
parent_task=wbs["ifc"] if wbs else None,
) )
ifcopenshell.api.run( ifcopenshell.api.run(
"sequence.edit_task", "sequence.edit_task",