Library-Dynamo-Python-CSharp/SetingInputToRunType2.py at master · nguyenngocdue/Library-Dynamo-Python-CSharp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
"""Copyright(c) 2023 by: duengocnguyen@gmail.com"""
'https://www.youtube.com/channel/UCt2JhCDDFxpYho575WTMZ4g'
"""________________Welcome to BIM3DM-DYNAMO API___________________"""
import clr
import sys
sys.path.append(r'C:\Program Files\Autodesk\Revit 2020\AddIns\DynamoForRevit\IronPython.StdLib.2.7.8')
import math
from System.Collections.Generic import *

clr.AddReference("ProtoGeometry")
from Autodesk.DesignScript.Geometry import *

clr.AddReference('RevitAPI')
import Autodesk
from Autodesk.Revit.DB import *
from Autodesk.Revit.DB.Structure import *

clr.AddReference('RevitAPIUI')
from Autodesk.Revit.UI import*
from  Autodesk.Revit.UI.Selection import*

clr.AddReference("RevitNodes")
import Revit
clr.ImportExtensions(Revit.Elements)
clr.ImportExtensions(Revit.GeometryConversion)

clr.AddReference("RevitServices")
import RevitServices
from RevitServices.Persistence import DocumentManager
from RevitServices.Transactions import TransactionManager
from Autodesk.DesignScript.Geometry import Point


path_to_check = r'C:\Users\NGUYEN NGOC DUE\AppData\Roaming\Dynamo\Dynamo Revit\2.10\packages\packages\BIM3DM\lib'
import sys
sys.path.append(path_to_check)
import BIM3DM
#########################################################################
def is_array(obj):
    return "List" in obj.__class__.__name__ or "list" in obj.__class__.__name__

def get_array_rank(array):
    if is_array(array):
        return 1 + max(get_array_rank(item) for item in array)
    else:
        return 0
def flatten_to_1d(arr):
    result = []
    def recursive_flatten(subarray):
        for item in subarray:
            if is_array(item):
                recursive_flatten(item)
            else:
                result.append(item)

    recursive_flatten(arr)
    return result
def returnOUT(fn, objects):
    rank = get_array_rank(objects)
    if rank == 0:  # a
        return fn(objects)
    elif rank == 1:  # [a]
        return [fn(*args) for args in objects]
    elif rank == 2:  # [[a]]
        return [fn(*i) for i in objects for i in j]
    else:
        elements = flatten_to_1d(objects)  # [a]
        return fn(elements)


#########################################################################
doc = DocumentManager.Instance.CurrentDBDocument
View = doc.ActiveView
uidoc = DocumentManager.Instance.CurrentUIApplication.ActiveUIDocument
app  = uidoc.Application
DB = Autodesk.Revit.DB
############################### FUNCTION #################################
def setLayoutAsMinimumClearSpacingForRebar(rebar, spacing, arrayLength, barsOnNormalSide=True, inclFirstBar=True, inclLastBar=True):
    TransactionManager.Instance.EnsureInTransaction(doc)
    rebar = UnwrapElement(rebar)
    re = rebar.GetShapeDrivenAccessor()
    re.SetLayoutAsMinimumClearSpacing(spacing,arrayLength,barsOnNormalSide,inclFirstBar,inclLastBar)
    TransactionManager.Instance.TransactionTaskDone()
    return rebar

############################### INPUT #################################
objects = UnwrapElement(IN[0])
spacing = IN[1]
arrayLength = IN[2]
barsOnNormalSide = IN[3]
inclFirstBar = IN[4]
inclLastBar = IN[5]
# Convert from mm to feet:
if IN[6]:
    spacing = spacing / 304.8
    arrayLength = arrayLength / 304.8

allVariable = [objects, spacing, arrayLength,barsOnNormalSide, inclFirstBar, inclLastBar]


############################### OUTPUT ##################################
rank = get_array_rank(objects)
if rank == 0:  # a
    OUT = setLayoutAsMinimumClearSpacingForRebar(*allVariable)
elif rank == 1:  # [a]
    OUT = [setLayoutAsMinimumClearSpacingForRebar(element, *allVariable) for element in objects]
elif rank == 2:  # [[a]]
    OUT = [setLayoutAsMinimumClearSpacingForRebar(i, *allVariable) for j in objects for i in j]
else:
    elements = flatten_to_1d(objects)  # [a]
    OUT = setLayoutAsMinimumClearSpacingForRebar(elements,*allVariable )