Support importing predecessor and successor relationships from P6

This commit is contained in:
Dion Moult
2021-04-27 12:30:09 +10:00
parent e0a5c9c8f3
commit 9164a7bcc3
+30 -13
View File
@@ -12,7 +12,8 @@ class P62Ifc:
self.work_plan = None self.work_plan = None
self.project = {} self.project = {}
self.wbs = {} self.wbs = {}
self.activity = {} self.activities = {}
self.relationships = {}
def execute(self): def execute(self):
self.parse_xml() self.parse_xml()
@@ -36,16 +37,22 @@ class P62Ifc:
} }
for activity in project.findall("pr:Activity", ns): for activity in project.findall("pr:Activity", ns):
self.wbs[activity.find("pr:WBSObjectId", ns).text]["activities"].append( activity_id = activity.find("pr:ObjectId", ns).text
{ self.wbs[activity.find("pr:WBSObjectId", ns).text]["activities"].append(activity_id)
"Name": activity.find("pr:Name", ns).text, self.activities[activity_id] = {
"Identification": activity.find("pr:Id", ns).text, "Name": activity.find("pr:Name", ns).text,
"StartDate": datetime.fromisoformat(activity.find("pr:StartDate", ns).text), "Identification": activity.find("pr:Id", ns).text,
"FinishDate": datetime.fromisoformat(activity.find("pr:FinishDate", ns).text), "StartDate": datetime.fromisoformat(activity.find("pr:StartDate", ns).text),
"Status": activity.find("pr:Status", ns).text, "FinishDate": datetime.fromisoformat(activity.find("pr:FinishDate", ns).text),
"ifc": None, "Status": activity.find("pr:Status", ns).text,
} "ifc": None,
) }
for relationship in project.findall("pr:Relationship", ns):
self.relationships[relationship.find("pr:ObjectId", ns).text] = {
"PredecessorActivity": relationship.find("pr:PredecessorActivityObjectId", ns).text,
"SuccessorActivity": relationship.find("pr:SuccessorActivityObjectId", ns).text,
}
def get_wbs(self, wbs): def get_wbs(self, wbs):
return {"Name": wbs.find("pr:Name", ns).text, "subtasks": []} return {"Name": wbs.find("pr:Name", ns).text, "subtasks": []}
@@ -55,6 +62,7 @@ class P62Ifc:
self.file = self.create_boilerplate_ifc() self.file = self.create_boilerplate_ifc()
work_schedule = self.create_work_schedule() work_schedule = self.create_work_schedule()
self.create_tasks(work_schedule) self.create_tasks(work_schedule)
self.create_rel_sequences()
def create_work_schedule(self): def create_work_schedule(self):
return ifcopenshell.api.run( return ifcopenshell.api.run(
@@ -78,8 +86,8 @@ class P62Ifc:
task=wbs["ifc"], task=wbs["ifc"],
attributes={"Name": wbs["Name"], "Identification": wbs["Code"]}, attributes={"Name": wbs["Name"], "Identification": wbs["Code"]},
) )
for activity in wbs["activities"]: for activity_id in wbs["activities"]:
self.create_task_from_activity(activity, wbs, work_schedule) self.create_task_from_activity(self.activities[activity_id], wbs, work_schedule)
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(
@@ -109,6 +117,15 @@ class P62Ifc:
}, },
) )
def create_rel_sequences(self):
for relationship in self.relationships.values():
ifcopenshell.api.run(
"sequence.assign_sequence",
self.file,
relating_process=self.activities[relationship["PredecessorActivity"]]["ifc"],
related_process=self.activities[relationship["SuccessorActivity"]]["ifc"],
)
def create_boilerplate_ifc(self): def create_boilerplate_ifc(self):
self.file = ifcopenshell.file(schema="IFC4") self.file = ifcopenshell.file(schema="IFC4")
self.work_plan = self.file.create_entity("IfcWorkPlan") self.work_plan = self.file.create_entity("IfcWorkPlan")