urgrip/src/main/java/urgrip/programnodes/GripperGripNodeContribution.java

package urgrip.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 GripperGripNodeContribution implements ProgramNodeContribution {
    private static final String KEY_DELAY = "KEY_DELAY";
    private final ProgramAPI programAPI;
    private final UndoRedoManager undoRedoManager;
    private final KeyboardInputFactory keyboardFactory;
    private final GripperGripNodeView view;
    private final DataModel model;

    public GripperGripNodeContribution(ProgramAPIProvider apiProvider, GripperGripNodeView 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.setSpeed(model.get(KEY_DELAY, 20));
    }

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

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

    @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(KEY_DELAY, 0));
        return keyboardInput;
    }

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

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

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

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