eu.stratosphere.arraymodel.functions.AbstractArrayModelFunction Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of array-datamodel Show documentation

The newest version!

/***********************************************************************************************************************
 * Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 **********************************************************************************************************************/
package eu.stratosphere.arraymodel.functions;

import java.lang.reflect.Method;

import eu.stratosphere.api.common.functions.AbstractFunction;
import eu.stratosphere.types.Value;

/**
 * Base class of all stubs in the array data model.
 */
public abstract class AbstractArrayModelFunction extends AbstractFunction {
	
	public Class[] getDataTypes(int input) {
		Method m = getUDFMethod();
		
		if (input == 0) {
			DataTypes.First typesFirst = m.getAnnotation(DataTypes.First.class);
			if (typesFirst == null) {
				DataTypes types = m.getAnnotation(DataTypes.class);
				return types == null ? null : types.value();
			} else {
				return typesFirst.value();
			}
		} else if (input == 1) {
			DataTypes.Second types = m.getAnnotation(DataTypes.Second.class);
			return types == null ? null : types.value();
		} else {
			throw new IndexOutOfBoundsException();
		}
	}
	
	public abstract Method getUDFMethod();
}