package controls.tableview;

import com.sun.javafx.runtime.VersionInfo;
import javafx.application.Application;
import javafx.collections.FXCollections;
import javafx.collections.ObservableList;
import javafx.scene.Scene;
import javafx.scene.control.*;
import javafx.scene.control.cell.PropertyValueFactory;
import javafx.scene.layout.StackPane;
import javafx.scene.layout.StackPaneBuilder;
import javafx.stage.Stage;
import javafx.util.Callback;

public class TableColumnPI extends Application {
    private TableView<Person> personTable = new TableView<>();
    private TableColumn<Person, String> nameCol = new TableColumn<>("name");
    private TableColumn<Person, Double> progressCol = new TableColumn<>("progress");

    @Override
    public void start(Stage primaryStage) {
        ObservableList<Person> data = FXCollections.observableArrayList();
        for (int i = 0; i < 40; i++) {
            data.add(new Person("person_" + i, i / 100d));
        }

        nameCol.setCellValueFactory(new PropertyValueFactory<Person, String>("name"));
        progressCol.setCellValueFactory(new PropertyValueFactory<Person, Double>("progress"));

        progressCol.setCellFactory(new Callback<TableColumn<Person, Double>, TableCell<Person, Double>>() {
            @Override
            public TableCell<Person, Double> call(TableColumn<Person, Double> arg0) {
                return new TableCell<Person, Double>() {
                    private ProgressIndicator pi = new ProgressIndicator(0);

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

                        if (empty) {
                            setText(null);
                            setGraphic(null);
                        } else {
                            pi.setProgress(getItem());
                            setGraphic(pi);
                            setContentDisplay(ContentDisplay.GRAPHIC_ONLY);
                        }
                    }
                };
            }
        });

        personTable.setItems(data);
        personTable.getColumns().addAll(nameCol, progressCol);
        StackPane root = StackPaneBuilder.create().children(personTable).build();
        primaryStage.setScene(new Scene(root, 300, 250));
        primaryStage.setTitle(VersionInfo.getRuntimeVersion());
        primaryStage.show();
    }

    public class Person {
        private String name;
        private Double progress;

        public Person(String name, Double progress) {
            this.name = name;
            this.progress = progress;
        }

        public String getName() {
            return name;
        }

        public Double getProgress() {
            return progress;
        }
    }

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