package treetableviewtest;

import com.sun.javafx.runtime.VersionInfo;
import javafx.application.Application;
import static javafx.application.Application.launch;
import javafx.beans.property.SimpleStringProperty;
import javafx.beans.value.ChangeListener;
import javafx.beans.value.ObservableValue;
import javafx.collections.FXCollections;
import javafx.collections.ObservableList;
import javafx.event.ActionEvent;
import javafx.event.EventHandler;
import javafx.scene.Node;
import javafx.scene.Scene;
import javafx.scene.control.Button;
import javafx.scene.control.SelectionMode;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeTableColumn;
import javafx.scene.control.TreeTableColumn.CellDataFeatures;
import javafx.scene.control.TreeTableView;
import javafx.scene.control.cell.ChoiceBoxTreeTableCell;
import javafx.scene.control.cell.TextFieldTreeTableCell;
import javafx.scene.layout.HBox;
import javafx.scene.paint.Color;
import javafx.scene.shape.Circle;
import javafx.scene.shape.PolygonBuilder;
import javafx.stage.Stage;
import javafx.util.Callback;

public class TreeTableViewTest extends Application {
    
    private ObservableList<String> idPool = FXCollections.observableArrayList();
    private TreeTableView ttv;
    
    @Override
    public void start(Stage primaryStage) {
        
        HBox root = new HBox(10.0);
        Scene scene = new Scene(root, 400, 250);
        
        initeTreeTableView();
        
        Button btnIncreaseColWidth = new Button("Set graphics");
        btnIncreaseColWidth.setOnAction(new EventHandler<ActionEvent>() {

            int index;
            Node[] graphicNodes = {
                new Circle(6.75, Color.RED),
                PolygonBuilder.create().fill(Color.GREEN).points(new Double[]{
                    0.0, 7.0,
                    5.0, 20.0,
                    15.0, 20.0,
                    20.0, 7.0,
                    10.0, 0.0,}).build()
            };
            
            @Override
            public void handle(ActionEvent event) {
                TreeItem root = ttv.getRoot();
                root.setGraphic(graphicNodes[index]);
                if (++index == graphicNodes.length) index = 0;
                
            }
        });
        
        root.getChildren().addAll(ttv, btnIncreaseColWidth);
        
        primaryStage.setTitle(VersionInfo.getRuntimeVersion());
        primaryStage.setScene(scene);
        primaryStage.show();
    }

    public static void main(String[] args) {
        launch(args);
    }

    private void initeTreeTableView() {
        ttv = new TreeTableView();
        ttv.getFocusModel().focusedIndexProperty().addListener(new ChangeListener<Number>() {

            @Override
            public void changed(ObservableValue<? extends Number> observable, Number oldValue, Number newValue) {
                System.out.println("Focused index = " + ttv.getFocusModel().getFocusedIndex());
            }
        });
        ttv.getSelectionModel().setSelectionMode(SelectionMode.MULTIPLE);
        ttv.setRoot(new TreeItem(new TreeTableViewTest.Data("Root", "42")));
        ttv.getRoot().setExpanded(true);
        
        for (int i = 0; i < 10; i++) {
            String id = "id - " + i;
            idPool.add(id);
            
            ttv.getRoot().getChildren().add(new TreeItem(new TreeTableViewTest.Data("item - " + i, id)));
        }
        
        final TreeTableColumn colName = new TreeTableColumn("Name");
        colName.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<TreeTableViewTest.Data, String>, ObservableValue<String>>() {
            @Override
            public ObservableValue<String> call(CellDataFeatures<TreeTableViewTest.Data, String> p) {
                return p.getValue().getValue().name;
            }
        });
        
        ttv.getColumns().addAll(colName);
        colName.setCellFactory(TextFieldTreeTableCell.forTreeTableColumn());
    }
    
    private class Data {
        public SimpleStringProperty name;
        public SimpleStringProperty id;
        
        public Data(String _name, String _id) {
            name = new SimpleStringProperty(_name);
            id = new SimpleStringProperty(_id);
        }
    }
}