/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2015, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Applied Energistics 2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.parts.automation;


import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.google.common.collect.ImmutableMap;

import net.minecraft.util.ResourceLocation;

import appeng.api.parts.IPartModel;
import appeng.core.AppEng;
import appeng.parts.PartModel;


/**
 * Contains a mapping from a Plane's connections to the models to use for that state.
 */
public class PlaneModels
{

	public static final ResourceLocation MODEL_CHASSIS_OFF = new ResourceLocation( AppEng.MOD_ID, "part/transition_plane_off" );
	public static final ResourceLocation MODEL_CHASSIS_ON = new ResourceLocation( AppEng.MOD_ID, "part/transition_plane_on" );
	public static final ResourceLocation MODEL_CHASSIS_HAS_CHANNEL = new ResourceLocation( AppEng.MOD_ID, "part/transition_plane_has_channel" );

	private final Map<PlaneConnections, IPartModel> modelsOff;

	private final Map<PlaneConnections, IPartModel> modelsOn;

	private final Map<PlaneConnections, IPartModel> modelsHasChannel;

	public PlaneModels( String prefixOff, String prefixOn )
	{
		Map<PlaneConnections, IPartModel> modelsOff = new HashMap<>();
		Map<PlaneConnections, IPartModel> modelsOn = new HashMap<>();
		Map<PlaneConnections, IPartModel> modelsHasChannel = new HashMap<>();

		for( PlaneConnections permutation : PlaneConnections.PERMUTATIONS )
		{
			ResourceLocation planeOff = new ResourceLocation( AppEng.MOD_ID, prefixOff + permutation.getFilenameSuffix() );
			ResourceLocation planeOn = new ResourceLocation( AppEng.MOD_ID, prefixOn + permutation.getFilenameSuffix() );

			modelsOff.put( permutation, new PartModel( MODEL_CHASSIS_OFF, planeOff ) );
			modelsOn.put( permutation, new PartModel( MODEL_CHASSIS_ON, planeOff ) );
			modelsHasChannel.put( permutation, new PartModel( MODEL_CHASSIS_HAS_CHANNEL, planeOn ) );
		}

		this.modelsOff = ImmutableMap.copyOf( modelsOff );
		this.modelsOn = ImmutableMap.copyOf( modelsOn );
		this.modelsHasChannel = ImmutableMap.copyOf( modelsHasChannel );
	}

	public IPartModel getModel( PlaneConnections connections, boolean hasPower, boolean hasChannel )
	{
		if( hasPower && hasChannel )
		{
			return this.modelsHasChannel.get( connections );
		}
		else if( hasPower )
		{
			return this.modelsOn.get( connections );
		}
		else
		{
			return this.modelsOff.get( connections );
		}
	}

	public List<IPartModel> getModels()
	{
		List<IPartModel> result = new ArrayList<>();
		this.modelsOff.values().forEach( result::add );
		this.modelsOn.values().forEach( result::add );
		this.modelsHasChannel.values().forEach( result::add );
		return result;
	}

}
