com.opengamma.language.function.FunctionProviderBean.java Source code

Introduction

Here is the source code for com.opengamma.language.function.FunctionProviderBean.java

Source

/**
 * Copyright (C) 2011 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */

package com.opengamma.language.function;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;

import org.springframework.beans.factory.InitializingBean;

import com.opengamma.util.ArgumentChecker;

/**
 * Produce a {@link FunctionProvider} from a set of {@link PublishedFunction} objects.
 */
public class FunctionProviderBean extends AbstractFunctionProvider implements InitializingBean {

    private Collection<PublishedFunction> _functions;

    public FunctionProviderBean(final Collection<PublishedFunction> functions) {
        setFunctions(functions);
    }

    public FunctionProviderBean(final PublishedFunction... functions) {
        setFunctions(Arrays.asList(functions));
    }

    public FunctionProviderBean() {
    }

    public void setFunctions(final Collection<PublishedFunction> functions) {
        ArgumentChecker.notNull(functions, "functions");
        _functions = new ArrayList<PublishedFunction>(functions);
    }

    public void addFunction(final PublishedFunction function) {
        if (_functions == null) {
            _functions = new ArrayList<PublishedFunction>();
        }
        _functions.add(function);
    }

    private Collection<PublishedFunction> getFunctionsInternal() {
        return _functions;
    }

    public Collection<PublishedFunction> getFunctions() {
        return Collections.unmodifiableCollection(getFunctionsInternal());
    }

    // InitializingBean

    @Override
    public void afterPropertiesSet() {
        ArgumentChecker.notNull(getFunctionsInternal(), "functions");
    }

    // AbstractFunctionProvider

    @Override
    protected void loadDefinitions(final Collection<MetaFunction> definitions) {
        for (PublishedFunction function : getFunctionsInternal()) {
            definitions.add(function.getMetaFunction());
        }
    }

}