2016-02-19 77 views
0

通過Dynamo論壇上的此主題:http://dynamobim.org/forums/topic/how-to-handle-variable-list-lengthnesting-in-custom-nodes-python/,我想找到一種方法來處理自定義節點的變量輸入。Revit API和Dynamo,處理變量輸入類型

下面是一個示例腳本,改變家族式的元素:

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

# Import ToDSType(bool) extension method 
clr.AddReference("RevitNodes") 
import Revit 
clr.ImportExtensions(Revit.Elements) 

# Import DocumentManager and TransactionManager 
clr.AddReference("RevitServices") 
import RevitServices 
from RevitServices.Persistence import DocumentManager 
from RevitServices.Transactions import TransactionManager 

from System.Collections.Generic import * 

# Import RevitAPI 
clr.AddReference("RevitAPI") 
import Autodesk 

doc = DocumentManager.Instance.CurrentDBDocument 
uiapp = DocumentManager.Instance.CurrentUIApplication 
app = uiapp.Application 


doc = DocumentManager.Instance.CurrentDBDocument 
app = DocumentManager.Instance.CurrentUIApplication.Application 


inputItems = [] 

for i in IN[0]: 
    inputItems.append(UnwrapElement(i)) 

toggle = IN[1] 
familyType = UnwrapElement(IN[2]).Id 
elementlist = [] 


TransactionManager.Instance.EnsureInTransaction(doc) 

if toggle == True: 
    for item in inputItems: 
     item.ChangeTypeId(familyType) 
     elementlist.append(item) 

else: 
    None 

TransactionManager.Instance.TransactionTaskDone() 

OUT = elementlist 

我想找到一種方法,我可以處理的輸入的列表,嵌套列表,或單項目(非列表)像下面的圖片:

enter image description here

後兩者,我可以只使用一個list.flatten。問題是找到一種方法來處理所有三種類型的列表/元素。

想法?

回答

1

在這種情況下,如果你的目標是使用Python來處理所有不同的輸入結構,我建議是這樣的:

,如果你只是想在項目上執行操作(比方說加+1),而不管數據結構的,我使用遞歸和簡單isinstance (IN[0], list)檢查像這樣:

#The inputs to this node will be stored as a list in the IN variable. 
dataEnteringNode = IN 

def ProcessList(_func, _list): 
    return map(lambda x: ProcessList(_func, x) if type(x)==list else _func(x), _list) 

def AddOne(item): 
    return item + 1 

if isinstance (IN[0], list): 
    output = ProcessList(AddOne, IN[0]) 
else: 
    output = AddOne(IN[0]) 

#Assign your output to the OUT variable. 
OUT = output 

下面是在發電機上的數據的不同的結構操作的一個例子:

enter image description here

這種方法非常好,因爲它可以輸出與輸入相同的結構。我使用了很多。

+0

This Works。雖然對於我所做的,定義輸入同樣好(var [])。我認爲允許輸出與輸入相同是一個很好的訣竅。 – stdmn

+0

當然,我明白,但問題是關於在這種情況下使用Python來處理不同的輸入結構。我相信我的回答是正確的,但是還有其他的可能性。他們在迪納摩論壇的討論中長大。我專注於用Python解決它。 – konrad