How to set a different cell factory on each row in javafx TreeTableView in the same column? - javafx

Question

I'm constructing a TreeTableView that has a column of data that has different type of data stored on each row. Some rows contain arbitrary strings, while other rows contain specific strings that are only available from a fixed set defined in a ComboBox.
This column of data must be editable by the user but I need to restrict what the users enters on each row depending on whether the row contains just a plain string or a combobox string.
My code looks something like:
TreeTableColumn<EIO, String> column = new TreeTableColumn<>( "Data" );
column.setEditable( true );
//if all data in this column contains arbitrary strings then I could do:
column.setCellFactory( c -> {
TextFieldTreeTableCell<EIO, String> cell = new TextFieldTreeTableCell<EIO, String>( new DefaultStringConverter() ) {
#Override
public void startEdit() {
TreeTableRow<EIO> row = getTableRow();
if( row == null ) {
return;
}
TreeItem<EIO> treeItem = getTreeItem( row.getIndex() );
if( treeItem == null || !treeItem.isLeaf() ) {
return;
}
super.startEdit();
}
};
return cell;
});
//if all data in the column contains string from 1 specific combobox than I could do:
column.setCellFactory( c -> {
ComboBoxTreeTableCell<EIO, String> cell = new ComboBoxTreeTableCell<EIO, String>( COMBO_OPTIONS ) {
#Override
public void startEdit() {
TreeTableRow<EIO> row = getTableRow();
if( row == null ) {
return;
}
TreeItem<EIO> treeItem = getTreeItem( row.getIndex() );
if( treeItem == null || !treeItem.isLeaf() ) {
return;
}
super.startEdit();
}
};
return cell;
});
But how do I set the cellFactory for this column based on which row in the column is being edited? The EIO instance has a method eio.getComboOptions() which will return null if this row contains an arbitrary string ... but how can I use that to defined a different factory per row?