Change focused cell when tab is pressed while editing

Author: nmeylan

src/array_table.rs

@@ -884,7 +884,6 @@ impl<'array> ArrayTable<'array> {
                                 row_index: table_row_index,
                                 is_pinned_column_table: pinned_column_table,
                             });
-                            ui.ctx().memory_mut(|m| m.request_focus(self.table_id));
 
                         } else {
                             textedit_response.request_focus();
@@ -1086,6 +1085,9 @@ impl<'array> ArrayTable<'array> {
                     if response.hovered() {
                         ui.ctx().set_cursor_icon(CursorIcon::Cell);
                     }
+                    if updated_value.is_some() {
+                        ui.ctx().memory_mut(|m| m.request_focus(self.table_id));
+                    }
                     Some(response)
                 });
             }

src/object_table.rs

@@ -105,11 +105,11 @@ impl ObjectTable {
                                         row_index: table_row_index,
                                         is_pinned_column_table: false,
                                     });
-                                    ui.ctx().memory_mut(|m| m.request_focus(self.table_id));
 
                                 } else {
                                     textedit_response.request_focus();
                                 }
+
                                 None
                             } else {
                                 let rect = ui.available_rect_before_wrap();
@@ -175,6 +175,9 @@ impl ObjectTable {
                     self.was_editing = true;
                 }
             });
+        if self.was_editing {
+            ui.ctx().memory_mut(|m| m.request_focus(self.table_id));
+        }
         array_response
     }