def getPhotoZ(o): if o.pztype == "none": pz = PhotoZBase() elif o.pztype == "gauss": pz = PhotoZGauss(options=o) elif o.pztype == "twopop": pz = PhotoZTwoPop(options=o) elif o.pztype == "hiddenvar": pz = PhotoZHiddenVar(options=o) elif o.pztype == "franzona": pz = PhotoZHist(options=o) else: print "Bad PZ type:", o.pztype stop() return pz