001/*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017package org.jetbrains.jet.utils;
018
019import com.intellij.util.NotNullFunction;
020import org.jetbrains.annotations.NotNull;
021
022import java.util.Map;
023
024public abstract class NotNullMemoizedFunction<K, V> extends NullableMemoizedFunction<K, V>  implements NotNullFunction<K, V> {
025
026    public static <K, V> NotNullFunction<K, V> create(@NotNull final NotNullFunction<K, V> compute) {
027        return new NotNullMemoizedFunction<K, V>() {
028            @NotNull
029            @Override
030            protected V compute(@NotNull K input) {
031                return compute.fun(input);
032            }
033        };
034    }
035
036    public NotNullMemoizedFunction(@NotNull Map<K, Object> map) {
037        super(map);
038    }
039
040    public NotNullMemoizedFunction() {
041        super();
042    }
043
044    @NotNull
045    @Override
046    public V fun(@NotNull K input) {
047        //noinspection ConstantConditions
048        return super.fun(input);
049    }
050
051    @NotNull
052    @Override
053    protected abstract V compute(@NotNull K input);
054}