def componentsintobranch(idf, branch, listofcomponents, fluid=None):
"""insert a list of components into a branch
fluid is only needed if there are air and water nodes in same object
fluid is Air or Water or ''.
if the fluid is Steam, use Water"""
if fluid is None:
fluid = ""
componentlist = [item[0] for item in listofcomponents]
# assumes that the nodes of the component connect to each other
# empty branch if it has existing components
thebranchname = branch.Name
thebranch = idf.removeextensibles("BRANCH",
thebranchname) # empty the branch
# fill in the new components with the node names into this branch
# find the first extensible field and fill in the data in obj.
e_index = idf.getextensibleindex("BRANCH", thebranchname)
theobj = thebranch.obj
modeleditor.extendlist(theobj, e_index) # just being careful here
for comp, compnode in listofcomponents:
theobj.append(comp.key)
theobj.append(comp.Name)
inletnodename = getnodefieldname(comp,
"Inlet_Node_Name",
fluid=fluid,
startswith=compnode)
theobj.append(comp[inletnodename])
outletnodename = getnodefieldname(comp,
"Outlet_Node_Name",
fluid=fluid,
startswith=compnode)
theobj.append(comp[outletnodename])
theobj.append("")
return thebranch
def componentsintobranch(idf, branch, listofcomponents, fluid=None):
"""insert a list of components into a branch
fluid is only needed if there are air and water nodes in same object
fluid is Air or Water or ''.
if the fluid is Steam, use Water"""
if fluid is None:
fluid = ''
componentlist = [item[0] for item in listofcomponents]
# assumes that the nodes of the component connect to each other
# empty branch if it has existing components
thebranchname = branch.Name
thebranch = idf.removeextensibles('BRANCH', thebranchname) # empty the branch
# fill in the new components with the node names into this branch
# find the first extensible field and fill in the data in obj.
e_index = idf.getextensibleindex('BRANCH', thebranchname)
theobj = thebranch.obj
modeleditor.extendlist(theobj, e_index) # just being careful here
for comp, compnode in listofcomponents:
theobj.append(comp.key)
theobj.append(comp.Name)
inletnodename = getnodefieldname(comp, "Inlet_Node_Name", fluid=fluid,
startswith=compnode)
theobj.append(comp[inletnodename])
outletnodename = getnodefieldname(comp, "Outlet_Node_Name",
fluid=fluid, startswith=compnode)
theobj.append(comp[outletnodename])
theobj.append('')
return thebranch
def test_extendlist():
"""py.test for extendlist"""
data = (([1,2,3], 2, 0, [1,2,3]), # lst, i, value, nlst
([1,2,3], 3, 0, [1,2,3,0]), # lst, i, value, nlst
([1,2,3], 5, 0, [1,2,3,0,0,0]), # lst, i, value, nlst
([1,2,3], 7, 0, [1,2,3,0,0,0,0,0]), # lst, i, value, nlst
)
for lst, i, value, nlst in data:
modeleditor.extendlist(lst, i, value=value)
assert lst == nlst
def test_extendlist():
"""py.test for extendlist"""
tdata = (
([1, 2, 3], 2, 0, [1, 2, 3]), # lst, i, value, nlst
([1, 2, 3], 3, 0, [1, 2, 3, 0]), # lst, i, value, nlst
([1, 2, 3], 5, 0, [1, 2, 3, 0, 0, 0]), # lst, i, value, nlst
([1, 2, 3], 7, 0, [1, 2, 3, 0, 0, 0, 0, 0]), # lst, i, value, nlst
)
for lst, i, value, nlst in tdata:
modeleditor.extendlist(lst, i, value=value)
assert lst == nlst
