meyfa/structogram2byob

package structogram2byob.blocks.special;

import java.util.List;

import scratchlib.objects.fixed.collections.ScratchObjectArray;
import scratchlib.objects.fixed.data.ScratchObjectSymbol;
import scratchlib.objects.fixed.data.ScratchObjectUtf8;
import structogram2byob.ScratchType;
import structogram2byob.blocks.Block;
import structogram2byob.blocks.BlockDescription;
import structogram2byob.blocks.BlockRegistry;
import structogram2byob.program.ScratchConversionException;
import structogram2byob.program.VariableContext;
import structogram2byob.program.VariableMap;
import structogram2byob.program.expressions.Expression;


/**
 * This is the "set (variable) to (any)" block.
 */
public class SetVariableBlock extends Block
{
    private static final BlockDescription description = new BlockDescription.Builder()
            .label("set").param(ScratchType.ANY).label("to").param(ScratchType.ANY).build();

    /**
     * The singleton instance.
     */
    public static final SetVariableBlock instance = new SetVariableBlock();

    private SetVariableBlock()
    {
        super(description, null);
    }

    @Override
    public ScratchObjectArray toScratch(List<Expression> params, VariableMap vars, BlockRegistry blocks)
            throws ScratchConversionException
    {
        String name = asVariableName(params.get(0));
        VariableContext context = vars.get(name);

        ScratchObjectArray a = new ScratchObjectArray();

        a.add(new ScratchObjectSymbol(context.requiresBYOB() ? "changeBlockVariable" : "changeVariable"));
        a.add(new ScratchObjectUtf8(name));
        a.add(new ScratchObjectSymbol("setVar:to:"));

        if (context.requiresBYOB()) {
            a.add(context.getWriteMarker());
        }

        a.add(params.get(1).toScratch(vars, blocks));

        return a;
    }
}