/*
 * Copyright (c) 2016-2017, Adam <[email protected]>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 *    list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package asm.attributes.code.instructions;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import asm.attributes.code.Instruction;
import asm.attributes.code.InstructionType;
import asm.attributes.code.Instructions;
import asm.attributes.code.Label;
import asm.attributes.code.instruction.types.JumpingInstruction;
import asm.execution.Frame;
import asm.execution.InstructionContext;
import asm.execution.Stack;
import asm.execution.StackContext;
import org.objectweb.asm.MethodVisitor;

public class LookupSwitch extends Instruction implements JumpingInstruction
{
	private List<Label> branchi = new ArrayList<>();
	private Label defi;

	private int[] match;

	public LookupSwitch(Instructions instructions, InstructionType type)
	{
		super(instructions, type);
	}

	@Override
	public Instruction clone()
	{
		LookupSwitch i = (LookupSwitch) super.clone();
		i.branchi = new ArrayList<>(branchi);
		return i;
	}

	@Override
	public void accept(MethodVisitor visitor)
	{
		visitor.visitLookupSwitchInsn(defi.getLabel(), match,
			branchi.stream().map(l -> l.getLabel()).toArray(org.objectweb.asm.Label[]::new));
	}

	@Override
	public InstructionContext execute(Frame frame)
	{
		InstructionContext ins = new InstructionContext(this, frame);
		Stack stack = frame.getStack();

		StackContext value = stack.pop();
		ins.pop(value);

		// N.B. lookupswitch isn't mappable, so frame.other is never mapped
		// (the frames it creates are dropped from the pending execution list
		// for other == null) - so the step executor won't map instructions
		// from the non-default branch
		for (Label i : branchi)
		{
			Frame other = frame.dup();
			other.jump(ins, i);

			ins.branch(other);
		}

		frame.jump(ins, defi);

		return ins;
	}

	@Override
	public boolean isTerminal()
	{
		return true;
	}

	@Override
	public List<Label> getJumps()
	{
		List<Label> list = new ArrayList<>();
		for (Label i : branchi)
		{
			list.add(i);
		}
		list.add(defi);
		return list.stream().distinct().collect(Collectors.toList());
	}

	@Override
	public void setJumps(List<Label> labels)
	{
		throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
	}

	@Override
	public void setLabel(org.objectweb.asm.Label label)
	{
		throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
	}

	public List<Label> getBranchi()
	{
		return branchi;
	}

	public void setBranchi(List<Label> branchi)
	{
		this.branchi = branchi;
	}

	public Label getDefi()
	{
		return defi;
	}

	public void setDefi(Label defi)
	{
		this.defi = defi;
	}

	public int[] getMatch()
	{
		return match;
	}

	public void setMatch(int[] match)
	{
		this.match = match;
	}
}