JavaFX TreeView : Access list of TreeCells

Question

我正在尝试从 TreeView 中获取当前存在的 TreeCell 的列表。我当前的尝试如下所示:

private Set<MyTreeCell> getMyTreeCells(){
    try {
        @SuppressWarnings("unchecked")
        Field f = ((VirtualContainerBase<TreeView<myNode>, TreeViewBehavior<myNode>, MyTreeCell>) myTreeView.skinProperty().get()).getClass().getDeclaredField("flow");
        f.setAccessible(true);
        @SuppressWarnings("unchecked")
        Field g = ((VirtualFlow<MyTreeCell>) f.get(((VirtualContainerBase<TreeView<myNode>, TreeViewBehavior<myNode>, ProofTreeCell>) proofTreeView.skinProperty().get()))).getClass().getDeclaredField("cells");
        g.setAccessible(true);
        @SuppressWarnings("unchecked")
        ArrayLinkedList<MyTreeCell> l = (ArrayLinkedList<MyTreeCell>) g.get(((VirtualFlow<MyTreeCell>) f.get(((TreeViewSkin<myNode>) myTreeView.skinProperty().get()))));
        Set<myTreeCell> s = new HashSet<>();
        s.addAll(l);
        return s;
    }
    catch (NoSuchFieldException e) {
        e.printStackTrace();
    }
    catch (SecurityException e) {
        e.printStackTrace();
    }
    catch (IllegalArgumentException e) {
        e.printStackTrace();
    }
    catch (IllegalAccessException e) {
        e.printStackTrace();
    }
    return null;
}

不幸的是，它抛出java.lang.NoSuchFieldException: flow。但为什么？显然，Field flow 存在于 VirtualContainerBase 类中。

请注意，我编写此代码只是为了好玩，而不是实际用于生产代码。我已经有了一个更干净的解决方案来解决这个问题。

Answer 1

VirtualContainerBase 声明 flow，但 myTreeView.skinProperty().get().getClass() 不返回 VirtualContainerBase。类，它返回TreeViewSkin.class。由于 getDeclaredField() 仅返回在该类中声明的字段，而不返回继承的字段，因此您会得到异常。

我认为你可以这样做

Field f = VirtualContainerBase.class.getDeclaredField("flow");
Field g = VirtualFlow.class.getDeclaredField("cells");

您还可以消除大部分类型转换，如本例所示:

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;

import com.sun.javafx.scene.control.skin.VirtualContainerBase;
import com.sun.javafx.scene.control.skin.VirtualFlow;

import javafx.application.Application;
import javafx.geometry.Insets;
import javafx.geometry.Pos;
import javafx.scene.Scene;
import javafx.scene.control.Button;
import javafx.scene.control.TreeCell;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeView;
import javafx.scene.layout.BorderPane;
import javafx.stage.Stage;

public class GetTreeCellsByReflection extends Application {

    @SuppressWarnings({ "restriction", "unchecked" })
    @Override
    public void start(Stage primaryStage) {
        TreeView<Integer> tree = createRandomTree(100);
        Button findCells = new Button("Find cells");
        findCells.setOnAction(e -> {
            try {
                Field flowField = VirtualContainerBase.class.getDeclaredField("flow");
                flowField.setAccessible(true);
                Field cellsField = VirtualFlow.class.getDeclaredField("cells");
                cellsField.setAccessible(true);
                Object flow = flowField.get(tree.getSkin());
                ((Iterable<TreeCell<?>>) cellsField.get(flow)).forEach(System.out::println);;
            } catch (Exception exc) {
                exc.printStackTrace();
            }
        });

        BorderPane root = new BorderPane(tree);
        BorderPane.setAlignment(findCells, Pos.CENTER);
        BorderPane.setMargin(findCells, new Insets(5));
        root.setBottom(findCells);

        Scene scene = new Scene(root, 600, 600);
        primaryStage.setScene(scene);
        primaryStage.show();
    }

    private TreeView<Integer> createRandomTree(int numItems) {
        Random rng = new Random();
        TreeItem<Integer> root = new TreeItem<>(0);
        List<TreeItem<Integer>> items = new ArrayList<>();
        items.add(root);
        for (int i = 1 ; i < numItems ; i++) {
            TreeItem<Integer> item = new TreeItem<>(i);
            items.get(rng.nextInt(items.size())).getChildren().add(item);
            items.add(item);
        }
        return new TreeView<>(root);
    }

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