Created
March 30, 2015 01:28
-
-
Save ksobon/86178ccef757ed1f58b6 to your computer and use it in GitHub Desktop.
correct framing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #Copyright(c) 2015, Konrad Sobon | |
| # @arch_laboratory, http://archi-lab.net | |
| import clr | |
| clr.AddReference('ProtoGeometry') | |
| from Autodesk.DesignScript.Geometry import * | |
| # Import RevitAPI | |
| clr.AddReference("RevitAPI") | |
| import Autodesk | |
| from Autodesk.Revit.DB import * | |
| from Autodesk.Revit.DB.Structure import * | |
| # Import ToDSType(bool) extension method | |
| clr.AddReference("RevitNodes") | |
| import Revit | |
| clr.ImportExtensions(Revit.Elements) | |
| clr.ImportExtensions(Revit.GeometryConversion) | |
| # Import DocumentManager and TransactionManager | |
| clr.AddReference("RevitServices") | |
| import RevitServices | |
| from RevitServices.Persistence import DocumentManager | |
| from RevitServices.Transactions import TransactionManager | |
| doc = DocumentManager.Instance.CurrentDBDocument | |
| columns = UnwrapElement(IN[0]) | |
| framing = UnwrapElement(IN[1]) | |
| def ProcessList(_func, _list): | |
| return map( lambda x: process_list(_func, x) if type(x)==list else _func(x), _list ) | |
| def FindClosestPoint(pointCloud, testPoint): | |
| distanceList = [] | |
| for point in pointCloud: | |
| distanceList.append(point.DistanceTo(testPoint)) | |
| index = distanceList.index(min(distanceList)) | |
| closestPoint = pointCloud[index] | |
| return closestPoint | |
| def GetStructuralColumnCurve(item): | |
| if item.StructuralType == StructuralType.Column: | |
| modelColumn = item.GetAnalyticalModel() | |
| if modelColumn.IsSingleCurve(): | |
| return modelColumn.GetCurve().ToProtoType() | |
| else: | |
| return None | |
| columnCurves = ProcessList(GetStructuralColumnCurve, columns) | |
| columnPoints = [] | |
| for i in columnCurves: | |
| columnPoints.append(i.PointAtParameter(0)) | |
| columnPoints.append(i.PointAtParameter(1)) | |
| #"Start" the transaction | |
| TransactionManager.Instance.EnsureInTransaction(doc) | |
| for i in framing: | |
| if i.StructuralType == StructuralType.Brace: | |
| dsCurve = i.Location.Curve.ToProtoType() | |
| dsStartPoint = dsCurve.PointAtParameter(0) | |
| dsEndPoint = dsCurve.PointAtParameter(1) | |
| newStartPoint = FindClosestPoint(columnPoints, dsStartPoint) | |
| newEndPoint = FindClosestPoint(columnPoints, dsEndPoint) | |
| newCurve = Autodesk.DesignScript.Geometry.Line.ByStartPointEndPoint(newStartPoint, newEndPoint) | |
| newRvtCurve = newCurve.ToRevitType() | |
| type = i.Symbol | |
| bipName = BuiltInParameter.INSTANCE_REFERENCE_LEVEL_PARAM | |
| levelId = i.get_Parameter(bipName).AsElementId() | |
| level = doc.GetElement(levelId) | |
| doc.Create.NewFamilyInstance(newRvtCurve, type, level, StructuralType.Brace) | |
| doc.Delete(i.Id) | |
| else: | |
| dsCurve = i.Location.Curve.ToProtoType() | |
| dsStartPoint = dsCurve.PointAtParameter(0) | |
| dsEndPoint = dsCurve.PointAtParameter(1) | |
| newStartPoint = FindClosestPoint(columnPoints, dsStartPoint) | |
| newEndPoint = FindClosestPoint(columnPoints, dsEndPoint) | |
| newCurve = Autodesk.DesignScript.Geometry.Line.ByStartPointEndPoint(newStartPoint, newEndPoint) | |
| i.Location.Curve = newCurve.ToRevitType() | |
| # "End" the transaction | |
| TransactionManager.Instance.TransactionTaskDone() | |
| #Assign your output to the OUT variable | |
| OUT = 0 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment