--- old/modules/controls/src/main/java/com/sun/javafx/scene/control/behavior/TreeTableCellBehavior.java 2015-09-03 14:51:02.536253400 -0700 +++ new/modules/controls/src/main/java/com/sun/javafx/scene/control/behavior/TreeTableCellBehavior.java 2015-09-03 14:51:01.932218900 -0700 @@ -25,7 +25,6 @@ package com.sun.javafx.scene.control.behavior; -import javafx.geometry.Bounds; import javafx.scene.Node; import javafx.scene.control.TableColumnBase; import javafx.scene.control.TablePositionBase; @@ -59,12 +58,12 @@ /** @{@inheritDoc} */ @Override protected TreeTableView getCellContainer() { - return getControl().getTreeTableView(); + return getNode().getTreeTableView(); } /** @{@inheritDoc} */ @Override protected TreeTableColumn getTableColumn() { - return getControl().getTableColumn(); + return getNode().getTableColumn(); } /** @{@inheritDoc} */ @@ -89,7 +88,7 @@ /** @{@inheritDoc} */ @Override protected boolean isTableRowSelected() { - return getControl().getTreeTableRow().isSelected(); + return getNode().getTreeTableRow().isSelected(); } /** @{@inheritDoc} */ @@ -112,15 +111,15 @@ } @Override protected boolean handleDisclosureNode(double x, double y) { - final TreeItem treeItem = getControl().getTreeTableRow().getTreeItem(); + final TreeItem treeItem = getNode().getTreeTableRow().getTreeItem(); - final TreeTableView treeTableView = getControl().getTreeTableView(); + final TreeTableView treeTableView = getNode().getTreeTableView(); final TreeTableColumn column = getTableColumn(); final TreeTableColumn treeColumn = treeTableView.getTreeColumn() == null ? treeTableView.getVisibleLeafColumn(0) : treeTableView.getTreeColumn(); if (column == treeColumn) { - final Node disclosureNode = getControl().getTreeTableRow().getDisclosureNode(); + final Node disclosureNode = getNode().getTreeTableRow().getDisclosureNode(); if (disclosureNode != null) { double startX = 0; for (TreeTableColumn tc : treeTableView.getVisibleLeafColumns()) { @@ -142,16 +141,16 @@ @Override protected void handleClicks(MouseButton button, int clickCount, boolean isAlreadySelected) { // handle editing, which only occurs with the primary mouse button - TreeItem treeItem = getControl().getTreeTableRow().getTreeItem(); + TreeItem treeItem = getNode().getTreeTableRow().getTreeItem(); if (button == MouseButton.PRIMARY) { if (clickCount == 1 && isAlreadySelected) { - edit(getControl()); + edit(getNode()); } else if (clickCount == 1) { // cancel editing edit(null); } else if (clickCount == 2 && treeItem.isLeaf()) { // attempt to edit - edit(getControl()); + edit(getNode()); } else if (clickCount % 2 == 0) { // try to expand/collapse branch tree item treeItem.setExpanded(! treeItem.isExpanded());