package com.infomancers.tests;

import com.infomancers.collections.yield.asm.NewMember;
import com.infomancers.collections.yield.asmbase.YielderInformationContainer;
import org.objectweb.asm.tree.LabelNode;

import java.util.Arrays;
import java.util.Queue;

/**
 * Created by IntelliJ IDEA.
 * User: aviadbendov
 * Date: Apr 3, 2009
 * Time: 11:23:10 PM
 * To change this template use File | Settings | File Templates.
 */
public class TestYIC implements YielderInformationContainer {
    private int counter;
    private final NewMember[] slots;
    private final LabelNode[] labels;

    public TestYIC(int counter, NewMember... slots) {
        this.counter = counter;
        this.slots = slots;
        this.labels = new LabelNode[counter];
    }

    public int getCounter() {
        return counter;
    }

    public Iterable<? extends NewMember> getSlots() {
        return Arrays.asList(slots);
    }

    public Queue<Integer> getLoads() {
        return null;  //To change body of implemented methods use File | Settings | File Templates.
    }

    public NewMember getSlot(int var) {
        for (NewMember slot : slots) {
            if (slot.getIndex() == var) {
                return slot;
            }
        }

        return null;
    }

    public int takeState() {
        return counter--;
    }

    public void setStateLabel(int state, LabelNode label) {
        labels[state - 1] = label;
    }

    public LabelNode[] getStateLabels() {
        return labels;
    }

    public LabelNode getStateLabel(int state) {
        if (labels[state - 1] == null) {
            labels[state - 1] = new LabelNode();
        }

        return labels[state - 1];
    }
}