From d6b4c1f4547d43078c2ae034c5e9ff21c38bdc0b Mon Sep 17 00:00:00 2001
From: Andrej730 <azhilenkov@gmail.com>
Date: Fri, 22 Sep 2023 12:16:36 +0500
Subject: [PATCH] Setup arrays constraints on loading .ifc #3753

---
 src/blenderbim/blenderbim/bim/import_ifc.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/blenderbim/blenderbim/bim/import_ifc.py b/src/blenderbim/blenderbim/bim/import_ifc.py
index ecfff3bcf0..db041a46d6 100644
--- a/src/blenderbim/blenderbim/bim/import_ifc.py
+++ b/src/blenderbim/blenderbim/bim/import_ifc.py
@@ -20,6 +20,7 @@ import os
 import re
 import bpy
 import time
+import json
 import bmesh
 import logging
 import mathutils
@@ -306,6 +307,7 @@ class IfcImporter:
         self.set_default_context()
         self.profile_code("Setting default context")
         self.setup_viewport_camera()
+        self.setup_arrays()
         self.update_progress(100)
         bpy.context.window_manager.progress_end()
 
@@ -1908,6 +1910,15 @@ class IfcImporter:
             bpy.ops.view3d.view_selected()
             bpy.ops.object.select_all(action="DESELECT")
 
+    def setup_arrays(self):
+        for element in self.file.by_type("IfcElement"):
+            pset_data = ifcopenshell.util.element.get_pset(element, "BBIM_Array")
+            if not pset_data["Data"]:  # skip array children
+                continue
+            for i in range(len(json.loads(pset_data["Data"]))):
+                tool.Blender.Modifier.Array.set_children_lock_state(element, i, True)
+                tool.Blender.Modifier.Array.constrain_children_to_parent(element)
+
 
 class IfcImportSettings:
     def __init__(self):