IF Statement with Python Script or DesignScript Code Block

Developers
1

I’m running into a wall while trying to work with IF statements in an efficient way. I just simply don’t know how to appropriately specify the code in the correct syntax. Here’s what I have so far. Any pointers would be greatly helpful.

import clr
import math
clr.AddReference('ProtoGeometry')
from Autodesk.DesignScript.Geometry import *
#The inputs to this node will be stored as a list in the IN variables.
theaterWomen = IN[0]
theaterMen = IN[1]

theaterTotalLoad = []
theaterWomenBasins = []
theaterWomenToilets = []
womenBasins = []

menAndWomen = theaterWomen + theaterMen

def calcTheaterWomensBasins(theaterWomen):
	if theaterWomen == 0:
		womenBasins = 0
	elif theaterWomen <= 40:
		womenBasins = 1
	elif theaterWomen <= 70:
		womenBasins = 3
	elif theaterWomen <= 100:
		womenBasins = 4
    	else:
    		womenBasins = int(math.ceil(((theaterWomen-100)/40)+4))
    	return womenBasins

    theaterTotalLoad.append(menAndWomen)
    theaterWomenBasins.append(womenBasins)

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

The interesting thing is I don’t get any errors when the code runs, but I continue to get an empty list. My inputs as of now are simply two integers (I’m using 75 and 25 respectively).

2

Design Script…

womenBasins=(women==0)?0:(women<=40?1:(women<=70?3:(women<=100?4:(Math.Round(Math.Ceiling(((women-100)/40)+4))))));

4 Likes
3

Concerning the Python code: You’ve defined a function but you’re not calling it anywhere in the code. In fact you could easily leave it out in this case since the code isn’t that complex. I don’t know exactly what the desired output is, but if you try this you will probably get something more to your liking:

theaterWomen = IN[0]
theaterMen = IN[1]

theaterTotalLoad = []
theaterWomenBasins = []
theaterWomenToilets = []
womenBasins = []

menAndWomen = theaterWomen + theaterMen

if theaterWomen == 0:
womenBasins = 0
elif theaterWomen <= 40:
womenBasins = 1
elif theaterWomen <= 70:
womenBasins = 3
elif theaterWomen <= 100:
womenBasins = 4
else:
womenBasins = int(math.ceil(((theaterWomen-100)/40)+4))

theaterTotalLoad.append(menAndWomen)
theaterWomenBasins.append(womenBasins)

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

4

Vikram,

Thanks for this. This is exactly the kind of solution I was looking for!

5

Makes sense. Thanks for the input. I think for now I will go with the Design Script route, but I will certainly keep this in my pocket. I’m sure I’ll need it.

6

Here’s a link to an alternative design script solution: