urgrip/src/main/java/uraxis/programnodes/GripperNodeContribution.java

package uraxis.programnodes;

import com.ur.urcap.api.contribution.ProgramNodeContribution;
import com.ur.urcap.api.contribution.program.ProgramAPIProvider;
import com.ur.urcap.api.domain.ProgramAPI;
import com.ur.urcap.api.domain.data.DataModel;
import com.ur.urcap.api.domain.script.ScriptWriter;
import com.ur.urcap.api.domain.undoredo.UndoRedoManager;
import com.ur.urcap.api.domain.userinteraction.keyboard.KeyboardInputCallback;
import com.ur.urcap.api.domain.userinteraction.keyboard.KeyboardInputFactory;
import com.ur.urcap.api.domain.userinteraction.keyboard.KeyboardNumberInput;

public class GripperNodeContribution implements ProgramNodeContribution {
    private final String POSITION="POSITION";
    private final String SPEED="SPEED";

    private final ProgramAPI programAPI;
    private final UndoRedoManager undoRedoManager;
    private final KeyboardInputFactory keyboardFactory;

    private final GripperNodeView view;
    private final DataModel model;


    public GripperNodeContribution(ProgramAPIProvider apiProvider, GripperNodeView view, DataModel model) {
        this.programAPI = apiProvider.getProgramAPI();
        this.undoRedoManager = apiProvider.getProgramAPI().getUndoRedoManager();
        this.keyboardFactory = apiProvider.getUserInterfaceAPI().getUserInteraction().getKeyboardInputFactory();

        this.view = view;
        this.model = model;
    }

    @Override
    public void openView() {
        view.setPosition(model.get(POSITION, 0));
        view.setSpeed(model.get(SPEED, 20));
    }

    @Override
    public void closeView() {
        model.set(POSITION, view.getPosition());
        model.set(SPEED, view.getSpeed());
    }

    @Override
    public String getTitle() {
        return "Gripper ";
    }

    @Override
    public void generateScript(ScriptWriter writer) {
        writer.appendLine("write_output_integer_register(3," + 3 + ")");
        writer.writeChildren();
    }

    public KeyboardNumberInput getKeyboardForPositionField() {
        KeyboardNumberInput<Integer> keyboardInput = keyboardFactory.createIntegerKeypadInput();
        keyboardInput.setInitialValue(model.get(POSITION, 0));
        return keyboardInput;
    }

    public KeyboardInputCallback<Integer> getCallbackForPositionField() {
        return new KeyboardInputCallback<Integer>() {
            @Override
            public void onOk(Integer value) {
                model.set(POSITION, value);
                view.setPosition(value);
            }
        };
    }

    public KeyboardNumberInput getKeyboardForSpeedField() {
        KeyboardNumberInput<Integer> keyboardInput = keyboardFactory.createIntegerKeypadInput();
        keyboardInput.setInitialValue(model.get(SPEED, 0));
        return keyboardInput;
    }

    public KeyboardInputCallback<Integer> getCallbackForSpeedField() {
        return new KeyboardInputCallback<Integer>() {
            @Override
            public void onOk(Integer value) {
                model.set(SPEED, value);
                view.setSpeed(value);
            }
        };
    }

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