Arch: Allow to export structural analysis model to IFC
This commit is contained in:
Yorik van Havre
180
src/Mod/Arch/exportIFCStructuralTools.py
Normal file
180
src/Mod/Arch/exportIFCStructuralTools.py
Normal file
@@ -0,0 +1,180 @@
|
||||
# ***************************************************************************
|
||||
# * Copyright (c) 2020 Yorik van Havre <yorik@uncreated.net> *
|
||||
# * *
|
||||
# * This program is free software; you can redistribute it and/or modify *
|
||||
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
||||
# * as published by the Free Software Foundation; either version 2 of *
|
||||
# * the License, or (at your option) any later version. *
|
||||
# * for detail see the LICENCE text file. *
|
||||
# * *
|
||||
# * This program 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 Library General Public License for more details. *
|
||||
# * *
|
||||
# * You should have received a copy of the GNU Library General Public *
|
||||
# * License along with this program; if not, write to the Free Software *
|
||||
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
||||
# * USA *
|
||||
# * *
|
||||
# ***************************************************************************
|
||||
|
||||
from __future__ import print_function
|
||||
|
||||
|
||||
__title__ = "FreeCAD structural IFC export tools"
|
||||
__author__ = "Yorik van Havre"
|
||||
__url__ = "https://www.freecadweb.org"
|
||||
|
||||
ALLOW_LINEAR_OBJECTS = True # allow non-solid objects (wires, etc) to become analytic objects?
|
||||
|
||||
structural_nodes = {} # this keeps track of nodes during this session
|
||||
scaling = 1.0 # this keeps track of scaling during this session
|
||||
|
||||
|
||||
def setup(ifcfile,ifcbin,scale):
|
||||
|
||||
"""Creates all the needed setup for structural model."""
|
||||
|
||||
global structural_nodes,scaling
|
||||
structural_nodes = {}
|
||||
scaling = scale
|
||||
import ifcopenshell
|
||||
uid = ifcopenshell.guid.new
|
||||
owh = ifcfile.by_type("IfcOwnerHistory")[0]
|
||||
prj = ifcfile.by_type("IfcProject")[0]
|
||||
ctx = createStructuralContext(ifcfile)
|
||||
if ifcfile.wrapped_data.schema_name == "IFC2X3":
|
||||
mod = ifcfile.createIfcStructuralAnalysisModel(uid(),owh,"Structural Analysis Model",None,None,"NOTDEFINED",None,None,None)
|
||||
else:
|
||||
pla = ifcbin.createIfcLocalPlacement()
|
||||
mod = ifcfile.createIfcStructuralAnalysisModel(uid(),owh,"Structural Analysis Model",None,None,"NOTDEFINED",None,None,None,pla)
|
||||
rel = ifcfile.createIfcRelDeclares(uid(),owh,None,None,prj,[mod])
|
||||
|
||||
|
||||
def createStructuralContext(ifcfile):
|
||||
|
||||
"""Creates an additional geometry context for structural objects. Returns the new context"""
|
||||
|
||||
contexts = ifcfile.by_type("IfcGeometricRepresentationContext")
|
||||
# filter out subcontexts
|
||||
contexts = [c for c in contexts if c.is_a() == "IfcGeometricRepresentationContext"]
|
||||
ctx = contexts[0] # arbitrarily take the first one...
|
||||
structcontext = ifcfile.createIfcGeometricRepresentationSubContext('Analysis','Axis',None,None,None,None,ctx,None,"GRAPH_VIEW",None)
|
||||
return structcontext
|
||||
|
||||
|
||||
def getStructuralContext(ifcfile):
|
||||
|
||||
"""Returns the structural context from the file"""
|
||||
for c in ifcfile.by_type("IfcGeometricRepresentationSubContext"):
|
||||
if c.ContextIdentifier == "Analysis":
|
||||
return c
|
||||
|
||||
|
||||
def createStructuralNode(ifcfile,ifcbin,point):
|
||||
|
||||
"""Creates a connection node at the given point"""
|
||||
|
||||
import ifcopenshell
|
||||
uid = ifcopenshell.guid.new
|
||||
owh = ifcfile.by_type("IfcOwnerHistory")[0]
|
||||
ctx = getStructuralContext(ifcfile)
|
||||
cpt = ifcbin.createIfcCartesianPoint(tuple(point))
|
||||
vtx = ifcfile.createIfcVertexPoint(cpt)
|
||||
rep = ifcfile.createIfcTopologyRepresentation(ctx,'Analysis','Vertex',[vtx])
|
||||
psh = ifcfile.createIfcProductDefinitionShape(None,None,[rep])
|
||||
# boundary conditions serve for ex. to create fixed nodes
|
||||
#cnd = ifcfile.createIfcBoundaryNodeCondition("Fixed",ifcfile.createIfcBoolean(True),ifcfile.createIfcBoolean(True),ifcfile.createIfcBoolean(True),ifcfile.createIfcBoolean(True),ifcfile.createIfcBoolean(True),ifcfile.createIfcBoolean(True))
|
||||
# for now we don't create any boundary condition
|
||||
cnd = None
|
||||
pla = ifcbin.createIfcLocalPlacement()
|
||||
prd = ifcfile.createIfcStructuralPointConnection(uid(),owh,'Vertex',None,None,pla,psh,cnd,None)
|
||||
return prd
|
||||
|
||||
|
||||
def createStructuralMember(ifcfile,ifcbin,obj):
|
||||
|
||||
"""Creates a structural member if possible. Returns the member"""
|
||||
|
||||
global structural_nodes
|
||||
prd = None
|
||||
import Draft
|
||||
import Part
|
||||
import ifcopenshell
|
||||
uid = ifcopenshell.guid.new
|
||||
owh = ifcfile.by_type("IfcOwnerHistory")[0]
|
||||
ctx = getStructuralContext(ifcfile)
|
||||
edges = None
|
||||
if Draft.getType(obj) not in ["Structure"]:
|
||||
if ALLOW_LINEAR_OBJECTS and obj.isDerivedFrom("Part::Feature"):
|
||||
if obj.Shape.Faces:
|
||||
return None
|
||||
elif not obj.Shape.Edges:
|
||||
return None
|
||||
else:
|
||||
edges = obj.Shape.Edges
|
||||
else:
|
||||
wire = Part.makePolygon([obj.Placement.multVec(n) for n in obj.Nodes])
|
||||
edges = wire.Edges
|
||||
if not edges:
|
||||
return None
|
||||
for edge in edges:
|
||||
if len(edge.Vertexes) > 1:
|
||||
# we don't care about curved edges just now...
|
||||
v0 = edge.Vertexes[0].Point.multiply(scaling)
|
||||
v1 = edge.Vertexes[-1].Point.multiply(scaling)
|
||||
cp1 = ifcbin.createIfcCartesianPoint(tuple(v0))
|
||||
cp2 = ifcbin.createIfcCartesianPoint(tuple(v1))
|
||||
upv = ifcbin.createIfcDirection((0,0,1))
|
||||
pla = ifcbin.createIfcLocalPlacement()
|
||||
vp1 = ifcfile.createIfcVertexPoint(cp1)
|
||||
vp2 = ifcfile.createIfcVertexPoint(cp2)
|
||||
edg = ifcfile.createIfcEdge(vp1,vp2)
|
||||
rep = ifcfile.createIfcTopologyRepresentation(ctx,'Analysis','Edge',[edg])
|
||||
psh = ifcfile.createIfcProductDefinitionShape(None,None,[rep])
|
||||
prd = ifcfile.createIfcStructuralCurveMember(uid(),owh,obj.Label,None,None,pla,psh,"RIGID_JOINED_MEMBER",upv)
|
||||
# check for existing connection nodes
|
||||
for v in [v0,v1]:
|
||||
vk = tuple(v)
|
||||
if vk in structural_nodes:
|
||||
if structural_nodes[vk]:
|
||||
n = structural_nodes[vk]
|
||||
else:
|
||||
# there is another member with same point, create a new node
|
||||
n = createStructuralNode(ifcfile,ifcbin,v)
|
||||
structural_nodes[vk] = n
|
||||
ifcfile.createIfcRelConnectsStructuralMember(uid(),None,None,None,prd,n,None,None,None,None);
|
||||
else:
|
||||
# just add the point, no other member using it yet
|
||||
structural_nodes[vk] = None
|
||||
return prd
|
||||
|
||||
|
||||
def createStructuralGroup(ifcfile):
|
||||
|
||||
"Assigns all structural objects found in the file to the structual model"""
|
||||
|
||||
import ifcopenshell
|
||||
uid = ifcopenshell.guid.new
|
||||
owh = ifcfile.by_type("IfcOwnerHistory")[0]
|
||||
edges = ifcfile.by_type("IfcStructuralCurveMember")
|
||||
verts = ifcfile.by_type("IfcStructuralPointConnection")
|
||||
model = ifcfile.by_type("IfcStructuralAnalysisModel")[0]
|
||||
if model:
|
||||
members = edges + verts
|
||||
if members:
|
||||
ifcfile.createIfcRelAssignsToGroup(uid(),owh,None,None,members,"PRODUCT",model)
|
||||
|
||||
|
||||
def associates(ifcfile,aobj,sobj):
|
||||
|
||||
"""Associates an arch object with a struct object"""
|
||||
|
||||
# This is probably not the right way to do this, ie. relate a structural
|
||||
# object with an IfcProduct. Needs to investigate more....
|
||||
|
||||
import ifcopenshell
|
||||
uid = ifcopenshell.guid.new
|
||||
owh = ifcfile.by_type("IfcOwnerHistory")[0]
|
||||
ifcfile.createIfcRelAssignsToProduct(uid(),owh,None,None,[sobj],None,aobj)
|
||||
Reference in New Issue
Block a user