Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-26 10:11:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix MSProject calendar day mapping at import, and add relationship type display in blenderBIM

Browse Source
This commit is contained in:
bosonprojets
2021-05-06 06:08:21 +00:00
parent 114214e2a1
commit a10f4c070a
1 changed files with 44 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcp6/ifcp6/msp2ifc.py
+44 -25
View File
@@ -19,13 +19,13 @@ class MSP2Ifc:
self.tasks = {} self.tasks = {}
self.relationships = {} self.relationships = {}
self.day_map = { self.day_map = {
"2": 1, "1": 1,
"3": 2, "2": 2,
"4": 3, "3": 3,
"5": 4, "4": 4,
"6": 5, "5": 5,
"7": 6, "6": 6,
"1": 7, "7": 7,
} }
def execute(self): def execute(self):
@@ -43,12 +43,27 @@ class MSP2Ifc:
self.parse_calendar_xml(project) self.parse_calendar_xml(project)
def parse_relationship_xml(self, task):
relationships = {}
if task.findall("pr:PredecessorLink", self.ns):
for relationship in task.findall("pr:PredecessorLink", self.ns):
relationships[task.find("pr:UID", self.ns).text] = {
"PredecessorTask": relationship.find("pr:PredecessorUID", self.ns).text,
"Type": relationship.find("pr:Type", self.ns).text
}
return relationships
def parse_task_xml(self, project): def parse_task_xml(self, project):
for task in project.find("pr:Tasks", self.ns): for task in project.find("pr:Tasks", self.ns):
task_id = task.find("pr:UID", self.ns).text task_id = task.find("pr:UID", self.ns).text
task_index_level = task.find("pr:OutlineLevel", self.ns).text task_index_level = task.find("pr:OutlineLevel", self.ns).text
wbs_id = task.find("pr:WBS", self.ns).text wbs_id = task.find("pr:WBS", self.ns).text
relationship = task.find("pr:PredecessorLink", self.ns) relationships = self.parse_relationship_xml(task)
# for relationship in task.findall("pr:PredecessorLink", self.ns):
# predecessor_task = relationship.find("pr:PredecessorUID", self.ns).text
# type = relationship.find("pr:Type", self.ns).text
outline_level = int(task.find("pr:OutlineLevel", self.ns).text) outline_level = int(task.find("pr:OutlineLevel", self.ns).text)
if outline_level != 0: if outline_level != 0:
@@ -66,7 +81,7 @@ class MSP2Ifc:
"Duration": ifcopenshell.util.date.ifc2datetime(task.find("pr:Duration", self.ns).text), "Duration": ifcopenshell.util.date.ifc2datetime(task.find("pr:Duration", self.ns).text),
"Priority": task.find("pr:Priority", self.ns).text, "Priority": task.find("pr:Priority", self.ns).text,
"CalendarUID": task.find("pr:CalendarUID", self.ns).text, "CalendarUID": task.find("pr:CalendarUID", self.ns).text,
"PredecessorTask": relationship.find("pr:PredecessorUID", self.ns).text if relationship else None, "PredecessorTasks": relationships if relationships else None,
"subtasks": [], "subtasks": [],
"ifc": None, "ifc": None,
} }
@@ -205,22 +220,26 @@ class MSP2Ifc:
def create_rel_sequences(self): def create_rel_sequences(self):
self.sequence_type_map = { self.sequence_type_map = {
"Start to Start": "START_START", "1": "START_START",
"Start to Finish": "START_FINISH", "2": "START_FINISH",
"Finish to Start": "FINISH_START", "3": "FINISH_START",
"Finish to Finish": "FINISH_FINISH", "4": "FINISH_FINISH",
"0": "NOTDEFINED",
} }
for task in self.tasks.values(): for task in self.tasks.values():
if not task["PredecessorTask"]: if not task["PredecessorTasks"]:
continue continue
rel_sequence = ifcopenshell.api.run( for predecessor in task["PredecessorTasks"].values():
"sequence.assign_sequence", rel_sequence = ifcopenshell.api.run(
self.file, "sequence.assign_sequence",
related_process = task["ifc"], self.file,
relating_process = self.tasks[task["PredecessorTask"]]["ifc"] related_process = task["ifc"],
) relating_process = self.tasks[predecessor["PredecessorTask"]]["ifc"]
ifcopenshell.api.run( )
"sequence.edit_sequence", if predecessor["Type"]:
self.file, ifcopenshell.api.run(
rel_sequence=rel_sequence, "sequence.edit_sequence",
) self.file,
rel_sequence=rel_sequence,
attributes={"SequenceType": self.sequence_type_map[predecessor["Type"]]}
)