def set_sites_depth_parameters(sites, gmpe): """ Need to select the appropriate z1pt0 value for different GMPEs. Note that these are required site parameters, so even though OQ has these equations built into the class in most cases. I have submitted an issue to OQ requesting subclasses of these methods that do not require the depth parameters in the SitesContext to make this easier. :param sites: A sites context. :param gmpe: A GMPE instance. :returns: A sites context with the depth parameters set for the requested GMPE. """ sites = Sites._addDepthParameters(sites) if gmpe == 'AbrahamsonEtAl2014()': sites.z1pt0 = sites.z1pt0_ask14_cal if gmpe == 'ChiouYoungs2014()': # Also BooreEtAl2014() if using subclass with depth parameter sites.z1pt0 = sites.z1pt0_cy14_cal if gmpe == 'CampbellBozorgnia2014()': sites.z2pt5 = sites.z2pt5_cb14_cal if gmpe == 'ChiouYoungs2008()': sites.z1pt0 = sites.z1pt0_cy08 if gmpe == 'CampbellBozorgnia2008()': sites.z2pt5 = sites.z2pt5_cb07 return sites