FXTreeCell.java

/**
 * erlyberly, erlang trace debugger
 * Copyright (C) 2016 Andy Till
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package ui;

import javafx.scene.Parent;
import javafx.scene.control.TreeCell;

public class FXTreeCell<T> extends TreeCell<T> {

    private final CellController<T> controller;

    private Parent aControl;

    public FXTreeCell(CellController<T> aController, Parent aControl) {
        controller = aController;
        this.aControl = aControl;

    }

    @Override
    public void cancelEdit() {
        super.cancelEdit();

        if(controller != null) {
            controller.cancelEdit();
        }
    }

    @Override
    public void commitEdit(T newValue) {
        super.commitEdit(newValue);

        if(controller != null) {
            controller.commitEdit(newValue);
        }
    }

    @Override
    public void startEdit() {
        super.startEdit();

        if(controller != null) {
            controller.startEdit();
        }
    }

    @Override
    protected void updateItem(T item, boolean empty) {
        super.updateItem(item, empty);

        if(item == null || empty) {
            setGraphic(null);
            setText(null);
        }
        else {
            setGraphic(aControl);
            setText(null);
        }

        if(controller != null) {
            controller.updateItem(item, empty);
        }
    }
}