/**
* Copyright (C) 2012 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.financial.analytics.model.curve;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import com.opengamma.core.change.ChangeEvent;
import com.opengamma.core.config.impl.ConfigItem;
import com.opengamma.engine.function.config.AbstractFunctionConfigurationBean;
import com.opengamma.engine.function.config.BeanDynamicFunctionConfigurationSource;
import com.opengamma.engine.function.config.FunctionConfiguration;
import com.opengamma.engine.function.config.FunctionConfigurationSource;
import com.opengamma.engine.function.config.VersionedFunctionConfigurationBean;
import com.opengamma.financial.analytics.curve.CurveConstructionConfiguration;
import com.opengamma.financial.analytics.curve.CurveConstructionConfigurationFunction;
import com.opengamma.financial.analytics.curve.CurveDefinition;
import com.opengamma.financial.analytics.curve.CurveGroupConfiguration;
import com.opengamma.financial.analytics.curve.CurveTypeConfiguration;
import com.opengamma.financial.analytics.curve.InflationCurveTypeConfiguration;
import com.opengamma.financial.analytics.curve.InflationIssuerCurveTypeConfiguration;
import com.opengamma.financial.analytics.curve.IssuerCurveTypeConfiguration;
import com.opengamma.financial.analytics.model.curve.forward.InstantaneousForwardCurveFunction;
import com.opengamma.financial.analytics.parameters.G2ppParameters;
import com.opengamma.financial.analytics.parameters.HullWhiteOneFactorParameters;
import com.opengamma.financial.config.ConfigMasterChangeProvider;
import com.opengamma.master.config.ConfigDocument;
import com.opengamma.master.config.ConfigMaster;
import com.opengamma.master.config.ConfigSearchRequest;
import com.opengamma.master.config.impl.ConfigSearchIterator;
import com.opengamma.util.ArgumentChecker;
import com.opengamma.util.money.Currency;
/**
* Function repository configuration source for the functions contained in this package and sub-packages.
*/
public class CurveFunctions extends AbstractFunctionConfigurationBean {
/**
* Default instance of a repository configuration source exposing the functions from this package and its sub-packages.
*
* @return the configuration source exposing functions from this package and its sub-packages
*/
public static FunctionConfigurationSource instance() {
return new CurveFunctions().getObjectCreating();
}
/**
* Returns a configuration populated with curve building functions.
*
* @param configMaster The config master
* @return A populated configuration
*/
public static FunctionConfigurationSource providers(final ConfigMaster configMaster) {
return new BeanDynamicFunctionConfigurationSource(ConfigMasterChangeProvider.of(configMaster)) {
@Override
protected VersionedFunctionConfigurationBean createConfiguration() {
final Providers providers = new Providers();
providers.setConfigMaster(configMaster);
return providers;
}
@Override
protected boolean isPropogateEvent(final ChangeEvent event) {
return Providers.isMonitoredType(event.getObjectId().getValue());
}
};
}
/**
* Returns a configuration populated with functions that supply model parameters (e.g. G2++ parameters).
*
* @param configMaster The config master
* @return A populated configuration
*/
public static FunctionConfigurationSource parameterProviders(final ConfigMaster configMaster) {
return new BeanDynamicFunctionConfigurationSource(ConfigMasterChangeProvider.of(configMaster)) {
@Override
protected VersionedFunctionConfigurationBean createConfiguration() {
final ParameterProviders providers = new ParameterProviders();
providers.setConfigMaster(configMaster);
return providers;
}
@Override
protected boolean isPropogateEvent(final ChangeEvent event) {
return ParameterProviders.isMonitoredType(event.getObjectId().getValue());
}
};
}
/**
* Function repository configuration source for the default functions contained in this package.
*/
public static class Defaults extends AbstractFunctionConfigurationBean {
/** The absolute tolerance used in root-finding by curve functions */
private double _absoluteTolerance = 1e-9;
/** The relative tolerance used in root-finding by curve functions */
private double _relativeTolerance = 1e-9;
/** The maximum number of iterations used in root-finding by curve functions */
private int _maxIterations = 1000;
/**
* Gets the absolute tolerance.
*
* @return The absolute tolerance
*/
public double getAbsoluteTolerance() {
return _absoluteTolerance;
}
/**
* Sets the absolute tolerance.
*
* @param absoluteTolerance The absolute tolerance
*/
public void setAbsoluteTolerance(final double absoluteTolerance) {
_absoluteTolerance = absoluteTolerance;
}
/**
* Gets the relative tolerance.
*
* @return The relative tolerance
*/
public double getRelativeTolerance() {
return _relativeTolerance;
}
/**
* Sets the relative tolerance.
*
* @param relativeTolerance The relative tolerance.
*/
public void setRelativeTolerance(final double relativeTolerance) {
_relativeTolerance = relativeTolerance;
}
/**
* Gets the maximum number of iterations.
*
* @return The maximum number of iterations
*/
public int getMaximumIterations() {
return _maxIterations;
}
/**
* Sets the maximum number of iterations
*
* @param maxIterations The maximum number of iterations
*/
public void setMaximumIterations(final int maxIterations) {
_maxIterations = maxIterations;
}
@Override
public void afterPropertiesSet() {
ArgumentChecker.notNegativeOrZero(getAbsoluteTolerance(), "absolute tolerance");
ArgumentChecker.notNegativeOrZero(getRelativeTolerance(), "relative tolerance");
ArgumentChecker.notNegativeOrZero(getMaximumIterations(), "maximum iterations");
super.afterPropertiesSet();
}
@Override
protected void addAllConfigurations(final List<FunctionConfiguration> functions) {
addCurveDefaults(functions);
}
/**
* Adds default values for absolute tolerance, relative tolerance and the maximum number of iterations for all curve building functions
*
* @param functions The list of function configurations.
*/
protected void addCurveDefaults(final List<FunctionConfiguration> functions) {
final String[] args = new String[3];
args[0] = Double.toString(getAbsoluteTolerance());
args[1] = Double.toString(getRelativeTolerance());
args[2] = Integer.toString(getMaximumIterations());
functions.add(functionConfiguration(CurveDefaults.class, args));
}
}
/**
* Function repository configuration source for curve functions based on the items defined in a {@link ConfigMaster}.
*/
public static class Providers extends VersionedFunctionConfigurationBean {
/** The configuration master */
private ConfigMaster _configMaster;
/**
* Sets the config master.
*
* @param configMaster The config master, not null
*/
public void setConfigMaster(final ConfigMaster configMaster) {
ArgumentChecker.notNull(configMaster, "config master");
_configMaster = configMaster;
}
/**
* Gets the config master.
*
* @return The config master
*/
public ConfigMaster getConfigMaster() {
return _configMaster;
}
/**
* Adds all interpolated curve building functions.
*
* @param functions The list of functions
* @param curveTypeConfigClasses The type of curves in a construction configuration
* @param curveConfigName The curve construction configuration name
*/
protected void addInterpolatedCurveBuildingFunctions(final List<FunctionConfiguration> functions, final Set<Class<? extends CurveTypeConfiguration>> curveTypeConfigClasses,
final String curveConfigName) {
if (curveTypeConfigClasses.contains(InflationCurveTypeConfiguration.class)) {
functions.add(functionConfiguration(InflationProviderDiscountingFunction.class, curveConfigName));
} else if (curveTypeConfigClasses.contains(InflationIssuerCurveTypeConfiguration.class)) {
functions.add(functionConfiguration(InflationIssuerProviderDiscountingFunction.class, curveConfigName));
} else if (curveTypeConfigClasses.contains(IssuerCurveTypeConfiguration.class)) {
functions.add(functionConfiguration(IssuerProviderDiscountingFunction.class, curveConfigName));
functions.add(functionConfiguration(IssuerProviderInterpolatedFunction.class, curveConfigName));
} else {
functions.add(functionConfiguration(MultiCurveDiscountingFunction.class, curveConfigName));
functions.add(functionConfiguration(HullWhiteOneFactorDiscountingCurveFunction.class, curveConfigName));
functions.add(functionConfiguration(MultiCurveInterpolatedFunction.class, curveConfigName));
}
functions.add(functionConfiguration(FXMatrixFunction.class, curveConfigName));
functions.add(functionConfiguration(CurveConstructionConfigurationFunction.class, curveConfigName));
}
/**
* Adds a function that constructs yield curves using the ISDA methodology.
*
* @param functions The list of functions
* @param curveConfigName The curve configuration name
*/
protected void addCurveBuildingFunctions(final List<FunctionConfiguration> functions, final String curveConfigName) {
functions.add(functionConfiguration(ISDACompliantCurveFunction.class, curveConfigName));
}
@Override
protected void addAllConfigurations(final List<FunctionConfiguration> functions) {
final ConfigSearchRequest<CurveDefinition> searchRequest = new ConfigSearchRequest<>();
searchRequest.setVersionCorrection(getVersionCorrection());
final Class<?>[] curveConstructionConfigurationClasses = new Class[] {CurveConstructionConfiguration.class };
for (final Class<?> klass : curveConstructionConfigurationClasses) {
searchRequest.setType(klass);
for (final ConfigDocument configDocument : ConfigSearchIterator.iterable(getConfigMaster(), searchRequest)) {
final String documentName = configDocument.getName();
final CurveConstructionConfiguration config = ((ConfigItem<CurveConstructionConfiguration>) configDocument.getConfig()).getValue();
/*
* We need the CurveTypeConfigurations of the curves contained within the CurveConstructionConfiguration to
* decided whether we want to add the curve building function for this CurveConstructionConfiguration.
*/
final Set<Class<? extends CurveTypeConfiguration>> allCurveTypeConfigs = extractCurveTypeConfigurationClasses(config);
addInterpolatedCurveBuildingFunctions(functions, allCurveTypeConfigs, documentName);
}
}
searchRequest.setType(CurveDefinition.class);
searchRequest.setVersionCorrection(getVersionCorrection());
for (final ConfigDocument configDocument : ConfigSearchIterator.iterable(getConfigMaster(), searchRequest)) {
final String documentName = configDocument.getName();
addCurveBuildingFunctions(functions, documentName);
}
}
private static boolean isMonitoredType(final String type) {
return CurveConstructionConfiguration.class.getName().equals(type) || CurveDefinition.class.getName().equals(type);
}
/**
* Extracts the CurveTypeConfiguration classes from a given CurveConstructionConfiguration.
* <p>
* This allows us to decide whether we want a function to be supported based on the contained CurveTypeConfigurations.
*
* @param config the CurveConstructionConfiguration to retrieve the curve types from.
* @return a Set of CurveTypeConfigurations.
*/
private static Set<Class<? extends CurveTypeConfiguration>> extractCurveTypeConfigurationClasses(final CurveConstructionConfiguration config) {
final Set<Class<? extends CurveTypeConfiguration>> allCurveTypeConfigs = new HashSet<>();
for (final CurveGroupConfiguration group : config.getCurveGroups()) {
for (final List<? extends CurveTypeConfiguration> curveTypeConfigs : group.getTypesForCurves().values()) {
for (final CurveTypeConfiguration curveTypeConfig : curveTypeConfigs) {
allCurveTypeConfigs.add(curveTypeConfig.getClass());
}
}
}
return allCurveTypeConfigs;
}
}
/**
* Function repository configuration source for curve parameter functions based on the items in a {@link ConfigMaster}
*/
public static class ParameterProviders extends VersionedFunctionConfigurationBean {
/** The configuration master */
private ConfigMaster _configMaster;
/**
* Sets the config master
*
* @param configMaster The config master, not null
*/
public void setConfigMaster(final ConfigMaster configMaster) {
ArgumentChecker.notNull(configMaster, "config master");
_configMaster = configMaster;
}
/**
* Gets the configuration master.
*
* @return The configuration master
*/
public ConfigMaster getConfigMaster() {
return _configMaster;
}
@Override
protected void addAllConfigurations(final List<FunctionConfiguration> functions) {
final ConfigSearchRequest<HullWhiteOneFactorParameters> hwSearchRequest = new ConfigSearchRequest<>();
hwSearchRequest.setType(HullWhiteOneFactorParameters.class);
hwSearchRequest.setVersionCorrection(getVersionCorrection());
for (final ConfigDocument configDocument : ConfigSearchIterator.iterable(getConfigMaster(), hwSearchRequest)) {
final String configurationName = configDocument.getName();
final HullWhiteOneFactorParameters hullWhiteParameters = ((ConfigItem<HullWhiteOneFactorParameters>) configDocument.getConfig()).getValue();
final Currency currency = hullWhiteParameters.getCurrency();
functions.add(functionConfiguration(HullWhiteOneFactorParametersFunction.class, configurationName, currency.getCode()));
}
final ConfigSearchRequest<G2ppParameters> g2ppSearchRequest = new ConfigSearchRequest<>();
g2ppSearchRequest.setType(G2ppParameters.class);
g2ppSearchRequest.setVersionCorrection(getVersionCorrection());
for (final ConfigDocument configDocument : ConfigSearchIterator.iterable(getConfigMaster(), g2ppSearchRequest)) {
final String configurationName = configDocument.getName();
final G2ppParameters g2ppParameters = ((ConfigItem<G2ppParameters>) configDocument.getConfig()).getValue();
final Currency currency = g2ppParameters.getCurrency();
functions.add(functionConfiguration(G2ppParametersFunction.class, configurationName, currency.getCode()));
}
}
private static boolean isMonitoredType(final String type) {
return HullWhiteOneFactorParameters.class.getName().equals(type) || G2ppParameters.class.getName().equals(type);
}
}
@Override
protected void addAllConfigurations(final List<FunctionConfiguration> functions) {
functions.add(functionConfiguration(HardCodedHullWhiteOneFactorParametersFunction.class));
functions.add(functionConfiguration(HardCodedG2ppParametersFunction.class));
functions.add(functionConfiguration(InstantaneousForwardCurveFunction.class));
}
}