/*******************************************************************************
*   Copyright 2015 Analog Devices, Inc.
*
*   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 com.analog.lyric.dimple.data;

import static java.util.Objects.*;

import org.eclipse.jdt.annotation.Nullable;

import com.analog.lyric.collect.ExtendedArrayList;
import com.analog.lyric.collect.NonNullListIndices;
import com.analog.lyric.collect.PrimitiveIterable;
import com.analog.lyric.dimple.model.core.FactorGraph;
import com.analog.lyric.dimple.model.core.IFactorGraphChild;
import com.analog.lyric.util.misc.Internal;

import net.jcip.annotations.NotThreadSafe;

/**
 * 
 * @since 0.08
 * @author Christopher Barber
 */
@NotThreadSafe
public class DenseFactorGraphData<K extends IFactorGraphChild, D extends IDatum> extends FactorGraphData<K,D>
{
	/*-------
	 * State
	 */
	
	private final ExtendedArrayList<D> _data;
	private int _size;

	/*--------------
	 * Construction
	 */
	
	/**
	 * @param graph
	 * @since 0.08
	 */
	public DenseFactorGraphData(DataLayerBase<K, ? super D> layer, FactorGraph graph, Class<K> keyType, Class<D> baseType)
	{
		super(layer, graph, keyType, baseType);
		_data = new ExtendedArrayList<>(graph.getOwnedVariables().size());
	}

	protected DenseFactorGraphData(DataLayerBase<K, ? super D> layer, FactorGraphData<K,D> other)
	{
		this(layer, other._graph, other._keyType, other._baseType);
		for (int i : other.getLocalIndices())
		{
			setByLocalIndex(i, _baseType.cast(requireNonNull(other.getByLocalIndex(i)).clone()));
		}
	}
	
	/**
	 * @category internal
	 */
	@Internal
	@Override
	public DenseFactorGraphData<K,D> clone(DataLayerBase<K, ? super D> newLayer)
	{
		return new DenseFactorGraphData<>(newLayer, this);
	}
	
	public static <K extends IFactorGraphChild, D extends IDatum> Constructor<K,D> constructorForType(
		final Class<K> keyType,	final Class<D> baseType)
	{
		return new Constructor<K,D> () {
			@Override
			public FactorGraphData<K,D> apply(DataLayerBase<K, ? super D> layer, FactorGraph graph)
			{
				return new DenseFactorGraphData<>(layer, graph, keyType, baseType);
			}
			
			@Override
			public boolean createOnRead()
			{
				return false;
			}
			
			@Override
			public Class<D> baseType()
			{
				return baseType;
			}
			
			@Override
			public Class<K> keyType()
			{
				return keyType;
			}
		};
	}
	
	/*-------------
	 * Map methods
	 */
	
	@Override
	public void clear()
	{
		_data.clear();
		_size = 0;
	}
	
	@Override
	public int size()
	{
		return _size;
	}
	
	/*-------------------------
	 * FactorGraphData methods
	 */
	
	@Override
	public boolean containsLocalIndex(int index)
	{
		return _data.getOrNull(index) != null;
	}
	
	@Override
	public @Nullable D getByLocalIndex(int index)
	{
		return _data.getOrNull(index);
	}
	
	@Override
	public PrimitiveIterable.OfInt getLocalIndices()
	{
		return new NonNullListIndices(_data);
	}
	
	@Override
	public boolean isView()
	{
		return false;
	}

	@Override
	public @Nullable D setByLocalIndex(int index, @Nullable D datum)
	{
		D prev = _data.set(index, _baseType.cast(datum));
		if (datum != null)
			++_size;
		if (prev != null)
			--_size;
		return prev;
	}

}